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

源码开发语言/平台
当前位置：首页 > 源码/资料 > Windows编程 > Shell编程 > 查看源码
dllregshex.cpp
资源名称：dllregshex_src.zip　[点击查看]
上传用户：sencon168
上传日期：2007-01-04
资源大小：22k
文件大小：9k
源码类别：

Shell编程

开发平台：

Visual C++

dllregshex.cpp：源码内容
  1. #include "dllreg_xhdr.h"
  2. #include <olectl.h>
  3. #include "dllregshex.h"
  4. #include "dllreg_util.h"
  5. #include "resource.h"
  7. #ifndef __CLSID_NTSrv_Defined__
  8. #define __CLSID_NTSrv_Defined__
  9. DEFINE_GUID(CLSID_DLLRegisterer, 0x25d84cb0, 0x7345, 0x11d3, 0xa4, 0xa1, 0x0, 0x80, 0xc8, 0xec, 0xfe, 0xd4);
  10. #endif
  12. UINT      g_cRefThisDll = 0;    
  13. HINSTANCE g_hmodThisDll = NULL;
  15. HBITMAP hBmp_Install = 0;
  16. HBITMAP hBmp_Uninstall  = 0;
  17. HBITMAP hBmp_About = 0;
  18. HMENU hSubMenu = 0;
  20. typedef struct
  21. {
  22.    HKEY  hRootKey;
  23.    LPTSTR lpszSubKey;
  24.    LPTSTR lpszValueName;
  25.    LPTSTR lpszData;
  26. } REGSTRUCT, *LPREGSTRUCT;
  28. STDAPI DllRegisterServer(void);
  29. STDAPI DllUnregisterServer(void);
  31. VOID   _LoadResources();
  32. VOID   _UnloadResources();
  34. STDAPI 
  35. DllRegisterServer(void)
  36. {
  37. HINSTANCE hInst = g_hmodThisDll;
  39. int      i;
  40. HKEY     hKey;
  41. LRESULT  lResult;
  42. DWORD    dwDisp;
  43. TCHAR    szSubKey[MAX_PATH];
  44. TCHAR    szCLSID[MAX_PATH];
  45. TCHAR    szModule[MAX_PATH];
  46. LPWSTR   pwszShellExt;
  48. StringFromIID(CLSID_DLLRegisterer, &pwszShellExt);
  50. if (pwszShellExt) 
  51. {
  52. WideCharToLocal(szCLSID, pwszShellExt, ARRAYSIZE(szCLSID));
  54. LPMALLOC pMalloc;
  55. CoGetMalloc(1, &pMalloc);
  56. if(pMalloc)
  57. {
  58. pMalloc->Free(pwszShellExt);
  59. pMalloc->Release();
  60. }
  61. }    
  63. GetModuleFileName(hInst, szModule, ARRAYSIZE(szModule));
  64. REGSTRUCT ShExClsidEntries[] = 
  65. {
  66. HKEY_CLASSES_ROOT, TEXT("CLSID\%s"), NULL,                   TEXT(DLLREGUNREGNAME),
  67. HKEY_CLASSES_ROOT, TEXT("CLSID\%s\InProcServer32"), NULL,                   TEXT("%s"),
  68. HKEY_CLASSES_ROOT, TEXT("CLSID\%s\InProcServer32"), TEXT("ThreadingModel"), TEXT("Apartment"),
  69. NULL,                NULL, NULL,                   NULL
  70. };
  72. for(i = 0; ShExClsidEntries[i].hRootKey; i++) 
  73.     {
  74. wsprintf(szSubKey, ShExClsidEntries[i].lpszSubKey, szCLSID);
  75. lResult = RegCreateKeyEx(ShExClsidEntries[i].hRootKey, szSubKey, 0, NULL, REG_OPTION_NON_VOLATILE,
  76. KEY_WRITE, NULL, &hKey, &dwDisp);
  78. if(NOERROR == lResult) 
  79. {
  80. TCHAR szData[MAX_PATH];
  81. wsprintf(szData, ShExClsidEntries[i].lpszData, szModule);
  82. lResult = RegSetValueEx(hKey, ShExClsidEntries[i].lpszValueName, 0, REG_SZ, 
  83. (LPBYTE)szData, lstrlen(szData) + 1);
  84. RegCloseKey(hKey);
  85. } 
  86. else
  87. return SELFREG_E_CLASS;
  88.    }                                
  90. REGSTRUCT OtherShExEntries[] = 
  91. {
  92. HKEY_LOCAL_MACHINE, TEXT("software\classes\clsid\"DLLREGUNREGNAME)   , NULL, TEXT("%s"),
  93. HKEY_CLASSES_ROOT, TEXT("dllfile\shellex\ContextMenuHandlers\"DLLREGUNREGNAME), NULL, TEXT("%s"),
  94. NULL, NULL, NULL,                   NULL
  95. };
  97. for (i = 0; OtherShExEntries[i].hRootKey; i++) 
  98.     {
  99. wsprintf(szSubKey, OtherShExEntries[i].lpszSubKey, szCLSID);
  100. lResult = RegCreateKeyEx(OtherShExEntries[i].hRootKey, szSubKey, 0, NULL, REG_OPTION_NON_VOLATILE,
  101. KEY_WRITE, NULL, &hKey, &dwDisp);
  103. if(NOERROR == lResult) 
  104. {
  105. TCHAR szData[MAX_PATH];
  106. wsprintf(szData, OtherShExEntries[i].lpszData, szCLSID);
  107. lResult = RegSetValueEx(hKey, OtherShExEntries[i].lpszValueName, 0, REG_SZ,
  108. (LPBYTE)szData, lstrlen(szData) + 1);
  109. RegCloseKey(hKey);
  110. } 
  111. else
  112. return SELFREG_E_CLASS;
  113. }
  115. OSVERSIONINFO  osvi;
  116. osvi.dwOSVersionInfoSize = sizeof(osvi);
  117. GetVersionEx(&osvi);
  118. if (VER_PLATFORM_WIN32_NT == osvi.dwPlatformId)  
  119. {
  120. lstrcpy( szSubKey, TEXT("Software\Microsoft\Windows\CurrentVersion\Shell Extensions\Approved"));
  121. lResult = RegCreateKeyEx(  HKEY_LOCAL_MACHINE, szSubKey, 0, NULL, REG_OPTION_NON_VOLATILE, KEY_WRITE,
  122. NULL, &hKey, &dwDisp);
  123. if(NOERROR == lResult) 
  124. {
  125. TCHAR szData[MAX_PATH];
  126. lstrcpy(szData, DLLREGUNREGNAME);
  127. lResult = RegSetValueEx(hKey, szCLSID, 0, REG_SZ, (LPBYTE)szData, lstrlen(szData) + 1);
  128. RegCloseKey(hKey);
  129. } 
  130. else
  131. return SELFREG_E_CLASS;
  132. }
  134. return S_OK;
  135. }
  137. STDAPI 
  138. DllUnregisterServer(void)
  139. {
  140. int      i;
  141. LRESULT  lResult;
  142. TCHAR    szSubKey[MAX_PATH];
  143. TCHAR    szCLSID[MAX_PATH];
  144. LPWSTR   pwszShellExt;
  146. StringFromIID(CLSID_DLLRegisterer, &pwszShellExt);
  148. if (pwszShellExt) 
  149. {
  150. WideCharToLocal(szCLSID, pwszShellExt, ARRAYSIZE(szCLSID));
  152. LPMALLOC pMalloc;   
  153. CoGetMalloc(1, &pMalloc);
  155. if(pMalloc)
  156. {
  157. pMalloc->Free(pwszShellExt);
  158. pMalloc->Release();
  159. }
  160. }    
  161.     
  162. REGSTRUCT ShExClsidEntries[] = 
  163. {
  164. HKEY_CLASSES_ROOT, TEXT("CLSID\%s\InProcServer32"), NULL, NULL,
  165. HKEY_CLASSES_ROOT, TEXT("CLSID\%s"), NULL, NULL,
  166. NULL, NULL, NULL, NULL
  167. };
  169. for(i = 0; ShExClsidEntries[i].hRootKey; i++) 
  170. {
  171. wsprintf(szSubKey, ShExClsidEntries[i].lpszSubKey, szCLSID);
  172. lResult = RegDeleteKey(ShExClsidEntries[i].hRootKey, szSubKey);
  173.  
  174. if(lResult != NOERROR) 
  175. return SELFREG_E_CLASS;
  176. }
  178. REGSTRUCT OtherShExEntries[] = 
  179. {
  180. HKEY_CLASSES_ROOT, TEXT("dllfile\shellex\ContextMenuHandlers\"DLLREGUNREGNAME), NULL, NULL,
  181. NULL, NULL, NULL,                   NULL
  182. };
  184. for(i = 0; OtherShExEntries[i].hRootKey; i++) 
  185. {
  186. wsprintf(szSubKey, OtherShExEntries[i].lpszSubKey, szCLSID);
  187. lResult = RegDeleteKey(OtherShExEntries[i].hRootKey, szSubKey);
  188. if(lResult != NOERROR) 
  189. return SELFREG_E_CLASS;
  190. }
  191.     
  192. OSVERSIONINFO  osvi;
  193. osvi.dwOSVersionInfoSize = sizeof(osvi);
  194. GetVersionEx(&osvi);
  196. if (VER_PLATFORM_WIN32_NT == osvi.dwPlatformId)  
  197. {
  198. lstrcpy( szSubKey, TEXT("Software\Microsoft\Windows\CurrentVersion\Shell Extensions\Approved\"DLLREGUNREGNAME));
  199. lResult = RegDeleteKey(HKEY_LOCAL_MACHINE, szSubKey);
  200. }
  202. return S_OK;
  203. }
  205. extern "C" 
  206. int APIENTRY
  207. DllMain(HINSTANCE hInstance, DWORD dwReason, LPVOID lpReserved)
  208. {
  209. _UNUSED_PARAMETER(lpReserved);
  211.     if (dwReason == DLL_PROCESS_ATTACH)
  212. {
  213.         g_hmodThisDll = hInstance;
  214. _LoadResources();
  215. }
  216. else
  217. _UnloadResources();
  219.     return 1;
  220. }
  222. VOID
  223. _LoadResources(VOID)
  224. {
  225. hBmp_Install = LoadBitmap(g_hmodThisDll, MAKEINTRESOURCE(IDB_INSTALLA));
  226. hBmp_Uninstall  = LoadBitmap(g_hmodThisDll, MAKEINTRESOURCE(IDB_UNINSTALLA));
  227. hBmp_About = LoadBitmap(g_hmodThisDll, MAKEINTRESOURCE(IDB_ABOUTA));
  228. }
  230. VOID
  231. _UnloadResources(VOID)
  232. {
  233. DeleteObject(hBmp_Install);
  234. DeleteObject(hBmp_Uninstall);
  235. DeleteObject(hBmp_About);
  237. DestroyMenu(hSubMenu);
  238. }
  240. STDAPI 
  241. DllCanUnloadNow(void)
  242. {
  243.     return (g_cRefThisDll == 0 ? S_OK : S_FALSE);
  244. }
  246. STDAPI 
  247. DllGetClassObject(REFCLSID rclsid, REFIID riid, LPVOID *ppvOut)
  248. {
  249.     *ppvOut = NULL;
  251.     if (IsEqualIID(rclsid, CLSID_DLLRegisterer))
  252.     {
  253.         CDllRegSxClassFactory *pcf = new CDllRegSxClassFactory;
  254.         return pcf->QueryInterface(riid, ppvOut);
  255.     }
  257.     return CLASS_E_CLASSNOTAVAILABLE;
  258. }
  260. CDllRegSxClassFactory::CDllRegSxClassFactory()
  261. {
  262.     m_cRef = 0L;
  263.     g_cRefThisDll++;
  264. }
  266. CDllRegSxClassFactory::~CDllRegSxClassFactory()
  267. {
  268.     g_cRefThisDll--;
  269. }
  271. STDMETHODIMP 
  272. CDllRegSxClassFactory::QueryInterface(REFIID riid, LPVOID FAR *ppv)
  273. {
  274.     *ppv = NULL;
  276.     if(IsEqualIID(riid, IID_IUnknown) || IsEqualIID(riid, IID_IClassFactory))
  277.     {
  278.         *ppv = (LPCLASSFACTORY)this;
  279.         AddRef();
  281.         return NOERROR;
  282.     }
  284.     return E_NOINTERFACE;
  285. }
  287. STDMETHODIMP_(ULONG) 
  288. CDllRegSxClassFactory::AddRef()
  289. {
  290.     return ++m_cRef;
  291. }
  293. STDMETHODIMP_(ULONG) 
  294. CDllRegSxClassFactory::Release()
  295. {
  296.     if (--m_cRef)
  297.         return m_cRef;
  299.     delete this;
  300.     return 0L;
  301. }
  303. STDMETHODIMP 
  304. CDllRegSxClassFactory::CreateInstance(LPUNKNOWN pUnkOuter, REFIID riid, LPVOID *ppvObj)
  305. {
  306.     *ppvObj = NULL;
  308.     if (pUnkOuter)
  309.      return CLASS_E_NOAGGREGATION;
  311.     LPCSHELLEXT pShellExt = new CShellExt();
  312.     if (NULL == pShellExt)
  313.      return E_OUTOFMEMORY;
  315.     return pShellExt->QueryInterface(riid, ppvObj);
  316. }
  318. STDMETHODIMP 
  319. CDllRegSxClassFactory::LockServer(BOOL fLock)
  320. {
  321. _UNUSED_PARAMETER(fLock);
  323.     return NOERROR;
  324. }
  326. CShellExt::CShellExt()
  327. {
  328.     m_cRef = 0L;
  329.     m_pDataObj = NULL;
  331.     g_cRefThisDll++;
  332. }
  334. CShellExt::~CShellExt()
  335. {
  336.     if (m_pDataObj)
  337.         m_pDataObj->Release();
  339.     g_cRefThisDll--;
  340. }
  342. STDMETHODIMP 
  343. CShellExt::QueryInterface(REFIID riid, LPVOID FAR *ppv)
  344. {
  345.     *ppv = NULL;
  346.     if (IsEqualIID(riid, IID_IShellExtInit) || IsEqualIID(riid, IID_IUnknown))
  347.      *ppv = (LPSHELLEXTINIT)this;
  348.     else if (IsEqualIID(riid, IID_IContextMenu))
  349.         *ppv = (LPCONTEXTMENU)this;
  351.     if(*ppv)
  352.     {
  353.         AddRef();
  354.         return NOERROR;
  355.     }
  357. return E_NOINTERFACE;
  358. }
  360. STDMETHODIMP_(ULONG) 
  361. CShellExt::AddRef()
  362. {
  363.     return ++m_cRef;
  364. }
  366. STDMETHODIMP_(ULONG) 
  367. CShellExt::Release()
  368. {
  369.     if(--m_cRef)
  370.         return m_cRef;
  372.     delete this;
  373.     return 0L;
  374. }
  376. STDMETHODIMP 
  377. CShellExt::Initialize(LPCITEMIDLIST pIDFolder, LPDATAOBJECT pDataObj, HKEY hRegKey)
  378. {
  379. _UNUSED_PARAMETER(hRegKey);
  380. _UNUSED_PARAMETER(pIDFolder);
  382. if (m_pDataObj)
  383.      m_pDataObj->Release();
  384.     if (pDataObj)
  385.     {
  386.      m_pDataObj = pDataObj;
  387.      pDataObj->AddRef();
  388.     }
  389.     return NOERROR;
  390. }