Linux TCP 多线程并发服务器实现详解_多线程并发的tcp服务端开发的实现原理

与多进程很相似，不多赘述，具体内容在多进程服务器的文章中

#include <stdio.h>
#include <arpa/inet.h>
#include <stdlib.h>
#include <pthread.h>
#include <sys/socket.h>
#include <unistd.h>
#include <sys/types.h>

#define SERV_PORT 5001
#define BACKLOG 5

void* fun(void *arg);

typedef struct		//自定义一种数据结构保存每个连接客户端的相关信息
{
	int fd;
	struct sockaddr_in addr;		//存放ip
	pthread_t id;		//存放线程号
}SockInfo;

int main(int argc, char *argv[])
{
	int rw_fd;
	int fd;
	int ret;
	socklen_t len = sizeof(struct sockaddr);
	struct sockaddr_in sin = {0};
	int i = 0;
	SockInfo info[32];

	for(i = 0; i < sizeof(info)/sizeof(info[0]); i++)
		info[i].fd = -1;		//初始化为-1

	//创建socket
	if((fd = socket(AF_INET, SOCK_STREAM, 0)) < 0)
	{
		perror("socket");
		exit(-1);
	}

	//初始化并绑定本机的ip和端口
	sin.sin_family = AF_INET;
	sin.sin_port = htons(SERV_PORT);
	//如果端口号需要输入则转化一下
	//int port = atoi(argv[1]); sin.sin_port = htons(port);
	sin.sin_addr.s_addr = htonl(INADDR_ANY);

	//避免出现bind error:address already in use的情况，设置端口复用
	//出现以上错误的原因是断开连接后需要等待2MSL时间
	int flag;
	setsockopt = (fd, SOL_SOCKET, SO_REUSEADDR, &flag, sizeof(flag));
	
	if((bind(fd, (struct sockaddr *)&sin, sizeof(sin))) < 0)
	{
		perror("bind");
		exit(-1);
	}

	//监听socket
	if(listen(fd, BACKLOG) < 0)
	{
		perror("listen");
		exit(-1);
	}

	while(1)		
	{
		//这样是为了选出未使用的最小的i，如果不这么做，如果已经连接的客户端又断开了，那他的i就无法
		//被使用，通过初始化fd为-1，并在断开的时候再次设置为-1，达到重复利用的目的
		for(i = 0; i < sizeof(info)/sizeof(info[0]); i++)
		{
			if(info[i].fd == -1)
				break;
		}

		//主进程接收客户端请求
		if((info[i].fd = accept(fd, (struct sockaddr *)&info[i].addr, &len)) < 0)		
		{
			perror("accept");
			exit(-1);
		}

		//创建子线程用于通信
		pthread_create(&info[i].id, NULL, fun, &info[i]);		

		//设置线程分离,线程结束后会自动回收资源
		pthread_detach(info[i].id);

		if(i == 32)		//数值跟创建的结构体数组的大小有关
			break;
	}
	close(fd);
	//只退出主线程，对子线程没有影响
	pthread_exit(NULL);
	return 0;
}

void* fun(void *arg)
{
	char ip[64];
	char buf[1024];
	SockInfo *info = (SockInfo *)arg;

	//接下来就是通信
	while(1)
	{
		//打印客户端ip和端口号
		printf("Cilent IP: %s, Port: %d\n\n", 
				inet_ntop(AF_INET,&info->addr.sin_addr.s_addr, ip, sizeof(ip)),
				ntohs(info->addr.sin_port));

		int len = read(info->fd, buf, sizeof(buf));
		if(len == -1)
		{
			perror("read error");
			pthread_exit(NULL);
		}
		else if(len == 0)
		{
			info->fd = -1;
			printf("客户端已经断开连接\n");
			close(info->fd);
			break;
		}
		else
		{
			printf("recv client: %s\n",buf);
			write(info->fd, buf, len);
		}
	}
	return NULL;
}

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133