博客园_飘遥的Blog

C#编码Email地址，防火防盗防垃圾邮件

2010-11-17T14:35:00Z

很多热心的博友为方便帮助他人，在博客上留下自己Email地址。为防止垃圾邮件，把邮件地址以图片的形式或以文本将@符号替换为(at)或#等形式发布。这样难免给他人带来一点不便。在浏览衣兄博客时，发现其邮件链接有些有趣，稍研究了一下，原来是将邮件地址拆分并进行了编码，看起来根本就不是邮件地址。这样就可以越过那些"邮件地址采集器"。在实验中发现将mailto:进行编码，Outlook也能识别，其他邮件客户端没有进行测试。
花了一袋烟的功夫写了个小工具。思路是：将邮件地址拼接成“mailto:youremail@domain.com?Subject=”的形式，然后随机拆分该字符串，编码后赋值到javascript变量。最后将javascript变量相加。编码方式是：将单字节字符编码为%XX的形式，(unicode)双字节字符编码为\uXXXX形式。具体实现见代码。
生成的效果见公告栏。

多少粘点代码吧(汗...)

// Divide the string to some parts.
protected List<string> PartString(string input)
{
    List<string> list = new List<string>();
    Random rnd = new Random();
    int len = input.Length;
    int idx = 0;

    while (len > 0)
    {
        int tmp = rnd.Next(1, len + 1);

        list.Add(input.Substring(idx, tmp));

        idx += tmp;
        len -= tmp;
    }

    return list;
}

// Encode string. ASCII: %xx format; double bytes: \uxxxx format.
protected string EncodeString(string input)
{
    StringBuilder output = new StringBuilder(input.Length * 4);

    foreach (char c in input)
    {
        int charValue = (int)c;

        if (charValue > 255)
        {
            output.AppendFormat("\\u{0:x4}", charValue);
        }
        else
        {
            output.AppendFormat("%{0:x2}", charValue);
        }
    }

    return output.ToString();
}

// Generate unique variable name.
protected string GenerateVarName()
{
    Random rnd = new Random();
    string output = ((char)((int)'a' + rnd.Next(0, 26))).ToString() + rnd.Next(0, 10);

    while (vars.ContainsKey(output))
    {
        output = ((char)((int)'a' + rnd.Next(0, 26))).ToString() + rnd.Next(0, 10);
    }

    return output;
}

演示地址：http://www.xianfen.net/Article148.aspx
代码下载：点这里

本文链接

C#调用BITS(Background Intellgent Transfer Service)

2010-09-23T15:20:00Z

