我正在尝试在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
那应该没问题。为什么我的缓冲区不能保存所有字节?
答案 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);
}
}
}