用汇编语言编写com组件[二]

来源:互联网 发布:仿淘宝会不会属于侵权 编辑:程序博客网 时间:2024/06/05 08:41
      组件对象模型com是以win32 dll或exe形式发布的执行代码组成的。Com是由一些对象和对象的接口组成,在com里,接口提供对象操作的机制。 而接口是由一个或者多个相关的方法、属性、事件组成的。在这里我们开发一个简单的但是功能齐全的一个进程内com组件(即以dll形式存在)。

这里假设你已经了解了com对象模型的基础知识,了解什么是虚表,什么是虚函数表指针。如果你不熟悉这些,建议看看《com本质论》这本书。
我们先来分析下进程内com服务的组成。由于它是一个dll形式发布的,其中包括5个重要的函数。其中后面的四个是要作为dll导出函数来导出的。

DllMain: 这是动态链接库德第一个入口函数,它在库被加载的时候被调用,通过这个函数,可以对客户端程序进行检查。
DllRegisterServer:通过这个函数能够实现组建的自我注册,注册信息作为资源保存在动态链接库中,这个函数能够读取资源,把信息写进注册表,使用
regsvr32.exe 注册组建时,实际上是调用了组件输出的这个函数。

DllUnregisterServer: 当一个组件不再使用时,这个组件应该能够提供自我卸载,regsvr32.exe能调用这个函数,实现这一步。
DllCanUnloadNow: com服务中的全局变量用于保存它的状态,客户端可以周期性的调用这个函数,检查组件服务器是否在使用,然后把它卸载。

DllGetClassObject: 这是完成组件输出的函数,这个输出需要3个参数,创建组件的GUID,要创建组件接口的GUID以及创建后指向对象的指针。如果组件对象或者接口不被支持,执行将失败。

到现在为止,我们应该注意到一件事情,就是如果不是因为间接访问,com将什么也不是。实际上,DllGetClassObject函数返回的对象不是我们要寻找的对象,它是类厂对象,一个类厂对象了解如何实例化其他任何的类。第一层的间接访问允许组件创建的细节被指定,如果它仅仅是简单而又直接的返回一个我们要寻找的对象指针,那么说明对象已经存在,那样,我们将不能设置和控制关于构造对象的任何参数。


DllGetClassObject返回一个IClassFactory接口,这个接口是从IUnknown派生的,另外他还有自己的两个重要的成员函数。
  HRESULT CreateInstance(
IUnknown * pUnkOuter,   //Pointer to outer object when part of an 
// aggregate REFIID riid, 
REFIID riid,           //Reference to the interface identifier
oid** ppvObject);       //Address of output variable that receives 
// the interface pointer requested in riid

HRESULT LockServer(BOOL fLock); 
//Increments or decrements the lock count 

LockServer用来控制类厂对象的引用计数,系统检查改计数以确定是否要卸载组件,即:控制类厂的生存期。
CreateInstance是最重要的,类厂组件的唯一功能是创建其它组件。一个类厂组件可以对应多种普通COM组件,但每个类厂组件的实例只能创建一种COM组件。
它接收一个接口GUID,返回该接口的指针。它并不接受组件的CLSID,所以一个类厂实例只能够创建一种COM组件,即传给 CoGetClassObject的CLSID对应的组件。

客户、COM库、组件dll、类厂、组件之间的交互过程:
1.    客户首先调用COM库的CoCreateInstance函数来创建COM组件。
2.    CoCreateInstance首先调用COM库的CoGetClassObject获取类厂。
3.    该函数具体是通过调用了组件DLL输出的DllGetClassObject来创建类厂。
4.     DllGetClassObject通过new函数产生一个Cfactory的对象,并通过QueryInterface获取其接口指针(一般是IclassFactory指针)。
5.     返回到COM库的CoCreateInstance调用刚才获得的接口指针(IclassFactory,类厂)的CreateInstance函数。
6.     该函数new指定的组件类,通过QueryInterface获得指定的接口
7.     CoCreateInstanse释放掉IclassFactory指针(通过Release),然后向客户程序返回获得的指针。
8.     可以在客户中使用获得的接口了。
在第6步中,根据不同的CLSID创建不同的组件,可以实现一个类厂供该DLL中多个组件共用。但只是类共用,不是实例共用。一旦在创建类厂时通过CoGetClassObject指定了CLSID,则只能创建该COM组件的实例。


