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

源码开发语言/平台
当前位置：首页 > 源码/资料 > Internet/网络编程 > Email客户端 > 查看源码
tcp.cpp
资源名称：FindMail.zip　[点击查看]
上传用户：feituo2008
上传日期：2013-02-02
资源大小：493k
文件大小：12k
源码类别：

Email客户端

开发平台：

Visual C++

tcp.cpp：源码内容
  1. #include "stdafx.h"
  2. #include <winsock.h>
  3. #include <stdio.h>
  4. #include <stdlib.h>
  5. #include <time.h>
  6. #include "util.h"
  8. #include "tcp.h"
  10. extern WriteLog(char *file_name, char *format, ...);
  11. extern WriteStat(char *format, ...);
  13. int wsa_ok =0;
  14. void peek_message(void);
  15. int get_local_ip(char *ip);
  17. class CTcpInit
  18. {
  19. public:
  20. CTcpInit()
  21. {
  22. tcp_init();
  23. }
  24. ~CTcpInit()
  25. {
  26. tcp_exit();
  27. }
  28. };
  30. CTcpInit g_ttt;
  32. int tcp_init()
  33. {
  34.     WSAData wsa;
  36.     //sd_bind =sd_connect =sd_accept =-1;
  37.     if(WSAStartup(MAKEWORD(1, 1), &wsa) !=0)
  38.         return -1;
  39.     wsa_ok =1;
  40. return 0;
  41. }
  43. int tcp_exit()
  44. {
  45.   if(wsa_ok) WSACleanup();
  47.   return 0;
  48. }
  50. int tcp_status(int  sd, char *type, int timeout)
  51. {
  52.   fd_set rset, wset, eset;
  53.   fd_set FAR *prset =NULL, *pwset =NULL, *peset =NULL;
  54.   struct timeval tval;
  55.   int i, status, err_no =0;
  56.   time_t t1, t2;
  57.   MSG msg;
  59.   tval.tv_sec =0;
  60.   tval.tv_usec =1;
  61.   time(&t1);
  62.   t2 =t1;
  63.   while(t2-t1 < timeout)
  64.   {
  65.     FD_ZERO(&rset);
  66.     FD_ZERO(&wset);
  67.     FD_ZERO(&eset);
  69.     for(i =0; i<(int)strlen(type); i++)
  70.     {
  71.       if(type[i] =='r') { FD_SET(sd, &rset); prset =&rset; }
  72.       if(type[i] =='w') { FD_SET(sd, &wset); pwset =&wset; }
  73.       if(type[i] =='e') { FD_SET(sd, &eset); peset =&eset; }
  74.     }
  75.     status =select(-1, prset, pwset, peset, &tval);
  76. err_no =WSAGetLastError();
  77. int err=GetLastError();
  78. //WriteStat("select err_no=%d, err=%d", err_no, err);
  79.     time(&t2);
  80. if(status ==0)
  81.     {
  82.       if(PeekMessage(&msg, 0, NULL, NULL, PM_REMOVE))
  83.       {
  84.         TranslateMessage(&msg);
  85.         DispatchMessage(&msg);
  86.         if(msg.message ==WM_QUIT)
  87. {
  88. //WSASetLastError(WSAETIMEOUT);
  89. return -1;
  90. }
  91.       }
  92.       if(t2-t1 <timeout) continue;
  93.       else
  94.       {
  95.         if(prset) FD_CLR((UINT)sd,&rset);
  96.         if(pwset) FD_CLR((UINT)sd,&wset);
  97.         if(peset) FD_CLR((UINT)sd,&eset);
  98.         WSASetLastError(WSAEWOULDBLOCK);
  99.         return -10;
  100.       }
  101.     }
  103.     if(peset && FD_ISSET(sd, peset))
  104.     {
  105.       if(prset !=NULL) FD_CLR((UINT)sd,&rset);
  106.       if(pwset !=NULL) FD_CLR((UINT)sd,&wset);
  107.       if(peset !=NULL) FD_CLR((UINT)sd,&eset);
  108.       //err_no =WSAGetLastError();
  109.       /*
  110.       len =sizeof(errno);
  111.       getsockopt(sd, SOL_SOCKET, SO_ERROR, (char *)&errno, &len);
  112.       */
  113.   WSASetLastError(err_no);
  114.       return -1;
  115.     }
  116.     if((prset && FD_ISSET(sd, prset)) || (pwset && FD_ISSET(sd, pwset)))
  117.     {
  118.       //err_no =WSAGetLastError();
  119.       /*
  120.       len =sizeof(errno);
  121.       getsockopt(sd, SOL_SOCKET, SO_ERROR, (char *)&errno, &len);
  122.       */
  123.     }
  124.     if(prset !=NULL) FD_CLR((UINT)sd,&rset);
  125.     if(pwset !=NULL) FD_CLR((UINT)sd,&wset);
  126.     if(peset !=NULL) FD_CLR((UINT)sd,&eset);
  128.     //if(status <0)
  129.       //err_no =WSAGetLastError();
  130. WSASetLastError(err_no);
  131.     if(err_no ==WSAEINTR)
  132. {
  133. return WSAEINTR;
  134. }
  135.     if(err_no)
  136.     {
  137.       return -1;
  138.     }
  139.     else break;
  140.   }
  142.   return 0;
  143. }
  145. int tcp_bind(HWND hWnd, int port)
  146. {
  147. struct sockaddr_in addr;
  148.     char temp[200];
  149.     int sd;
  151.     sd =-1;
  152. if((sd =(int)socket(AF_INET, SOCK_STREAM, IPPROTO_TCP)) <0)
  153. {
  154. sprintf(temp, "socket failed! errno:%d", WSAGetLastError());
  155. return -1;
  156. }
  158. memset(&addr, 0, sizeof(addr));
  160. addr.sin_family =AF_INET;
  161. addr.sin_port =htons((unsigned short)port);
  162.     int l =1;
  163.     setsockopt(sd, SOL_SOCKET, SO_REUSEADDR, (char *)&l, sizeof(l));
  164.     /*setsockopt(sd, SOL_SOCKET, SO_REUSEPORT, (char *)&l, sizeof(l));*/
  165. if(bind(sd, (struct sockaddr *)&addr, sizeof(addr)) <0)
  166. {
  167.       sprintf(temp, "bind failed! errno:%d", WSAGetLastError());
  168.   closesocket(sd);
  169.       return -1;
  170. }
  171. if(hWnd && WSAAsyncSelect(sd, hWnd, WM_TCP, FD_ACCEPT) !=0)
  172.     {
  173.       sprintf(temp, "tcp_bind:WSAAsyncSelect failed!");
  174.       closesocket(sd);
  175.       return -2;
  176.     }
  177.     //sd_bind =sd;
  178.     listen(sd, 5);
  179.     return sd;
  180. }
  182. int tcp_accept(int sd, int timeout)
  183. {
  184.   int sd_acc =-1;
  185.   struct sockaddr_in sa;
  186.   int len;
  187.   /*unsigned long l;*/
  189.   if(tcp_status(sd, "rw", timeout) <0)
  190.     return -1;
  191.   len =sizeof(sa);
  192.   if((sd_acc =(int)accept(sd, (struct sockaddr *)&sa, &len)) <0)
  193.     return -1;
  194.   //sd_accept =sd_acc;
  195.   /*  l =1;
  196.     if(ioctlsocket(sd_acc, FIONBIO, &l) <0)
  197.     {
  198.       closesocket(sd);
  199.       return -1;
  200.     }*/
  201.   return sd_acc;
  202. }
  204. int tcp_connect(char *hostname, int port, int timeout, int f_noblock)
  205. {
  206.     struct hostent *hp;
  207.     struct sockaddr_in addr;
  208.     char temp[200];
  209.     unsigned long ul;
  210.     long l;
  211.     int sd =-1, ret;
  212.     time_t t1, t2;
  214.     sd =-1;
  215. memset(&addr, 0, sizeof(addr));
  216. if((sd =(int)socket(PF_INET, SOCK_STREAM, IPPROTO_TCP)) <0)
  217. {
  218. sprintf(temp, "socket failed! errno:%d", WSAGetLastError());
  219. return -1;
  220. }
  222. memset(&addr, 0, sizeof(addr));
  224.     ul =inet_addr(hostname);
  225.     if(ul ==0xffffffff)
  226.     {
  227.   if((hp =gethostbyname(hostname)) ==NULL)
  228.   {
  229.         sprintf(temp, "gethostbyname and inet_addr failed! errno:%d", WSAGetLastError());
  230.     closesocket(sd);
  231.         return -1;
  232.       }
  233.   memcpy(&addr.sin_addr, hp->h_addr, hp->h_length);
  234. }
  235.     else  addr.sin_addr.s_addr=ul;
  237. addr.sin_family =PF_INET;
  238. addr.sin_port =htons((unsigned short)port);
  239.     l =1;
  240.     if(f_noblock && ioctlsocket(sd, FIONBIO, (unsigned long *)&l) <0)
  241.     {
  242.       closesocket(sd);
  243.       return -1;
  244.     }
  245. /*if(setsockopt(sd, IPPROTO_TCP, TCP_NODELAY, (char *)&l, sizeof(l)) <0)
  246.     {
  247.       closesocket(sd);
  248.       return -1;
  249.     }*/
  250.     time(&t1);
  251.     while((ret =connect(sd, (struct sockaddr *)&addr, sizeof(addr))) !=0)
  252.     {
  253.       time(&t2);
  254.       if((t2 -t1) > timeout)
  255.       {
  256.         closesocket(sd);
  257.         return -1;
  258.       }
  259.       peek_message();
  260.       int err_no =WSAGetLastError();
  261.   if(ret ==SOCKET_ERROR)
  262.   {
  263.   if(err_no ==WSAEISCONN) break;
  264.       else if(err_no ==WSAEWOULDBLOCK /*|| err_no ==WSAEINPROGRESS*/ || err_no ==WSAEALREADY)
  265.              continue;
  266.           else
  267.       {
  268. //  char temp[256];
  269.     //GetErrString(temp, err_no);
  270. closesocket(sd);
  271.         return -1;
  272.       }
  273.   }
  274. }
  275.     if(tcp_status(sd, "we", timeout) <0)
  276.     {
  277.       sprintf(temp, "status:连接服务器失败!n检查服务器端程序是否运行n且主机地址是否%s, 端口是否%dn"
  278.                     "errno=%d",
  279.             hostname, port, WSAGetLastError());
  280.       closesocket(sd);
  281.       return -1;
  282.     }
  284.     //sd_connect =sd;
  285.     return sd;
  286. }
  288. void tcp_close(int sd)
  289. {
  290. tcp_disconnect(sd);
  291. }
  293. void tcp_disconnect(int sd)
  294. {
  295.   if(sd >0)
  296.   {
  297.     closesocket(sd);
  298.     //if(sd ==sd_connect) sd_connect =-1;
  299.     //if(sd ==sd_accept) sd_accept =-1;
  300.     //if(sd ==sd_bind) sd_bind =-1;
  301.   }
  302. }
  304. int tcp_send(int sd, char *buf, int len, int timeout)
  305. {
  306.   int len1, len_send =0;
  307.   time_t t1, t2;
  309.   len_send =0;
  310.   time(&t1);
  311.   t2 =t1;
  312.   //WriteLog("comm.log", "tcp_send:len=%d", len);
  313.   while(len_send <len)
  314.   {
  315.     if(t2-t1 >timeout)
  316.       return len_send;
  317.     if(tcp_status(sd, "w", timeout-(int)(t2-t1)) <0)
  318. {
  319.       return len_send;
  320. }
  321.     if((len1 =send(sd, &buf[len_send], len-len_send, 0)) <=0)
  322.     {
  323.       if(len1 ==SOCKET_ERROR && GetLastError() ==WSAEWOULDBLOCK)
  324.       {
  325.         time(&t2);
  326.         continue;
  327.       }
  328.       return len_send;
  329.     }
  330.     len_send +=len1;
  331.     time(&t2);
  332.   }
  334.   return len_send;
  335. }
  337. int tcp_recv(int sd, char *buf, int len, int timeout)
  338. {
  339.   int len1, len_recv;
  340.   time_t t2, t1;
  342.   len_recv =0;
  343.   time(&t1);
  344.   t2 =t1;
  346.   int err =0;
  347.   while(len_recv <len)
  348.   {
  349.     if(timeout >0)
  350. {
  351. if(t2 -t1 >timeout)
  352. {
  353. WSASetLastError(err);
  354. if(len_recv ==0) return -1;
  355. return len_recv;
  356. }
  357. if(tcp_status(sd, "r", timeout-(int)(t2-t1)) <0)
  358. {
  359. err =WSAGetLastError();
  360. //WriteStat("tcp_status err=%d", WSAGetLastError());
  361. WSASetLastError(err);
  362. if(len_recv ==0) return -1;
  363. return len_recv;
  364. }
  365. }
  367.      if((len1 =recv(sd, &buf[len_recv], len-len_recv, 0)) <=0)
  368.      {
  369. err =WSAGetLastError();
  370. //WriteStat("recv err=%d, len1=%d", err, len1);
  371.        if(timeout ==0) break;
  372.        if(len1 ==SOCKET_ERROR && err ==WSAEWOULDBLOCK)
  373.        {
  374.          time(&t2);
  375.          if(timeout <=0) break;
  376.  continue;
  377.        }
  378.    WSASetLastError(err);
  379.       if(len_recv ==0) return -1;
  380.        return len_recv;
  381.     }
  383.     len_recv +=len1;
  384.     if(timeout <=0) break;
  385.     time(&t2);
  386.   }
  387.   WSASetLastError(err);
  388.   if(len_recv ==0) return -1;
  389.   return len_recv;
  390. }
  392. int tcp_gethostnamebyip(char *ip, char *host)
  393. {
  394.   struct hostent *hp;
  395.   struct in_addr ul;
  397.   host[0] =0;
  398.   ul.S_un.S_addr =inet_addr(ip);
  399.   if(ul.S_un.S_addr ==0xFFFFFFFF) return -1; // ip error or ip is hostname
  401.   hp =gethostbyaddr((char *)&ul, 4, AF_INET);
  402.   if(hp ==NULL) return -1;  // can not get hostname
  403.   strcpy(host, hp->h_name);
  405.   return 0;
  406. }
  408. int get_local_ip(char *ip)
  409. {
  410.   struct hostent *hp;
  411.   char host[50], *p;
  413.   if(gethostname(host, sizeof(host)) <0) return -1;
  415.   hp =gethostbyname(host);
  416.   if(hp ==NULL) return -1;
  417.   p =(char *)hp->h_addr;
  418.   wsprintf(ip, "%d.%d.%d.%d", (int)p[0]&0xFF, (int)p[1]&0xFF, (int)p[2]&0xFF, (int)p[3]&0xFF);
  420.   return 0;
  421. }
  423. char *get_remote_ip(int sd, char *ip)
  424. {
  425.   struct sockaddr_in addr_in;
  426.   int len =sizeof(addr_in);
  427.   char *p1;
  429.   if(sd <0) return  NULL;
  430.   if(getpeername(sd, (struct sockaddr *)&addr_in, &len) <0)
  431.     return NULL;
  432.   p1 =(char *)&addr_in.sin_addr;
  433.   sprintf(ip, "%d.%d.%d.%d", ((int)p1[0]) &0xff, ((int)p1[1]) &0xff, (int)p1[2] &0xff, (int)p1[3]&0xff);
  434.  
  435.   return ip;
  436. }
  438. unsigned short tcp_htons(unsigned short s)
  439. {
  440.   return htons(s);
  441. }
  443. unsigned short tcp_ntohs(unsigned short s)
  444. {
  445.   return ntohs(s);
  446. }
  448. unsigned long int tcp_htonl(unsigned long int l)
  449. {
  450.   return htonl(l);
  451. }
  453. unsigned long int tcp_ntohl(unsigned long int l)
  454. {
  455.   return ntohl(l);
  456. }
  458. float tcp_htonf(float f)
  459. {
  460.   unsigned char *p, p0, p1;
  462.   if(htons(1) ==1) return f;
  463.   p =(unsigned char *)&f;
  464.   p0 =p[0];
  465.   p1 =p[1];
  466.   p[0] =p[3];
  467.   p[3] =p0;
  468.   p[1] =p[2];
  469.   p[2] =p1;
  471.   return f;
  472. }
  474. float tcp_ntohf(float f)
  475. {
  476.   unsigned char *p, p0, p1;
  478.   if(ntohs(1) ==1) return f;
  480.   p =(unsigned char *)&f;
  481.   p0 =p[0];
  482.   p1 =p[1];
  483.   p[0] =p[3];
  484.   p[3] =p0;
  485.   p[1] =p[2];
  486.   p[2] =p1;
  487.   return f;
  488. }
  490. double tcp_htond(double d)
  491. {
  492.   unsigned char *p, p0, p1, p2, p3;
  494.   if(htons(1) ==1) return d;
  496.   p =(unsigned char *)&d;
  497.   p0 =p[0];
  498.   p1 =p[1];
  499.   p2 =p[2];
  500.   p3 =p[3];
  501.   p[0] =p[7];
  502.   p[7] =p0;
  503.   p[1] =p[6];
  504.   p[6] =p1;
  505.   p[2] =p[5];
  506.   p[5] =p2;
  507.   p[3] =p[4];
  508.   p[4] =p3;
  510.   return d;
  511. }
  513. double tcp_ntohd(double d)
  514. {
  515.   unsigned char *p, p0, p1, p2, p3;
  517.   if(ntohs(1) ==1) return d;
  519.   p =(unsigned char *)&d;
  520.   p0 =p[0];
  521.   p1 =p[1];
  522.   p2 =p[2];
  523.   p3 =p[3];
  524.   p[0] =p[7];
  525.   p[7] =p0;
  526.   p[1] =p[6];
  527.   p[6] =p1;
  528.   p[2] =p[5];
  529.   p[5] =p2;
  530.   p[3] =p[4];
  531.   p[4] =p3;
  533.   return d;
  534. }
  536. unsigned __int64 tcp_htonh(unsigned __int64 d)
  537. {
  538.   unsigned char *p, p0, p1, p2, p3;
  540.   if(htons(1) ==1) return d;
  542.   p =(unsigned char *)&d;
  543.   p0 =p[0];
  544.   p1 =p[1];
  545.   p2 =p[2];
  546.   p3 =p[3];
  547.   p[0] =p[7];
  548.   p[7] =p0;
  549.   p[1] =p[6];
  550.   p[6] =p1;
  551.   p[2] =p[5];
  552.   p[5] =p2;
  553.   p[3] =p[4];
  554.   p[4] =p3;
  556.   return d;
  557. }
  559. unsigned __int64 tcp_ntohh(unsigned __int64 d)
  560. {
  561.   unsigned char *p, p0, p1, p2, p3;
  563.   if(ntohs(1) ==1) return d;
  565.   p =(unsigned char *)&d;
  566.   p0 =p[0];
  567.   p1 =p[1];
  568.   p2 =p[2];
  569.   p3 =p[3];
  570.   p[0] =p[7];
  571.   p[7] =p0;
  572.   p[1] =p[6];
  573.   p[6] =p1;
  574.   p[2] =p[5];
  575.   p[5] =p2;
  576.   p[3] =p[4];
  577.   p[4] =p3;
  579.   return d;
  580. }
  582. void peek_message(void)
  583. {
  584.   MSG msg;
  586.   if(PeekMessage(&msg, NULL, 0, 0, PM_REMOVE))
  587.   {
  588.     TranslateMessage(&msg);
  589.     DispatchMessage(&msg);
  590.   }
  591. }