tb_long_t tb_charset_utf16_get(tb_static_stream_ref_t sstream, tb_bool_t be, tb_uint32_t* ch)
{
// init
tb_byte_t const* p = tb_static_stream_pos(sstream);
tb_byte_t const* q = p;
tb_size_t n = tb_static_stream_left(sstream);
// not enough? break it
tb_check_return_val(n > 1, -1);
// the first character
tb_uint32_t c = be? tb_bits_get_u16_be(p) : tb_bits_get_u16_le(p);
p += 2;
// large?
if (c >= 0xd800 && c <= 0xdbff)
{
// not enough? break it
tb_check_return_val(n > 3, -1);
// the next character
tb_uint32_t c2 = be? tb_bits_get_u16_be(p) : tb_bits_get_u16_le(p);
if (c2 >= 0xdc00 && c2 <= 0xdfff)
{
c = ((c - 0xd800) << 10) + (c2 - 0xdc00) + 0x0010000;
p += 2;
};
};
// next
if (p > q) tb_static_stream_skip(sstream, p - q);
// set character
*ch = c;
// ok?
return p > q? 1 : 0;
}
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;
}