在这里我们将深入c++对象模型,来看下一些内部的实现细节。通常编译器来处理这些。com的设计者充分利用了这些,因此,我们需要了解它。
当我们用汇编写一个常规的程序时,我依靠编译器为我们创建代码段和数据段,内存中的一块区域是我们执行的代码,另一块区域保存了我们需要的数据。
C++运行时动态内存分配,给每一个类实例,每一个小的代码段它自己的数据段。换句话讲,一个类的实例就是这个数据段,每一个类实例的数据描述都是保存在一个动态的数据区域。
或许你听说过c++传递对象成员函数参数时,有一个隐藏的参数,即this指针。当一个人为对象写一个低层的代码时(在c++中编译器会作这个工作,你不需要考虑),
你首先遇到的问题是”我在给哪个对象写代码?“

This指针是一个简单的指针,它指向这个动态数据内存区域的这个类对象实例。当一个类对象函数被调用时,this指针就会被悄悄地传递过去。当这个对象的私有数据被访问时,类的代码区域就会使用this指针,来找到它的对象实例的数据。

对于一个com接口指针跟this指针很类似。使用中,com是一个接口规范,让你看不到它的代码实现。

; declare the ClassFactory object structure 
ClassFactoryObject STRUCT 
lpVtbl   DWORD 0 ; function table pointer 
nRefCount   DWORD 0 ; object and reference count
ClassFactoryObject ENDS 

; declare the MyCom object structure 
MyComObject STRUCT 
lpVtbl   DWORD 0 ; function table pointer
nRefCount   DWORD 0 ; reference count
nValue   DWORD 0 ; interface private data
MyComObject ENDS 

第一个lpVtbl是一个虚表指针,它指向一个虚函数表,我用它来控制每个接口的私有数据。就像这里的nRefCount和nValue。
这些结构所在的动态内存是通过CoTaskMemAlloc和CoTaskMemFree这两个API函数来分配和释放的。这两个函数是由ole32.dll导出的。Ole32.dll还导出了很多的函数,例如比对GUIDs值和把转换GUIDs为字符串,或者把字符串转换为GUIDs。


为了举例说明com接口的工作原理,我们创建一个简单接口IMyCom(注:所有的com接口都有一个“I”前缀。同其他接口一样,他派生于IUnknown接口,也就是说他的前三个函数是QueryInterface, AddRef, 和Release。下面我们添加几个接口函数。下面看到的是c风格的函数原形:
 
HRESULT SetValue(long *pVal); 
HRESULT GetValue(long newVal); 
HRESULT RaiseValue(long newVal); 

其中,SetValue 和 GetValue用于读,设置我们接口的数据成员。RaiseValue用于增加这个数据的值。
这个结构在内存中的形式如下: 

          

客户端仅仅拥有一个分布式结构的指针(ppv)这个名字来源于它的c++形式的定义("pointer to pointer to (void)."),当创建类实例的时候,这个对象数据块是动态分配和初始化的,虚函数表vtable和server functions是静态的,他们在编译时定义好。

有一点需要注意的是,虚函数表拥有的是函数指针,而并非是函数本身。因此,我们可以修改虚函数表中指向的例程,就可以简单的"override“一个派生函数。

在例子中,IClassFactory和IMyCom都是派生于IUnknown接口,都继承了QueryInterface,但是他们支持不同的接口,它们需要指向不同的例程,返回不同的结果。
因此,它们有各自的QueryInterface例程(QueryInterfaceCF 和 QueryInterfaceMC)被不同的虚函数表指向。
同样的,AddRef和Release也要被不同的支持他们的接口来定制。


类型库:
每一个com接口都是从系统注册表中得到信息,这些接口的定义都是由一个被称为接口定义语言(IDL)来描述的,在windows平台下,使用MIDL进行编译。我们可以利用vc开发环境,通过向导来创建一个原始的接口定义文件。

---------------------------------------------------------------------------------------------------------------------On WinTel platforms, 
我建一个ATL工程,命名为MyComApp,然后选择“insert a new ATL object“,然后选择“Simple Object”,命名为:MyCom。这样就创建了一个空的IMyCom接口,然后通过右键菜单,我们添加属性SetValue和GetValue,并增加一个RaiseValue方法。然后我们保存退出工程,拷贝MyComApp.idl文件到我的汇编工程目录。
下面就是这个idl文件的内容:
// MyCom.idl : IDL source for MyCom.dll 
// 
// This file will be processed by the MIDL tool to 
// produce the type library (MyCom.tlb) and marshalling code

import "oaidl.idl"; 
import "ocidl.idl"; 

object, 
uuid(F8CE5E41-1135-11d4-A324-0040F6D487D9), 
helpstring("IMyCom Interface"), 
pointer_default(unique) 

interface IMyCom : IUnknown 

[propget, helpstring("property Value")] 
HRESULT Value([out, retval] long *pVal); 
[propput, helpstring("property Value")] 
HRESULT Value([in] long newVal); 
[helpstring("method Raise")] 
HRESULT Raise(long Value); 
}; 

