跨进程实现在Tree中快速定位节点
前些日子写软件时，需要实现一个功能，就是在Tree中快速定位节点，比如注册表编辑器左边的Tree，只要给出Tree中的节点路径（以“\”分隔），就可以快速将树展开，并将当前节点定位到指定的节点。功能的实现并不难，但稍有些麻烦。原因在于，如果是本进程中的Tree，只要发消息就可以了，但如果是另外一个进程中的Tree，就要在那个进程中申请内存，将Tree节点的文字复制到这块内存，然后再把这块内存的数据复制到本进程的一块内存中，才能与指定的节点路径相比较。由于这个功能还有一些可一般化的东西，所以就写了一个DLL，只要给出Tree的句柄和节点路径，就可以展开这颗树并定位节点。
DLL源代码如下：
|
/********************************************************************/ /* 文件名: Tree.cpp */ /* */ /* 功能: 标准 DLL ---- 跨进程展开 SysTreeView32 中指定的节点 */ /* */ /* 作者: 卢培培 (goodname008) 时间: 2005.02.18 */ /* */ /* BLOG: http://blog.csdn.net/goodname008 */ /********************************************************************/ #include "stdafx.h" #include "Tree.h" #include "commctrl.h" #include <string> using namespace std; BOOL APIENTRY DllMain( HANDLE hModule, DWORD ul_reason_for_call, LPVOID lpReserved) { switch (ul_reason_for_call) { case DLL_PROCESS_ATTACH: case DLL_THREAD_ATTACH: case DLL_THREAD_DETACH: case DLL_PROCESS_DETACH: break; } return TRUE; } /********************************************************************/ /* 功 能: 跨进程展开 SysTreeView32 中指定的节点 /* /* 参 数: hTreeWnd SysTreeView32 的句柄 /* lpszPath SysTreeView32 中的节点路径(忽略大小写) /* /* 返回值: TRUE 成功 /* FALSE 失败(节点路径不存在时会返回失败, 但仍然展开) /* /* 说 明: 在节点路径不存在的情况下, 本函数会尽可能展开存在的节点 /********************************************************************/ TREE_API BOOL APIENTRY ExpandTreeNode(HWND hTreeWnd, LPCSTR lpszPath) { string szPath = lpszPath; if (szPath.empty()) return FALSE; DWORD dwProcessID = NULL; GetWindowThreadProcessId(hTreeWnd, &dwProcessID); if (!dwProcessID) return FALSE; HANDLE hProcess = NULL; hProcess = OpenProcess(PROCESS_VM_OPERATION | PROCESS_VM_READ | PROCESS_VM_WRITE | PROCESS_QUERY_INFORMATION, FALSE, dwProcessID); if (!hProcess) return FALSE; TVITEM tvItem, *pItem = NULL; ZeroMemory(&tvItem, sizeof(TVITEM)); pItem = (TVITEM *)VirtualAllocEx(hProcess, NULL, sizeof(TVITEM), MEM_COMMIT, PAGE_READWRITE); tvItem.mask = TVIF_TEXT; tvItem.cchTextMax = 512; tvItem.pszText = (LPSTR)VirtualAllocEx(hProcess, NULL, 512, MEM_COMMIT, PAGE_READWRITE); tvItem.hItem = TreeView_GetRoot(hTreeWnd); if (!tvItem.hItem) return FALSE; string szPathNode; string::size_type nBackslashPos = -1; char szItemText[512] = {'\0'}; do { szPathNode = szPath.substr(nBackslashPos + 1, szPath.find('\\', nBackslashPos + 1) - nBackslashPos - 1); do { if (!WriteProcessMemory(hProcess, pItem, &tvItem, sizeof(TVITEM), NULL)) return FALSE; if (!TreeView_GetItem(hTreeWnd, pItem)) return FALSE; if (!ReadProcessMemory(hProcess, tvItem.pszText, szItemText, 512, NULL)) return FALSE; if (lstrcmpi(szPathNode.c_str(), szItemText) == 0) { TreeView_SelectItem(hTreeWnd, tvItem.hItem); if (TreeView_Expand(hTreeWnd, tvItem.hItem, TVE_EXPAND)) { tvItem.hItem = TreeView_GetChild(hTreeWnd, tvItem.hItem); if (!tvItem.hItem) return FALSE; } } else { tvItem.hItem = TreeView_GetNextSibling(hTreeWnd, tvItem.hItem); if (!tvItem.hItem) return FALSE; } } while(lstrcmpi(szPathNode.c_str(), szItemText) != 0); nBackslashPos = szPath.find('\\', nBackslashPos + 1); } while(nBackslashPos != -1); VirtualFreeEx(hProcess, tvItem.pszText, NULL, MEM_RELEASE); VirtualFreeEx(hProcess, pItem, NULL, MEM_RELEASE); CloseHandle(hProcess); return TRUE; } |
头文件源代码：
|
#ifdef TREE_EXPORTS #define TREE_API __declspec(dllexport) #else #define TREE_API __declspec(dllimport) #endif
TREE_API BOOL APIENTRY ExpandTreeNode(HWND hTreeWnd, LPCSTR lpszPath); |
DEF文件如下：
|
LIBRARY Tree
EXPORTS ExpandTreeNode @1 |
| 欢迎访问最专业的网吧论坛,无盘论坛,网吧经营,网咖管理,网吧专业论坛
https://bbs.txwb.com
关注天下网吧微信/下载天下网吧APP/天下网吧小程序，一起来超精彩
|
本文来源：vczx 作者：佚名
天下网吧 网吧天下
闽公网安备35010202000238号
