Jason_liu

导航

进程间通讯-WriteProcessMemory和ReadProcessMemory

最近在写个程序的时候需要在进程间通讯,具体需求是这样。

1.       主要有两个进程:一个进程作为被请求进程,我们称为 SERVER 进程;另一个进程是请求进程,称为 CLIENG 进程。

2.       SERVER 进程提供一些服务,其完成计算功能;而 CLIENT 进程需要在它执行完计算之后将结果取会。

 

由于计算结果可能是一个结构,也可能是一个复杂的数据,所以通过消息来在进程传递信息是有限的。另一方面一般是单方向的通讯,实际上这里的需求有一个双向性,看下图:

 

 


这里两个进程都可以有自己的窗口,因此实际上我们可以通过消息来通知对方。但仔细一想,请求服务通过windows消息是没有问题的,通知结果通过消息是不妥当的,实际上我们需要在请求服务完以后立即得到执行结果,而使用windows消息可能有时间上的问题,而且同步非常麻烦。

想要的结果是在 CLIENT 请求服务以后立即得到返回结果,但我们可以改变一下思路就容易多了:结果不是有 SERVER 返回,而由 CLIENG 自己去获取。这样,我们可以在请求消息的时候使用 SendMessage 来发送这个请求。这是必须的, SendMessage 发送消息是同步的方式,必须等到消息处理完毕之后才返回,而这正是我们想要的结果。那么 CLIENT 怎么样才能获得 SERVER 进程的结果来。

我们知道, WINDOWS 下每个进程都有自己的地址空间,一般情况下一个进程访问你一个进程的地址是不正确的,最简单的是提示该地址无效,程序崩溃。当然还是有很多中方式来读取对方进程的地址空间上的数据。

1.  可以做一个 DLL ,利用注入 DLL 的方式,来将该 DLL 注入到远程进程的地址空间上,然后通过 DLL 的 API 来读取。这个方式有点麻烦,还必须写一个 DLL ,还要注入 DLL 。

2.  使用 WriteProcessMemory 和 ReadProcessMemory 来读写远程进程的内存。这种方式相对比较简单,其有一个参数远程进程的 HANDLE 指示你需要读写哪个进程的内存。当然,使用这两个函数是需要远程进程的地址空间的,这个地址是通过 VirtualAllocEx 来分配的,其通过VirtualFreeEx 来释放。这两个 API 也有一个进程句柄参数。

有人问,为什么不使用 WM_COPYDATA 来传递大块数据?好,该消息是可以传递大块数据,但我们的应用中需要消息的双向,所以这里使用WM_COPYDATA 是不合适的。

 

下面我们看一下第 2 种方法的具体步骤:

1.  找到对方进程的窗口。

2.  找到他的进程 ID

3.  使用 PROCESS_VM_OPERATION| PROCESS_VM_WRITE|PROCESS_VM_READ 标志来打开该进程,得到该继承的 HANDLE 。

4.  使用 VirtualAllocEx 来在该进程上分配适当大小的内存,得到一个地址,这个地址是远程进程的,通过不同的方式来修改该地址上的值是无效的。

5.  如果需要传递一些参数到远程进程,我们可以在该内存上写一些内容,通过 WriteProcessMemory 来完成

6.  使用 SendMessage 来发送请求服务消息,同时将上面分配的内存地址作为参数传递给远程进程。

7.  远程进程得到指定消息后处理该消息,取得参数,计算结果,将结果写到指定的地址。由于这个地址是远程进程自己的地址空间,其操作这块内存的方法没有什么特别之处。

8.  SendMessage 消息返回, CLIENT 知道后,从上面的内存地址中读取返回结果;这里必须使用 ReadProcessMemory 来读取。好了,整个过程结束。

9.  调用 VritualFreeEx 将上面的内存释放。

这里需要强调两点:

1.  打开进程的时候必须设置对虚拟内存可操作、可写、可读,如果只是可写,那么 ReadProcessMemory 将读取不正确。

2.  必须释放该内存。

 

下面是部分程序:

CLIENT 程序:

#define     WM_COMPAREIMAGE WM_USER +100

 

void CTestCompareDlg::OnBnClickedButton1()

{

    HANDLE hProcess = NULL;

    DWORD dwProcessId = 0;

    HWND hServerWnd = ::FindWindow(NULL,"CompareServer");

    if(hServerWnd == NULL)

    {

       //Need create the process

       return ;

    }

    ::GetWindowThreadProcessId(hServerWnd,&dwProcessId);

    hProcess = OpenProcess(PROCESS_VM_OPERATION|

       PROCESS_VM_WRITE|PROCESS_VM_READ,FALSE,dwProcessId);

    if(hProcess == NULL) return ;

 

    MyInfo * pMyInfo = NULL;

 

    pMyInfo = (MyInfo *)VirtualAllocEx(hProcess,NULL,

       sizeof(MyInfo),MEM_COMMIT,PAGE_READWRITE);

 

    if(pMyInfo == NULL) return ;

    MyInfo myInfo;

    myInfo.blue = 20.01;

    myInfo.red = 3333;

 

    WriteProcessMemory(hProcess,pMyInfo,&myInfo,sizeof(MyInfo),NULL);

 

    ::SendMessage(hServerWnd,WM_COMPAREIMAGE,sizeof(MyInfo),(LPARAM)pMyInfo);

   

    DWORD dwRead = 0;

    MyInfo myInfo2;

    BOOL bRet = ::ReadProcessMemory(hProcess,pMyInfo,&myInfo2,sizeof(MyInfo),&dwRead);

 

    dwRead = GetLastError();

    m_log.Format("red =%.2f,blue=%.2f",myInfo2.blue,myInfo2.red);

    TRACE(m_log);

    VirtualFreeEx(hProcess,pMyInfo,0,MEM_RELEASE);

    UpdateData(FALSE);

}

 

SERVER 程序:

LRESULT    CMyWindow::OnCompareImage(HWND hWnd,WPARAM wParam,LPARAM lParam)

{

    if(wParam <sizeof(MyInfo)) return -1;

    MyInfo * pMyInfo = (MyInfo *)lParam;

 

    sprintf(m_strLog,"client:red=%.2f,blue=%.2f",pMyInfo->red,pMyInfo->blue);

    ::TextOut(GetDC(hWnd),0,50,m_strLog,strlen(m_strLog));

    pMyInfo->blue = 1.0;

    pMyInfo->red = 2.0;

    return 0;

}

源码下载:CompareServer.zip

posted on 2012-10-13 18:06  Jason_liu  阅读(3730)  评论(0编辑  收藏  举报