Linux下實現基於多線程的echo程序

准備開始寫一些Linux 下網絡編程以及多線程的blog，就從這個簡單的echo程序開始吧。

在echo的服務端使用多線程與客戶進行通信，可以實現一個服務端程序同時連接多個客戶的功能。那麼，到底在服務端要有多少個線程呢？

答案是：1+1+n。一個主線程，一個監聽（listen）線程，與n個客戶進行通信的n個線程。

服務端代碼：

#include "../unp.h"
#include <pthread.h>
    
void* str_echo(void* argv)
{
    int *pcommu = (int*)argv;
    int sockfd =*pcommu;
        
    ssize_t n;
    char buf[MAXLINE];
    
    while ( (n = read(sockfd,buf,MAXLINE)) > 0) {
        write(sockfd,buf,n);
    }
}
    
void* listen_func(void* argv)
{
    int* plistenfd = (int*)argv;
    int listenfd = *plistenfd;
        
    listen(listenfd,LISTENQ);
    for (;;) {
        struct sockaddr_in cliaddr;
        socklen_t clilen = sizeof(cliaddr);
            int connfd = accept(listenfd,(SA*)&cliaddr,&clilen);
        pthread_t communication;//通信線程
        pthread_create(&communication,NULL,&str_echo,&connfd);
    }
}
    
    
int main()
{
    int listenfd;
        
    struct sockaddr_in servaddr;
    
    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,(SA*)&servaddr,sizeof(servaddr));
        
    pthread_t listenPthread;//監聽線程
    pthread_create(&listenPthread,NULL,&listen_func,&listenfd);
        
    while(1);
    
    return 0;
}

返回欄目頁：http://www.bianceng.cn/OS/Linux/

客戶端代碼：

#include "../unp.h"
    
void str_cli(FILE *fp, int sockfd);
int main(int argc, char **argv)
{
    int sockfd;
    struct sockaddr_in servaddr,cliaddr;
    if (argc != 2) {
        printf("no IPaddress\n");
        return 0;
    }
    
    sockfd = socket(AF_INET, SOCK_STREAM, 0);
       
    bzero(&servaddr, sizeof(servaddr));
    servaddr.sin_family = AF_INET;
    servaddr.sin_port = htons(SERV_PORT);
    inet_pton(AF_INET, argv[1],&servaddr.sin_addr);
    
    connect(sockfd, (SA*) &servaddr, sizeof(servaddr));
    
    str_cli(stdin,sockfd);
    
    exit(0);
}
    
void str_cli(FILE *fp, int sockfd)
{
    char sendline[MAXLINE],recvline[MAXLINE];
    int n;
    while (fgets(sendline,MAXLINE,fp) != NULL) {
        write(sockfd, sendline, strlen(sendline));
    
        if(n = read(sockfd, recvline, MAXLINE) ) {
            recvline[n] = 0;
            fputs(recvline, stdout);
        }
    }
}

ps，之前在看UNP，客戶端代碼是UNP裡面的。我主要是把UNP裡面的服務端的程序改成了多線程的，在原書裡是基於多進程的，這樣的話服務端一旦運行就堵塞在監聽那，改成多線程，單獨拿出一個線程來監聽可以解決這種情況，主線程可以去做其他的操作，而不會阻塞在listen ：）。

pps，我在github上放了一個即時聊天的小程序，服務端也是基於多線程的，不過是window下的，而且用的原始的socket api，准備有時間改成linux+muduo庫，感興趣請戳：github地址

作者：cnblogs NicGanon