BITS(Background Intellgent Transfer Service)是微软推出的用来实现Client和Server之间进行文件传输的技术。微软只提供了基于COM的接口，而没有提供基于托管代码的封装。如果想用.NET上语言(如C#)需要我们手工对COM接口进行封装或使用第三方的封装，BITS有多个版本，找到合适的比较麻烦。MS没有提供TLB文件，也没把TLB文件嵌入到实现BITS的dll中，也就不能用tlbimp.exe工具生成COM的包装。如果有TLB文件，则可以用tlbimp.exe生成COM的包装程序集。

我在查看文件Windows SDK的时候，发现里面包含了各个版本的BITS的idl文件，midl.exe编译idl文件，发现生成TLB文件，tlbimp.exe发现能生成包装程序集，但有个问题，无论用哪个版本的idl文件，都只能生成BITS 1.0的COM接口的包装程序集；并且如果用bits4_0.idl生成的包装类，在创建BackgroundCopyManager4_0Class对象时，在两台Win7机器上实验，都出现异常，异常信息为：
Retrieving the COM class factory for component with CLSID {BB6DF56B-CACE-11DC-9992-0019B93A3A84} failed due to the following error: 80080005.
暂时没找到解决办法，我十分怀疑是MS的BUG。

4.0的不能用，就用3.0的，可只生成1.0的包装程序集不行啊。下面对bits3_0.idl文件进行手术进行手术：

library BackgroundCopyManager3_0
{
    [
        uuid(659cdea7-489e-11d9-a9cd-000d56965251),
        helpstring("Background copy manager 3.0")
    ]
    coclass BackgroundCopyManager3_0
    {
        [default] interface IBackgroundCopyManager;
    };
    interface IBackgroundCopyJob4;

    interface IBackgroundCopyJobHttpOptions;
    interface IBackgroundCopyCallback2;
    interface IBackgroundCopyFile3;
    interface IBitsPeerCacheAdministration;
}

粗体部分是我们手工添加的，目的是生成TLB文件时，将粗体部分的interface包含进去，tlbimp.exe对这些接口进行包装。
命令行为：

midl bits3_0.idl
tlbimp bits3_0.tlb

最终生成BackgroundCopyManager3_0.dll文件，在.NET项目中添加引用即可。如果要从低版本的接口获得高版本的接口，需要低版本的接口调用QueryInterface，但对COM接口的包装会屏蔽引用计数及QueryInterface。翻一翻System.Runtime.InteropServices命名空间会发现Marshal类中有个QueryInterface的方法：

public static int QueryInterface(IntPtr pUnk, ref Guid iid, out IntPtr ppv);

这个问题解决。

写代码测试一下：

using System;
using System.Runtime.InteropServices;
using BackgroundCopyManager3_0;

namespace www.Xianfen.Net
{
    class Program
    {
        static void Main(string[] args)
        {
            BackgroundCopyManager3_0Class mgr = new BackgroundCopyManager3_0Class();
            GUID guid;
            IBackgroundCopyJob job;

            //create job
            mgr.CreateJob("job", BG_JOB_TYPE.BG_JOB_TYPE_DOWNLOAD, out guid, out job);

            //新版本interface的GUID.
            Guid gJob4 = new Guid("659cdeae-489e-11d9-a9cd-000d56965251");

            //获取老版本interface的指针
            IntPtr pUnknwJob = Marshal.GetIUnknownForObject(job);
            IntPtr pUnknwJob4;

            //获取新版本interface的指针
            int hresult = Marshal.QueryInterface(pUnknwJob, ref gJob4, out pUnknwJob4);

            //获取新版本interface实例
            IBackgroundCopyJob4 job4 = (IBackgroundCopyJob4)Marshal.GetObjectForIUnknown(pUnknwJob4);

            job4.AddFile("http://www.xianfen.net/GRMSDKX_EN_DVD.iso", "C:\\WinSDK.iso");
            job4.SetPriority(BG_JOB_PRIORITY.BG_JOB_PRIORITY_FOREGROUND);
            //各种操作。。。
        }
    }
}

下载：修改后的bits3_0.idl、TLB文件及程序集下载

本文链接

发布一个注册类型库(TypeLib)的小工具

2010-09-23T00:57:00Z

用于在注册表中注册和反注册TypeLib。使用方法：可以用RegTypeLib.exe /? 查看帮助。

RegTypeLib v1.0 (c) Zhenxing Zhou

A tool to register TypeLib or unregister TypeLib
Using RegTypeLib.exe /? to show help
Using RegTypeLib.exe /r <TypeLib file path> to register TypeLib
Using RegTypeLib.exe /u <TypeLib guid> to unregister TypeLib
Example:
RegTypeLib.exe /r D:\a.tlb
RegTypeLib.exe /u {bb6df56e-cace-11dc-9992-0019b93a3a84}

源码如下：

#include <Windows.h>
#include <stdio.h>

void ShowErrorMessage();
void ShowUsage();

int wmain(int argc, WCHAR* argv[])
{
    if(argc < 2)
    {
        ShowErrorMessage();
        return 1;
    }

    if(wcslen(argv[1]) != 2)
    {
        ShowErrorMessage();
        return 1;
    }

    if(argv[1][0] != L'/' && argv[1][0] != L'-')
    {
        ShowErrorMessage();
        return 1;
    }

    if(argv[1][1] == L'?')
    {
        ShowUsage();
        return 0;
    }

    if(argc < 3)
    {
        ShowErrorMessage();
        return 1;
    }

    if(tolower(argv[1][1]) == L'r')
    {
        HANDLE hFile = CreateFile(argv[2], 0, 0, NULL, OPEN_EXISTING, 0, NULL);

        if(hFile == INVALID_HANDLE_VALUE)
        {
            wprintf(L"The file %s does not exists!\r\n", argv[2]);
            return 1;
        }

        CloseHandle(hFile);

        HRESULT hr = S_OK;
        ITypeLib *pTypeLib;

        hr = LoadTypeLibEx(argv[2], REGKIND_REGISTER, &pTypeLib);

        if(FAILED(hr))
        {
            wprintf(L"Register TypeLib failed. Error Code: %lu", hr);
            return 1;
        }
        else
        {
            wprintf(L"Register TypeLib succeed.");
            pTypeLib->Release();
        }
    }
    else if(tolower(argv[1][1]) == L'u')
    {
        HRESULT hr = S_OK;
        GUID guid;

        hr = CLSIDFromString(argv[2], &guid);

        if(FAILED(hr))
        {
            wprintf(L"Can't convert %s to GUID. Error Code: %lu", argv[2], hr);
        }
        else
        {
            hr = UnRegisterTypeLib(guid, 1, 0, LANG_NEUTRAL, SYS_WIN32);

            if(FAILED(hr))
            {
                wprintf(L"Can't unregister the TypeLib. Error Code: %lu", hr);
                return 1;
            }
            else
            {
                wprintf(L"Unregister TypeLib %s successfully.", argv[2]);
            }
        }
    }
    else
    {
        ShowErrorMessage();
        return 1;
    }

    return 0;
}

void ShowErrorMessage()
{
    wprintf(L"The syntax of the command is incorrect.\r\n\r\n");

    ShowUsage();
}

void ShowUsage()
{
    wprintf(L"RegTypeLib v1.0 (c) Zhenxing Zhou\r\n\r\n");
    wprintf(L"A tool to register TypeLib or unregister TypeLib\r\n");
    wprintf(L"Using RegTypeLib.exe /? to show help\r\n");
    wprintf(L"Using RegTypeLib.exe /r <TypeLib file path> to register TypeLib\r\n");
    wprintf(L"Using RegTypeLib.exe /u <TypeLib guid> to unregister TypeLib\r\n");
    wprintf(L"Example:\r\n");
    wprintf(L"\tRegTypeLib.exe /r D:\\a.tlb\r\n");
    wprintf(L"\tRegTypeLib.exe /u {bb6df56e-cace-11dc-9992-0019b93a3a84}\r\n");
}

下载：下载源文件及可执行文件

本文链接

【软件发布】发布一个查单词的小工具

2010-09-20T06:22:00Z

之前看海量的英文文档，英语水平有限，需要经常查单词，用了几个工具，发现窗口切换来切换去太麻烦，鼠标取词总是烦我，于是自己做了个超简洁的。原则是：需要时输入即查，不用时别烦我，别让我切换窗口！OK，小作品诞生了。(传给朋友用，朋友发现还不错，就放到网上了。我也放出来吧，毕竟是官方的^_^)。

大小有30KB，绿色免安装。界面如下：

这个小窗口总是显示在所有窗口上方，其位置可以随意放置。

查询单词后，显示结果列表：

查询结果来自Dict.cn，支持单词发音，例句朗读等。
支持更换皮肤颜色：

支持透明度选择：

运行环境：Vista以上系统，否则需要安装.NET Framework2.0。
下载地址：MiniDict.zip

本文链接

COM 入门(4)

2010-09-01T01:41:00Z

本篇介绍COM自动化，对脚本的支持。

1. 实现IDispatch接口
COM自动化是指实现了IDispatch接口，这样可以供VBScript，JScript能脚本调用。
实现IDispatch接口的对象，对应到VBScript和JScript中的Object。
IDispatch接口定义为：

interface IDispatch : IUnknown
{
    HRESULT GetTypeInfoCount(
        [out] UINT* pCountTypeInfo);

    HRESULT GetTypeInfo(
        [in] UINT iTInfo,
        [in] LCID lcid,
        [out] ITypeInfo** ppTInfo);

    HRESULT GetIDsOfNames(
        [in] REFIID riid,
        [in, size_is(cNames)] LPOLESTR* rgszNames,
        [in] UINT cNames,
        [in] LCID lcid,
        [out, size_is(cNames)] DISPID* rgDispId);

    HRESULT Invoke(
        [in] DISPID dispIdMember,
        [in] REFIID riid,
        [in] LCID lcid,
        [in] WORD wFlags,
        [in, out] DISPPARAMS* pDispParams,
        [out] VARIANT* pVarResult,
        [out] EXCEPINFO* pExcepInfo,
        [out] UINT* puArgErr);
};

方法数量不多，但很复杂，特别是Invoke方法，不过微软提供了一些辅助方法来简化实现过程。VARIANT在C/C++中的定义是一个结构体，比较复杂，具体内容可参考Windows SDK。

idl中提供了dispinterface关键字实现IDispatch接口，因为功能限制的原因不推荐使用，推荐的是实现双接口(Dual Interface)，既可以通过用户定义的接口也可以通过IDispatch接口访问对象。
BeginningCOM示例中，IDL接口的定义：

[object, uuid(93C3840F-AD5A-4020-AAAB-313C4B61B184), dual]
interface IBeginningCOM : IDispatch
{
    [id(1)] HRESULT Sum([in] int a, [in] int b, [out, retval] int *sum);
    [id(2), propget] HRESULT Num([out, retval] int *pVal);
    [id(2), propput] HRESULT Num([in] int val);
}

dual标记该接口为双接口，id为成员的ID。

GetTypeInfoCount方法获取对象提供的类型信息的数量，如果对象提供类型信息，pCountTypeInfo设为1，否则为0。

STDMETHODIMP BeginningCOM::GetTypeInfoCount(UINT* pCountTypeInfo)
{
*pCountTypeInfo = 1;
return S_OK;
}

GetTypeInfo获取类型信息，我们直接用API取的其信息返回即可。
取得类型信息作为类的成员变量：

ITypeLib *pTypeLib;
HRESULT hr = S_OK;

hr = LoadRegTypeLib(LIBID_BEGINNINGCOMLib, 1, 0, LANG_NEUTRAL, &pTypeLib);

if(SUCCEEDED(hr))
{
pTypeLib->GetTypeInfoOfGuid(IID_IBeginningCOM, &m_pTypeInfo);
//...

GetTypeInfo的实现：

STDMETHODIMP BeginningCOM::GetTypeInfo(UINT iTypeInfo, LCID lcid,
    ITypeInfo** ppITypeInfo)
{
    if(*ppITypeInfo != NULL)
    {
        return DISP_E_BADINDEX;
    }

    m_pTypeInfo->AddRef();
    *ppITypeInfo = m_pTypeInfo;

    return S_OK;
}

GetIDsOfNames函数获取成员的ID，上面取得了TypeInfo，那么调用DispGetIDsOfNames可根据成员名取得其ID，其实现为：

STDMETHODIMP BeginningCOM::GetIDsOfNames(REFIID riid,
    LPOLESTR* rgszNames, UINT cNames, LCID lcid,
    DISPID* rgDispId)
{
    if(riid != IID_NULL)
    {
        return DISP_E_UNKNOWNINTERFACE;
    }

    return DispGetIDsOfNames(m_pTypeInfo, rgszNames, cNames, rgDispId);
}

riid不使用，并且必须为IID_NULL。

Invoke函数用于调用类的成员函数。

STDMETHODIMP BeginningCOM::Invoke(DISPID dispIdMember, REFIID riid,
    LCID lcid, WORD wFlags, DISPPARAMS* pDispParams,
    VARIANT* pVarResult, EXCEPINFO* pExcepInfo,
    UINT* puArgErr)
{
    if(riid != IID_NULL)
    {
        return DISP_E_UNKNOWNINTERFACE;
    }

    return DispInvoke(this, m_pTypeInfo, dispIdMember, wFlags, pDispParams, pVarResult, pExcepInfo, puArgErr);
}

2. 脚本语言调用COM组件
JS:

var obj = new ActiveXObject("BeginningCOM.1");
WScript.Echo(obj.Sum(3,6));

VBS:

dim obj
set obj = CreateObject("BeginningCOM.1")
msgbox (obj.Sum(5,6))

HTML:

HTML在浏览器中运行中会出现不安全控件的提示，下一篇提供解决方法。

源文件下载：BeginningCOM4.zip

本文链接

COM 入门(3)

2010-08-31T14:41:00Z

本篇介绍COM注册TypeLib，CLR程序调用COM组件；注册ProgID及ProgID和CLSID相互映射等。

1. 注册TypeLib
IDL在MIDL编译后会生成一个.tlb文件，这个文件包含TypeLibrary信息，为使我们写的COM组件供.NET或VB等程序调用，注册类型库是必须的。我们可以直接注册生成的tlb文件，但为了便于部署，可以将tlb文件作为资源嵌dll中。
新建一个rc文件，添加如下代码：

1 TypeLib "Debug\\BeginningCOM.tlb"

这里的TypeLib的ID通常为1，注册TypeLib的时候直接使用dll路径及文件名即可，否则需要在路径文件名后加上"\TypeLib在rc文件中的ID"。
注册TypeLib可以使用直接写注册表的方式和调用API注册，建议使用后者，因为调用API时会对TypeLib的合法性进行验证。

//register typelib
ITypeLib *pTypeLib;
hr = LoadTypeLibEx(szFileName, REGKIND_REGISTER, &pTypeLib);
pTypeLib -> Release();

//unregister type
UnRegisterTypeLib(LIBID_BEGINNINGCOMLib, 1, 0, LANG_NEUTRAL, SYS_WIN32);

LoadTypeLibEx函数会加载COM组件，取得ITypeLib接口实例并注册TypeLib。
LIBID_BEGINNINGCOMLib是MIDL编译idl文件时从idl文件获得的library的ID。

COM类型库注册后会在注册表中写入如下信息:

[HKEY_CLASSES_ROOT\TypeLib\{D9161D4D-66C0-4AE6-9264-C322BDE034C7}\1.0]
@="BeginningCOMLib"

[HKEY_CLASSES_ROOT\TypeLib\{D9161D4D-66C0-4AE6-9264-C322BDE034C7}\1.0\0\win32]
@="E:\\Projects\\BeginningCOM\\Debug\\BeginningCOM.dll"
//...

2. CLR调用COM组件
CLR调用COM组件可以用tlbimp工具生成代理类，也可以直接在VS中添加引用，这里我们仅实验后者。
VS中新建一个Console Application，添加引用，COM列表中，Componment Name列中列出了我们刚刚创建的COM组建，添加BeginningCOMLib类型库。然后编写代码：

static void Main(string[] args)
{
    BeginningCOM bc = new BeginningCOM();
    Console.WriteLine(bc.Sum(1, 2));

    bc.Num = int.MaxValue;
    Console.WriteLine(bc.Num);

    Console.ReadKey();
}

运行，输出了我们预期的结果。64位系统要注意，如果COM组件编译为x86平台，.NET程序也要编译为x86平台，否则会抛出类没有注册的错误。

3. 注册ProgID
ProgID和CLSID基本上是一一对应的，CLSID是GUID形式，能保证其唯一性；而ProgID比较易读易记，并且脚本语言一般用ProgID来调用COM组件。
注册ProgID只需在注册表中写入一些键值即可，ProgID习惯的命名方式是组件.版本。为注册ProgID，RegTable数组的内容如下：

可以使用ProgIDFromCLSID函数，将CLSID转换为ProgID；使用CLSIDFromProgID将ProgID转换为CLSID，具体使用见MSDN。

COM有关的注册表内容:http://msdn.microsoft.com/en-us/library/ms678477(VS.85).aspx

源文件下载：BeginningCOM3.zip

本文链接

COM 入门(2)

2010-08-28T15:46:00Z

上一篇文章简单演示了COM组件的编写，注册及调用，本篇实现COM的自注册和反注册，定义一个有实际功能的接口及实现IClassFactory接口。

1. 实现COM组件的自注册和反注册
实现COM组件的自注册和反注册，本质上就是写注册表与删注册表。需要在DLL中引出两个函数：DllRegisterServer和DllUnregisterServer，让这两个函数实现注册表操作。

LPCTSTR RegTable[][3] =
{
    {L"CLSID\\{586CDC7B-09F1-4f44-A110-F0E604AED81E}", 0, L"BeginningCOM"},
    {L"CLSID\\{586CDC7B-09F1-4f44-A110-F0E604AED81E}\\InprocServer32", 0, (LPCTSTR)-1},
    {L"CLSID\\{586CDC7B-09F1-4f44-A110-F0E604AED81E}\\InprocServer32", L"ThreadingModel", L"Both"}
};

STDAPI DllUnregisterServer()
{
    HRESULT hr = S_OK;

    int regCount = sizeof(RegTable) / sizeof(*RegTable);

    for(int i = regCount - 1; i >= 0; i--)
    {
        LSTATUS error = RegDeleteKey(HKEY_CLASSES_ROOT, RegTable[i][0]);

        if(error != ERROR_SUCCESS)
        {
            hr = S_FALSE;
        }
    }

    return hr;
}

STDAPI DllRegisterServer(void)
{
    HRESULT hr = S_OK;
    TCHAR szFileName[MAX_PATH];

    ZeroMemory(&szFileName, MAX_PATH * sizeof(TCHAR));

    GetModuleFileName(g_hModule, szFileName, MAX_PATH);

    int regCount = sizeof(RegTable) / sizeof(*RegTable);

    for(int i = 0; i < regCount; i++)
    {
        if(RegTable[i][2] == (LPCTSTR)-1)
        {
            RegTable[i][2] = szFileName;
        }

        HKEY hKey;
        LSTATUS error = ::RegCreateKey(HKEY_CLASSES_ROOT, RegTable[i][0], &hKey);

        if(error == ERROR_SUCCESS)
        {
            error = RegSetValueEx(hKey, RegTable[i][1], 0, REG_SZ, (const BYTE*)RegTable[i][2], (lstrlen(RegTable[i][2]) + 1) * sizeof(TCHAR));
            RegCloseKey(hKey);
        }

        if(error != ERROR_SUCCESS)
        {
            DllUnregisterServer();
        }
    }

    return hr;
}

现将需要修改的注册表内容放到一个二维数组中，第一维内容分别是注册表键，项，值。-1为占位符，标识该位置放置COM组件的路径，在写注册表之前将其替换。
删注册表时，应该先删子键，否则删除会失败。
同时需要在BeginningCOM.def中导出这两个函数：

LIBRARY    "BeginningCOM"
EXPORTS
    DllGetClassObject private
    DllRegisterServer private
    DllUnregisterServer private

这样就可以用regsvr32.exe注册和反注册COM组件。

2. 自定义接口
可以用IDL文件定义，也可以在代码中直接继承IUnknown接口实现。用IDL定义接口的好处是，MIDL工具能自动生成GUID描述，接口存根/代理等。IDL全称为：Interface Definition Language，其语法类似于C语言，去掉了存在二义性的内容，并进行了扩展。作为示例，我们定义一个IBeginningCOM的接口，里面添加一个Sum函数和一个Num属性。

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

[object, uuid(93C3840F-AD5A-4020-AAAB-313C4B61B184)]
interface IBeginningCOM : IUnknown
{
    HRESULT Sum([in] int a, [in] int b, [out, retval] int *sum);
    [propget] HRESULT Num([out, retval] int *pVal);
    [propput] HRESULT Num([in] int val);
}

[
    uuid(D9161D4D-66C0-4ae6-9264-C322BDE034C7),
    version(1.0),
    helpstring("BeginningCOMLib")
]
library BEGINNINGCOMLib
{
    importlib("stdole32.tlb");
    importlib("stdole2.tlb");

    [
        uuid(586CDC7B-09F1-4f44-A110-F0E604AED81E),
        helpstring("BeginningCOM Lib")
    ]
    coclass BeginningCOM
    {
        [default] interface IBeginningCOM;
    };
};

引入的oaidl.idl等包含了系统标准接口声明等，直接引入即可。接口的object和接口的名字这两个特性是必须的，为防止接口重名，用GUID来表示接口的名字。propget和proppub表示方法映射到属性，如VB中，在不支持属性的语言，如C++，映射为get_XXX，put_XXX。每个IDL文件只能有一个library标识，内部可以有多个coclass。
IDL文件经MIDL工具编译后，生成四个文件：
*_h.h        接口说明文件
*_i.c        GUID描述文件
*_p.c        接口存根/代理实现文件
dlldata.c    包含存根/代理相关内容

//BeginningCOM继承自IBeginningCOM接口
//...
class BeginningCOM : public IBeginningCOM
//...

//实现IBeginningCOM接口成员，属性需要分别实现set/put。
//...
STDMETHODIMP BeginningCOM::Sum(int a, int b, int *sum)
{
    *sum = a + b;
    return S_OK;
}

STDMETHODIMP BeginningCOM::get_Num(int *pVal)
{
    *pVal = m_Num;
    return S_OK;
}

STDMETHODIMP BeginningCOM::put_Num(int val)
{
    m_Num = val;
    return S_OK;
}

3. 实现IClassFactory接口
IClassFactory是系统定义的对象创建的标准接口。IClassFactory有两个方法：LockServer和CreateInstance。LockServer用于进程外激活时，COM内部对其进行调用，目前我们都是进程内激活，可先不实现；CreateInstance用于创建请求的类对象，该方法的第一个参数在聚合时使用。
实现IClassFactory：

#pragma once
#include "unknwn.h"

class BeginningCOMClassFactory :
    public IClassFactory
{
public:
    BeginningCOMClassFactory(VOID);    //IUknown
    STDMETHODIMP QueryInterface(REFIID riid, VOID** ppv);
    STDMETHODIMP_(ULONG) AddRef(VOID);
    STDMETHODIMP_(ULONG) Release(VOID);

    //IClassFactory
    STDMETHODIMP CreateInstance(LPUNKNOWN punkOuter, REFIID riid, VOID** ppv);
    STDMETHODIMP LockServer(BOOL fLock);

protected:
    ULONG m_ulRefCount;
};

//...
STDMETHODIMP BeginningCOMClassFactory::CreateInstance(LPUNKNOWN punkOuter, REFIID riid, VOID** ppv)
{
    BeginningCOM *pbc;
    HRESULT hr;

    //暂不支持聚合
    if(punkOuter != NULL)
    {
        return CLASS_E_NOAGGREGATION;
    }

    pbc = new BeginningCOM;

    if(pbc == NULL)
    {
        return E_OUTOFMEMORY;
    }

    pbc -> AddRef();
    hr = pbc -> QueryInterface(riid, ppv);
    pbc -> Release();

    return hr;
}

STDMETHODIMP BeginningCOMClassFactory::LockServer(BOOL fLock)
{
    return E_FAIL;
}

DllGetClassObject返回是否实现IClassFactory即可。
STDAPI DllGetClassObject(REFCLSID rclsid, REFIID riid, void **ppv)
{
    if(rclsid == CLSID_BeginningCOM)
    {
        BeginningCOMClassFactory *pbc = new BeginningCOMClassFactory;

        if(pbc == NULL)
        {
            return E_OUTOFMEMORY;
        }

        return pbc -> QueryInterface(riid, ppv);
    }

    *ppv = 0;

    return CLASS_E_CLASSNOTAVAILABLE;
}

实现了IClassFactory接口，创建COM对象的代码可以通过查询IClassFactory接口，然后调用该接口的CreateInstance方法：

HRESULT hr = NULL;
IClassFactory *pcf;
IBeginningCOM *pbc;

hr = CoGetClassObject(CLSID_BeginningCOM, CLSCTX_INPROC_SERVER, NULL, IID_IClassFactory, (void**)&pcf);

if(SUCCEEDED(hr))
{
hr = pcf -> CreateInstance(NULL, IID_IBeginningCOM, (void**)&pbc);
//...

也可以用CoCreateInstance创建，该函数包装了上面的两布，在分布式环境下能减少一次客户端与服务器之间的通信。

//...
hr = CoCreateInstance(CLSID_BeginningCOM, NULL, CLSCTX_INPROC_SERVER, IID_IBeginningCOM, (void**)&pbc);
//...

源程序下载：BeginningCOM2.zip

本文链接

COM 入门(1)

2010-08-28T10:00:00Z

最近需要用到COM知识，总结一下刚学到的知识。

1. 什么是COM
COM全称是Component Object Model，中文译为组件对象模型。COM组件在物理上是一些DLL或EXE文件；COM组件实现二进制级别的代码重用；COM是与程序设计语言无关，理论上任何语言都可以开发和调用COM组件；COM组件用引用计数实现生命周期的自我管理；COM组件调用者能够查询它所支持的接口；COM组件的位置对调用者是透明的；COM组件依赖于注册表；COM组件都要直接或间接的实现IUnknown接口……

2. IUnknown接口
所有COM组件都直接或间接实现IUnknown接口，IUnknown接口是COM的根接口，其声明为：

interface IUnknown
{
    virtual STDMETHODIMP QueryInterface(REFIID riid, VOID** ppv) = 0;
    virtual STDMETHODIMP_(ULONG) AddRef(VOID) = 0;
    virtual STDMETHODIMP_(ULONG) = 0;
};

标准C++中没有interface关键字，这里的interface是typedef struct interface。
ULONG，VOID等都是windows对C/C++中标准数据类型的宏定义，目的是屏蔽不同平台的差异，并且使编码风格统一。
STDMETHODIMP等价于HRESULT __stdcall
STDMETHODIMP_(DataType)等价于DataType __stdcall

COM规定，调用函数的方式必须为__stdcall，这是pascal语言缺省的调用函数的方式：函数的参数从右到左依次压栈，函数退出时自己清空占用的堆栈。与之相对的为__cdecl，这是C/C++语言缺省的调用函数的方式：函数的参数从右到左依次压栈，函数退出时，由调用者清空函数占用的堆栈。

QueryInterface函数，客户通过该函数查询COM实现的接口；riid是接口的标识，为GUID形式；返回值标识查询的接口是否实现，如果实现了，则返回S_OK，并且ppv指向接口的实例，否则返回E_NOINTERFACE，ppv指向的内容无效。QueryInterface函数隔离了不同编程语言构造对象实例的差异。
AddRef和Release函数实现了COM对象生命周期的自我管理。实现该接口的类需要有一个ULONG型成员记录其实例的引用计数，如果AddRef一次，引用计数加1，否则引用计数减1，如果引用计数为0时，就释放该实例。这两个函数隔离了不同变成语言释放对象实例的差异。

3. 实现一个最最简单的COM组件

//BeginningCOM.h
#ifndef __BEGINNINGCOM_H__
#define __BEGINNINGCOM_H__

#pragma once
#include <windows.h>

// {7BB69A25-68E4-427a-BE4B-B06ED17531AA}
CLSID CLSID_BeginningCOM =
{ 0x7bb69a25, 0x68e4, 0x427a, { 0xbe, 0x4b, 0xb0, 0x6e, 0xd1, 0x75, 0x31, 0xaa } };

#endif //__BEGINNINGCOM_H__

//BeginningCOM.cpp
#include "BeginningCOM.h"

class BeginningCOM : public IUnknown
{
public:
    BeginningCOM(VOID);
    STDMETHODIMP QueryInterface(REFIID riid, VOID** ppv);
    STDMETHODIMP_(ULONG) AddRef(VOID);
    STDMETHODIMP_(ULONG) Release(VOID);

protected:
    ULONG m_ulRefCount;
};

BeginningCOM::BeginningCOM(VOID) : m_ulRefCount(0)
{
}

STDMETHODIMP BeginningCOM::QueryInterface(REFIID riid, VOID** ppv)
{
    if(riid == IID_IUnknown)
    {
        *ppv = static_cast<IUnknown*>(this);
    }
    else
    {
        *ppv = NULL;

        return E_NOINTERFACE;
    }

    reinterpret_cast<IUnknown*>(*ppv) -> AddRef();
    return S_OK;
}

STDMETHODIMP_(ULONG) BeginningCOM::AddRef(VOID)
{
    return InterlockedIncrement(&m_ulRefCount);
}

STDMETHODIMP_(ULONG) BeginningCOM::Release(VOID)
{
    ULONG tmp = InterlockedDecrement(&m_ulRefCount);

    if(tmp == 0)
    {
        delete this;
    }

    return tmp;
}

// DLL entry point.
BOOL APIENTRY DllMain( HMODULE hModule,
                      DWORD  ul_reason_for_call,
                      LPVOID lpReserved
                      )
{
    switch (ul_reason_for_call)
    {
    case DLL_PROCESS_ATTACH:
    case DLL_THREAD_ATTACH:
    case DLL_THREAD_DETACH:
    case DLL_PROCESS_DETACH:
        break;
    }
    return TRUE;
}

STDAPI DllGetClassObject(REFCLSID rclsid, REFIID riid, void **ppv)
{
    if(rclsid == CLSID_BeginningCOM)
    {
        BeginningCOM *pbc = new BeginningCOM;

        if(pbc == NULL)
        {
            return E_OUTOFMEMORY;
        }

        return pbc -> QueryInterface(riid, ppv);
    }

    *ppv = 0;

    return CLASS_E_CLASSNOTAVAILABLE;
}

BeginningCOM.h中，先定义一个GUID作为实现接口的class的ID。BeginningCOM.cpp中实现IUnknown接口。
DllGetClassObject是客户调用COM组件的入口，需要导出这个函数：

//BeginningCOM.def
LIBRARY "BeginningCOM"
EXPORTS
DllGetClassObject private

4. 注册COM组件
在HKEY_CLASSES_ROOT\CLSID键下注册COM组件的class信息。创建一个reg文件，导入注册表即可。

//regsvr.reg
Windows Registry Editor Version 5.00

[HKEY_CLASSES_ROOT\CLSID\{7BB69A25-68E4-427a-BE4B-B06ED17531AA}]
@="BeginningCOM"

[HKEY_CLASSES_ROOT\CLSID\{7BB69A25-68E4-427a-BE4B-B06ED17531AA}\InprocServer32]
@="E:\\Projects\\BeginningCOM\\Debug\\BeginningCOM.dll"
"ThreadingModel"="Both"

@="BeginningCOM"是class的描述信息，可忽略。
"ThreadingModel"="Both"是COM组件的套件类型，后面介绍。

相应的卸载COM组件的reg文件内容应该为：

//unregsvr.reg
Windows Registry Editor Version 5.00

[-HKEY_CLASSES_ROOT\CLSID\{7BB69A25-68E4-427a-BE4B-B06ED17531AA}]
@="BeginningCOM"

需要注意的是，在Win64中，如果COM组件编译选择的是X86，那么注册表会进行重定向，读取位置变为HKEY_CLASSES_ROOT\Wow6432Node\CLSID，因此，注册时，键值应该写在这个路径下。见《Win64 注册表重定机制向导致程序运行异常》

5. 调用COM组件
要调用COM组件，必须向客户公布接口ID，class ID等，我们这里把需要公布的信息放到BeginningCOM.h中。
下面的代码用于调用刚创建的COM组件。

#include <Windows.h>
#include <tchar.h>
#include <iostream>
#include "../BeginningCOM/BeginningCOM.h"
using namespace std;

int _tmain(int argc, _TCHAR* argv[])
{
    CoInitialize(NULL);

    HRESULT hr = NULL;
    IUnknown *puk;

    hr = CoGetClassObject(CLSID_BeginningCOM, CLSCTX_INPROC_SERVER, NULL, IID_IUnknown, (void**)&puk);

    if(SUCCEEDED(hr))
    {
        //do nothing
        puk -> Release();
    }
    else
    {
        cout << "Failed to create object"<<endl;
    }

    CoUninitialize();

    return 0;
}

在使用COM组件前，需要初始化COM调用环境，CoInitialize初始化单线程套间，只有一个保留的参数，CoInitializeEx除了一个保留的参数，还可以指定初始化的套件类型。

Author: Zhenxing Zhou
Blog: http://www.xianfen.net/

本文链接

C#内嵌汇编代码的讨论

2010-06-22T09:08:00Z

很久之前整理了一篇《C# 调用非托管程序》文章，在博客园zhongzf同学《在.net程序中嵌入asm汇编代码》进行了简单的讨论，现在才有时间整理。
《C# 调用非托管程序》最后一种方法通俗的讲是构造符合汇编代码(机器代码)格式的数据，把该数据当作可执行代码执行。Windows提供了DEP(Data Execution Prevention 数据执行保护)机制，也就是Windows会试图阻止程序运行非可执行内存区域的可执行代码。如果开启DEP，《C# 调用非托管程序》一文中最后一种方法执行会失败；如果关闭DEP，Windows XP SP2中执行该代码会成功，但Vista/Win7由于安全性增强的原因，该代码执行会失败。
是不是这用嵌入汇编代码的方式在开启DEP及Vista/Win7中一定不能使用呢？答案在下面分析中得出。
如果我们把保存汇编代码的内存区域标记为可执行，上面的方法也许可用。

Win API VirtualAlloc中有参数，可以指定新分配内存的权限。
使用完内存后，调用VirtualFree释放。

沿着这样的思路，将《C# 调用非托管程序》一文中最后一种方法修改如下(篇幅原因简化了注释)：

/*修改记录
    2008-5-11 8:07 曲滨
        >> 基本实现预期功能
        [!] 明天进行优化

    2008-5-12 15:54 曲滨
        [E] 优化完成
        [N] 加入 NativeCodeHelper 类便于使用

    2010-6-17 周振兴
        修改兼容性，可在开启DEP及Vista/Win7中运行。
*/

namespace NShellNativeCode
{
    using System;
    using System.Runtime.InteropServices;

    delegate int AddProc(int p1, int p2);
    class Program
    {
        static void Main(string[] args)
        {
            byte[] codeBytes = {
                  0x8B, 0x44, 0x24, 0x08    // mov eax,[esp+08h]
                , 0x8B, 0x4C, 0x24, 0x04    // mov ecx,[esp+04h]
                , 0x03, 0xC1                // add    eax,ecx
                , 0xC3                        // ret
                };

            /*
            上面的字节数组，就是下面函数的本机代码；
            int add(int x,int y) {
                return x+y;
            }

            */

            IntPtr handle = IntPtr.Zero;
            handle = VirtualAlloc(
                IntPtr.Zero,
                codeBytes.Length,
                MEM_COMMIT | MEM_RESERVE,
                PAGE_EXECUTE_READWRITE);

            try
            {

                Marshal.Copy(codeBytes, 0, handle, codeBytes.Length);

                AddProc add
                   = Marshal.GetDelegateForFunctionPointer(handle, typeof(AddProc)) as AddProc;

                int r = add(1976, 1);

                Console.WriteLine("本机代码返回:{0}", r);

            }
            finally
            {
                VirtualFree(handle, 0, MEM_RELEASE);
            }

            Console.ReadLine();
        }

        //Windows API
        [DllImport("Kernel32.dll", EntryPoint = "VirtualAlloc")]
        public static extern IntPtr VirtualAlloc(IntPtr address, int size, uint allocType, uint protect);

        [DllImport("Kernel32.dll", EntryPoint = "VirtualFree")]
        public static extern bool VirtualFree(IntPtr address, int size, uint freeType);

        //flags
        const uint MEM_COMMIT = 0x1000;
        const uint MEM_RESERVE = 0x2000;

        const uint PAGE_EXECUTE_READWRITE = 0x40;

        const uint MEM_RELEASE = 0x8000;
    }
}

事实证明，这种嵌入汇编代码的方式在开启DEP及Vista/Win7可运行。
Win7 (X86)开启DEP环境下测试通过。
注：codeBytes数组中是X86汇编代码，如果要在X64中运行，需修改该代码！

本文链接

仿Win7显示桌面的工具

2010-01-29T20:42:00Z

Win7改进了许多用户体验，其中右下角显示桌面的按钮是最舒心的功能之一。为了在没装Win7的开发机上体验这个功能，写了下面的小工具。在用户体验上，右下角是一个无限大的区域，因为鼠标被限制在屏幕范围内；与屏幕中某块孤立范围相比，操作的命中率大大提高。