I/O 复用完成单线程并发服务器的处 理和实现

I/O 复用完成单线程并发服务器的处 理和实现

实验目的

利用 I/O 复用实现一个单线程并发服务器， 使其可以并发接收客户端传来的数据， 并将收到的数据原封不动发回给客户端。
实验原理
1.(1)创建套接字并将其绑定在服务器的熟知端口上（本实验采用 12345 号端口），将改 套接字加入一个表中，该表中的项是可以进行 I/O 的描述符
(2)使用 select 在已有套接字上等待 I/O
(3)如果最初的套接字准备就绪，调用 accpet 获取下一个连接，并将该连接的套接字 加入到表中
(4)如果最初的套接字以外的套接字就绪，使用 read 获取下一个请求，构造响应，用write 将响应发回给客户端
(5)回到步骤 2
2.select 函数用于在非阻塞中，当一个套接字或一组套接字有信号时通知你，系统提供select 函数来实现多路复用输入/输出模型，原型：
参数 maxfd 是需要监视的最大的文件描述符值+1；
rdset,wrset,exset 分别对应于需要检测的可读文件描述符的集合，可写文件描述符 的集 合及异常文件描述符的集合。
struct timeval 结构用于描述一段时间长度，如果在这个时间内，需要监视的描述 符没有事件发生则函数返回，返回值为 0。
fd_set 是一组文件描述字(fd)的集合，它用一位来表示一个 fd，对于 fd_set 类型通 过下面四个宏来操作：
FD_ZERO(fd_set *fdset);将指定的文件描述符集清空，在对文件描述符集合进行设 置前，必须对其进行初始化，如果不清空，由于在系统分配内存空间后，通常并不作清空处理，所以结果是不可知的。
FD_SET(fd_set *fdset);用于在文件描述符集合中增加一个新的文件描述符。
FD_CLR(fd_set *fdset);用于在文件描述符集合中删除一个文件描述符。
FD_ISSET(int fd,fd_set *fdset);用于测试指定的文件描述符是否在该集合中。
timeout 用于描述一段时间长度，如果在这个时间内，需要监视的描述符没有事件发生则函数返回，返回值为 0。 有三种可能：
1.timeout=NULL（阻塞：select 将一直被阻塞，直到某个文件描述符上发生了事 件）
2.timeout 所指向的结构设为非零时间（等待固定时间：如果在指定的时间段里有 事件发生或者时间耗尽，函数均返回）
3.timeout 所指向的结构，时间设为 0（非阻塞：仅检测描述符集合的状态，然后 立即返回，并不等待外部事件的发生）还有 getdtablesize()，它是用来返回这个进程的文件描述表的项数，也就是说这个进 程打开的文件的数目

实验内容

服务器端：

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <pthread.h>
#define PORT 1028
#define LISTENQ 10
#define BUFFSIZE 1024 

// echo 服务函数
int echo(int s_c) {
	ssize_t s_z = 0;
	char buffer[BUFFSIZE];
	char buff_t[100] = { 0 };
	// 清零
	memset(buff_t, 0, sizeof(buff_t));
	memset(buffer, 0, BUFFSIZE);
	// 等待客服端发送数据
	s_z = read(s_c, buffer, sizeof(buffer)); 
	sprintf(buff_t, " :%d\n", s_c);
	strcpy(buffer + strlen(buffer) - 1, buff_t);
	// 反馈给客户端数据
	write(s_c, buffer, strlen(buffer)+1); 
	
	// 返回个数
	return s_z;
}

int passiveTCP() {
	// 创建一个套接字
	int fd = socket(AF_INET, SOCK_STREAM, 0);
	// 创建一个地址结构体
	struct sockaddr_in addr;
	// 地址结构体初始化
	addr.sin_family = AF_INET; 
	addr.sin_port = htons(PORT); 
	addr.sin_addr.s_addr = INADDR_ANY;
	// 端口ip绑定
	bind(fd, (struct sockaddr*)&addr, sizeof(addr)); 
	// 监听
	listen(fd, LISTENQ);

	return fd;
}
int main() {
	int serv_fd = passiveTCP(); 
	int cli_fd;
	struct sockaddr_in cli_addr; 
	socklen_t len = sizeof(cli_addr);
	//下列函数都在实验原理中有说明
	fd_set rfds; // 可读文件描述符的集合 
	fd_set afds; // 用于保存创建的套接字描述符的集合
	int temp, nfds; 
	FD_ZERO(&afds); // 将指定文件描述符集合清空
	FD_SET(serv_fd, &afds); // 在文件描述集合中增加一个新的文件描述符
	nfds = getdtablesize(); // 返回这个进程的文件描述表的项数
//	printf("%d\n", nfds);
	while (1) {
			
		//printf("等待连接\n");
		//调用 select。初始套接字就绪后调用 accpet
		memcpy(&rfds, &afds, sizeof(rfds));
		// 非阻塞
		if (select(nfds, &rfds, (fd_set*)0, (fd_set*)0, (struct timeval*)0) < 0) { 
			printf("select error\n");
			continue;
		}
		else {
			printf("select\n");
		}
		// 测试指定文件描述符是否在该集合中
		if (FD_ISSET(serv_fd, &rfds)) {
			cli_fd = accept(serv_fd, (struct sockaddr*)&cli_addr, &len); 
			if (cli_fd < 0) {
				printf("accept error\n");
			}
			else {
				printf("accept success\n");
				FD_SET(cli_fd, &afds); //当 accpet 成功时将新的套接字加入表中
			}
		}
		for (temp = 0; temp < nfds; temp++) {
			if (temp != serv_fd && FD_ISSET(temp, &rfds)) {	
				if (echo(temp) == 0) {//初始套接字以外的套接字就绪后调用构造响应
					close(temp); // 处理结束关闭客户端通信套接字 
					FD_CLR(temp, &afds); // 在文件描述符集合中删除一个文件描述符
				}
			}
		}
	}
	close(serv_fd); 
	return 0;
}

客户端：

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <unistd.h>
#define PORT 1028
#define IP "127.0.0.1" 

int main(int argc, char *argv[]) {
	//创建一个套接字
	int fd = socket(AF_INET, SOCK_STREAM, 0); 
	struct sockaddr_in addr; // 定义一个地址结构体
	//初始化地址结构体
	addr.sin_family = AF_INET;
	addr.sin_port = htons(PORT);
	inet_pton(AF_INET, IP, &(addr.sin_addr.s_addr));
	// 连接请求
	if (connect(fd, (struct sockaddr*)&addr, sizeof(addr)) < 0) { 
		printf("error\n");
	}
	else {
		printf("success\n");
	}
	char buffer[1024]; 
	ssize_t size = 0;
	// 清零
	memset(buffer, 0 , sizeof(buffer));
	printf("please input msg to server:\n");
	// 从stdio 读取数据
	size = read(0, buffer, sizeof(buffer));
	
	// 验证并发性
	if( size > 0 && argc > 1){
		int sleep_s = 0;
		printf("收到客服端反馈，休息%ss\n", argv[1]);
		sleep_s = atoi(argv[1]);
		sleep(sleep_s);
		printf("休息结束\n");
	}
	// 发送给服务器
	write(fd, buffer, size);
	// 获取服务器的回发
	size = read(fd, buffer, sizeof(buffer)); 
	printf("the msg from server is:\n");
	// 将buffer 里面的内容写到 stdout
	write(1, buffer, size);
	close(fd);
	return 0;
}

实验结果：