2020-08-21 09:05:32 +08:00
|
|
|
#include <stdio.h>
|
|
|
|
#include <stdlib.h>
|
|
|
|
#include <getopt.h>
|
|
|
|
#include "lib_acl.h"
|
|
|
|
#include "rfc1035.h"
|
|
|
|
|
|
|
|
static int __qid = 0;
|
|
|
|
|
|
|
|
static void build_request(const char *domain)
|
|
|
|
{
|
|
|
|
char *sbuf;
|
|
|
|
char buf[1024];
|
|
|
|
int ret;
|
|
|
|
|
|
|
|
memset(buf, 0, sizeof(buf));
|
|
|
|
|
|
|
|
ret = rfc1035BuildAQuery(domain, buf, sizeof(buf), __qid++, NULL);
|
|
|
|
if (ret <= 0) {
|
|
|
|
printf("rfc1035BuildAQuery error, domain=%s\r\n", domain);
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
printf(">>ret=%d\r\n", ret);
|
|
|
|
sbuf = (char*) acl_base64_encode(buf, ret);
|
|
|
|
printf(">>%s\r\n", sbuf);
|
|
|
|
acl_myfree(sbuf);
|
|
|
|
}
|
|
|
|
|
|
|
|
static void parse_result(const char *filepath)
|
|
|
|
{
|
|
|
|
rfc1035_message *answers;
|
|
|
|
ssize_t len;
|
|
|
|
int i, ret;
|
|
|
|
char *buf = acl_vstream_loadfile2(filepath, &len);
|
|
|
|
|
|
|
|
if (buf == NULL) {
|
|
|
|
printf("load from file %s error %s\r\n", filepath, acl_last_serror());
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
ret = rfc1035MessageUnpack(buf, (int) len, &answers);
|
|
|
|
if (ret <= 0) {
|
|
|
|
printf("invalid answers\r\n");
|
|
|
|
if (ret == 0) {
|
|
|
|
rfc1035MessageDestroy(answers);
|
|
|
|
}
|
|
|
|
acl_myfree(buf);
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
for (i = 0; i < ret; i++) {
|
|
|
|
if (answers->answer[i].type == RFC1035_TYPE_A) {
|
|
|
|
char ip[64];
|
|
|
|
struct in_addr sin_addr;
|
|
|
|
memcpy(&sin_addr, answers->answer[i].rdata, 4);
|
|
|
|
inet_ntop(AF_INET, &sin_addr, ip, sizeof(ip));
|
2020-09-01 17:09:26 +08:00
|
|
|
printf(">>ip=%s, ttl=%d\r\n", ip, answers->answer[i].ttl);
|
2020-08-25 10:22:27 +08:00
|
|
|
} else if (answers->answer[i].type == RFC1035_TYPE_CNAME) {
|
|
|
|
char name[128];
|
|
|
|
size_t n = sizeof(name) > answers->answer[i].rdlength
|
|
|
|
? answers->answer[i].rdlength : sizeof(name);
|
|
|
|
ACL_SAFE_STRNCPY(name, answers->answer[i].rdata, n);
|
2020-09-01 17:09:26 +08:00
|
|
|
printf(">>cname=%s, ttl=%d\r\n", name, answers->answer[i].ttl);
|
2020-08-21 09:05:32 +08:00
|
|
|
} else {
|
|
|
|
printf("not support type=%d\n", answers->answer[i].type);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
acl_myfree(buf);
|
|
|
|
}
|
|
|
|
|
|
|
|
static void usage(const char *procname)
|
|
|
|
{
|
2020-08-25 10:22:27 +08:00
|
|
|
printf("usage: %s -h [help] -e [encoding] -d [decoding] xxx\r\n", procname);
|
2020-08-21 09:05:32 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
int main(int argc, char *argv[])
|
|
|
|
{
|
|
|
|
int ch, encoding = 1;
|
|
|
|
ACL_VSTRING *buf = acl_vstring_alloc(64);
|
|
|
|
|
|
|
|
while ((ch = getopt(argc, argv, "hed")) > 0) {
|
|
|
|
switch (ch) {
|
|
|
|
case 'h':
|
|
|
|
usage(argv[0]);
|
|
|
|
acl_vstring_free(buf);
|
|
|
|
return 0;
|
|
|
|
case 'e':
|
|
|
|
encoding = 1;
|
|
|
|
break;
|
|
|
|
case 'd':
|
|
|
|
encoding = 0;
|
|
|
|
break;
|
|
|
|
default:
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
if (argv[optind] == NULL) {
|
|
|
|
usage(argv[0]);
|
|
|
|
return 1;
|
|
|
|
}
|
|
|
|
|
|
|
|
printf("xx=%s, optind=%d\r\n", argv[optind] ? argv[optind] : "null", optind);
|
|
|
|
acl_vstring_strcpy(buf, argv[optind]);
|
|
|
|
|
|
|
|
if (encoding) {
|
|
|
|
build_request(acl_vstring_str(buf));
|
|
|
|
} else {
|
|
|
|
parse_result(acl_vstring_str(buf));
|
|
|
|
}
|
|
|
|
|
|
|
|
acl_vstring_free(buf);
|
|
|
|
return (0);
|
|
|
|
}
|