fix length computation in dn_expand
authorRich Felker <dalias@aerifal.cx>
Wed, 14 Aug 2013 22:18:46 +0000 (18:18 -0400)
committerRich Felker <dalias@aerifal.cx>
Wed, 14 Aug 2013 22:18:46 +0000 (18:18 -0400)
there are two possible points where the length is evaluated: either
the first 'compression' jump, or the null terminator if no jumps have
taken place yet. the previous code only measured the length of the
first component.

src/network/dn_expand.c

index 72a80e8..f817d05 100644 (file)
@@ -6,7 +6,7 @@ int __dn_expand(const unsigned char *base, const unsigned char *end, const unsig
        const unsigned char *p = src;
        int len = -1, j;
        if (space > 256) space = 256;
        const unsigned char *p = src;
        int len = -1, j;
        if (space > 256) space = 256;
-       if (p==end) return -1;
+       if (p==end || !*p) return -1;
        for (;;) {
                if (*p & 0xc0) {
                        if (p+1==end) return -1;
        for (;;) {
                if (*p & 0xc0) {
                        if (p+1==end) return -1;
@@ -16,11 +16,13 @@ int __dn_expand(const unsigned char *base, const unsigned char *end, const unsig
                        p = base+j;
                } else if (*p) {
                        j = *p+1;
                        p = base+j;
                } else if (*p) {
                        j = *p+1;
-                       if (len < 0) len = p+1-src;
                        if (j>=end-p || j>space) return -1;
                        while (--j) *dest++ = *p++;
                        *dest++ = *++p ? '.' : 0;
                        if (j>=end-p || j>space) return -1;
                        while (--j) *dest++ = *p++;
                        *dest++ = *++p ? '.' : 0;
-               } else return len;
+               } else {
+                       if (len < 0) len = p+1-src;
+                       return len;
+               }
        }
 }
 
        }
 }