• 第22章 DLL注入和API拦截(1)


    22.1 注入的一个例子(跨进程子类化窗口)

     

      ①子类化窗口可以改变窗口的行为,让发往该窗口的消息重新发到我们指定的过程来处理。但这种行为只能在本进程中(如A),对于从一个进程(如B)去子类化另一个进程(如A)时,会遇到无法跨越进程地址空间的边界问题。

      ②上图,进程B试图调用SetWindowLongPtr将进程A中的hWnd窗口过程,重新指定为MySubClassProc来处理(注意,MySubClassProc在进程B的地址空间中),而不是hWnd窗口的标准窗口处理过程。但因该行为是跨进程的,如果被允许的话,当系统要执行进程A的hWnd窗口过程时,就会试图调用进程A地址空间中另一个地址(该地址的值与进程B中的地址相同),原因是不同进程的地址空间是相互隔离的,所以会出现内存访问违规的错误。因此,为了避免这种错误,微软不允许SetWindowLongPrt修改别的进程的窗口过程

      ③要实现跨进程子类化窗口,须用到“DLL注入”技术,将DLL注入到进程的地址空间中。

    22.2 使用注册表来注入DLL

    (1)注册表项的使用

     

      ①AppInit_Dlls键可能会包含一个或一组Dll的文件名(用空格或逗号分隔,第一个Dll的文件名可包含路径,但其他Dll包含的路径会被忽略)。为了让系统使用这个注册表项,还须将LoadAppInit_Dlls的值设为1。

      ②因为路径可能会被忽略,所以最好将自己的Dll放到Windows的系统目录中,这样就不必指定路径了。

    (2)User32.dll的映射

      ①当User32.dll被映射到一个新的进程时,该进程会收到DLL_PROCESS_ATTACH通知。User32.dll对它进行处理的时候,会取得上述注册表键的值,并调用LoadLibrary来载入这个字符串中指定的每个DLL。

      ②当系统载入每个DLL时,会调用它们的DllMain函数,并将参数fdwReason设为DLL_PROCESS_ATTACH,这样每个DLL就能够对自己进行初始化。

      ③由于被注入的DLL是在进程生命期的早期被载入,所以调用Kernel32.dll中的函数是没问题的,但其他DLL中的函数可能会导致问题。

    (3)注册表注入DLL的缺点

      ①基于GUI的应用程序都使用了User32.dll,但大多数基于CUI的应用程序不会使用它。因此想要将DLL注入到编译器或链接器,这种方法是行不通的。

      ②DLL会被映射进每个基于GUI的应用程序中,导致这些进程的线程都在运行我们的代码,如果我们的代码进入无限循环或错误地访问了内存,那么会影响到“容器”进程(即其他基于GUI的所有进程)的行为和健壮性

      ③我们的DLL被映射到每个基于GUI应用程序中,其生命期会直到应用程序终止(因为进程终止时,才会发送DLL_PROCESS_DETACH通知给User32.dll,在这里会卸载到我们的DLL)。而理想的情况下,DLL应注入到我们需要的进程中,并且映射的时间越短越好。

    22.3 使用Windows钩子来注入DLL

    22.3.1 钩子的类型

    (1)按作用范围

      ①局部钩子:仅钩挂属于自身进程的事件

      ②远程钩子:可钩挂自身进程,还可以钩挂其他进程中发生的事件。又分为基于线程和系统范围的。

    (2)按监视消息的类型分类

    钩子名称

    监视消息的类型和时机

    WH_CALLWNDPROC

    每当调用SendMessage函数时,函数将消息发送给目标窗口前首先调用钩子函数

    WH_CALLWNDPROCRET

    每当调用SendMessage函数时,函数将消息发送给目标窗口过程后再调用钩子函数

    WH_GETMESSAGE

    每当调用GetMessage或PeekMessage函数时,函数从程序的消息队列中获取一个消息后调用钩子函数

    WH_KEYBOARD

    每当调用GetMessage或PeekMessage函数时,如果从消息队列中得到的是WM_KEYUP或WM_KEYDOWN消息时,则调用钩子函数

    WH_MOUSE

    每当调用GetMessage或PeekMessage函数时,如果从消息队列中得到的是鼠标消息,则调用钩子函数

    WH_HARDWARE

    每当调用GetMessage或PeekMessage函数时,如果从消息队列中得到的是非鼠标和键盘消息时,则调用钩子函数

    WH_MSGFILTER

    当用户对对话框、菜单和滚动条有所操作时,系统在发送对应的消息之前调用钩子函数,这种钩子只能是局部的

    WH_SYSMSGFILTER

    同WH_MSGFILTER,不过是系统范围的

    WH_SHELL

    当Windows shell程序准备接收一些通知事件前调用钩子函数,如shell被激活或重绘等。外壳应用程序是不接受WH_SHELL消息的,要使应用程序能够接收WH_SHELL,必须调用SystemParametersInfo函数注册该消息。WM_SHELL共有5种情况:

    ①TaskBar需要重绘某个按钮时

    ②当系统地要显示关于Taskbar的一个程序的最小化形式

    ③当目前的键盘布局状态改变。

    ④当按Ctrl+Esc去执行TaskManager(或相等级别的程序)时。

    ⑤只要有个top-level、unowned窗口被创建、起作用或被摧毁

    WH_DEBUG

    用来给其他钩子函数除错

    WH_CBT

    当基于计算机的训练(CBT)事件发生时调用钩子函数

    WH_JOURNALRECORD

    日志记录钩子,用来记录发送给系统消息队列的所有消息

    WH_JOURNALPLAYBACK

    日志回放钩子,用来回放日志记录钩子记录的系统事件

    WH_FOREGROUNDIDLE

    系统空闲钩子,当系统空闲的时候调用钩子函数,这样就可以在这里安排一些低优先级的任务

    22.3.2 钩子程序的结构:

    (1)主要的3个功能模块

    模块

    作用

    ①主程序

    用来实现界面或者其他功能

    ②钩子回调函数

    用来接收系统发过来的消息

    ①对于局部钩子,这些模块可在同一个EXE文件中

    ②对于远程钩子这个回调函数必须在一个DLL中

    ③钩子的安装和卸载模块

    调用SetWindowsHookEx和UnhookWindowsHookEx函数。

    ①这部分虽然没有要求,但一般也放在DLL中,因为钩子创建以后会得到一个钩子句柄,这个句柄要在钩子回调函数中,以及卸载钩子的时候用到。

    ②如果把这部分代码放在主程序中的话,还需要创建一个函数将它传回给动态链接库,所以不如直接放在库中

    (2)钩子的安装:SetWindowsHookEx

    参数

    说明

    int idHook

    要安装的钩子的类型

    HOOKPROC lpfn

    钩子回调函数(在我们的地址空间或某个DLL)

    HINSTANCE hInstDll

    标识一个DLL,这个DLL包含了钩子回调函数。

    如果是局部钩子这里须为NULL。

    DWORD dwThreadId

    给哪个线程安装钩子。如果为0,表示给系统中所有的GUI线程安装钩子。

    (3)举例说明钩子的工作过程——以WM_GETMESSAGE为例(进程A监视进程B的窗口消息,设钩子回调函数为GetMsgProc)

      ①当进程B的一个线程准备GetMessage一条窗口消息时。系统会检查该线程是否安装了WH_GETMESSAGE钩子。

      ②系统检查GetMsgProc所在的DLL是否己被映射进程B的地址空中。

      ③如果DLL尚未被映射,那么系统会强制将该DLL映射到进程B的地址空间中,并将进程B中该DLL的锁计数器递增

      ④由于安装钩子时参数hInstDll和回调函数的地址都是进程A地址空间中的值。当被映射到进程B时,系统会检查该DLL被映射的基地址是否与进程A中的位置相同。

        Ⅰ如果相同,那么在两个进程的地址空间中,GetMsgProc函数位于相同的位置,在这情况下,系统可以直接在进程B(课本这里写A,是否是错误?)的地址空间中调用GetMsgProc。

        Ⅱ如果不同,则得通过GetMsgProcB = hInstDllB + (GetMsgProcA - hInstDllA)公式计算GetMsgProc在进程B中的虚拟地址。

      ⑤系统在进程B中递增该DLL的锁计数器,这是因为调用GetMsgProc函数时须先锁定,以防止该DLL被别的线程卸载掉。当GetMsgProc返回后,系统递减该DLL在进程B中的锁计数器。

    (4)钩子的卸载:BOOL UnhookWindowsHookEx(HHOOK hHook)

      ①当一个线程调用UnhookWindowsHookEx时,系统会遍历其内部维护的一个进程列表,该列表记录着那些己经注入过该DLL的进程。

      ②并将这些进程中该DLL的锁计数递减(注意每个进程都有各自对该DLL的锁计数器)。当某个进程的锁计数器减到0时,系统会自动从该进程的地址空间中撤消对该DLL的映射。

    22.3.3 钩子链

    (1)钩子链

      ①每一个钩子都有一个与之相关联的指针列表,称之为钩子链表,由系统来维护。这个列表的指针指向由应用程序定义的、后面由钩子调用的回调函数,也就是该钩子的各个处理子程。

      ②多个程序可以同时安装同一种钩子(如WH_GETMESSAGE),这些钩子组成一个钩子链,最近加入的钩子放在链表的头部,Windows负责为每种钩子维护一个钩子链。

      ③当一个事件发生的时候,Windows调用最后安装的钩子,然后由这个钩子的回调函数发起调用下一个钩子的动作,Windows收到这个动作后,再从链表中取出下一个钩子的地址并将调用传递下去。

    (2)CallNextHookEx函数:调用下一个钩子回调函数的函数。

    【Desktop Item Position Saver(DIPS)工具】——用来保存/恢复桌面图标的位置

    (1)可能遇到的几个问题

      ①获得桌面的ListView控件窗口句柄的方法

      A、XP下

      //桌面的ListView控件是ProgMan窗口的孙子窗口

      hWndLV = GetFirstChild(GetFirstChild(FindWindow(TEXT("ProgMan"),NULL)));

      B、Win8下

    HWND GetDestopListViewHandle(){
        HWND hWnd = NULL,hWndItem;
        HWND hWndParent = FindWindowEx(0,0,TEXT("WorkerW"),NULL);
        while (hWndParent && (!hWnd)){
    
            hWndItem = FindWindowEx(hWndParent, 0, TEXT("SHELLDLL_DefView"), NULL);
            if (hWndItem){
                hWnd = FindWindowEx(hWndItem, 0, TEXT("SysListView32"), TEXT("FolderView"));
                break;
            }
            hWndParent = FindWindowEx(0, hWndParent, TEXT("WorkerW"), NULL);
        }
        return hWnd;
    }

      ②获取创建窗口的线程ID:dwThreadID = GetWindowThreadProcessID(hwnd,NULL)

      ③为Windows资源管理器创建一个“隐藏窗口”用于与DIPS.exe通信(利用窗口消息)

      ④卸载钩子时,须先销毁“隐藏窗口”这个对话框,再清除钩子。否则当对话框收到一下条消息时会导致Windows资源管理器的线程引发访问违规。

    (2)实例源码

    【动态链接库】

    /************************************************************************
    Module:  DIPSLib.h
    Notices: Copyright(c) 2008 Jeffrey Richter & Christophe Nasarre
    ************************************************************************/
    #pragma  once
    #include <windows.h>
    
    #if !defined(DIPSLIBAPI)
    #define DIPSLIBAPI __declspec(dllimport)
    #endif
    
    //////////////////////////////////////////////////////////////////////////
    
    //导出的外部函数原型
    DIPSLIBAPI  BOOL WINAPI  SetDIPSHook(DWORD dwThreadId);
    
    
    //////////////////////////////////////////////////////////////////////////

    //DIPSLib.cpp

    /************************************************************************
    Module:  DIPSLib.cpp
    Notices: Copyright(c) 2008 Jeffrey Richter & Christophe Nasarre
    ************************************************************************/
    
    #include "../../CommonFiles/CmnHdr.h"
    
    
    #define DIPSLIBAPI __declspec(dllexport)
    #include "DIPSLib.h"
    #include "resource.h"
    
    //////////////////////////////////////////////////////////////////////////
    #ifdef _DEBUG
    //这个函数会强制调试器被调用
    void ForceDebugBreak(){
        __try{ DebugBreak();}
        __except (UnhandledExceptionFilter(GetExceptionInformation())){}
    }
    #else
    #define ForceDebugBreak()
    #endif
    
    //////////////////////////////////////////////////////////////////////////
    //前置声明
    LRESULT  WINAPI GetMsgProc(int nCode, WPARAM wParam, LPARAM lParam);
    
    INT_PTR  WINAPI Dlg_Proc(HWND hWnd, UINT uMsg, WPARAM wParam, LPARAM lParam);
    
    //////////////////////////////////////////////////////////////////////////
    //通知编译器将g_hHook变量放在共享数据段中,以便让不同进程共享
    #pragma data_seg("Shared")
    HHOOK  g_hHook = NULL;
    DWORD  g_dwThreadIdDIPS = 0; //安装钩子的线程(这里是DIPS.exe中一个线程)
    #pragma data_seg()
    
    #pragma comment(linker,"/section:Shared,rws") //可读可读、可共享
    
    //////////////////////////////////////////////////////////////////////////
    
    HINSTANCE g_hInstDll = NULL;
    
    //////////////////////////////////////////////////////////////////////////
    BOOL WINAPI DllMain(HINSTANCE hInstDll, DWORD fdwReason, PVOID fImpLoad){
        switch (fdwReason)
        {
        case DLL_PROCESS_ATTACH:
            g_hInstDll = hInstDll;
            break;
    
        case DLL_THREAD_ATTACH:
            break;
    
        case DLL_THREAD_DETACH:
            break;
    
        case DLL_PROCESS_DETACH:
            break;
        }
        return (TRUE);
    }
    
    //////////////////////////////////////////////////////////////////////////
    BOOL WINAPI SetDIPSHook(DWORD dwThreadId){
        BOOL bOk = FALSE;
    
        if (dwThreadId != 0){
            chASSERT(g_hHook == NULL);//断言钩子还未被安装
    
            //保存我们的线程ID,以便在GetMsgProc函数里可以通过客户程序(DIPS.exe)与
            //服务程序(Windows资源管理器)间进行通信
            g_dwThreadIdDIPS = GetCurrentThreadId();
    
            //为指定的线程安装WH_GETMESSAGE钩子
            g_hHook = SetWindowsHookEx(WH_GETMESSAGE, GetMsgProc, g_hInstDll, dwThreadId);
    
            bOk = (g_hHook != NULL);
            if (bOk){
                //钩子安装成功,发送一条消息到Windows资源管理器的消息队列里,这时资源管理器
                //会检测到一条消息,并调用GetMessage,于是钩子函数就开始工作。
                bOk = PostThreadMessage(dwThreadId, WM_NULL, 0, 0);
            }
    
        } else{
            //当传入dwThreadId为0时,表示要卸载钩子
            chASSERT(g_hHook != NULL);//断言钩子己被安装,
            bOk = UnhookWindowsHookEx(g_hHook);
            g_hHook = NULL;
        }
    
        return (bOk);
    }
    
    //////////////////////////////////////////////////////////////////////////
    LRESULT WINAPI GetMsgProc(int nCode, WPARAM wParam, LPARAM lParam){
        static BOOL bFirstTime = TRUE;
    
        if (bFirstTime){
            //Dll刚被注入
            bFirstTime = FALSE;
    
            //当调试被注入DLL的进程时,要取消注释
            //ForceDebugBreak();
    
            //创建DIPS服务窗口,以处理客户请求
            CreateDialog(g_hInstDll, MAKEINTRESOURCE(IDD_DIPS), NULL, Dlg_Proc);
    
            //通知DIPS应用程序服务窗口己创建完毕,可以接收请求了
            PostThreadMessage(g_dwThreadIdDIPS, WM_NULL, 0, 0);
        }
        return (CallNextHookEx(g_hHook, nCode, wParam, lParam));
    }
    
    //////////////////////////////////////////////////////////////////////////
    void Dlg_OnClose(HWND hWnd){
        DestroyWindow(hWnd);
    }
    
    //////////////////////////////////////////////////////////////////////////
    static const TCHAR g_szRegSubKey[] = 
            TEXT("Software\Wintellect\Desktop Item Position Saver");
    
    //////////////////////////////////////////////////////////////////////////
    void SaveListViewItemPositions(HWND hWndLV){
        int nMaxItems = ListView_GetItemCount(hWndLV);
    
        //删除先前的旧信息
        LONG lRet = RegDeleteKey(HKEY_CURRENT_USER, g_szRegSubKey);
    
        //创建新的注册表项
        HKEY hKey;
        lRet = RegCreateKeyEx(HKEY_CURRENT_USER, g_szRegSubKey, 0, NULL, REG_OPTION_NON_VOLATILE,
                              KEY_SET_VALUE, NULL, &hKey, NULL);
    
        chASSERT(lRet == ERROR_SUCCESS);
    
        for (int nItem = 0; nItem < nMaxItems;nItem++){
    
            //获取ListItem项的名称及位置
            TCHAR szName[MAX_PATH];
            ListView_GetItemText(hWndLV, nItem, 0, szName, _countof(szName));
    
            POINT pt;
            ListView_GetItemPosition(hWndLV, nItem, &pt);
    
            //保存名称及位置
            lRet = RegSetValueEx(hKey, szName, 0, REG_BINARY, (PBYTE)&pt, sizeof(pt));
    
            chASSERT(lRet == ERROR_SUCCESS);
        }
    
        RegCloseKey(hKey);
    }
    
    //////////////////////////////////////////////////////////////////////////
    void RestoreListViewItemPosition(HWND hWndLV){
        HKEY hKey;
        LONG lRet = RegOpenKeyEx(HKEY_CURRENT_USER, g_szRegSubKey, 0, KEY_QUERY_VALUE, &hKey);
    
        if (lRet != ERROR_SUCCESS)
            return;
            
        //如果ListView打开了自动排列属性,则暂时关闭
        DWORD dwStyle = GetWindowStyle(hWndLV);
        if (dwStyle & LVS_AUTOARRANGE)
            SetWindowLong(hWndLV, GWL_STYLE, dwStyle & ~LVS_AUTOARRANGE);
    
        lRet = NO_ERROR;
        for (int nIndex = 0; lRet != ERROR_NO_MORE_ITEMS; nIndex++){
            TCHAR szName[MAX_PATH];
            DWORD cbValueName = _countof(szName);//必须要在循环体内重新设置该值,因为每次会被RegEnumValue改变,用来接受下一个szName  
    
            POINT pt;
            DWORD cbData = sizeof(pt), nItem;//必须要在循环体内重新设置cbData,因为每次会被RegEnumValue改变
    
            //从注册表中读取名称及位置信息
            DWORD dwType;
    
            lRet = RegEnumValue(hKey, nIndex, szName, &cbValueName, NULL, &dwType, (PBYTE)&pt, &cbData);
    
            if (lRet == ERROR_NO_MORE_ITEMS)
                continue;
    
            if ((dwType == REG_BINARY) && (cbData = sizeof(pt))){
                //查找ListView控件相应的项目
                LV_FINDINFO  lvfi;
                lvfi.flags = LVFI_STRING;
                lvfi.psz = szName;
                nItem = ListView_FindItem(hWndLV, -1, &lvfi);
                if (nItem != -1){
                    ListView_SetItemPosition(hWndLV, nItem, pt.x, pt.y);
                }
            }
        }
        //将自动排列属性恢复为原来的值
        SetWindowLong(hWndLV, GWL_STYLE, dwStyle);
        RegCloseKey(hKey);
    }
    
    //////////////////////////////////////////////////////////////////////////
    INT_PTR WINAPI Dlg_Proc(HWND hWnd, UINT uMsg, WPARAM wParam, LPARAM lParam){
        switch (uMsg)
        {
            chHANDLE_DLGMSG(hWnd, WM_CLOSE, Dlg_OnClose); //隐藏服务窗口
        case WM_APP:
             //当调试被注入DLL的进程时,要取消注释
             //ForceDebugBreak();
    
            //lParam:表示保存或恢复操作。wParam:桌面ListView的句柄
            if (lParam)
                SaveListViewItemPositions((HWND)wParam);
            else
                RestoreListViewItemPosition((HWND)wParam);
            break;
        }
        return (FALSE);
    }

    //resource.h

    //{{NO_DEPENDENCIES}}
    // Microsoft Visual C++ 生成的包含文件。
    // 供 22_DipsLib.rc 使用
    //
    #define IDD_DIPS                        101
    
    // Next default values for new objects
    // 
    #ifdef APSTUDIO_INVOKED
    #ifndef APSTUDIO_READONLY_SYMBOLS
    #define _APS_NEXT_RESOURCE_VALUE        102
    #define _APS_NEXT_COMMAND_VALUE         40001
    #define _APS_NEXT_CONTROL_VALUE         1001
    #define _APS_NEXT_SYMED_VALUE           101
    #endif
    #endif

    //rc文件

    // Microsoft Visual C++ generated resource script.
    //
    #include "resource.h"
    
    #define APSTUDIO_READONLY_SYMBOLS
    /////////////////////////////////////////////////////////////////////////////
    //
    // Generated from the TEXTINCLUDE 2 resource.
    //
    #include "winres.h"
    
    /////////////////////////////////////////////////////////////////////////////
    #undef APSTUDIO_READONLY_SYMBOLS
    
    /////////////////////////////////////////////////////////////////////////////
    // 中文(简体,中国) resources
    
    #if !defined(AFX_RESOURCE_DLL) || defined(AFX_TARG_CHS)
    LANGUAGE LANG_CHINESE, SUBLANG_CHINESE_SIMPLIFIED
    
    #ifdef APSTUDIO_INVOKED
    /////////////////////////////////////////////////////////////////////////////
    //
    // TEXTINCLUDE
    //
    
    1 TEXTINCLUDE 
    BEGIN
        "resource.h"
    END
    
    2 TEXTINCLUDE 
    BEGIN
        "#include ""winres.h""
    "
        ""
    END
    
    3 TEXTINCLUDE 
    BEGIN
        "
    "
        ""
    END
    
    #endif    // APSTUDIO_INVOKED
    
    
    /////////////////////////////////////////////////////////////////////////////
    //
    // Dialog
    //
    
    IDD_DIPS DIALOG DISCARDABLE  0, 0, 132, 13
    STYLE WS_CAPTION
    CAPTION "Wintellect DIPS"
    FONT 8, "MS Shell Dlg"
    BEGIN
    END
    
    /////////////////////////////////////////////////////////////////////////////
    //
    // DESIGNINFO
    //
    
    #ifdef APSTUDIO_INVOKED
    GUIDELINES DESIGNINFO
    BEGIN
        IDD_DIALOG1, DIALOG
        BEGIN
            LEFTMARGIN, 7
            RIGHTMARGIN, 302
            TOPMARGIN, 7
            BOTTOMMARGIN, 169
        END
    END
    #endif    // APSTUDIO_INVOKED
    
    #endif    // 中文(简体,中国) resources
    /////////////////////////////////////////////////////////////////////////////
    
    
    
    #ifndef APSTUDIO_INVOKED
    /////////////////////////////////////////////////////////////////////////////
    //
    // Generated from the TEXTINCLUDE 3 resource.
    //
    
    
    /////////////////////////////////////////////////////////////////////////////
    #endif    // not APSTUDIO_INVOKED

    【DIPS工具源码】

    /************************************************************************
    Module: DIPS.cpp
    Notices: Copyright(c) 2008 Jeffrey Richter & Christophe Nasarre
    ************************************************************************/
    
    #include "../../CommonFiles/CmnHdr.h"
    #include <tchar.h>
    #include "resource.h"
    #include "../22_DipsLib/DIPSLib.h"
    
    #pragma  comment(lib,"../../Debug/22_DIPSLib.lib")
    //////////////////////////////////////////////////////////////////////////
    BOOL Dlg_OnInitDialog(HWND hWnd, HWND hWndFocus, LPARAM lParam){
        chSETDLGICONS(hWnd, IDI_DIPS);
        return (TRUE);
    }
    
    //////////////////////////////////////////////////////////////////////////
    void Dlg_OnCommand(HWND hwnd, int id, HWND hwndCtrl, UINT codeNotity){
        switch (id)
        {
        case IDC_SAVE:
        case IDC_RESTORE:
        case IDCANCEL:
            EndDialog(hwnd, id);
            break;
        }
    }
    
    //////////////////////////////////////////////////////////////////////////
    INT_PTR WINAPI Dlg_Proc(HWND hWnd, UINT uMsg, WPARAM wParam, LPARAM lParam){
        switch (uMsg){
            chHANDLE_DLGMSG(hWnd, WM_INITDIALOG, Dlg_OnInitDialog);
            chHANDLE_DLGMSG(hWnd, WM_COMMAND, Dlg_OnCommand);
        }
    
        return (FALSE);
    }
    
    HWND GetDestopListViewHandle(){
        HWND hWnd = NULL,hWndItem;
        HWND hWndParent = FindWindowEx(0,0,TEXT("WorkerW"),NULL);
        while (hWndParent && (!hWnd)){
    
            hWndItem = FindWindowEx(hWndParent, 0, TEXT("SHELLDLL_DefView"), NULL);
            if (hWndItem){
                hWnd = FindWindowEx(hWndItem, 0, TEXT("SysListView32"), TEXT("FolderView"));
                break;
            }
            hWndParent = FindWindowEx(0, hWndParent, TEXT("WorkerW"), NULL);
        }
        return hWnd;
    }
    
    //////////////////////////////////////////////////////////////////////////
    int WINAPI _tWinMain(HINSTANCE hInstExe, HINSTANCE, PTSTR pszCmdLine, int){
        //将命令行转为大写字符串
        CharUpperBuff(pszCmdLine, 1);
        TCHAR cWhatToDo = pszCmdLine[0];
    
        if ((cWhatToDo !=TEXT('S')) && (cWhatToDo != TEXT('R'))){
            cWhatToDo = 0; //无效的命令行参数
        }
    
        //如果是从GUI界面启动本程序,而不是命令行启动
        if (cWhatToDo == 0){
            //无命令行参数时,显示对话框以让用户相应的选择
            switch (DialogBox(hInstExe, MAKEINTRESOURCE(IDD_DIPS), NULL, Dlg_Proc)){
            case IDC_SAVE:
                cWhatToDo = TEXT('S');
                break;
            case IDC_RESTORE:
                cWhatToDo = TEXT('R');
                break;
            }
        }
    
        //表示用户什么也不做
        if (cWhatToDo == 0){
            return (0);
        }
    
        //获取桌面listView的窗口句柄,他是
        
        HWND hWndLV = GetDestopListViewHandle();
        chASSERT(IsWindow(hWndLV));
    
        //安装钩子到Explorer进程的地址空间。当安装完成后,Explorer里会创建一个“隐藏”
        //的窗口,我们可以向这个窗口发送消息,以便告诉他我们要想做的事情
        chASSERT(SetDIPSHook(GetWindowThreadProcessId(hWndLV, NULL)));
    
        //阻塞等待DIPS服务器窗口创建完毕,当创建完毕后,这里GetMessage会收到通知,线程被唤醒
        MSG msg;
        GetMessage(&msg, NULL, 0, 0);//阻塞等待
    
        //查找服务端这个被隐藏的窗口
        HWND hWndDIPS = FindWindow(NULL, TEXT("Wintellect DIPS"));
    
        chASSERT(IsWindow(hWndDIPS)); //断言窗口是存在的
    
        //告诉服务端的窗口进行保存或恢复桌面图标操作
        BOOL bSave = (cWhatToDo == TEXT('S'));
        SendMessage(hWndDIPS, WM_APP, (WPARAM)hWndLV, bSave);
    
        //销毁隐形窗口,使用SendMessage可以保证在函数返回后,窗口己这被销毁
        SendMessage(hWndDIPS, WM_CLOSE, 0, 0);
    
        chASSERT(!IsWindow(hWndDIPS)); //断言窗口己被销毁
    
        //卸载钩子
        SetDIPSHook(0);
        return (0);
    }

    //resource.h

    //{{NO_DEPENDENCIES}}
    // Microsoft Visual C++ 生成的包含文件。
    // 供 22_DIPS.rc 使用
    //
    #define IDC_RESTORE                     3
    #define IDI_DIPS                        101
    #define IDD_DIPS                        102
    #define IDC_SAVE                        1000
    
    // Next default values for new objects
    // 
    #ifdef APSTUDIO_INVOKED
    #ifndef APSTUDIO_READONLY_SYMBOLS
    #define _APS_NEXT_RESOURCE_VALUE        102
    #define _APS_NEXT_COMMAND_VALUE         40001
    #define _APS_NEXT_CONTROL_VALUE         1001
    #define _APS_NEXT_SYMED_VALUE           101
    #endif
    #endif

    //rc文件

    // Microsoft Visual C++ generated resource script.
    //
    #include "resource.h"
    
    #define APSTUDIO_READONLY_SYMBOLS
    /////////////////////////////////////////////////////////////////////////////
    //
    // Generated from the TEXTINCLUDE 2 resource.
    //
    #include "winres.h"
    
    /////////////////////////////////////////////////////////////////////////////
    #undef APSTUDIO_READONLY_SYMBOLS
    
    /////////////////////////////////////////////////////////////////////////////
    // 中文(简体,中国) resources
    
    #if !defined(AFX_RESOURCE_DLL) || defined(AFX_TARG_CHS)
    LANGUAGE LANG_CHINESE, SUBLANG_CHINESE_SIMPLIFIED
    
    /////////////////////////////////////////////////////////////////////////////
    //
    // Icon
    //
    
    // Icon with lowest ID value placed first to ensure application icon
    // remains consistent on all systems.
    IDI_DIPS                ICON                    "DIPS.ico"
    
    #ifdef APSTUDIO_INVOKED
    /////////////////////////////////////////////////////////////////////////////
    //
    // TEXTINCLUDE
    //
    
    1 TEXTINCLUDE 
    BEGIN
        "resource.h"
    END
    
    2 TEXTINCLUDE 
    BEGIN
        "#include ""winres.h""
    "
        ""
    END
    
    3 TEXTINCLUDE 
    BEGIN
        "
    "
        ""
    END
    
    #endif    // APSTUDIO_INVOKED
    
    
    /////////////////////////////////////////////////////////////////////////////
    //
    // Dialog
    //
    IDD_DIPS DIALOGEX 0, 0, 208, 95
    STYLE DS_SETFONT | DS_3DLOOK | DS_CENTER | WS_MINIMIZEBOX | WS_CAPTION | WS_SYSMENU
    CAPTION "桌面图标位置保存工具"
    FONT 8, "MS Shell Dlg", 0, 0, 0x1
    BEGIN
        CTEXT           "DIPS: 桌面图标位置保存工具
    版权所有 (c) 2016 by 浅墨浓香.", IDC_STATIC, 4, 4, 200, 20, SS_SUNKEN, WS_EX_DLGMODALFRAME | WS_EX_STATICEDGE
        LTEXT           "这个工具可以保存或恢复桌面图标的位置.
    
    保存位置: DIPS S
    恢复位置: DIPS R", IDC_STATIC, 4, 32, 200, 36
        DEFPUSHBUTTON   "&保存", IDC_SAVE, 4, 76, 50, 14
        PUSHBUTTON      "&恢复", IDC_RESTORE, 79, 76, 50, 14
        PUSHBUTTON      "取消", IDCANCEL, 154, 76, 50, 14
    END
    
    /////////////////////////////////////////////////////////////////////////////
    //
    // DESIGNINFO
    //
    
    #ifdef APSTUDIO_INVOKED
    GUIDELINES DESIGNINFO
    BEGIN
        IDD_DIALOG1, DIALOG
        BEGIN
            LEFTMARGIN, 7
            RIGHTMARGIN, 302
            TOPMARGIN, 7
            BOTTOMMARGIN, 169
        END
    END
    #endif    // APSTUDIO_INVOKED
    
    #endif    // 中文(简体,中国) resources
    /////////////////////////////////////////////////////////////////////////////
    
    
    
    #ifndef APSTUDIO_INVOKED
    /////////////////////////////////////////////////////////////////////////////
    //
    // Generated from the TEXTINCLUDE 3 resource.
    //
    
    
    /////////////////////////////////////////////////////////////////////////////
    #endif    // not APSTUDIO_INVOKED
  • 相关阅读:
    qt映射器QSignalMapper的理解
    win10环境下安装Qt4.8、PyQt及development tools
    BZOJ1040 基环森林 找环+基础树形DP
    贪心 BZOJ1034
    HDU5293 树链剖分+树形DP
    BZOJ 1028 BZOJ 1029 //贪心
    BZOJ1025
    仙人掌图判定及求直径HDU3594 BZOJ1023
    BZOJ1021
    BZOJ1022
  • 原文地址:https://www.cnblogs.com/5iedu/p/5180828.html
Copyright © 2020-2023  润新知