首先讲一下两方的通信过程:服务端启动后就进行监听,客户端主动连接服务端,连接成功后为其建立一个线程接收控制命令并进行处理。
下面讲解客户端的实现。
客户端的功能其实很简单,只要连接上服务端后就基本什么不用做了,当用户点击“发送控制”按钮后根据控制选项构造不同的命令进行发送。
下面是连接服务端的代码:
// 得到服务端IP BYTE ch1,ch2,ch3,ch4; m_edtServer.GetAddress(ch1,ch2,ch3,ch4); m_strServer.Format("%u.%u.%u.%u",ch1,ch2,ch3,ch4); WSADATA ws; int ret; struct sockaddr_in server; if(WSAStartup(MAKEWORD(2,2),&ws) != 0) { return; } if((sClient = socket(AF_INET,SOCK_STREAM,0))==INVALID_SOCKET) { return; } server.sin_family = AF_INET; server.sin_port = htons(m_nPort); server.sin_addr.s_addr = inet_addr(m_strServer); if(connect(sClient,(struct sockaddr *)&server,sizeof(server)) == 0) { m_edtStatus.SetWindowText("已连接上 TCP端口:12345"); } |
下面是“发送控制”按钮的响应函数,本示例程序中只有三种基本功能:信息发送(使对方弹出一个对话框,显示您所发送过去的信息)、系统控制(包括关机、重启、截获屏幕、弹出/关闭光驱5个子功能)、鼠标控制(包括随机移动、禁用输入、交换左右键3个子功能)。我使用了三个单选框来确定是哪类基本类型的控制,下拉框来进行子功能选择。因此每次控制要发送两次控制,第一次确定基本功能,第二次确定子功能。
char CmdBuffer[1024]; char CmdType[5]; CString strBuffer; int iSelect; HANDLE hThread; DWORD dwThread; // 构造命令 if(m_rdoMsg.GetCheck() == 1) { // 发送消息 m_edtMsg.GetWindowText(strBuffer); // 得到输入框中的内容 sprintf(CmdBuffer,"%s",strBuffer); sprintf(CmdType,"%c",’S’); } else if(m_rdoCtrl.GetCheck() == 1) { // 系统控制 // 只发送下拉框返回的选项索引号,服务端直接根据该索引确定子功能命令 iSelect = m_cmbCtrl.GetCurSel(); sprintf(CmdBuffer,"%d",iSelect); sprintf(CmdType,"%c",’C’); } else if(m_rdoMouse.GetCheck() == 1) { // 鼠标控制 iSelect = m_cmbMouse.GetCurSel(); sprintf(CmdBuffer,"%d",iSelect); sprintf(CmdType,"%c",’M’); } else { return; } // 首先发送命令基本类型 int ret = send(sClient,CmdType,strlen(CmdType) + 1,0); if((ret == SOCKET_ERROR) || (ret == 0)) { return; } // 发送子功能号 ret = send(sClient,CmdBuffer,strlen(CmdBuffer) + 1,0); if((ret == SOCKET_ERROR) || (ret == 0)) { return; } return; |
下面再贴一段如何使服务端截获到的屏幕图像显示在picture控件中,因为时间比较紧张,我没有把图像传输这一块儿做好,就给偷了个懒,服务端截获到屏幕后直接保存在C盘根目录下,而控制端直接到该位置读取,哈哈,专门跟老师说了下,老师考虑到时间限制就放我了一马,其实就是用某些编码算法将图片压缩一下就可以了,例如JPEG,懒得弄了。
HBITMAP hBitmap = (HBITMAP)LoadImage(AfxGetInstanceHandle(),"C:\\test.bmp", IMAGE_BITMAP,0,0,LR_CREATEDIBSECTION|LR_LOADFROMFILE); CDC *pDc = GetDlgItem(IDC_FILE_STATIC)->GetDC(); CDC screen; screen.CreateCompatibleDC(pDc); CRect rect; GetClientRect(rect); HBITMAP OldBitmap = (HBITMAP)screen.SelectObject(hBitmap); pDc->BitBlt(0,0,rect.Width() ,rect.Height() ,&screen,0,0,SRCCOPY); return; |
下面来看服务端,老规矩,先来看监听的函数。
WSADATA ws; int iAddrSize; HANDLE hThread; DWORD dwThread; struct sockaddr_in local,client; if(WSAStartup(MAKEWORD(2,2),&ws) != 0) { return; } if((sListen = socket(AF_INET,SOCK_STREAM,0))==INVALID_SOCKET) { return; } local.sin_family = AF_INET; local.sin_port = htons(12345); local.sin_addr.S_un.S_addr = INADDR_ANY; if(bind(sListen,(struct sockaddr*)&local,sizeof(struct sockaddr)) == SOCKET_ERROR) { closesocket(sListen); return; } listen(sListen,5); iAddrSize = sizeof(client); sClient = accept(sListen,(struct sockaddr *)&client,&iAddrSize); if(sClient == INVALID_SOCKET) { closesocket(sListen); return; } // 创建一个会话线程 hThread = CreateThread(NULL,0,ClientThread,(LPVOID)sClient,0,&dwThread); if(hThread == NULL) { return; } CloseHandle(hThread); closesocket(sListen); WSACleanup(); return; 再来看线程函数的代码。 SOCKET sock = (SOCKET)param; char szBuff[MAX_PATH]; int ret; while(1) { ret = recv(sock,szBuff,sizeof(szBuff),0); if(ret == 0) break; else if(ret == SOCKET_ERROR) break; else szBuff[ret] = ’\0’; switch(szBuff[0]) { case ’S’: // 显示消息 ret = recv(sock,szBuff,sizeof(szBuff),0); szBuff[ret] = ’\0’; ShowMessage(szBuff); break; case ’C’: // 系统控制 ret = recv(sock,szBuff,sizeof(szBuff),0); szBuff[ret] = ’\0’; ret = atoi(szBuff); SystemControl(ret); break; case ’M’: // 鼠标控制 ret = recv(sock,szBuff,sizeof(szBuff),0); szBuff[ret] = ’\0’; ret = atoi(szBuff); MouseControl(ret); break; default: break; } } return 0; |
接下来就是三个处理函数了,全部贴出来。注意这几个函数都要定义成全局的,不要定义成类的成员函数,那样的话在线程函数里面访问不到。
void ShowMessage(LPCTSTR msg) { ::MessageBox(NULL,msg,"信息",MB_OK); } void SystemControl(int select) { switch(select) { case 0: OpenCDoor(); // 打开光驱 break; case 1: CloseCDoor(); // 关闭光驱 break; case 2: SnapScreen(); // 截获屏幕 break; case 3: PreProcess(); // 提升进程权限 RebootSystem(); // 重启系统 break; case 4: PreProcess(); ShutDown(); //关闭系统 break; default: break; } } void MouseControl(int select) { int i = 0; int nX = 0; int nY = 0; switch(select) { case 0: // 25秒内随机移动鼠标 for(i = 0; i <= 49; i ++) { nX = rand() % 1024; nY = rand() % 768; ::SetCursorPos(nX,nY); Sleep(500); } break; case 1: // 10秒内锁定鼠标键盘响应 BlockInput(TRUE); Sleep(10000); BlockInput(FALSE); break; case 2: // 10秒内交换鼠标左右键¸´ SwapMouseButton(TRUE); Sleep(10000); SwapMouseButton(FALSE); break; default: break; } } 最后把一些函数的详细过程贴出来。 //截获屏幕,该段代码来自《黑客防线》 BOOL SnapScreen() { CDC dc; int nWidth; int nHeight; dc.CreateDC("DISPLAY",NULL,NULL,NULL); nWidth = GetDeviceCaps(dc,HORZRES); nHeight = GetDeviceCaps(dc,VERTRES); CDC dcMem; dcMem.CreateCompatibleDC(&dc); CBitmap bitmap; bitmap.CreateCompatibleBitmap(&dc,nWidth,nHeight); CBitmap *pOldBitmap = dcMem.SelectObject(&bitmap); dcMem.BitBlt(0,0,nWidth,nHeight,&dc,0,0,SRCCOPY); dcMem.SelectObject(pOldBitmap); CString strFile = "C:\\test.bmp"; SaveBitmapToFile(dc.GetSafeHdc(),bitmap,strFile); dc.DeleteDC(); return TRUE; } // SaveBitmapToFile函数将截获到的图像保存问bmp文件 BOOL SaveBitmapToFile(HDC hDc, CBitmap &bitmap, LPCTSTR lpszFileName) { BOOL ret = TRUE; BITMAP btm; bitmap.GetBitmap(&btm); DWORD size = btm.bmWidthBytes * btm.bmHeight; HGLOBAL hMem = GlobalAlloc(GMEM_FIXED|GMEM_ZEROINIT,size); if(hMem == NULL) return FALSE; LPSTR lpData = (LPSTR)GlobalLock(hMem); BITMAPINFOHEADER bih; bih.biSize = sizeof(BITMAPINFOHEADER); bih.biWidth = btm.bmWidth; bih.biHeight = btm.bmHeight; bih.biPlanes = 1; bih.biBitCount = btm.bmBitsPixel; bih.biCompression = 0; bih.biSizeImage = size; bih.biXPelsPerMeter = 0; bih.biYPelsPerMeter = 0; bih.biClrUsed = 0; bih.biClrImportant = 0; if(GetDIBits(hDc,bitmap,0,bih.biHeight,lpData,(BITMAPINFO *) &bih,DIB_RGB_COLORS) == 0) { GlobalFree(hMem); return FALSE; } BITMAPFILEHEADER bfh; bfh.bfType = ((WORD)(’M’ << 8) | ’B’); bfh.bfReserved1 = 0; bfh.bfReserved2 = 0; bfh.bfSize = sizeof(bfh) + size; bfh.bfOffBits = sizeof(bfh); CFile bf; if(bf.Open(lpszFileName,CFile::modeCreate|CFile::modeWrite)) { bf.WriteHuge(&bfh,sizeof(BITMAPFILEHEADER)); bf.WriteHuge(&bih,sizeof(BITMAPINFOHEADER)); bf.WriteHuge(lpData,size); bf.Close(); } else ret = FALSE; GlobalFree(hMem); return ret; } // 提升进程权限的代码 BOOL PreProcess() { HANDLE hToken; TOKEN_PRIVILEGES tkp; if(OpenProcessToken(GetCurrentProcess(), TOKEN_ADJUST_PRIVILEGES|TOKEN_QUERY,&hToken) == 0) return FALSE; if(LookupPrivilegeValue(NULL,SE_SHUTDOWN_NAME,&tkp.Privileges[0].Luid) == 0) return FALSE; tkp.PrivilegeCount=1; tkp.Privileges[0].Attributes=SE_PRIVILEGE_ENABLED; if(AdjustTokenPrivileges(hToken,false,&tkp,0,(PTOKEN_PRIVILEGES)NULL,0) == 0) return FALSE; return TRUE; } |