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

源码开发语言/平台
当前位置：首页 > 源码/资料 > Internet/网络编程 > P2P编程 > 查看源码
CSClient.cpp
资源名称：p2p_vod.rar　[点击查看]
上传用户：liguizhu
上传日期：2015-11-01
资源大小：2422k
文件大小：18k
源码类别：

P2P编程

开发平台：

Visual C++

CSClient.cpp：源码内容
  1. /*
  2.  *  Openmysee
  3.  *
  4.  *  This program is free software; you can redistribute it and/or modify
  5.  *  it under the terms of the GNU General Public License as published by
  6.  *  the Free Software Foundation; either version 2 of the License, or
  7.  *  (at your option) any later version.
  8.  *
  9.  *  This program is distributed in the hope that it will be useful,
  10.  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  11.  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12.  *  GNU General Public License for more details.
  13.  *
  14.  *  You should have received a copy of the GNU General Public License
  15.  *  along with this program; if not, write to the Free Software
  16.  *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  17.  *
  18.  */
  19. #include "StdAfx.h"
  20. #include "CSClient.h"
  21. #include "Communicator.h"
  24. namespace NPLayer1 {
  26. CSClient::CSClient(Communicator* c) :comm(c) {
  27. memset(recvBuf, 0, MAX_UDP_LEN);
  28. maxFailCount = MAX_LOGIN_FAIL_COUNT;
  29. bStopLogin = false;
  30. }
  32. CSClient::~CSClient() {
  33. if(m_Socket != INVALID_SOCKET)
  34. closesocket(m_Socket);
  35. }
  37. P2P_RETURN_TYPE CSClient::Init() {
  38. // 初始化成员
  39. m_Socket = INVALID_SOCKET;
  40. isReqResSucceed = false;
  41. content = recvBuf;
  43. isLogin = false;
  44. loginFailCount = 0;
  46. // Create UDP Socket
  47. m_Socket = socket(AF_INET, SOCK_DGRAM, 0);
  48. if (m_Socket == INVALID_SOCKET) {
  49. comm->logFile.StatusErr("Creating UDP socket", WSAGetLastError());
  50. return PRT_NET;
  51. }
  53. // SO_EXCLUSIVEADDRUSE is only supported in 2000/XP/2003 and higher OS.
  54. if(comm->osvi.dwMajorVersion > 5) {
  55. // 防止窃听, 设定独占端口
  56. BOOL bExAddrUse = TRUE;
  57. if(setsockopt(m_Socket, SOL_SOCKET, ((int)(~SO_REUSEADDR)), (const char*)&bExAddrUse, sizeof(BOOL)) == SOCKET_ERROR) {
  58. comm->logFile.StatusErr("Setting UDP socket as SO_EXCLUSIVEADDRUSE", WSAGetLastError());
  59. return PRT_NET;
  60. }
  61. }
  63. //int newSize = SO_MAX_MSG_SIZE*100;
  64. //setsockopt(m_Socket, SOL_SOCKET, SO_RCVBUF, (const char*)&newSize, sizeof(int));
  65. //setsockopt(m_Socket, SOL_SOCKET, SO_SNDBUF, (const char*)&newSize, sizeof(int));
  67. sockaddr_in addr;
  68. memset(&addr, 0, sizeof(addr));
  69. addr.sin_family = AF_INET;
  70. // 绑定所有本机IP
  71. addr.sin_addr.s_addr = INADDR_ANY;
  73. USHORT port = 0;
  74. for(int i = 0; i < 1000; ++i) {
  75. port = static_cast<USHORT>(rand(&comm->ctx))%10000 + 50000;
  77. addr.sin_port = htons(port);
  78. if(bind(m_Socket, (sockaddr*)&addr, sizeof(addr)) != SOCKET_ERROR)
  79. break;
  80. comm->logFile.StatusOut("Bind UDP socket at port %d. Failed", port);
  81. }
  82. if(i == 1000)
  83. return PRT_INIT_BIND;
  85. comm->logFile.StatusOut("Bind UDP socket at port %d.", port);
  87. return PRT_OK;
  88. }
  90. BOOL CSClient::ParseMsg(sockaddr_in& addr, int& addrlen) {
  91. CriticalSection::Owner lock(sendCS);
  92. int recvSize = recvfrom(m_Socket, recvBuf, MAX_UDP_LEN, 0, (sockaddr*)&addr, &addrlen);
  93. if(recvSize == SOCKET_ERROR)
  94. return FALSE;
  96. // 如果来源地址不是TS，就返回
  97. if(memcmp(&addr, &comm->trackerIP, sizeof(comm->trackerIP)) != 0)
  98. return FALSE;
  100. // 如果过小，则不是正常的包
  101. if(recvSize < sizeof(UINT)+sizeof(UINT8))
  102. return FALSE;
  104. // 把移动指针放到数据的起始地址
  105. content = recvBuf;
  107. // 读取消息大小
  108. UINT msgSize;
  109. CopyMoveSrc(&msgSize, content, sizeof(msgSize));
  110. if(msgSize != recvSize)
  111. return FALSE;
  113. // 读取消息类型
  114. UINT8 msgType;
  115. CopyMoveSrc(&msgType, content, sizeof(msgType));
  117. if(!SwitchMsg(msgType)) {
  118. // 重新登录
  119. SendLogout();
  120. SendLogin();
  121. return FALSE;
  122. }
  123. return TRUE;
  124. }
  126. BOOL CSClient::SwitchMsg(UINT8 msgType) {
  127. switch(msgType) {
  128. case TS2NP_WELCOME:
  129. if(!OnWelcome())
  130. return FALSE;
  131. break;
  132. case TS2NP_PEERS:
  133. if(!OnPeers())
  134. return FALSE;
  135. break;
  136. case TS2NP_CONNECT_TO:
  137. if(!OnConnectTo())
  138. return FALSE;
  139. break;
  140. case TS2NP_RESINFO:
  141. if(!OnResInfo())
  142. return FALSE;
  143. break;
  144. case TS2NP_MSG:
  145. if(!OnMsg())
  146. return FALSE;
  147. break;
  148. default:
  149. return FALSE;
  150. }
  151. return TRUE;
  152. }
  154. BOOL CSClient::OnWelcome() {
  155. // 复制验证码
  156.     CopyMoveSrc(checkCode, content, 7);
  157. // 复制本机地址
  158. CopyMoveSrc(&comm->localAddress, content, sizeof(comm->localAddress));
  160. assert(comm->localAddress.outerIP.sin_addr.s_addr != 0);
  162. comm->logFile.StatusOut("CSClient Logon TS...%s", comm->p2pMgr.FormatIPAddress(comm->localAddress));
  164. isLogin = true;
  165. loginFailCount = 0;
  167. lastRecvPeers = timeGetTime();
  169. // 如果当前资源没有告知TS，则发送ReqRes
  170. if(!isReqResSucceed && comm->currRes) {
  171. SendReqRes();
  172. }
  174. return TRUE;
  175. }
  176. BOOL CSClient::OnPeers() {
  177. // 是否需要尝试连接CP
  178. bool tryCP = false;
  180. // 复制CP地址个数
  181. UINT8 listSize = 0;
  182. CopyMoveSrc(&listSize, content, sizeof(listSize));
  183. if(listSize > 0 && listSize < 100) {
  184. // 复制CP地址
  185. for(UINT8 i = 0; i < listSize; ++i) {
  186. PeerInfoWithAddr nAddr;
  187. CopyMoveSrc(&nAddr.outerIP, content, sizeof(NormalAddress));
  188. nAddr.subnetIP.sin_addr.s_addr = 0xffffffff;
  189. nAddr.subnetIP.sin_port = nAddr.outerIP.sin_port;
  190. comm->logFile.StatusOut("CSClient got CP address %s...", comm->p2pMgr.FormatIPAddress(nAddr));
  191. nAddr.layer = 0;
  192. nAddr.isCachePeer = true;
  194. if(nAddr.outerIP.sin_addr.s_addr != 0) {
  195. comm->p2pMgr.AddPeerInfo(nAddr);
  196. // 如果得到CP地址，就尝试连接
  197. tryCP = true;
  198. }
  199. }
  200. }
  202. // 复制NP地址个数
  203. CopyMoveSrc(&listSize, content, sizeof(listSize));
  204. if(listSize > 0 && listSize < 100) {
  205. for(UINT8 i = 0; i < listSize; ++i) {
  206. PeerInfoWithAddr nAddr;
  207. CopyMoveSrc(&nAddr, content, sizeof(nAddr));
  208. assert(nAddr.outerIP.sin_addr.s_addr != 0);
  209. nAddr.isCachePeer = false;
  210. // put in known np list
  211. comm->p2pMgr.AddPeerInfo(nAddr);
  212. }
  213. comm->logFile.StatusOut("CSClient got %d NP.", listSize);
  214. }
  216. lastRecvPeers = timeGetTime();
  218. // 尝试连接CP
  219. if(tryCP)
  220. comm->tryClient.Try();
  221. return TRUE;
  222. }
  224. BOOL CSClient::OnConnectTo() {
  225. // 复制NP的地址
  226. P2PAddress nAddr;
  227. CopyMoveSrc(&nAddr, content, sizeof(nAddr));
  228. // 是否使用free outgoing
  229. bool connectForFree;
  230. CopyMoveSrc(&connectForFree, content, sizeof(connectForFree));
  231. // add to connectto peer list
  232. comm->p2pMgr.AddConnectTo(nAddr, connectForFree);
  234. comm->logFile.StatusOut("CSClient got ConnectTo.");
  235. return TRUE;
  236. }
  238. BOOL CSClient::OnResInfo() {
  239. // TODO: 接收返回的资源信息
  240. return TRUE;
  241. }
  243. BOOL CSClient::OnMsg() {
  244. // 复制错误代码
  245. UINT16 errCode;
  246. CopyMoveSrc(&errCode, content, sizeof(errCode));
  248. // 是否需要退出
  249. bool shouldQuit;
  250. CopyMoveSrc(&shouldQuit, content, sizeof(shouldQuit));
  252. // 是否不能继续登录
  253. bool bShouldStopLogin = false;
  255. // 根据错误代码处理
  256. switch(errCode) {
  257. case ERR_PROTOCOL_FORMAT:
  258. comm->logFile.StatusOut("TS sent ERR_PROTOCOL_FORMAT error!");
  259. bShouldStopLogin = true;
  260. break;
  261. case ERR_AUTHORIZATION:
  262. comm->logFile.StatusOut("TS sent ERR_AUTHORIZATION error!");
  263. bShouldStopLogin = true;
  264. break;
  265. case ERR_INTERNAL:
  266. comm->logFile.StatusOut("TS sent ERR_INTERNAL error!");
  267. break;
  268. case ERR_LOW_VERSION:
  269. comm->logFile.StatusOut("TS sent ERR_LOW_VERSION error!");
  270. // 发送版本太低的消息给外界
  271. comm->PostErrMessage(PNT_LOW_VERSION, 0, true);
  272. bShouldStopLogin = true;
  273. break;
  274. case ERR_NO_SUCH_PEER:
  275. comm->logFile.StatusOut("TS sent ERR_NO_SUCH_PEER error!");
  276. break;
  277. case ERR_NO_SUCH_RES:
  278. comm->logFile.StatusOut("TS sent ERR_NO_SUCH_RES error!");
  279. break;
  280. case ERR_CHECK_BYTES:
  281. comm->logFile.StatusOut("TS sent ERR_CHECK_BYTES error!");
  282. break;
  283. case ERR_ADD_RES_OK:
  284. // 请求资源成功！
  285. isReqResSucceed = true;
  286. comm->logFile.StatusOut("TS sent ERR_ADD_RES_OK!");
  288. // 发送第一次Report
  289. SendReport(true);
  290. break;
  291. default:
  292. shouldQuit = true;
  293. }
  294. if(shouldQuit) {
  295. isLogin = false;
  296. loginFailCount = 0;
  297. }
  298. if(bShouldStopLogin)
  299. bStopLogin = true;
  301. return TRUE;
  302. }
  304. void CSClient::SendLogin() {
  305. CriticalSection::Owner lock(sendCS);
  306. if(isLogin)
  307. return;
  308. if(bStopLogin)
  309. return;
  311. if(!SendBegin(NP2TS_LOGIN))
  312. return;
  314. // 用户ID和密码
  315. CopyMoveDes(content, &comm->userID, sizeof(comm->userID));
  316. CopyMoveDes(content, comm->userPass, MD5_LEN);
  318. // 版本号和监听端口
  319. CopyMoveDes(content, &comm->cfgData.COMMUNICATOR_VERSION, 
  320. sizeof(comm->cfgData.COMMUNICATOR_VERSION));
  321. // 写入监听端口
  322. CopyMoveDes(content, &comm->localAddress.subnetIP.sin_port, sizeof(comm->localAddress.subnetIP.sin_port));
  324. // 本机IP列表
  325. in_addr* addrList = NULL;
  326. UINT8 ipSize = GetLocalIPList(addrList);
  327. if(ipSize == 0) {
  328. comm->logFile.StatusErr("Get Local IP List", WSAGetLastError());
  329. return;
  330. }
  331. CopyMoveDes(content, &ipSize, sizeof(ipSize));
  332. CopyMoveDes(content, addrList, ipSize*sizeof(in_addr));
  333. delete [] addrList;
  334. addrList = NULL;
  336. comm->logFile.StatusOut("CSClient sent NP2TS_LOGIN.");
  337. isReqResSucceed = false;
  339. if(loginFailCount == 2) {
  340. // 两次登录失败，设置默认CP地址
  341. if(comm->currRes)
  342. comm->currRes->SetDefaultCP();
  343. }
  345. loginFailCount++;
  346. if(loginFailCount == maxFailCount) {
  347. // 发送登陆失败的消息给外界
  348. comm->PostErrMessage(PNT_CANNOT_LOGON_TS, 0, true);
  349. loginFailCount = 0;
  350. // 按MAX_LOGIN_FAIL_COUNT倍递增
  351. maxFailCount *= MAX_LOGIN_FAIL_COUNT;
  352. }
  354. SendEnd();
  355. }
  357. void CSClient::SendResList() {
  358. CriticalSection::Owner lock(sendCS);
  359. if(!isLogin || !comm->currRes)
  360. return;
  361. if(!SendBegin(NP2TS_RES_LIST))
  362. return;
  364. // 要共享的资源个数
  365. UINT8 resNum = 1;
  366. CopyMoveDes(content, &resNum, sizeof(resNum));
  368. // 依次写入所有要共享的资源
  369. CopyMoveDes(content, comm->currRes->GetHashCode().data(), MD5_LEN);
  370. // send block intervals
  371. UINT8 intervalNum = MAX_BLOCK_INTERVALS;
  372. comm->currRes->GetAllIntervals((BlockInterval*)(content+sizeof(intervalNum)), intervalNum);
  373. CopyMoveDes(content, &intervalNum, sizeof(intervalNum));
  374. content += intervalNum*sizeof(BlockInterval);
  376. comm->logFile.StatusOut("CSClient sent NP2TS_RES_LIST.");
  378. SendEnd();
  379. }
  381. void CSClient::SendReqRes() {
  382. CriticalSection::Owner lock(sendCS);
  383. assert(comm->currRes);
  385. // 如果已经登录，则发送请求资源；否则，登录成功之后再发送
  386. if(!isLogin) {
  387. comm->logFile.StatusOut("NOT LOGIN YET, first request will be sent after recving TS2NP_WELCOME...");
  388. return;
  389. }
  390. if(!comm->currRes)
  391. return;
  393. if(!SendBegin(NP2TS_REQ_RES))
  394. return;
  396. // request resource md5
  397. CopyMoveDes(content, comm->currRes->GetHashCode().data(), MD5_LEN);
  399. // send block intervals
  400. UINT8 intervalNum = MAX_BLOCK_INTERVALS;
  401. comm->currRes->GetAllIntervals((BlockInterval*)(content+sizeof(intervalNum)), intervalNum);
  402. CopyMoveDes(content, &intervalNum, sizeof(intervalNum));
  403. content += intervalNum*sizeof(BlockInterval);
  405. // current block ID
  406. UINT currBlockID = comm->currRes->GetPlayingBlock();
  407. CopyMoveDes(content, &currBlockID, sizeof(currBlockID));
  409. // need CP?
  410. bool needCP = (comm->p2pMgr.GetCPConCount(true) < MAX_CONNECTION_OF_CP);
  411. CopyMoveDes(content, &needCP, sizeof(needCP));
  413. comm->logFile.StatusOut("CSClient sent NP2TS_REQ_RES.");
  415. SendEnd();
  417. // 发送了请求资源的消息，等待请求成功的消息返回
  418. isReqResSucceed = false;
  419. }
  421. void CSClient::SendDelRes(BaseResource* res) {
  422. CriticalSection::Owner lock(sendCS);
  423. if(!res || !isLogin)
  424. return;
  425. if(!SendBegin(NP2TS_DEL_RES))
  426. return;
  428. // request resource md5
  429. CopyMoveDes(content, res->GetHashCode().data(), MD5_LEN);
  431. comm->logFile.StatusOut("CSClient sent NP2TS_DEL_RES(%s).", res->GetHashCode().data());
  433. SendEnd();
  434. }
  436. void CSClient::SendReport(bool bRefresh) {
  437. CriticalSection::Owner lock(sendCS);
  438. if(!isLogin)
  439. return;
  441. if(!SendBegin(NP2TS_REPORT))
  442. return;
  444. // 本机信息
  445. CorePeerInfo thisPeer;
  446. comm->p2pMgr.GetSelfInfo(thisPeer);
  447. CopyMoveDes(content, &thisPeer, sizeof(thisPeer));
  449. // 如果没有当前资源，则发送bRefresh=true和intervalNum=0
  450. if(!comm->currRes)
  451. bRefresh = true;
  453. // 是否更新全部区间列表
  454. CSClient::CopyMoveDes(content, &bRefresh, sizeof(bRefresh));
  456. // send block intervals
  457. // TODO: 这里存在重复代码，有待改进
  458. UINT8 intervalNum = comm->currRes?MAX_BLOCK_INTERVALS:0;
  459. if(bRefresh) {
  460. if(comm->currRes) {
  461. // 如果是更新全部区间，则取得当前资源全部区间的列表，并发送
  462. comm->currRes->GetAllIntervals((BlockInterval*)(content+sizeof(intervalNum)), intervalNum);
  463. }
  464. CSClient::CopyMoveDes(content, &intervalNum, sizeof(intervalNum));
  465. content += intervalNum*sizeof(BlockInterval);
  466. assert(intervalNum <= 20);
  467. }
  468. else {
  469. // 如果是更新变化的区间，则先后写入增加的区间和减少的区间
  470. comm->currRes->GetDiffIntervals((BlockInterval*)(content+sizeof(intervalNum)), intervalNum, true, true);
  471. CSClient::CopyMoveDes(content, &intervalNum, sizeof(intervalNum));
  472. content += intervalNum*sizeof(BlockInterval);
  473. assert(intervalNum <= 20);
  475. comm->currRes->GetDiffIntervals((BlockInterval*)(content+sizeof(intervalNum)), intervalNum, true, false);
  476. CSClient::CopyMoveDes(content, &intervalNum, sizeof(intervalNum));
  477. content += intervalNum*sizeof(BlockInterval);
  478. assert(intervalNum <= 20);
  479. }
  481. // 发送本机传输信息
  482. TransferInfo ti;
  483. comm->p2pMgr.GetTransferInfo(ti);
  484. CSClient::CopyMoveDes(content, &ti, sizeof(TransferInfo));
  486. comm->logFile.StatusOut("CSClient sent NP2TS_REPORT.");
  488. SendEnd();
  489. }
  491. void CSClient::SendNeedPeers() {
  492. CriticalSection::Owner lock(sendCS);
  493.   if(!comm->currRes || !isLogin)
  494. return;
  496. // 如果请求资源还没有成功，就发送请求消息，然后返回
  497. if(!isReqResSucceed) {
  498. SendReqRes();
  499. return;
  500. }
  502. // 如果当前下载速度大于直播的码率，则不必发送NeedPeers
  503. if(comm->p2pMgr.GetCurDownSpeed()/1024 > comm->currRes->GetBitRate())
  504. return;
  506. // 如果已知的Peer列表足够多，就不用发送NeedPeers
  507. if(comm->p2pMgr.GetKnownPeerCount() + comm->p2pMgr.GetAllCount() >= MIN_KNOWN_PEERS)
  508. return;
  510. if(!SendBegin(NP2TS_NEED_PEERS))
  511. return;
  513. // 本机是否需要CachePeer的地址
  514. bool needCP = (comm->p2pMgr.GetCPConCount(true) < MAX_CONNECTION_OF_CP);
  515. CopyMoveDes(content, &needCP, sizeof(needCP));
  518. UINT currBlockID = comm->currRes->GetPlayingBlock();
  519. CopyMoveDes(content, &currBlockID, sizeof(currBlockID));
  521. // 本机在P2P网络中所属的层
  522. UINT8 layer = comm->p2pMgr.GetSelfLayer();
  523. CopyMoveDes(content, &layer, sizeof(layer));
  525. comm->logFile.StatusOut("CSClient sent NP2TS_NEED_PEERS.");
  527. SendEnd();
  528. }
  530. void CSClient::SendQueryRes() {
  531. // TODO: 什么时候发送请求的，目前客户端不关心现在有多少频道(md5)，每个频道有多少人
  532. }
  534. void CSClient::SendLogout() {
  535. CriticalSection::Owner lock(sendCS);
  536.   if(!comm->currRes || !isLogin)
  537. return;
  539. if(!SendBegin(NP2TS_LOGOUT))
  540. return;
  542. SendEnd();
  544. comm->logFile.StatusOut("CSClient sent NP2TS_LOGOUT.");
  546. // 退出登录
  547. isLogin = false;
  548. loginFailCount = 0;
  549. }
  551. void CSClient::SendReport2() {
  552. CriticalSection::Owner lock(sendCS);
  553. if(!comm->currRes || !isLogin)
  554. return;
  556. if(!SendBegin(NP2TS_REPORT2))
  557. return;
  559. UINT16 value = 0;
  561. // playing block
  562. UINT currBlockID = comm->currRes->GetPlayingBlock();
  563. CopyMoveDes(content, &currBlockID, sizeof(currBlockID));
  565. // 当前缓冲的时间，没有缓冲的话就是0
  566. value = comm->currRes->GetBufferingTime();
  567. CopyMoveDes(content, &value, sizeof(value));
  569. // 之前缓冲的次数
  570. value = comm->currRes->GetBufferCount();
  571. CopyMoveDes(content, &value, sizeof(value));
  573. // 之前所有缓冲所用的时间
  574. value = comm->currRes->GetBufferTime();
  575. CopyMoveDes(content, &value, sizeof(value));
  577. // connect Fail Count
  578. value = comm->p2pMgr.GetConnectFailCount();
  579. CopyMoveDes(content, &value, sizeof(value));
  581. // incoming connection count
  582. value = comm->p2pMgr.GetTotalIncomingCount();
  583. CopyMoveDes(content, &value, sizeof(value));
  585. // outgoing connection count
  586. value = comm->p2pMgr.GetTotalOutgoingCount();
  587. CopyMoveDes(content, &value, sizeof(value));
  589. // avg incoming connection elapsed time
  590. value = comm->p2pMgr.GetAvgIncomingTime();
  591. CopyMoveDes(content, &value, sizeof(value));
  593. // avg outgoing connection elapsed time
  594. value = comm->p2pMgr.GetAvgOutgoingTime();
  595. CopyMoveDes(content, &value, sizeof(value));
  597. // message percent
  598. float msgPercent = comm->p2pMgr.GetMessagePercent();
  599. CopyMoveDes(content, &msgPercent, sizeof(msgPercent));
  601. // 发送本机传输信息
  602. TransferInfo ti;
  603. comm->p2pMgr.GetTransferInfo(ti);
  604. CSClient::CopyMoveDes(content, &ti, sizeof(TransferInfo));
  606. comm->logFile.StatusOut("CSClient sent NP2TS_REPORT2.");
  608. SendEnd();
  609. }
  611. BOOL CSClient::SendBegin(UINT8 msgType) {
  612. // 先留着消息大小不写，到最后再写
  613. content = recvBuf+sizeof(UINT);
  615. // 消息类型
  616. CopyMoveDes(content, &msgType, sizeof(msgType));
  618. if(msgType != NP2TS_LOGIN) {
  619. // TS的验证码
  620. CopyMoveDes(content, checkCode, 7);
  621. }
  623. return TRUE;
  624. }
  626. void CSClient::SendEnd() {
  627. // 消息的大小就是移动的指针减去初始的指针
  628. UINT msgSize = content-recvBuf;
  629. memcpy(recvBuf, &msgSize, sizeof(msgSize));
  631. // 发送
  632. sockaddr_in addr;
  633. memcpy(&addr, &comm->trackerIP, sizeof(comm->trackerIP));
  635. int ret = sendto(m_Socket, (const char*)recvBuf, msgSize, 
  636. 0, (SOCKADDR*)&addr, sizeof(addr));
  637. if(ret == SOCKET_ERROR) {
  638. comm->logFile.StatusErr("sendto() error, rebind udp socket", WSAGetLastError());
  639. // 重新连接TS
  640. Reinit();
  641. }
  642. }
  644. void CSClient::CopyMoveSrc(void * des, const char *& src, size_t size) {
  645. assert(des && src);
  646. if(!des || !src)
  647. return;
  648. memcpy(des, src, size);
  649. src += size;
  650. }
  652. void CSClient::CopyMoveDes(char *& des, const void * src, size_t size) {
  653. assert(des && src);
  654. if(!des || !src)
  655. return;
  656. memcpy(des, src, size);
  657. des += size;
  658. }
  660. UINT8 CSClient::GetLocalIPList(in_addr*& addrList) {
  661. UINT8 ipCount = 0;
  662. addrList = NULL;
  664. // Get local host name
  665. char szHostName[128] = "";
  666. if(gethostname(szHostName, sizeof(szHostName)))
  667. return 0;
  669. // Get local IP addresses
  670. hostent *pHost = 0;
  671. if((pHost = gethostbyname(szHostName)) == NULL)
  672. return 0;
  674. // 计算IP地址个数
  675. while(pHost->h_addr_list[ipCount]) {
  676. ipCount++;
  677. // 注意不能越界了，超过0xff个IP地址将不被支持
  678. if(ipCount == 0xff)
  679. break;
  680. }
  682. // 分配数组，要在外部释放
  683. addrList = new in_addr[ipCount];
  684. if(!addrList)
  685. return 0;
  687. // 复制IP列表
  688. for(int i = 0; i < ipCount; ++i) {
  689. memcpy(&addrList[i], pHost->h_addr_list[i], pHost->h_length);
  690. }
  692. // 返回IP地址个数
  693. return ipCount;
  694. }
  696. // 重新初始化
  697. void CSClient::Reinit() {
  698. if(m_Socket != INVALID_SOCKET)
  699. closesocket(m_Socket);
  701. Init();
  703. SendLogin();
  704. }
  706. }