進程間通訊-WriteProcessMemory和ReadProcessMemory

 

最近在寫個程序的時候需要在進程間通訊,具體需求是這樣。

1.       主要有兩個進程:一個進程作爲被請求進程,我們稱爲 SERVER 進程;另一個進程是請求進程,稱爲 CLIENG 進程。

2.       SERVER 進程提供一些服務,其完成計算功能;而 CLIENT 進程需要在它執行完計算之後將結果取會。

 

由於計算結果可能是一個結構,也可能是一個複雜的數據,所以通過消息來在進程傳遞信息是有限的。另一方面一般是單方向的通訊,實際上這裏的需求有一個雙向性,看下圖:

ProcessComm.JPG


這裏兩個進程都可以有自己的窗口,因此實際上我們可以通過消息來通知對方。但仔細一想,請求服務通過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;

}

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章