From 13dac0ee2a4534574fcc566009e7810ae7608e33 Mon Sep 17 00:00:00 2001 From: akallabeth Date: Wed, 15 Apr 2020 17:49:41 +0200 Subject: [PATCH] Fixed possible int overflow. --- libfreerdp/codec/progressive.c | 22 ++++++++++++---------- 1 file changed, 12 insertions(+), 10 deletions(-) diff --git a/libfreerdp/codec/progressive.c b/libfreerdp/codec/progressive.c index bb58ed1a8..bfa2dc5da 100644 --- a/libfreerdp/codec/progressive.c +++ b/libfreerdp/codec/progressive.c @@ -1920,7 +1920,7 @@ static INLINE INT32 progressive_wb_read_region_header(PROGRESSIVE_CONTEXT* progr UINT16 blockType, UINT32 blockLen, PROGRESSIVE_BLOCK_REGION* region) { - size_t offset, len; + size_t len; memset(region, 0, sizeof(PROGRESSIVE_BLOCK_REGION)); if (Stream_GetRemainingLength(s) < 12) @@ -1965,35 +1965,37 @@ static INLINE INT32 progressive_wb_read_region_header(PROGRESSIVE_CONTEXT* progr } len = Stream_GetRemainingLength(s); - offset = (region->numRects * 8); - if (len < offset) + if (len / 8 < region->numRects) { WLog_Print(progressive->log, WLOG_ERROR, "ProgressiveRegion data short for region->rects"); return -1015; } + len -= region->numRects * 8ULL; - offset += (region->numQuant * 5); - if (len < offset) + if (len / 5 < region->numQuant) { WLog_Print(progressive->log, WLOG_ERROR, "ProgressiveRegion data short for region->cQuant"); return -1018; } + len -= region->numQuant * 5ULL; - offset += (region->numProgQuant * 16); - if (len < offset) + if (len / 16 < region->numProgQuant) { WLog_Print(progressive->log, WLOG_ERROR, "ProgressiveRegion data short for region->cProgQuant"); return -1021; } + len -= region->numProgQuant * 16ULL; - offset += region->tileDataSize; - if (len < offset) + if (len < region->tileDataSize) { WLog_Print(progressive->log, WLOG_ERROR, "ProgressiveRegion data short for region->tiles"); return -1024; } - + len -= region->tileDataSize; + if (len > 0) + WLog_Print(progressive->log, WLOG_DEBUG, + "Unused byes detected, %" PRIuz " bytes not processed", len); return 0; }