高并发的socket实现

方法1  使用select 

方法2  使用epoll

首先linux平台下的socket最大连接数是1024，其实去除一些系统用的文字描述符，就会少于1024了，但是一般的服务器1000多个连接哪能够呢？用命令ulimit -n可以查询最大文件描述符 也可以改写 ulimit -n 65536

select是I/O多路转换技术

服务器端

#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 <stdlib.h>

int main()
{
    int server_sockfd, client_sockfd;
    int server_len, client_len;
    struct sockaddr_in server_address;
    struct sockaddr_in client_address;
    int result;
    fd_set readfds, testfds;
    char rec_buf[1024];


    server_sockfd = socket(AF_INET, SOCK_STREAM, 0);

    server_address.sin_family = AF_INET;
    server_address.sin_addr.s_addr = htonl(INADDR_ANY);
    server_address.sin_port = htons(5555);
    server_len = sizeof(server_address);

    bind(server_sockfd, (struct sockaddr *)&server_address, server_len);


    listen(server_sockfd, 5);

    FD_ZERO(&readfds);
    FD_SET(server_sockfd, &readfds);

    memset(rec_buf,0,1024);
    while(1) {
        char ch;
        int fd;
        int nread;

        testfds = readfds;

        printf("server waiting\n");
        result = select(FD_SETSIZE, &testfds, (fd_set *)0, 
            (fd_set *)0, (struct timeval *) 0);

        if(result < 1) {
            perror("server5");
            exit(1);
        }


        for(fd = 0; fd < FD_SETSIZE; fd++) {
            if(FD_ISSET(fd,&testfds)) {


                if(fd == server_sockfd) {
                    client_len = sizeof(client_address);
                    client_sockfd = accept(server_sockfd, 
                        (struct sockaddr *)&client_address, &client_len);
                    FD_SET(client_sockfd, &readfds);
                    printf("adding client on fd %d\n", client_sockfd);
                }


                else {
                    ioctl(fd, FIONREAD, &nread);

                    if(nread == 0) {
                        close(fd);
                        FD_CLR(fd, &readfds);
                        printf("removing client on fd %d\n", fd);
                    }

                    else {
                        read(fd, rec_buf, 100);
			printf("rec from %s",rec_buf);
                    }
                }
            }
        }
    }
}

客户机

/*  Make the necessary includes and set up the variables.  */

#include <sys/types.h>
#include <sys/socket.h>
#include <stdio.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <stdlib.h>

int main()
{
    int sockfd;
    int len;
    struct sockaddr_in address;
    int result;
    char ch = 'A';
    char snd_buf[1024];

/*  Create a socket for the client.  */

    sockfd = socket(AF_INET, SOCK_STREAM, 0);

/*  Name the socket, as agreed with the server.  */

    address.sin_family = AF_INET;
    address.sin_addr.s_addr = inet_addr("127.0.0.1");
    address.sin_port = htons(5555);
    len = sizeof(address);

/*  Now connect our socket to the server's socket.  */

    result = connect(sockfd, (struct sockaddr *)&address, len);

    if(result == -1) {
        perror(" client connect failed");
        exit(1);
    }

/*  We can now read/write via sockfd.  */
    memset(snd_buf,0,1024); 
    sprintf(snd_buf,"my pid is %d",getpid());
    len=strlen(snd_buf);
   if(len>0){
		write(sockfd,snd_buf,len);
    }
   while(1){
	sleep(100);   
   }
    close(sockfd);
    exit(0);
}

shell 启动代码

#!/bin/bash

./select &
for (( i=0;i<2200;i++))
do

	./client &

done

shell 结束代码

#!/bin/bash
killall client
killall select

epoll代码

#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 <stdlib.h>
#include <sys/epoll.h>


#define M_MAXFD 1024
struct epoll_event ev,events[M_MAXFD];
int epfd;

int main()
{
    int server_sockfd, client_sockfd;
    int server_len, client_len;
    struct sockaddr_in server_address;
    struct sockaddr_in client_address;
    int result;
    char rec_buf[1024];
    int nfds,i,sockfd;


    server_sockfd = socket(AF_INET, SOCK_STREAM, 0);

    server_address.sin_family = AF_INET;
    server_address.sin_addr.s_addr = htonl(INADDR_ANY);
    server_address.sin_port = htons(5555);
    server_len = sizeof(server_address);

    bind(server_sockfd, (struct sockaddr *)&server_address, server_len);


    listen(server_sockfd, 5);
    		
    epfd=epoll_create(M_MAXFD);
    ev.data.fd = server_sockfd;
    ev.events = EPOLLIN | EPOLLET;

    epoll_ctl(epfd,EPOLL_CTL_ADD,server_sockfd,&ev);



    memset(rec_buf,0,1024);
    while(1) {
        int nread;
	nfds = epoll_wait(epfd,events,1024,500);
	for(i=0;i<nfds;++i)
	{
		if(events[i].data.fd==server_sockfd)
		{
                        
                        client_len = sizeof(client_address);
			client_sockfd=accept(server_sockfd,(struct sockaddr * )&client_address,&client_len);
			if(client_sockfd<0)
			{
				perror("client_sockfd<0");
				exit(1);
			}
			printf("connect success %d \n",client_sockfd);
			ev.data.fd=client_sockfd;
			ev.events=EPOLLIN | EPOLLET;
			epoll_ctl(epfd,EPOLL_CTL_ADD,client_sockfd,&ev);
		}
		else if(events[i].events & EPOLLIN)
		{
			if((sockfd=events[i].data.fd)<0)continue;
			printf("reading %d \n",sockfd);
                       result=read(sockfd, rec_buf, 100);
		       if(result==0){
			     printf("client close %d \n",sockfd);
		       	     close(sockfd);
		       }
		       else{
		       
			printf("rec from %s\n",rec_buf);
		       }


			ev.events=EPOLLIN | EPOLLET;
			ev.data.fd=sockfd;
			epoll_ctl(epfd,EPOLL_CTL_MOD,sockfd,&ev);	
		
		}
		
	}	


    }
}