-

Author: vishalcoc44

src/test/java/org/xerial/snappy/fuzz/BitShuffleFuzzer.java

@@ -27,7 +27,7 @@ public class BitShuffleFuzzer {
 
   @FuzzTest
   public static void fuzzerTestOneInput(FuzzedDataProvider data) {
-    switch (data.consumeInt(0, 8)) {
+    switch (data.consumeInt(0, 9)) {
       case 0:
         fuzzBitshuffle(data.consumeInts(SIZE), BitShuffle::shuffle, BitShuffle::unshuffleIntArray, "int[]");
         break;
@@ -55,6 +55,9 @@ public static void fuzzerTestOneInput(FuzzedDataProvider data) {
       case 8:
         fuzzBitshuffle(new short[0], BitShuffle::shuffle, BitShuffle::unshuffleShortArray, "empty short[]");
         break;
+      case 9:
+        fuzzBitshuffle(new long[0], BitShuffle::shuffle, BitShuffle::unshuffleLongArray, "empty long[]");
+        break;
     }
   }
 

src/test/java/org/xerial/snappy/fuzz/SnappyCombinedFuzzer.java

@@ -263,7 +263,7 @@ private static void testFramed(FuzzedDataProvider data) {
     }
 
     private static void testCrc32C(FuzzedDataProvider data) {
-        switch (data.consumeInt(0, 3)) {
+        switch (data.consumeInt(0, 2)) {
             case 0:
                 runFuzz(() -> {
                     byte[] input = data.consumeBytes(data.consumeInt(0, 4096));
@@ -304,28 +304,6 @@ private static void testCrc32C(FuzzedDataProvider data) {
                     }
                 });
                 break;
-            case 3:
-                runFuzz(() -> {
-                    byte[] input = data.consumeBytes(data.consumeInt(0, 4096));
-                    PureJavaCrc32C crc = new PureJavaCrc32C();
-                    crc.update(input, 0, input.length);
-                    long value = crc.getValue();
-                    byte[] serialized = new byte[12];
-                    serialized[0] = (byte) (value >>> 56);
-                    serialized[1] = (byte) (value >>> 48);
-                    serialized[2] = (byte) (value >>> 40);
-                    serialized[3] = (byte) (value >>> 32);
-                    serialized[4] = (byte) (value >>> 24);
-                    serialized[5] = (byte) (value >>> 16);
-                    serialized[6] = (byte) (value >>> 8);
-                    serialized[7] = (byte) value;
-                    crc.reset();
-                    crc.update(serialized, 0, 8);
-                    if (crc.getValue() != value) {
-                        throw new IllegalStateException("CRC32C serialization roundtrip failed");
-                    }
-                });
-                break;
         }
     }
 
@@ -496,7 +474,7 @@ private static void testHadoopStream(FuzzedDataProvider data) {
     }
 
     private static void testByteBuffer(FuzzedDataProvider data) {
-        switch (data.consumeInt(0, 3)) {
+        switch (data.consumeInt(0, 1)) {
             case 0:
                 runFuzz(() -> {
                     byte[] input = data.consumeBytes(data.consumeInt(0, 4096));
@@ -522,28 +500,6 @@ private static void testByteBuffer(FuzzedDataProvider data) {
                 });
                 break;
             case 1:
-                runFuzz(() -> {
-                    byte[] input = data.consumeBytes(data.consumeInt(0, 4096));
-                    ByteBuffer src = ByteBuffer.wrap(input);
-                    ByteBuffer dst = ByteBuffer.allocate(Snappy.maxCompressedLength(input.length));
-                    int compressed = Snappy.compress(src, dst);
-                    
-                    dst.limit(compressed);
-                    dst.position(0);
-                    ByteBuffer uncompressedBuf = ByteBuffer.allocate(input.length);
-                    int uncompressed = Snappy.uncompress(dst, uncompressedBuf);
-                    
-                    uncompressedBuf.limit(uncompressed);
-                    uncompressedBuf.position(0);
-                    byte[] result = new byte[uncompressed];
-                    uncompressedBuf.get(result);
-                    
-                    if (!Arrays.equals(input, result)) {
-                        throw new IllegalStateException("Heap ByteBuffer compress failed");
-                    }
-                });
-                break;
-            case 2:
                 runFuzz(() -> {
                     byte[] input = data.consumeBytes(data.consumeInt(0, 4096));
                     ByteBuffer src = ByteBuffer.allocateDirect(input.length);
@@ -557,20 +513,6 @@ private static void testByteBuffer(FuzzedDataProvider data) {
                     }
                 });
                 break;
-            case 3:
-                runFuzz(() -> {
-                    byte[] input = data.consumeBytes(data.consumeInt(0, 4096));
-                    if (input.length > 0) {
-                        ByteBuffer src = ByteBuffer.wrap(input);
-                        ByteBuffer dst = ByteBuffer.allocate(Snappy.maxCompressedLength(input.length));
-                        try {
-                            Snappy.compress(src, dst);
-                        } catch (Exception e) {
-                            // Expected for invalid input during fuzzing
-                        }
-                    }
-                });
-                break;
         }
     }
 }