RCPServer.cpp
上传用户:hyz2004817
上传日期:2022-03-30
资源大小:226k
文件大小:25k
源码类别:

远程控制编程

开发平台:

Visual C++

  1. // RCPServer.cpp : Defines the entry point for the application.
  2. //
  3. #include "stdafx.h"
  4. #include "resource.h"
  5. //#include "WSAErrorString.h"
  6. //#include "id_handler.h"
  7. // 时间的限制,以分钟为单位
  8. #define TIME_LIMIT 10
  9. // 服务器端主窗口的类名
  10. #define SERVER_WINDOW_CLASSNAME "drm_rca_server_window_class"
  11. // 服务器端主窗口的标题名
  12. #define SERVER_WINDOW_CAPTIONNAME "RCA 服务器"
  13. // 客户端主窗口的类名
  14. #define CLIENT_WINDOW_CLASSNAME "drm_rca_client_window_class"
  15. // 用于创建命名信标对象,控制服务器同一时刻最多只能运行两个实例
  16. #define RCA_RUNNING_CONTROL_SEMAPHORE_NAME "rca_running_semphore_control"
  17. // 当前软件的版本号, 语法 MAKEWORD( 次版本号, 主版本号)
  18. #define CURRENT_VERSION MAKEWORD( 0, 2 )
  19. // 自定义版本更新消息, WPARAM 为版本验证值,LPARAM 为传送消息的线程的 ID
  20. // WM_VERSIONUPDATE 消息有两个类型, 一种为
  21. #define WM_VERSIONUPDATE (WM_USER +3)
  22. #define VERSION_HIGH 1
  23. #define VERSION_LOW 2
  24. #define VERSION_EQUAL 3
  25. // 自定义消息 假如正在工作的 RCA 版本较低, 一个更高版本的 RCA 将会通知低版本的退出
  26. // 然后高版本的将等待低版本的传递 WM_OLDVERSIONEXIT 消息,以表示低版本的已经退出
  27. // 这时高版本的将可以继续运行
  28. #define WM_OLDVERSIONEXIT (WM_USER +4)
  29. #define WM_RCASHELLICON (WM_USER +5)
  30. WORD winsock_version = MAKEWORD(2, 2);
  31. WSADATA wsadata;
  32. ///////////////////////////
  33. //RCAInit 函数初始化的变量
  34. CHAR g_pszOS[6]; // OS version info
  35. CHAR g_pszComputerName[MAX_COMPUTERNAME_LENGTH+1]; // computer Name
  36. CHAR g_pszUserName[UNLEN +1]; // current thread's user name
  37. CHAR* g_pszSystemDir; // system directory
  38. CHAR* g_pszWindowsDir; // windows setup directory
  39. HINSTANCE g_hInstance;
  40. OSVERSIONINFO g_SysVer; // OS version struct
  41. MEMORYSTATUS g_MemoryStatus; // memory status struct 
  42. BOOL g_bCanAcceptEx; // 初始化时确定的布尔变量, 说明操作系统是否支持 
  43. // AcceptEx函数
  44. HWND g_hWnd; // main window handle;
  45. NOTIFYICONDATA g_nid;
  46. HICON g_hBig;
  47. HICON g_hSmall;
  48. HMENU g_hShellIconMenu;
  49. HANDLE g_hRunningControlSemaphore;
  50. DWORD g_dwAnotherRCAVertionUpdateThreadId;
  51. BOOL g_bMustExit = FALSE;
  52. HANDLE g_hExitEvent;
  53. //////////////////////////////////////////////
  54. //
  55. // 服务开始使用的全局变量
  56. //
  57. HANDLE g_hServiceMainThreadHandle = NULL;
  58. DWORD g_dwServiceMainThreadID = NULL;
  59. BOOL g_bServiceMainThreadMustExit = FALSE;
  60. SOCKET g_sockService = NULL;
  61. SOCKADDR_IN g_addrServer; // 服务器地址描述
  62. WORD g_wServerPort; // 服务器端口号, 网络字节顺序
  63. ///////////////////////////////////////////////
  64. //
  65. //  有关安全方面的全局变量
  66. //
  67. BOOL g_bAllAccess = TRUE;
  68. ///////////////////////////////////////////////////
  69. //
  70. // 函数原型声明
  71. //
  72. BOOL WINAPI RCAInit();
  73. BOOL WINAPI RCAUninit();
  74. BOOL WINAPI CanGoon( DWORD * result ); 
  75. void WINAPI OutputAddr( unsigned int uIP );
  76. LRESULT CALLBACK MainWndProc( HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam );
  77. DWORD WINAPI RCAService( PVOID pVoid );
  78. BOOL WINAPI ServiceMainThreadStartup();
  79. BOOL WINAPI ServiceInit();
  80. BOOL WINAPI ServiceUninit();
  81. BOOL WINAPI AccessControl( RCAREQUESTHEADER * pRequestHead, DWORD dwRemoteAddress, WORD wPort );
  82. DWORD WINAPI RequestHandle( SOCKET accept, DWORD dwRemoteAddress );
  83. BOOL WINAPI GetRCARequestIDHandler( DWORD dwRequestID, RCAREQUESTIDHANDLER * pHandler);
  84. /////////////////////////////////////////////////
  85. //
  86. // 通过客户端请求的操作 ID , 查找相应的 ID 处理函数
  87. //
  88. //
  89. BOOL WINAPI GetRCARequestIDHandler( DWORD dwRequestID, RCAREQUESTIDHANDLER * pHandler)
  90. {
  91. switch ( dwRequestID )
  92. {
  93. case 0x10: // 16 进制 10 号, 10 进制 16 号, 创建进程执行 
  94. *pHandler = RCAID_0X10_HANDLER;
  95. break;
  96. case 0x20:
  97. *pHandler = RCAID_0X20_HANDLER;
  98. break;
  99. case 0x40:
  100. *pHandler = RCAID_0X40_HANDLER;
  101. break;
  102. case 0x41:
  103. *pHandler = RCAID_0X41_HANDLER;
  104. break;
  105. case 0x110:
  106. *pHandler = RCAID_0X110_HANDLER;
  107. break;
  108. default:
  109. return FALSE;
  110. }
  111. return TRUE;
  112. }
  113. //////////////////////////////////////////////////////
  114. //
  115. // 说明:
  116. // 一个客户端的请求已经到来, 先分析请求是否为 RCA 请求
  117. // 请求 如果是一个 RCA 请求, 分两步处理:
  118. // 1. 对请求的访问控制, 这将涉及到发出请求的客户端 IP 地址,
  119. // 请求的操作 ID. 这些由 AccessControl 函数完成, 如果 AccessControl 
  120. // 返回 TRUE , 则允许执行这个请求, 返回 FALSE, 将拒绝请求.
  121. //
  122. // 2.  如果允许执行请求, 将调用 ID 分流模块, 
  123. // 以将请求正确的传递到相应的处理程序
  124. //
  125. DWORD WINAPI RequestHandle( PVOID p )
  126. {
  127. RCAREQUESTHANDLEDATA * pData = (RCAREQUESTHANDLEDATA *)p;
  128. SOCKET sockRemote = pData->sockRemote;
  129. HANDLE hEventArray[2];
  130. char headBuffer[40];
  131. PRCAREQUESTHEADER pRCAHead = NULL;
  132. int ret;
  133. int nBytes = sizeof(RCAREQUESTHEADER);
  134. int nLeft = nBytes;
  135. int i = 0;
  136. BOOL bExit = FALSE;
  137. hEventArray[0] = CreateEvent( NULL , FALSE, FALSE, NULL );
  138. hEventArray[1] = g_hExitEvent;
  139. if( hEventArray[0] == NULL )
  140. goto end_request_handle;
  141. ret = WSAEventSelect( sockRemote, hEventArray[0], FD_READ | FD_WRITE | FD_CLOSE );
  142. if( ret != 0 )
  143. {
  144. DEBUGOUTPUT1( WSAERRORSTRING );
  145. exit(0);
  146. }
  147. BOOL bRecv = TRUE;
  148. while( 1 )
  149. {
  150. ret = RCARecv_EventSelectIO( sockRemote, hEventArray, headBuffer, sizeof(RCAREQUESTHEADER));
  151. if( ret != sizeof(RCAREQUESTHEADER) )
  152. break;
  153. pRCAHead = (RCAREQUESTHEADER*)headBuffer;
  154. // 确定前四个字节为一个 ANSI C 字符串 "RCA"
  155. if( 0 != lstrcmp( pRCAHead ->rcaID, "RCA" ) )
  156. break;
  157. // 确保版本支持, 当前只支持 1.0 的版本
  158. // 保证协议版本在 1.0 或 1.0 以下
  159. if( pRCAHead->wMajorVersion > 1 )
  160. break;
  161. if( (pRCAHead->wMajorVersion == 1) && (pRCAHead->wMinorVersion!=0) )
  162. break;
  163. /////////////////////////////////////
  164. //
  165. // 至此验证完毕. 大致上符合一个 RCA 的请求
  166. // 现在将调用 AccessControl 函数, 确定此请求是否接受
  167. if( !AccessControl( pRCAHead, pData->dwRemoteAddress, (WORD)pData->dwRemotePort ) )
  168. break;
  169. RCAREQUESTIDHANDLER pFunc; // ID 处理函数的指针
  170. //nBytes = pRCAHead->requestBytes - 40; // 需要接收的字节数, 为请求头中 requestBytes 字段 减去 40
  171. if( GetRCARequestIDHandler( pRCAHead->dwRequestID, &pFunc) )
  172. {
  173. if( TRUE == (*pFunc)( sockRemote , pData, pRCAHead, hEventArray) )
  174. {
  175. nLeft = sizeof(RCAREQUESTHEADER); // 重置接收参数
  176. i = 0; // 重置接收参数
  177. continue; // 继续接收下一个请求
  178. }
  179. else
  180. break;  // 结束请求处理
  181. // 省略了一些..... 以后加入
  182. }
  183. }
  184.  // 结束请求并释放资源
  185. end_request_handle:
  186. if( hEventArray[0] != NULL )
  187. CloseHandle( hEventArray[0] );
  188. // shutdown( pData->sockRemote, SD_BOTH );
  189. closesocket( pData->sockRemote );
  190. free( p ); // 释放
  191. return 0;
  192. }
  193. //////////////////////////////////////////////////////
  194. //
  195. // 访问控制的实现函数
  196. // 
  197. BOOL WINAPI AccessControl( RCAREQUESTHEADER * pRequestHead, DWORD dwRemoteAddress, WORD wPort )
  198. {
  199. // 现在不实现. 当前版本将接受所有的请求 
  200. return TRUE;
  201. }
  202. ////////////////////////////////////////////////////
  203. //
  204. // 当监听服务线程结束时必须调用的清除资源的操作
  205. BOOL WINAPI ServiceUninit()
  206. {
  207. int ret;
  208. closesocket( g_sockService );
  209. ret = WSACleanup();
  210. if ( ret != 0 )
  211. {
  212. DEBUGOUTPUT1( WSAERRORSTRING );
  213. return FALSE;
  214. }
  215. return TRUE;
  216. }
  217. ////////////////////////////////////////////////////////////
  218. //
  219. // 监听服务线程需要的初始化操作
  220. //
  221. BOOL WINAPI ServiceInit()
  222. {
  223.   char str[256];
  224. str[0] = '0';
  225. int ret;
  226. ret = WSAStartup( winsock_version, &wsadata );
  227. if( ret != 0 )
  228. {
  229. DEBUGOUTPUT1( WSAERRORSTRING );
  230. return FALSE;
  231. }
  232. g_sockService = socket( AF_INET, SOCK_STREAM, IPPROTO_TCP );
  233. if( g_sockService == INVALID_SOCKET )
  234. {
  235. DEBUGOUTPUT1( WSAERRORSTRING );
  236. return FALSE;
  237. }
  238. g_wServerPort = htons( 4813);
  239. g_addrServer.sin_family = AF_INET;
  240. g_addrServer.sin_port = g_wServerPort;  // 4813
  241. g_addrServer.sin_addr.S_un.S_addr = htonl( INADDR_ANY ); // 监听任何地址
  242. ret = bind( g_sockService,  (SOCKADDR*)&g_addrServer, sizeof( g_addrServer ) );
  243. if( ret != 0 )
  244. {
  245. DEBUGOUTPUT2( WSAERRORSTRING, "无法将套接字绑定到本地地址" );
  246. closesocket( g_sockService );
  247. return FALSE;
  248. }
  249. return TRUE;
  250. }
  251. ////////////////////////////
  252. //
  253. // 启动主服务线程
  254. //
  255. // 参数: 无
  256. //
  257. // 返回值: TRUE --- 服务线程已启动
  258. // FALSE --- 服务线程无法启动
  259. //
  260. BOOL WINAPI ServiceMainThreadStartup()
  261. {
  262. g_hServiceMainThreadHandle = 
  263. CreateThread( 
  264. NULL, // 默认的安全属性
  265. 0, // 默认的堆栈大小
  266. RCAService, // 线程函数是 RCAService
  267. NULL, // 无参数
  268. CREATE_SUSPENDED, // 线程创建后将其挂起
  269. &g_dwServiceMainThreadID // 将线程 ID 保存在全局变量 g_dwServiceMainThreadID 中
  270. );
  271. if( g_hServiceMainThreadHandle == 0 )
  272. return FALSE;
  273. else
  274. return TRUE;
  275. }
  276. DWORD WINAPI RCAService( PVOID pVoid )
  277. {
  278. int ret;
  279. static DWORD first = GetTickCount();
  280. static DWORD last;
  281. first = GetTickCount();
  282. if( listen( ::g_sockService, 64 ) != 0 )
  283. {
  284. PostMessage( g_hWnd, WM_DESTROY, 0, 0 );
  285. return 0;
  286. }
  287. SOCKET sockAccept;
  288. HANDLE hEventArray[2];
  289. SOCKADDR_IN clientAddr;
  290. int addrLen;
  291. DWORD dwTempThreadID;
  292. DWORD dwWaitResult;
  293. HANDLE hTemp;
  294. WSANETWORKEVENTS networkEvents;
  295. addrLen = sizeof( clientAddr );
  296. // 创建事件对象, 并将监听插座置为异步模式, 此异步模型为异步事件模型
  297. hEventArray[0] = CreateEvent( NULL, FALSE, FALSE, NULL );
  298. ret = WSAEventSelect( g_sockService, hEventArray[0], FD_ACCEPT );
  299. if( ret != 0 )
  300. {
  301. DEBUGOUTPUT1( WSAERRORSTRING );
  302. exit(0); // 危险代码 !
  303. }
  304. hEventArray[1] = g_hExitEvent;
  305. while( 1 )
  306. {
  307. //服务主线程开始对指定端口号进行监听,最大连接缓冲 64 个
  308. sockAccept = accept( g_sockService , (SOCKADDR*)&clientAddr, &addrLen );
  309. if( sockAccept == INVALID_SOCKET )
  310. {
  311. if( WSAEWOULDBLOCK != WSAGetLastError() )
  312. {
  313. DEBUGOUTPUT1( WSAERRORSTRING );
  314. exit(0);
  315. }
  316. } else
  317. {
  318. // 注: 这里必须在堆上分配内存.
  319. // 一个很常见的编程错误就是给一个刚创建的线程传递在栈上分配的数据
  320. RCAREQUESTHANDLEDATA * pData = 
  321. (RCAREQUESTHANDLEDATA*) new BYTE[sizeof(RCAREQUESTHANDLEDATA)];
  322. if( pData == NULL ) // 内存不足, 很少见的情况
  323. {
  324. closesocket( sockAccept );
  325. continue;
  326. }
  327. pData->sockRemote = sockAccept;
  328. pData->dwRemoteAddress = clientAddr.sin_addr.S_un.S_addr;
  329. pData->dwRemotePort = clientAddr.sin_port;
  330. hTemp = CreateThread (
  331. NULL,
  332. 0,
  333. RequestHandle,
  334. pData,
  335. 0,
  336. &dwTempThreadID);
  337. if( hTemp != 0 )
  338. CloseHandle( hTemp );
  339. continue;
  340. }
  341. if ( (GetTickCount() - first) > TIME_LIMIT * 60 * 1000 )
  342. {
  343. PostMessage(g_hWnd, WM_DESTROY, 0, 0 );
  344. return 0;
  345. }
  346. dwWaitResult = WaitForMultipleObjects( 2,  hEventArray, FALSE, INFINITE );
  347. // 一个网络事件发生后, 用 WSAEnumNetworkEvents 列举网络事件
  348. if( dwWaitResult == (WAIT_OBJECT_0 + 1) )
  349. {
  350. CloseHandle( hEventArray[0] );
  351. return 0;
  352. }
  353. if( dwWaitResult == WAIT_OBJECT_0 )
  354. {
  355. ret = WSAEnumNetworkEvents( g_sockService, hEventArray[0], &networkEvents );
  356. if( networkEvents.lNetworkEvents & FD_ACCEPT )
  357. {
  358. // FD_ACCEPT 事件没产生错误,将再次 accept( ... ) 以接受连接
  359. if( networkEvents.iErrorCode[ FD_ACCEPT_BIT ] == 0 )
  360. {
  361. // 接受连接
  362. continue;
  363. else
  364. {
  365. DEBUGOUTPUT1( WSAERRORSTRING );
  366. }
  367. }
  368. }// service while
  369. CloseHandle( hEventArray[0] );
  370. return 0;
  371. }
  372. VOID WINAPI OnShellIconMessage( HWND hwnd, WPARAM w, LPARAM l )
  373. {
  374. static int x;
  375. static int y;
  376. POINT p;
  377. // 筛选图标 ID
  378. switch( w )
  379. {
  380. case 1: // RCA SERVER ICON
  381. // 筛选消息
  382. switch( l )
  383. {
  384. case WM_RBUTTONUP:
  385. GetCursorPos( &p );
  386. // 假如选中了"退出"命令
  387. if( 2 == TrackPopupMenuEx( g_hShellIconMenu, 
  388. TPM_RIGHTBUTTON | TPM_NONOTIFY | TPM_RETURNCMD | 
  389. TPM_BOTTOMALIGN | TPM_RIGHTALIGN, p.x, p.y ,hwnd, NULL ) )
  390. DestroyWindow( hwnd );
  391. break;
  392. }
  393. break;
  394. }
  395. }
  396. LRESULT CALLBACK MainWndProc( HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam )
  397. {
  398. switch( msg )
  399. {
  400. case WM_RCASHELLICON:
  401. OnShellIconMessage( hwnd, wParam, lParam );
  402. return 0;
  403. case WM_CREATE:
  404. g_nid.cbSize = sizeof(NOTIFYICONDATA );
  405. g_nid.hIcon  = g_hSmall;
  406. g_nid.hWnd  = hwnd;
  407. lstrcpy( g_nid.szTip, "RCA SERVER v0.8 beta n使用限时:10分 n端口: 4813" );
  408. g_nid.uCallbackMessage = (WM_RCASHELLICON);
  409. g_nid.uID    = 1;
  410. g_nid.uFlags    = NIF_ICON | NIF_MESSAGE | NIF_TIP;
  411. return Shell_NotifyIcon( NIM_ADD, &g_nid );
  412. case WM_VERSIONUPDATE:
  413. DWORD iCmp;
  414. if( CURRENT_VERSION > wParam ) 
  415. iCmp = (DWORD)VERSION_LOW;
  416. else if( CURRENT_VERSION == wParam )
  417. iCmp = (DWORD)VERSION_EQUAL;
  418. else
  419. iCmp = (DWORD)VERSION_HIGH;
  420. PostThreadMessage( lParam, WM_VERSIONUPDATE, iCmp, CURRENT_VERSION );
  421. if( iCmp == VERSION_HIGH ) // 正在运行的 RCA 版本较低,应该退出,现在尝试退出
  422. {
  423. DestroyWindow( hwnd );
  424. PostThreadMessage( lParam, WM_OLDVERSIONEXIT, 0, 0 );
  425. }
  426. return 0;
  427. case WM_DESTROY:
  428. g_bMustExit = TRUE;
  429. DWORD ret;
  430. SetEvent( g_hExitEvent );
  431. ret = WaitForSingleObject ( ::g_hServiceMainThreadHandle, 10000 );
  432. if ( ret != WAIT_OBJECT_0 )
  433. {
  434. MessageBox( hwnd, "严重错误", "报告", MB_ICONINFORMATION );
  435. exit(0);
  436. }
  437. PostQuitMessage(0);
  438. CloseHandle( g_hExitEvent );
  439. Shell_NotifyIcon( NIM_DELETE, &g_nid );
  440. break;
  441. case WM_CLOSE:
  442. DestroyWindow( hwnd );
  443. break;
  444. case WM_TIMER:
  445. if( wParam == 3 )
  446. DestroyWindow( hwnd );
  447. break;
  448. default:
  449. return DefWindowProc(hwnd, msg, wParam, lParam );
  450. }
  451. return DefWindowProc(hwnd, msg, wParam, lParam );
  452. }
  453. INT WINAPI WinMain( HINSTANCE, HINSTANCE, LPSTR, INT )
  454. //int _cdecl main()
  455. {
  456. //ShellExecute( NULL, NULL, "notepad", "", NULL, SW_SHOWNORMAL );
  457. g_hInstance = GetModuleHandle(NULL);
  458. // 执行必须的初始化
  459. if( RCAInit() )
  460. {
  461. DWORD dwResult;
  462. //////////////////////////////////////
  463. // 
  464. // CanGoon 函数进行必要的版本控制. 如果函数返回 TRUE,
  465. // 将表示当前这个 RCAServer 可以运行,具体说明参考 CanGoon 函数
  466. // 
  467. //
  468. if( CanGoon( &dwResult ) ) 
  469. {
  470. g_hExitEvent = CreateEvent( NULL, TRUE, FALSE, NULL );
  471. // 创建服务主线程, 线程将监听客户的请求
  472. if( ServiceInit() == TRUE )
  473. {
  474. if( ServiceMainThreadStartup() == TRUE )
  475. {
  476. ResumeThread( ::g_hServiceMainThreadHandle );
  477. } else
  478. goto init_failed;
  479. } else
  480. goto init_failed;
  481. SetTimer( g_hWnd, 3, TIME_LIMIT * 60 * 1000, NULL );
  482. MSG msg;
  483. while( GetMessage( &msg, NULL, 0, 0 ) )
  484. {
  485.  
  486. TranslateMessage( &msg );
  487. DispatchMessage( &msg );
  488. }
  489. init_failed:
  490. if( g_hRunningControlSemaphore != NULL )
  491. ReleaseSemaphore( g_hRunningControlSemaphore, 1, NULL );
  492. SAFE_CLOSE_HANDLE( g_hRunningControlSemaphore );
  493. ServiceUninit();
  494. }
  495. RCAUninit();
  496. }
  497. else 
  498. {
  499. //RCAUninit();
  500. return -1;
  501. }
  502. return 0;
  503. }
  504. ///////////////////////////////////////////////
  505. //
  506. // 函数原型: BOOL RCAInit();
  507. //
  508. // 功能: 执行时必须进行的初始化部分
  509. //
  510. // 返回值 TRUE -- 初始化成功, RCA将继续运行
  511. // FALSE-- 失败, RCA 必须退出
  512. //
  513. BOOL WINAPI RCAInit() 
  514. {
  515. // 获取版本信息 ---- g_SysVer;
  516. g_SysVer.dwOSVersionInfoSize = sizeof(OSVERSIONINFO);
  517. GetVersionEx( &g_SysVer);
  518. switch( g_SysVer.dwPlatformId )
  519. {
  520. case VER_PLATFORM_WIN32s:// Windows 3.x , return FALSE, 不支持
  521. return FALSE;
  522. case VER_PLATFORM_WIN32_WINDOWS: // Win9x/ME
  523. if( g_SysVer.dwMinorVersion == 0 )
  524. lstrcpy( g_pszOS, "WIN95" );
  525. else
  526. lstrcpy( g_pszOS, "WIN98" );
  527. g_bCanAcceptEx = FALSE; // Win9x/ME 不支持 AcceptEx 函数
  528. break;
  529. case VER_PLATFORM_WIN32_NT: // Windows NT/2000;
  530. switch( g_SysVer.dwMajorVersion )
  531. {
  532. case 4:
  533. lstrcpy( g_pszOS, "WINNT" ); break;
  534. case 5:
  535. if( g_SysVer.dwMinorVersion > 0 )
  536. lstrcpy( g_pszOS, "WINXP" ); break;
  537. }
  538. else
  539. {
  540. lstrcpy( g_pszOS, "WIN2K" ); break;
  541. }
  542. default:
  543. return FALSE;
  544. }
  545. g_bCanAcceptEx = TRUE; // Windows NT/2000 支持 套接字函数 AcceptEx
  546. break;
  547. default: // 无法识别
  548. return FALSE;
  549. }
  550. // 获取计算机名 g_pszComputerName
  551. DWORD dwSize = MAX_COMPUTERNAME_LENGTH +1;
  552. if( !GetComputerName( g_pszComputerName, &dwSize ) )
  553. {
  554. DEBUGOUTPUT1( "Call GetComputerName failed" );
  555. return FALSE;
  556. }
  557. DEBUGOUTPUT2( "计算机名: ", g_pszComputerName );
  558. //////////////////////////////////////////////////
  559. // 获系统目录路径 g_pszSystemDir;
  560. //
  561. // example:
  562. // for win9x c:Windowssystem
  563. // for winnt/2000 c:winntsystem32
  564. dwSize = 0;
  565. dwSize = GetSystemDirectory( NULL, 0 );
  566. if( dwSize == NULL )
  567. return FALSE;
  568. dwSize ++;
  569. g_pszSystemDir = NULL;
  570. g_pszSystemDir = new CHAR[ dwSize ];
  571. if( g_pszSystemDir == NULL )
  572. {
  573. DEBUGOUTPUT1( "ERROR - not enough memory" );
  574. return FALSE;
  575. }
  576. dwSize = GetSystemDirectory( g_pszSystemDir, dwSize );
  577. DEBUGOUTPUT2( "当前系统目录是: ", g_pszSystemDir );
  578. ////////////////////////////////////////////////////
  579. // 获取当前用户名
  580. dwSize = UNLEN +1;
  581. if( ! GetUserName( g_pszUserName, &dwSize ) )
  582. {
  583. DEBUGOUTPUT1( "Call GetUserName failed" );
  584. return FALSE;
  585. }
  586. DEBUGOUTPUT2( "当前用户名:", g_pszUserName );
  587. /////////////////////////////////////////////////////
  588. // 获取 Windows  目录
  589. // 
  590. // example: 
  591. // for Win9x C:Windows
  592. // for WinNT/2000 C:winnt
  593. dwSize = 0;
  594. dwSize = GetWindowsDirectory( NULL, 0 );
  595. if( dwSize == NULL )
  596. return FALSE;
  597. dwSize ++;
  598. g_pszWindowsDir = NULL;
  599. g_pszWindowsDir = new CHAR[ dwSize ];
  600. if( g_pszWindowsDir == NULL )
  601. {
  602. DEBUGOUTPUT1( "ERROR - not enough memory" );
  603. delete[] g_pszSystemDir;
  604. return FALSE;
  605. }
  606. dwSize = GetWindowsDirectory( g_pszWindowsDir, dwSize );
  607. DEBUGOUTPUT2( "Windows 目录是: ", g_pszWindowsDir );
  608. ////////////////////////////////////////////////////////
  609. // 获取内存信息
  610. //
  611. GlobalMemoryStatus( &g_MemoryStatus );
  612. g_hBig = LoadIcon( g_hInstance, MAKEINTRESOURCE(IDI_BIG) );
  613. g_hSmall = LoadIcon( g_hInstance, MAKEINTRESOURCE(IDI_SMALL) );
  614. g_hShellIconMenu = CreatePopupMenu();
  615. AppendMenu( g_hShellIconMenu,
  616. MF_STRING, 1, "打开 RCA 服务管理器(&O) (beta版未实现)" );
  617. AppendMenu( g_hShellIconMenu,
  618. MF_SEPARATOR, 0, NULL );
  619. AppendMenu( g_hShellIconMenu,
  620. MF_STRING, 2, "退出(&E)" );
  621. return TRUE;
  622. }
  623. /////////////////////////////////////////////////////////////
  624. //
  625. // 原型: BOOL RCAUninit();
  626. // 
  627. // 功能: 程序退出时,进行清除操作
  628. //
  629. // 返回值: 返回 TRUE 表示成功清除, FALSE 表示失败
  630. //
  631. BOOL WINAPI RCAUninit()
  632. {
  633. DestroyMenu( g_hShellIconMenu );
  634. SAFE_DELETE_ARRAY( g_pszSystemDir );
  635. SAFE_DELETE_ARRAY( g_pszWindowsDir );
  636. return TRUE;
  637. }
  638. //////////////////////////////////////////////////////////////
  639. //
  640. // 原型: void OutputAddr( unsigned int u );
  641. //
  642. // 功能: 在控制台输出一个描述 IP 地址的无符号整数
  643. //
  644. // 注: 这个无符号整数已经被字节反序编码过的,
  645. // 输出时最低字节将被先输出,最高字节最后输出
  646. //
  647. // 返回值: 无
  648. //
  649. void WINAPI OutputAddr( unsigned int u )
  650. {
  651. cout<< ((u & 0xFF000000) >> 24) << "." 
  652. << ((u & 0x00ff0000) >> 16) << "."
  653. << ((u & 0x0000ff00) >> 8) << "."
  654. << ((u & 0x000000ff) >> 0) ;
  655. }
  656. ////////////////////////////////////////////////////////////
  657. //
  658. // CanGoon 函数进行必要的版本控制. 如果函数返回 TRUE,
  659. // 将表示当前这个 RCAServer 可以运行,这将发生在以下两种情况:
  660. // 
  661. // 1. 当前系统没有任何 RCAServer 在运行,这时本 RCAServer 将顺利运行。
  662. // 这是通过一个 Semaphore 对象对正在运行的任何版本的 RCAServer 进行控制实现的
  663. // 2. 首先说明一点,Semaphore 对象将保证当前系统绝不会有三个或三个以上的无论什么版本的RCAServer 同时运行
  664. // 若当前系统有一个 RCAServer 在运行, CanGoon 将试图查找正运行的 RCAServer 的主窗口, 找不到将退出
  665. // 找到后会给窗口 Post 一个 WM_VERSIONUPDATE 消息, 以校验两个 RCAServer 的版本到底谁年轻谁 TMD 老该死
  666. //
  667. // 几种情况:
  668. // 更早运行的 RCAServer 版本低, 并且没有活动的连接,更早运行的 RCAServer 版本将退出
  669. // 新版本的将继续初始化并替代老版本的执行 RCP 服务
  670. //    
  671. // 更早运行的 RCAServer 版本低,但有活动的连接,新版本的将等待一段时间(不长,几秒钟),若这
  672. // 若这段时间内老版本的所有连接已关闭, 老的将退出,由新的替代运行,若这段时间连接仍未关闭,
  673. // 新的高版本的 RCAServer 将退出
  674. //
  675. // 更早运行的 RCAServer 版本高, 简单, 新的低版本 RCAServer 直接退出.
  676. //
  677. // 返回值: TRUE --- 此 RCAServer 实例将可以继续运行
  678. // FALSE --- 此 RCAServer 实例必须退出
  679. //
  680. BOOL WINAPI CanGoon( DWORD * result )
  681. {
  682. // 
  683. BOOL bAlreadyRun = FALSE;
  684. DWORD dwWaitResult;
  685. g_hRunningControlSemaphore = 
  686. CreateSemaphore( NULL, 2, 2, RCA_RUNNING_CONTROL_SEMAPHORE_NAME );
  687. if( !g_hRunningControlSemaphore )
  688. return FALSE;
  689. if( GetLastError() == ERROR_ALREADY_EXISTS ) // 信标对象已经存在, 将试图找 
  690. { // RCA SERVER 主窗口, 并进行版本更新
  691. dwWaitResult = WaitForSingleObject( g_hRunningControlSemaphore, 500 );
  692. if( dwWaitResult == WAIT_TIMEOUT )
  693. {
  694. SAFE_CLOSE_HANDLE( g_hRunningControlSemaphore );
  695. return FALSE;
  696. }
  697. bAlreadyRun = TRUE;
  698. HWND oldRCAMainWnd = FindWindow( SERVER_WINDOW_CLASSNAME, 
  699.  SERVER_WINDOW_CAPTIONNAME);
  700. if( !oldRCAMainWnd ) // 如果找不到,先休眠 2 秒, 再次试图查找窗口
  701. Sleep(2000);
  702. oldRCAMainWnd = FindWindow( SERVER_WINDOW_CLASSNAME, 
  703. SERVER_WINDOW_CAPTIONNAME );
  704. if( !oldRCAMainWnd ) // 如果还找不到,返回 FALSE,必须退出,信标对象无法准确定位
  705. return FALSE;
  706. PostMessage( oldRCAMainWnd, WM_VERSIONUPDATE, 
  707.  CURRENT_VERSION, GetCurrentThreadId() );
  708. MSG msg;
  709. int count = 4; // 循环 4 次
  710. while( count ) // 等待正在工作的 RCA 回递版本更新的消息
  711. {
  712. Sleep( 500 );  // 
  713. if( PeekMessage( &msg, NULL, WM_VERSIONUPDATE, WM_VERSIONUPDATE, PM_REMOVE ) )
  714. {
  715. DWORD iCmp = msg.wParam; // wParam 为版本比较结果信息
  716. WORD oldVersion = LOWORD(msg.lParam); // lParam 的低字为正在工作的 RCA 版本
  717. // 正在工作的RCA 是否可以退出, 假如正在工作的 RCA 版本低,
  718. // 但有连接在工作中,则为 FALSE,即无法退出
  719. BOOL bOldRCARunning = (BOOL)HIWORD(msg.lParam); 
  720. if( iCmp == VERSION_LOW )  // 本版本低, 必须退出
  721. {
  722. ReleaseSemaphore( g_hRunningControlSemaphore , 1, NULL );
  723. SAFE_CLOSE_HANDLE( g_hRunningControlSemaphore );
  724. return FALSE;
  725. } else if( iCmp == VERSION_EQUAL ) // 正在工作的版本与本版本相同,也退出
  726. {
  727. ReleaseSemaphore( g_hRunningControlSemaphore, 1, NULL );
  728. SAFE_CLOSE_HANDLE( g_hRunningControlSemaphore );
  729. return FALSE;
  730. } else if( iCmp == VERSION_HIGH ) // 本版本高,可以开始更新
  731. {
  732. BOOL bOldRCAExit = FALSE;
  733. count = 4;
  734. while( count )
  735. {
  736. Sleep( 500 ); 
  737. if( PeekMessage( &msg, NULL, WM_OLDVERSIONEXIT, WM_OLDVERSIONEXIT, PM_REMOVE ) )
  738. {
  739. bOldRCAExit = TRUE;
  740. break;
  741. } else 
  742. count-- ;
  743. }
  744. if( bOldRCAExit == FALSE ) // 虽然正在工作的版本低,但无法退出(有连接等)
  745. {
  746. ReleaseSemaphore( g_hRunningControlSemaphore, 1, NULL );
  747. SAFE_CLOSE_HANDLE( g_hRunningControlSemaphore );
  748. return FALSE;
  749. }
  750. break; // 跳出版本更新的消息通信循环
  751. }
  752. } else
  753. count --;
  754. }
  755. }
  756. WNDCLASSEX wcex;
  757. wcex.cbClsExtra = 0;
  758. wcex.cbSize = sizeof( wcex );
  759. wcex.cbWndExtra = 0;
  760. wcex.hbrBackground = (HBRUSH)(COLOR_WINDOW );
  761. wcex.hCursor = LoadCursor( NULL, IDC_ARROW );
  762. wcex.hIcon = g_hBig;
  763. wcex.hIconSm = g_hSmall;
  764. wcex.hInstance = g_hInstance;
  765. wcex.lpfnWndProc = MainWndProc;
  766. wcex.lpszClassName = SERVER_WINDOW_CLASSNAME;
  767. wcex.lpszMenuName = NULL;
  768. wcex.style = CS_HREDRAW | CS_VREDRAW;
  769. if( !RegisterClassEx( &wcex ) )
  770. {
  771. RCAUninit();
  772. return -1;
  773. }
  774. g_hWnd = CreateWindow( SERVER_WINDOW_CLASSNAME, SERVER_WINDOW_CAPTIONNAME,
  775. WS_OVERLAPPEDWINDOW, CW_USEDEFAULT, 
  776. CW_USEDEFAULT, 400, 300,
  777. NULL, NULL, g_hInstance, NULL );
  778. if( g_hWnd == NULL )
  779. {
  780. RCAUninit();
  781. return -1;
  782. }
  783. ShowWindow( g_hWnd, SW_HIDE);
  784. UpdateWindow( g_hWnd );
  785. return TRUE;
  786. }
  787.  /* // 接收请求头
  788. while ( nLeft > 0 )
  789. {
  790. ret = recv( sockRemote, &(headBuffer[i]), nLeft, 0 );
  791. if( ret == SOCKET_ERROR )
  792. {
  793. if( WSAGetLastError() != WSAEWOULDBLOCK )
  794. {
  795. bRecv = FALSE;
  796. break;
  797. }
  798. else 
  799. {
  800. nLeft -= ret;
  801. i += ret;
  802. continue;
  803. }
  804. DWORD dw = WaitForSingleObject( hEvent, INFINITE );
  805. WSANETWORKEVENTS netEvents;
  806. ret = WSAEnumNetworkEvents( sockRemote, hEvent, &netEvents );
  807. if( ret != 0 )
  808. {
  809. DEBUGOUTPUT1( WSAERRORSTRING );
  810. bRecv = FALSE;
  811. break;
  812. }
  813. if( netEvents.lNetworkEvents & FD_READ )
  814. {
  815. if( netEvents.iErrorCode[FD_READ_BIT] != 0 )
  816. {
  817. DEBUGOUTPUT1( WSAERRORSTRING );
  818. bRecv = FALSE;
  819. break;
  820. } else
  821. continue;
  822. }
  823. if( netEvents.lNetworkEvents & FD_CLOSE )
  824. {
  825. bRecv = FALSE;
  826. break;
  827. }
  828. } // recv while
  829. */