]> git.ipfire.org Git - thirdparty/zstd.git/commitdiff
Fix bitshift error
authorStella Lau <laus@fb.com>
Thu, 29 Jun 2017 22:40:49 +0000 (15:40 -0700)
committerStella Lau <laus@fb.com>
Thu, 29 Jun 2017 22:40:49 +0000 (15:40 -0700)
lib/decompress/huf_decompress.c

index cbf78e807264f9c6ab9d37097f2c33a7e0236885..68938aebd2de0d40b282e4ef86e0090aa8b2381b 100644 (file)
@@ -110,8 +110,8 @@ size_t HUF_readDTableX2_wksp(HUF_DTable* DTable, const void* src, size_t srcSize
     huffWeight = (BYTE *)((U32 *)workSpace + spaceUsed32);
     spaceUsed32 += ALIGN(HUF_SYMBOLVALUE_MAX + 1, sizeof(U32)) >> 2;
 
-    if ((spaceUsed32 >> 2) > wkspSize)
-            return ERROR(tableLog_tooLarge);
+    if ((spaceUsed32 << 2) > wkspSize)
+        return ERROR(tableLog_tooLarge);
     workSpace = (U32 *)workSpace + spaceUsed32;
     wkspSize -= (spaceUsed32 << 2);
 
@@ -524,7 +524,7 @@ size_t HUF_readDTableX4_wksp(HUF_DTable* DTable, const void* src,
     weightList = (BYTE *)((U32 *)workSpace + spaceUsed32);
     spaceUsed32 += ALIGN(HUF_SYMBOLVALUE_MAX + 1, sizeof(U32)) >> 2;
 
-    if ((spaceUsed32 >> 2) > wkspSize)
+    if ((spaceUsed32 << 2) > wkspSize)
         return ERROR(tableLog_tooLarge);
     workSpace = (U32 *)workSpace + spaceUsed32;
     wkspSize -= (spaceUsed32 << 2);