经常用到的几个自定义函数：
1、开启监听的函数
http://linux.chinaunix.net/bbs/viewthread.php?tid=786283&extra=
view plain
copy to clipboard
print
?
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; }
/*----------------------源代码代码开始--------------------------------------------*/
int
OpenSCPServer(int port, int total, int sendbuflen, int recvbuflen, int blockORnot, int reuseORnot) {
/*************************关于本函数************************************
*function_name: OpenSCPServer
*参数说明：port整数型监听端口号，total整数型监听个数，sendbuflen整数型发送缓冲区大小
* recvbuflen整数型接收缓冲区大小，blockORnot整数型是否阻塞，reuseORnot整数型是否端口重用
*purpose: 用来建立一个tcp服务端socket
*tidied by: zhoulifa(zhoulifa@163.com) 周立发(http://zhoulifa.9999mb.com)
Linux爱好者 Linux知识传播者 SOHO族 开发者 最擅长C语言
*date time:2006-07-05 20:00:00
*Note: 任何人可以任意复制代码并运用这些文档，当然包括你的商业用途
* 但请遵循GPL
*Thanks to: Paul Sheer 感谢Paul Sheer在select_tut的man手册里提供了这份源代码
*Hope:希望越来越多的人贡献自己的力量，为科学技术发展出力
*Note:要使用此函数需要自定义一个全局变量char errorMessage[1024];并包含GetCurrentTime.h头文件
*********************************************************************/
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);/*block or not*/
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、连接服务器的函数
view plain
copy to clipboard
print
?
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; }
/*----------------------源代码代码开始--------------------------------------------*/
int
ConnectSCPServer(char * serverip, int serverport, int blockORnot) {
/*************************关于本函数************************************
*function_name: ConnectSCPServer
*参数说明：serverip服务器IP地址或主机名，serverport服务器端口，blockORnot整数型是否阻塞
*purpose: 用来建立一个tcp客户端socket
*tidied by: zhoulifa(zhoulifa@163.com) 周立发(http://zhoulifa.9999mb.com)
Linux爱好者 Linux知识传播者 SOHO族 开发者 最擅长C语言
*date time:2006-07-05 20:40:00
*Note: 任何人可以任意复制代码并运用这些文档，当然包括你的商业用途
* 但请遵循GPL
*Thanks to: Paul Sheer 感谢Paul Sheer在select_tut的man手册里提供了这份源代码
*Hope:希望越来越多的人贡献自己的力量，为科学技术发展出力
*Note:要使用此函数需要自定义一个全局变量char errorMessage[1024];并包含自己编写的GetCurrentTime.h头文件
*********************************************************************/
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); //block or not
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
view plain
copy to clipboard
print
?
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; }
/*----------------------源代码代码开始--------------------------------------------*/
int
Send(int sock, char * buf, size_t size, int flag, int timeout) {
/*************************关于本函数************************************
*function_name: Send
*参数说明：sock整数型socket，buf待发送的内容，size要发送的大小，flag发送选项，timeout超时时间值
*purpose: 用来通过一个socket在指定时间内发送数据
*tidied by: zhoulifa(zhoulifa@163.com) 周立发(http://zhoulifa.9999mb.com)
Linux爱好者 Linux知识传播者 SOHO族 开发者 最擅长C语言
*date time:2006-07-05 20:58:00
*Note: 任何人可以任意复制代码并运用这些文档，当然包括你的商业用途
* 但请遵循GPL
*Thanks to: Paul Sheer 感谢Paul Sheer在select_tut的man手册里提供了这份源代码
*Hope:希望越来越多的人贡献自己的力量，为科学技术发展出力
*Note:要使用此函数需要自定义一个全局变量char errorMessage[1024];并包含自己编写的GetCurrentTime.h头文件
*********************************************************************/
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
view plain
copy to clipboard
print
?
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; }
