C ++ recvfrom不接收所有(二进制)数据

时间:2012-09-13 18:34:41

标签: c++ linux dns binary

我正在尝试在Linux 2.6.32上的C ++中接收DNS请求(带有DNS标头的DNS内容)。当我使用netcat在UDP上写一个字符串时,代码有效。但如果我做“nslookup google.com”,它只会收到:

root@EliteBook:/home/.../dns# xxd request.bin  
0000000: 8a2b 01

如果我使用“tcpdump -x'udp port 53',它会向我显示更多字节。

我的C ++代码:

#include <string>
#include <iostream>
#include <string.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <stdio.h>
#include <stdlib.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <unistd.h>

#define BUFLEN 1024*6
#define NPACK 10
#define PORT 53

using namespace std;

void diep(const string &s)
{
        perror(s.c_str());
        exit(1);
}

int main()
{
        struct sockaddr_in si_me;
        struct sockaddr si_other;
        int slen=sizeof(si_other);
        int s;                                          // UDP Socket
        char buf[BUFLEN];
        buf[0] = 'X';
        buf[1] = 'Y';
        buf[2] = 'Z';

        if ((s=socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP))==-1)
                diep("Failed to create socket!");

        memset((char *) &si_me, 0, sizeof(si_me));
        si_me.sin_family = AF_INET;
        si_me.sin_port = htons(PORT);
        si_me.sin_addr.s_addr = htonl(INADDR_ANY);
        if (bind(s, (const sockaddr*) &si_me, sizeof(si_me))==-1)
                diep("bind");

        for (int i=0; i<NPACK; i++) {
                if (recvfrom(s, buf, BUFLEN, 0, &si_other, (socklen_t*) &slen)==-1)
                        diep("recvfrom()");

                struct sockaddr_in *si_other_in = (struct sockaddr_in *)&si_other;

                cout << "Pkg arrived: " << buf << " " << inet_ntoa(si_other_in->sin_addr) << ":" << ntohs(si_other_in->sin_port) << endl;

                FILE * pFile;
                pFile = fopen ("request.bin","wb");
                if (pFile!=NULL)
                {
                        fputs (buf, pFile);
                        fclose (pFile);
                }
        }
        close(s);

        return 0;
}

编译
g++ -Wall main.cpp -o dns

那么,我的错是什么,我忘记了什么或者我是否必须使用其他调用来读取二进制数据?

PS:Localhost是/etc/resolv.conf中的第一个名称服务器

修改

我看了一下recvfrom的返回值,它显示了我:

size: **27**
Pkg arrived: �. 127.0.0.1:36686
size: **45**
Pkg arrived: l 127.0.0.1:38952

那应该没问题。为什么我的缓冲区不能保存所有字节?

1 个答案:

答案 0 :(得分:2)

您大多忽略了recvfrom()的返回值。该返回值告诉您收到了多少字节。

来自man page

  

成功完成后,recvfrom()以字节

返回消息的长度

没有这些信息,您无法知道您没有收到所有字节。

  

那么,我的错是什么,我忘记了什么或者我是否必须使用其他调用来读取二进制数据?

你正在读取二进制数据,但是你正确地写它。您对std::cout << buf的呼叫和对fputs(buf, pFile)的呼叫均以字符串运行,而不是二进制数据。具体来说,它们每个都写入字节,但不包括第一个零值字节。

要修复您的计划,请记录来自recvfrom的答案,跳过对cout << buf的通话,并将fputs()替换为fwrite

// UNTESTED
for (int i=0; i<NPACK; i++) {
    int bufsize;
    if ( (bufsize = recvfrom(s, buf, BUFLEN, 0, &si_other, (socklen_t*) &slen))==-1)
        diep("recvfrom()");

        struct sockaddr_in *si_other_in = (struct sockaddr_in *)&si_other;

        cout << "Pkg arrived: " << bufsize << " " << inet_ntoa(si_other_in->sin_addr) << ":" << ntohs(si_other_in->sin_port) << endl;

        FILE * pFile;
        pFile = fopen ("request.bin","wb");
        if (pFile!=NULL)
        {
            fwrite(buf, 1, bufsize, pFile);
            fclose (pFile);
        }
    }
}