网络编程--day3--实现TCP并发服务器

一、使用多进程实现TCP的并发服务

#include <myhead.h>
#define SER_PORT 6688			// 服务器端口号
#define SER_IP "192.168.18.128" // 服务器ip地址

// 定义信号处理函数
void handler(int signo)
{
	// 判断要处理的信号
	if (signo == SIGCHLD)
	{										   // 将僵尸进程处理殆尽
		while (waitpid(-1, NULL, WNOHANG) > 0) // waitpid(-1,NULL,WNOHANG)
			;								   // -1是回收任意子进程,,NULL是不接收子进程死亡后的状态exti(0);NOHANG是非阻塞态
	}
}

int main(int argc, const char *argv[])
{
	// 将子进程的SIGCHLD（17）信号
	// 当子进程退出时，会向其父进程发送该信号
	if (signal(SIGCHLD, handler) == SIG_ERR)
	{
		perror("signal error");
		return -1;
	}

	// 1、为通信创建一个端点
	int sfd = socket(AF_INET, SOCK_STREAM, 0);
	// 参数1：说明使用的是ipv4通信域
	// 参数2：说明使用的是TCP面向连接的通信方式
	// 参数3：由于参数2中已经指定通信方式，填0即可
	if (sfd == -1)
	{
		perror("socket error");
		return -1;
	}
	printf("socket success sfd = %d\n", sfd); // 3

	// 调用端口号快速重用函数
	int reuse = 1;
	if (setsockopt(sfd, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof(reuse)) == -1)
	{
		perror("setsockopt error");
		return -1;
	}

	// 2、绑定ip和端口号
	// 2.1 准备地址信息结构体
	struct sockaddr_in sin;
	sin.sin_family = AF_INET;				 // 通信域
	sin.sin_port = htons(SER_PORT);			 // 端口号
	sin.sin_addr.s_addr = inet_addr(SER_IP); // ip地址

	// 2.2 绑定工作
	if (bind(sfd, (struct sockaddr *)&sin, sizeof(sin)) == -1)
	{
		perror("bind error");
		return -1;
	}
	printf("bind success\n");

	// 3、将套接字设置成被动监听状态
	if (listen(sfd, 128) == -1)
	{
		perror("listen error");
		return -1;
	}
	printf("listen success\n");

	// 4、阻塞等待客户端的连接
	// 4.1 定义用于接受客户端信息的容器
	struct sockaddr_in cin;
	socklen_t addrlen = sizeof(cin);
	int newfd = -1; // 客户端套接字变量

	while (1)
	{
		// 父进程
		newfd = accept(sfd, (struct sockaddr *)&cin, &addrlen);
		if (newfd == -1)
		{
			perror("accept error");
			return -1;
		}
		printf("[%s:%d]:发来连接请求\n", inet_ntoa(cin.sin_addr), ntohs(cin.sin_port));

		pid_t pid = fork(); // 创建子进程
		if (pid > 0)
		{
			// 父进程体
			// 关闭newfd
			close(newfd);
		}
		else if (pid == 0)
		{
			// 关闭sfd
			close(sfd);

			// 5、与客户端进行相互通信
			char rbuf[128] = ""; // 读取消息内容的容器

			while (1)
			{
				// 清空容器
				bzero(rbuf, sizeof(rbuf));

				// 从套接字中读取数据
				// int res = read(newfd, rbuf, sizeof(rbuf));
				int res = recv(newfd, rbuf, sizeof(rbuf), 0);
				if (res == 0)
				{
					printf("客户端已经下线\n");
					break;
				}

				// 将读取的消息展示出来
				printf("[%s:%d]:%s\n", inet_ntoa(cin.sin_addr), ntohs(cin.sin_port), rbuf);

				// 将收到的消息处理一下，回复给客户端
				strcat(rbuf, "*_*");

				// 讲消息发送给客户端
				// write(newfd, rbuf, strlen(rbuf));
				send(newfd, rbuf, strlen(rbuf), 0);
				printf("发送成功\n");
			}

			// 6、关闭套接字
			close(newfd);

			// 退出子进程
			exit(EXIT_SUCCESS);
		}
		else
		{
			perror("fork error");
			return -1;
		}
	} // while(1)
	close(sfd);

	return 0;
}

