Linux下基于多线程的echo

32324

准备开始写一些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;
}
客户端代码：
#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地址