select 示例

server

/**
 * select server端测试代码
 */
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <netinet/in.h>
#include <errno.h>
#include <sys/types.h>
#include <sys/socket.h> 
#include <time.h>
 
#define PORT 8888
#define ADDR "127.0.0.1"
#define MAX_FD 1024

int server_init()
{
	struct sockaddr_in server_addr, client_addr;
	socklen_t addr_len;
	int listen_fd, client_fd;
	fd_set rfds;
	struct timeval timeout;
	int result;
	int maxfd;
	
	listen_fd = socket(AF_INET, SOCK_STREAM, 0);
	if (listen_fd < 0) {
		perror("create listen fd fail\n");
		return -1;
	}
	
	server_addr.sin_addr.s_addr = inet_addr(ADDR);
	server_addr.sin_family = AF_INET;
	server_addr.sin_port = htons(PORT);
	if (bind(listen_fd, (struct sockaddr*)&server_addr, sizeof(server_addr)) < 0) {
		perror("bind fail\n");
		return -1;
	}
	listen(listen_fd, 5);
	timeout.tv_sec = 30;
	timeout.tv_usec = 0;
	FD_ZERO(&rfds);
	FD_SET(listen_fd, &rfds);
	maxfd = MAX_FD;
	fprintf(stdout, "server start, listen on port %u....\n", PORT);
	while(1) {
		int i = 0;
		result = select(maxfd, &rfds, NULL, NULL, NULL);
		if (result < 0) {
			perror("select fail\n");
			return -1;
		}
		if (result == 0) {
			fprintf(stdout, "select timeout\n");
			continue;
		}
		for (i = 0;i <= result; i++) {
			if (FD_ISSET(listen_fd, &rfds)) {
				fprintf(stdout, "server fd can read\n");
				client_fd = accept(listen_fd, (struct sockaddr*)&client_addr,
							&addr_len);
				if (client_fd < 0) {
					perror("accept fail\n");
					return -1;
				}
				fprintf(stdout, "server recv a client, addr(%s), port(%u)\n",
					inet_ntoa(client_addr.sin_addr.s_addr), ntohs(client_addr.sin_port));
				FD_SET(client_fd, &rfds);
			if (FD_ISSET(client_fd, &rfds)) {
				fprintf(stdout, "client can read:\n");
				close(client_fd);
			}
			}
		}
		
	}
	
}
int main()
{
	server_init();
}
 

client：

#include <sched.h>
#include <stdlib.h>
#include <stdio.h>
#include <unistd.h>
#include <sys/epoll.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#define SERV_PORT 8888
#define SERV_ADDR "127.0.0.1"
#define BUF_SIZE 1024

void connect_server()
{
	int client_fd;
	struct sockaddr_in serv_addr;
	char send_buf[BUF_SIZE] = {0};
	
	serv_addr.sin_family = AF_INET;
	serv_addr.sin_port = htons(SERV_PORT);
	serv_addr.sin_addr.s_addr = inet_addr(SERV_ADDR);

	client_fd = socket(AF_INET, SOCK_STREAM, 0);
	if (client_fd < 0)
	{
		fprintf(stderr, "create socket fail\n");
		exit(1);
	}
	if (connect(client_fd, (struct sockaddr*)&serv_addr, sizeof(serv_addr)) < 0)
	{
		fprintf(stderr, "connect to server fail\n");
		exit(1);
	}
	close(client_fd);
	sleep(9);
	printf("client close\n");
	
}

int main()
{
	connect_server();
}