Linux多线程很简单

最新推荐文章于 2024-08-17 09:28:44 发布

小麦大大

最新推荐文章于 2024-08-17 09:28:44 发布

阅读量350

点赞数

分类专栏： linux网络与系统编程

linux网络与系统编程专栏收录该内容

52 篇文章 13 订阅

订阅专栏

多线程服务器

多线程服务器是对多进程的服务器的改进，由于多进程服务器在创建进程时要消耗较大的系统资源，所以用线程来取代进程，这样服务处理程序可以较快的创建。据统计，创建线程与创建进程要快10100倍，所以又把线程称为“轻量级”进程。线程与进程不同的是：一个进程内的所有线程共享相同的全局内存、全局变量等信息，这种机制又带来了同步问题。

以下是多线程服务器模板:

在这里插入图片描述

示例代码如下：


#include <stdio.h>
#include <stdlib.h>
#include <string.h>						
#include <unistd.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>					
#include <pthread.h>

 

/************************************************************************

函数名称：	void *client_process(void *arg)
函数功能：	线程函数,处理客户信息
函数参数：	已连接套接字
函数返回：	无

************************************************************************/

void *client_process(void *arg)
{

	int recv_len = 0;
	char recv_buf[1024] = "";	// 接收缓冲区
	int connfd = (int)arg; // 传过来的已连接套接字

	// 接收数据
	while((recv_len = recv(connfd, recv_buf, sizeof(recv_buf), 0)) > 0)
	{
		printf("recv_buf: %s\n", recv_buf); // 打印数据
		send(connfd, recv_buf, recv_len, 0); // 给客户端回数据
	}

	
	printf("client closed!\n");
	close(connfd);	//关闭已连接套接字

	return 	NULL;

}

 
//===============================================================
// 语法格式：	void main(void)
// 实现功能：	主函数，建立一个TCP并发服务器
// 入口参数：	无
// 出口参数：	无
//===============================================================

int main(int argc, char *argv[])
{

	int sockfd = 0;				// 套接字
	int connfd = 0
	int err_log = 0;
	
	struct sockaddr_in my_addr;	// 服务器地址结构体
	unsigned short port = 8080; // 监听端口
	pthread_t thread_id;
	
	printf("TCP Server Started at port %d!\n", port);
	sockfd = socket(AF_INET, SOCK_STREAM, 0);   // 创建TCP套接字
	if(sockfd < 0)
	{

		perror("socket error");
		exit(-1);
	}

	

	bzero(&my_addr, sizeof(my_addr));	   // 初始化服务器地址
	my_addr.sin_family = AF_INET;
	my_addr.sin_port   = htons(port);
	my_addr.sin_addr.s_addr = htonl(INADDR_ANY);
	printf("Binding server to port %d\n", port);
	// 绑定
	err_log = bind(sockfd, (struct sockaddr*)&my_addr, sizeof(my_addr));
	
	if(err_log != 0)
	{
		perror("bind");
		close(sockfd);		
		exit(-1);
	}


	// 监听，套接字变被动
	err_log = listen(sockfd, 10);

	if( err_log != 0)
	{
		perror("listen");
		close(sockfd);		
		exit(-1);
	}

	printf("Waiting client...\n");

	while(1)
	{

		char cli_ip[INET_ADDRSTRLEN] = "";	   // 用于保存客户端IP地址
		struct sockaddr_in client_addr;		   // 用于保存客户端地址
		socklen_t cliaddr_len = sizeof(client_addr);   // 必须初始化!!!

		//获得一个已经建立的连接	
		connfd = accept(sockfd, (struct sockaddr*)&client_addr, &cliaddr_len);   							

		if(connfd < 0)
		{
			perror("accept this time");
			continue;
		}

		// 打印客户端的 ip 和端口
		inet_ntop(AF_INET, &client_addr.sin_addr, cli_ip, INET_ADDRSTRLEN);
		printf("----------------------------------------------\n");
		printf("client ip=%s,port=%d\n", cli_ip,ntohs(client_addr.sin_port));


		if(connfd > 0)
		{
			//由于同一个进程内的所有线程共享内存和变量，因此在传递参数时需作特殊处理，值传递。 
			//(void *)connfd注意这个参数。传值和传地址很不一样
			pthread_create(&thread_id, NULL, (void *)client_process, (void *)connfd);  //创建线程
			pthread_detach(thread_id); // 线程分离，结束时自动回收资源

		}

	}

	close(sockfd);
	return 0;

}

注意，上面例子给线程传参有很大的局限性，最简单的一种情况，如果我们需要给线程传多个参数，这时候我们需要结构体传参，这种值传递编译都通不过，这里之所以能够这么值传递，是因为， int 长度时 4 个字节， void * 长度也是 4 个字节。

int connfd = accept(sockfd, (struct sockaddr*)&client_addr, &cliaddr_len);
pthread_create(&thread_id, NULL, (void *)client_process, (void *)connfd);

如果考虑类型匹配的话，应该是这么传参，pthread_create()最后一个参数应该传地址（ &connfd ），而不是值：

int connfd = accept(sockfd, (struct sockaddr*)&client_addr, &cliaddr_len);
pthread_create(&thread_id, NULL, (void *)client_process, (void *)&connfd);

