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

源码开发语言/平台
当前位置：首页 > 源码/资料 > Internet/网络编程 > P2P编程 > 查看源码
wsocket.cpp
资源名称：P2PSource-PeerCast.rar　[点击查看]
上传用户：chn_coc
上传日期：2007-12-20
资源大小：563k
文件大小：10k
源码类别：

P2P编程

开发平台：

Windows_Unix

wsocket.cpp：源码内容
  1. // ------------------------------------------------ // File : wsocket.cpp // Date: 4-apr-2002 // Author: giles // Desc:  // Windows version of ClientSocket. Handles the nitty gritty of actually // reading and writing TCP // // (c) 2002 peercast.org // ------------------------------------------------ // This program is free software; you can redistribute it and/or modify // it under the terms of the GNU General Public License as published by // the Free Software Foundation; either version 2 of the License, or // (at your option) any later version. // This program is distributed in the hope that it will be useful, // but WITHOUT ANY WARRANTY; without even the implied warranty of // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the // GNU General Public License for more details. // ------------------------------------------------
  2. // TODO: fix socket closing
  4. #include "winsock2.h"
  5. #include <windows.h> #include <stdio.h> #include "wsocket.h" #include "stats.h"
  7. // --------------------------------------------------
  8. void WSAClientSocket::init()
  9. {
  10. WORD wVersionRequested;
  11. WSADATA wsaData;
  12. int err;
  13.     
  14. wVersionRequested = MAKEWORD( 2, 0 );
  15. err = WSAStartup( wVersionRequested, &wsaData );
  16. if ( err != 0 ) throw SockException("Unable to init sockets");     //LOG4("WSAStartup:  OK"); } // -------------------------------------------------- bool ClientSocket::getHostname(char *str,unsigned int ip) { HOSTENT *he; ip = htonl(ip); he = gethostbyaddr((char *)&ip,sizeof(ip),AF_INET); if (he) { strcpy(str,he->h_name); return true; }else return false; } // -------------------------------------------------- unsigned int ClientSocket::getIP(char *name) { char szHostName[256]; if (!name) { if (gethostname(szHostName, sizeof(szHostName))==0) name = szHostName; else return 0; } HOSTENT *he = WSAClientSocket::resolveHost(name); if (!he) return 0; LPSTR lpAddr = he->h_addr_list[0]; if (lpAddr) { struct in_addr  inAddr; memmove (&inAddr, lpAddr, 4); return inAddr.S_un.S_un_b.s_b1<<24 |    inAddr.S_un.S_un_b.s_b2<<16 |    inAddr.S_un.S_un_b.s_b3<<8 |    inAddr.S_un.S_un_b.s_b4; } return 0; } // --------------------------------------------------
  17. void WSAClientSocket::setLinger(int sec)
  18. {
  19. linger linger;
  20. linger.l_onoff = (sec>0)?1:0;
  21.     linger.l_linger = sec;
  23. if (setsockopt(sockNum, SOL_SOCKET, SO_LINGER, (const char *)&linger, sizeof (linger)) == -1) 
  24. throw SockException("Unable to set LINGER");
  25. }
  27. // -------------------------------------------------- void WSAClientSocket::setBlocking(bool yes) { unsigned long op = yes ? 0 : 1; if (ioctlsocket(sockNum, FIONBIO, &op) == SOCKET_ERROR) throw SockException("Can`t set blocking"); } // -------------------------------------------------- void WSAClientSocket::setNagle(bool on) {     int nodelay = (on==false); if (setsockopt(sockNum, SOL_SOCKET, TCP_NODELAY, (char *)&nodelay, sizeof nodelay) == -1)  throw SockException("Unable to set NODELAY"); } // -------------------------------------------------- void WSAClientSocket::setReuse(bool yes) { unsigned long op = yes ? 1 : 0; if (setsockopt(sockNum,SOL_SOCKET,SO_REUSEADDR,(char *)&op,sizeof(unsigned long)) == -1)  throw SockException("Unable to set REUSE"); } // -------------------------------------------------- HOSTENT *WSAClientSocket::resolveHost(const char *hostName) { HOSTENT *he; if ((he = gethostbyname(hostName)) == NULL) { // if failed, try using gethostbyaddr instead unsigned long ip = inet_addr(hostName); if (ip == INADDR_NONE) return NULL; if ((he = gethostbyaddr((char *)&ip,sizeof(ip),AF_INET)) == NULL) return NULL; } return he; } // -------------------------------------------------- void WSAClientSocket::open(Host &rh) { sockNum = socket (AF_INET, SOCK_STREAM, IPPROTO_TCP); if (sockNum == INVALID_SOCKET) throw SockException("Can`t open socket"); setBlocking(false); #ifdef DISABLE_NAGLE setNagle(false); #endif host = rh; memset(&remoteAddr,0,sizeof(remoteAddr)); remoteAddr.sin_family = AF_INET; remoteAddr.sin_port = htons(host.port); remoteAddr.sin_addr.S_un.S_addr = htonl(host.ip); } // -------------------------------------------------- void WSAClientSocket::checkTimeout(bool r, bool w) {     int err = WSAGetLastError();     if (err == WSAEWOULDBLOCK)     {
  28. timeval timeout; fd_set read_fds; fd_set write_fds;
  29. timeout.tv_sec = 0;
  30. timeout.tv_usec = 0;
  32.         FD_ZERO (&write_fds);
  33. if (w)
  34. {
  35. timeout.tv_sec = (int)this->writeTimeout/1000;
  36. FD_SET (sockNum, &write_fds);
  37. }
  39.         FD_ZERO (&read_fds);
  40. if (r)
  41. {
  42. timeout.tv_sec = (int)this->readTimeout/1000;
  43.         FD_SET (sockNum, &read_fds);
  44. }
  46. timeval *tp;
  47. if (timeout.tv_sec)
  48. tp = &timeout;
  49. else
  50. tp = NULL;
  52. int r=select (NULL, &read_fds, &write_fds, NULL, tp);         if (r == 0)
  53. throw TimeoutException();
  54. else if (r == SOCKET_ERROR) throw SockException("select failed."); }else{ char str[32]; sprintf(str,"%d",err); throw SockException(str); }
  55. }
  56. // --------------------------------------------------
  57. void WSAClientSocket::checkTimeout2(bool r, bool w)
  58. {
  59.     {
  61. timeval timeout;
  62. fd_set read_fds;
  63. fd_set write_fds;
  65. timeout.tv_sec = 0;
  66. timeout.tv_usec = 0;
  68.         FD_ZERO (&write_fds);
  69. if (w)
  70. {
  71. timeout.tv_sec = (int)this->writeTimeout/1000;
  72. FD_SET (sockNum, &write_fds);
  73. }
  75.         FD_ZERO (&read_fds);
  76. if (r)
  77. {
  78. timeout.tv_sec = (int)this->readTimeout/1000;
  79.         FD_SET (sockNum, &read_fds);
  80. }
  82. timeval *tp;
  83. if (timeout.tv_sec)
  84. tp = &timeout;
  85. else
  86. tp = NULL;
  89. int r=select (NULL, &read_fds, &write_fds, NULL, tp);
  91.         if (r == 0)
  92. throw TimeoutException();
  93. else if (r == SOCKET_ERROR)
  94. throw SockException("select failed.");
  95. }
  96. }
  98. // -------------------------------------------------- Host WSAClientSocket::getLocalHost() { struct sockaddr_in localAddr; int len = sizeof(localAddr);     if (getsockname(sockNum, (sockaddr *)&localAddr, &len) == 0) return Host(SWAP4(localAddr.sin_addr.s_addr),0); else return Host(0,0); } // -------------------------------------------------- void WSAClientSocket::connect() { if (::connect(sockNum,(struct sockaddr *)&remoteAddr,sizeof(remoteAddr)) == SOCKET_ERROR) checkTimeout(false,true); } // -------------------------------------------------- int WSAClientSocket::read(void *p, int l) {
  99. int bytesRead=0; while (l) { int r = recv(sockNum, (char *)p, l, 0); if (r == SOCKET_ERROR) {
  100. // non-blocking sockets always fall through to here
  101. checkTimeout(true,false);
  102. }else if (r == 0) {
  103. throw EOFException("Closed on read");
  105. }else { stats.add(Stats::BYTESIN,r); if (host.localIP()) stats.add(Stats::LOCALBYTESIN,r); updateTotals(r,0); bytesRead += r; l -= r; p = (char *)p+r; }
  106. } return bytesRead; } // --------------------------------------------------
  107. int WSAClientSocket::readUpto(void *p, int l)
  108. {
  109. int bytesRead=0;
  110. while (l)
  111. {
  112. int r = recv(sockNum, (char *)p, l, 0);
  113. if (r == SOCKET_ERROR)
  114. {
  115. // non-blocking sockets always fall through to here
  116. checkTimeout(true,false);
  118. }else if (r == 0)
  119. {
  120. break;
  121. }else
  122. {
  123. stats.add(Stats::BYTESIN,r);
  124. if (host.localIP())
  125. stats.add(Stats::LOCALBYTESIN,r);
  126. updateTotals(r,0);
  127. bytesRead += r;
  128. l -= r;
  129. p = (char *)p+r;
  130. }
  131. }
  132. return bytesRead;
  133. }
  134. // -------------------------------------------------- void WSAClientSocket::write(const void *p, int l) { while (l) { int r = send(sockNum, (char *)p, l, 0);
  135. if (r == SOCKET_ERROR) {
  136. checkTimeout(false,true);
  137. }
  138. else if (r == 0) { throw SockException("Closed on write"); }
  139. else
  140. if (r > 0) {
  141. stats.add(Stats::BYTESOUT,r); if (host.localIP()) stats.add(Stats::LOCALBYTESOUT,r); updateTotals(0,r); l -= r; p = (char *)p+r;
  142. } }
  143. } // -------------------------------------------------- void WSAClientSocket::bind(Host &h) { struct sockaddr_in localAddr; if ((sockNum = socket (PF_INET, SOCK_STREAM, IPPROTO_TCP)) == -1) throw SockException("Can`t open socket"); setBlocking(false);
  144. setReuse(true); memset(&localAddr,0,sizeof(localAddr)); localAddr.sin_family = AF_INET; localAddr.sin_port = htons(h.port); localAddr.sin_addr.s_addr = INADDR_ANY; if( ::bind (sockNum, (sockaddr *)&localAddr, sizeof(localAddr)) == -1) throw SockException("Can`t bind socket"); if (::listen(sockNum,SOMAXCONN)) throw SockException("Can`t listen",WSAGetLastError()); host = h; } // -------------------------------------------------- ClientSocket *WSAClientSocket::accept() { int fromSize = sizeof(sockaddr_in); sockaddr_in from; int conSock = ::accept(sockNum,(sockaddr *)&from,&fromSize); if (conSock ==  INVALID_SOCKET) return NULL;     WSAClientSocket *cs = new WSAClientSocket(); cs->sockNum = conSock; cs->host.port = from.sin_port; cs->host.ip = from.sin_addr.S_un.S_un_b.s_b1<<24 |   from.sin_addr.S_un.S_un_b.s_b2<<16 |   from.sin_addr.S_un.S_un_b.s_b3<<8 |   from.sin_addr.S_un.S_un_b.s_b4; cs->setBlocking(false); #ifdef DISABLE_NAGLE cs->setNagle(false); #endif return cs; } // -------------------------------------------------- void WSAClientSocket::close() { if (sockNum) {
  145. shutdown(sockNum,SD_SEND);
  147. setReadTimeout(2000);
  148. try
  149. {
  150. //char c;
  151. //while (readUpto(&c,1)!=0);
  152. //readUpto(&c,1);
  153. }catch(StreamException &) {}
  155. if (closesocket(sockNum))
  156. LOG_ERROR("closesocket() error");
  158. sockNum=0; } }
  159. // --------------------------------------------------
  160. bool WSAClientSocket::readReady()
  161. {
  162. timeval timeout;
  163. fd_set read_fds;
  165. timeout.tv_sec = 0;
  166. timeout.tv_usec = 0;
  168.     FD_ZERO (&read_fds);
  169.     FD_SET (sockNum, &read_fds);
  171. return select (sockNum+1, &read_fds, NULL, NULL, &timeout) == 1;
  172. }