C ++同时接收2个或更多UDP消息

时间:2015-01-16 03:24:24

标签: c++ sockets udp memset

我正在尝试在main.cxx中接收UDP消息。我创建了一个UDP服务器方法getUdp(char * buffer)来侦听我的while(true)无限循环中传入的UDP消息。

这是我面临的问题。该UDP服务器能够一次侦听一条消息。当两个或多个UDP消息在相同的时间到来时,它不会排队到缓冲区中。我认为这是因为套接字是每次在无限循环中调用方法时,getUdp()方法打开一个套接字,获取消息并关闭套接字,导致服务器无法对消息进行排队。

如何调整此代码以接收2条或更多UDP消息? 感谢任何建议。

感谢。

UdpServer.cpp

#include <stdio.h>
#include <iostream>
#include <sstream>
#include <stdlib.h>
#include <string.h>
#include <winsock.h>
#include <time.h>

#include "UdpServer.h"
#include "stdafx.h"

using namespace std;

#pragma comment(lib, "ws2_32.lib")
#define BUFFER_SIZE 4096

void getUDP(char *buffer);

void UDPServer::MyUDP::getUDP(char *buffer)
{
    WSADATA w;                          /* Used to open windows connection */
    int client_length;                  /* Length of client struct */
    int bytes_received;                 /* Bytes received from client */
    SOCKET sd;                          /* Socket descriptor of server */
    struct sockaddr_in server;          /* Information about the server */
    struct sockaddr_in client;          /* Information about the client */
    struct hostent *hp;                 /* Information about this computer */
    char host_name[256];                /* Name of the server */
    time_t current_time;                /* Current time */

    /* Open windows connection */
    if (WSAStartup(0x0101, &w) != 0)
    {
        fprintf(stderr, "Could not open Windows connection.\n");
    }

    /* Open a datagram socket */
    sd = socket(AF_INET, SOCK_DGRAM, 0);
    if (sd == INVALID_SOCKET)
    {
        fprintf(stderr, "Could not create socket.\n");
        WSACleanup();
    }

    /* Clear out server struct */
    memset((void *)&server, '\0', sizeof(struct sockaddr_in));

    /* Set family and port */
    server.sin_family = AF_INET;
    server.sin_port = htons(11000);

    /* Set address automatically if desired */
    /* Get host name of this computer */
    gethostname(host_name, sizeof(host_name));
    hp = gethostbyname(host_name);

    /* Check for NULL pointer */
    if (hp == NULL)
    {
        fprintf(stderr, "Could not get host name.\n");
        closesocket(sd);
        WSACleanup();
    }

    unsigned int a = 127;
    unsigned int b = 0;
    unsigned int c = 0;
    unsigned int d = 1;

    /* Assign the address */
    server.sin_addr.S_un.S_un_b.s_b1 = a;
    server.sin_addr.S_un.S_un_b.s_b2 = b;
    server.sin_addr.S_un.S_un_b.s_b3 = c;
    server.sin_addr.S_un.S_un_b.s_b4 = d;

    /* Bind address to socket */
    if (bind(sd, (struct sockaddr *)&server, sizeof(struct sockaddr_in)) == -1)
    {
        fprintf(stderr, "Could not bind name to socket.\n");
        closesocket(sd);
        WSACleanup();
    }

    /* Print out server information */
    printf("Server running on %u.%u.%u.%u\n", (unsigned char)server.sin_addr.S_un.S_un_b.s_b1,
                                                (unsigned char)server.sin_addr.S_un.S_un_b.s_b2,
                                                (unsigned char)server.sin_addr.S_un.S_un_b.s_b3,
                                                (unsigned char)server.sin_addr.S_un.S_un_b.s_b4);
    printf("Press CTRL + C to quit\n");

    /* Loop and get data from clients */
    client_length = (int)sizeof(struct sockaddr_in);

    /* Receive bytes from client */
    bytes_received = recvfrom(sd, buffer, BUFFER_SIZE, 0, (struct sockaddr *)&client, &client_length);

    if (bytes_received < 0)
    {
        fprintf(stderr, "Could not receive datagram.\n");
        closesocket(sd);
        WSACleanup();
    }

    current_time = time(NULL);
    closesocket(sd);
    WSACleanup();
}

main.cxx

int main(int argc, char** argv)
{
    while(true)
    {
        //! wait for UDP message
        UDPServer::MyUDP myudp;
        myudp.getUDP(buffer);

        if(buffer[0] != 0)
        {
            string udpMsg(buffer);

            if(udpMsg == "ProcessThisMessage")
            {
                memset(&buffer[0], 0, sizeof(buffer));

                cout << "UDP Message: " + udpMsg;
            }

            ...
        }
    }
}

1 个答案:

答案 0 :(得分:3)

  

我认为这是因为套接字是每次调用方法时   在无限循环中,getUdp()方法打开一个套接字,得到   消息并关闭套接字,导致服务器无法使用   排队邮件。

你的直觉是正确的。当你有一个绑定到端口的UDP套接字时,网络堆栈将为你缓冲(有限数量)传入的UDP数据包,这样(假设你以相对及时的方式调用recv()),不应该得到传入的数据包丢失。但是当你关闭socket()时,缓冲区被释放,当然在没有套接字被绑定到UDP端口并且收到UDP数据包的时候,传入的UDP数据包将被简单地丢弃(即根本不被缓冲) )因为没有套接字绑定到该端口。

  

如何调整此代码以接收2条或更多UDP消息?   感谢任何建议。

从概念上讲,至少,您需要将getUdp()方法拆分为三个独立的部分:一个Setup()部分,在程序启动时调用一次,一个Receive()部分(仅包含) recv()调用,您可以根据需要多次调用,接收下一个数据包,最后一个Cleanup()部分关闭套接字并关闭TCP堆栈(只有当您的程序时才会调用它)即将退出)。这样,UDP套接字在程序运行的整个过程中保持有效并绑定到端口,这样OS就可以可靠地缓冲传入的UDP数据包,通过recv()为程序提供。