511 lines
18 KiB
Diff
511 lines
18 KiB
Diff
--- a/dns.c
|
|
+++ b/dns.c
|
|
@@ -918,6 +918,507 @@ void restell(char *s)
|
|
fputs("\r",stderr);
|
|
}
|
|
|
|
+#ifdef __UCLIBC__
|
|
+
|
|
+static const char digits[] = "0123456789";
|
|
+#define __set_errno(e) (errno = (e))
|
|
+
|
|
+#define NS_PUT16(s, cp) do { \
|
|
+ register u_int16_t t_s = (u_int16_t)(s); \
|
|
+ register u_char *t_cp = (u_char *)(cp); \
|
|
+ *t_cp++ = t_s >> 8; \
|
|
+ *t_cp = t_s; \
|
|
+ (cp) += NS_INT16SZ; \
|
|
+} while (0)
|
|
+
|
|
+
|
|
+
|
|
+#define NS_PUT32(l, cp) do { \
|
|
+ register u_int32_t t_l = (u_int32_t)(l); \
|
|
+ register u_char *t_cp = (u_char *)(cp); \
|
|
+ *t_cp++ = t_l >> 24; \
|
|
+ *t_cp++ = t_l >> 16; \
|
|
+ *t_cp++ = t_l >> 8; \
|
|
+ *t_cp = t_l; \
|
|
+ (cp) += NS_INT32SZ; \
|
|
+} while (0)
|
|
+
|
|
+
|
|
+void
|
|
+ns_put16(u_int src, u_char *dst) {
|
|
+ NS_PUT16(src, dst);
|
|
+}
|
|
+
|
|
+void
|
|
+ns_put32(u_long src, u_char *dst) {
|
|
+ NS_PUT32(src, dst);
|
|
+}
|
|
+
|
|
+void __putshort(u_int16_t src, u_char *dst) { ns_put16(src, dst); }
|
|
+void __putlong(u_int32_t src, u_char *dst) { ns_put32(src, dst); }
|
|
+
|
|
+int
|
|
+mklower(int ch) {
|
|
+ if (ch >= 0x41 && ch <= 0x5A)
|
|
+ return (ch + 0x20);
|
|
+ return (ch);
|
|
+}
|
|
+
|
|
+
|
|
+static int
|
|
+dn_find(const u_char *domain, const u_char *msg,
|
|
+ const u_char * const *dnptrs,
|
|
+ const u_char * const *lastdnptr)
|
|
+{
|
|
+ const u_char *dn, *cp, *sp;
|
|
+ const u_char * const *cpp;
|
|
+ u_int n;
|
|
+
|
|
+ for (cpp = dnptrs; cpp < lastdnptr; cpp++) {
|
|
+ sp = *cpp;
|
|
+ /*
|
|
+ * terminate search on:
|
|
+ * root label
|
|
+ * compression pointer
|
|
+ * unusable offset
|
|
+ */
|
|
+ while (*sp != 0 && (*sp & NS_CMPRSFLGS) == 0 &&
|
|
+ (sp - msg) < 0x4000) {
|
|
+ dn = domain;
|
|
+ cp = sp;
|
|
+ while ((n = *cp++) != 0) {
|
|
+ /*
|
|
+ * check for indirection
|
|
+ */
|
|
+ switch (n & NS_CMPRSFLGS) {
|
|
+ case 0: /* normal case, n == len */
|
|
+ if (n != *dn++)
|
|
+ goto next;
|
|
+ for ((void)NULL; n > 0; n--)
|
|
+ if (mklower(*dn++) !=
|
|
+ mklower(*cp++))
|
|
+ goto next;
|
|
+ /* Is next root for both ? */
|
|
+ if (*dn == '\0' && *cp == '\0')
|
|
+ return (sp - msg);
|
|
+ if (*dn)
|
|
+ continue;
|
|
+ goto next;
|
|
+
|
|
+ case NS_CMPRSFLGS: /* indirection */
|
|
+ cp = msg + (((n & 0x3f) << 8) | *cp);
|
|
+ break;
|
|
+
|
|
+ default: /* illegal type */
|
|
+ __set_errno (EMSGSIZE);
|
|
+ return (-1);
|
|
+ }
|
|
+ }
|
|
+ next:
|
|
+ sp += *sp + 1;
|
|
+ }
|
|
+ }
|
|
+ __set_errno (ENOENT);
|
|
+ return (-1);
|
|
+}
|
|
+
|
|
+
|
|
+int
|
|
+ns_name_pack(const u_char *src, u_char *dst, int dstsiz,
|
|
+ const u_char **dnptrs, const u_char **lastdnptr)
|
|
+{
|
|
+ u_char *dstp;
|
|
+ const u_char **cpp, **lpp, *eob, *msg;
|
|
+ const u_char *srcp;
|
|
+ int n, l, first = 1;
|
|
+
|
|
+ srcp = src;
|
|
+ dstp = dst;
|
|
+ eob = dstp + dstsiz;
|
|
+ lpp = cpp = NULL;
|
|
+ if (dnptrs != NULL) {
|
|
+ if ((msg = *dnptrs++) != NULL) {
|
|
+ for (cpp = dnptrs; *cpp != NULL; cpp++)
|
|
+ (void)NULL;
|
|
+ lpp = cpp; /* end of list to search */
|
|
+ }
|
|
+ } else
|
|
+ msg = NULL;
|
|
+
|
|
+ /* make sure the domain we are about to add is legal */
|
|
+ l = 0;
|
|
+ do {
|
|
+ n = *srcp;
|
|
+ if ((n & NS_CMPRSFLGS) != 0 && n != 0x41) {
|
|
+ __set_errno (EMSGSIZE);
|
|
+ return (-1);
|
|
+ }
|
|
+ if (n == 0x41)
|
|
+ n = *++srcp / 8;
|
|
+ l += n + 1;
|
|
+ if (l > MAXCDNAME) {
|
|
+ __set_errno (EMSGSIZE);
|
|
+ return (-1);
|
|
+ }
|
|
+ srcp += n + 1;
|
|
+ } while (n != 0);
|
|
+
|
|
+ /* from here on we need to reset compression pointer array on error */
|
|
+ srcp = src;
|
|
+ do {
|
|
+ /* Look to see if we can use pointers. */
|
|
+ n = *srcp;
|
|
+ if (n != 0 && n != 0x41 && msg != NULL) {
|
|
+ l = dn_find(srcp, msg, (const u_char * const *)dnptrs,
|
|
+ (const u_char * const *)lpp);
|
|
+ if (l >= 0) {
|
|
+ if (dstp + 1 >= eob) {
|
|
+ goto cleanup;
|
|
+ }
|
|
+ *dstp++ = (l >> 8) | NS_CMPRSFLGS;
|
|
+ *dstp++ = l % 256;
|
|
+ return (dstp - dst);
|
|
+ }
|
|
+ /* Not found, save it. */
|
|
+ if (lastdnptr != NULL && cpp < lastdnptr - 1 &&
|
|
+ (dstp - msg) < 0x4000 && first) {
|
|
+ *cpp++ = dstp;
|
|
+ *cpp = NULL;
|
|
+ first = 0;
|
|
+ }
|
|
+ }
|
|
+ /* copy label to buffer */
|
|
+ if ((n & NS_CMPRSFLGS) != 0 && n != 0x41) { /* Should not happen. */
|
|
+ goto cleanup;
|
|
+ }
|
|
+ if (n == 0x41) {
|
|
+ n = *++srcp / 8;
|
|
+ if (dstp + 1 >= eob)
|
|
+ goto cleanup;
|
|
+ *dstp++ = 0x41;
|
|
+ }
|
|
+ if (dstp + 1 + n >= eob) {
|
|
+ goto cleanup;
|
|
+ }
|
|
+ memcpy(dstp, srcp, n + 1);
|
|
+ srcp += n + 1;
|
|
+ dstp += n + 1;
|
|
+ } while (n != 0);
|
|
+
|
|
+ if (dstp > eob) {
|
|
+cleanup:
|
|
+ if (msg != NULL)
|
|
+ *lpp = NULL;
|
|
+ __set_errno (EMSGSIZE);
|
|
+ return (-1);
|
|
+ }
|
|
+ return (dstp - dst);
|
|
+}
|
|
+
|
|
+
|
|
+int
|
|
+ns_name_pton(const char *src, u_char *dst, size_t dstsiz) {
|
|
+ u_char *label, *bp, *eom;
|
|
+ int c, n, escaped;
|
|
+ char *cp;
|
|
+
|
|
+ escaped = 0;
|
|
+ bp = dst;
|
|
+ eom = dst + dstsiz;
|
|
+ label = bp++;
|
|
+
|
|
+ while ((c = *src++) != 0) {
|
|
+ if (escaped) {
|
|
+ if ((cp = strchr(digits, c)) != NULL) {
|
|
+ n = (cp - digits) * 100;
|
|
+ if ((c = *src++) == 0 ||
|
|
+ (cp = strchr(digits, c)) == NULL) {
|
|
+ __set_errno (EMSGSIZE);
|
|
+ return (-1);
|
|
+ }
|
|
+ n += (cp - digits) * 10;
|
|
+ if ((c = *src++) == 0 ||
|
|
+ (cp = strchr(digits, c)) == NULL) {
|
|
+ __set_errno (EMSGSIZE);
|
|
+ return (-1);
|
|
+ }
|
|
+ n += (cp - digits);
|
|
+ if (n > 255) {
|
|
+ __set_errno (EMSGSIZE);
|
|
+ return (-1);
|
|
+ }
|
|
+ c = n;
|
|
+ } else if (c == '[' && label == bp - 1 && *src == 'x') {
|
|
+ /* Theoretically we would have to handle \[o
|
|
+ as well but we do not since we do not need
|
|
+ it internally. */
|
|
+ *label = 0x41;
|
|
+ label = bp++;
|
|
+ ++src;
|
|
+ while (isxdigit (*src)) {
|
|
+ n = *src > '9' ? *src - 'a' + 10 : *src - '0';
|
|
+ ++src;
|
|
+ if (! isxdigit(*src)) {
|
|
+ __set_errno (EMSGSIZE);
|
|
+ return (-1);
|
|
+ }
|
|
+ n <<= 4;
|
|
+ n += *src > '9' ? *src - 'a' + 10 : *src - '0';
|
|
+ if (bp + 1 >= eom) {
|
|
+ __set_errno (EMSGSIZE);
|
|
+ return (-1);
|
|
+ }
|
|
+ *bp++ = n;
|
|
+ ++src;
|
|
+ }
|
|
+ *label = (bp - label - 1) * 8;
|
|
+ if (*src++ != ']' || *src++ != '.') {
|
|
+ __set_errno (EMSGSIZE);
|
|
+ return (-1);
|
|
+ }
|
|
+ escaped = 0;
|
|
+ label = bp++;
|
|
+ if (bp >= eom) {
|
|
+ __set_errno (EMSGSIZE);
|
|
+ return (-1);
|
|
+ }
|
|
+ continue;
|
|
+ }
|
|
+ escaped = 0;
|
|
+ } else if (c == '\\') {
|
|
+ escaped = 1;
|
|
+ continue;
|
|
+ } else if (c == '.') {
|
|
+ c = (bp - label - 1);
|
|
+ if ((c & NS_CMPRSFLGS) != 0) { /* Label too big. */
|
|
+ __set_errno (EMSGSIZE);
|
|
+ return (-1);
|
|
+ }
|
|
+ if (label >= eom) {
|
|
+ __set_errno (EMSGSIZE);
|
|
+ return (-1);
|
|
+ }
|
|
+ *label = c;
|
|
+ /* Fully qualified ? */
|
|
+ if (*src == '\0') {
|
|
+ if (c != 0) {
|
|
+ if (bp >= eom) {
|
|
+ __set_errno (EMSGSIZE);
|
|
+ return (-1);
|
|
+ }
|
|
+ *bp++ = '\0';
|
|
+ }
|
|
+ if ((bp - dst) > MAXCDNAME) {
|
|
+ __set_errno (EMSGSIZE);
|
|
+ return (-1);
|
|
+ }
|
|
+ return (1);
|
|
+ }
|
|
+ if (c == 0 || *src == '.') {
|
|
+ __set_errno (EMSGSIZE);
|
|
+ return (-1);
|
|
+ }
|
|
+ label = bp++;
|
|
+ continue;
|
|
+ }
|
|
+ if (bp >= eom) {
|
|
+ __set_errno (EMSGSIZE);
|
|
+ return (-1);
|
|
+ }
|
|
+ *bp++ = (u_char)c;
|
|
+ }
|
|
+ c = (bp - label - 1);
|
|
+ if ((c & NS_CMPRSFLGS) != 0) { /* Label too big. */
|
|
+ __set_errno (EMSGSIZE);
|
|
+ return (-1);
|
|
+ }
|
|
+ if (label >= eom) {
|
|
+ __set_errno (EMSGSIZE);
|
|
+ return (-1);
|
|
+ }
|
|
+ *label = c;
|
|
+ if (c != 0) {
|
|
+ if (bp >= eom) {
|
|
+ __set_errno (EMSGSIZE);
|
|
+ return (-1);
|
|
+ }
|
|
+ *bp++ = 0;
|
|
+ }
|
|
+ if ((bp - dst) > MAXCDNAME) { /* src too big */
|
|
+ __set_errno (EMSGSIZE);
|
|
+ return (-1);
|
|
+ }
|
|
+ return (0);
|
|
+}
|
|
+
|
|
+
|
|
+
|
|
+int
|
|
+ns_name_compress(const char *src, u_char *dst, size_t dstsiz,
|
|
+ const u_char **dnptrs, const u_char **lastdnptr)
|
|
+{
|
|
+ u_char tmp[NS_MAXCDNAME];
|
|
+
|
|
+ if (ns_name_pton(src, tmp, sizeof tmp) == -1)
|
|
+ return (-1);
|
|
+ return (ns_name_pack(tmp, dst, dstsiz, dnptrs, lastdnptr));
|
|
+}
|
|
+
|
|
+
|
|
+int
|
|
+dn_comp(const char *src, u_char *dst, int dstsiz,
|
|
+ u_char **dnptrs, u_char **lastdnptr)
|
|
+{
|
|
+ return (ns_name_compress(src, dst, (size_t)dstsiz,
|
|
+ (const u_char **)dnptrs,
|
|
+ (const u_char **)lastdnptr));
|
|
+}
|
|
+
|
|
+
|
|
+
|
|
+
|
|
+int
|
|
+res_nmkquery(res_state statp,
|
|
+ int op, /* opcode of query */
|
|
+ const char *dname, /* domain name */
|
|
+ int class, int type, /* class and type of query */
|
|
+ const u_char *data, /* resource record data */
|
|
+ int datalen, /* length of data */
|
|
+ const u_char *newrr_in, /* new rr for modify or append */
|
|
+ u_char *buf, /* buffer to put query */
|
|
+ int buflen) /* size of buffer */
|
|
+{
|
|
+ register HEADER *hp;
|
|
+ register u_char *cp;
|
|
+ register int n;
|
|
+ u_char *dnptrs[20], **dpp, **lastdnptr;
|
|
+
|
|
+#ifdef DEBUG
|
|
+ if (statp->options & RES_DEBUG)
|
|
+ printf(";; res_nmkquery(%s, %s, %s, %s)\n",
|
|
+ _res_opcodes[op], dname, p_class(class), p_type(type));
|
|
+#endif
|
|
+ /*
|
|
+ * Initialize header fields.
|
|
+ */
|
|
+ if ((buf == NULL) || (buflen < HFIXEDSZ))
|
|
+ return (-1);
|
|
+ memset(buf, 0, HFIXEDSZ);
|
|
+ hp = (HEADER *) buf;
|
|
+ /* We randomize the IDs every time. The old code just
|
|
+ incremented by one after the initial randomization which
|
|
+ still predictable if the application does multiple
|
|
+ requests. */
|
|
+#if 0
|
|
+ hp->id = htons(++statp->id);
|
|
+#else
|
|
+ hp->id = htons(statp->id);
|
|
+ int randombits;
|
|
+ do
|
|
+ {
|
|
+#ifdef RANDOM_BITS
|
|
+ RANDOM_BITS (randombits);
|
|
+#else
|
|
+ struct timeval tv;
|
|
+ gettimeofday (&tv, NULL);
|
|
+ randombits = (tv.tv_sec << 8) ^ tv.tv_usec;
|
|
+#endif
|
|
+ }
|
|
+ while ((randombits & 0xffff) == 0);
|
|
+ statp->id = (statp->id + randombits) & 0xffff;
|
|
+#endif
|
|
+ hp->opcode = op;
|
|
+ hp->rd = (statp->options & RES_RECURSE) != 0;
|
|
+ hp->rcode = NOERROR;
|
|
+ cp = buf + HFIXEDSZ;
|
|
+ buflen -= HFIXEDSZ;
|
|
+ dpp = dnptrs;
|
|
+ *dpp++ = buf;
|
|
+ *dpp++ = NULL;
|
|
+ lastdnptr = dnptrs + sizeof dnptrs / sizeof dnptrs[0];
|
|
+ /*
|
|
+ * perform opcode specific processing
|
|
+ */
|
|
+ switch (op) {
|
|
+ case QUERY: /*FALLTHROUGH*/
|
|
+ case NS_NOTIFY_OP:
|
|
+ if ((buflen -= QFIXEDSZ) < 0)
|
|
+ return (-1);
|
|
+ if ((n = dn_comp(dname, cp, buflen, dnptrs, lastdnptr)) < 0)
|
|
+ return (-1);
|
|
+ cp += n;
|
|
+ buflen -= n;
|
|
+ __putshort(type, cp);
|
|
+ cp += INT16SZ;
|
|
+ __putshort(class, cp);
|
|
+ cp += INT16SZ;
|
|
+ hp->qdcount = htons(1);
|
|
+ if (op == QUERY || data == NULL)
|
|
+ break;
|
|
+ /*
|
|
+ * Make an additional record for completion domain.
|
|
+ */
|
|
+ buflen -= RRFIXEDSZ;
|
|
+ n = dn_comp((char *)data, cp, buflen, dnptrs, lastdnptr);
|
|
+ if (n < 0)
|
|
+ return (-1);
|
|
+ cp += n;
|
|
+ buflen -= n;
|
|
+ __putshort(T_NULL, cp);
|
|
+ cp += INT16SZ;
|
|
+ __putshort(class, cp);
|
|
+ cp += INT16SZ;
|
|
+ __putlong(0, cp);
|
|
+ cp += INT32SZ;
|
|
+ __putshort(0, cp);
|
|
+ cp += INT16SZ;
|
|
+ hp->arcount = htons(1);
|
|
+ break;
|
|
+
|
|
+ case IQUERY:
|
|
+ /*
|
|
+ * Initialize answer section
|
|
+ */
|
|
+ if (buflen < 1 + RRFIXEDSZ + datalen)
|
|
+ return (-1);
|
|
+ *cp++ = '\0'; /* no domain name */
|
|
+ __putshort(type, cp);
|
|
+ cp += INT16SZ;
|
|
+ __putshort(class, cp);
|
|
+ cp += INT16SZ;
|
|
+ __putlong(0, cp);
|
|
+ cp += INT32SZ;
|
|
+ __putshort(datalen, cp);
|
|
+ cp += INT16SZ;
|
|
+ if (datalen) {
|
|
+ memcpy(cp, data, datalen);
|
|
+ cp += datalen;
|
|
+ }
|
|
+ hp->ancount = htons(1);
|
|
+ break;
|
|
+
|
|
+ default:
|
|
+ return (-1);
|
|
+ }
|
|
+ return (cp - buf);
|
|
+}
|
|
+
|
|
+int
|
|
+res_mkquery(int op, /* opcode of query */
|
|
+ const char *dname, /* domain name */
|
|
+ int class, int type, /* class and type of query */
|
|
+ const u_char *data, /* resource record data */
|
|
+ int datalen, /* length of data */
|
|
+ const u_char *newrr_in, /* new rr for modify or append */
|
|
+ u_char *buf, /* buffer to put query */
|
|
+ int buflen) /* size of buffer */
|
|
+{
|
|
+ return (res_nmkquery(&_res, op, dname, class, type,
|
|
+ data, datalen,
|
|
+ newrr_in, buf, buflen));
|
|
+}
|
|
+
|
|
+#endif
|
|
|
|
void dorequest(char *s,int type,word id)
|
|
{
|