准備開始寫一些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