Signed-off-by: André Roth <neolynx@xxxxxxxxx> --- lib/include/libdvbv5/descriptors.h | 2 -- lib/libdvbv5/descriptors.c | 44 ++++++++++++++++++------------------ 2 files changed, 22 insertions(+), 24 deletions(-) diff --git a/lib/include/libdvbv5/descriptors.h b/lib/include/libdvbv5/descriptors.h index 6f89aeb..36bcc61 100644 --- a/lib/include/libdvbv5/descriptors.h +++ b/lib/include/libdvbv5/descriptors.h @@ -78,8 +78,6 @@ void dvb_desc_default_print (struct dvb_v5_fe_parms *parms, const struct dvb_de for( _struct *_desc = (_struct *) _tbl->descriptor; _desc; _desc = (_struct *) _desc->next ) \ if(_desc->type == _type) \ -ssize_t dvb_desc_init(const uint8_t *buf, struct dvb_desc *desc); - uint32_t bcd(uint32_t bcd); void hexdump(struct dvb_v5_fe_parms *parms, const char *prefix, const unsigned char *buf, int len); diff --git a/lib/libdvbv5/descriptors.c b/lib/libdvbv5/descriptors.c index a2176b4..626f81d 100644 --- a/lib/libdvbv5/descriptors.c +++ b/lib/libdvbv5/descriptors.c @@ -56,12 +56,11 @@ #include <libdvbv5/desc_partial_reception.h> #include <libdvbv5/desc_extension.h> -ssize_t dvb_desc_init(const uint8_t *buf, struct dvb_desc *desc) +static void dvb_desc_init(uint8_t type, uint8_t length, struct dvb_desc *desc) { - desc->type = buf[0]; - desc->length = buf[1]; + desc->type = type; + desc->length = length; desc->next = NULL; - return 2; } void dvb_desc_default_init(struct dvb_v5_fe_parms *parms, const uint8_t *buf, struct dvb_desc *desc) @@ -94,16 +93,27 @@ char *default_charset = "iso-8859-1"; char *output_charset = "utf-8"; void dvb_parse_descriptors(struct dvb_v5_fe_parms *parms, const uint8_t *buf, - uint16_t section_length, struct dvb_desc **head_desc) + uint16_t buflen, struct dvb_desc **head_desc) { - const uint8_t *ptr = buf; + const uint8_t *ptr = buf, *endbuf = buf + buflen; struct dvb_desc *current = NULL; struct dvb_desc *last = NULL; - while (ptr < buf + section_length) { - unsigned desc_type = ptr[0]; - int desc_len = ptr[1]; + + *head_desc = NULL; + + while (ptr + 2 <= endbuf ) { + uint8_t desc_type = ptr[0]; + uint8_t desc_len = ptr[1]; size_t size; + ptr += 2; /* skip type and length */ + + if (ptr + desc_len > endbuf) { + dvb_logerr("short read of %zd/%d bytes parsing descriptor %#02x", + endbuf - ptr, desc_len, desc_type); + return; + } + switch (parms->verbose) { case 0: case 1: @@ -119,11 +129,6 @@ void dvb_parse_descriptors(struct dvb_v5_fe_parms *parms, const uint8_t *buf, hexdump(parms, "content: ", ptr + 2, desc_len); } - if (desc_len > section_length - 2) { - dvb_logwarn("descriptor type %0x02x is too big", - desc_type); - return; - } dvb_desc_init_func init = dvb_descriptors[desc_type].init; if (!init) { @@ -133,21 +138,16 @@ void dvb_parse_descriptors(struct dvb_v5_fe_parms *parms, const uint8_t *buf, size = dvb_descriptors[desc_type].size; } if (!size) { - dvb_logwarn("descriptor type 0x%02x has no size defined", desc_type); - size = 4096; - } - if (ptr + 2 >= buf + section_length) { - dvb_logwarn("descriptor type 0x%02x is truncated: desc len %d, section len %zd", - desc_type, desc_len, section_length - (ptr - buf)); + dvb_logerr("descriptor type 0x%02x has no size defined", desc_type); return; } - current = calloc(1, size); + current = malloc(size); if (!current) { dvb_perror("Out of memory"); return; } - ptr += dvb_desc_init(ptr, current); /* the standard header was read */ + dvb_desc_init(desc_type, desc_len, current); /* initialize the standard header */ init(parms, ptr, current); if (!*head_desc) *head_desc = current; -- 1.7.10.4 -- To unsubscribe from this list: send the line "unsubscribe linux-media" in the body of a message to majordomo@xxxxxxxxxxxxxxx More majordomo info at http://vger.kernel.org/majordomo-info.html