Linux下TCP/IP网络编程示例——实现服务器/客户端通信（四）

Linux下TCP/IP网络编程示例——实现服务器/客户端通信（四）：基于socket实现基本的服务器与客户端通信，使用多路复用（select）实现服务端的多并发功能。

客户端代码同之前一样。

服务端 server.c

#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <stdlib.h>
#include <errno.h>
#include <signal.h>
#include <fcntl.h>
#include <arpa/inet.h>
#include <sys/time.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>

#define SERVER_PORT		6000
#define MAX_CLIENT_NUM	100

int listen_fd = -1;
int g_client_fd[MAX_CLIENT_NUM];
unsigned int g_client_num = 0;

static void set_sockfd_nonblock(int fd);

/*
** 按Ctrl+C退出server程序时，执行下面操作关闭监听套接字
*/
void signal_handler(int arg)
{
	printf("close listen_fd(signal = %d)\n", arg);
	close(listen_fd);
	exit(0);
}

int main(int argc, const char *argv[])
{
	int i = 0;
	int new_fd = -1;
	int maxfd = -1;
	fd_set readfds, errorfds, allfds;
	struct timeval timeout;
	struct sockaddr_in server;
	struct sockaddr_in client;
	socklen_t saddrlen = sizeof(server);
	socklen_t caddrlen = sizeof(client);

	signal(SIGINT, signal_handler);

	for (i=0; i<MAX_CLIENT_NUM; i++)
	{
		g_client_fd[i] = -1;
	}
	memset(&server, 0, sizeof(server));
	memset(&client, 0, sizeof(client));
	memset(&timeout, 0, sizeof(timeout));

	FD_ZERO(&readfds);
	FD_ZERO(&errorfds);
	FD_ZERO(&allfds);

	listen_fd = socket(AF_INET, SOCK_STREAM, 0);
	if (listen_fd < 0)
	{
		printf("socket error!\n");
		return -1;
	}

	server.sin_family = AF_INET;
	server.sin_port = htons(SERVER_PORT);
	server.sin_addr.s_addr = INADDR_ANY;

	if (bind(listen_fd, (struct sockaddr *)&server, saddrlen) < 0)
	{
		printf("bind error!\n");
		return -1;
	}

	if (listen(listen_fd, 5) < 0)
	{
		printf("listen error!\n");
		return -1;
	}

	maxfd = listen_fd;

	int ret = 0;
	size_t rsize = 0;
	char rbuf[200] = {0};
	while (1)
	{
		timeout.tv_sec = 1;
		timeout.tv_usec = 0;

		FD_SET(listen_fd, &allfds);
		memcpy(&readfds, &allfds, sizeof(fd_set));
		memcpy(&errorfds, &allfds, sizeof(fd_set));

		ret = select(maxfd + 1, &readfds, NULL, &errorfds, &timeout);
		if (ret < 0)
		{
			perror("select error");
			continue;
		}
		else if (ret == 0)
		{
			continue;
		}

		if (FD_ISSET(listen_fd, &readfds))
		{
			new_fd = accept(listen_fd, (struct sockaddr *)&client, &caddrlen);
			if (new_fd < 0)
			{
				perror("accept error");
				return -1;
			}
			printf("new client connected(%d).IP:%s,port:%u\n", new_fd, inet_ntoa(client.sin_addr), ntohs(client.sin_port));
			
			set_sockfd_nonblock(new_fd);

			for (i=0; i<MAX_CLIENT_NUM; i++)
			{
				if (g_client_fd[i] == -1)
				{
					g_client_fd[i] = new_fd;
					g_client_num++;
					break;
				}
			}

			if (i >= MAX_CLIENT_NUM)
			{
				printf("reached the maximum number of connections!\n");
				close(new_fd);
			}
			else
			{
				if (new_fd > maxfd)
				{
					maxfd = new_fd;
				}
				FD_SET(new_fd, &allfds);
			}
		}

		for (i=0; i<MAX_CLIENT_NUM; i++)
		{
			if ((g_client_fd[i] != -1) && (FD_ISSET(g_client_fd[i], &readfds)))
			{
				rsize = read(g_client_fd[i], rbuf, sizeof(rbuf));
				if (rsize < 0)
				{
					printf("read error!\n");
					continue;
				}
				else if (rsize == 0)
				{
					printf("client (fd = %d) is closed!\n", g_client_fd[i]);
					FD_CLR(g_client_fd[i], &allfds);
					close(g_client_fd[i]);
					g_client_fd[i] = -1;
					g_client_num--;
					continue;
				}
				else
				{
					printf("recv from link=%d:%s\n", g_client_fd[i], rbuf);
				}
			}

			if ((g_client_fd[i] != -1) && (FD_ISSET(g_client_fd[i], &errorfds)))
			{
				printf("client fd(%d) error!\n", g_client_fd[i]);
				FD_CLR(g_client_fd[i], &allfds);
				close(g_client_fd[i]);
				g_client_fd[i] = -1;
				g_client_num--;
			}
		}

		if (g_client_num == 0)
		{
			printf("all client has closed!\n");
		}
	}

	close(listen_fd);

	return 0;
}

/*
** 设置套接字为非阻塞模式
*/
static void set_sockfd_nonblock(int fd)
{
	int flag = -1;

	flag = fcntl(fd, F_GETFL, 0);
	flag |= O_NONBLOCK;
	fcntl(fd, F_SETFL, flag);
}