经常用到的几个自定义函数:
1、开启监听的函数
http://linux.chinaunix.net/bbs/viewthread.php?tid=786283&extra=
-
- int
- OpenSCPServer(int port, int total, int sendbuflen, int recvbuflen, int blockORnot, int reuseORnot) {
-
-
-
-
-
-
-
-
-
-
-
-
-
-
- int sockfd = 0, ret = 0, opt = 0, flags=1;
- struct sockaddr_in laddr;
-
- ret = sockfd = socket(PF_INET, SOCK_STREAM, 0);
- if(ret < 0) {
- sprintf(errorMessage, "OpenTCPServer socket() error! return:%d, errno=%d, errortext:'%s' %s", ret, errno, strerror(errno), GetCurrentTime(0, 0));
- return -1;
- }
-
- ret = setsockopt(sockfd, SOL_SOCKET, SO_REUSEADDR, &reuseORnot, sizeof(int));
- if(ret < 0) {
- sprintf(errorMessage, "OpenTCPServer setsockopt() reuse error! return:%d, errno=%d, errortext:'%s' %s", ret, errno, strerror(errno), GetCurrentTime(0, 0));
- return -2;
- }
-
- ret = setsockopt(sockfd, SOL_SOCKET, SO_RCVBUF, &recvbuflen, sizeof(int));
- if ( ret < 0) {
- sprintf(errorMessage, "OpenTCPServer setsockopt() recvbuf error! return:%d, errno=%d, errortext:'%s' %s", ret, errno, strerror(errno), GetCurrentTime(0, 0));
- return -3;
- }
-
- ret = setsockopt(sockfd, SOL_SOCKET, SO_SNDBUF, &sendbuflen, sizeof(int));
- if (ret < 0) {
- sprintf(errorMessage, "OpenTCPServer setsockopt() sendbuf error! return:%d, errno=%d, errortext:'%s' %s", ret, errno, strerror(errno), GetCurrentTime(0, 0));
- return -4;
- }
-
- ioctl(sockfd,FIONBIO,&blockORnot);
-
- laddr.sin_family = PF_INET;
- laddr.sin_port = htons(port);
- laddr.sin_addr.s_addr = INADDR_ANY;
- bzero(&(laddr.sin_zero), 8);
-
- ret = bind(sockfd, (struct sockaddr *)&laddr, sizeof(struct sockaddr));
- if(ret < 0) {
- sprintf(errorMessage, "OpenTCPServer bind() error! return:%d, errno=%d, errortext:'%s' %s", ret, errno, strerror(errno), GetCurrentTime(0, 0));
- close(sockfd);
- return -5;
- }
- ret = listen(sockfd, total);
- if(ret < 0) {
- sprintf(errorMessage, "OpenTCPServer listen() error! return:%d, errno=%d, errortext:'%s' %s", ret, errno, strerror(errno), GetCurrentTime(0, 0));
- close(sockfd);
- return -6;
- }
- sprintf(errorMessage, "OpenTCPServer opened on port.%d(%d) OK, socket(%d), buf(%d:%d)! %s", port, total, sockfd, sendbuflen, recvbuflen, GetCurrentTime(0, 0));
- return sockfd;
- }
-
2、连接服务器的函数
-
- int
- ConnectSCPServer(char * serverip, int serverport, int blockORnot) {
-
-
-
-
-
-
-
-
-
-
-
-
-
- int serversock = 0, ret = 0;
- unsigned long addr;
- struct sockaddr_in sin;
- struct hostent *he;
-
- if((he=gethostbyname(serverip))== 0) {
- sprintf(errorMessage, "ConnectSCPServer IP address '%s' error! return:-1 %s", serverip, GetCurrentTime(0, 0));
- return -1;
- }
-
- serversock = socket(PF_INET, SOCK_STREAM, 0);
- if(serversock == -1) {
- sprintf(errorMessage, "ConnectSCPServer socket() error! return:-2, errno=%d, errortext:'%s' %s", errno, strerror(errno), GetCurrentTime(0, 0));
- return -2;
- }
-
- ioctl(serversock, FIONBIO, &blockORnot);
-
- memset((char*)&sin, 0, sizeof(struct sockaddr_in));
- sin.sin_family = PF_INET;
- sin.sin_port = htons(serverport);
- sin.sin_addr = *((struct in_addr *)he->h_addr);
-
- ret = connect(serversock, (struct sockaddr *)&sin, sizeof(sin));
-
- if(ret == -1) {
- sprintf(errorMessage, "ConnectSCPServer connect() error! return:-3, errno=%d, errortext:'%s' %s", errno, strerror(errno), GetCurrentTime(0, 0));
- close(serversock);
- return -3;
- }
-
- return serversock;
- }
-
3、发送数据函数Send
-
- int
- Send(int sock, char * buf, size_t size, int flag, int timeout) {
-
-
-
-
-
-
-
-
-
-
-
-
-
- int i = 0, ret = 0, intretry = 0;
-
- struct timeval tival;
- fd_set writefds;
- int maxfds = 0;
-
- tival.tv_sec = timeout;
- tival.tv_usec = 0;
-
- FD_ZERO(&writefds);
-
- if(sock > 0) {
- FD_SET(sock, &writefds);
- maxfds=((sock > maxfds)?sock:maxfds);
- }
- else {
- sprintf(errorMessage, "Send socket:%d error! return:-2 %s", sock, GetCurrentTime(0, 0));
- return -2;
- }
-
- ret = select(maxfds + 1, NULL, &writefds, NULL, &tival);
- if(ret <= 0) {
- if(ret < 0) sprintf(errorMessage, "Send socket:%d select() error! return:%d, errno=%d, errortext:'%s' %s", sock, ret, errno, strerror(errno), GetCurrentTime(0, 0));
- else sprintf(errorMessage, "Send socket:%d select timeout(%d)! %s", sock, timeout, GetCurrentTime(0, 0));
- close(sock);
- return -3;
- }
- if(!(FD_ISSET(sock, &writefds))) {
- sprintf(errorMessage, "Send socket:%d not in writefds! %s", sock, GetCurrentTime(0, 0));
- close(sock);
- return -4;
- }
-
- while(i < size) {
- ret = send(sock, buf + i, size - i, flag);
- if(ret <= 0) {
- sprintf(errorMessage, "Send socket:%d send() error! return:%d, errno=%d, errortext:'%s' %s", sock, ret, errno, strerror(errno), GetCurrentTime(0, 0));
-
- if (EINTR == errno)
- if(intretry < 10) {intretry++;continue;}
- else sprintf(errorMessage, "Send socket:%d send() error!EINTR 10 times! %s", sock, GetCurrentTime(0, 0));
-
- close(sock);
- return -1;
- }
- else i += ret;
- }
- sprintf(errorMessage, "Send socket:%d send() OK! %d/%d bytes sent! %s", sock, i, size, GetCurrentTime(0, 0));
- return i;
- }
-
-
接收数据函数Recv
-
- int
- Recv(int sock, char * buf, size_t size, int flag, int timeout) {
-
-
-
-
-
-
-
-
-
-
-
-
-
- int i = 0, ret = 0, intretry = 0;
-
- struct timeval tival;
- fd_set readfds;
- int maxfds = 0;
-
- tival.tv_sec = timeout;
- tival.tv_usec = 0;
-
- FD_ZERO(&readfds);
-
- if(sock > 0) {
- FD_SET(sock, &readfds);
- maxfds=((sock > maxfds)?sock:maxfds);
- }
- else {
- sprintf(errorMessage, "Recv socket:%d error! return:-2 %s", sock, GetCurrentTime(0, 0));
- return -2;
- }
-
- ret = select(maxfds + 1, &readfds, NULL, NULL, &tival);
- if(ret <= 0) {
- if(ret < 0) sprintf(errorMessage, "Recv socket:%d select() error! return:%d, errno=%d, errortext:'%s' %s", sock, ret, errno, strerror(errno), GetCurrentTime(0, 0));
- else sprintf(errorMessage, "Recv socket:%d select timeout(%d)! %s", sock, timeout, GetCurrentTime(0, 0));
- close(sock);
- return -3;
- }
- if(!(FD_ISSET(sock, &readfds))) {
- sprintf(errorMessage, "Recv socket:%d not in readfds! %s", sock, GetCurrentTime(0, 0));
- close(sock);
- return -4;
- }
- while(i < size) {
- ret = recv(sock, buf + i, size - i, flag);
- if(ret <= 0){
- sprintf(errorMessage, "Recv socket:%d recv() error! return:%d, errno=%d, errortext:'%s' %s", sock, ret, errno, strerror(errno), GetCurrentTime(0, 0));
- if(errno == EINTR)
- if(intretry < 10) {intretry++;continue;}
- else sprintf(errorMessage, "Recv socket:%d recv() error! EINTR 10 times! %s", sock, GetCurrentTime(0, 0));
- close(sock);
- return -1;
- }
- else i += ret;
- }
- sprintf(errorMessage, "Recv socket:%d recv() OK! %d/%d bytes received! %s", sock, i, size, GetCurrentTime(0, 0));
- return i;
- }
Trackback: http://tb.blog.csdn.net/TrackBack.aspx?PostId=1807179
发表于 @ 2007年10月10日 01:07:00|评论(loading...)|编辑