二、使用多线程实现TCP的并发服务

#include <myhead.h>
#define SER_PORT 6688			// 服务器端口号
#define SER_IP "192.168.18.128" // 服务器ip地址

// 定义一个结构体类型，用于向线程体函数传递参数
struct Info
{
	int newfd;
	struct sockaddr_in cin;
};

// 定义线程体函数
void *deal_cli_msg(void *arg)
{
	// 解析传递进来的数据
	int newfd = ((struct Info *)arg)->newfd;
	struct sockaddr_in cin = ((struct Info *)arg)->cin;

	// 5、与客户端进行相互通信
	char rbuf[128] = ""; // 读取消息内容的容器

	while (1)
	{
		// 清空容器
		bzero(rbuf, sizeof(rbuf));

		// 从套接字中读取数据
		// int res = read(newfd, rbuf, sizeof(rbuf));
		int res = recv(newfd, rbuf, sizeof(rbuf), 0);
		if (res == 0)
		{
			printf("客户端已经下线\n");
			break;
		}

		// 将读取的消息展示出来
		printf("[%s:%d]:%s\n", inet_ntoa(cin.sin_addr), ntohs(cin.sin_port), rbuf);

		// 将收到的消息处理一下，回复给客户端
		strcat(rbuf, "*_*");

		// 讲消息发送给客户端
		// write(newfd, rbuf, strlen(rbuf));
		send(newfd, rbuf, strlen(rbuf), 0);
		printf("发送成功\n");
	}

	// 6、关闭套接字
	close(newfd);

	// 退出线程
	pthread_exit(NULL);
}

int main(int argc, const char *argv[])
{
	// 1、为通信创建一个端点
	int sfd = socket(AF_INET, SOCK_STREAM, 0);
	// 参数1：说明使用的是ipv4通信域
	// 参数2：说明使用的是TCP面向连接的通信方式
	// 参数3：由于参数2中已经指定通信方式，填0即可
	if (sfd == -1)
	{
		perror("socket error");
		return -1;
	}
	printf("socket success sfd = %d\n", sfd); // 3

	// 2、绑定ip和端口号
	// 2.1 准备地址信息结构体
	struct sockaddr_in sin;
	sin.sin_family = AF_INET;				 // 通信域
	sin.sin_port = htons(SER_PORT);			 // 端口号
	sin.sin_addr.s_addr = inet_addr(SER_IP); // ip地址

	// 2.2 绑定工作
	if (bind(sfd, (struct sockaddr *)&sin, sizeof(sin)) == -1)
	{
		perror("bind error");
		return -1;
	}
	printf("bind success\n");

	// 3、将套接字设置成被动监听状态
	if (listen(sfd, 128) == -1)
	{
		perror("listen error");
		return -1;
	}
	printf("listen success\n");

	// 4、阻塞等待客户端的连接
	// 4.1 定义用于接受客户端信息的容器
	struct sockaddr_in cin;
	socklen_t addrlen = sizeof(cin);

	while (1)
	{
		// accept函数会预选一个当前未分配的最小的文件描述符
		// 即使在阻塞过程中，有更小的文件描述符产生，本次操作的文件描述符也不会更改了
		int newfd = accept(sfd, (struct sockaddr *)&cin, &addrlen);
		if (newfd == -1)
		{
			perror("accept error");
			return -1;
		}

		printf("[%s:%d]:发来连接请求, newfd = %d\n", inet_ntoa(cin.sin_addr), ntohs(cin.sin_port), newfd);

		// 定义要传递数据的结构体变量
		struct Info buf = {newfd, cin};

		// 创建分支线程，用于通信
		pthread_t tid = -1;
		if (pthread_create(&tid, NULL, deal_cli_msg, &buf) == -1)
		{
			printf("pthread_create error\n");
			return -1;
		}

		// 将线程设置成分离态
		pthread_detach(tid);
	}
	close(sfd);

	return 0;
}