但是，如果按地址传递的话，又会有这么一个问题，假如有多个客户端要连接这个服务器，正常的情况下，一个客户端连接对应一个 connfd，相互之间独立不受影响，但是，假如多个客户端同时连接这个服务器，A 客户端的连接套接字为 connfd，服务器正在用这个 connfd 处理数据，还没有处理完，突然来了一个 B 客户端，accept()之后又生成一个 connfd, 因为是地址传递， A 客户端的连接套接字也变成 B 这个了，这样的话，服务器肯定不能再为 A 客户端服务器了，这时候，我们就需要考虑多任务的互斥或同步问题了，这里通过互斥锁来解决这个问题，确保这个connfd值被一个临时变量保存过后，才允许修改。


#include <pthread.h>

pthread_mutex_t mutex;	// 定义互斥锁，全局变量

pthread_mutex_init(&mutex, NULL); // 初始化互斥锁，互斥锁默认是打开的

// 上锁，在没有解锁之前，pthread_mutex_lock()会阻塞

pthread_mutex_lock(&mutex);	
int connfd = accept(sockfd, (struct sockaddr*)&client_addr, &cliaddr_len);

//给回调函数传的参数，&connfd，地址传递
pthread_create(&thread_id, NULL, (void *)client_process, (void *)&connfd);  //创建线程

// 线程回调函数
void *client_process(void *arg)
{

	int connfd = *(int *)arg; // 传过来的已连接套接字
	// 解锁，pthread_mutex_lock()唤醒，不阻塞
	pthread_mutex_unlock(&mutex);
	return 	NULL;

}

完整代码：


#include <stdio.h>
#include <stdlib.h>
#include <string.h>						
#include <unistd.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>					
#include <pthread.h>

pthread_mutex_t mutex;	// 定义互斥锁，全局变量

 
/************************************************************************

函数名称：	void *client_process(void *arg)
函数功能：	线程函数,处理客户信息
函数参数：	已连接套接字
函数返回：	无
************************************************************************/

void *client_process(void *arg)
{

	int recv_len = 0;
	char recv_buf[1024] = "";	// 接收缓冲区
	int connfd = *(int *)arg; // 传过来的已连接套接字

	

	// 解锁，pthread_mutex_lock()唤醒，不阻塞
	pthread_mutex_unlock(&mutex); 

	// 接收数据
	while((recv_len = recv(connfd, recv_buf, sizeof(recv_buf), 0)) > 0)
	{
		printf("recv_buf: %s\n", recv_buf); // 打印数据
		send(connfd, recv_buf, recv_len, 0); // 给客户端回数据

	}

	
	printf("client closed!\n");
	close(connfd);	//关闭已连接套接字
	return 	NULL;

}

 

//===============================================================

// 语法格式：	void main(void)
// 实现功能：	主函数，建立一个TCP并发服务器
// 入口参数：	无
// 出口参数：	无
//===============================================================

int main(int argc, char *argv[])
{

	int sockfd = 0;				// 套接字
	int connfd = 0;
	int err_log = 0;
	struct sockaddr_in my_addr;	// 服务器地址结构体
	unsigned short port = 8080; // 监听端口
	pthread_t thread_id;
	pthread_mutex_init(&mutex, NULL); // 初始化互斥锁，互斥锁默认是打开的

	printf("TCP Server Started at port %d!\n", port);

	sockfd = socket(AF_INET, SOCK_STREAM, 0);   // 创建TCP套接字
	if(sockfd < 0)
	{
		perror("socket error");
		exit(-1);

	}

	

	bzero(&my_addr, sizeof(my_addr));	   // 初始化服务器地址
	my_addr.sin_family = AF_INET;
	my_addr.sin_port   = htons(port);
	my_addr.sin_addr.s_addr = htonl(INADDR_ANY);

	printf("Binding server to port %d\n", port);

	

	// 绑定

	err_log = bind(sockfd, (struct sockaddr*)&my_addr, sizeof(my_addr));
	if(err_log != 0)
	{
		perror("bind");
		close(sockfd);		
		exit(-1);
	}

	
	// 监听，套接字变被动
	err_log = listen(sockfd, 10);
	if( err_log != 0)
	{

		perror("listen");
		close(sockfd);		
		exit(-1);
	}

	printf("Waiting client...\n");

	while(1)
	{

		char cli_ip[INET_ADDRSTRLEN] = "";	   // 用于保存客户端IP地址
		struct sockaddr_in client_addr;		   // 用于保存客户端地址
		socklen_t cliaddr_len = sizeof(client_addr);   // 必须初始化!!!

		// 上锁，在没有解锁之前，pthread_mutex_lock()会阻塞
		pthread_mutex_lock(&mutex);	
		//获得一个已经建立的连接	
		connfd = accept(sockfd, (struct sockaddr*)&client_addr, &cliaddr_len);   							

		if(connfd < 0)
		{
			perror("accept this time");
			continue;

		}
		// 打印客户端的 ip 和端口

		inet_ntop(AF_INET, &client_addr.sin_addr, cli_ip, INET_ADDRSTRLEN);
		printf("----------------------------------------------\n");

		printf("client ip=%s,port=%d\n", cli_ip,ntohs(client_addr.sin_port));

		if(connfd > 0)
		{

			//给回调函数传的参数，&connfd，地址传递
			pthread_create(&thread_id, NULL, (void *)client_process, (void *)&connfd);  //创建线程
			pthread_detach(thread_id); // 线程分离，结束时自动回收资源

		}

	}

	close(sockfd);
	return 0;

}