基于TCP的客户端--服务器

服务器端代码实现：

#include <stdio.h>
#include <stdlib.h>
#include <signal.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>

//信号的处理非常类似于中断的处理，当有信号产生时，执行这个函数
void handler_signal(int signum)
{
    //非阻塞方式调用，有子进程退出时，对子进程进行收尸
    waitpid(-1,NULL,WNOHANG);
    return ;
}

int do_client(int connect_fd)
{
    int n;
    char buf[1024];

    while(1)
    {
        n = read(connect_fd,buf,sizeof(buf) - 1);
        if(n <= 0)
            break;

        buf[n] = '\0';
        printf("Read %d bytes : %s\n",n,buf);

        if(strncmp(buf,"quit",4)==0)
            break;
    }

    //子进程直接退出，让父进程对其收尸
    exit(EXIT_SUCCESS);
}

//./server   ip   port
int main(int argc, const char *argv[])
{
    int pid;
    int sockfd;
    int connect_fd;
    struct sockaddr_in server_addr;
    struct sockaddr_in peer_addr;
    socklen_t  addrlen = sizeof(struct sockaddr);

    if(argc < 3)
    {
        fprintf(stderr,"Usage : %s  \n",argv[0]);
        exit(EXIT_FAILURE);
    }

    //为了防止将尸态子进程的出现，子进程退出时，父进程需要对子进程收尸
    //设置SIGCHLD信号处理方式:捕捉
    if(signal(SIGCHLD,handler_signal) == SIG_ERR)
    {
        perror("Fail to signal");
        exit(EXIT_FAILURE);
    }

    //1.创建流式套接字
    sockfd = socket(AF_INET,SOCK_STREAM,0);
    if(sockfd < 0){
        perror("Fail to socket");
        exit(EXIT_FAILURE);
    }

    //2.填充服务器的IP地址和端口信息
    server_addr.sin_family = AF_INET;
    server_addr.sin_port = htons(atoi(argv[2]));
    server_addr.sin_addr.s_addr = inet_addr(argv[1]);
    //3.把服务器的端口和IP地址信息和流式套接字绑定
    if(bind(sockfd,(struct sockaddr *)&server_addr,sizeof(server_addr)) < 0)
    {
        perror("Fail to bind");
        exit(EXIT_FAILURE);
    }

    //4.设置流式套接字为监听模式
    //在服务器端的某个应用程序需要一种状态去等候客户端的通信邀请，这种状态称之
    //为“监听”，监听本身不具备连接的功能，它仅仅是一种等候。
    listen(sockfd,128);

    printf("listen ...\n");

    while(1)
    {
        //在请求队列中提取客户端的请求，提取成功之后会新创建一个新客户端的套接字，
        //这个新的套接字才是和客户端真正进行通信的套接字，而前面的套接字是用于实
        //现监听的。
        connect_fd = accept(sockfd,(struct sockaddr *)&peer_addr,&addrlen);
        if(connect_fd < 0){
            perror("Fail to accept");
            exit(EXIT_FAILURE);
        }

        printf("connect_fd : %d\n",connect_fd);

        printf("-----------------------------\n");
        printf("Port : %d\n",ntohs(peer_addr.sin_port));
        printf("Ip   : %s\n",inet_ntoa(peer_addr.sin_addr));
        printf("-----------------------------\n");

        pid = fork();
        if(pid < 0){
            perror("Fail to fork");
            exit(EXIT_FAILURE);
        }

        if(pid == 0)
        {
            //子进程中读取客户端的发送的数据，并且把父进程中拷贝过来的监听的套接字关闭
            close(sockfd);
            do_client(connect_fd);
        }

        //为了防止文件描述符浪费，父子进程在最后执行完之后都应该关闭已经打开的文件描述符
        close(connect_fd);
    }

    return 0;
}

客户端代码实现：

#include <stdio.h>
#include <stdlib.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>

//./server   ip   port
int main(int argc, const char *argv[])
{
    int n;
    int sockfd;
    int connect_fd;
    char buf[1024];
    struct sockaddr_in server_addr;
    struct sockaddr_in peer_addr;
    socklen_t  addrlen = sizeof(struct sockaddr);

    if(argc < 3)
    {
        fprintf(stderr,"Usage : %s  \n",argv[0]);
        exit(EXIT_FAILURE);
    }

    //1.创建流式套接字
    sockfd = socket(AF_INET,SOCK_STREAM,0);
    if(sockfd < 0){
        perror("Fail to socket");
        exit(EXIT_FAILURE);
    }

    //2.填充服务器IP地址和端口信息
    server_addr.sin_family = AF_INET;
    server_addr.sin_port = htons(atoi(argv[2]));
    server_addr.sin_addr.s_addr = inet_addr(argv[1]);
    //3.链接服务器(实质上是完成3次握手，链接成功之后Linux内核会把请求放置在内核的请求队列中)
    // 特别注意:这个函数仅仅只在客户端使用，主要的目的就是完成和服务器的3次握手
    if(connect(sockfd,(struct sockaddr *)&server_addr,sizeof(server_addr)) < 0)
    {
        perror("Fail to connect");
        exit(EXIT_FAILURE);
    }

    while(1)
    {
        printf("client >");
        fgets(buf,sizeof(buf),stdin);
        buf[strlen(buf) - 1] = '\0';

        //给服务器发送数据
        n = write(sockfd,buf,strlen(buf));
        if(n < 0){
            perror("Fail to write");
            exit(EXIT_FAILURE);
        }
    }

    return 0;
}

运行测试结果如下：

ubuntu@ubuntu:~/network_teacher/day02/tcp/server$ ./server 192.168.127.131 8888
listen ...
connect_fd : 4
-----------------------------
Port : 53984
Ip : 192.168.127.131
-----------------------------
Read 6 bytes : hello
Read 5 bytes : world
Read 3 bytes : xxx

ubuntu@ubuntu:~/network_teacher/day02/tcp/client$ ./client 192.168.127.131 8888
client >hello
client >world
client >xxx
client >