windows之桌面程序引導功能

本文轉載自查看原文 2016-07-29 23:54 1904 windows 桌面向導桌面引導/ 技術文章

經常使用windows系統的同學可能都會遇到這樣一種情況，剛按照完的應用程序，可能會在桌面產生一個提示信息，指示當前快捷方式可以使用了，並給出相應的文字說明，指示該快捷方式的功能。那么大家有沒有考慮過這供功能是怎么實現的呢，使用一般的窗口spy工具應該都能抓取windows系統桌面使用的窗口類，我使用彗星小助手抓取了下，效果如圖1所示。原來windows桌面窗口是一個listview，那么問題就變得簡單了，我們只需要拿到窗口句柄，並向其發送消息LVM_GETITEMCOUNT消息，獲取到窗口item數量后，在發送LVM_GETITEMTEXTA消息，並讀取每一項的文本名稱，在和我們自己想要的桌面文件名稱對比，如果是目標文件在發送LVM_GETITEMRECT消息，就可以拿到具體的桌面文件位置。下面看具體的實現過程。

圖1

看到這篇文章的同學，我推薦一下這個彗星小助手工具，功能挺豐富的，看圖1的標題欄就知道了，在沒有發現彗星小助手工具之前我采集屏幕顏色一致使用的是picpick，也挺好用，推薦大家使用。除此之外還推薦大家：everything硬盤搜索工具、小屁孩桌面便簽

一、首先是獲取桌面窗口句柄

獲取桌面窗口句柄需要區分xp系統和win7系統，首先我們按win7方式獲取，如果獲取失敗，那么我們再按win xp方式獲取，代碼如下，拿到桌面窗口句柄后，我們就可以進一步的給桌面窗口發送消息，並獲取桌面圖標相關信息，不過在此之前我們還需要做一件事，就是區分當前系統的位數，因為32位系統和64位系統的結構有所不一樣，需要我們分開做處理。

 1 CRect cRect;
 2     HWND hDeskWnd = nullptr;//桌面上SysListView32的窗口句柄
 3     HWND hWnd = ::FindWindow(L"WorkerW", nullptr);//先當WIN7系統查找
 4     while (hWnd)
 5     {
 6         HWND hShellView = ::FindWindowEx(hWnd, nullptr, L"SHELLDLL_DefView", nullptr);
 7         if (hShellView)
 8         {
 9             hDeskWnd = ::FindWindowEx(hShellView, nullptr, L"SysListView32", nullptr);
10             break;
11         }
12         hWnd = ::GetWindow(hWnd, GW_HWNDNEXT);
13     }
14 
15     if (hDeskWnd == nullptr)
16     {
17         qDebug() << QStringLiteral("WIN7系統查找方式失敗");
18         //如果沒找到，再按XP方式查找
19         hWnd = ::FindWindow(L"Progman", L"Program Manager");
20         if (hWnd)
21         {
22             hWnd = ::FindWindowEx(hWnd, nullptr, L"SHELLDLL_DefView", nullptr);
23             hDeskWnd = ::FindWindowEx(hWnd, nullptr, L"SysListView32", nullptr);
24         }
25     }
26 
27     if (hDeskWnd == nullptr)
28     {
29         QMessageBox::warning(nullptr, QStringLiteral("警告"), QStringLiteral("查找桌面窗口句柄失敗"));
30         return false;
31     }

判斷系統位數的方法不難，代碼如下：

 1 BOOL Is64Bit_OS()//判斷操作系統位數
 2     {
 3         BOOL bRetVal = false;
 4         SYSTEM_INFO si = { 0 };
 5         LPFN_PGNSI pGNSI = (LPFN_PGNSI)GetProcAddress(GetModuleHandle(L"kernel32.dll"), "GetNativeSystemInfo");
 6         if (pGNSI == nullptr)
 7         {
 8             return false;
 9         }
10         pGNSI(&si);
11         if (si.wProcessorArchitecture == PROCESSOR_ARCHITECTURE_AMD64
12             || si.wProcessorArchitecture == PROCESSOR_ARCHITECTURE_IA64)
13         {
14             bRetVal = true;
15         }
16         else
17         {
18             // 32 位操作系統
19             //_tprintf(L"is 32 bit OS\r\n");
20         }
21         return bRetVal;
22     }

在桌桌面引導的時候你可能還需要顯示桌面，我使用的代碼如下：

 1 void ShowDesktop(void)//顯示桌面
 2     {
 3         CoInitialize(0);
 4         IShellDispatch4 * pdisp = nullptr;
 5         CoCreateInstance(CLSID_Shell, nullptr, CLSCTX_ALL, __uuidof(IShellDispatch4), (void **)&pdisp);
 6         if (pdisp)
 7         {
 8             pdisp->ToggleDesktop();//這句是用來切換桌面的
 9             pdisp->Release();
10         }
11     }

二、獲取桌面圖標位置

從第一步我們可以成功拿到桌面窗口句柄，接下來我們就只需要給桌面窗口發送一些列的消息，然后拿到我們自己需要的信息。下面我一步一步講解，貼上關鍵代碼。

注意：32位系統和64位系統發送消息的流程一模一樣，只是定義的結構體不一樣，因此我只講解32位的代碼，64位的自己一看便明白，文章最后我會附上相關的示例demo。

1、首先是打開一個已存在的進程對象，並返回進程的句柄

1 GetWindowThreadProcessId(hDeskWnd, &PID); 2 HANDLE hProcess = OpenProcess(PROCESS_ALL_ACCESS, false, PID);

