C語言學習之Linux下TCP服務器與客戶端的實現

客戶端代碼如下：

#include <stdlib.h>
#include <stdio.h>
#include <errno.h>
#include <string.h>
#include <netdb.h>
#include <sys/types.h>
#include <netinet/in.h>
#include <sys/socket.h>

#define portnumber 3333

int main(int argc, char *argv[])
{
    int sockfd;
    char buffer[1024];
    struct sockaddr_in server_addr;
    struct hostent *host;

    /* 使用hostname查詢host 名字 */
    if(argc!=2)
    {
        fprintf(stderr,"Usage:%s hostname \a\n",argv[0]);
        exit(1);
    }

    if((host=gethostbyname(argv[1]))==NULL)
    {
        fprintf(stderr,"Gethostname error\n");
        exit(1);
    }

    /* 客戶程序開始建立 sockfd描述符 */
    if((sockfd=socket(AF_INET,SOCK_STREAM,0))==-1) // AF_INET:Internet;SOCK_STREAM:TCP
    {
        fprintf(stderr,"Socket Error:%s\a\n",strerror(errno));
        exit(1);
    }

    /* 客戶程序填充服務端的資料 */
    bzero(&server_addr,sizeof(server_addr)); // 初始化,置0
    server_addr.sin_family=AF_INET;          // IPV4
    server_addr.sin_port=htons(portnumber);  // (將本機器上的short數據轉化爲網絡上的short數據)端口號
    server_addr.sin_addr=*((struct in_addr *)host->h_addr); // IP地址

    /* 客戶程序發起連接請求 */
    if(connect(sockfd,(struct sockaddr *)(&server_addr),sizeof(struct sockaddr))==-1)
    {
        fprintf(stderr,"Connect Error:%s\a\n",strerror(errno));
        exit(1);
    }

    /* 連接成功了 */
    printf("Please input char:\n");

    /* 發送數據 */
    fgets(buffer,1024,stdin);
    send(sockfd,buffer,strlen(buffer),0);

    /* 結束通訊 */
    close(sockfd);
    exit(0);
}

服務器代碼如下：

#include <stdlib.h>
#include <stdio.h>
#include <errno.h>
#include <string.h>
#include <netdb.h>
#include <sys/types.h>
#include <netinet/in.h>
#include <sys/socket.h>

#define portnumber 3333//服務器的端口號和客戶端的端口號一定相等

int main(int argc, char *argv[])
{
    int sockfd,new_fd;
    struct sockaddr_in server_addr;
    struct sockaddr_in client_addr;
    int sin_size;
    int nbytes;
    char buffer[1024];
 

    /* 服務器端開始建立sockfd描述符 */
    if((sockfd=socket(AF_INET,SOCK_STREAM,0))==-1) // AF_INET:IPV4;SOCK_STREAM:TCP
    {
        fprintf(stderr,"Socket error:%s\n\a",strerror(errno));
        exit(1);
    }

    /* 服務器端填充 sockaddr結構 */
    bzero(&server_addr,sizeof(struct sockaddr_in)); // 初始化,置0
    server_addr.sin_family=AF_INET;                 // Internet
    server_addr.sin_addr.s_addr=htonl(INADDR_ANY);  // (將本機器上的long數據轉化爲網絡上的long數據)和任何主機通信  //INADDR_ANY 表示可以接收任意IP地址的數據，即綁定到所有的IP
    server_addr.sin_port=htons(portnumber);         // (將本機器上的short數據轉化爲網絡上的short數據)端口號

    /* 捆綁sockfd描述符到IP地址 */
    if(bind(sockfd,(struct sockaddr *)(&server_addr),sizeof(struct sockaddr))==-1)
    {
        fprintf(stderr,"Bind error:%s\n\a",strerror(errno));
        exit(1);
    }

    /* 設置允許連接的最大客戶端數 */
    if(listen(sockfd,5)==-1)
    {
        fprintf(stderr,"Listen error:%s\n\a",strerror(errno));
        exit(1);
    }

    while(1)
    {
        /* 服務器阻塞,直到客戶程序建立連接 */
        sin_size=sizeof(struct sockaddr_in);
        if((new_fd=accept(sockfd,(struct sockaddr *)(&client_addr),&sin_size))==-1)
        {
            fprintf(stderr,"Accept error:%s\n\a",strerror(errno));
            exit(1);
        }
        fprintf(stderr,"Server get connection from %s\n",inet_ntoa(client_addr.sin_addr)); // 將網絡地址轉換成.字符串

        //if((nbytes=read(new_fd,buffer,1024))==-1) //此處可以寫爲if((nbytes=recv(new_fd,buffer,1024,0))==-1)
        if((nbytes=recv(new_fd,buffer,1024,0))==-1)
        {
            fprintf(stderr,"Read Error:%s\n",strerror(errno));
            exit(1);
        }  
        buffer[nbytes]='\0';
        printf("Server received %s\n",buffer);

        /* 這個通訊已經結束 */
        close(new_fd);
        /* 循環下一個 */
    }

    /* 結束通訊 */
    close(sockfd);
    exit(0);
}