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

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

P2P编程

开发平台：

Visual C++

ChaosFilter.cpp：源码内容
  1. // chaosfilter.cpp : 定义 DLL 应用程序的入口点。
  2. //
  4. #include "stdafx.h"
  5. #include "chaosfilter.h"
  6. extern "C" BOOL WINAPI DllEntryPoint(HINSTANCE, ULONG, LPVOID);
  7. HANDLE g_hModule = NULL;
  8. BOOL APIENTRY DllMain( HANDLE hModule, 
  9.                        DWORD  ul_reason_for_call, 
  10.                        LPVOID lpReserved
  11.  )
  12. {
  13. switch (ul_reason_for_call)
  14. {
  15. case DLL_PROCESS_ATTACH:
  16. case DLL_THREAD_ATTACH:
  17. case DLL_THREAD_DETACH:
  18. case DLL_PROCESS_DETACH:
  19. break;
  20. }
  22. g_hModule = hModule;
  23.     return DllEntryPoint((HINSTANCE)hModule, ul_reason_for_call, lpReserved);;
  25. }
  26. //
  27. ////////////////////////////////////////////////////////////////////////
  28. //
  29. // Exported entry points for registration and unregistration 
  30. // (in this case they only call through to default implementations).
  31. //
  32. ////////////////////////////////////////////////////////////////////////
  33. namespace {
  35. BOOL GetRegString(HKEY hkey, LPCTSTR regToBeGet, LPCTSTR valueName, LPTSTR pdata)
  36. {
  37. HKEY hkSub;
  38. DWORD dis;
  40. if(RegCreateKeyEx(hkey, regToBeGet, 0, _T(""), 0, 
  41. KEY_READ, NULL, &hkSub, &dis) != ERROR_SUCCESS)
  42. return FALSE;
  44. DWORD len = 0;
  46. if(RegQueryValueEx(hkSub, valueName, 0, NULL, NULL, &len) != ERROR_SUCCESS)
  47. return FALSE;
  49. if(RegQueryValueEx(hkSub, valueName, 0, NULL, (BYTE*)pdata, &len) != ERROR_SUCCESS)
  50. return FALSE;
  52. RegCloseKey(hkSub);
  53. return TRUE;
  54. }
  56. BOOL SetRegEx(HKEY hkey, LPCTSTR regToBeSet, LPCTSTR valueName, BYTE* pdata, DWORD type, UINT size)
  57. {
  58. HKEY hkSub;
  59. DWORD dis;
  61. if(RegCreateKeyEx(hkey, regToBeSet, 0, _T(""), 0, 
  62. KEY_WRITE, NULL, &hkSub, &dis) != ERROR_SUCCESS)
  63. return FALSE;
  65. if(RegSetValueEx(hkSub, valueName, 0, type, pdata, size) != ERROR_SUCCESS)
  66. return FALSE;
  68. RegCloseKey(hkSub);
  69. return TRUE;
  70. }
  72. BOOL SetRegString(HKEY hkey, LPCTSTR regToBeSet, LPCTSTR valueName, LPCTSTR data)
  73. {
  74. return SetRegEx(hkey, regToBeSet, valueName, (BYTE*) data, REG_SZ, (UINT) _tcslen(data));
  76. }
  78. BOOL SetRegDWord(HKEY hkey, LPCTSTR regToBeSet, LPCTSTR valueName, DWORD word)
  79. {
  80. return SetRegEx(hkey, regToBeSet, valueName, (BYTE*) &word, REG_DWORD, sizeof(DWORD));
  81. }
  83. BOOL SetRegBin(HKEY hkey, LPCTSTR regToBeSet, LPCTSTR valueName, BYTE* data, UINT size)
  84. {
  85. return SetRegEx(hkey, regToBeSet, valueName, data, REG_BINARY, size);
  86. }
  88. BOOL CopyRegString(HKEY hkey, LPCTSTR regToBeCopy, LPCTSTR regCopyTo, LPCTSTR valueName) {
  90. LPTSTR temp = new TCHAR[512];
  91. memset(temp, 0, 512*sizeof(TCHAR));
  93. if(!GetRegString(hkey, regToBeCopy, valueName, temp))
  94. {
  95. delete[] temp;
  96. return FALSE;
  97. }
  99. if(!SetRegString(hkey, regCopyTo, valueName, temp))
  100. {
  101. delete[] temp;
  102. return FALSE;
  103. }
  105. delete[] temp;
  106. return TRUE;
  107. }
  109. }
  111. STDAPI DllRegisterServer()
  112. {
  113. SetRegString(HKEY_CLASSES_ROOT, _T("Media Type\Extensions\.gtv"), _T("Source Filter"), Chaos_GUID);
  114. SetRegString(HKEY_CLASSES_ROOT, _T("HTTP\Extensions"), _T(".gtv"), Chaos_GUID);
  116. SetRegString(HKEY_CLASSES_ROOT, _T(".gtv"), NULL, _T("gtv_auto_file"));
  117. // SetRegString(HKEY_CLASSES_ROOT, _T(".gtv"), _T("PerceivedType"), _T("video")); //和WMP8.0冲突。去掉
  119. DWORD nUrlListCount = 65536;
  120. SetRegBin(HKEY_CLASSES_ROOT, _T("gtv_auto_file"), _T("EditFlags"), (BYTE*) &nUrlListCount, sizeof(DWORD));
  121. SetRegBin(HKEY_LOCAL_MACHINE, _T("SOFTWARE\Classes\gtv_auto_file"), _T("EditFlags"), (BYTE*) &nUrlListCount, sizeof(DWORD));
  123. LPTSTR playeropencommand = new TCHAR[MAX_PATH + 100];
  124. LPTSTR playerplaycommand = new TCHAR[MAX_PATH + 100];
  125. memset(playeropencommand, 0, (MAX_PATH+100)*sizeof(TCHAR));
  126. memset(playerplaycommand, 0, (MAX_PATH+100)*sizeof(TCHAR));
  127. playeropencommand[0] = _T('"');
  128. LPTSTR bufHead = playeropencommand + 1;
  130. BOOL wmplayerOK = FALSE;
  131. //使用WMP8.0/9.0
  132. if(GetRegString(HKEY_LOCAL_MACHINE, _T("SOFTWARE\Microsoft\Multimedia\WMPlayer"), _T("Player.Path"), bufHead))
  133. wmplayerOK = TRUE;
  134. //使用MPlayer2
  135. else if(GetRegString(HKEY_LOCAL_MACHINE, _T("SOFTWARE\Microsoft\Multimedia\MPlayer2"), _T("Player.Path"), bufHead))
  136. wmplayerOK = FALSE;
  137. else
  138. ;
  140. playeropencommand[_tcslen(playeropencommand)] = _T('"');
  141. _tcscpy(playerplaycommand, playeropencommand);
  143. //组合出open和play命令行
  144. TCHAR tmp[100];
  145. memset(tmp, 0, 100*sizeof(TCHAR));
  146. if(wmplayerOK)
  147. GetRegString(HKEY_LOCAL_MACHINE, _T("SOFTWARE\Microsoft\Multimedia\WMPlayer"), _T("Shell.Open"), tmp);
  148. else
  149. GetRegString(HKEY_LOCAL_MACHINE, _T("SOFTWARE\Microsoft\Multimedia\MPlayer2"), _T("Shell.Open"), tmp);
  151. size_t len = strlen(playeropencommand);
  152. LPTSTR appendpos = playeropencommand + len;
  153. _tcscpy(appendpos, tmp);
  155. memset(tmp, 0, 100*sizeof(TCHAR));
  156. if(wmplayerOK)
  157. GetRegString(HKEY_LOCAL_MACHINE, _T("SOFTWARE\Microsoft\Multimedia\WMPlayer"), _T("Shell.Play"), tmp);
  158. else
  159. GetRegString(HKEY_LOCAL_MACHINE, _T("SOFTWARE\Microsoft\Multimedia\MPlayer2"), _T("Shell.Play"), tmp);
  161. appendpos = playerplaycommand + len;
  162. _tcscpy(appendpos, tmp);
  164. SetRegString(HKEY_CLASSES_ROOT, _T("gtv_auto_file\shell\open\command"), _T(""), playeropencommand);
  165. SetRegString(HKEY_CLASSES_ROOT, _T("gtv_auto_file\shell\play\command"), _T(""), playerplaycommand);
  167. //只尝试用WMP的图标,得不到就算了.
  168. CopyRegString(HKEY_CLASSES_ROOT, _T("Applications\wmplayer.exe\DefaultIcon"), _T("gtv_auto_file\DefaultIcon"), _T(""));
  170. CopyRegString(HKEY_CLASSES_ROOT, _T("Applications\notepad.exe\shell\open\command"), _T("gtv_auto_file\shell\edit\command"), _T(""));
  172. SetRegString(HKEY_LOCAL_MACHINE, _T("SOFTWARE\Microsoft\MediaPlayer\Player\Extensions\Descriptions"),
  173. _T("18123"), _T("高维视讯直播文件 (*.gtv)"));
  175. SetRegString(HKEY_LOCAL_MACHINE, _T("SOFTWARE\Microsoft\MediaPlayer\Player\Extensions\MUIDescriptions"),
  176. _T("18123"), _T("高维视讯直播文件"));
  178. SetRegString(HKEY_LOCAL_MACHINE, _T("SOFTWARE\Microsoft\MediaPlayer\Player\Extensions\Types"),
  179. _T("18123"), _T("*.gtv"));
  181. DWORD value = 1;
  182. SetRegDWord(HKEY_CURRENT_USER, _T("SOFTWARE\Microsoft\MediaPlayer\Player\Extensions\.gtv"), _T("Permissions"), value);
  183. SetRegDWord(HKEY_CURRENT_USER, _T("SOFTWARE\Microsoft\MediaPlayer\Player\Extensions\.gtv"), _T("Runtime"), value);
  185. //以下同样地处理ZZL
  187. SetRegString(HKEY_CLASSES_ROOT, _T("Media Type\Extensions\.zzl"), _T("Source Filter"), Chaos_GUID);
  189. SetRegString(HKEY_CLASSES_ROOT, _T("HTTP\Extensions"), _T(".zzl"), Chaos_GUID);
  191. SetRegString(HKEY_CLASSES_ROOT, _T(".zzl"), NULL, _T("zzl_auto_file"));
  192. // SetRegString(HKEY_CLASSES_ROOT, _T(".zzl"), _T("PerceivedType"), _T("video"));
  194. nUrlListCount = 65536;
  195. SetRegBin(HKEY_CLASSES_ROOT, _T("zzl_auto_file"), _T("EditFlags"), (BYTE*) &nUrlListCount, sizeof(DWORD));
  196. SetRegBin(HKEY_LOCAL_MACHINE, _T("SOFTWARE\Classes\zzl_auto_file"), _T("EditFlags"), (BYTE*) &nUrlListCount, sizeof(DWORD));
  198. SetRegString(HKEY_CLASSES_ROOT, _T("zzl_auto_file\shell\open\command"), _T(""), playeropencommand);
  199. SetRegString(HKEY_CLASSES_ROOT, _T("zzl_auto_file\shell\play\command"), _T(""), playerplaycommand);
  200. //只尝试用WMP的图标,得不到就算了.
  201. CopyRegString(HKEY_CLASSES_ROOT, _T("Applications\wmplayer.exe\DefaultIcon"), _T("gtv_auto_file\DefaultIcon"), _T(""));
  203. SetRegString(HKEY_LOCAL_MACHINE, _T("SOFTWARE\Microsoft\MediaPlayer\Player\Extensions\Descriptions"),
  204. _T("18124"), _T("高维视讯点播文件 (*.zzl)"));
  206. SetRegString(HKEY_LOCAL_MACHINE, _T("SOFTWARE\Microsoft\MediaPlayer\Player\Extensions\MUIDescriptions"),
  207. _T("18124"), _T("高维视讯点播文件"));
  209. SetRegString(HKEY_LOCAL_MACHINE, _T("SOFTWARE\Microsoft\MediaPlayer\Player\Extensions\Types"),
  210. _T("18124"), _T("*.zzl"));
  212. value = 1;
  213. SetRegDWord(HKEY_CURRENT_USER, _T("SOFTWARE\Microsoft\MediaPlayer\Player\Extensions\.zzl"), _T("Permissions"), value);
  214. SetRegDWord(HKEY_CURRENT_USER, _T("SOFTWARE\Microsoft\MediaPlayer\Player\Extensions\.zzl"), _T("Runtime"), value);
  216.     //注册GAOV协议
  217.     SetRegString(HKEY_CLASSES_ROOT, _T("GAOV"), _T("Source Filter"), Chaos_GUID);
  218.     SetRegString(HKEY_CLASSES_ROOT, _T("GAOV"), _T("URL Protocol"), _T(""));
  219.     SetRegString(HKEY_CLASSES_ROOT, _T("GAOV"), NULL, _T("URL: Mysee Media Protocol"));
  220.     SetRegString(HKEY_CLASSES_ROOT, _T("GAOV\shell\open\command"), _T(""), playeropencommand);
  222.     //让WMP允许我们的GAOV和GVWRAP协议
  223.     SetRegDWord(HKEY_CURRENT_USER, _T("SOFTWARE\Microsoft\MediaPlayer\Player\Schemes\GAOV"), _T("Runtime"), 1);
  225.     //命令行的作用已经结束
  226. delete[] playeropencommand;
  227. delete[] playerplaycommand;
  229. return AMovieDllRegisterServer2(TRUE);
  230. }
  232. STDAPI DllUnregisterServer()
  233. {
  234. HKEY hkSub;
  235. DWORD dis;
  236. LONG lRes;
  237. lRes = RegCreateKeyEx(HKEY_CLASSES_ROOT, _T("HTTP\Extensions"), 0, _T(""), 0, 
  238. KEY_READ | KEY_WRITE, NULL, &hkSub, &dis);
  239. RegDeleteValue(hkSub, _T(".gtv"));
  240. RegDeleteValue(hkSub, _T(".zzl"));
  241. RegCloseKey(hkSub);
  242. lRes = RegCreateKeyEx(HKEY_CLASSES_ROOT, _T("Applications\wmplayer.exe\SupportedTypes"), 0, _T(""), 0, 
  243. KEY_READ | KEY_WRITE, NULL, &hkSub, &dis);
  244. RegDeleteValue(hkSub, _T(".gtv"));
  245. RegDeleteValue(hkSub, _T(".zzl"));
  246. RegCloseKey(hkSub);
  248. RegDeleteKey(HKEY_CLASSES_ROOT, _T("Media Type\Extensions\.gtv"));
  249. RegDeleteKey(HKEY_CLASSES_ROOT, _T(".gtv"));
  250. RegDeleteKey(HKEY_CLASSES_ROOT, _T("gtv_auto_file\DefaultIcon"));
  251. RegDeleteKey(HKEY_CLASSES_ROOT, _T("gtv_auto_file\EditFlags"));
  252. RegDeleteKey(HKEY_CLASSES_ROOT, _T("gtv_auto_file\shell\open\command"));
  253. RegDeleteKey(HKEY_CLASSES_ROOT, _T("gtv_auto_file\shell\play\command"));
  254. RegDeleteKey(HKEY_CLASSES_ROOT, _T("gtv_auto_file\shell\edit\command"));
  255. RegDeleteKey(HKEY_CLASSES_ROOT, _T("gtv_auto_file\shell\open"));
  256. RegDeleteKey(HKEY_CLASSES_ROOT, _T("gtv_auto_file\shell\play"));
  257. RegDeleteKey(HKEY_CLASSES_ROOT, _T("gtv_auto_file\shell\edit"));
  258. RegDeleteKey(HKEY_CLASSES_ROOT, _T("gtv_auto_file\shell"));
  259. RegDeleteKey(HKEY_CLASSES_ROOT, _T("gtv_auto_file"));
  261. RegDeleteKey(HKEY_CLASSES_ROOT, _T("Media Type\Extensions\.zzl"));
  262. RegDeleteKey(HKEY_CLASSES_ROOT, _T(".zzl"));
  263. RegDeleteKey(HKEY_CLASSES_ROOT, _T("zzl_auto_file\DefaultIcon"));
  264. RegDeleteKey(HKEY_CLASSES_ROOT, _T("zzl_auto_file\EditFlags"));
  265. RegDeleteKey(HKEY_CLASSES_ROOT, _T("zzl_auto_file\shell\open\command"));
  266. RegDeleteKey(HKEY_CLASSES_ROOT, _T("zzl_auto_file\shell\play\command"));
  267. RegDeleteKey(HKEY_CLASSES_ROOT, _T("zzl_auto_file\shell\edit\command"));
  268. RegDeleteKey(HKEY_CLASSES_ROOT, _T("zzl_auto_file\shell\open"));
  269. RegDeleteKey(HKEY_CLASSES_ROOT, _T("zzl_auto_file\shell\play"));
  270. RegDeleteKey(HKEY_CLASSES_ROOT, _T("zzl_auto_file\shell\edit"));
  271. RegDeleteKey(HKEY_CLASSES_ROOT, _T("zzl_auto_file\shell"));
  272. RegDeleteKey(HKEY_CLASSES_ROOT, _T("zzl_auto_file"));
  274. lRes = RegCreateKeyEx(HKEY_LOCAL_MACHINE, _T("SOFTWARE\Microsoft\MediaPlayer\Player\Extensions\Descriptions"), 0, _T(""), 0, 
  275. KEY_READ | KEY_WRITE, NULL, &hkSub, &dis);
  276. RegDeleteValue(hkSub, _T("18123"));
  277. RegDeleteValue(hkSub, _T("18124"));
  278. RegCloseKey(hkSub);
  279. lRes = RegCreateKeyEx(HKEY_LOCAL_MACHINE, _T("SOFTWARE\Microsoft\MediaPlayer\Player\Extensions\MUIDescriptions"), 0, _T(""), 0, 
  280. KEY_READ | KEY_WRITE, NULL, &hkSub, &dis);
  281. RegDeleteValue(hkSub, _T("18123"));
  282. RegDeleteValue(hkSub, _T("18124"));
  283. RegCloseKey(hkSub);
  284. lRes = RegCreateKeyEx(HKEY_LOCAL_MACHINE, _T("SOFTWARE\Microsoft\MediaPlayer\Player\Extensions\Types"), 0, _T(""), 0, 
  285. KEY_READ | KEY_WRITE, NULL, &hkSub, &dis);
  286. RegDeleteValue(hkSub, _T("18123"));
  287. RegDeleteValue(hkSub, _T("18124"));
  288. RegCloseKey(hkSub);
  290. lRes = RegCreateKeyEx(HKEY_CURRENT_USER, _T("SOFTWARE\Microsoft\MediaPlayer\Player\Extensions\.gtv"), 0, _T(""), 0, 
  291. KEY_READ | KEY_WRITE, NULL, &hkSub, &dis);
  292. RegDeleteValue(hkSub, _T("Permissions"));
  293. RegDeleteValue(hkSub, _T("Runtime"));
  294. RegCloseKey(hkSub);
  295. lRes = RegCreateKeyEx(HKEY_CURRENT_USER, _T("SOFTWARE\Microsoft\MediaPlayer\Player\Extensions\.zzl"), 0, _T(""), 0, 
  296. KEY_READ | KEY_WRITE, NULL, &hkSub, &dis);
  297. RegDeleteValue(hkSub, _T("Permissions"));
  298. RegDeleteValue(hkSub, _T("Runtime"));
  299. RegCloseKey(hkSub);
  301.     //反注册GAOV协议
  302. RegDeleteKey(HKEY_CLASSES_ROOT, _T("GAOV\shell\open\command"));
  303. RegDeleteKey(HKEY_CLASSES_ROOT, _T("GAOV\shell\open"));
  304. RegDeleteKey(HKEY_CLASSES_ROOT, _T("GAOV\shell"));
  305. RegDeleteKey(HKEY_CLASSES_ROOT, _T("GAOV"));
  306. RegDeleteKey(HKEY_CURRENT_USER, _T("SOFTWARE\Microsoft\MediaPlayer\Player\Schemes\GAOV"));
  308. return AMovieDllRegisterServer2(FALSE);
  309. }