Linux TCP server系列（6）-select模式下的多线程server

目标：

修改上一篇的select模式下的server，让它使用多线程来处理客户端请求（多进程的模式已经在上篇中加了注释）。

思路：

（1）服务器

我们已经在之前的客户端模型多个并发用户的过程中使用过多线程的技术了（其中还涉及到多线程利用条件变量进行线程同步），在这里我们可以很轻松的在上篇文章代码中加入线程部分代码。

//for thread

int *lptr;

pthread_t pid;

//for thread

for(i=0;i<=maxi;i++)

{

if((sockfd=client[i]) <0)

continue;

if(FD_ISSET(sockfd,&rset))

{

//thread client

lptr=(int*)malloc(sizeof(int));

*lptr=sockfd;

interrerr=pthread_create(&pid,NULL,threadPerClient,lptr);

if(errerr!=0)

printf("err%s",strerror(errno));

FD_CLR(sockfd,&allset); //短连接后关闭socket，服务器发不过来

client[i]=-1;

printf("canread : %d,%d,%d/n",i,sockfd,nready);

if(--nready<=0)

break;

//threadclient

}

}

但是！！！！真的那么轻松就可以加入吗？为什么我不直接在pthread_create中传入sockfd，而是要再new一个整型数然后赋值给它再传递过去呢？

答案很明显，如果直接使用sockfd的指针，那么在下次sockfd被client[i]赋值时，sockfd的值变了！而这时线程的实际操作的正是这个sockfd（不只因为线程共享变量，同时还因为传递的是指针）！所以我们复制一个新值，让线程自己处理对应的fd。

这样就完了？很明显不会那么简单，让我们看看线程处理函数就知道。

void*threadPerClient(void *arg)

{

int connfd=*((int*)arg);

free(arg); //防止内存泄露

pthread_detach(pthread_self());

printf("client from %d(socketnum)/n",connfd);

str_echo(connfd);

close( connfd );

return NULL;

}

看到注释了吗？“防止内存泄露“！！这是很容易发生的事，在刚才的代码中new一个新对象，那么我们就必须在使用后释放。除此以外我们还注意到这里调用了pthread_detach函数，因为线程分为joinable和unjoinable两种。

如果线程是joinable状态，当线程函数自己返回退出时或pthread_exit时都不会释放线程所占用堆栈和线程描述符（总计8K多）。只有当你调用了pthread_join之后这些资源才会被释放。若是unjoinable状态的线程，这些资源在线程函数退出时或pthread_exit时自动会被释放。

unjoinable属性可以在pthread_create时指定，或在线程创建后在线程中pthread_detach自己,
如：pthread_detach(pthread_self())，将状态改为unjoinable状态，确保资源的释放。或者将线程置为 joinable,然后适时调用pthread_join.

（其实就类似于进程退出时父亲进程的wait）

（2）客户端

无需修改

代码：


server.cpp

