程序師世界是廣大編程愛好者互助、分享、學習的平台,程序師世界有你更精彩!
首頁
編程語言
C語言|JAVA編程
Python編程
網頁編程
ASP編程|PHP編程
JSP編程
數據庫知識
MYSQL數據庫|SqlServer數據庫
Oracle數據庫|DB2數據庫
 程式師世界 >> 編程語言 >> Visual Basic語言 >> VB.NET >> 【VB.NET】也談跨進程消息鉤子

【VB.NET】也談跨進程消息鉤子

編輯:VB.NET

寫給VB.NET程序員^_^

老生常談了:子類化。

我們都知道在VB6裡面可以用API函數來進行子類化,以處理自身的窗體過程;如果跨進程,這就麻煩了,由於我們的函數在我們的進程中(廢話),而目標進程的窗口的消息處理函數在目標進程(還是廢話),所以只能想辦法把我們的代碼放到對方進程中去執行——並且要告知我們的進程得到了什麼消息。恐怕寫匯編就有點嚇人了,於是大家都寫DLL,其原理就是把回調函數放到一個DLL裡面注入到對方進程,DLL去修改目標窗口的默認處理函數——把消息發送給我們。

當然也有“另類”一點的:http://www.it-berater.org/ThueDownloads/index.shtml上面有一個DLL包,其中含有一個dssubcls.dll,用它,可以輕松的完成我們的工作:就像調用一個API一樣簡單,而且在我們的程序中使用回調函數!呵呵,省去了自己寫DLL的麻煩之後,這些好處足以吸引各位觀眾了吧?

好了,VB6的代碼大家可以在下載的壓縮包中找到,作者提供了一個以記事本為基礎的實例(在\dssubcls目錄下),非常詳細無需詳細敘述了。關鍵是在VB.NET裡面如何使用它——如何聲明API,如何進行回調,看用來子類化的API的VB6聲明先:

Declare Function SubClass& Lib "dssubcls" (ByVal HwndSubclass&, _

Optional ByVal Address& = 0, _

Optional ByVal OldStyle& = 0, _

Optional ByVal NewStyle& = 0, _

Optional ByVal Ext& = 0, _

Optional ByVal SubClass& = 0)

轉化成VB.NET的聲明類似下面的樣子(習慣使然,我把&展開成了As Integer):

Declare Function SubClass Lib "dssubcls" (ByVal HwndSubclass As Integer, Optional ByVal Address As Integer = 0, Optional ByVal OldStyle As Integer = 0, Optional ByVal NewStyle As Integer = 0, Optional ByVal Ext As Integer = 0, Optional ByVal SubClass As Integer = 0) As Integer

這不是很好嘛?問題來了,這樣的聲明在VB6裡面可以使用Addressof function來傳入第二個參數(參見你下載的源碼),但是在VB.NET裡面直接Addressof就不成了——我們需要委托一個回調:

Private Delegate Function HookCallBack(ByVal wMsg As Integer, ByVal wParam As Integer, ByVal lParam As Integer) As Integer

這個委托,對應的是以下函數:

Private Function mCallback(ByVal wMsg As Integer, ByVal wParam As Integer, ByVal lParam As Integer) As Integer

‘在這裡處理得到的消息

End Function

使用時,需要注意先實例化這個委托:

Private fix_COCD = New HookCallBack(AddressOf mCallback)

此時,fix_COCD就是我們的mCallback函數引用了,用更直觀的觀點來看,fix_COCD就是一個指向mCallback的指針,相當於VB6裡面的Addressof function得到的結果,看似問題解決了,於是我們寫了以下代碼來搞對方的進程窗體消息:

SubClass(Handle, fix_COCD, 0, 0, 0, 1)   '修改處理函數

問題真是接踵而至!IDE提示變量類型不符!!事實確實如此,我們把一個HookCallBack類型當做Integer來傳遞,無法通過檢查,那麼強行轉換吧?當然,你可以去試試。這時,我所做的是,修改這個API聲明:

Private Declare Function SubClass Lib "dssubcls" (ByVal HwndSubclass As Integer, Optional ByVal Address As HookCallBack = Nothing, Optional ByVal OldStyle As Integer = 0, Optional ByVal NewStyle As Integer = 0, Optional ByVal Ext As Integer = 0, Optional ByVal SubClass As Integer = 0) As Integet

使之符合我們的調用?有點倒行逆施?並非如此,當你習慣了修改API聲明之後,會發現有些事變得如此簡單,有些事需要你重新認識——對於WIN32 API也是如此。

至此,大功告成:

較為完整的代碼如下:

Private Declare Function SubClass Lib "dssubcls" (ByVal HwndSubclass As Integer, Optional ByVal Address As HookCallBack = Nothing, Optional ByVal OldStyle As Integer = 0, Optional ByVal NewStyle As Integer = 0, Optional ByVal Ext As Integer = 0, Optional ByVal SubClass As Integer = 0) As Integer
    Private Declare Function UseSendMessage Lib "dssubcls" (ByVal use As Integer) As Integer
    '實例化的委托
    Private fix_COCD = New HookCallBack(AddressOf mCallback)
    '委托
    Private Delegate Function HookCallBack(ByVal wMsg As Integer, ByVal wParam As Integer, ByVal lParam As Integer) As Integer
    Public Sub Hook(ByVal Handle As Integer)
        proc = SubClass(Handle, fix_COCD, 0, 0, 0, 1)   '修改處理函數
        UseSendMessage(1)
    End Sub

    Private Function mCallback(ByVal wMsg As Integer, ByVal wParam As Integer, ByVal lParam As Integer) As Integer

