3.txt

linux环境下的TCP网络服务器端实现原代码

  


用gcc tcpforwardport.c -o MyProxy编译此程序后运行效果如下： 

    当有用户访问本机的8000端口时，MyProxy程序将把此请求转发到172.16.100.218主机的80端口，即实现了一个http代理。 

关于select函数： 
其函数原型为： 
int select(int n, fd_set *readfds, fd_set *writefds, fd_set *exceptfds, struct timeval *timeout); 
此函数的功能是由内核检测在timeout时间内，是否有readfds，writefds，exceptfds三个句柄集(file descriptors)里的某个句柄（file descriptor)的状态符合寻求，即readfds句柄集里有句柄可读或writefds句柄集里有可写或exceptfds句柄集里有例外发生，任何一个有变化函数就立即返回，返回值为timeout发生状态变化的句柄个数。 
n是所有readfds，writefds，exceptfds三个句柄集(file descriptors)里编号最大值加1。比如：要检测两个socket句柄fd1和fd2在timeout时间内是否分别可读和可写就可以这样： 
先把两个句柄集(file descriptors)清零： 
        FD_ZERO (&readfds); 
        FD_ZERO (&writefds); 
然后把fd1加入读检测集： 
        FD_SET (fd1, &readfds); 
然后把fd2加入写检测集： 
        FD_SET (fd2, &writefds); 
再给timeout设置值，timeout是这样的一个结构： 
              struct timeval { 
                  long    tv_sec;         /* seconds */ 
                  long    tv_usec;        /* microseconds */ 
              }; 
你可以这样赋值： 
        timeout.tv_sec=1; 
        timeout.tv_uec=0; 
表示检测在1秒钟内是否有句柄状态发生变化。 
如果有句柄发生变化，就可以用FD_ISSET检测各个句柄，比如： 
                FD_ISSET (fd1, &readfds);//检测是否fd1变成可读的了 
                FD_ISSET (fd2, &writefds);//检测是否fd2变成可写的了 
示意程序代码如下： 
/*----------------------示意代码开始--------------------------------------------*/ 
    fd1 = socket();//创建一个socket 
    fd2 = socket();//创建一个socket 
    while(1)  { 
        FD_ZERO (&readfds); 
        FD_ZERO (&writefds); 
        FD_SET (fd1, &readfds); 
        FD_SET (fd2, &writefds); 
        timeout.tv_sec=1; 
        timeout.tv_uec=0; 
        ret = select(fd1>fd2?(fd1+1):(fd2+1), &readfds, &writefds, NULL, &timeout); 
        if(ret < 0) {printf("系统错误，select出错，错误代码：%d, 错误信息：%s", errno, strerror(errno));} 
        else if(ret == 0) {printf("select超时返回，没有任何句柄状态发生变化！");} 
        //有句柄状态发生了变化 
        if(FD_ISSET(fd1, &readfds)) { 
            fd1有数据可读; 
            fd1里的数据被读出来; 
        } 
        if(FD_ISSET(fd2, &writefds)) { 
            fd2可写; 
            fd2里发送数据给对方; 
        } 
    } 
/*----------------------示意代码结束--------------------------------------------*/ 

经常用到的几个自定义函数： 
1、开启监听的函数 

/*----------------------源代码代码开始--------------------------------------------*/ 
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([e-mail]) 周立发(http://zhoulifa.bokee.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、连接服务器的函数 

/*----------------------源代码代码开始--------------------------------------------*/ 
int 
ConnectSCPServer(char * serverip, int serverport, int blockORnot)    { 
/*************************关于本函数************************************ 
*function_name: ConnectSCPServer 
*参数说明：serverip服务器IP地址或主机名，serverport服务器端口，blockORnot整数型是否阻塞 
*purpose: 用来建立一个tcp客户端socket 
*tidied by: zhoulifa([e-mail]) 周立发(http://zhoulifa.bokee.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; 
    }