2、在一些機器上，可能會存在打開進程失敗的情況，比如我在xp系統上就出現這個問題，那么這個時候就需要提升進程的訪問權限，提升進程訪問權限代碼如下：

 1 BOOL EnableDebugPrivilege()
 2     {
 3         HANDLE hToken;
 4         BOOL fOk = FALSE;
 5         if (OpenProcessToken(GetCurrentProcess(), TOKEN_ADJUST_PRIVILEGES, &hToken))
 6         {
 7             TOKEN_PRIVILEGES tp;
 8             tp.PrivilegeCount = 1;
 9             LookupPrivilegeValue(NULL, SE_DEBUG_NAME, &tp.Privileges[0].Luid);
10 
11             tp.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED;
12             AdjustTokenPrivileges(hToken, FALSE, &tp, sizeof(tp), NULL, NULL);
13 
14             fOk = (GetLastError() == ERROR_SUCCESS);
15             CloseHandle(hToken);
16         }
17         return fOk;
18     }

3、如果進程打開失敗，則提升進程訪問權限，並在此打開進程

 1 if (hProcess == nullptr)
 2         {
 3             unsigned short errorCode = GetLastError();
 4             qDebug() << QStringLiteral("獲取進程句柄操作失敗32 hWnd=%1,PID=%2,errorCode=%3")
 5                 .arg((int)hDeskWnd).arg(PID).arg(errorCode);
 6             if (errorCode == 5)
 7             {
 8                 qDebug() << QStringLiteral("拒絕訪問");
 9                 if (EnableDebugPrivilege())
10                 {
11                     qDebug() << QStringLiteral("提升進程訪問權限成功");
12                 }
13                 else
14                 {
15                     qDebug() << QStringLiteral("提升進程訪問權限失敗");
16                 }
17 
18                 GetWindowThreadProcessId(hDeskWnd, &PID);
19                 hProcess = OpenProcess(PROCESS_ALL_ACCESS, false, PID);
20                 if (hProcess == nullptr)
21                 {
22                     qDebug() << QStringLiteral("進程依然打開失敗");
23                 }
24                 else
25                 {
26                     qDebug() << QStringLiteral("提升權限后開成功");
27                 }
28             }
29         }

4、進程打開成功后，想桌面程序發送LVM_GETITEMCOUNT消息，並獲取桌面文件個數

5、遍歷桌面所有文件，並發送LVM_GETITEMTEXTA消息，拿到文件的名稱，如果是目標文件則進行下一步，否則一直遍歷，直到遍歷完所有文件並結束

6、如果是目標文件，則發送LVM_GETITEMRECT消息，拿到文件在位置，並返回。

下面我提上完整的獲取桌面圖標位置代碼(32系統上)

 1 if (hProcess)
 2         {
 3             LVITEMA * pLVITEM = (LVITEMA *)VirtualAllocEx(hProcess, nullptr, sizeof(LVITEM), MEM_COMMIT, PAGE_READWRITE);
 4             char* pszText = (char *)VirtualAllocEx(hProcess, nullptr, 512, MEM_COMMIT, PAGE_READWRITE);
 5             RECT* pItemRc = (RECT *)VirtualAllocEx(hProcess, nullptr, sizeof(RECT), MEM_COMMIT, PAGE_READWRITE);
 6             RECT rc;
 7 
 8             if (!pItemRc || !pLVITEM)
 9             {
10                 qDebug() << QStringLiteral("無法分配內存！");
11             }
12             else
13             {
14                 LVITEMA LVITEM;
15                 LVITEM.mask = LVIF_TEXT;
16                 LVITEM.cchTextMax = 512;
17                 LVITEM.pszText = pszText;
18                 char ItemBuf[512];
19                 int nCount = ::SendMessage(hDeskWnd, LVM_GETITEMCOUNT, 0, 0);
20 
21                 for (int iItem = 0; iItem < nCount; ++iItem)
22                 {
23                     LVITEM.iItem = iItem;
24                     LVITEM.iSubItem = 0;
25                     //將設置好的結構插入目標進程
26                     WriteProcessMemory(hProcess, pLVITEM, &LVITEM, sizeof(LVITEM), nullptr);
27                     //發送LVM_GETITEM消息
28                     BOOL r = (BOOL)::SendMessage(hDeskWnd, LVM_GETITEMTEXTA, iItem, (LPARAM)pLVITEM);
29                     //獲取pszText
30                     ReadProcessMemory(hProcess, pszText, ItemBuf, 512, nullptr);
31                     QString str = QString::fromLocal8Bit(ItemBuf);
32                     //AfxMessageBox(str);
33                     if (str == strIconName)
34                     {
35                         ::SendMessage(hDeskWnd, LVM_GETITEMRECT, iItem, (LPARAM)pItemRc);
36                         ReadProcessMemory(hProcess, pItemRc, &rc, sizeof(RECT), nullptr);
37                         memcpy(lpRect, &rc, sizeof(RECT));
38                         bRet = true;
39                         break;
40                     }
41 
42                 }
43                 VirtualFreeEx(hProcess, pLVITEM, 0, MEM_RELEASE);
44                 VirtualFreeEx(hProcess, pszText, 0, MEM_RELEASE);
45                 VirtualFreeEx(hProcess, pItemRc, 0, MEM_RELEASE);//釋放內存
46             }
47             CloseHandle(hProcess);
48         }