uuid(F8CE5E42-1135-11d4-A324-0040F6D487D9), 
version(1.0), 
helpstring("MyComApp 1.0 Type Library") 

library MyComLib 

importlib("stdole32.tlb"); 
importlib("stdole2.tlb"); 

uuid(F8CE5E43-1135-11d4-A324-0040F6D487D9), 
helpstring("MyCom Class") 

coclass MyCom 

[default] interface IMyCom; 
}; 
}; 

这个文件可以被用来作为原型进一步进行接口定义。注意这里面有三个GUIDs,一个是为接口,一个是为coclass,一个是为类型库。对于新的应用,它们的值一定不同。
透过这个定义的文件结构,我们很容易了解他的内容。 

[propget, helpstring("property Value")] HRESULT Value([out, retval] long *pVal); [propput, helpstring("property Value")] HRESULT Value([in] long newVal); [helpstring("method Raise")] HRESULT Raise(long Value); 

下面是这些接口在masm32中的定义: 

GetValue PROTO :DWORD, :DWORD 
SetValue PROTO :DWORD, :DWORD 
RaiseValue PROTO :DWORD, :DWORD 
  
他们有很大的不同,但是原因很简单。类型库中的接口是作为通用的,可以直接被客户端象VB来使用。
为了创建类型库,可以使用MIDL命令行来编译idl文件 :
MIDL MyCom.idl 
编译产生的几个文件,除了MyCom.tlb外,其他的都可以忽略,接下来我们需要把类型库添加到dll资源文件中。例如:
1 typelib MyCom.tlb 

让他作为资源文件中的第一个元素是很重要的,后续我们将会使用LoadTypeLib API函数来使用这个库,同时这个函数也是希望在第一位置发现这个库。

注册组件:

DllRegisterServer 和 DllUnregisterServer 为我们注册组件和注销组件用.内容如下:
 
HKEY_CLASSES_ROOT/CMyCom 
(Default) "CMyCom simple client" 
HKEY_CLASSES_ROOT/CMyCom/CLSID 
(Default) "{A21A8C43-1266-11D4-A324-0040F6D487D9}" 

HKEY_CLASSES_ROOT/CLSID/{A21A8C43-1266-11D4-A324-0040F6D487D9} 
(Default) "CMyCom simple client" 
HKEY_CLASSES_ROOT/CLSID/{A21A8C43-1266-11D4-A324-0040F6D487D9}/CMyCom 
(Default) "CMyCom" 
HKEY_CLASSES_ROOT/CLSID/{A21A8C43-1266-11D4-A324-0040F6D487D9}/InprocServer32 
(Default) "C:/MASM32/MYCOM/MYCOM.DLL" 
ThreadingModel "Single" 

