基于select的回射服务器

server

#include <sys/types.h>
#include <sys/socket.h>
#include <stdio.h>
#include <netinet/in.h>
#include <sys/time.h>
#include <sys/ioctl.h>
#include <unistd.h>
#include <arpa/inet.h>
#include <string.h>
#include <stdlib.h> //exit
#define MAXLINE 1024
int main()
{
	int i, maxi, maxfd, listenfd, connfd, sockfd;
	int nready, client[FD_SETSIZE];
	ssize_t n;
	fd_set rset, allset;
	char buf[MAXLINE];
	socklen_t clilen;
	struct sockaddr_in cliaddr, servaddr;
	int result;

	listenfd = socket(AF_INET, SOCK_STREAM, 0); //建立服务器端socket
	bzero(&servaddr, sizeof(servaddr));
	servaddr.sin_family = AF_INET;
	servaddr.sin_addr.s_addr = htonl(INADDR_ANY);
	servaddr.sin_port = htons(9734);

	bind(listenfd, (struct sockaddr *)&servaddr, sizeof(servaddr));
	if (listen(listenfd, 5) == -1)
		exit(0);

	maxfd = listenfd; /* initialize */
	maxi = -1;		  /* index into client[] array 储存client[]用了多少位*/
	for (i = 0; i < FD_SETSIZE; i++)
		client[i] = -1; /* -1 indicates available entry */
	FD_ZERO(&allset);
	FD_SET(listenfd, &allset); //将服务器端socket加入到集合中

	while (1)
	{
		rset = allset;										 /* structure assignment */
		nready = select(maxfd + 1, &rset, NULL, NULL, NULL); // nready储存所有描述符集已就绪的总位数
		if (nready < 1)
		{
			perror("server5");
			exit(1);
		}

		if (FD_ISSET(listenfd, &rset))
		{ /* new client connection */
			clilen = sizeof(cliaddr);
			connfd = accept(listenfd, (struct sockaddr *)&cliaddr, &clilen);
			printf("new client: %s, port %d\n",
				   inet_ntop(AF_INET, &cliaddr.sin_addr, NULL, 4),
				   ntohs(cliaddr.sin_port)); //输出新连接的地址信息和端口号

			for (i = 0; i < FD_SETSIZE; i++)
				//将第一个-1的client[i]取出来存connfd
				if (client[i] < 0)
				{
					client[i] = connfd; /* save descriptor */
					break;
				}
			if (i == FD_SETSIZE)
				if (i == FD_SETSIZE)
					// FD_SETSIZE个的set都用完了，客户端连接过多了。
					perror("too many clients");

			FD_SET(connfd, &allset); /* add new descriptor to set */
			if (connfd > maxfd)
				maxfd = connfd; /* for select */
			if (i > maxi)
				maxi = i; /* max index in client[] array */

			if (--nready <= 0)
				continue; /* no more readable descriptors */
		}
		for (i = 0; i <= maxi; i++) {	/* check all clients for data */
			if ( (sockfd = client[i]) < 0)
				continue;//若client[i]=-1 继续循环
			if (FD_ISSET(sockfd, &rset)) {
				if ( (n = read(sockfd, buf, MAXLINE)) == 0) {
						/*若连接已被客户端关闭 */
					close(sockfd);
					FD_CLR(sockfd, &allset);
					client[i] = -1;
				} else
					write(sockfd, buf, n);

				if (--nready <= 0)
					break;				/* no more readable descriptors */
			}
		}
	}
}

client

#include <sys/types.h>
#include <sys/socket.h>
#include <stdio.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <string.h>
#include <stdlib.h> //exit
#include <errno.h>
#define MAXLINE 1024
int main()
{
    int client_sockfd;
    int len;
    struct sockaddr_in server; //服务器端网络地址结构体
    int result;
    char recvline[MAXLINE], sendline[MAXLINE];
    client_sockfd = socket(AF_INET, SOCK_STREAM, 0); //建立客户端socket
    // memset(&client_sockfd, 0, sizeof(sockaddr));
    server.sin_family = AF_INET;
    inet_pton(AF_INET, "127.0.0.1", &server.sin_addr);
    server.sin_port = htons(9734);

    len = sizeof(server);
    if (connect(client_sockfd, (struct sockaddr *)&server, len) == -1)
    {
        perror("oops: client2");
        exit(1);
    }

    while (1)
    {
        fgets(sendline, 1024, stdin);

        if ((send(client_sockfd, sendline, strlen(sendline), 0)) < 0)
        {
            printf("send mes error: %s errno : %d", strerror(errno), errno);
            exit(0);
        }
        int n = recv(client_sockfd, &recvline, MAXLINE, 0);
        recvline[n] = '\0';
        printf("recv msg from server:%s", recvline);
    }
    close(client_sockfd);
    exit(0);
}

结果