End Function

用這個代碼的時候,可能會碰見一些“意外情況“,例如wm_datacopy,此時,我們需要進一步去獲取LPARTM所指向的結構並對其進行解析(我們要讀的是對方窗口所在進程的內存,具體地址由lParam確定——實際上lParam一直是一個指針——IntPrt,但它與Integer完全就是一回事(如果你使用VB2005可能需要使用Intprt.toint32或intprt=new intprt(integer)這些):

Public Class GetMsg
    Public Declare Function ReadProcessMemory Lib "kernel32" (ByVal hProcess As Integer, ByVal lpBaseAddress As Integer, ByVal lpBuffer() As Byte, ByVal nSize As Integer, ByRef lpNumberOfBytesWritten As Integer) As Integer
    Public Declare Function ReadProcessMemory Lib "kernel32" (ByVal hProcess As Integer, ByVal lpBaseAddress As Integer, ByRef int As Integer, ByVal nSize As Integer, ByRef lpNumberOfBytesWritten As Integer) As Integer
    Public Declare Function OpenProcess Lib "kernel32" (ByVal dwDesiredAccess As Integer, ByVal bInheritHandle As Integer, ByVal dwProcessId As Integer) As Integer
    Public Declare Function CloseHandle Lib "kernel32" (ByVal hObject As Integer) As Integer
    Private hProc As IntPtr
    Sub New(ByVal PID As Integer)
        hProc = OpenProcess(&HFFFF, False, PID)
    End Sub

    Function readmsg(ByVal address As Integer) As Byte()
        Dim buf(19) As Byte
        ReadProcessMemory(hProc, address, buf, 20, 0)
        Return buf
    End Function

    Protected Overrides Sub Finalize()
        CloseHandle(hProc)
        MyBase.Finalize()
    End Sub
End Class

這個類提供了Readmsg方法來讀取一些內容——但這並不是完整的,我們知道,LPARAM指向的結構是這樣的:

<StructLayout(LayoutKind.Sequential)> _
Public Structure COPYDATASTRUCT
    Public dwData As Integer
    Public cbData As Integer
    Public lpData As IntPtr
End Structure

其中dwData我們不是很關心,當然其中也可能存在一些有用信息(這裡不想多說,網上有些文章純屬誤導)

而cbData是一個長度:lpData的長度

lpData這裡被聲明為指針,看起來更直觀了——它就是地址

有了地址和長度,如何讀取代碼就自己寫吧。

提示一下:參考我重載的ReadProcessMemory可能對你有不少幫助。

當然,上面提到的只是“特殊情況”中的一個典型,還有很多時候,進程是用自定義消息(>&H40A)來傳遞數據的,例如我所開發的這個工程,打印mCallBack的參數後,得到的是如下結果(十六進制,只提取了有用的信息):

473  14  42257D0

其中lParam就是一個指針,我讀了其中的一部分:

Function readmsg(ByVal address As Integer) As Byte()
        Dim buf(19) As Byte
        ReadProcessMemory(hProc, address, buf, 20, 0)
        Return buf
End Function

現在就明白為什麼上面的代碼是那樣了:)

然後進行了一個處理,得到了我想要的信息:

'消息解碼後得到的移動棋子信息:玩家,起X,起Y,止X,止Y,棋子編號,走棋總步數

Event Move(ByVal player As Byte, ByVal sx As Byte, ByVal sy As Byte, ByVal dx As Byte, ByVal dy As Byte, ByVal name As Byte, ByVal [step] As Byte)
    Private Function mCallback(ByVal wMsg As Integer, ByVal wParam As Integer, ByVal lParam As Integer) As Integer
        If wParam = &H14 Then
            Dim s As Byte() = msg.readmsg(lParam)
            RaiseEvent Move(s(1), s(10), s(11), s(12), s(13), s(14), s(16))
        End If
End Function

當然,在我的工程裡面重載的ReadProcessMemory並沒有被使用。

補充一下咯:

在VB.NET中,處理自己的窗體的消息只需要重載窗體消息處理過程就可以了,無需子類化:)

有補充一下:

對於wm_datacopy來說,還有一些數據獲取的問題沒有說清楚,實際上都可以用一些方法來解決。

真的要回家咯。餓。。貼一個重載可能更說明問題:

Protected Overrides Sub DefWndProc(ByRef m As System.Windows.Forms.Message)
        Select Case m.Msg
            Case WM_COPYDATA
                Dim mMessage As New COPYDATASTRUCT()
                mMessage = DirectCast(m.GetLParam(mMessage.[GetType]()), COPYDATASTRUCT)

               Dim bs(mMessage.cbData - 1) As Byte
               Marshal.Copy(mMessage.lpData, bs, 0, mMessage.cbData)

               debig.print System.BitConverter.ToString(bs, 0, mMessage.cbData)

        End Select

End Sub

  1. 上一頁:
  2. 下一頁:
Copyright © 程式師世界 All Rights Reserved