开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > Internet/网络编程 > P2P编程 > 查看源码
TVSourceFilter.cpp
资源名称：p2p_vod.rar　[点击查看]
上传用户：liguizhu
上传日期：2015-11-01
资源大小：2422k
文件大小：27k
源码类别：

P2P编程

开发平台：

Visual C++

TVSourceFilter.cpp：源码内容
  1. #include "stdafx.h"
  2. #include <process.h>
  3. #include <Dmoreg.h>
  4. #include <dmodshow.h>
  5. #include <atlconv.h>
  6. #include "networkproxy.h"
  7. #include "graphrebuildhelper.h"
  8. //
  9. extern HANDLE g_hModule;
  11. HRESULT CTVStreamSource::EnumFilters (IFilterGraph *pGraph) 
  12. {
  13. IEnumFilters* pEnum  = NULL;
  14. IBaseFilter*  pFilter = NULL;;
  15. ULONG cFetched;
  17. HRESULT hr = pGraph->EnumFilters(&pEnum);
  18. if (FAILED(hr))
  19. return hr;
  20. //
  21. while(pEnum->Next(1, &pFilter, &cFetched) == S_OK)
  22. {
  23. FILTER_INFO FilterInfo;
  24. hr = pFilter->QueryFilterInfo(&FilterInfo);
  25. if (FAILED(hr))
  26. {
  27. ASSERT(FALSE);
  28. // Maybe the next one will work.
  29. continue;  
  30. }
  31. //
  32. GUID guid;
  33. HRESULT hr = pFilter->GetClassID(&guid);
  34. if (hr == S_OK)
  35. {
  36. if (CLSID_DSoundRender == guid ||
  37. CLSID_AudioRender == guid )
  38. {
  39. if (NULL != m_pAudioRender)
  40. {
  41. m_pAudioRender->Release();
  42. m_pAudioRender = NULL;
  43. }
  44. //
  45. m_pAudioRender = pFilter;
  46. pFilter->AddRef();
  47. }
  48. //
  49. if (CLSID_VideoMixingRenderer  == guid ||
  50. CLSID_VideoRendererDefault == guid||
  51. CLSID_VideoRenderer        == guid)
  52. {
  53. if (NULL != m_pVideoRender)
  54. {
  55. m_pVideoRender->Release();
  56. m_pVideoRender = NULL;
  57. }
  58. //
  59. m_pVideoRender = pFilter;
  60. pFilter->AddRef();
  61. }
  63. }
  64. // The FILTER_INFO structure holds a pointer to the Filter Graph
  65. // Manager, with a reference count that must be released.
  66. if (FilterInfo.pGraph != NULL)
  67. {
  68. FilterInfo.pGraph->Release();
  69. }
  70. pFilter->Release();
  71. }
  72. //
  73. pEnum->Release();
  74. return S_OK;
  75. }
  78. //
  79. unsigned __stdcall CTVStreamSource::ThreadFunc( void* pArguments )
  80. {
  81. ASSERT(NULL != pArguments);
  82. CTVStreamSource* pTVSS = (CTVStreamSource*)pArguments;
  83. //
  84. CoInitialize(NULL);
  85. //
  86. MSG msg;
  87. WORD m_accRequest = 0; //累积的请求数目
  88. WORD m_needRequest = 2;
  89. BOOL m_videoavaible = FALSE;
  90. BOOL m_audioavaible = FALSE;
  91. while(GetMessage(&msg, NULL, 0, 0))
  92. {
  93.         if(msg.message == MEDIA_ERROR_STOP)
  94.         {
  95.             pTVSS->m_pAudioPin->KillPin();
  96.             pTVSS->m_pVideoPin->KillPin();
  97.             IGraphConfig* pConfig = pTVSS->GetGCF();
  98.   if(!pConfig) {
  99. assert(0);
  100. continue;
  101. }
  102. IMediaControl* pIMediaControl = NULL;
  103. pConfig->QueryInterface(IID_IMediaControl, (void**)&pIMediaControl);
  105. //错误信息因此停止图。
  106. HRESULT hr = pIMediaControl->Stop();
  107.             pIMediaControl->Release();
  109.         }
  110. else if (msg.message == MEDIATYPE_CHANGED)
  111. {
  112. m_accRequest++;
  113. if(LOWORD(msg.wParam) == 0)
  114. m_audioavaible = HIWORD(msg.wParam) == 1? TRUE:FALSE;
  115. else
  116. m_videoavaible = HIWORD(msg.wParam) == 1? TRUE:FALSE;
  118.             if(m_accRequest < m_needRequest)    //如果需要等待,我们要开始一个flush操作,以避免另一个pin卡死在deliver上面！
  119.             {
  120.                 HRESULT hr;
  121.                 if(m_audioavaible == FALSE)
  122.                 {
  123.                     hr = pTVSS->m_pAudioPin->BeginFlushData();
  124.                     assert(!FAILED(hr));
  125.                 }
  126.                 if(m_videoavaible == FALSE)
  127.                 {
  128.                     hr = pTVSS->m_pVideoPin->BeginFlushData();
  129.                     assert(!FAILED(hr));
  130.                 }
  131.             }
  132.             else //累积请求数目足够，开始动态重连
  133. {
  134. m_needRequest = m_accRequest = 0;
  135. if(m_audioavaible)
  136. m_needRequest++;
  137. if(m_videoavaible)
  138. m_needRequest++;
  140.                 //flush停止
  141.                 pTVSS->m_pAudioPin->EndFlushData();
  142.                 pTVSS->m_pVideoPin->EndFlushData();
  144.                 //预先加入filter到cache里面以提高重连的成功率
  145. /*                pTVSS->AddFilterToCache();
  146.                 GraphRebuildHelper helper;
  148. //                IMediaControl* pIMediaControl = NULL;
  149. // pConfig->QueryInterface(IID_IMediaControl, (void**)&pIMediaControl);
  151.                 pTVSS->m_bDoRealStop = false;
  152.                 HRESULT hr = helper.stopgraph(pTVSS->GetFG());
  153.                 pTVSS->m_bDoRealStop = true;
  155.                 BOOL anypinOK = TRUE;
  156.                 if(m_videoavaible == TRUE)
  157.                 {
  158.                     hr = helper.startbuilding(pTVSS->GetFG(), pTVSS->m_pVideoPin, false);
  159.                     if(FAILED(hr))
  160.                     {
  161.                         char msg[100];
  162.                         wsprintf(msg, "在连接视频时的DShow错误%X", hr);
  163.                         MessageBox(NULL,msg,"错误", MB_OK);
  164.                         anypinOK = FALSE;
  165.                         pTVSS->m_pVideoPin->KillPin();
  166.                         m_needRequest--;
  167.                     }
  168.                 }
  170.                 if(m_audioavaible == TRUE)
  171.                 {
  172.                     hr = helper.startbuilding(pTVSS->GetFG(), pTVSS->m_pAudioPin, true);
  173.                     if(FAILED(hr))
  174.                     {
  175.                         char msg[100];
  176.                         wsprintf(msg, "在连接音频时的DShow错误%X", hr);
  177.                         MessageBox(NULL,msg,"错误", MB_OK);
  178.                         anypinOK = FALSE;
  179.                         pTVSS->m_pAudioPin->KillPin();
  180.                         m_needRequest--;
  181.                     }
  182.                 }*/
  183.                 
  184. IGraphConfig* pConfig = pTVSS->GetGCF();
  185. if(!pConfig) {
  186. assert(0);
  187. continue;
  188. }
  189.                 
  190.                 //先把Audio和Video Renderer Filter枚举到
  191. pTVSS->EnumFilters (pTVSS->GetFG());
  192. //
  193. IMediaControl* pIMediaControl = NULL;
  194. pConfig->QueryInterface(IID_IMediaControl, (void**)&pIMediaControl);
  196. //动态重连前，先把图stop or pause
  197. pTVSS->m_bDoRealStop = false;
  198. HRESULT hr = pIMediaControl->Stop();
  199. //                HRESULT hr = pTVSS->Stop();
  200. assert(hr == S_OK);
  201. pTVSS->m_bDoRealStop = true;
  202.                 Sleep(500);
  203. //
  204.   
  205. if (FALSE == pTVSS->m_iSAddFilterToCache)
  206. pTVSS->AddFilterToCache();
  208. BOOL anypinOK = TRUE;
  209. if(m_audioavaible == TRUE)
  210. {
  211. //
  212. IPin* pin = NULL;
  213. ULONG cFetched = 0;
  214. //
  215. IEnumPins* ppEnum = NULL;
  216. //
  218. if (S_OK == pTVSS->m_pAudioRender->EnumPins(&ppEnum))
  219. {
  220. ppEnum->Reset();
  221. ppEnum->Next(1, &pin, &cFetched);
  222. //
  223. HRESULT hr = S_FALSE;
  224. const AM_MEDIA_TYPE *pmtFirstConnection = pTVSS->m_pAudioPin->GetType();
  225. hr = pConfig->Reconnect(pTVSS->m_pAudioPin,
  226. pin, 
  227. pmtFirstConnection,
  228. NULL,
  229. NULL, 
  230. AM_GRAPH_CONFIG_RECONNECT_CACHE_REMOVED_FILTERS |
  231. AM_GRAPH_CONFIG_RECONNECT_USE_ONLY_CACHED_FILTERS);
  232. //
  234. if(FAILED(hr))
  235.                         {
  236. anypinOK = FALSE;
  237.                             char msg[100];
  238.                             wsprintf(msg, "在连接音频时的DShow错误%X", hr);
  239.                             MessageBox(NULL,msg,"错误", MB_OK);
  240.                             pTVSS->m_pAudioPin->KillPin();
  241.                             m_needRequest--;
  242.                         }
  243. }
  245. if(ppEnum)
  246. ppEnum->Release();
  247. if(pin)
  248. pin->Release();
  249. }
  251. if(m_videoavaible == TRUE)
  252. {
  253. //
  254. IPin* pin = NULL;
  255. ULONG cFetched = 0;
  256. //
  257. IEnumPins* ppEnum = NULL;
  258. //
  259. if (S_OK == pTVSS->m_pVideoRender->EnumPins(&ppEnum))
  260. {
  261. ppEnum->Reset();
  262. if (S_OK == ppEnum->Next(1, &pin, &cFetched))
  263. {
  264. hr = S_FALSE;
  265. const AM_MEDIA_TYPE *pmtFirstConnection = pTVSS->m_pVideoPin->GetType();
  266. hr= pConfig->Reconnect((IPin*)(pTVSS->m_pVideoPin),
  267. pin,
  268. pmtFirstConnection,
  269. NULL,
  270. NULL,
  271. AM_GRAPH_CONFIG_RECONNECT_CACHE_REMOVED_FILTERS |
  272. AM_GRAPH_CONFIG_RECONNECT_USE_ONLY_CACHED_FILTERS);
  274. if(FAILED(hr))
  275.                             {
  276. anypinOK = FALSE;
  277.                                 char msg[100];
  278.                                 wsprintf(msg, "在连接音频时的DShow错误%X", hr);
  279.                                 MessageBox(NULL,msg,"错误", MB_OK);
  280.                                 pTVSS->m_pAudioPin->KillPin();
  281.                                 m_needRequest--;
  282.                             }
  283. }
  284. }
  286. if(ppEnum)
  287. ppEnum->Release();
  288. if(pin)
  289. pin->Release();
  290. }
  291.                 
  293. m_audioavaible = m_videoavaible = FALSE;
  294. /* if (!anypinOK)
  295. {
  296. MessageBox(NULL, "由于一个DShow错误，播放将无法继续，需要停止后重新播放","错误", MB_OK);
  297. }*/
  299. //                hr = helper.rungraph(pTVSS->GetFG());
  300.                 hr = pIMediaControl->Run();
  301.                 if(FAILED(hr))
  302.                 {
  303.                     char msg[100];
  304.                     wsprintf(msg, "由于DShow错误%X, 无法重新运行", hr);
  305.                     MessageBox(NULL,msg,"错误", MB_OK);
  306.                 }
  307. //                CRefTime rf;
  308. //                pTVSS->Run(rf.GetUnits());
  310. if(pIMediaControl)
  311. pIMediaControl->Release();
  312. if(pConfig)
  313. pConfig->Release();
  314. }
  315. }
  316. //
  317. TranslateMessage(&msg);
  318. DispatchMessage(&msg);
  319. }
  320. //
  321. //
  322. CoUninitialize();
  323. //
  324. // _endthreadex( 0 );
  325. //
  326. return 0;
  327. }
  328. // filter info
  329. const AMOVIESETUP_MEDIATYPE sudPinTypes =
  330. {
  331.     &MEDIATYPE_Video,            // Major type
  332.     &MEDIASUBTYPE_NULL           // Minor type
  333. };
  335. const AMOVIESETUP_PIN sudPins =
  336. {
  337.     L"Video",                  // Pin string name
  338.     FALSE,                      // Is it rendered
  339.     TRUE,                       // Is it an output
  340.     FALSE,                      // Allowed none
  341.     FALSE,                      // Likewise many
  342.     &CLSID_NULL,                // Connects to filters
  343.     NULL,                       // Connects to pin
  344.     1,                          // Number of types
  345.     &sudPinTypes                // Pin information
  346. };
  348. const AMOVIESETUP_FILTER sudTVStreamSource =
  349. {
  350.     &CLSID_TVStreamSource,    // clsID
  351.     L"Chaos Filter",       // strName
  352.     MERIT_DO_NOT_USE,         // dwMerit
  353.     1,                        // nPins
  354.     &sudPins                  // lpPin
  355. };
  357. // Global data
  358. CFactoryTemplate g_Templates[] = 
  359. {
  360.     { L"Chaos Filter", &CLSID_TVStreamSource, 
  361.   CTVStreamSource::CreateInstance, NULL,
  362.   &sudTVStreamSource },
  363. };
  365. int g_cTemplates = sizeof(g_Templates) / sizeof(g_Templates[0]);
  367. CTVStreamSource* g_pfilter = NULL;
  368. //
  369. CTVStreamSource::CTVStreamSource(LPUNKNOWN pUnk, HRESULT *phr)
  370. : CSource(NAME("Chaos filter"), pUnk, CLSID_TVStreamSource),
  371.   m_pComm(NULL), m_pFileReader(NULL), 
  372.   m_pFileName(NULL), m_isAudioOnly(FALSE), m_bCanDynamicReconnect(FALSE),
  373.   m_bUpdateComplete(FALSE), m_pVideoPin(NULL), m_pAudioPin(NULL)
  374. {
  375. m_pAudioRender = NULL;
  376. m_pVideoRender = NULL;
  378. // Try calling GetVersionEx using the OSVERSIONINFOEX structure.
  379. // If that fails, try using the OSVERSIONINFO structure.
  380. ZeroMemory(&osvi, sizeof(OSVERSIONINFOEX));
  381. osvi.dwOSVersionInfoSize = sizeof(OSVERSIONINFOEX);
  382. if(!GetVersionEx ((OSVERSIONINFO *) &osvi)) {
  383. osvi.dwOSVersionInfoSize = sizeof (OSVERSIONINFO);
  384. if(!GetVersionEx ((OSVERSIONINFO*) &osvi)) {
  385. assert(0);
  386. }
  387. }
  389. //
  390. g_pfilter = this;
  391. //
  392. m_hThread = NULL;
  393. m_iSAddFilterToCache = FALSE;
  394. //
  395. //
  396. SetLastError(0);
  397. m_bDoRealStop = true;
  398. }
  399. //
  400. CTVStreamSource::~CTVStreamSource()
  401. {
  402. if(m_hThread != NULL)
  403. {
  404. PostThreadMessage(m_iThreadID, WM_QUIT, 0, NULL);
  405. if (WAIT_TIMEOUT == WaitForSingleObject(m_hThread, 5000))
  406. {
  407. TerminateThread(m_hThread, 0);
  408. }
  409. }
  410. //
  411. if (NULL != m_pAudioRender)
  412. {
  413. m_pAudioRender->Release();
  414. m_pAudioRender = NULL;
  415. }
  416. //
  417. if (NULL != m_pVideoRender)
  418. {
  419. m_pVideoRender->Release();
  420. m_pVideoRender = NULL;
  421. }
  422. //
  423. SAFE_DELETE(m_pVideoPin);
  424. SAFE_DELETE(m_pAudioPin);
  425. //
  426. if(NULL != m_pComm)
  427. m_pComm->Stop();
  428. //
  429. g_pfilter = NULL;
  430. //
  431. SAFE_DELETE(m_pComm);
  432. SAFE_DELETE(m_pFileReader);
  433. //
  434. SAFE_ARRAYDELETE(m_pFileName);
  436. }
  437. //
  438. STDMETHODIMP CTVStreamSource::JoinFilterGraph(IFilterGraph *pGraph, LPCWSTR pName)
  439. {
  440. CBaseFilter::JoinFilterGraph(pGraph, pName);
  442. if(pGraph != NULL)
  443. {
  444. IGraphConfig* pGCF = GetGCF();
  445. if(pGCF != NULL)
  446. {
  447. m_bCanDynamicReconnect = TRUE;
  448. pGCF->Release();
  449. }
  450. }
  451.     if(m_bCanDynamicReconnect == FALSE)
  452.     {
  453.         MessageBox(NULL, "DirectX版本太低, 请安装9.0以上的版本", "错误", MB_OK);
  454.         return E_FAIL;
  455.     }
  457. // m_bCanDynamicReconnect = FALSE;
  459.     SAFE_DELETE(m_pVideoPin);
  460. SAFE_DELETE(m_pAudioPin);
  461. //
  462. HRESULT hr;
  463. HRESULT* phr = &hr;
  464. m_pVideoPin = new CTVStreamSourceVideoPin(phr, this);
  465.     if (phr)
  466. {
  467. *phr = (NULL == m_pVideoPin) ? E_OUTOFMEMORY : S_OK;
  468. }
  469. //
  470. m_pAudioPin = new CTVStreamSourceAudioPin(phr, this);
  471. //
  472.     if (phr)
  473. {
  474. *phr = (NULL == m_pAudioPin) ? E_OUTOFMEMORY : S_OK;
  475. }
  476. //
  477. m_pAudioPin->SetAnotherPin(m_pVideoPin);
  478. m_pVideoPin->SetAnotherPin(m_pAudioPin);
  480. if (NULL != pGraph && NULL == m_hThread)
  481. {
  482. /*
  483. IMediaFilter* pIMediaFilter = NULL;
  484. //
  485. pGraph->QueryInterface(IID_IMediaFilter, (void**)&pIMediaFilter);
  486. pIMediaFilter->SetSyncSource(NULL);
  487. pIMediaFilter->Release();*/
  489. // AddFilterToCache();
  490. //create thread
  491. m_hThread = (HANDLE) _beginthreadex( NULL, 0, &ThreadFunc, this, 0, &m_iThreadID );
  492. }
  493. //
  494. return S_OK;
  495. }
  496. //
  497. IBaseFilter* CTVStreamSource::CreateDMO( REFCLSID clsidDMO, REFCLSID catDMO)
  498. {
  499. // Create the DMO Wrapper filter.
  500. IBaseFilter *pFilter = NULL;
  501. HRESULT hr = CoCreateInstance(CLSID_DMOWrapperFilter, 
  502.   NULL, 
  503.   CLSCTX_INPROC_SERVER,
  504.   IID_IBaseFilter, 
  505.   reinterpret_cast<void**>(&pFilter));
  506. //
  508. if (SUCCEEDED(hr)) 
  509. {
  510. // Query for IDMOWrapperFilter.
  511. IDMOWrapperFilter *pDmoWrapper = NULL;
  512. hr = pFilter->QueryInterface(IID_IDMOWrapperFilter, 
  513. reinterpret_cast<void**>(&pDmoWrapper));
  515. if (SUCCEEDED(hr)) 
  516. {     
  517. // Initialize the filter.
  518. hr = pDmoWrapper->Init(clsidDMO, catDMO); 
  519. pDmoWrapper->Release();
  521. if (SUCCEEDED(hr)) 
  522. {
  523. return pFilter;
  524. }
  525. }
  527. }
  528. //
  529. return NULL;
  530. }
  531. //
  532. void CTVStreamSource::AddFilterToCache()
  533. {
  534. m_iSAddFilterToCache = TRUE;
  535. IGraphConfig* pConfig = GetGCF();
  536. if(!pConfig)
  537. return;
  538. //vp7: {083863F1-70DE-11D0-BD40-00A0C911CE86}{C204438D-6E1A-4309-B09C-0C0F749863AF}
  539. GUID rclsid_vp7 = {0xc204438d, 0x6e1a, 0x4309, {0xb0, 0x9c, 0x0c, 0x0f, 0x74, 0x98, 0x63, 0xaf}};
  540. IBaseFilter* pIBaseFilter = NULL;
  541. if (S_OK == CoCreateInstance(rclsid_vp7, 
  542.  NULL,
  543.  CLSCTX_INPROC_SERVER,
  544.  IID_IBaseFilter,
  545.  (void**)&pIBaseFilter))
  546. {
  547. pConfig->AddFilterToCache(pIBaseFilter);
  548. pIBaseFilter->Release();
  549. pIBaseFilter = NULL;
  550. }
  551. else
  552. assert(0);
  553. // WMVideo Decoder DMO: {82D353DF-90BD-4382-8BC2-3F6192B76E34}{4A69B442-28BE-4991-969C-B500ADF5D8A8}
  554. GUID rclsid_wmv = {0x82D353DF, 0x90BD, 0x4382, {0x8B, 0xC2, 0x3F, 0x61, 0x92, 0xB7, 0x6E, 0x34}};
  555. pIBaseFilter = CreateDMO( rclsid_wmv, DMOCATEGORY_VIDEO_DECODER);
  556. if (NULL != pIBaseFilter)
  557. {
  558. pConfig->AddFilterToCache(pIBaseFilter);
  559. pIBaseFilter->Release();
  560. pIBaseFilter = NULL;
  561. }
  562. else
  563. assert(0);
  565. // WMAudio Decoder DMO: {2EEB4ADF-4578-4D10-BCA7-BB955F56320A}{57F2DB8B-E6BB-4513-9D43-DCD2A6593125}
  566. GUID rclsid_wma = {0x2EEB4ADF, 0x4578, 0x4D10, {0xBC, 0xA7, 0xBB, 0x95, 0x5F, 0x56, 0x32, 0x0A}};
  567. pIBaseFilter = CreateDMO( rclsid_wma, DMOCATEGORY_AUDIO_DECODER);
  568. if (NULL != pIBaseFilter)
  569. {
  570. pConfig->AddFilterToCache(pIBaseFilter);
  571. pIBaseFilter->Release();
  572. pIBaseFilter = NULL;
  573. }
  574. else
  575. assert(0);
  576. /*
  577. //gsm{083863F1-70DE-11D0-BD40-00A0C911CE86}{6A08CF80-0E18-11CF-A24D-0020AFD79767}
  578. GUID rclsid_gsm = {0x6A08CF80, 0x0E18, 0x11CF, {0xA2, 0x4D, 0x00, 0x20, 0xAF, 0xD7, 0x97, 0x67}};
  579. //
  580. pIBaseFilter = NULL;
  581. if (S_OK == CoCreateInstance(rclsid_gsm, 
  582.  NULL,
  583.  CLSCTX_INPROC_SERVER,
  584.  IID_IBaseFilter,
  585.  (void**)&pIBaseFilter))
  586. {
  587. pConfig->AddFilterToCache(pIBaseFilter);
  588. pIBaseFilter->Release();
  589. pIBaseFilter = NULL;
  590. }
  591. else
  592. assert(0);
  593. */
  594. // 直接无法创建，只能从现有的filter中寻找（所以默认的音频必须是GSM，这样才能找到）
  595. // gsm
  596. pIBaseFilter = FindFilterFromFriendlyName(CLSID_AudioCompressorCategory, "GSM 6.10");
  597.     if (NULL != pIBaseFilter) 
  598. {
  599. pConfig->AddFilterToCache(pIBaseFilter);
  600. pIBaseFilter->Release();
  601. pIBaseFilter = NULL;
  602. }
  603. else
  604. assert(0);
  605. //
  606. pConfig->Release();
  607. }
  608. //
  609. IFilterGraph* CTVStreamSource::GetFG()
  610. {
  611. return m_pGraph;
  612. }
  613. //
  614. IGraphConfig* CTVStreamSource::GetGCF()
  615. {
  616. IGraphConfig* pConfig = NULL;
  617. GetFG()->QueryInterface(IID_IGraphConfig, (void**)&pConfig);
  618. //
  619. return pConfig;
  620. }
  621. //
  622. bool CTVStreamSource::UpdateAndStartLocalServer()
  623. {
  624. // 1. get module path & system path
  625. TCHAR lpModulePath[MAX_PATH_EX] = _T("");
  626. DWORD dwLen = GetModuleFileName((HMODULE)g_hModule, lpModulePath, MAX_PATH_EX);
  627. if(dwLen == 0)
  628. return false;
  629. TCHAR* idx = _tcsrchr(lpModulePath, _T('\'));
  630. if(idx == NULL)
  631. return false;
  632. *(idx+1) = _T('');
  634. TCHAR lpSystemPath[MAX_PATH_EX] = _T("");
  635. dwLen = ::GetSystemDirectory(lpSystemPath, MAX_PATH_EX);
  636. if(dwLen == 0)
  637. return false;
  638. _tcscat(lpSystemPath, "\");
  640. // 2. check existance of "layer1.dll" "localsrv.exe" "vp7dec.ax"
  641. bool bForceUpdate = false;
  642. if(!DoesFileExist(lpModulePath, _T("layer1.dll")))
  643. bForceUpdate = true;
  644. if(!DoesFileExist(lpModulePath, _T("localsrv.exe")))
  645. bForceUpdate = true;
  646. if(!DoesFileExist(lpSystemPath, _T("vp7dec.ax")))
  647. bForceUpdate = true;
  649. // 3. 如果需要强制升级,则不能失败,而且要显示升级界面;
  650. //    如果不需要,则允许升级失败,且不用显示升级界面
  651. STARTUPINFO si;
  652. PROCESS_INFORMATION pi;
  654. ZeroMemory( &si, sizeof(si) );
  655. si.cb = sizeof(si);
  656. ZeroMemory( &pi, sizeof(pi) );
  658. string filename = lpModulePath;
  659. filename.append(_T("xtvupdater.exe"));
  660. if( CreateProcess(  // No module name (use command line). 
  661. filename.data(), 
  662. bForceUpdate?NULL:_T(" /S "), // 如果不需要强制升级,则使用/S参数避免显示升级界面
  663. NULL,             // Process handle not inheritable. 
  664. NULL,             // Thread handle not inheritable. 
  665. FALSE,            // Set handle inheritance to FALSE. 
  666. 0,                // No creation flags. 
  667. NULL,             // Use parent's environment block. 
  668. NULL,             // Use parent's starting directory. 
  669. &si,              // Pointer to STARTUPINFO structure.
  670. &pi )             // Pointer to PROCESS_INFORMATION structure.
  671. ) 
  672. {
  673. // Wait until child process exits.
  674. WaitForSingleObject( pi.hProcess, INFINITE );
  676. // Close process and thread handles. 
  677. CloseHandle( pi.hProcess );
  678. CloseHandle( pi.hThread );
  679. }
  680. else if(bForceUpdate) // 强制升级的话，这里可不能失败
  681. return false;
  683. // 4. 如果强制升级，那么升级成功了吗？
  684. if(bForceUpdate)
  685. {
  686. if(!DoesFileExist(lpModulePath, _T("layer1.dll")))
  687. return false;
  688. if(!DoesFileExist(lpModulePath, _T("localsrv.exe")))
  689. return false;
  690. if(!DoesFileExist(lpSystemPath, _T("vp7dec.ax")))
  691. return false;
  692. }
  694. // 5. 启动LocalServer，它会自动保证同时只存在一个实例的
  695. filename = lpModulePath;
  696. filename.append(_T("localsrv.exe"));
  697. // Start the child process. 
  698. if( !CreateProcess(  // No module name (use command line). 
  699. filename.data(), 
  700. NULL,             // Command line. 
  701. NULL,             // Process handle not inheritable. 
  702. NULL,             // Thread handle not inheritable. 
  703. FALSE,            // Set handle inheritance to FALSE. 
  704. 0,                // No creation flags. 
  705. NULL,             // Use parent's environment block. 
  706. NULL,             // Use parent's starting directory. 
  707. &si,              // Pointer to STARTUPINFO structure.
  708. &pi )             // Pointer to PROCESS_INFORMATION structure.
  709. ) 
  710. {
  711. // 启动失败可不行
  712.         //MessageBox(NULL, "启动localsrv失败，也许直接执行它可以解决问题", "错误", MB_OK);
  713. return false;
  714. }
  715. // Close process and thread handles. 
  716. CloseHandle( pi.hProcess );
  717. CloseHandle( pi.hThread );
  719. return true;
  720. }
  722. bool CTVStreamSource::DoesFileExist(LPCTSTR path, LPCTSTR filename)
  723. {
  724. if(!path || !filename)
  725. return false;
  727. // 查找文件是否存在
  728. WIN32_FIND_DATA fileData;
  730. string match = path;
  731. match.append(filename);
  732. HANDLE hFind = FindFirstFile(match.data(), &fileData);
  733. if(hFind == INVALID_HANDLE_VALUE)
  734. return false;
  735. return true;
  736. }
  738. CUnknown * WINAPI CTVStreamSource::CreateInstance(LPUNKNOWN pUnk, HRESULT * phr)
  739. {
  740.     CTVStreamSource* pNewFilter = new CTVStreamSource(pUnk, phr);
  741. //
  742.     if (phr)
  743. {
  744. *phr = (NULL==pNewFilter) ? E_OUTOFMEMORY : S_OK;
  745. }
  746. //
  747.     return pNewFilter;
  748. }
  749. //
  750. HRESULT CTVStreamSource::_LoadFile(const char* filename)
  751. {
  752. ASSERT(strlen(filename) > 0);
  753. //
  754. SAFE_DELETE(m_pFileReader);
  755. //
  756. m_pFileReader = new ZZLFileReader();
  757. bool ret = m_pFileReader->Init(filename);
  758. //
  759. if(!ret) 
  760. {
  761. MessageBox(NULL, _T("zzl文件格式错误，无法播放！"), _T("致命错误"), MB_OK|MB_ICONSTOP);
  762. return VFW_E_CANNOT_RENDER;
  763. }
  764. //
  765. m_pVideoPin->SetMediaTime();
  766. m_pAudioPin->SetMediaTime();
  767. //
  768. return S_OK;
  770. }
  771. //
  772. HRESULT CTVStreamSource::_LoadURL(char* strURL)
  773. {
  774. //
  775. m_pComm = new NetworkProxy();
  776. //
  777. bool ret = false;
  778. if (NULL != m_pComm)
  779. {
  780. ret = m_pComm->Request(strURL);
  781. }
  782. //
  783. if(!ret) 
  784. {
  785. return VFW_E_CANNOT_RENDER;
  786. }
  787. //
  788. m_pVideoPin->SetMediaTime();
  789. m_pAudioPin->SetMediaTime(); 
  790. // 
  791. return S_OK;
  792. }
  793. //
  794. STDMETHODIMP CTVStreamSource::Load(LPCOLESTR lpwszFileName, const AM_MEDIA_TYPE *pmt) 
  795. {
  796. if(!m_bUpdateComplete)
  797. {
  798. if(!UpdateAndStartLocalServer())
  799. {
  800. // 1. get module path & system path
  801. TCHAR lpModulePath[MAX_PATH_EX] = _T("");
  802. DWORD dwLen = GetModuleFileName((HMODULE)g_hModule, lpModulePath, MAX_PATH_EX);
  803. if(dwLen == 0)
  804. return false;
  805. TCHAR* idx = _tcsrchr(lpModulePath, _T('\'));
  806. if(idx == NULL)
  807. return false;
  808. *(idx+1) = _T('');
  810. TCHAR lpSystemPath[MAX_PATH_EX] = _T("");
  811. dwLen = ::GetSystemDirectory(lpSystemPath, MAX_PATH_EX);
  812. if(dwLen == 0)
  813. return false;
  814. _tcscat(lpSystemPath, "\");
  816. if(!DoesFileExist(lpModulePath, _T("layer1.dll")) || !DoesFileExist(lpSystemPath, _T("vp7dec.ax"))) {
  817. MessageBox(NULL, _T("自动下载失败！请到www.mysee.com下载并进行安装！"), _T("错误"), MB_OK|MB_ICONSTOP);
  818. return VFW_E_CANNOT_RENDER;
  819. }
  820. }
  821. m_bUpdateComplete = TRUE;
  822. }
  824. TCHAR szFinal[MAX_PATH_EX] = _T(""); 
  825. //
  826. if(lpwszFileName) 
  827. {
  828. // 将WCHAR转换成TCHAR
  829. SAFE_ARRAYDELETE(m_pFileName);
  830. m_pFileName = new WCHAR[MAX_PATH_EX];
  831. //
  832. wcscpy(m_pFileName, lpwszFileName);
  833. }
  835. if(0 == WideCharToMultiByte(CP_ACP, WC_COMPOSITECHECK, m_pFileName, -1, szFinal, sizeof(szFinal), NULL, NULL)) 
  836. {
  837. MessageBox(NULL, _T("错误的URL，无法播放!"), _T("致命错误"), MB_OK|MB_ICONSTOP);
  838. return VFW_E_CANNOT_RENDER;
  839. }
  841. // gaov protocol
  842. if(strnicmp(szFinal, "gaov://", 7) == 0)
  843.     {
  844.         return _LoadURL(szFinal);
  845.     }
  847. //
  848. char* index = strrchr(szFinal, '.');
  849. if(index != NULL)
  850. {
  851. index++;
  852. if(strnicmp(index, "gtv", 3) == 0) 
  853. {
  854. //ASSERT(FALSE);
  855. // gtv file
  856. }else if(strnicmp(index, "zzl", 3) == 0)
  857. {
  858. return _LoadFile(szFinal);
  859. }
  860. else 
  861. {
  862. index = NULL;
  863. }
  864. }
  865. //
  866. if(index == NULL) 
  867. {
  868. MessageBox(NULL, _T("无法识别的URL，后缀名必是gtv或者zzl!"), _T("致命错误"), MB_OK|MB_ICONSTOP);
  869. return VFW_E_CANNOT_RENDER;
  870. }
  871. //
  872. return _LoadURL(szFinal);
  873. }
  875. STDMETHODIMP CTVStreamSource::GetCurFile(LPOLESTR *ppszFileName, AM_MEDIA_TYPE *pmt)
  876. {
  877. CheckPointer(ppszFileName, E_POINTER);
  878. *ppszFileName = NULL;
  880. if(m_pFileName != NULL) 
  881. {
  882. *ppszFileName = (LPOLESTR) CoTaskMemAlloc(MAX_PATH_EX);
  883. if (*ppszFileName != NULL)
  884. {
  885. CopyMemory(*ppszFileName, m_pFileName, MAX_PATH_EX);
  886. }
  887. }
  889. if(pmt != NULL) 
  890. {
  891. // TODO
  892. NOTE("GetCurFile TODO");
  893. }
  894. return S_OK;
  895. }
  897. STDMETHODIMP CTVStreamSource::NonDelegatingQueryInterface(REFIID riid, void ** ppv) 
  898. {
  899. CheckPointer(ppv,E_POINTER);
  901. if (riid == IID_IFileSourceFilter) 
  902. {
  903. return GetInterface((IFileSourceFilter *) this, ppv);
  904. }
  905. else if(riid == IID_ISettings)
  906. {
  907. return GetInterface((ISettings *) this, ppv);
  908. }
  909. else
  910. {
  911. return CSource::NonDelegatingQueryInterface(riid, ppv);
  912. }
  913. }
  914. //
  915. STDMETHODIMP_(void) CTVStreamSource::SetPopMsgHWND(HWND pop) 
  916. {
  917. }
  918. //
  919. STDMETHODIMP CTVStreamSource::Stop()   
  920. {
  921. CAutoLock cObjectLock(m_pLock);
  923. if(m_bDoRealStop) {
  925. if(NULL != m_pComm)
  926. m_pComm->Stop();
  927. }
  928. //
  929. HRESULT hr = CSource::Stop();
  930. //
  931. return hr;
  932. }
  933. //
  934. /*
  935. STDMETHODIMP CTVStreamSource::Run(REFERENCE_TIME tStart) 
  936. {
  937. CAutoLock cObjectLock(m_pLock);
  938. //
  939. return CSource::Run(tStart);
  940. }
  942. STDMETHODIMP CTVStreamSource::Pause() 
  943. {
  944. CAutoLock cObjectLock(m_pLock); 
  945. //SetBaseRef(_I64_MAX);
  946. //
  947. return CSource::Pause();
  948. }*/
  951. //
  952. LONGLONG CTVStreamSource::GetBaseRef()
  953. {
  954. CAutoLock l(&m_critllBase);
  955. return m_llBaseRef;
  956. }
  957. //
  958. void CTVStreamSource::SetBaseRef( LONGLONG llBaseRef)
  959. {
  960. CAutoLock l(&m_critllBase);
  961. m_llBaseRef = llBaseRef;
  963. }
  966. LONGLONG CTVStreamSource::GetRawAudioRef()
  967. {
  968. CAutoLock l(&m_critllBase);
  969. return m_rawAudiotime;
  970. }
  972. void CTVStreamSource::SetRawAudioRef(LONGLONG llRawRef)
  973. {
  974. CAutoLock l(&m_critllBase);
  975. m_rawAudiotime = llRawRef;
  976. }
  979. //
  980. IBaseFilter* CTVStreamSource::FindFilterFromFriendlyName(REFCLSID clsid, string str)
  981. {
  982. // Create the System Device Enumerator.
  983. HRESULT hr;
  984. IBaseFilter *pFilter = NULL;
  985. ICreateDevEnum* pSysDevEnum = NULL;
  986. hr = CoCreateInstance(CLSID_SystemDeviceEnum, NULL, CLSCTX_INPROC_SERVER,
  987. IID_ICreateDevEnum, (void **)&pSysDevEnum);
  989. // Obtain a class enumerator for the video compressor category.
  990. IEnumMoniker* pEnumCat = NULL;
  991. hr = pSysDevEnum->CreateClassEnumerator(clsid, &pEnumCat, 0);
  993. if (hr == S_OK) 
  994. {
  995. // Enumerate the monikers.
  996. IMoniker* pMoniker = NULL;
  997. ULONG cFetched;
  998. while(pEnumCat->Next(1, &pMoniker, &cFetched) == S_OK)
  999. {
  1000. IPropertyBag* pPropBag = NULL;
  1001. hr = pMoniker->BindToStorage(0, 0, IID_IPropertyBag, (void **)&pPropBag);
  1002. if(hr != S_OK)
  1003. break;
  1005. // To retrieve the friendly name of the filter, do the following:
  1006. VARIANT varName;
  1007. VariantInit(&varName);
  1008. hr = pPropBag->Read(L"FriendlyName", &varName, 0);
  1009. if(pPropBag)
  1010. pPropBag->Release();
  1011. if (SUCCEEDED(hr))
  1012. {
  1013. // Display the name in your UI somehow.
  1014. USES_CONVERSION;
  1016. //string str = W2CA(varName.bstrVal);
  1017. //const char* pStr = str.data();
  1018. if (stricmp(W2CA(varName.bstrVal), str.data()) == 0)
  1019. {
  1020. //ASSERT(FALSE);
  1021. VariantClear(&varName);
  1022. // To create an instance of the filter, do the following:
  1023. pMoniker->BindToObject(NULL, NULL, IID_IBaseFilter, (void**)&pFilter);
  1024. pFilter->AddRef();
  1026. pMoniker->Release();
  1027. pMoniker = NULL;
  1028. break;
  1029. }
  1030. }
  1031. //
  1032. VariantClear(&varName);
  1033. if(pMoniker) {
  1034. pMoniker->Release();
  1035. pMoniker = NULL;
  1036. }
  1037. }//end while
  1039. }
  1040. if(pSysDevEnum)
  1041. pSysDevEnum->Release();
  1042. if(pEnumCat)
  1043. pEnumCat->Release();
  1045. if(!pFilter)
  1046. ASSERT(FALSE);
  1048. return pFilter;
  1049. }
  1050. //