attempting to add a dns packet parser
This commit is contained in:
parent
c6e3b6204b
commit
42fe77bde5
@ -894,7 +894,7 @@ for (i = 0; i < 5; i++)
|
|||||||
MIO_DNS_RCODE_BADCOOKIE /* rcode */
|
MIO_DNS_RCODE_BADCOOKIE /* rcode */
|
||||||
};
|
};
|
||||||
|
|
||||||
//mio_svc_dnc_sendreq (dnc, &qhdr, qrs, MIO_COUNTOF(qrs), &qedns, MIO_NULL);
|
mio_svc_dnc_sendreq (dnc, &qhdr, qrs, MIO_COUNTOF(qrs), &qedns, MIO_NULL);
|
||||||
mio_svc_dnc_sendmsg (dnc, &rhdr, qrs, MIO_COUNTOF(qrs), rrs, MIO_COUNTOF(rrs), &qedns, MIO_NULL);
|
mio_svc_dnc_sendmsg (dnc, &rhdr, qrs, MIO_COUNTOF(qrs), rrs, MIO_COUNTOF(rrs), &qedns, MIO_NULL);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
246
mio/lib/dns.c
246
mio/lib/dns.c
@ -387,113 +387,6 @@ static mio_dns_msg_t* build_dns_msg (mio_svc_dnc_t* dnc, mio_dns_bdns_t* bdns, m
|
|||||||
return msg;
|
return msg;
|
||||||
}
|
}
|
||||||
|
|
||||||
static int parse_dns_pkt (mio_svc_dnc_t* dnc, mio_dns_pkt_t* pkt, mio_oow_t len)
|
|
||||||
{
|
|
||||||
|
|
||||||
mio_t* mio = dnc->mio;
|
|
||||||
mio_dns_bdns_t* bdns;
|
|
||||||
mio_uint16_t i, rrc;
|
|
||||||
mio_uint8_t* dn;
|
|
||||||
mio_uint8_t* pktend = (mio_uint8_t*)pkt + len;
|
|
||||||
|
|
||||||
MIO_ASSERT (mio, len >= MIO_SIZEOF(*pkt));
|
|
||||||
return 0;
|
|
||||||
/* TODO: */
|
|
||||||
bdns->id = mio_ntoh16(pkt->id);
|
|
||||||
|
|
||||||
bdns->qr = pkt->qr & 0x01;
|
|
||||||
bdns->opcode = pkt->opcode & 0x0F;
|
|
||||||
bdns->aa = pkt->aa & 0x01;
|
|
||||||
bdns->tc = pkt->tc & 0x01;
|
|
||||||
bdns->rd = pkt->rd & 0x01;
|
|
||||||
bdns->ra = pkt->ra & 0x01;
|
|
||||||
bdns->ad = pkt->ad & 0x01;
|
|
||||||
bdns->cd = pkt->cd & 0x01;
|
|
||||||
bdns->rcode = pkt->rcode & 0x0F;
|
|
||||||
|
|
||||||
/*
|
|
||||||
bdns->qdcount = mio_ntoh16(pkt->qdcount);
|
|
||||||
bdns->ancount = mio_ntoh16(pkt->ancount);
|
|
||||||
bdns->nscount = mio_ntoh16(pkt->nscount);
|
|
||||||
bdns->arcount = mio_ntoh16(pkt->arcount);
|
|
||||||
*/
|
|
||||||
dn = (mio_uint8_t*)(pkt + 1);
|
|
||||||
|
|
||||||
rrc = mio_ntoh16(pkt->qdcount);
|
|
||||||
for (i = 0; i < rrc; i++)
|
|
||||||
{
|
|
||||||
mio_oow_t totlen, seglen;
|
|
||||||
mio_dns_qrtr_t* qrtr;
|
|
||||||
|
|
||||||
if (dn >= pktend)
|
|
||||||
{
|
|
||||||
mio_seterrbfmt (mio, MIO_EINVAL, "invalid packet");
|
|
||||||
return -1;
|
|
||||||
}
|
|
||||||
|
|
||||||
totlen = 0;
|
|
||||||
while ((seglen = *dn++) > 0)
|
|
||||||
{
|
|
||||||
if (seglen > 64)
|
|
||||||
{
|
|
||||||
/* compressed. pointer to somewhere else */
|
|
||||||
/* TODO: */
|
|
||||||
}
|
|
||||||
|
|
||||||
totlen += seglen;
|
|
||||||
dn += seglen;
|
|
||||||
}
|
|
||||||
|
|
||||||
qrtr = (mio_dns_qrtr_t*)dn;
|
|
||||||
dn += MIO_SIZEOF(*qrtr);
|
|
||||||
}
|
|
||||||
|
|
||||||
rrc = mio_ntoh16(pkt->ancount);
|
|
||||||
for (i = 0; i < rrc; i++)
|
|
||||||
{
|
|
||||||
mio_oow_t totlen, seglen;
|
|
||||||
mio_dns_rrtr_t* rrtr;
|
|
||||||
|
|
||||||
if (dn >= pktend)
|
|
||||||
{
|
|
||||||
mio_seterrbfmt (mio, MIO_EINVAL, "invalid packet");
|
|
||||||
return -1;
|
|
||||||
}
|
|
||||||
|
|
||||||
totlen = 0;
|
|
||||||
while ((seglen = *dn++) > 0)
|
|
||||||
{
|
|
||||||
totlen += seglen;
|
|
||||||
dn += seglen;
|
|
||||||
}
|
|
||||||
|
|
||||||
rrtr = (mio_dns_rrtr_t*)dn;
|
|
||||||
dn += MIO_SIZEOF(*rrtr) + rrtr->dlen;
|
|
||||||
}
|
|
||||||
|
|
||||||
#if 0
|
|
||||||
for (i = 0; i < bdns->nscount; i++)
|
|
||||||
{
|
|
||||||
}
|
|
||||||
|
|
||||||
for (i = 0; i < bdns->arcount; i++)
|
|
||||||
{
|
|
||||||
#if 0
|
|
||||||
if (*dn == 0)
|
|
||||||
{
|
|
||||||
rrtr = (mio_dns_rrtr_t*)(dn + 1);
|
|
||||||
if (rrtr->qtype == MIO_CONST_HTON16(MIO_DNS_QTYPE_OPT)
|
|
||||||
{
|
|
||||||
/* edns */
|
|
||||||
}
|
|
||||||
}
|
|
||||||
#endif
|
|
||||||
}
|
|
||||||
#endif
|
|
||||||
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
/* ----------------------------------------------------------------------- */
|
/* ----------------------------------------------------------------------- */
|
||||||
|
|
||||||
static int dnc_on_read (mio_dev_sck_t* dev, const void* data, mio_iolen_t dlen, const mio_sckaddr_t* srcaddr)
|
static int dnc_on_read (mio_dev_sck_t* dev, const void* data, mio_iolen_t dlen, const mio_sckaddr_t* srcaddr)
|
||||||
@ -528,8 +421,6 @@ static int dnc_on_read (mio_dev_sck_t* dev, const void* data, mio_iolen_t dlen,
|
|||||||
if (dev == (mio_dev_sck_t*)reqmsg->dev && pkt->id == reqpkt->id) /* TODO: check the source address against the target address */
|
if (dev == (mio_dev_sck_t*)reqmsg->dev && pkt->id == reqpkt->id) /* TODO: check the source address against the target address */
|
||||||
{
|
{
|
||||||
MIO_DEBUG1 (mio, "received dns response...id %d\n", id);
|
MIO_DEBUG1 (mio, "received dns response...id %d\n", id);
|
||||||
/* TODO: parse the response... perform actual work. pass the result back?? */
|
|
||||||
/*parse_dns_pkt (dnc, pkt, dlen);*/
|
|
||||||
|
|
||||||
if (MIO_LIKELY(reqmsg->ctx))
|
if (MIO_LIKELY(reqmsg->ctx))
|
||||||
((mio_svc_dnc_on_reply_t)reqmsg->ctx) (dnc, reqmsg, MIO_ENOERR, data, dlen);
|
((mio_svc_dnc_on_reply_t)reqmsg->ctx) (dnc, reqmsg, MIO_ENOERR, data, dlen);
|
||||||
@ -709,10 +600,6 @@ int mio_svc_dnc_sendreq (mio_svc_dnc_t* dnc, mio_dns_bdns_t* bdns, mio_dns_bqr_t
|
|||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
static void on_dnc_resolve_reply (mio_svc_dnc_t* dnc, mio_dns_msg_t* reqmsg, mio_errnum_t status, const void* data, mio_oow_t dlen)
|
|
||||||
{
|
|
||||||
}
|
|
||||||
|
|
||||||
int mio_svc_dnc_resolve (mio_svc_dnc_t* dnc, const mio_bch_t* qname, mio_dns_qtype_t qtype, mio_svc_dnc_on_reply_t on_reply)
|
int mio_svc_dnc_resolve (mio_svc_dnc_t* dnc, const mio_bch_t* qname, mio_dns_qtype_t qtype, mio_svc_dnc_on_reply_t on_reply)
|
||||||
{
|
{
|
||||||
static mio_dns_bdns_t qhdr =
|
static mio_dns_bdns_t qhdr =
|
||||||
@ -749,3 +636,136 @@ int mio_svc_dnc_resolve (mio_svc_dnc_t* dnc, const mio_bch_t* qname, mio_dns_qty
|
|||||||
return mio_svc_dnc_sendreq(dnc, &qhdr, &qr, 1, &qedns, on_reply);
|
return mio_svc_dnc_sendreq(dnc, &qhdr, &qr, 1, &qedns, on_reply);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
/* ----------------------------------------------------------------------- */
|
||||||
|
|
||||||
|
static mio_uint8_t* parse_question (mio_t* mio, mio_uint8_t* dn, mio_uint8_t* pktend)
|
||||||
|
{
|
||||||
|
mio_oow_t totlen, seglen;
|
||||||
|
mio_dns_qrtr_t* qrtr;
|
||||||
|
|
||||||
|
if (dn >= pktend)
|
||||||
|
{
|
||||||
|
mio_seterrbfmt (mio, MIO_EINVAL, "invalid packet");
|
||||||
|
return MIO_NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
totlen = 0;
|
||||||
|
while ((seglen = *dn++) > 0)
|
||||||
|
{
|
||||||
|
if (seglen > 64)
|
||||||
|
{
|
||||||
|
/* compressed. pointer to somewhere else */
|
||||||
|
/* TODO: */
|
||||||
|
}
|
||||||
|
|
||||||
|
totlen += seglen;
|
||||||
|
dn += seglen;
|
||||||
|
}
|
||||||
|
|
||||||
|
qrtr = (mio_dns_qrtr_t*)dn;
|
||||||
|
dn += MIO_SIZEOF(*qrtr);
|
||||||
|
|
||||||
|
return dn;
|
||||||
|
}
|
||||||
|
|
||||||
|
static mio_uint8_t* parse_answer (mio_t* mio, mio_uint8_t* dn, mio_uint8_t* pktend)
|
||||||
|
{
|
||||||
|
mio_oow_t totlen, seglen;
|
||||||
|
mio_dns_rrtr_t* rrtr;
|
||||||
|
|
||||||
|
if (dn >= pktend)
|
||||||
|
{
|
||||||
|
mio_seterrbfmt (mio, MIO_EINVAL, "invalid packet");
|
||||||
|
return MIO_NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
totlen = 0;
|
||||||
|
while ((seglen = *dn++) > 0)
|
||||||
|
{
|
||||||
|
totlen += seglen;
|
||||||
|
dn += seglen;
|
||||||
|
}
|
||||||
|
|
||||||
|
rrtr = (mio_dns_rrtr_t*)dn;
|
||||||
|
dn += MIO_SIZEOF(*rrtr) + rrtr->dlen;
|
||||||
|
|
||||||
|
return dn;
|
||||||
|
}
|
||||||
|
|
||||||
|
int mio_dns_parse_packet (mio_svc_dnc_t* dnc, mio_dns_pkt_t* pkt, mio_oow_t len, mio_dns_bdns_t* bdns)
|
||||||
|
{
|
||||||
|
mio_t* mio = dnc->mio;
|
||||||
|
mio_uint16_t i, rrc;
|
||||||
|
mio_uint8_t* dn;
|
||||||
|
mio_uint8_t* pktend = (mio_uint8_t*)pkt + len;
|
||||||
|
|
||||||
|
MIO_ASSERT (mio, len >= MIO_SIZEOF(*pkt));
|
||||||
|
|
||||||
|
bdns->id = mio_ntoh16(pkt->id);
|
||||||
|
|
||||||
|
bdns->qr = pkt->qr & 0x01;
|
||||||
|
bdns->opcode = pkt->opcode & 0x0F;
|
||||||
|
bdns->aa = pkt->aa & 0x01;
|
||||||
|
bdns->tc = pkt->tc & 0x01;
|
||||||
|
bdns->rd = pkt->rd & 0x01;
|
||||||
|
bdns->ra = pkt->ra & 0x01;
|
||||||
|
bdns->ad = pkt->ad & 0x01;
|
||||||
|
bdns->cd = pkt->cd & 0x01;
|
||||||
|
bdns->rcode = pkt->rcode & 0x0F;
|
||||||
|
|
||||||
|
/*
|
||||||
|
bdns->qdcount = mio_ntoh16(pkt->qdcount);
|
||||||
|
bdns->ancount = mio_ntoh16(pkt->ancount);
|
||||||
|
bdns->nscount = mio_ntoh16(pkt->nscount);
|
||||||
|
bdns->arcount = mio_ntoh16(pkt->arcount);
|
||||||
|
*/
|
||||||
|
|
||||||
|
dn = (mio_uint8_t*)(pkt + 1);
|
||||||
|
|
||||||
|
rrc = mio_ntoh16(pkt->qdcount);
|
||||||
|
for (i = 0; i < rrc; i++)
|
||||||
|
{
|
||||||
|
dn = parse_question(mio, dn, pktend);
|
||||||
|
if (!dn) return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
rrc = mio_ntoh16(pkt->ancount);
|
||||||
|
for (i = 0; i < rrc; i++)
|
||||||
|
{
|
||||||
|
dn = parse_answer(mio, dn, pktend);
|
||||||
|
if (!dn) return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
rrc = mio_ntoh16(pkt->nscount);
|
||||||
|
for (i = 0; i < rrc; i++)
|
||||||
|
{
|
||||||
|
dn = parse_answer(mio, dn, pktend);
|
||||||
|
if (!dn) return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
rrc = mio_ntoh16(pkt->arcount);
|
||||||
|
for (i = 0; i < rrc; i++)
|
||||||
|
{
|
||||||
|
dn = parse_answer(mio, dn, pktend);
|
||||||
|
if (!dn) return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
#if 0
|
||||||
|
for (i = 0; i < bdns->arcount; i++)
|
||||||
|
{
|
||||||
|
#if 0
|
||||||
|
if (*dn == 0)
|
||||||
|
{
|
||||||
|
rrtr = (mio_dns_rrtr_t*)(dn + 1);
|
||||||
|
if (rrtr->qtype == MIO_CONST_HTON16(MIO_DNS_QTYPE_OPT)
|
||||||
|
{
|
||||||
|
/* edns */
|
||||||
|
}
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user