准备开始写一些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地址