查询进程加载的动态库

一个动态连接库，里面实现了几个常用的功能函数，比如把dll插入指定进程、查找一进程中调用了那些dll并显示dll的所在路径和关机等功能函数.

MyDll.h

#ifndef _MYDLLMYDLL #define _MYDLLMYDLL #include <windows.h> #include <Tlhelp32.h> #include <stdio.h> typedef struct Module{ char szExePath[MAX_PATH]; struct Module *next; }ModuleInfo,*pModule; //@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@函数@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@ int CloseComputer(); //关机 int ReComputer(); //重起 bool WINAPI EnablePrivilege(PCHAR PrivilegeName); //调整进程级别 bool InsertDll(LPCTSTR lpszDll,char* lpszProcess); //使用远程线程插入dll DWORD FindProcessId(char* lpszProcess); //查找进程PID pModule FindExeDll(char* exeName); //查找指定的进程调用的动态连接库 pModule FindDllExe(char* dllName); //查找指定的动态连接库被那些进程调用 pModule SnapshotProcess(); //进程快照 void SelfDelete(); //自删除函数 #endif

MyDll.cpp

/* * MyDll常用功能函数 * 无版权欢迎任意修改 * http://www.kusoft.org * By Wt0x00 Make:2007 */ #include "MyDll.h" /******************************************************/ 功能描述：调整进程的权限 参数说明： 1.PrivilegeName:要调整权限的名字。 /******************************************************/ bool WINAPI EnablePrivilege(PCHAR PrivilegeName) { HANDLE hProc,hToken; TOKEN_PRIVILEGES TP; //获得进程句柄 hProc=GetCurrentProcess(); //打开进程令牌环 if(!OpenProcessToken(hProc,TOKEN_ADJUST_PRIVILEGES,&hToken)) { return false; } //获得进程本地唯一ID if(!LookupPrivilegeValue(NULL,PrivilegeName,&TP.Privileges[0].Luid)) { CloseHandle(hToken); return false; } TP.Privileges[0].Attributes=SE_PRIVILEGE_ENABLED; TP.PrivilegeCount=1; //调整权限 if(!AdjustTokenPrivileges(hToken,false,&TP,sizeof(TP),0,0)) { CloseHandle(hToken); return false; } CloseHandle(hToken); return true; } //重起 int ReComputer() { //判断系统的版本 DWORD dwVersion = GetVersion(); //Windows 2000/XP/2003的需要调整进程的级别。 if(dwVersion < 0x80000000) { if(EnablePrivilege(SE_SHUTDOWN_NAME)) { return(ExitWindowsEx(EWX_SHUTDOWN|EWX_REBOOT,0)); } } else { return(ExitWindowsEx(EWX_SHUTDOWN|EWX_REBOOT,0)); } return 0; } //关机 int CloseComputer() { DWORD dwVersion = GetVersion(); if(dwVersion < 0x80000000) { if(EnablePrivilege(SE_SHUTDOWN_NAME)) { return(ExitWindowsEx(EWX_SHUTDOWN|EWX_POWEROFF,0)); } } else { return(ExitWindowsEx(EWX_SHUTDOWN|EWX_REBOOT,0)); } return 0; } /******************************************************/ 功能描述：查找进程并且返回其进程ID 参数说明： 1.lpszProcess:要查找的进程名。 /******************************************************/ DWORD FindProcessId(char* lpszProcess) { DWORD dwRet=0; //创建进程快照 HANDLE hSnapshot=CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS,0); //进程信息数据结构 PROCESSENTRY32 pe32; pe32.dwSize=sizeof(PROCESSENTRY32); //得到快照第一个进程信息 Process32First(hSnapshot,&pe32); do { strupr(pe32.szExeFile); strupr(lpszProcess); if(lstrcmpi(pe32.szExeFile,lpszProcess)==0) { dwRet=pe32.th32ProcessID; break; } }while(Process32Next(hSnapshot,&pe32));//获取下一个 CloseHandle(hSnapshot); return dwRet; } /******************************************************/ 功能描述：把DLL插入到指定的目标进程中 参数说明： 1.lpszDll：要插入的动态连接库名; 2.lpszProcess:要插入的进程。 返回值：插入成功返回TRUE，插入失败返回FALSE /******************************************************/ bool InsertDll(LPCTSTR lpszDll,char* lpszProcess) { DWORD dwSize,dwWritten,dwRet; char FilePath[256]; //获取DLL的路径 GetFullPathName(lpszDll,256,FilePath,NULL); dwSize = strlen(FilePath)+1; dwRet = FindProcessId(lpszProcess); if(0==dwRet) { return false; } if(!EnablePrivilege(SE_DEBUG_NAME)) { return false; } //打开进程，获得该进程句柄 HANDLE hProcess = OpenProcess(PROCESS_CREATE_THREAD|PROCESS_VM_OPERATION|PROCESS_VM_WRITE ,FALSE,dwRet); if(NULL==hProcess) { return false; } //在进程中申请一快内存 LPVOID lpBuf = (LPVOID)VirtualAllocEx(hProcess,NULL,dwSize,MEM_COMMIT,PAGE_READWRITE); if (lpBuf == NULL) { CloseHandle(hProcess); return false; } //把DLL写入申请到的内存 if(!WriteProcessMemory(hProcess,lpBuf,(LPVOID)FilePath,dwSize,&dwWritten)) { if(dwWritten!=dwSize) { VirtualFreeEx(hProcess,lpBuf,dwSize,MEM_DECOMMIT); CloseHandle( hProcess ); return false; } else CloseHandle(hProcess); return false; } DWORD dwID; //获得LoadLibraryA函数的地址 LPVOID pFunc = GetProcAddress(GetModuleHandle("kernel32.dll"),"LoadLibraryA"); //创建远程线程 HANDLE hThread = CreateRemoteThread(hProcess,NULL,0,(LPTHREAD_START_ROUTINE)pFunc ,lpBuf,0,&dwID); if(hThread == NULL) { VirtualFreeEx( hProcess, lpBuf, dwSize, MEM_DECOMMIT ); CloseHandle( hProcess ); return false; } if(WaitForSingleObject( hThread, INFINITE ) == WAIT_FAILED) { return false; } VirtualFreeEx( hProcess, lpBuf, dwSize, MEM_DECOMMIT ); CloseHandle( hThread ); CloseHandle( hProcess ); return true; } /*-----------------------进程操作相关函数------------------------*/ /******************************************************/ 功能描述：查找指定的进程调用的动态连接库 参数说明： 1.exeName：要查找的进程名; 返回值：如果成功返回链表的头head，失败返回NULL /******************************************************/ pModule FindExeDll(char* exeName) { bool FINDP = false; PROCESSENTRY32* pInfo = new PROCESSENTRY32; MODULEENTRY32* dInfo = new MODULEENTRY32; pInfo->dwSize = sizeof(PROCESSENTRY32); dInfo->dwSize = sizeof(MODULEENTRY32); ModuleInfo *p,*head = new ModuleInfo; head->next = NULL; ZeroMemory(head->szExePath,MAX_PATH); HANDLE hProcess=CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS,0); HANDLE hModule; EnablePrivilege(SE_DEBUG_NAME); Process32First(hProcess,pInfo); do { strupr(exeName); strupr(pInfo->szExeFile); if(!strcmp(exeName,pInfo->szExeFile)) { hModule=CreateToolhelp32Snapshot(TH32CS_SNAPMODULE,pInfo->th32ProcessID); Module32First(hModule, dInfo); do { p = new ModuleInfo; strcpy(p->szExePath,dInfo->szExePath); p->next = head->next; head->next = p; }while(Module32Next(hModule,dInfo)); FINDP = true; break; } }while(Process32Next(hProcess, pInfo)); if(FINDP) return head; return NULL; } /******************************************************/ 功能描述：查找指定的动态连接库被那些进程调用 参数说明： 1.dllName：要查找的动态连接库; 返回值：如果成功返回链表的头head，失败返回NULL /******************************************************/ pModule FindDllExe(char* dllName) { bool FINDP = false; PROCESSENTRY32* pInfo = new PROCESSENTRY32; MODULEENTRY32* dInfo = new MODULEENTRY32; pInfo->dwSize = sizeof(PROCESSENTRY32); dInfo->dwSize = sizeof(MODULEENTRY32); ModuleInfo *p,*head = new ModuleInfo; head->next = NULL; ZeroMemory(head->szExePath,MAX_PATH); HANDLE hProcess=CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS,0); HANDLE hModule; EnablePrivilege(SE_DEBUG_NAME); Process32First(hProcess,pInfo); do { hModule=CreateToolhelp32Snapshot(TH32CS_SNAPMODULE,pInfo->th32ProcessID); Module32First(hModule, dInfo); char temp[MAX_PATH]; strcpy(temp,dInfo->szExePath); do { strupr(dllName); strupr(dInfo->szModule); if(!strcmp(dllName,dInfo->szModule)) { p = new ModuleInfo; strcpy(p->szExePath,temp); p->next = head->next; head->next = p; #ifndef _FFFFFF #define _FFFFFF FINDP = TRUE; #endif } }while(Module32Next(hModule,dInfo)); }while(Process32Next(hProcess, pInfo)); if(FINDP) return head; return NULL; } /******************************************************/ 功能描述：创造所有进程的快照 返回值：如果成功返回链表的头head，失败返回NULL /******************************************************/ pModule SnapshotProcess() { PROCESSENTRY32* pInfo = new PROCESSENTRY32; MODULEENTRY32* dInfo = new MODULEENTRY32; pInfo->dwSize = sizeof(PROCESSENTRY32); dInfo->dwSize = sizeof(MODULEENTRY32); ModuleInfo *p,*head; if(head=new ModuleInfo,NULL == head)return NULL; head->next = NULL; ZeroMemory(head->szExePath,MAX_PATH); HANDLE hProcess=CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS,0); HANDLE hModule; EnablePrivilege(SE_DEBUG_NAME); Process32First(hProcess,pInfo); do { hModule=CreateToolhelp32Snapshot(TH32CS_SNAPMODULE,pInfo->th32ProcessID); Module32First(hModule, dInfo); if(p=new ModuleInfo,NULL==p)return NULL; strcpy(p->szExePath,dInfo->szExePath); p->next = head->next; head->next = p; }while(Process32Next(hProcess, pInfo)); return head; } /*------------------------------------------------------------*/ //自删除 void SelfDelete() { TCHAR szModule [MAX_PATH], szComspec[MAX_PATH], szParams [MAX_PATH]; //获取文件的路径和环境变量"COMSPEC"的值即mcd路径 if((GetModuleFileName(0,szModule,MAX_PATH)!=0) && (GetShortPathName(szModule,szModule,MAX_PATH)!=0) && (GetEnvironmentVariable("COMSPEC",szComspec,MAX_PATH)!=0)) { //设置命令行参数 lstrcpy(szParams," /c del "); lstrcat(szParams, szModule); lstrcat(szParams, " > nul"); lstrcat(szComspec, szParams); printf(szComspec); //设置数据结构数据 STARTUPINFO si={0}; PROCESS_INFORMATION pi={0}; si.cb = sizeof(si); si.dwFlags = STARTF_USESHOWWINDOW; si.wShowWindow = SW_HIDE; // increase resource allocation to program SetPriorityClass(GetCurrentProcess(), REALTIME_PRIORITY_CLASS); SetThreadPriority(GetCurrentThread(), THREAD_PRIORITY_TIME_CRITICAL); if(CreateProcess(0, szComspec, 0, 0, 0,CREATE_SUSPENDED| DETACHED_PROCESS, 0, 0, &si, &pi)) { SetPriorityClass(pi.hProcess,IDLE_PRIORITY_CLASS); SetThreadPriority(pi.hThread,THREAD_PRIORITY_IDLE); ResumeThread(pi.hThread); return; } else { SetPriorityClass(GetCurrentProcess(), NORMAL_PRIORITY_CLASS); SetThreadPriority(GetCurrentThread(), THREAD_PRIORITY_NORMAL); } } return; }