使用WinSock处理多个用户

时间:2018-05-11 17:57:50

标签: c++ sockets tcp winsock

我正在关注显示如何在不使用多线程的情况下接受多个连接的this tutorial

我遇到的问题是accept()函数始终返回SOCKET_ERROR

我希望服务器接受多个客户端。我想如果有人能指出我做错了什么。

这是我的服务器代码:

#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

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];

int accept(_client*);
int send(_client*, char*,int);
int recv(_client*, char*, int);
void Server_Status(int );
void char_message(char*);
void accept_clients();
void recv_client();

int main() {
    //int res;
    int i = 1;
    int port = 5150;
    SOCKET sock;
    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);
     x->cs = accept(sock, (SOCKADDR*)&x->addr, &x->i);
     if(x->cs != INVALID_SOCKET ) {
         std::cout << x->cs << std::endl;
         x->con = true;
         FD_ZERO(&x->set);
         FD_SET(x->cs,&x->set);
         printf("accepted client");
         return true;
     }
     //printf("failed to accept client");
     return false;
 }

 int send(_client* x, char* buffer, int sz) {
    x->i = send(x->cs, buffer, sz, 0);
    if(x-> i == SOCKET_ERROR || x->i == 0 ) {
        return false;
    }
    return true;
 }

 int recv(_client *x, char* buffer, int sz) {
     if(FD_ISSET(x->cs,&x->set)) {
        x->i = recv(x->cs,buffer,sz, 0);
        if(x->i == 0) {
            return false;
        }
        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 chat_message(char* s) {
    int len = strlen(s);
    for(int i = 0; i< MAX_CONS; i++) {
        if(client[i].con) {
            send(&client[i], s, len);
        }           
    }
}

void recv_client() {
    char buffer[BF_SZ];
    for(int i = 0; i < MAX_CONS; i++) {
        if(client[i].con) {
            if(recv(&client[i],buffer, BF_SZ)) {
                if(buffer[0] == '/') {
                    if(strcmp(buffer, "/server_bang") == 0) {
                        chat_message("** Hi**");
                    }                   
                }
                else {
                    chat_message(buffer);
                }
            }
        }       
    }
}

1 个答案:

答案 0 :(得分:2)

  

我遇到的问题是SOCKET_ERROR函数始终返回SOCKET

这是因为您传递了一个无效的WSAGetLastError()句柄(如果您在收到SOCKET_ERROR之后打算与WSAENOTSOCK核对,则可能会返回accept())。

原因是因为您的accept()包装函数正在使用名为SOCKET全局sock变量调用Winsock的main()函数,但您的{ {1}}函数永远不会初始化该变量!而是初始化一个名为SOCKET本地sock变量。你需要:

  • 删除main()的本地sock变量,以便main()使用全局sock变量。

  • 摆脱全局sock变量,让main()将其本地sock变量作为输入参数传递给accept_clients()函数,然后可以将其作为输入参数传递给accept()函数。

相关问题