tb_long_t tb_charset_ascii_get(tb_static_stream_ref_t sstream, tb_bool_t be, tb_uint32_t* ch) { *ch = tb_static_stream_read_u8(sstream); return 1; }
static tb_bool_t tb_dns_looker_resp_done(tb_dns_looker_t* looker, tb_ipaddr_ref_t addr) { // the rpkt and size tb_byte_t const* rpkt = tb_static_buffer_data(&looker->rpkt); tb_size_t size = tb_static_buffer_size(&looker->rpkt); tb_assert_and_check_return_val(rpkt && size >= TB_DNS_HEADER_SIZE, tb_false); // init stream tb_static_stream_t stream; tb_static_stream_init(&stream, (tb_byte_t*)rpkt, size); // init header tb_dns_header_t header; header.id = tb_static_stream_read_u16_be(&stream); tb_static_stream_skip(&stream, 2); header.question = tb_static_stream_read_u16_be(&stream); header.answer = tb_static_stream_read_u16_be(&stream); header.authority = tb_static_stream_read_u16_be(&stream); header.resource = tb_static_stream_read_u16_be(&stream); // trace tb_trace_d("response: size: %u", size); tb_trace_d("response: id: 0x%04x", header.id); tb_trace_d("response: question: %d", header.question); tb_trace_d("response: answer: %d", header.answer); tb_trace_d("response: authority: %d", header.authority); tb_trace_d("response: resource: %d", header.resource); tb_trace_d(""); // check header tb_assert_and_check_return_val(header.id == TB_DNS_HEADER_MAGIC, tb_false); // skip questions, only one question now. // name + question1 + question2 + ... tb_assert_and_check_return_val(header.question == 1, tb_false); #if 1 tb_static_stream_skip_cstr(&stream); tb_static_stream_skip(&stream, 4); #else tb_char_t* name = tb_static_stream_read_cstr(&stream); //name = tb_dns_decode_name(name); tb_assert_and_check_return_val(name, tb_false); tb_static_stream_skip(&stream, 4); tb_trace_d("response: name: %s", name); #endif // decode answers tb_size_t i = 0; tb_size_t found = 0; for (i = 0; i < header.answer; i++) { // decode answer tb_dns_answer_t answer; tb_trace_d("response: answer: %d", i); // decode dns name tb_char_t const* name = tb_dns_decode_name(&stream, answer.name); tb_used(name); tb_trace_d("response: name: %s", name); // decode resource answer.res.type = tb_static_stream_read_u16_be(&stream); answer.res.class_ = tb_static_stream_read_u16_be(&stream); answer.res.ttl = tb_static_stream_read_u32_be(&stream); answer.res.size = tb_static_stream_read_u16_be(&stream); tb_trace_d("response: type: %d", answer.res.type); tb_trace_d("response: class: %d", answer.res.class_); tb_trace_d("response: ttl: %d", answer.res.ttl); tb_trace_d("response: size: %d", answer.res.size); // is ipv4? if (answer.res.type == 1) { // get ipv4 tb_byte_t b1 = tb_static_stream_read_u8(&stream); tb_byte_t b2 = tb_static_stream_read_u8(&stream); tb_byte_t b3 = tb_static_stream_read_u8(&stream); tb_byte_t b4 = tb_static_stream_read_u8(&stream); // trace tb_trace_d("response: ipv4: %u.%u.%u.%u", b1, b2, b3, b4); // save the first ip if (!found) { // save it if (addr) { // init ipv4 tb_ipv4_t ipv4; ipv4.u8[0] = b1; ipv4.u8[1] = b2; ipv4.u8[2] = b3; ipv4.u8[3] = b4; // save ipv4 tb_ipaddr_ipv4_set(addr, &ipv4); } // found it found = 1; // trace tb_trace_d("response: "); break; } } else { // decode rdata answer.rdata = (tb_byte_t*)tb_dns_decode_name(&stream, answer.name); // trace tb_trace_d("response: alias: %s", answer.rdata? (tb_char_t const*)answer.rdata : ""); } // trace tb_trace_d("response: "); } // found it? tb_check_return_val(found, tb_false); #if 0 // decode authorities for (i = 0; i < header.authority; i++) { // decode answer tb_dns_answer_t answer; tb_trace_d("response: authority: %d", i); // decode dns name tb_char_t* name = tb_dns_decode_name(&stream, answer.name); tb_trace_d("response: name: %s", name? name : ""); // decode resource answer.res.type = tb_static_stream_read_u16_be(&stream); answer.res.class_ = tb_static_stream_read_u16_be(&stream); answer.res.ttl = tb_static_stream_read_u32_be(&stream); answer.res.size = tb_static_stream_read_u16_be(&stream); tb_trace_d("response: type: %d", answer.res.type); tb_trace_d("response: class: %d", answer.res.class_); tb_trace_d("response: ttl: %d", answer.res.ttl); tb_trace_d("response: size: %d", answer.res.size); // is ipv4? if (answer.res.type == 1) { tb_byte_t b1 = tb_static_stream_read_u8(&stream); tb_byte_t b2 = tb_static_stream_read_u8(&stream); tb_byte_t b3 = tb_static_stream_read_u8(&stream); tb_byte_t b4 = tb_static_stream_read_u8(&stream); tb_trace_d("response: ipv4: %u.%u.%u.%u", b1, b2, b3, b4); } else { // decode data answer.rdata = tb_dns_decode_name(&stream, answer.name); tb_trace_d("response: server: %s", answer.rdata? answer.rdata : ""); } tb_trace_d("response: "); } for (i = 0; i < header.resource; i++) { // decode answer tb_dns_answer_t answer; tb_trace_d("response: resource: %d", i); // decode dns name tb_char_t* name = tb_dns_decode_name(&stream, answer.name); tb_trace_d("response: name: %s", name? name : ""); // decode resource answer.res.type = tb_static_stream_read_u16_be(&stream); answer.res.class_ = tb_static_stream_read_u16_be(&stream); answer.res.ttl = tb_static_stream_read_u32_be(&stream); answer.res.size = tb_static_stream_read_u16_be(&stream); tb_trace_d("response: type: %d", answer.res.type); tb_trace_d("response: class: %d", answer.res.class_); tb_trace_d("response: ttl: %d", answer.res.ttl); tb_trace_d("response: size: %d", answer.res.size); // is ipv4? if (answer.res.type == 1) { tb_byte_t b1 = tb_static_stream_read_u8(&stream); tb_byte_t b2 = tb_static_stream_read_u8(&stream); tb_byte_t b3 = tb_static_stream_read_u8(&stream); tb_byte_t b4 = tb_static_stream_read_u8(&stream); tb_trace_d("response: ipv4: %u.%u.%u.%u", b1, b2, b3, b4); } else { // decode data answer.rdata = tb_dns_decode_name(&stream, answer.name); tb_trace_d("response: alias: %s", answer.rdata? answer.rdata : ""); } tb_trace_d("response: "); } #endif // ok return tb_true; }