开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > Internet/网络编程 > ICQ/即时通讯 > 查看源码
ServerService.cpp
资源名称：QQ2004_Source_0.01.rar　[点击查看]
上传用户：hysujiao87
上传日期：2007-12-02
资源大小：156k
文件大小：13k
源码类别：

ICQ/即时通讯

开发平台：

C/C++

ServerService.cpp：源码内容
  1. #include "stdafx.h"
  2. #include "serverservice.h"
  4. #include "MainFrm.h"
  5. #include "xmlparser.h"
  6. #include "QQSocket.h"
  8. CServerService::CServerService(void)
  9. {
  10. _listenThread = NULL;
  11. _list = NULL;
  12. }
  14. CServerService::~CServerService(void)
  15. {
  16. }
  18. HRESULT CServerService::dealLogOnRequest(CXMLParser *request, CElement *responseRoot, CUserList &list)
  19. {
  20. DEAL_REQUEST_BEGIN()
  22. CUserData *userData = list.getUserByID(requestRoot->getChildContent(_T("userID")));
  23. if (userData != NULL)
  24. {
  25. if (requestRoot->getChildContent(_T("password")) != userData->_password)
  26. userData = NULL;
  27. }
  28. if (userData == NULL)
  29. {
  30. responseRoot->AddChild(_T("message"), _T("用户名或密码错误."));
  31. responseRoot->AddChild(_T("result"), _T("1"));
  32. return S_OK;
  33. }
  35. responseRoot->AddChild(_T("result"), _T("0"));
  36. responseRoot->AddChild(_T("nickname"), userData->_nickname);
  38. return S_OK;
  39. }
  41. HRESULT CServerService::dealLogOffRequest(CXMLParser *request, CElement *responseRoot, CUserList &list)
  42. {
  43. DEAL_REQUEST_BEGIN()
  45. CUserData *userData = list.getUserByID(requestRoot->getChildContent(_T("userID")));
  46. if (userData != NULL)
  47. {
  48. userData->_online = FALSE;
  49. userData->_lastReport = 0;
  50. ::ZeroMemory(&userData->_sockAddr, sizeof(sockaddr_in));
  51. }
  53. return S_OK;
  54. }
  56. HRESULT CServerService::dealRegisterRequest(CXMLParser *request, CElement *responseRoot, CUserList &list)
  57. {
  58. DEAL_REQUEST_BEGIN()
  60. CUserData *userData = list.getUserByID(requestRoot->getChildContent(_T("userID")));
  61. if (userData != NULL)
  62. {
  63. responseRoot->AddChild(_T("result"), _T("1"));
  64. responseRoot->AddChild(_T("message"), _T("这个ID已经有人使用了."));
  65. return S_OK;
  66. }
  68. CUserData newUser;
  69. newUser._clientID = requestRoot->getChildContent(_T("userID"));
  70. newUser._password = requestRoot->getChildContent(_T("password"));
  71. newUser._nickname = requestRoot->getChildContent(_T("nickname"));
  73. list.addUser(&newUser);
  74. responseRoot->AddChild(_T("result"), _T("0"));
  75. return S_OK;
  76. }
  78. HRESULT CServerService::dealQueryUserRequest(CXMLParser *request, CElement *responseRoot, CUserList &list)
  79. {
  80. DEAL_REQUEST_BEGIN()
  82. CString userID = requestRoot->getChildContent(_T("userID"));
  83. CUserData *userData = list.getUserByID(userID);
  84. if (userData == NULL)
  85. {
  86. responseRoot->AddChild(_T("result"), _T("1"));
  87. }
  88. else
  89. {
  90. responseRoot->AddChild(_T("result"), _T("0"));
  91. responseRoot->AddChild(_T("userID"), userData->_clientID);
  92. responseRoot->AddChild(_T("nickname"), userData->_nickname);
  93. if (userData->_online)
  94. {
  95. responseRoot->AddChild(_T("online"), _T("true"));
  96. responseRoot->AddChild(_T("ipAddr"), CQQSocket::IPAddrToString(userData->_sockAddr));
  97. CString port;
  98. port.Format("%d", userData->_sockAddr.sin_port);
  99. responseRoot->AddChild(_T("port"), port);
  101. }
  102. else
  103. responseRoot->AddChild(_T("online"), _T("false"));
  104. }
  106. return S_OK;
  107. }
  109. HRESULT CServerService::dealAddFriendRequest(CXMLParser *request, CElement *responseRoot, CUserList &list)
  110. {
  111. DEAL_REQUEST_BEGIN()
  113. CUserData *friendData = list.getUserByID(requestRoot->getChildContent(_T("friendID")));
  114. if (friendData != NULL)
  115. {
  116. CUserData *userData = list.getUserByID(requestRoot->getChildContent(_T("userID")));
  117. HRESULT ret = userData->addFriend(friendData->_clientID);
  118. if (ret == S_OK)
  119. {
  120. responseRoot->AddChild(_T("result"), _T("0"));
  121. return S_OK;
  122. }
  124. if (ret == S_FALSE)
  125. {
  126. responseRoot->AddChild(_T("result"), _T("1"));
  127. responseRoot->AddChild(_T("message"), _T("该用户已经在好友列表中."));
  128. return S_OK;
  129. }
  131. responseRoot->AddChild(_T("result"), _T("500"));
  132. responseRoot->AddChild(_T("message"), _T("服务器内部错误."));
  133. return S_OK;
  134. }
  136. responseRoot->AddChild(_T("result"), _T("2"));
  137. responseRoot->AddChild(_T("message"), _T("该用户不存在."));
  138. return S_OK;
  139. }
  141. HRESULT CServerService::dealDownloadFriendsRequest(CXMLParser *request, CElement *responseRoot, CUserList &list)
  142. {
  143. DEAL_REQUEST_BEGIN()
  145. CUserData *userData = list.getUserByID(requestRoot->getChildContent(_T("userID")));
  146. if (userData != NULL)
  147. {
  148. responseRoot->AddChild(_T("result"), _T("0"));
  149. for (int i = 0; i < userData->getFriendCount(); i++)
  150. {
  151. responseRoot->AddChild(_T("friend"), userData->getFriendID(i));
  152. }
  153. }
  154. else
  155. {
  156. responseRoot->AddChild(_T("result"), _T("1"));
  157. responseRoot->AddChild(_T("message"), _T("要查询好友的用户不存在."));
  158. }
  160. return S_OK;
  161. }
  163. HRESULT CServerService::dealOnlineRequest(CXMLParser *request, CElement *responseRoot, const sockaddr_in& addrFrom, CUserList &list)
  164. {
  165. DEAL_REQUEST_BEGIN()
  167. CUserData *friendData = list.getUserByID(requestRoot->getChildContent(_T("userID")));
  168. if (friendData != NULL)
  169. {
  170. CString csPort = requestRoot->getChildContent(_T("port"));
  171. short port = _ttoi(csPort);
  173. memcpy(&friendData->_sockAddr, &addrFrom, sizeof(sockaddr_in));
  174. friendData->_sockAddr.sin_port = port;
  176. friendData->_lastReport = ::GetTickCount();
  177. friendData->_online = TRUE;
  178. }
  180. return S_OK;
  181. }
  183. HRESULT CServerService::dealRequest(BYTE *data, 
  184. int dataLen, 
  185. const SOCKET &sock, 
  186. const sockaddr_in &addrFrom, 
  187. CUserList* list)
  188. {
  189. _ASSERTE(data != NULL);
  190. if (data == NULL)
  191. return E_INVALIDARG;
  193. CXMLParser request;
  194. if (request.LoadXML(data, dataLen) != 0)
  195. return E_FAIL;
  197. CElement *requestRoot = request.get_root();
  198. _ASSERTE(requestRoot->get_tag() == _T("REQUEST"));
  200. CElement *type = requestRoot->GetFirstChildByTag(_T("type"));
  201. _ASSERTE(type != NULL);
  202. if (type == NULL)
  203. return E_FAIL;
  205. if (type->get_content() != _T("ONLINE"))
  206. {
  207. CString log = logMessage(&request, addrFrom);
  208. TCHAR *msg = new TCHAR[log.GetLength() + 1];
  209. _tcscpy(msg, log);
  210. AfxGetMainWnd()->PostMessage(WM_SERVICE_NOTIFY, (WPARAM)msg, 0);
  211. }
  213. CXMLParser response;
  214. CElement *responseRoot = response.createElement(_T("RESPONSE"), _T(""));
  216. // Add the new request deal code here.
  217. if (type->get_content() == _T("LOGON"))
  218. {
  219. HRESULT ret = dealLogOnRequest(&request, responseRoot, *list);
  220. if (ret != S_OK)
  221. {
  222. delete responseRoot;
  223. return ret;
  224. }
  225. }
  227. if (type->get_content() == _T("REGISTER"))
  228. {
  229. HRESULT ret = dealRegisterRequest(&request, responseRoot, *list);
  230. if (ret != S_OK)
  231. {
  232. delete responseRoot;
  233. return ret;
  234. }
  235. }
  237. if (type->get_content() == _T("ADDFRIEND"))
  238. {
  239. HRESULT ret = dealAddFriendRequest(&request, responseRoot, *list);
  240. if (ret != S_OK)
  241. {
  242. delete responseRoot;
  243. return ret;
  244. }
  245. }
  247. if (type->get_content() == _T("DOWNLOADFRIENDS"))
  248. {
  249. HRESULT ret = dealDownloadFriendsRequest(&request, responseRoot, *list);
  250. if (ret != S_OK)
  251. {
  252. delete responseRoot;
  253. return ret;
  254. }
  255. }
  257. if (type->get_content() == _T("QUERYUSER"))
  258. {
  259. HRESULT ret = dealQueryUserRequest(&request, responseRoot, *list);
  260. if (ret != S_OK)
  261. {
  262. delete responseRoot;
  263. return ret;
  264. }
  265. }
  267. if (type->get_content() == _T("ONLINE"))
  268. {
  269. dealOnlineRequest(&request, responseRoot, addrFrom, *list);
  270. delete responseRoot;
  271. return S_OK;
  272. }
  274. if (type->get_content() == _T("LOGOFF"))
  275. {
  276. dealLogOffRequest(&request, responseRoot, *list);
  277. delete responseRoot;
  278. return S_OK;
  279. }
  280. // End of adding request deal code.
  282. response.set_root(responseRoot);
  283. if (response.BuildXML() != 0)
  284. {
  285. TRACE("ERROR: Failed to build the response xml.n");
  286. return E_FAIL;
  287. }
  288. CString xml = response.get_xml();
  290. int ret = send(sock, ((LPCTSTR)xml), xml.GetLength() * sizeof(TCHAR), 0);
  291. if (ret == SOCKET_ERROR)
  292. return E_FAIL;
  294. return S_OK;
  295. }
  297. UINT CServerService::serviceProc(LPVOID wParam)
  298. {
  299. auto_ptr<LISTEN_PARAM> param((LISTEN_PARAM*)wParam);
  301. SOCKET sock = socket(AF_INET, SOCK_STREAM, 0);
  302. sockaddr_in addrListen;
  303. addrListen.sin_family = AF_INET;
  304. addrListen.sin_addr.S_un.S_addr = INADDR_ANY;
  305. addrListen.sin_port = htons(6000);
  306. int ret = bind(sock, (SOCKADDR*)&addrListen, sizeof(sockaddr_in));
  307. if (ret != 0)
  308. {
  309. ::closesocket(sock);
  311. int error = ::WSAGetLastError();
  312. TRACE("ERROR: Failed to bind the server listen socket (%d).n", error);
  313. return -1;
  314. }
  316. ret = listen(sock, 5);
  317. if (ret != 0)
  318. {
  319. ::closesocket(sock);
  321. int error = ::WSAGetLastError();
  322. TRACE("ERROR: Failed to listen the server listen socket (%d).n", error);
  323. return -1;
  324. }
  326. u_long arg = TRUE;
  327. ret = ioctlsocket(sock, FIONBIO, &arg);
  328. if (ret != 0)
  329. {
  330. ::closesocket(sock);
  332. int error = ::WSAGetLastError();
  333. TRACE("ERROR: Failed to ioctlsocket the socket to non-blocking mode (%d).n", error);
  334. return -1;
  335. }
  337. while(TRUE)
  338. {
  339. while(TRUE)
  340. {
  341. if (::WaitForSingleObject(*(param->finishEvent), 0) == WAIT_OBJECT_0)
  342. {
  343. ::closesocket(sock);
  345. TRACE("NOTIFY: The server's listening thread is finished.n");
  346. return 0;
  347. }
  349. fd_set readset;
  350. FD_ZERO(&readset);
  351. FD_SET(sock, &readset);
  352. TIMEVAL time;
  353. time.tv_sec = 0;
  354. time.tv_usec = 1;
  356. int ret = select(0, &readset, NULL, NULL, &time);
  357. if (ret == SOCKET_ERROR)
  358. {
  359. ::closesocket(sock);
  361. int error = ::WSAGetLastError();
  362. TRACE("ERROR: Failed to select the socket states (%d).n", error);
  363. return -1;
  364. }
  366. if (ret != 0)
  367. break ;
  368. }
  370. sockaddr_in addrFrom;
  371. ::ZeroMemory(&addrFrom, sizeof(sockaddr_in));
  372. int addrLen = sizeof(sockaddr_in);
  373. SOCKET acceptedSock = accept(sock, (SOCKADDR*)&addrFrom, &addrLen);
  374. _ASSERTE(acceptedSock != INVALID_SOCKET);
  375. if (acceptedSock == INVALID_SOCKET)
  376. {
  377. ::closesocket(sock);
  379. int error = ::WSAGetLastError();
  380. TRACE("ERROR: Failed to accept a connect (%d).n", error);
  381. return -1;
  382. }
  384. u_long arg = FALSE;
  385. ret = ioctlsocket(acceptedSock, FIONBIO, &arg);
  386. if (ret != 0)
  387. {
  388. ::closesocket(acceptedSock);
  389. ::closesocket(sock);
  391. int error = ::WSAGetLastError();
  392. TRACE("ERROR: Failed to ioctlsocket the socket to non-blocking mode (%d).n", error);
  393. return -1;
  394. }
  396. while(TRUE)
  397. {
  398. if (::WaitForSingleObject(*(param->finishEvent), 0) == WAIT_OBJECT_0)
  399. {
  400. ::closesocket(acceptedSock);
  401. ::closesocket(sock);
  403. TRACE("NOTIFY: The server's listening thread is finished.n");
  404. return 0;
  405. }
  407. DWORD dataLen = 0;
  408. ret = ::ioctlsocket(acceptedSock, FIONREAD, &dataLen);
  409. if (ret != 0)
  410. {
  411. ::closesocket(acceptedSock);
  412. ::closesocket(sock);
  414. int error = ::WSAGetLastError();
  415. TRACE("ERROR: Failed to ioctlsocket() on the accepted socket (%d).n", error);
  416. return -1;
  417. }
  418. if (dataLen <= 0)
  419. {
  420. Sleep(50);
  421. continue ;
  422. }
  424. auto_ptr<BYTE> data(new BYTE[dataLen]);
  425. dataLen = recv(acceptedSock, (char*)(data.get()), dataLen, 0);
  426. if (dataLen == SOCKET_ERROR)
  427. {
  428. ::closesocket(acceptedSock);
  429. ::closesocket(sock);
  431. int error = ::WSAGetLastError();
  432. TRACE("ERROR: Failed to receive the request (%d).n", error);
  433. return -1;
  434. }
  436.             dealRequest(data.get(), dataLen, acceptedSock, addrFrom, param->list);
  437. break ;
  438. }
  439. closesocket(acceptedSock);
  440. }
  442. return -1;
  443. }
  445. HRESULT CServerService::startListen(CUserList *list)
  446. {
  447. _ASSERTE(_listenThread == NULL);
  448. if (_listenThread != NULL)
  449. return E_UNEXPECTED;
  451. _list = list;
  453. LISTEN_PARAM *param = new LISTEN_PARAM;
  454. param->finishEvent = &_finishEvent;
  455. param->list = list;
  457. CWinThread *pThread = AfxBeginThread(CServerService::serviceProc, (LPVOID)param, 0, 0, CREATE_SUSPENDED);
  458. _ASSERTE(pThread != NULL);
  459. if (pThread == NULL)
  460. return E_FAIL;
  462. HANDLE hProcess = ::GetCurrentProcess();
  463. BOOL ret = ::DuplicateHandle(hProcess, 
  464. pThread->m_hThread, 
  465. hProcess, 
  466. &_listenThread, 
  467. 0, 
  468. TRUE, 
  469. DUPLICATE_SAME_ACCESS);
  470. _ASSERTE(ret != FALSE);
  471. if (ret == FALSE)
  472. return E_FAIL;
  474. pThread->ResumeThread();
  476. return S_OK;
  477. }
  479. HRESULT CServerService::stopListen()
  480. {
  481. if(_listenThread == NULL)
  482. return S_OK;
  484. _finishEvent.SetEvent();
  486. DWORD ret = ::WaitForSingleObject(_listenThread, 5000);
  487. _ASSERTE(ret == WAIT_OBJECT_0);
  488. if (ret != WAIT_OBJECT_0)
  489. return E_FAIL;
  491. ::CloseHandle(_listenThread);
  492. _listenThread = NULL;
  494. return S_OK;
  495. }
  497. CString CServerService::logMessage(CXMLParser *msg, const sockaddr_in &addrFrom)
  498. {
  499. CString notifyMsg = _T("Received a request from ");
  500. notifyMsg += CQQSocket::IPAddrToString(addrFrom);
  502. CElementList list;
  503. msg->get_root()->GetChildrenByTag(NULL, list);
  505. POSITION pos = list.GetHeadPosition();
  506. while(pos != NULL)
  507. {
  508. CElement *element = list.GetNext(pos);
  509. _ASSERTE(element != NULL);
  510. if (element == NULL)
  511. return _T("");
  513. CString csElement;
  514. csElement.Format(" %s=%s", element->get_tag(), element->get_content());
  515. notifyMsg += csElement;
  516. }
  518. return notifyMsg;
  519. }