准备开始写一些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)) > ) {
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,); 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(); return ;
}
客户端代码:
#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 != ) {
printf("no IPaddress\n");
return ;
} sockfd = socket(AF_INET, SOCK_STREAM, ); bzero(&servaddr, sizeof(servaddr));
servaddr.sin_family = AF_INET;
servaddr.sin_port = htons(SERV_PORT);
inet_pton(AF_INET, argv[],&servaddr.sin_addr); connect(sockfd, (SA*) &servaddr, sizeof(servaddr)); str_cli(stdin,sockfd); exit();
} 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] = ;
fputs(recvline, stdout);
}
}
}
ps,之前在看UNP,客户端代码是UNP里面的。我主要是把UNP里面的服务端的程序改成了多线程的,在原书里是基于多进程的,这样的话服务端一旦运行就堵塞在监听那,改成多线程,单独拿出一个线程来监听可以解决这种情况,主线程可以去做其他的操作,而不会阻塞在listen :)。
pps,我在github上放了一个即时聊天的小程序,服务端也是基于多线程的,不过是window下的,而且用的原始的socket api,准备有时间改成linux+muduo库,感兴趣请戳:github地址