Winsock2听循环

时间:2014-01-04 08:57:21

标签: c++ multithreading loops winsock

你好再次精彩的stackoverflow社区!上次我问了一个问题,它很快得到解决,而且我希望这会以同样的方式进行。 :)

所以我玩弄了winsock,我希望能够同时将多个客户端连接到我的服务器。我觉得这可以通过每次客户端连接时创建线程,套接字创建,绑定和监听的循环来实现,但是我这样做的努力只出现了“listen failed with error”。两个不同的取决于我尝试过的。 我用谷歌搜索它只是为了找到最先进的MSDN样本,这是最容易的,但仍然太难的例子。 有人有什么简单的建议吗? (旁边的问题:我似乎无法获得“mrecv()”来返回整个“recvbuf”变量。我得到的只是一个字母。我知道这是我正在制作的新手错误,但我无法弄清楚。:/这个问题可以等到以后。)

(这是目前为止的服务器代码:)

#undef UNICODE

#define WIN32_LEAN_AND_MEAN

#include <windows.h>
#include <winsock2.h>
#include <ws2tcpip.h>
#include <stdlib.h>
#include <stdio.h>
#include <iostream>

// Need to link with Ws2_32.lib
#pragma comment (lib, "Ws2_32.lib")
// #pragma comment (lib, "Mswsock.lib")

int minitialize();
int msend(char msendbuf[512]);
char mrecv();
int mshutdown();

#define DEFAULT_BUFLEN 512
#define DEFAULT_PORT "10150"
 WSADATA wsaData;
 int iResult;

 SOCKET ListenSocket = INVALID_SOCKET;
 SOCKET ClientSocket = INVALID_SOCKET;

 struct addrinfo *result = NULL;
 struct addrinfo hints;

 int iSendResult;
 char recvbuf[DEFAULT_BUFLEN];
 int recvbuflen = DEFAULT_BUFLEN;

int main(void) 
{
    minitialize();
    mrecv();

    char mmessage[512];
    if (strncmp(mmessage,"shutdown",(strlen(mmessage))) == 0) {mshutdown();}
    std::cin.getline(mmessage, 512);
    msend(mmessage);

    // shutdown the connection since we're done
    mshutdown();
    std::cin.ignore();

    return 0;
}

int msend(char msendbuf[512])   // Send a message
{
    int iResult3 = send( ClientSocket, msendbuf, 512, 0 );
    if (iResult3 == SOCKET_ERROR) {
        printf("send failed with error: %d\n", WSAGetLastError());
        closesocket(ClientSocket);
        WSACleanup();
        return 1;
    }

    printf("Bytes Sent: %ld\n", iResult);
            std::cout<<"msendbuf: "<<msendbuf<<"\n";
            std::cin.ignore();
}

char mrecv() //Recieve a message
{
    int iResult2 = recv(ClientSocket, recvbuf, 512, 0);
        if (iResult2 > 0) {
            printf("Bytes received: %d\n", iResult2);
        std::cout<<"recvbuf: "<<recvbuf<<"\n";
        }
        else if (iResult2 == 0)
            printf("Connection closing...\n");
        else  {
            printf("recv failed with error: %d\n", WSAGetLastError());
            closesocket(ClientSocket);
            WSACleanup();
            return 1;
        }
    return *recvbuf;
}

int minitialize()   //initialize the winsock server
{
    // Initialize Winsock
    iResult = WSAStartup(MAKEWORD(2,2), &wsaData);
    if (iResult != 0) {
        printf("WSAStartup failed with error: %d\n", iResult);
        return 1;
    }

    ZeroMemory(&hints, sizeof(hints));
    hints.ai_family = AF_INET;
    hints.ai_socktype = SOCK_STREAM;
    hints.ai_protocol = IPPROTO_TCP;
    hints.ai_flags = AI_PASSIVE;

    // Resolve the server address and port
    iResult = getaddrinfo(NULL, DEFAULT_PORT, &hints, &result);
    if ( iResult != 0 ) {
        printf("getaddrinfo failed with error: %d\n", iResult);
        WSACleanup();
        return 1;
    }

    // Create a SOCKET for connecting to server
    ListenSocket = socket(result->ai_family, result->ai_socktype, result->ai_protocol);
    if (ListenSocket == INVALID_SOCKET) {
        printf("socket failed with error: %ld\n", WSAGetLastError());
        freeaddrinfo(result);
        WSACleanup();
        return 1;
    }

    // Setup the TCP listening socket
    iResult = bind( ListenSocket, result->ai_addr, (int)result->ai_addrlen);
    if (iResult == SOCKET_ERROR) {
        printf("bind failed with error: %d\n", WSAGetLastError());
        freeaddrinfo(result);
        closesocket(ListenSocket);
        WSACleanup();
        return 1;
    }

    freeaddrinfo(result);

    iResult = listen(ListenSocket, SOMAXCONN);
    if (iResult == SOCKET_ERROR) {
        printf("listen failed with error: %d\n", WSAGetLastError());
        closesocket(ListenSocket);
        WSACleanup();
        return 1;
    }

    // Accept a client socket
    ClientSocket = accept(ListenSocket, NULL, NULL);
    if (ClientSocket == INVALID_SOCKET) {
        printf("accept failed with error: %d\n", WSAGetLastError());
        closesocket(ListenSocket);
        WSACleanup();
        return 1;
    }

    // No longer need server socket
    closesocket(ListenSocket);
}

int mshutdown()     //shutdown the server
{
    iResult = shutdown(ClientSocket, SD_SEND);
    if (iResult == SOCKET_ERROR) {
        printf("shutdown failed with error: %d\n", WSAGetLastError());
        closesocket(ClientSocket);
        WSACleanup();
        return 1;
    }

    // cleanup
    closesocket(ClientSocket);
    WSACleanup();
    return 0;
}

是的,如果您想知道,我计划将所有这些全局变量放在各自的本地函数中。我现在只需要它们就可以清除一点杂乱,而且似乎没有引起任何问题。

2 个答案:

答案 0 :(得分:1)

mrecv()定义为

char mrecv()

在函数mrecv()中,它返回一个字符串。因为mrecv()定义应该返回一个char,所以返回recvbuf的第一个char。编译器不会抱怨return *recvbuf。即使意图返回单个字符,这在语法上也是正确的。

accept()将文件描述符返回给客户端的连接。返回accept()后,您可以生成一个新进程来使用此描述符处理消息,同时原始进程返回到接受连接。或者,您可以创建一个新线程来处理来自此描述符的消息,同时原始线程恢复接受连接。您需要循环accept()

这是一个可用于服务器的骨架模板。您已在minitialize()上正确填写了所有内容。在下面的示例代码中,mrecv()接受一个参数,即连接客户端的文件描述符newconn。

socket()
bind()
listen(mysock, 10);
while(keep listening) {
    newconn = accept(mysock, &peeraddr, &peeraddrlen)
    if(newconn > 0) {
        if(CreateThread(<security attribute>, <stack size>, (void *)&mrecv, (void *)&newconn, <creation flag>, <threadid>))
            perror("Unable to create thread\n");
    }
}
closesocket(mysock)

答案 1 :(得分:1)

  

我觉得这可以通过每次客户端连接时创建线程,创建套接字,绑定和监听的循环来实现

没有。您所需要做的就是接受客户端连接并使用该套接字启动一个线程。严格保留监听套接字。它不受accept()操作的影响,也不需要以任何方式重建它。