首页 > 代码库 > Linux下基于多线程的echo

Linux下基于多线程的echo

    准备开始写一些Linux 下网络编程以及多线程的blog,就从这个简单的echo程序开始吧。

 

    在echo的服务端使用多线程与客户进行通信,可以实现一个服务端程序同时连接多个客户的功能。那么,到底在服务端要有多少个线程呢?

    答案是:1+1+n。一个主线程,一个监听(listen)线程,与n个客户进行通信的n个线程。

    服务端代码:

 1 #include "../unp.h"
 2 #include <pthread.h>
 3 
 4 void* str_echo(void* argv)
 5 {
 6     int *pcommu = (int*)argv;
 7     int sockfd =*pcommu;
 8     
 9     ssize_t n;
10     char buf[MAXLINE];
11 
12     while ( (n = read(sockfd,buf,MAXLINE)) > 0) {
13         write(sockfd,buf,n);
14     }
15 }
16 
17 void* listen_func(void* argv)
18 {
19     int* plistenfd = (int*)argv;
20     int listenfd = *plistenfd;
21     
22     listen(listenfd,LISTENQ);
23     for (;;) {
24         struct sockaddr_in cliaddr;
25         socklen_t clilen = sizeof(cliaddr);
26             int connfd = accept(listenfd,(SA*)&cliaddr,&clilen);
27         pthread_t communication;//通信线程
28         pthread_create(&communication,NULL,&str_echo,&connfd);
29     }
30 }
31 
32 
33 int main()
34 {
35     int listenfd;
36     
37     struct sockaddr_in servaddr;
38 
39     listenfd = socket(AF_INET,SOCK_STREAM,0);
40 
41     bzero(&servaddr,sizeof(servaddr));
42     servaddr.sin_family = AF_INET;
43     servaddr.sin_addr.s_addr = htonl(INADDR_ANY);
44     servaddr.sin_port = htons(SERV_PORT);
45 
46     bind(listenfd,(SA*)&servaddr,sizeof(servaddr));
47     
48     pthread_t listenPthread;//监听线程
49     pthread_create(&listenPthread,NULL,&listen_func,&listenfd);
50     
51     while(1);
52 
53     return 0;
54 }

客户端代码:

 1 #include "../unp.h"
 2 
 3 void str_cli(FILE *fp, int sockfd);
 4 int main(int argc, char **argv)
 5 {
 6     int sockfd;
 7     struct sockaddr_in servaddr,cliaddr;
 8     if (argc != 2) {
 9         printf("no IPaddress\n");
10         return 0;
11     }
12 
13     sockfd = socket(AF_INET, SOCK_STREAM, 0);
14    
15     bzero(&servaddr, sizeof(servaddr));
16     servaddr.sin_family = AF_INET;
17     servaddr.sin_port = htons(SERV_PORT);
18     inet_pton(AF_INET, argv[1],&servaddr.sin_addr);
19 
20     connect(sockfd, (SA*) &servaddr, sizeof(servaddr));
21 
22     str_cli(stdin,sockfd);
23 
24     exit(0);
25 }
26 
27 void str_cli(FILE *fp, int sockfd)
28 {
29     char sendline[MAXLINE],recvline[MAXLINE];
30     int n;
31     while (fgets(sendline,MAXLINE,fp) != NULL) {
32         write(sockfd, sendline, strlen(sendline));
33 
34         if(n = read(sockfd, recvline, MAXLINE) ) {
35             recvline[n] = 0;
36             fputs(recvline, stdout);
37         }
38     }
39 }

 

ps,之前在看UNP,客户端代码是UNP里面的。我主要是把UNP里面的服务端的程序改成了多线程的,在原书里是基于多进程的,这样的话服务端一旦运行就堵塞在监听那,改成多线程,单独拿出一个线程来监听可以解决这种情况,主线程可以去做其他的操作,而不会阻塞在listen  :)。

 

pps,我在github上放了一个即时聊天的小程序,服务端也是基于多线程的,不过是window下的,而且用的原始的socket api,准备有时间改成linux+muduo库,感兴趣请戳:github地址