引言
前幾天山寨了ATL的COM_INTERFACE，了解了一個(gè)COM類的如何進(jìn)行通用的組織。今天再來(lái)學(xué)習(xí)下COM協(xié)議，看看如何實(shí)現(xiàn)一個(gè)COM組件——當(dāng)然，也是不能用ATL的，不然就學(xué)不到什么了。
COM DLL說(shuō)簡(jiǎn)單簡(jiǎn)單，說(shuō)復(fù)雜也很復(fù)雜。說(shuō)簡(jiǎn)單呢，其實(shí)貌似只要導(dǎo)出下面這五個(gè)函數(shù)就可以了：
DllCanUnloadNow
DllGetClassObject
DllRegisterServer
DllUnregisterServer
DllInstall
（我有點(diǎn)懷疑但不確定DllInstall是不是后來(lái)加的，本文中我們先不理它。）
前四個(gè)函數(shù)中，后兩個(gè)是注冊(cè)與反注冊(cè)，就是寫寫注冊(cè)表的事情，簡(jiǎn)單。前面兩個(gè)，特別是DllGetClassObject，比較關(guān)鍵。
先不研究這些，我們先按前兩天的方法寫個(gè)COM類吧。
準(zhǔn)備一個(gè)COM類
首先，我們定義一個(gè)接口 ISampleInterface，以及一個(gè)類CSampleClass
Interface.h
| #include <Unknwn.h> struct __declspec(uuid("{83C783E3-F989-4E0D-BFC5-631273EDFFDA}")) ISampleInterface : public IUnknown { STDMETHOD(SampleMethod)() PURE; }; class __declspec(uuid("{0DECBFF5-A8A5-49E8-9962-3D18AAC6088E}")) SampleClass; |
與前面不同的是，這里兩個(gè)聲明都寫上了一個(gè)UUID，前一個(gè)是接口ID（IID），后一個(gè)是類ID（CLSID）。這個(gè)文件將是提供給COM的使用者的。
SampleClass.h
| class SampleClass : public xl::ComClass<SampleClass>, xl::IUnknownImpl<ISampleInterface> { public: SampleClass(); ~SampleClass(); public: STDMETHOD(SampleMethod)(); public: XL_COM_INTERFACE_BEGIN(SampleClass) XL_COM_INTERFACE(ISampleInterface) XL_COM_INTERFACE_END() }; |
SampleClass.cpp
| SampleClass::SampleClass() { InterlockedIncrement(&g_nModuleCount); } SampleClass::~SampleClass() { InterlockedDecrement(&g_nModuleCount); } STDMETHODIMP SampleClass::SampleMethod() { MessageBox(NULL, _T("SampleMethod called."), _T("Info"), MB_OK | MB_ICONINFORMATION); return S_OK; } |
SampleMethod 就簡(jiǎn)單調(diào)用一個(gè)MessageBox意思一下。構(gòu)造函數(shù)和析構(gòu)函數(shù)中兩行先不看，后面解釋。
實(shí)現(xiàn)DllCanUnloadNow
這個(gè)函數(shù)的MSDN文檔見(jiàn)：
http://msdn.microsoft.com/en-us/library/windows/desktop/ms690368.aspx
函數(shù)原型為：
HRESULT __stdcall DllCanUnloadNow(void);
函數(shù)實(shí)現(xiàn)的要求是，當(dāng)此DLL內(nèi)的所有COM對(duì)象都消亡了的時(shí)候，返回S_OK；如果還有COM對(duì)象存在，就返回S_FALSE。
現(xiàn)在我們有一個(gè)COM類SampleClass，它可能被創(chuàng)建一次，然后引用計(jì)數(shù)加加減減；也可能被創(chuàng)建多次，每個(gè)實(shí)例的引用計(jì)數(shù)同樣會(huì)被加加減減。當(dāng)引用計(jì)數(shù)被減為0的時(shí)候，對(duì)象將消亡（析構(gòu)函數(shù)被調(diào)用）。因此，我們?cè)趯?duì)象的構(gòu)造和析構(gòu)的地方埋點(diǎn)就可以了。
定義一個(gè)全局變量：
LONG g_nModuleCount = 0;
然后看到剛才的SampleClass的構(gòu)造函數(shù)和析構(gòu)函數(shù)中的灰色代碼，就可以存對(duì)象的創(chuàng)建/銷毀計(jì)數(shù)了。對(duì)于多個(gè)COM對(duì)象的情況，也可以這么搞。
然后，DllCanUnloadNow的實(shí)現(xiàn)就很簡(jiǎn)單了：
| STDAPI DllCanUnloadNow() { return g_nModuleCount > 0 ? S_FALSE : S_OK; } |
實(shí)現(xiàn)DllGetClassObject
終于到關(guān)鍵的地方了。以前一直弄不懂這個(gè)函數(shù)。它的MSDN文檔見(jiàn)：
http://msdn.microsoft.com/en-us/library/windows/desktop/ms680760.aspx
函數(shù)原型為：
HRESULT __stdcall DllGetClassObject(REFCLSID rclsid, REFIID riid, LPVOID *ppv);
第一個(gè)參數(shù)是要?jiǎng)?chuàng)建的對(duì)象的CLSID，很明確。第二個(gè)參數(shù)有點(diǎn)迷惑，MSDN原文是：
A reference to the identifier of the interface that the caller is to use to communicate with the class object. Usually, this is IID_IClassFactory (defined in the OLE headers as the interface identifier for IClassFactory).
“Usually”，它是IID_IClassFactory。我不知道有沒(méi)有不“Usually”的情況，也不知道這個(gè)接口原先的設(shè)計(jì)意圖是什么。如果可能，其實(shí)完全可以繞開(kāi)類廠機(jī)制，直接用想要使用的那個(gè)接口的IID，貌似整套機(jī)制也能運(yùn)轉(zhuǎn)……有木有達(dá)人解釋下它的淵源？
不過(guò)呢，目前我就把它當(dāng)作IID_IClassFactory，其他一律不支持。
我們先要實(shí)現(xiàn)一個(gè)“類廠”——一個(gè)繼承于IClassFactory的COM類：
ClassFactory.h
| class ClassFactory : public xl::ComClass<ClassFactory>, public xl::IClassFactoryImpl<> { public: ClassFactory(REFCLSID rclsid); ~ClassFactory(); public: STDMETHOD(CreateInstance)(_In_opt_ IUnknown *pUnkOuter, _In_ REFIID riid, _COM_Outptr_ void **ppvObject); public: XL_COM_INTERFACE_BEGIN(ClassFactory) XL_COM_INTERFACE(IClassFactory) XL_COM_INTERFACE_END() private: CLSID m_clsid; }; |
IClassFactory有兩個(gè)方法，CreateInstance和LockServer。我們只實(shí)現(xiàn)前者，它用于創(chuàng)建一個(gè)對(duì)象，第一個(gè)參數(shù)不理它，第二個(gè)參數(shù)是IID，第三個(gè)參數(shù)用于輸出。CLSID由構(gòu)造函數(shù)傳入，保存在m_clsid中——類廠是與COM類一一對(duì)應(yīng)的。
ClassFactory.cpp
| ClassFactory::ClassFactory(REFCLSID rclsid) { m_clsid = rclsid; } ClassFactory::~ClassFactory() { } STDMETHODIMP ClassFactory::CreateInstance(_In_opt_ IUnknown *pUnkOuter, _In_ REFIID riid, _COM_Outptr_ void **ppvObject) { if (riid == __uuidof(ISampleInterface) && m_clsid == __uuidof(SampleClass)) { ISampleInterface *p = new SampleClass; p->QueryInterface(riid, ppvObject); return S_OK; } return CLASS_E_CLASSNOTAVAILABLE; } |
這里只對(duì)IID為__uuidof(ISampleInterface)且CLSID為__uuidof(SampleClass))的情況作了響應(yīng)，創(chuàng)建一個(gè)SampleClass對(duì)象。注意到創(chuàng)建對(duì)象之后有一次QueryInterface，在這里面會(huì)做一次AddRef操作，因此引用計(jì)數(shù)此時(shí)為1。其實(shí)AddRef可以不出現(xiàn)在代碼中，需要的時(shí)候就用QueryInterface代替。
這些都準(zhǔn)備好了，最后來(lái)實(shí)現(xiàn)DllGetClassObject：
| STDAPI DllGetClassObject(_In_ REFCLSID rclsid, _In_ REFIID riid, _Outptr_ LPVOID *ppv) { if (riid == __uuidof(IClassFactory) && rclsid == __uuidof(SampleClass)) { IClassFactory *p = new ClassFactory(rclsid); p->QueryInterface(riid, ppv); return S_OK; } return CLASS_E_CLASSNOTAVAILABLE; } |
實(shí)現(xiàn)DllRegisterServer和DllUnregisterServer
這個(gè)我就直接貼代碼了：
| STDAPI DllRegisterServer(void) { TCHAR szModulePath[MAX_PATH] = {}; GetModuleFileName(g_hModule, szModulePath, ARRAYSIZE(szModulePath)); xl::Registry::SetString(HKEY_CLASSES_ROOT, _T("CLSID\\{0DECBFF5-A8A5-49E8-9962-3D18AAC6088E}\\InprocServer32"), _T(""), szModulePath); xl::Registry::SetString(HKEY_CLASSES_ROOT, _T("CLSID\\{0DECBFF5-A8A5-49E8-9962-3D18AAC6088E}\\ProgID"), _T(""), _T("Streamlet.COMProvider.SampleClass.1")); xl::Registry::SetString(HKEY_CLASSES_ROOT, _T("Streamlet.COMProvider.SampleClass.1"), _T(""), _T("SampleClass Class")); xl::Registry::SetString(HKEY_CLASSES_ROOT, _T("Streamlet.COMProvider.SampleClass.1\\CLSID"), _T(""), _T("{0DECBFF5-A8A5-49E8-9962-3D18AAC6088E}")); return S_OK; } STDAPI DllUnregisterServer(void) { xl::Registry::DeleteKeyRecursion(HKEY_CLASSES_ROOT, _T("CLSID\\{0DECBFF5-A8A5-49E8-9962-3D18AAC6088E}")); xl::Registry::DeleteKeyRecursion(HKEY_CLASSES_ROOT, _T("Streamlet.COMProvider.SampleClass.1")); return S_OK; } |
注意，我這里并沒(méi)有寫得很全，只是注冊(cè)幾項(xiàng)必要的。至此，我們的COM組件實(shí)現(xiàn)完畢。
調(diào)用COM組件
編譯剛才的DLL，并使用regsvr32注冊(cè)（注意管理員權(quán)限）。然后寫一個(gè)小程序來(lái)調(diào)用之：
| #include <tchar.h> #include <Objbase.h> #include "../COMProvider/Interface.h" int _tmain(int argc, TCHAR *argv[]) { HRESULT hr = CoInitialize(NULL); ISampleInterface *pSampleInterface = nullptr; hr = CoCreateInstance(__uuidof(SampleClass), nullptr, CLSCTX_INPROC_SERVER, __uuidof(ISampleInterface), (LPVOID *)&pSampleInterface); if (SUCCEEDED(hr)) { pSampleInterface->SampleMethod(); pSampleInterface->Release(); } CoUninitialize(); return 0; } |
運(yùn)行結(jié)果：
上述例子代碼見(jiàn)COMProtocol.rar（http://pan.baidu.com/s/1c0GSI7u），庫(kù)依賴見(jiàn)http://xllib.codeplex.com/。當(dāng)然，現(xiàn)在只是簡(jiǎn)單地迎合了一下CoCreateInstance，還有許多其他事情要做，且聽(tīng)下回分解。
posted on 2012-09-07 00:23
溪流 閱讀(6419)
評(píng)論(5) 編輯 收藏 引用 所屬分類:
C++ 、
Windows 、
COM