网络通讯:select模型

//file: server.cpp
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <stdio.h>
#include <stdlib.h>
#include <strings.h>
#include <sys/wait.h>
#include <string.h>
#include <errno.h>
#define DEFAULT_PORT 6666

int main(int argc, char ** argv){
	int serverfd, acceptfd; 
	struct sockaddr_in my_addr;
	struct sockaddr_in their_addr; 
	unsigned int sin_size, myport = 6666, lisnum = 5;
	
	if((serverfd = socket(AF_INET, SOCK_STREAM, 0)) == -1){
		perror("socket");
		return -1;
	}
	printf("socket ok\n");
	my_addr.sin_family = AF_INET;
	my_addr.sin_port = htons(DEFAULT_PORT);
	my_addr.sin_addr.s_addr = INADDR_ANY;
	bzero(&(my_addr.sin_zero), 0);
	if((bind(serverfd, (struct sockaddr*)&my_addr, sizeof(struct sockaddr)) == -1){
		perror("bind");
		return -2;
	}
	printf("bind ok \n");
	if(listen(serverfd, lisnum) == -1){
		perror("listen");
		return -3;
	}
	printf("listen ok \n");
	
	fd_set client_fdset;     // 监控文件描述符集合
	int maxsock;             // 监控文件描述符中最大的文件号
	struct timeval tv;       // 超时返回时间
	int client_sockfd[5];    // 存放活动的socket
	bzero((void*)client_sockfd, sizeof(client_sockfd));
	int conn_amount = 0;     // 用来记录描述符数目
	maxsock = serverfd;
	char buffer[1024];
	int ret = 0;
	while(1){
		// 初始化文件描述符号到集合
		FD_ZERO(&client_fdset);
		// 加入服务器描述符
		FD_SET(serverfd, &client_fdset);	
		// 设置超时时间
		tv.tv_sec = 30;  // 30s
		tv.tv_usec = 0; 
		// 把活动的句柄加入到文件描述符中
		for(int i = 0; i< 5;i++){
			if(client_sockfd[i] != 0){
				FD_SET(client_sockfd[i], &client_fdset);
			}
		}
		
		// select函数
		ret = select(maxsock + 1, &client_fdset, NULL, NULL, &tv);
		if(ret < 0){
			perror("select error!\n");
			break;
		}else if(ret == 0){
			printf("timeout!\n");
			continue;
		}
		
		// 轮询各个文件描述符
		for(int i = 0;i< conn_amount; ++i){
			// FD_ISSET检查client_sockfd是否可读写,>0可读写
			if(FD_ISSET(client_sockfd[i], &client_fdset)){
				printf("start recv from client[%d]:\n", i);
				ret = recv(client_sockfd[i], buffer, 1024, 0);
				if(ret <= 0){
					printf(*client[%d] close\n", i);
					close(client_sockfd[i]);
					FD_CLR(client_sockfd[i], &client_fdset);
					client_sockfd[i] = 0;
				}else{
					printf("recv from client[%d]:%s\n", i, buffer);
				}
			}
		}
		// 检查是否又新的连接,如果又,接收连接,加入到client_sockfd中
		int sock_client = 0;
		if(FD_ISSET(serverfd, &client_fdset)){
			// 接收连接
			struct sockaddr_in client_addr;
			size_t size = sizeof(struct sockaddr_in);
			sock_client = accept(serverfd, (struct sockaddr*)(&client_addr), (unsigned int *)(&size));
			if(sock_client <= 0){
				perror("accept error !\n");
				continue;
			}
		}
		// 把连接加入都文件描述符集合中
		if(conn_amount < 5){
			client_sockfd[conn_amount++] = sock_client;
			bzero(buffer, 1024);
			strcpy(buffer, "this is server! welcome!\n");
			send(sock_client, buffer, 1024, 0);
			....
		}
	}
	
	// 关闭套接字
	for(int i = 0;i< 5;i++){
		if(client_sockfd[i] != 0){
			close(client_sockfd[i]);
		}
	}
	close(serverfd);
	return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值