1 // VisVim.cpp : Defines the initialization routines for the DLL. 2 // 3 4 #include "stdafx.h" 5 #include <initguid.h> 6 #include "VisVim.h" 7 #include "DSAddIn.h" 8 #include "Commands.h" 9 10 #ifdef _DEBUG 11 #define new DEBUG_NEW 12 #undef THIS_FILE 13 static char THIS_FILE[] = __FILE__; 14 15 #endif 16 17 CComModule _Module; 18 19 BEGIN_OBJECT_MAP (ObjectMap) 20 OBJECT_ENTRY (CLSID_DSAddIn, CDSAddIn) 21 END_OBJECT_MAP () 22 23 class CVisVimApp : public CWinApp 24 { 25 public: 26 CVisVimApp (); 27 28 //{{AFX_VIRTUAL(CVisVimApp) 29 public: 30 virtual BOOL InitInstance (); 31 virtual int ExitInstance (); 32 //}}AFX_VIRTUAL 33 34 //{{AFX_MSG(CVisVimApp) 35 //}}AFX_MSG 36 DECLARE_MESSAGE_MAP () 37 }; 38 39 BEGIN_MESSAGE_MAP (CVisVimApp, CWinApp) 40 //{{AFX_MSG_MAP(CVisVimApp) 41 //}}AFX_MSG_MAP 42 END_MESSAGE_MAP () 43 44 // The one and only CVisVimApp object 45 CVisVimApp theApp; 46 47 CVisVimApp::CVisVimApp () 48 { 49 } 50 51 BOOL CVisVimApp::InitInstance () 52 { 53 _Module.Init (ObjectMap, m_hInstance); 54 return CWinApp::InitInstance (); 55 } 56 57 int CVisVimApp::ExitInstance () 58 { 59 _Module.Term (); 60 return CWinApp::ExitInstance (); 61 } 62 63 // Special entry points required for inproc servers 64 // 65 66 STDAPI DllGetClassObject (REFCLSID rclsid, REFIID riid, LPVOID * ppv) 67 { 68 AFX_MANAGE_STATE (AfxGetStaticModuleState ()); 69 return _Module.GetClassObject (rclsid, riid, ppv); 70 } 71 72 STDAPI DllCanUnloadNow (void) 73 { 74 AFX_MANAGE_STATE (AfxGetStaticModuleState ()); 75 return (AfxDllCanUnloadNow () == S_OK && _Module.GetLockCount () == 0) 76 ? S_OK : S_FALSE; 77 } 78 79 // By exporting DllRegisterServer, you can use regsvr32.exe 80 // 81 STDAPI DllRegisterServer (void) 82 { 83 AFX_MANAGE_STATE (AfxGetStaticModuleState ()); 84 HRESULT hRes; 85 86 // Registers object, typelib and all interfaces in typelib 87 hRes = _Module.RegisterServer (TRUE); 88 if (FAILED (hRes)) 89 // Hack: When this fails we might be a normal user, while the 90 // admin already registered the module. Returning S_OK then 91 // makes it work. When the module was never registered it 92 // will soon fail in another way. 93 // old code: return hRes; 94 return S_OK; 95 96 _ATL_OBJMAP_ENTRY *pEntry = _Module.m_pObjMap; 97 CRegKey key; 98 LONG lRes = key.Open (HKEY_CLASSES_ROOT, _T ("CLSID")); 99 100 if (lRes == ERROR_SUCCESS) 101 { 102 USES_CONVERSION; 103 LPOLESTR lpOleStr; 104 105 StringFromCLSID (*pEntry->pclsid, &lpOleStr); 106 LPTSTR lpsz = OLE2T (lpOleStr); 107 108 lRes = key.Open (key, lpsz); 109 if (lRes == ERROR_SUCCESS) 110 { 111 CString strDescription; 112 113 strDescription.LoadString (IDS_VISVIM_DESCRIPTION); 114 key.SetKeyValue (_T ("Description"), strDescription); 115 } 116 CoTaskMemFree (lpOleStr); 117 } 118 119 if (lRes != ERROR_SUCCESS) 120 hRes = HRESULT_FROM_WIN32 (lRes); 121 122 return hRes; 123 124 } 125 126 // DllUnregisterServer - Removes entries from the system registry 127 // 128 STDAPI DllUnregisterServer (void) 129 { 130 AFX_MANAGE_STATE (AfxGetStaticModuleState ()); 131 132 HRESULT hRes = S_OK; 133 _Module.UnregisterServer (); 134 return hRes; 135 } 136 137 138 // Debugging support 139 140 // GetLastErrorDescription is used in the implementation of the VERIFY_OK 141 // macro, defined in stdafx.h. 142 143 #ifdef _DEBUG 144 145 void GetLastErrorDescription (CComBSTR & bstr) 146 { 147 CComPtr < IErrorInfo > pErrorInfo; 148 if (GetErrorInfo (0, &pErrorInfo) == S_OK) 149 pErrorInfo->GetDescription (&bstr); 150 } 151 152 #endif //_DEBUG 153