tcp.cpp
资源名称:update.zip [点击查看]
上传用户:wangying89
上传日期:2007-01-07
资源大小:61k
文件大小:7k
源码类别:
Ftp客户端
开发平台:
Visual C++
- #include <windows.h>
- #include <winsock.h>
- #include <stdio.h>
- #include <stdlib.h>
- #include <time.h>
- #include "tcp.h"
- int sd_connect =-1, sd_bind =-1, sd_accept =-1;
- int wsa_ok =0;
- int tcp_init()
- {
- WSAData wsa;
- sd_bind =sd_connect =sd_accept =-1;
- if(WSAStartup(MAKEWORD(1, 1), &wsa) !=0)
- return -1;
- wsa_ok =1;
- return 0;
- }
- int tcp_exit()
- {
- if(sd_connect >=0) closesocket(sd_connect);
- if(sd_accept >=0) closesocket(sd_accept);
- if(sd_bind >=0) closesocket(sd_bind);
- if(wsa_ok) WSACleanup();
- sd_connect =sd_accept =sd_bind =-1;
- return 0;
- }
- int tcp_status(int sd, char *type, int timeout)
- {
- fd_set rset, wset, eset;
- fd_set FAR *prset =NULL, *pwset =NULL, *peset =NULL;
- struct timeval tval;
- int i, status, err_no =0;
- time_t t1, t2;
- MSG msg;
- tval.tv_sec =0;
- tval.tv_usec =0;
- time(&t1);
- t2 =t1;
- while(t2-t1 < timeout)
- {
- FD_ZERO(&rset);
- FD_ZERO(&wset);
- FD_ZERO(&eset);
- for(i =0; i<(int)strlen(type); i++)
- {
- if(type[i] =='r') { FD_SET(sd, &rset); prset =&rset; }
- if(type[i] =='w') { FD_SET(sd, &wset); pwset =&wset; }
- if(type[i] =='e') { FD_SET(sd, &eset); peset =&eset; }
- }
- status =select(sd+1, prset, pwset, peset, &tval);
- time(&t2);
- if(status ==0)
- {
- if(PeekMessage(&msg, 0, NULL, NULL, PM_REMOVE))
- {
- TranslateMessage(&msg);
- DispatchMessage(&msg);
- if(msg.message ==WM_QUIT) return -1;
- }
- if(t2-t1 <timeout) continue;
- else
- {
- if(prset) FD_CLR((UINT)sd,&rset);
- if(pwset) FD_CLR((UINT)sd,&wset);
- if(peset) FD_CLR((UINT)sd,&eset);
- SetLastError(WSAETIMEDOUT);
- return -10;
- }
- }
- if(peset && FD_ISSET(sd, peset))
- {
- if(prset !=NULL) FD_CLR((UINT)sd,&rset);
- if(pwset !=NULL) FD_CLR((UINT)sd,&wset);
- if(peset !=NULL) FD_CLR((UINT)sd,&eset);
- err_no =WSAGetLastError();
- /*
- len =sizeof(errno);
- getsockopt(sd, SOL_SOCKET, SO_ERROR, (char *)&errno, &len);
- */
- return -1;
- }
- if((prset && FD_ISSET(sd, prset)) || (pwset && FD_ISSET(sd, pwset)))
- {
- err_no =WSAGetLastError();
- /*
- len =sizeof(errno);
- getsockopt(sd, SOL_SOCKET, SO_ERROR, (char *)&errno, &len);
- */
- }
- if(prset !=NULL) FD_CLR((UINT)sd,&rset);
- if(pwset !=NULL) FD_CLR((UINT)sd,&wset);
- if(peset !=NULL) FD_CLR((UINT)sd,&eset);
- if(status <0)
- err_no =WSAGetLastError();
- if(err_no ==WSAEINTR) return WSAEINTR;
- if(err_no)
- {
- return -1;
- }
- else break;
- }
- return 0;
- }
- int tcp_bind(HWND hWnd, int port)
- {
- struct sockaddr_in addr;
- char temp[200];
- int sd;
- sd =-1;
- if((sd =socket(AF_INET, SOCK_STREAM, IPPROTO_TCP)) <0)
- {
- sprintf(temp, "socket failed! errno:%d", WSAGetLastError());
- return -1;
- }
- memset(&addr, 0, sizeof(addr));
- addr.sin_family =AF_INET;
- addr.sin_port =htons((unsigned short)port);
- int l =1;
- setsockopt(sd, SOL_SOCKET, SO_REUSEADDR, (char *)&l, sizeof(l));
- /*setsockopt(sd, SOL_SOCKET, SO_REUSEPORT, (char *)&l, sizeof(l));*/
- if(bind(sd, (struct sockaddr *)&addr, sizeof(addr)) <0)
- {
- sprintf(temp, "bind failed! errno:%d", WSAGetLastError());
- closesocket(sd);
- return -1;
- }
- if(hWnd && WSAAsyncSelect(sd, hWnd, WM_TCP, FD_ACCEPT) !=0)
- {
- sprintf(temp, "tcp_bind:WSAAsyncSelect failed!");
- closesocket(sd);
- return -2;
- }
- sd_bind =sd;
- listen(sd_bind, 5);
- return sd;
- }
- int tcp_accept(int sd, int timeout)
- {
- int sd_acc =-1;
- struct sockaddr_in sa;
- int len;
- /*unsigned long l;*/
- if(tcp_status(sd, "rw", timeout) <0)
- return -1;
- len =sizeof(sa);
- if((sd_acc =accept(sd, (struct sockaddr *)&sa, &len)) <0)
- return -1;
- sd_accept =sd_acc;
- /* l =1;
- if(ioctlsocket(sd_acc, FIONBIO, &l) <0)
- {
- closesocket(sd);
- return -1;
- }*/
- return sd_acc;
- }
- int tcp_connect(char *hostname, int port, int timeout)
- {
- struct hostent *hp;
- struct sockaddr_in addr;
- char temp[200];
- unsigned long ul;
- long l;
- int sd =-1;
- sd =-1;
- if((sd =socket(AF_INET, SOCK_STREAM, IPPROTO_TCP)) <0)
- {
- sprintf(temp, "socket failed! errno:%d", WSAGetLastError());
- return -1;
- }
- memset(&addr, 0, sizeof(addr));
- ul =inet_addr(hostname);
- if(ul ==0xffffffff)
- {
- if((hp =gethostbyname(hostname)) ==NULL)
- {
- sprintf(temp, "gethostbyname and inet_addr failed! errno:%d", WSAGetLastError());
- closesocket(sd);
- return -1;
- }
- memcpy(&addr.sin_addr, hp->h_addr, hp->h_length);
- }
- else addr.sin_addr.s_addr=ul;
- addr.sin_family =AF_INET;
- addr.sin_port =htons((unsigned short)port);
- l =1;
- /*if(ioctlsocket(sd, FIONBIO, (unsigned long *)&l) <0)
- {
- closesocket(sd);
- return -1;
- }*/
- if(setsockopt(sd, IPPROTO_TCP, TCP_NODELAY, (char *)&l, sizeof(l)) <0)
- {
- closesocket(sd);
- return -1;
- }
- while(connect(sd, (struct sockaddr *)&addr, sizeof(addr)) !=0)
- {
- int err_no =WSAGetLastError();
- if(err_no ==WSAEINTR) continue;
- if(err_no ==WSAEINPROGRESS) break;
- sprintf(temp, "connect:连接服务器失败!n检查服务器端程序是否运行n且主机地址是否%s, 端口是否%dn errno=%d",
- hostname, port, err_no);
- closesocket(sd);
- return -1;
- }
- if(tcp_status(sd, "w", timeout) <0)
- {
- sprintf(temp, "status:连接服务器失败!n检查服务器端程序是否运行n且主机地址是否%s, 端口是否%dn"
- "errno=%d",
- hostname, port, WSAGetLastError());
- closesocket(sd);
- return -1;
- }
- sd_connect =sd;
- return sd;
- }
- void tcp_close(int sd)
- {
- tcp_disconnect(sd);
- }
- void tcp_disconnect(int sd)
- {
- if(sd >0)
- {
- closesocket(sd);
- if(sd ==sd_connect) sd_connect =-1;
- if(sd ==sd_accept) sd_accept =-1;
- if(sd ==sd_bind) sd_bind =-1;
- }
- }
- int tcp_send(int sd, char *buf, int len, int timeout)
- {
- int len1, len_send =0;
- time_t t1, t2;
- len_send =0;
- time(&t1);
- t2 =t1;
- while(len_send <len)
- {
- if(t2-t1 >timeout)
- return len_send;
- if(tcp_status(sd, "w", timeout-(t2-t1)) <0)
- return len_send;
- if((len1 =send(sd, &buf[len_send], len-len_send, 0)) <=0)
- {
- /*if(GetLastError() ==WSAEWOULDBLOCK)
- {
- time(&t2);
- continue;
- }*/
- return len_send;
- }
- len_send +=len1;
- time(&t2);
- }
- return len_send;
- }
- int tcp_recv(int sd, char *buf, int len, int timeout)
- {
- int len1, len_recv;
- time_t t2, t1;
- len_recv =0;
- time(&t1);
- t2 =t1;
- while(len_recv <len)
- {
- if(t2 -t1 >timeout)
- return len_recv;
- if(tcp_status(sd, "r", timeout-(t2-t1)) <0)
- return len_recv;
- if((len1 =recv(sd, &buf[len_recv], len-len_recv, 0)) <=0)
- {
- /*if(timeout ==0) break;
- if(GetLastError() ==WSAEWOULDBLOCK)
- {
- time(&t2);
- continue;
- }*/
- return len_recv;
- }
- len_recv +=len1;
- time(&t2);
- }
- return len_recv;
- }