Ensure sufficient buffer size when converting nlabels to string.
This commit is contained in:
parent
4bed5363e7
commit
db4e276662
16
mdns.c
16
mdns.c
@ -90,18 +90,26 @@ uint8_t *join_nlabel(const uint8_t *n1, const uint8_t *n2) {
|
||||
char *nlabel_to_str(const uint8_t *name) {
|
||||
char *label, *labelp;
|
||||
const uint8_t *p;
|
||||
size_t buf_len = 256;
|
||||
|
||||
assert(name != NULL);
|
||||
|
||||
label = labelp = malloc(256);
|
||||
label = labelp = malloc(buf_len);
|
||||
|
||||
for (p = name; *p; p++) {
|
||||
strncpy(labelp, (char *) p + 1, *p);
|
||||
labelp += *p;
|
||||
uint8_t label_len = *p;
|
||||
if (buf_len <= label_len)
|
||||
break;
|
||||
|
||||
strncpy(labelp, (char *) p + 1, label_len);
|
||||
labelp += label_len;
|
||||
|
||||
*labelp = '.';
|
||||
labelp++;
|
||||
|
||||
p += *p;
|
||||
buf_len -= label_len + 1;
|
||||
|
||||
p += label_len;
|
||||
}
|
||||
|
||||
*labelp = '\0';
|
||||
|
Loading…
Reference in New Issue
Block a user