【网络编程三】网络通信之多进程(线程)实现TCP通信(一)

【多进程】

【多线程】

/*************************************************************************
    > File summary: 多线程服务器
    > Author: 
    > Mail: xls0930@163.com 
    > Created Time: Sun 27 Jan 2019 10:08:19 PM CST
 ************************************************************************/

#include<stdio.h>
#include<string.h>
#include<arpa/inet.h>
#include<pthread.h>
#include<ctype.h>
#include<unistd.h>
#include<fcntl.h>

#define MAXLINE 8192
#define SERV_PORT 8000

struct s_info
{
    struct sockaddr_in cliaddr;
    int connfd;
};

// 回调函数
void *do_work(void *arg)
{
    int n,i;
    struct s_info *ts = (struct s_info *)arg;
    char buf[MAXLINE];
    char str[INET_ADDRSTRLEN];

    while(1)
    {
        // 读出客户端的信息,关于read的返回值:返回n(n>0)表示读出n字节的数据,返回0表示已经读到结尾(可以理解为对端已关闭)
        n = read(ts->connfd,buf,MAXLINE);
        if(n == 0)
        {
            printf("the client %d close...
", ts->connfd);
            //跳出循环
            break;
        }
        //打印输出客户端的ip+port
        printf("received from %s at PORT %d
",
                inet_ntop(AF_INET, &(*ts).cliaddr.sin_addr,str,sizeof(str)),
        
                ntohs((*ts).cliaddr.sin_port));
        //对读出的信息进行小写转大写操作
        for(i = 0; i < n;i++)
        {
            buf[i] = toupper(buf[i]);
        }
        // 写到服务器的标准输出
        write(STDOUT_FILENO,buf,n);
        // 写回客户端
        write(ts->connfd,buf,n);
    }
    close(ts->connfd);
    return (void *)0;
}

int main()
{
    struct sockaddr_in servaddr,cliaddr;
    socklen_t cliaddr_len;
    int listenfd,connfd;
    pthread_t tid;

    struct s_info ts[256];    // 创建结构体数组
    int i = 0;

    listenfd = socket(AF_INET,SOCK_STREAM,0);

    bzero(&servaddr, sizeof(servaddr));
    servaddr.sin_family = AF_INET;
    servaddr.sin_addr.s_addr = htonl(INADDR_ANY);
    servaddr.sin_port = htons(SERV_PORT);

    bind(listenfd, (struct sockaddr *)&servaddr, sizeof(servaddr));

    listen(listenfd, 128);

    printf("accepting client connect ...
");

    while(1)
    {
        cliaddr_len = sizeof(cliaddr);
        // listenfd做监听,connfd与客户端进行读写操作,connfd中存放着客户端的信息
        connfd = accept(listenfd, (struct sockaddr *)&cliaddr, &cliaddr_len);
        ts[i].cliaddr = cliaddr;
        ts[i].connfd = connfd;

        pthread_create(&tid,NULL,do_work,(void*)&ts[i]);
        // 回收子线程
        pthread_detach(tid);
        i++;
    }
    return 0;

}

我们可以使用nc命令来模拟客户端，对代码进行测试。