HKEY_CLASSES_ROOT/TypeLib/{A21A8C42-1266-11D4-A324-0040F6D487D9} 
(Default) (value not set) 
HKEY_CLASSES_ROOT/TypeLib/{A21A8C42-1266-11D4-A324-0040F6D487D9}/1.0 
(Default) "MyCom 1.0 Type Library" 
HKEY_CLASSES_ROOT/TypeLib/{A21A8C42-1266-11D4-A324-0040F6D487D9}/1.0/0 
(Default) (value not set) 
HKEY_CLASSES_ROOT/TypeLib/{A21A8C42-1266-11D4-A324-0040F6D487D9}/1.0/0/win32 
(Default) " C:/masm32/COM/MyCom /MYCOM.DLL" 
HKEY_CLASSES_ROOT/TypeLib/{A21A8C42-1266-11D4-A324-0040F6D487D9}/1.0/FLAGS 
(Default) "O" 
HKEY_CLASSES_ROOT/TypeLib/{A21A8C42-1266-11D4-A324-0040F6D487D9}/1.0/HELPDIR 
(Default) "C:/masm32/COM/MyCom" 

有一个键值是变化的,它是服务dll自身的路径和文件名,在我的系统上,我把它放置在 "C:/MASM32/COM/MYCOM/MYCOM.DLL",当我注册组件的时候,这个可以被检测到,DllRegisterServer通过调用GetModuleFileName可以发现dll自身的存储位置。

这里有大量的信息是关于这个com服务的,但是我们仅仅需要传递{A21A8C43-1266-11D4-A324-0040F6D487D9}这个ID和一个有效的接口ID给CoCreateInstance函数来实例化我们的com服务。这个函数将会跟踪注册表设置,利用CLSID来发现创建组件需要的东西,一旦它创建了组件,它将加载类型库,以获取更多需要的信息。

对我们来说非常幸运,最后的5个注册入口项通过RegisterTypeLib函数可以完成。在DllRegisterServer中,我们通过一些列的注册表函数来设置前面5项键值。然后调用RegisterTypeLib。 DllUnregisterServer函数删除DllRegisterServer中的注册表项,然后调用UnRegisterTypeLib。注意不要完全删除HKEY_CLASSES_ROOT/CLSID/

实现 Unknown

AddRef_MC proc this_:DWORD
mov eax, this_
inc (MyComObject ptr [eax]).nRefCount
mov eax, (MyComObject ptr [eax]).nRefCount
ret         ; note we return the object count
AddRef_MC endp


Release_MC proc this_:DWORD
    mov eax, this_
    dec (MyComObject ptr [eax]).nRefCount
    mov eax, (MyComObject ptr [eax]).nRefCount
    .IF (eax == 0)
        ; the reference count has dropped to zero
        ; no one holds reference to the object
        ; so let's delete it
        invoke  CoTaskMemFree, this_
        dec MyCFObject.nRefCount
  xor eax, eax    ; clear eax (count = 0)
   .ENDIF
    ret         ; note we return the object count
Release_MC endp

MyCom自己的成员实现:
GetValue proc this_:DWORD, pval:DWORD
    mov eax, this_
    mov eax, (MyComObject ptr [eax]).nValue
    mov edx, pval
    mov [edx], eax
    xor eax, eax        ; return S_OK
    ret
GetValue endp

SetValue proc this_:DWORD, val:DWORD
    mov eax, this_
    mov edx, val
    mov (MyComObject ptr [eax]).nValue, edx
    xor eax, eax        ; return S_OK
    ret
SetValue endp

RaiseValue  PROC this_:DWORD, val:DWORD
    mov eax, this_
    mov edx, val
    add (MyComObject ptr [eax]).nValue, edx
    xor eax, eax        ; return S_OK    
    ret
RaiseValue  ENDP


MyCom.dll, 这个com服务工程需要以下5个文件来编译:

MyCom.asm  汇编源程序

MyCom.idl  IDL文件,用于编译产生MyCom.tlb 

MyCom.tlb  类型库,需要一个rc资源文件

rsrc.rc    资源文件,从中可以获得类型库信息

MyCom.DEF  标准的dll输出文件

编译后,代码不会做任何事情,直到我们注册它,我们可以使用命令行:
regsvr32 MyCom.dll注册。