управление несколькими сокетами с помощью select () - PullRequest
Новая
       10

управление несколькими сокетами с помощью select ()

0 голосов
/ 15 мая 2018

У меня возникли проблемы с выбором функции select () для корректной работы с несколькими клиентами и получения от них сообщений без блокировки для каждой операции.

Я узнал, что select определяет состояние одного или нескольких сокетов, поэтому каждый раз, когда соединение было принято, я помещал сокет в набор, используя макросы, чтобы определить, какой из них готов к чтению. Я заметил, что мой код никогда не входит в функцию accept_clients (). Я хотел бы, чтобы кто-нибудь посмотрел на мой код и помог ему заставить его работать.

Вот мой код. 

#pragma comment(lib, "ws2_32.lib")
#include <WinSock2.h>
#include <stdio.h>
#include <signal.h>
#include <conio.h>
#include <assert.h>
#include <Windows.h>
#include <iostream>
//sockets
#define CLIENT_CON 10
#define CLIENT_DIS 20
#define BF_SZ 100
#define MAX_CONS 5
#define TIMEVAL_SEC 0
#define TIMEVAL_USEC 10
SOCKET sock, clien;
int PR_CONS = 0;

 struct _client
{
    bool con; // Set true if a client is connected
    sockaddr_in addr; // Client info like ip address
    SOCKET cs; // Client socket
    //fd_set set; // used to check if there is data in the socket
    int i; // any piece of additional info
};
_client client[10];
fd_set sset;

 int accept(_client*);
 int recv(_client*, char*, int);
 void Server_Status(int );
 void accept_clients();
 void recv_client();
int main() {
    //int res;
    int i = 1;
    int port = 5150;
    WSADATA ws;
    printf("\t Echo Server (Multiple client support)\n");
    sockaddr_in ser;
    ser.sin_family = AF_INET;
    ser.sin_addr.S_un.S_addr = INADDR_ANY;
    ser.sin_port = htons(port);
    WSAStartup(MAKEWORD(2,2),&ws);

    sock = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);

    setsockopt(sock, SOL_SOCKET, SO_REUSEADDR, (char*)&i,sizeof(i));
    bind(sock,(SOCKADDR*)&ser, sizeof(ser));
    listen(sock,5);
    printf("listening \n");
    unsigned long b= 1;
    ioctlsocket(sock,FIONBIO,&b);

    for(int i = 0; i < MAX_CONS; i++) {
        client[i].con = false;
    }
    while(true) {
        accept_clients();
        recv_client();
    }

}
 int accept(_client* x) {
     x->i = sizeof(sockaddr_in);
     int isClient = 0;
     unsigned long b= 1;
     FD_ZERO(&sset);
     FD_SET(sock, &sset);
     isClient = select(0, &sset, NULL,NULL, 0);
     if(isClient > 0 && PR_CONS <  MAX_CONS) {
         x->cs = accept(sock, (SOCKADDR*)&x->addr, &x->i);
         ioctlsocket(x->cs,FIONBIO,&b);
             if(x->cs != INVALID_SOCKET ) {
             x->con = true;
             FD_ZERO(&sset);
             FD_SET(x->cs,&sset);
             return true;
         }
     }
     //printf("failed to accept client");
     return false;
 }

 int recv(_client *x, char* buffer, int sz) {
     int r;
     /*timeval timeOut;
     timeOut.tv_sec = TIMEVAL_SEC;
     timeOut.tv_usec = TIMEVAL_USEC;*/
     r = select(0,&sset, NULL,NULL,0);
    if(r >  0) {
        if((FD_ISSET(x->cs,&sset)) ) {
            x->i = recv(x->cs,buffer,sz, 0);
            if(x->i > 0) {
                Server_Status(CLIENT_DIS);
                FD_CLR(x->cs,&sset);
                return 0;
            }
            return true;
          }
        }
         return false;
 }

 void accept_clients() {
     for(int i = 0; i < MAX_CONS;  i++) {
         if(!client[i].con) {
            if(accept(&client[i])) {
                Server_Status(CLIENT_CON);
            }
        }
    }
 }
void Server_Status(int msg) {
    if(msg == CLIENT_CON) {
        PR_CONS++;
        printf("client has connected");
    }
    else if(msg == CLIENT_DIS) {
        PR_CONS--;
        printf("client has disconnected");
    }
    else {
        printf("we got unknown message");
    }
}


void recv_client() {
    printf("problem?");
    char buffer[BF_SZ];
    for(int i = 0; i < MAX_CONS; i++) {
        if(client[i].con) {
            if(recv(&client[i],buffer, BF_SZ)) {
                std::cout << buffer << std::endl;
            }
        }
    }

}

1 Ответ

0 голосов
/ 15 мая 2018

Если количество клиентов не фиксировано (всегда MAX_CONS), и ничто не может быть заменено до того, как все они подключены), вы неправильно используете select.

Более распространенное использование (псевдокод) 

s = listening_socket
put s in a fd_set
loop on select
   if s is ready for read:
      new_con = accept
      store new_con in a client array
      put s in the fd_set
   else
      find one element of the read-ready set other than s (*)
      recv data from that client
      process data

(*) вы можете начать с первого элемента, в этом случае первые клиенты будут иметь немного более высокий приоритет, или начать с последнего использованного элемента и переходить к первому после последнего (циклический перебор) вдать всем клиентам одинаковый приоритет

...