emileclarkb/hammer
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

All helper functions for DNS parser done and compiling.

Browse source
This commit is contained in:
Meredith L. Patterson 2012-05-26 15:59:06 +02:00
parent 0c324adaa6
commit 3b336d0932
1 changed files with 20 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

23
examples/dns.c
View file

@ -81,6 +81,23 @@ char* get_domain(const HParsedToken *t) {
} }
} }
uint8_t* get_cs(const HCountedArray *arr) {
uint8_t *ret = h_arena_malloc(arr->arena, sizeof(uint8_t)*arr->used);
for (size_t i=0; i<arr->used; ++i)
ret[i] = arr->elements[i]->uint;
return ret;
}
uint8_t** get_txt(const HCountedArray *arr) {
uint8_t **ret = h_arena_malloc(arr->arena, sizeof(uint8_t*)*arr->used);
for (size_t i=0; i<arr->used; ++i) {
uint8_t *tmp = h_arena_malloc(arr->arena, sizeof(uint8_t)*arr->elements[i]->seq->used);
for (size_t j=0; j<arr->elements[i]->seq->used; ++j)
tmp[j] = arr->elements[i]->seq->elements[j]->uint;
}
return ret;
}
void set_rr(struct dns_rr rr, HCountedArray *rdata) { void set_rr(struct dns_rr rr, HCountedArray *rdata) {
uint8_t *data = h_arena_malloc(rdata->arena, sizeof(uint8_t)*rdata->used); uint8_t *data = h_arena_malloc(rdata->arena, sizeof(uint8_t)*rdata->used);
for (size_t i=0; i<rdata->used; ++i) for (size_t i=0; i<rdata->used; ++i)
@ -218,8 +235,8 @@ void set_rr(struct dns_rr rr, HCountedArray *rdata) {
if (!r) if (!r)
rr.type = 0; rr.type = 0;
else { else {
rr.hinfo.cpu = get_cs(r->ast->seq->elements[0]); rr.hinfo.cpu = get_cs(r->ast->seq->elements[0]->seq);
rr.hinfo.os = get_cs(r->ast->seq->elements[1]); rr.hinfo.os = get_cs(r->ast->seq->elements[1]->seq);
} }
break; break;
} }
@ -252,7 +269,7 @@ void set_rr(struct dns_rr rr, HCountedArray *rdata) {
rr.type = 0; rr.type = 0;
else { else {
rr.txt.count = r->ast->seq->elements[0]->seq->used; rr.txt.count = r->ast->seq->elements[0]->seq->used;
rr.txt.txt_data = get_txt(r->ast->seq->elements[0]); rr.txt.txt_data = get_txt(r->ast->seq->elements[0]->seq);
} }
break; break;
} }