#include <sys/types.h>
#include <sys/socket.h>
#include <sys/time.h>
#include <netinet/in.h>
#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <fcntl.h>
#include <algorithm>
#include <set>
using namespace std;

int main()
{
int listener;
struct sockaddr_in addr;
char buf[1024];
int bytes_read;
listener = socket(AF_INET, SOCK_STREAM, 0);
if(listener < 0)
    {
     perror("socket");
     exit(1);
    }
fcntl(listener, F_SETFL, O_NONBLOCK);
addr.sin_family = AF_INET;
addr.sin_port = htons(8001);
addr.sin_addr.s_addr = INADDR_ANY;
if(bind(listener, (struct sockaddr *)&addr, sizeof(addr)) < 0)
    {
     perror("bind");
     exit(2);
    }
listen(listener, 2);
set<int> clients;
clients.clear();
while(1)
    {
     // Заполняем множество сокетов
     fd_set readset;
     FD_ZERO(&readset);
     FD_SET(listener, &readset);
     for(set<int>::iterator it = clients.begin(); it != clients.end(); it++)
        FD_SET(*it, &readset);
     // Задаём таймаут
     timeval timeout;
     timeout.tv_sec = 15;
     timeout.tv_usec = 0;
     // Ждём события в одном из сокетов
     int mx = max(listener, *max_element(clients.begin(), clients.end()));
     if(select(mx+1, &readset, NULL, NULL, &timeout) <= 0)
        {
         perror("select");
         exit(3);
        }
     // Определяем тип события и выполняем соответствующие действия
     if(FD_ISSET(listener, &readset))
        {
         // Поступил новый запрос на соединение, используем accept
         int sock = accept(listener, NULL, NULL);
         if(sock < 0)
            {
             perror("accept");
             exit(3);
            }
        fcntl(sock, F_SETFL, O_NONBLOCK);
        clients.insert(sock);
        }
for(set<int>::iterator it = clients.begin(); it != clients.end(); it++)
        {
         if(FD_ISSET(*it, &readset))
            {
             memcpy(buf,"\0",1024);
             // Поступили данные от клиента, читаем их
             bytes_read = recv(*it, buf, 1024, 0);
             buf[bytes_read-1]=0;
             printf("Get %d bytes:%s\n",bytes_read-1,buf);
             if(!strcmp(buf,"quit"))
	     {
	      printf("Exiting...\n");
	      exit(1);
	     }
             if(bytes_read <= 0)
                {
                 // Соединение разорвано, удаляем сокет из множества
                 close(*it);
                 clients.erase(*it);
                 continue;
                }
             // Отправляем данные обратно клиенту
             send(*it, buf, bytes_read, 0);
            }
        }
}
return 0;
}