Linux常用開發伺服器的代碼[Linux zhoulifa ]

經常用到的幾個自定義函數：

1、開啟監聽的函數

http://linux.chinaunix.net/bbs/viewthread.php?tid=786283&extra=

/*************************關于本函數************************************  

*function_name: OpenSCPServer  

*參數說明：port整數型監聽端口号，total整數型監聽個數，sendbuflen整數型發送緩沖區大小  

*          recvbuflen整數型接收緩沖區大小，blockORnot整數型是否阻塞，reuseORnot整數型是否端口重用  

*purpose: 用來建立一個tcp服務端socket  

*tidied by: zhoulifa([email protected]) 周立發(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  

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));   

        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));   

        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);   

        sprintf(errorMessage, "OpenTCPServer listen() error! return:%d, errno=%d, errortext:'%s' %s", ret, errno, strerror(errno), GetCurrentTime(0, 0));   

        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;   

}   

/*************************關于本函數************************************

*function_name: OpenSCPServer

*參數說明：port整數型監聽端口号，total整數型監聽個數，sendbuflen整數型發送緩沖區大小

*          recvbuflen整數型接收緩沖區大小，blockORnot整數型是否阻塞，reuseORnot整數型是否端口重用

*purpose: 用來建立一個tcp服務端socket

*tidied by: zhoulifa([email protected]) 周立發(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

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));

        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));

        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);

        sprintf(errorMessage, "OpenTCPServer listen() error! return:%d, errno=%d, errortext:'%s' %s", ret, errno, strerror(errno), GetCurrentTime(0, 0));

        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、連接配接伺服器的函數

*function_name: ConnectSCPServer  

*參數說明：serverip伺服器IP位址或主機名，serverport伺服器端口，blockORnot整數型是否阻塞  

*purpose: 用來建立一個tcp用戶端socket  

*date time:2006-07-05 20:40:00  

*Note:要使用此函數需要自定義一個全局變量char errorMessage[1024];并包含自己編寫的GetCurrentTime.h頭檔案  

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));   

    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));   

    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 serversock;   

*function_name: ConnectSCPServer

*參數說明：serverip伺服器IP位址或主機名，serverport伺服器端口，blockORnot整數型是否阻塞

*purpose: 用來建立一個tcp用戶端socket

*date time:2006-07-05 20:40:00

*Note:要使用此函數需要自定義一個全局變量char errorMessage[1024];并包含自己編寫的GetCurrentTime.h頭檔案

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));

    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));

    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 serversock;

3、發送資料函數Send

*function_name: Send  

*參數說明：sock整數型socket，buf待發送的内容，size要發送的大小，flag發送選項，timeout逾時時間值  

*purpose: 用來通過一個socket在指定時間内發送資料  

*date time:2006-07-05 20:58:00  

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));   

    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);   

    if(!(FD_ISSET(sock, &writefds)))    {   

        sprintf(errorMessage, "Send socket:%d not in writefds! %s", sock, GetCurrentTime(0, 0));   

    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;   

*function_name: Send

*參數說明：sock整數型socket，buf待發送的内容，size要發送的大小，flag發送選項，timeout逾時時間值

*purpose: 用來通過一個socket在指定時間内發送資料

*date time:2006-07-05 20:58:00

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));

    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);

    if(!(FD_ISSET(sock, &writefds)))    {

        sprintf(errorMessage, "Send socket:%d not in writefds! %s", sock, GetCurrentTime(0, 0));

    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

*function_name: Recv  

*參數說明：sock整數型socket，buf接收資料的緩沖區，size要接收資料的大小，flag接收選項，timeout逾時時間值  

*purpose: 用來從一個socket在指定時間内讀取資料  

*date time:2006-07-05 21:10:00  

*********************************************************************/ 

Recv(int sock, char * buf, size_t size, int flag, int timeout)    

{   

    fd_set readfds;   

    FD_ZERO(&readfds);   

        FD_SET(sock, &readfds);   

        sprintf(errorMessage, "Recv socket:%d error! return:-2 %s", sock, GetCurrentTime(0, 0));   

    ret = select(maxfds + 1, &readfds, NULL, NULL, &tival);   

        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));   

    if(!(FD_ISSET(sock, &readfds)))    {   

        sprintf(errorMessage, "Recv socket:%d not in readfds! %s", sock, GetCurrentTime(0, 0));   

        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));   

    sprintf(errorMessage, "Recv socket:%d recv() OK! %d/%d bytes received! %s", sock, i, size, GetCurrentTime(0, 0));   

}