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

源码开发语言/平台
当前位置：首页 > 源码/资料 > Internet/网络编程 > 网格计算 > 查看源码
Service.cpp
资源名称：fmailstable.rar　[点击查看]
上传用户：maryhy001
上传日期：2007-05-02
资源大小：2317k
文件大小：24k
源码类别：

网格计算

开发平台：

Visual C++

Service.cpp：源码内容
  1. #include "stdafx.h"
  2. #include "Server.h"
  3. #include "Service.h"
  4. #include "passwrd.h"
  6. safe_vector<PCLIENT_STATE> m_omiclientstates;
  8. bool ModifyClientState(string clienthdid, bool state)
  9. {
  10. if(!theApp.m_dbcont.IsActived()) return false;
  12. CADODataSet rs;
  13. rs.SetConnection(theApp.m_dbcont);
  15. char sql[MAX_SQL_SIZE];
  16. memset(sql, 0x0, sizeof(sql));
  17. sprintf(sql, SQL::SERVER_UPDATE_AGSTATE, (state ? 1 : 0), clienthdid.c_str());
  19. if( rs.Open(sql) )
  20. {
  21. for(int i = 0; i < m_omiclientstates.size(); ++i)
  22. {
  23. PCLIENT_STATE pitem = m_omiclientstates[i];
  24. if(NULL == pitem)
  25. {
  26. m_omiclientstates.pop_at(i--);
  27. continue;
  28. }
  29. if(!pitem->clienthdid.compare(clienthdid.c_str()))
  30. {
  31. pitem->interval = 0;
  32. break;
  33. }
  34. }
  35. return true;
  36. }
  37. return false;
  38. }
  40. //update the client state list.
  41. void UpdateClientState(void)
  42. {
  43. if(!theApp.m_dbcont.IsActived())
  44. {
  45. //TRACE0("The db connection is not active yetn");
  46. return ;
  47. }
  49. try
  50. {
  51. for(int i = 0; i < m_omiclientstates.size(); ++i)
  52. {
  53. PCLIENT_STATE pitem = m_omiclientstates[i];
  54. if(NULL == pitem)
  55. {
  56. m_omiclientstates.pop_at(i--);
  57. continue;
  58. }
  59. pitem->interval++;
  61. //update the client database state flag.
  62. if(pitem->interval > 300)
  63. {
  64. theApp.m_syslog.Print(LL_INFO, "UpdateClientState : 侦测到 [%s] 客户端连接状态改变...n", pitem->clienthdid.c_str());
  66. //restore to zero.
  67. pitem->interval = 0;
  68. ModifyClientState(pitem->clienthdid, false);
  69. }
  70. }
  71. }
  72. catch (...)
  73. {
  74. return ;
  75. }
  76. }
  78. //为以后扩充用---部门根据IP地址来划分
  79. int GetClientDepart(string &clienthdid)
  80. {
  81. return -1;
  82. }
  84. /********************************************************
  85.  *  Class Name : CService. *
  86.  * Purpose : Service objects manager class. *
  87.  *  File  Name : Service.h / Service.cpp *
  88.  *------------------------------------------------------*
  89.  * Author : Devia Lee. Date: 2004-06-01 *
  90.  ********************************************************/
  91. CService::CService()
  92. {
  93. m_bopened  = false;
  94. m_omiservice = NULL;
  95. }
  97. CService::~CService()
  98. {
  99. this->Stop();
  100. }
  102. bool CService::IsOpened(void)
  103. {
  104. return m_bopened;
  105. }
  107. bool CService::Start(const int serverport)
  108. {
  109. theApp.m_syslog.Print(LL_INFO, "CService::Start : 开始创建服务对象...n");
  111. if(NULL == this->m_omiservice)
  112. {
  113. this->m_omiservice = new CServiceThread;
  114. if(!this->m_omiservice->Create(this, false, serverport))
  115. {
  116. theApp.m_syslog.Print(LL_ERROR, "CService::Start : 创建服务对象主线程失败!!!n");
  118. this->m_bopened = false;
  119. return false;
  120. }
  121. theApp.m_syslog.Print(LL_INFO, "CService::Start : 创建服务对象主线程成功!!!n");
  122. }
  124. theApp.m_syslog.Print(LL_INFO, "CService::Start : 创建服务对象成功!!!n");
  126. this->m_bopened = true;
  127. return true;
  128. }
  130. void CService::Stop()
  131. {
  132. theApp.m_syslog.Print(LL_INFO, "CService::Stop : 开始销毁服务对象...n");
  134. if(NULL != this->m_omiservice)
  135. {
  136. this->m_omiservice->Stop();
  137. _DELETE(this->m_omiservice);
  139. theApp.m_syslog.Print(LL_INFO, "CService::Stop : 服务对象主线程已销毁!!!n");
  140. }
  141. theApp.m_syslog.Print(LL_INFO, "CService::Stop : 销毁服务对象成功!!!n");
  143. this->m_bopened = false;
  144. }
  146. /********************************************************
  147.  *  Class Name : CServiceThread. *
  148.  * Purpose : Main service thread class. *
  149.  *  File  Name : Service.h / Service.cpp *
  150.  *------------------------------------------------------*
  151.  * Author : Devia Lee. Date: 2004-06-01 *
  152.  ********************************************************/
  153. CServiceThread::CServiceThread()
  154. {
  155. this->m_service = NULL;
  156. this->m_svrsock = NULL;
  157. }
  159. CServiceThread::~CServiceThread()
  160. {
  161. this->OnStop();
  162. }
  164. //create the acceptor socket listening thread.
  165. bool CServiceThread::Create(CService *pService, bool bCreateAndSuspended, const uint unSvrPort)
  166. {
  167. assert( NULL != pService && NULL == this->m_svrsock );
  169. theApp.m_syslog.Print(LL_INFO, "CServiceThread::Create : 开始创建服务主线程...n");
  171. this->m_service = pService;
  173. //create the listening server socket object.
  174. this->m_svrsock = new CServiceSocket();
  175. this->m_svrsock->Set_Socket_Address(unSvrPort);
  176. try{
  177. if ( this->m_svrsock->Create() )
  178. {
  179. if(this->m_svrsock->Open())
  181. theApp.m_syslog.Print(LL_INFO, "CServiceThread::Create : 创建客户端Socket连接成功!!!n");
  182. else
  183. theApp.m_syslog.Print(LL_ERROR, "CServiceThread::Create : 创建客户端Socket连接失败...n");
  184. }
  185. }
  186. catch ( CNetSockException *ex ) {
  188. theApp.m_syslog.Print(LL_ERROR, "CServiceThread::Create : 创建客户端Socket连接发生异常, ErrorCode = %dn", ex->GetSockErrCode());
  190. _DELETE( this->m_svrsock );
  191. throw ex;
  192. return false;
  193. }
  194. return CThread::Create(bCreateAndSuspended, NULL);
  195. }
  197. //overrides execute function.
  198. void CServiceThread::Execute()
  199. {
  200. while(!this->CanBeTerminate())
  201. {
  202. UpdateClientState();
  204. Sleep(100);
  205. }
  206. }
  208. //start, stop, pause and resume event process functions.
  209. void CServiceThread::OnStart()
  210. {
  211. }
  212. void CServiceThread::OnStop()
  213. {
  214. //close the server socket
  215. if ( this->m_svrsock && this->m_svrsock->Open(false) )
  216. {
  217. _DELETE( this->m_svrsock );
  218. theApp.m_syslog.Print(LL_INFO, "CServiceThread::OnStop : 服务主线程退出!!!n");
  219. }
  220. }
  221. void CServiceThread::OnSuspend()
  222. {
  223. assert(NULL != this->m_svrsock);
  224. theApp.m_syslog.Print(LL_INFO, "CServiceThread::OnSuspend : 服务主线程挂起!!!n");
  225. }
  226. void CServiceThread::OnResume()
  227. {
  228. assert(NULL != this->m_svrsock);
  229. theApp.m_syslog.Print(LL_INFO, "CServiceThread::OnStop : 服务主线程恢复运行!!!n");
  230. }
  234. /********************************************************
  235.  *  Class Name : CServiceSocket. *
  236.  * Purpose : Service socket object class. *
  237.  *  File  Name : Service.h / Service.cpp *
  238.  *------------------------------------------------------*
  239.  * Author : Devia Lee. Date: 2004-06-01 *
  240.  ********************************************************/
  241. CServiceSocket::CServiceSocket()
  242. {
  243. }
  245. CServiceSocket::~CServiceSocket()
  246. {
  247. theApp.m_syslog.Print(LL_INFO, "CServiceSocket::~CServiceSocket : 开始销毁服务端Socket...n");
  249. //destroy the vector.
  250. for(int i = 0; i = (int)m_omicltthreads.size() > 0; )
  251. {
  252. CClientThread *pthread = m_omicltthreads[i-1];
  253. if(NULL == pthread)
  254. {
  255. m_omicltthreads.pop_at(i-1);
  256. continue;
  257. }
  259. if(pthread->GetSockObj()->Open(false))
  260. {
  261. theApp.m_syslog.Print(LL_INFO, "CServiceSocket::~CServiceSocket : 销毁 [%s] 客户端处理线程.n", 
  262. pthread->GetSockObj()->Get_RemoteHost_IP().c_str());
  264. pthread->Stop();
  265. _DELETE(pthread);
  266. m_omicltthreads.pop_back();
  267. }
  268. }
  270. //destroy the list.
  271. for(i = 0; i = m_omiclientstates.size() > 0; )
  272. {
  273. PCLIENT_STATE pitem = m_omiclientstates[i-1];
  274. if(NULL == pitem)
  275. {
  276. m_omiclientstates.pop_at(i-1);
  277. continue;
  278. }
  279. _DELETE(pitem);
  280. m_omiclientstates.pop_back();
  281. }
  282. theApp.m_syslog.Print(LL_INFO, "CServiceSocket::~CServiceSocket : 销毁服务端Socket结束!!!n");
  283. }
  285. //===============================================================
  286. //
  287. // Process the client's connected event and update clients list.
  288. //
  289. //===============================================================
  290. void CServiceSocket::OnClientConnect(CXWinSocket *pClientSocket)
  291. {
  292. CXWinServerSocket::OnClientConnect(pClientSocket);
  294. if(NULL == pClientSocket) return ;
  296. string sremoteip = pClientSocket->Get_RemoteHost_IP();
  297. theApp.m_syslog.Print(LL_INFO, "CServiceSocket::OnClientConnect : 侦听到远程客户端 [%s] 连接...n", sremoteip.c_str());
  299. if(!theApp.m_dbcont.IsActived())
  300. {
  301. theApp.m_syslog.Print(LL_ERROR, "CServiceSocket::OnClientConnect : 数据库连接异常,客户端 [%s] 连接被异常终止.n", sremoteip.c_str());
  303. pClientSocket->Open(false);
  304. return ;
  305. }
  307. //@1---检查该客户端是否已经存在于列表中,如果存在则直接关闭Socket返回.
  308. CClientThread *pthread = NULL;
  309. for(int i = 0; i < m_omicltthreads.size(); ++i)
  310. {
  311. pthread = m_omicltthreads[i];
  312. if(NULL == pthread)
  313. {
  314. m_omicltthreads.pop_at(i--);
  315. continue;
  316. }
  317. if( pthread->GetSockObj()->Get_RemoteHost_IP() == pClientSocket->Get_RemoteHost_IP() ||
  318. pthread->GetSockObj()->Get_RemoteHost_Name() == pClientSocket->Get_RemoteHost_Name())
  319. {
  320. theApp.m_syslog.Print(LL_WARNING, "CServiceSocket::OnClientConnect : [%s] 客户端已在服务器注册，连接终止!!!n", sremoteip.c_str());
  322. pClientSocket->Open(false);
  323. return ;
  324. }
  325. }
  326. //@3---为客户端创建处理线程
  328. theApp.m_syslog.Print(LL_INFO, "CServiceSocket::OnClientConnect : 准备为 [%s] 客户端创建处理线程...n", sremoteip.c_str());
  330. string remotename = pClientSocket->Get_RemoteHost_Name(), remoteip = pClientSocket->Get_RemoteHost_IP();
  332. pthread = new CClientThread(pClientSocket);
  334. if(pthread->Create()){
  336. m_omicltthreads.push_back(pthread);
  337. theApp.m_syslog.Print(LL_ERROR, "CServiceSocket::OnClientConnect : 为 [%s] 客户端创建处理线程成功!!!n", sremoteip.c_str());
  338. }
  339. else{
  341. //如果创建处理线程失败则恢复数据库状态标志
  342. ModifyClientState(remoteip, false);
  344. _DELETE(pthread);
  345. pClientSocket->Open(false);
  347. theApp.m_syslog.Print(LL_ERROR, "CServiceSocket::OnClientConnect : 为 [%s] 客户端创建处理线程失败!!!n", sremoteip.c_str());
  348. }
  349. }
  351. //=============================================================================
  352. //
  353. // Process the client's disconnected event and update clients list.
  354. //
  355. //=============================================================================
  356. void CServiceSocket::OnClientClose(CXWinSocket *pClientSocket)
  357. {
  358. if(NULL == pClientSocket) return ;
  360. try
  361. {
  362. string sremoteip;
  364. CClientThread *pthread = NULL;
  365. for(int i = 0; i < m_omicltthreads.size(); ++i)
  366. {
  367. pthread = m_omicltthreads[i];
  368. if(NULL == pthread)
  369. {
  370. m_omicltthreads.pop_at(i--);
  371. continue;
  372. }
  373. if(pthread->GetSockObj()->Get_Socket_Handle() == pClientSocket->Get_Socket_Handle())
  374. {
  375. //@1
  376. sremoteip = pthread->m_sremoteip;
  378. for(int j = m_omiclientstates.size(); j > 0; --j)
  379. {
  380. PCLIENT_STATE pitem = m_omiclientstates[j-1];
  381. if(NULL == pitem)
  382. {
  383. m_omiclientstates.pop_at(j-1);
  384. continue;
  385. }
  386. if(0 == pitem->clienthdid.compare(pthread->m_sclienthdid.c_str()))
  387. {
  388. if(!ModifyClientState(pitem->clienthdid, false))
  389. {
  390. theApp.m_syslog.Print(LL_ERROR, "CServiceSocket::OnClientClose : 修改 [%s] 客户端状态失败.n", pitem->clienthdid.c_str());
  391. return ;
  392. }
  393. _DELETE(pitem);
  395. m_omiclientstates.pop_at(j-1);
  397. break;
  398. }
  399. }
  400. //@2
  401. pthread->Stop();
  402. m_omicltthreads.pop_at(i);
  403. _DELETE(pthread);
  405. break;
  406. }
  407. }
  408. theApp.m_syslog.Print(LL_INFO, "CServiceSocket::OnClientClose : 客户端 [%s] 已端开.n", sremoteip.c_str());
  410. CXWinServerSocket::OnClientClose(pClientSocket);
  411. }
  412. catch (...) {
  413. CXWinServerSocket::OnClientClose(pClientSocket);
  414. }
  415. }
  418. /********************************************************
  419.  *  Class Name : CClientThread. *
  420.  * Purpose : Client processing thread class. *
  421.  *  File  Name : Service.h / Service.cpp *
  422.  *------------------------------------------------------*
  423.  * Author : Devia Lee. Date: 2004-06-01 *
  424.  ********************************************************/
  425. CClientThread::CClientThread(CXWinSocket *pClientSocket)
  426. {
  427. assert(NULL != pClientSocket && pClientSocket->Socket_IsOpened());
  428. this->m_pcltsock = pClientSocket;
  429. this->m_bfirststate = true;
  430. }
  432. CClientThread::~CClientThread()
  433. {
  434. }
  436. //send answer to peer.
  437. bool CClientThread::SendAnswer(RAGPACKTYPE akp, bool bSuccessfully)
  438. {
  439. KAGENTPACK agkpack;
  440. int size = sizeof(agkpack);
  442. memset(&agkpack, 0x0, size);
  444. agkpack.type = KAGPACKTYPE(akp);
  445. agkpack.ret  = bSuccessfully ? K_OK : K_FAILED;
  447. //for(int i = 0; i < 3; ++i)
  448. {
  449. int sended = this->m_pcltsock->SendBuf(&agkpack, size);
  450. if(-1 == sended)
  451. {
  452. theApp.m_syslog.Print(LL_ERROR, "CClientThread::SendAnswer : 发送客户端应答 [%s] 发生Socket异常,客户端处理线程终止!!!n", this->m_sremoteip.c_str());
  454. this->Stop();
  455. return false;
  456. }
  457. else if(size == sended)
  458. {
  459. return true;
  460. }
  461. }
  462. return false;
  463. }
  465. //overrides execute function.
  466. void CClientThread::Execute()
  467. {
  468. RAGENTPACK agrpack;
  469. int size = 0;
  471. this->m_sremotehost = this->m_pcltsock->Get_RemoteHost_Name();
  472. this->m_sremoteip   = this->m_pcltsock->Get_RemoteHost_IP();
  474. theApp.m_syslog.Print(LL_INFO, "CClientThread::Execute : 客户端 [%s] 线程开始运行...n", this->m_sremoteip.c_str());
  476. while(!this->CanBeTerminate() && theApp.m_dbcont.IsActived() && 
  477. m_pcltsock->Socket_IsValid() && m_pcltsock->Socket_IsOpened())
  478. {
  479. size = sizeof(agrpack);
  480. memset(&agrpack, 0x0, size);
  482. //@1---recv the package head.
  483. if(size != m_pcltsock->ReceiveBuf(&agrpack, size))
  484. {
  485. Sleep(10);
  486. continue;
  487. }
  488. else if(agrpack.type != RAG_STATE && 
  489. agrpack.type != RAG_VERIFYPWD &&
  490. agrpack.type != RAG_EXECUSRSQL)
  491. {
  492. Sleep(10);
  493. continue;
  494. }
  496. //@2---recv the package extra data.
  497. if(agrpack.size > 0)
  498. {
  499. agrpack.data = new char[agrpack.size+1];
  500. memset(agrpack.data, 0x0, agrpack.size+1);
  502. if(agrpack.size != m_pcltsock->ReceiveBuf(agrpack.data, agrpack.size))
  503. {
  504. delete []agrpack.data;
  506. Sleep(10);
  507. continue;
  508. }
  509. }
  511. //@4---process the client's requirements.
  512. switch(agrpack.type){
  513. case RAG_STATE:
  514. {
  515. theApp.m_syslog.Print(LL_INFO, "CClientThread::Execute : 接收到客户端 [%s] 状态包...n", this->m_sremoteip.c_str());
  517. this->ProcessRAGState(agrpack);
  518. }
  519. break;
  521. case RAG_VERIFYPWD:
  522. {
  523. theApp.m_syslog.Print(LL_INFO, "CClientThread::Execute : 接收到客户端 [%s] 用户身份校验包...n", this->m_sremoteip.c_str());
  525. this->ProcessRAGAuthor(agrpack);
  526. }
  527. break;
  529. case RAG_EXECUSRSQL:
  530. {
  531. theApp.m_syslog.Print(LL_INFO, "CClientThread::Execute : 接收到客户端 [%s] SQL请求包...n", this->m_sremoteip.c_str());
  533. this->ProcessRAGUsrSql(agrpack);
  534. }
  535. break;
  536. }
  538. //free the memory space
  539. if(agrpack.size > 0)
  540. {
  541. delete []agrpack.data;
  542. }
  543. Sleep(10);
  544. }
  545. }
  547. //=============================================================================
  548. //
  549. // Process the client's timed state requirement and update the database info.
  550. //
  551. //=============================================================================
  552. void CClientThread::ProcessRAGState(RAGENTPACK &agrpack)
  553. {
  554. theApp.m_syslog.Print(LL_INFO, "CClientThread::ProcessRAGState : 开始处理客户端 [%s] 状态请求包...n", this->m_sremoteip.c_str());
  556. //////////////////////////////////////////////////////////////////////////
  558. //get hostname and hardware id.
  560. char *p = (char*)agrpack.data;
  561. char *phdid = strchr(p, '|');
  562. if(!phdid) return ;
  564. char *phostname = phdid + 1;
  565. if(!phostname) return ;
  567. *phdid = ''; phdid = p;
  569. //////////////////////////////////////////////////////////////////////////
  571. string  remotename = phostname,
  572. remoteip = this->m_sremoteip;
  574. this->m_sremotehost = (char*)phostname;
  575. bool occurerror = false, bsuccessfully = true;
  577. char sql[MAX_SQL_SIZE];
  578. char *pshdid = new char[strlen(phdid)+1];
  580. memset(pshdid, '', strlen(phdid)+1);
  581. _snprintf(pshdid, agrpack.size, "%s", phdid);
  583. m_sclienthdid = pshdid;
  585. sprintf(sql, SQL::SERVER_UPDATE_AGINFO, remotename.c_str(), remoteip.c_str(), 1, pshdid);
  587. CADODataSet rs;
  589. try
  590. {
  591. //@1---更新该客户端状态
  593. if(this->m_bfirststate)
  594. {
  595. PCLIENT_STATE pitem = new CLIENT_STATE;
  596. pitem->clienthdid = pshdid;
  597. pitem->interval = 0;
  599. m_omiclientstates.push_back(pitem);
  601. this->m_bfirststate = false;
  602. }
  603. ModifyClientState(pshdid, true);
  605. //@2---看该客户端是否已经在数据库中注册
  607. rs.SetConnection(theApp.m_dbcont);
  609. if(!rs.Open(sql))
  610. {
  611. occurerror = true; bsuccessfully = false;
  612. goto ACTION;
  613. }
  614. rs.Close();
  616. sprintf(sql, SQL::SERVER_GET_AGENTISREG, pshdid);
  617. if(!rs.Open(sql))
  618. {
  619. occurerror = true; bsuccessfully = false;
  620. goto ACTION;
  621. }
  622. //如果该客户端没有注册过则注册之...
  623. if(rs.Fields().FieldByName("CLTCOUNT").getValue().AsInteger() == 0)
  624. {
  625. rs.Close();
  626. sprintf(sql, SQL::SERVER_AGENT_REGISTER, remotename.c_str(), remoteip.c_str(), pshdid, GetClientDepart(remoteip), 1, remotename.c_str());
  627. if(!rs.Open(sql))
  628. {
  629. occurerror = true; bsuccessfully = false;
  630. goto ACTION;
  631. }
  632. }
  633. bsuccessfully = ModifyClientState(pshdid, true);
  635. ACTION:
  636. this->SendAnswer(agrpack.type, bsuccessfully);
  637. if(occurerror)
  638. {
  639. theApp.m_syslog.Print(LL_ERROR, "CClientThread::ProcessRAGState : 处理发生错误, [%s] 线程自动终止!!!n", this->m_sremoteip.c_str());
  641. this->Stop();
  642. }
  643. theApp.m_syslog.Print(LL_INFO, "CClientThread::ProcessRAGState : 客户端 [%s] 状态请求处理完毕.n", this->m_sremoteip.c_str());
  645. if(pshdid) { delete []pshdid; pshdid = NULL; }
  646. }
  647. catch (CADOException *ex) {
  649. theApp.m_syslog.Print(LL_ERROR, "CClientThread::ProcessRAGState : 处理时发生ADO异常, [%s] 线程自动终止!!!n", this->m_sremoteip.c_str());
  651. if(pshdid) { delete []pshdid; pshdid = NULL; }
  653. this->SendAnswer(agrpack.type, false);
  655. this->Stop();
  657. delete ex;
  658. }
  659. }
  661. //============================================================
  662. //
  663. // Process the client's authentication requirement and 
  664. // return the result(OK or FAILED).
  665. //
  666. //============================================================
  667. void CClientThread::ProcessRAGAuthor(RAGENTPACK &agrpack)
  668. {
  669. string sremoteip = this->m_sremoteip;
  671. theApp.m_syslog.Print(LL_INFO, "CClientThread::ProcessRAGAuthor : 开始处理客户端 [%s] 用户身份校验请求包...n", sremoteip.c_str());
  673. bool bsuccessful = false;
  675. if(!this->m_pcltsock || !this->m_pcltsock->Socket_IsOpened()) return ;
  677. #define JUNKUSRNAME "@#$&^!fack@you@  ^R~"
  678. #define JUNKPASSWRD "@!@#Z$&^"
  680. //TRACE0((char*)agrpack.data);
  681. AUTHORINFO authorinfo;
  682. memset(&authorinfo, 0x0, sizeof(AUTHORINFO));
  683. memcpy(&authorinfo, agrpack.data, agrpack.size);
  685. //unencrypting the user name.
  686. vncDecryptBytes((unsigned char*)authorinfo.username, ENCRYPTKEY);
  688. //unencrypting the password
  689. CPasswrd::ToText passwrd_encry(authorinfo.passwrd);
  691. char username[MAX_USR_SIZE+1], passwrd[MAXPWLEN+1], sql[MAX_SQL_SIZE];
  692. memset(username, '', sizeof(username));
  693. memcpy(username, authorinfo.username, MAX_USR_SIZE);
  695. memset(passwrd, '', sizeof(passwrd));
  696. memcpy(passwrd, (const char*)passwrd_encry, MAXPWLEN);
  698. CADODataSet rs;
  700. try
  701. {
  702. rs.SetConnection(theApp.m_dbcont);
  704. sprintf(sql, SQL::SERVER_VERIFY_SYSUSR, username, passwrd);
  705. if(!rs.Open(sql))
  706. {
  707. memcpy(authorinfo.username, JUNKUSRNAME, MAX_USR_SIZE);
  708. memcpy(authorinfo.passwrd, JUNKPASSWRD, MAXPWLEN);
  709. goto ACTION;
  710. }
  711. //if not exist then send an junk info.
  712. if(0 == rs.Fields().FieldByName("SYSUSRCNT").getValue().AsInteger())
  713. {
  714. memcpy(authorinfo.username, JUNKUSRNAME, MAX_USR_SIZE);
  715. memcpy(authorinfo.passwrd, JUNKPASSWRD, MAXPWLEN);
  716. }
  717. else
  718. {
  719. vncEncryptBytes((unsigned char*)username, ENCRYPTKEY);
  720. memset(&authorinfo, 0x0, sizeof(authorinfo));
  722. CPasswrd::FromText encryor(passwrd);
  723. memcpy(authorinfo.username, username, MAX_USR_SIZE);
  724. memcpy(authorinfo.passwrd, (const char*)encryor, MAXPWLEN);
  726. bsuccessful = true;
  727. }
  728. rs.Close();
  730. theApp.m_syslog.Print(LL_INFO, "CClientThread::ProcessRAGAuthor : 客户端 [%s] 用户身份校验请求处理完毕.n", sremoteip.c_str());
  731. }
  732. catch (CADOException *ex) {
  734. theApp.m_syslog.Print(LL_ERROR, "CClientThread::ProcessRAGAuthor :处理时发生ADO异常!!!n");
  736. delete ex;
  737. }
  739. ACTION:
  740. if(this->SendAnswer(RAG_VERIFYPWD, bsuccessful) && bsuccessful)
  741. {
  742. if(-1 == this->m_pcltsock->SendBuf(&authorinfo, sizeof(AUTHORINFO)))
  743. {
  744. theApp.m_syslog.Print(LL_ERROR, "CClientThread::ProcessRAGAuthor : 处理时发生Socket异常, [%s] 自动退出!!!n", sremoteip.c_str());
  746. this->Stop();
  747. }
  748. }
  749. }
  751. //==========================================================================
  752. //
  753. // Process the client's SQL requirement and return the recordset file(XML).
  754. //
  755. //==========================================================================
  756. void CClientThread::ProcessRAGUsrSql(RAGENTPACK &agrpack)
  757. {
  758. if(!this->m_pcltsock || !this->m_pcltsock->Socket_IsOpened()) return ;
  760. string sremoteip = this->m_sremoteip;
  762. theApp.m_syslog.Print(LL_INFO, "CClientThread::ProcessRAGUsrSql : 开始处理客户端 [%s] SQL处理请求包...n", sremoteip.c_str());
  764. //@1
  765. if(!this->SendAnswer(agrpack.type)) return;
  767. EXECSQLINFO execsqlinfo;
  768. memcpy(&execsqlinfo, agrpack.data, sizeof(EXECSQLINFO));
  770. //@2---query the dataset and transfer to the client.
  771. CADODataSet rs;
  773. try
  774. {
  775. rs.SetConnection(theApp.m_dbcont);
  776. if(rs.Open((const char*)execsqlinfo.sql))
  777. {
  778. //if the recordset has exists some records then 
  779. //return the recordset xml file to client peer.
  780. if(execsqlinfo.needretval)
  781. {
  782. //@1---save record set into xml file.
  783. char szRSXMLFile[_MAX_PATH];
  784. memset(szRSXMLFile, 0x0, sizeof(szRSXMLFile));
  786. theApp._GetTempFileName("xml", szRSXMLFile);
  787. rs.SaveToXMLFile((char*)szRSXMLFile);
  788. rs.Close();
  790. //@2---send the xml file and delete the temporary xml file.
  791. if(!SendFile((char*)&szRSXMLFile[0]))
  792. {
  793. theApp.m_syslog.Print(LL_INFO, "CClientThread::ProcessRAGUsrSql : 客户端 [%s] SQL处理中文件发送失败!!!n", sremoteip.c_str());
  794. }
  795. else
  796. {
  797. theApp.m_syslog.Print(LL_INFO, "CClientThread::ProcessRAGUsrSql : 客户端 [%s] SQL处理中文件发送成功!!!n", sremoteip.c_str());
  798. }
  799. DeleteFile((LPCSTR)szRSXMLFile);
  800. }
  801. this->SendAnswer(RAG_EXECUSRSQL, true);
  803. theApp.m_syslog.Print(LL_INFO, "CClientThread::ProcessRAGUsrSql : 客户端 [%s] SQL处理请求处理完毕.n", sremoteip.c_str());
  804. }
  805. }
  806. catch (CADOException *ex) {
  808. theApp.m_syslog.Print(LL_ERROR, "CClientThread::ProcessRAGUsrSql : 处理时发生ADO异常!!!n");
  810. this->SendAnswer(RAG_EXECUSRSQL, false);
  812. delete ex;
  813. }
  814. }
  816. //==========================================================================
  817. //
  818. // Process the client's SQL requirement and return the recordset file(XML).
  819. // (sub procedure for sending the recordset file(XML).)
  820. //
  821. //==========================================================================
  822. bool CClientThread::SendFile(LPCSTR lpszFile)
  823. {
  824. if(!this->m_pcltsock || !this->m_pcltsock->Socket_IsOpened()) return false;
  826. //@1---open the file.
  827. HANDLE hFile = CreateFile(lpszFile, GENERIC_READ, FILE_SHARE_READ, 
  828. NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
  830. if(INVALID_HANDLE_VALUE == hFile)
  831. {
  832. //TRACE0("Failed to open the specialed filen");
  833. return false;
  834. }
  836. //@2--send the file size.
  837. DWORD sizehigh = 0, sizelower = 0;
  838. sizelower = GetFileSize(hFile, &sizehigh);
  840. char cRes = 'O';
  841. int size = 0, errtimes = 0;
  842. do{
  843. size = sizeof(DWORD);
  844. if(size != this->m_pcltsock->SendBuf(&sizelower, size)) ++errtimes;
  845. else{
  846. //@3--wait for client...
  847. size = sizeof(char);
  849. if(size != this->m_pcltsock->ReceiveBuf(&cRes, size)) ++errtimes;
  850. else break;
  851. }
  852. }
  853. while(errtimes < 3);
  854. //3.1
  855. if(errtimes == 3)
  856. {
  857. //TRACE0("Failed to send the file size infon");
  858. CloseHandle(hFile);
  859. return false;
  860. }
  862. //@4
  863. int   nsended = 0, maxsize = 0;
  864. DWORD dwleft  = sizelower, dwread = 0;
  865. char  buf[MAX_NET_PACKET];
  867. //need encrypting process???
  868. while(dwleft > 0)
  869. {
  870. memset((char*)buf, 0x0, sizeof(buf));
  872. maxsize = (dwleft > MAX_NET_PACKET) ? MAX_NET_PACKET : dwleft;
  873. ReadFile(hFile, buf, maxsize, &dwread, NULL);
  875. XorEncryptStream(buf, dwread);
  877. nsended = this->m_pcltsock->SendBuf(buf, dwread);
  878. if(nsended < 0)
  879. {
  880. CloseHandle(hFile);
  881. return false;
  882. }
  884. //wait for client...
  885. size = sizeof(char);
  886. if(size != this->m_pcltsock->ReceiveBuf(&cRes, size))
  887. {
  888. //TRACE0("Failed to send the file size infon");
  889. CloseHandle(hFile);
  890. return false;
  891. }
  892. if('O' == cRes){
  893. dwleft -= nsended;
  894. }
  895. else{
  896. CloseHandle(hFile);
  897. return false;
  898. }
  899. }
  900. CloseHandle(hFile);
  901. return true;
  902. }