linux下socket编程，使用多线程实现多客户端与服务器通信

多线程实现多个客户端与服务器通信，服务器主线程一直不停监听客户端的连接，每来一个客户端连接，服务器主控线程就创建一个线程去处理这个客户端数据，服务器端的线程负责把客户端的发来的数据从小写变成大写再写回客户端。

服务器端代码：server.c

/* server.c */
#include 
#include 
#include 
#include 
#include 
#include  "wrap.h"
#include 

#define MAXLINE 80
#define SERV_PORT 8000

struct s_info {
	struct sockaddr_in cliaddr;
	int connfd;
};
void *do_work(void *arg)
{
	int n,i;
	struct s_info *ts = (struct s_info*)arg;
	char buf[MAXLINE];//在线程自己的用户空间栈开辟的,该线程运行结束的时候,主控线程就不能操作这块内存了
	char str[INET_ADDRSTRLEN];//INET_ADDRSTRLEN 是宏16个字节

//在创建线程前设置线程创建属性,设为分离态,效率高
	pthread_detach(pthread_self());

	while (1) {
	n = read(ts->connfd, buf, MAXLINE);
	if (n == 0) {
		printf("the other side has been closed.\n");
		break;
		}
	printf("received from %s at PORT %d\n",
	inet_ntop(AF_INET, &(*ts).cliaddr.sin_addr, str, sizeof(str)),ntohs((*ts).cliaddr.sin_port));

	for (i = 0; i < n; i++)
	    buf[i] = toupper(buf[i]);
	write(ts->connfd, buf, n);
	}
	close(ts->connfd);
}
int main(void)
{
	struct sockaddr_in servaddr, cliaddr;
	socklen_t cliaddr_len;
	int listenfd, connfd;
	int i = 0;
	pthread_t tid;
	struct s_info ts[3497];

	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, (struct sockaddr *)&servaddr, sizeof(servaddr));
	listen(listenfd, 20);

	printf("Accepting connections ...\n");
	while (1) {
		cliaddr_len = sizeof(cliaddr);
		connfd = Accept(listenfd, (struct sockaddr *)&cliaddr, &cliaddr_len);
		ts[i].cliaddr = cliaddr;
		ts[i].connfd = connfd;
/* 达到线程最大数时，pthread_create出错处理, 增加服务器稳定性 */
		pthread_create(&tid, NULL, do_work, (void*)&ts[i]);//把accept得到的客户端信息传给线程，让线程去和客户端进行数据的收发
		i++;
	}
return 0;
}

客户端代码：client.c

/* client.c */
#include 
#include 
#include 
#include 
#include "wrap.h"
#include 
#include 


#define MAXLINE 80
#define SERV_PORT 8000

int main(int argc, char *argv[])
{
	struct sockaddr_in servaddr;
	char buf[MAXLINE];
	int sockfd, n;
	sockfd = socket(AF_INET, SOCK_STREAM, 0);
	bzero(&servaddr, sizeof(servaddr));
	servaddr.sin_family = AF_INET;
	inet_pton(AF_INET, "127.0.0.1", &servaddr.sin_addr.s_addr);//本地ip
	servaddr.sin_port = htons(SERV_PORT);

	connect(sockfd, (struct sockaddr *)&servaddr, sizeof(servaddr));

	while (fgets(buf, MAXLINE, stdin) != NULL) {
		write(sockfd, buf, strlen(buf));
		n = read(sockfd, buf, MAXLINE);
		if (n == 0)
		printf("the other side has been closed.\n");
		else
		write(STDOUT_FILENO, buf, n);
		}
		close(sockfd);
	return 0;
}

运行结果如下：显示的是开了五个客户端的效果

要解释一下server.c的代码中的struct s_info ts[3497]，这块的意思是服务器进程能创建的线程的最大数，这样要根据你自己的机器来定，可以使用下面代码来查看你的电脑可以创建多少个线程：

#include 
#include 
#include 
#include 

void *th_fun(void *arg)
{
        while(1)
        sleep(1);
}
int main(void)
{
        pthread_t tid;
        int i=1,err;
        while(1){
                err = pthread_create(&tid,NULL,th_fun,NULL);
                if(err!=0){
                        printf("%s\n",strerror(err));
                        exit(1);
                }
                printf("%d\n",i++);
        }

}

那么创建的最大线程个数可以改变吗，答案是肯定的。一个进程可以创建的线程数由可用虚拟空间和线程的栈的大小共同决定，只要虚拟空间足够，那么新线程的建立就会成功。可以通过减小你线程栈的大小就可以实现了增加线程个数。

linux下ulimit -a可以查看线程的属性，如下图，可以看到线程的栈大小为8192KB

我们可以使用 ulimit -s 4096来改变栈的大小，这样可以创建的线程个数就多一倍了