1 #include<sys/types.h>
2 #include<sys/socket.h>
3 #include<strings.h>
4 #include<arpa/inet.h>
5 #include<unistd.h>
6 #include<stdlib.h>
7 #include<stdio.h>
8 #include<string.h>
9 #include<errno.h>
10 #include<signal.h>
11 #include<sys/wait.h>
12 #include<sys/time.h>
13 #include<pthread.h>
14
15 #define LISTEN_PORT 84
16
17 //服务器处理
18 void str_echo(int sockfd)            // 服务器收到客户端的消息后的响应
19 {
20     ssize_t n;
21 char line[512];
22
23     printf("ready to read/n");
24
25 while( (n=read(sockfd,line,512))>0 )      //阻塞IO版本
26     {
27             line
='/0';
28             printf("Client Diary: %s/n",line);
29
30 char msgBack[512];
31             snprintf(msgBack,sizeof(msgBack),"recv: %s/n",line);
32             write(sockfd,msgBack,strlen(msgBack));
33             bzero(&line,sizeof(line));
34     }
35     printf("end read/n");
36 }
37
38 //子进程结束的信号处理
39 float timeuse;
40 void sig_child(int signo)         //父进程对子进程结束的信号处理
41 {
42     pid_t pid;
43 int   stat;
44
45 struct timeval tpstart,tpend;
46 //float timeuse;
47     gettimeofday(&tpstart,NULL);
48
49 while( (pid=waitpid(-1,&stat,WNOHANG))>0)
50     printf("child %d terminated/n",pid);
51
52     gettimeofday(&tpend,NULL);
53
54     timeuse+=1000000*(tpend.tv_sec-tpstart.tv_sec)+tpend.tv_usec-tpstart.tv_usec;
55 //timeuse/=1000000;
56     printf("Use Time:%f/n",timeuse/1000000);
57 return;
58 }
59
60
61
62 //使用线程代替进程来处理客户请求
63 void* threadPerClient(void*arg)
64 {
65 int connfd=*((int*)arg);
66     free(arg);                //防止内存泄露
67     pthread_detach(pthread_self());
68
69     printf("client from %d(socket num)/n",connfd);
70     str_echo(connfd);
71
72     close( connfd );
73 return NULL;
74 }
75
76
77 int main(int argc, char**argv)
78 {
79 int listenfd, connfd;
80     pid_t childpid;
81     socklen_t chilen;
82
83 struct sockaddr_in chiaddr,servaddr;
84
85 //for select
86 int i,maxi,maxfd,sockfd;
87 int nready,client[FD_SETSIZE];
88     ssize_t n;
89     fd_set rset,allset;
90 //for select
91
92     listenfd=socket(AF_INET,SOCK_STREAM,0);
93 if(listenfd==-1)
94     {
95         printf("socket established error: %s/n",(char*)strerror(errno));
96     }
97
98     bzero(&servaddr,sizeof(servaddr));
99     servaddr.sin_family=AF_INET;
100     servaddr.sin_addr.s_addr=htonl(INADDR_ANY);
101     servaddr.sin_port=htons(LISTEN_PORT);
102
103 int bindc=bind(listenfd,(struct sockaddr *)&servaddr,sizeof(servaddr));
104 if(bindc==-1)
105     {
106         printf("bind error: %s/n",strerror(errno));
107     }
108
109     listen(listenfd,SOMAXCONN);               //limit是SOMAXCONN
110
111 //initial "select" elements
112     maxfd=listenfd;            //新增listenfd，所以更新当前的最大fd
113     maxi=-1;
114 for(i=0;i<FD_SETSIZE;i++)
115       client[i] =-1;
116     FD_ZERO(&allset);
117     FD_SET(listenfd,&allset);
118 //end initial
119
120 int cliconn=0;
121     signal(SIGCHLD,sig_child);
122 for(;;)
123     {
124         rset=allset;
125         nready=select(maxfd+1,&rset,NULL,NULL,NULL); //一开始select监听的是监听口
126 //如果有timeout设置，那么每次select之前都要再重新设置一下timeout的值
127 //因为select会修改timeout的值。
128 if(FD_ISSET(listenfd,&rset))
129         {
130             chilen=sizeof(chiaddr);
131
132             connfd=accept(listenfd,(struct sockaddr*)&chiaddr,&chilen);
133 //阻塞在accept，直到三次握手成功了才返回
134 if(connfd==-1)
135             printf("accept client error: %s/n",strerror(errno));
136 else
137             printf("client connected:%d/n",++cliconn);
138
139 for(i=0;i<FD_SETSIZE;i++)
140             {
141 if (client[i]<0)
142                 {
143                     client[i]=connfd;    //找一个最小的插进入
144 break;
145                 }
146             }
147 if(i==FD_SETSIZE)
148             {
149                 printf("too many clients/n");
150                 exit(0);
151             }
152             FD_SET(connfd,&allset);   //新加入的描述符，还没判断是否可以或者写，所以后面使用rset而不是allset
153
154
155
156 if(connfd>maxfd)
157               maxfd=connfd;
158 if(i>maxi)
159               maxi=i;
160 if(--nready<=0)
161 continue;
162         }
163
164 //for thread
165 int*lptr;
166         pthread_t pid;
167 //for thread
168
169 for(i=0;i<=maxi;i++)
170         {
171 if( (sockfd=client[i]) <0)
172 continue;
173 if(FD_ISSET(sockfd,&rset))
174             {
175 //thread client
176                 lptr=(int*)malloc(sizeof(int));
177 *lptr=sockfd;
178 //pthread_t pid;
179
180 int errerr=pthread_create(&pid,NULL,threadPerClient,lptr);
181 if(errerr!=0)
182                 printf("err %s",strerror(errno));
183
184                 FD_CLR(sockfd,&allset);                                //短连接后关闭socket，服务器发不过来
185                 client[i]=-1;
186
187                 printf("can read : %d,%d,%d/n",i,sockfd,nready);
188 if(--nready<=0)
189 break;
190 //thread client
191
192             }
193         }
194
195     }
196 }

作者：Aga.J

出处：http://www.cnblogs.com/aga-j

本文版权归作者和博客园共有，欢迎转载，但未经作者同意必须保留此段声明，且在文章页面明显位置给出原文连接，否则保留追究法律责任的权利。