diff options
author | Filipe Laíns <lains@archlinux.org> | 2020-06-25 20:40:06 +0100 |
---|---|---|
committer | Tomasz Moń <desowin@gmail.com> | 2020-07-07 14:46:37 +0000 |
commit | 3c37494ea67d8e991dfcfc95645a6b5a07374731 (patch) | |
tree | bff06261babef29c7cbcb84f6ddb74dd810e9253 /epan/dissectors/packet-usb-hid.c | |
parent | 2b7975db547badc3c4c7f4d7111759aeea076466 (diff) |
USB HID: fix memory leak when saving report descriptors
Change-Id: I5cb5598be557823cf3b39cda30bed6febee297d3
Signed-off-by: Filipe Laíns <lains@archlinux.org>
Reviewed-on: https://code.wireshark.org/review/37561
Petri-Dish: Tomasz Moń <desowin@gmail.com>
Tested-by: Petri Dish Buildbot
Reviewed-by: Tomasz Moń <desowin@gmail.com>
Diffstat (limited to 'epan/dissectors/packet-usb-hid.c')
-rw-r--r-- | epan/dissectors/packet-usb-hid.c | 21 |
1 files changed, 14 insertions, 7 deletions
diff --git a/epan/dissectors/packet-usb-hid.c b/epan/dissectors/packet-usb-hid.c index de82cfc641..8a43b79d3e 100644 --- a/epan/dissectors/packet-usb-hid.c +++ b/epan/dissectors/packet-usb-hid.c @@ -3473,9 +3473,7 @@ insert_report_descriptor(packet_info *pinfo, report_descriptor_t *data) {0, NULL} }; - /* only insert report descriptor the first time we parse it */ - if (!PINFO_FD_VISITED(pinfo) && parse_report_descriptor(data)) - wmem_tree_insert32_array(report_descriptors, key, data); + wmem_tree_insert32_array(report_descriptors, key, data); } /* Returns usage page string */ @@ -3948,7 +3946,6 @@ dissect_usb_hid_get_report_descriptor(packet_info *pinfo _U_, proto_tree *parent proto_tree *tree; int old_offset=offset; struct usb_hid_global_state initial_global; - report_descriptor_t *data = wmem_new(wmem_file_scope(), report_descriptor_t); memset(&initial_global, 0, sizeof(struct usb_hid_global_state)); @@ -3957,11 +3954,21 @@ dissect_usb_hid_get_report_descriptor(packet_info *pinfo _U_, proto_tree *parent tree = proto_item_add_subtree(item, ett_usb_hid_report); offset = dissect_usb_hid_report_item(pinfo, tree, tvb, offset, usb_conv_info, &initial_global); - if (usb_conv_info) { + /* only insert report descriptor the first time we parse it */ + if (!PINFO_FD_VISITED(pinfo) && usb_conv_info) { + wmem_allocator_t *scope = wmem_file_scope(); + report_descriptor_t *data = wmem_new(scope, report_descriptor_t); + data->usb_info = *usb_conv_info; data->desc_length = offset - old_offset; - data->desc_body = (guint8*) tvb_memdup(wmem_file_scope(), tvb, old_offset, data->desc_length); - insert_report_descriptor(pinfo, data); + data->desc_body = (guint8*) tvb_memdup(scope, tvb, old_offset, data->desc_length); + + if (parse_report_descriptor(data)) { + insert_report_descriptor(pinfo, data); + } else { + wmem_free(scope, data->desc_body); + wmem_free(scope, data); + } } proto_item_set_len(item, offset-old_offset); |