1 // Copyright 2021 Code Intelligence GmbH 2 // 3 // Licensed under the Apache License, Version 2.0 (the "License"); 4 // you may not use this file except in compliance with the License. 5 // You may obtain a copy of the License at 6 // 7 // http://www.apache.org/licenses/LICENSE-2.0 8 // 9 // Unless required by applicable law or agreed to in writing, software 10 // distributed under the License is distributed on an "AS IS" BASIS, 11 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12 // See the License for the specific language governing permissions and 13 // limitations under the License. 14 15 package test; 16 17 import com.code_intelligence.jazzer.api.FuzzedDataProvider; 18 import java.util.ArrayList; 19 import java.util.Arrays; 20 import java.util.stream.Collectors; 21 22 class FuzzTargetWithDataProvider { assertEqual(T a, T b)23 public static <T extends Comparable<T>> void assertEqual(T a, T b) { 24 if (a.compareTo(b) != 0) { 25 throw new IllegalArgumentException("Expected: " + a + ", got: " + b); 26 } 27 } 28 fuzzerTestOneInput(FuzzedDataProvider data)29 public strictfp static void fuzzerTestOneInput(FuzzedDataProvider data) { 30 assertEqual(true, data.consumeBoolean()); 31 32 assertEqual((byte) 0x7F, data.consumeByte()); 33 assertEqual((byte) 0x14, data.consumeByte((byte) 0x12, (byte) 0x22)); 34 35 assertEqual(0x12345678, data.consumeInt()); 36 assertEqual(-0x12345600, data.consumeInt(-0x12345678, -0x12345600)); 37 assertEqual(0x12345679, data.consumeInt(0x12345678, 0x12345679)); 38 39 assertEqual(true, Arrays.equals(new byte[] {0x01, 0x02}, data.consumeBytes(2))); 40 41 assertEqual("jazzer", data.consumeString(6)); 42 assertEqual("ja\u0000zer", data.consumeString(6)); 43 assertEqual("ۧ", data.consumeString(2)); 44 45 assertEqual("jazzer", data.consumeAsciiString(6)); 46 assertEqual("ja\u0000zer", data.consumeAsciiString(6)); 47 assertEqual("\u0062\u0002\u002C\u0043\u001F", data.consumeAsciiString(5)); 48 49 assertEqual(true, 50 Arrays.equals(new boolean[] {false, false, true, false, true}, data.consumeBooleans(5))); 51 assertEqual(true, 52 Arrays.equals(new long[] {0x0123456789abdcefL, 0xfedcba9876543210L}, data.consumeLongs(2))); 53 54 assertEqual((float) 0.28969181, data.consumeProbabilityFloat()); 55 assertEqual(0.086814121166605432, data.consumeProbabilityDouble()); 56 assertEqual((float) 0.30104411, data.consumeProbabilityFloat()); 57 assertEqual(0.96218831486039413, data.consumeProbabilityDouble()); 58 59 assertEqual((float) -2.8546307e+38, data.consumeRegularFloat()); 60 assertEqual(8.0940194040236032e+307, data.consumeRegularDouble()); 61 assertEqual((float) 271.49084, data.consumeRegularFloat((float) 123.0, (float) 777.0)); 62 assertEqual(30.859126145478349, data.consumeRegularDouble(13.37, 31.337)); 63 64 assertEqual((float) 0.0, data.consumeFloat()); 65 assertEqual((float) -0.0, data.consumeFloat()); 66 assertEqual(Float.POSITIVE_INFINITY, data.consumeFloat()); 67 assertEqual(Float.NEGATIVE_INFINITY, data.consumeFloat()); 68 assertEqual(true, Float.isNaN(data.consumeFloat())); 69 assertEqual(Float.MIN_VALUE, data.consumeFloat()); 70 assertEqual(-Float.MIN_VALUE, data.consumeFloat()); 71 assertEqual(Float.MIN_NORMAL, data.consumeFloat()); 72 assertEqual(-Float.MIN_NORMAL, data.consumeFloat()); 73 assertEqual(Float.MAX_VALUE, data.consumeFloat()); 74 assertEqual(-Float.MAX_VALUE, data.consumeFloat()); 75 76 assertEqual(0.0, data.consumeDouble()); 77 assertEqual(-0.0, data.consumeDouble()); 78 assertEqual(Double.POSITIVE_INFINITY, data.consumeDouble()); 79 assertEqual(Double.NEGATIVE_INFINITY, data.consumeDouble()); 80 assertEqual(true, Double.isNaN(data.consumeDouble())); 81 assertEqual(Double.MIN_VALUE, data.consumeDouble()); 82 assertEqual(-Double.MIN_VALUE, data.consumeDouble()); 83 assertEqual(Double.MIN_NORMAL, data.consumeDouble()); 84 assertEqual(-Double.MIN_NORMAL, data.consumeDouble()); 85 assertEqual(Double.MAX_VALUE, data.consumeDouble()); 86 assertEqual(-Double.MAX_VALUE, data.consumeDouble()); 87 88 int[] array = {0, 1, 2, 3, 4}; 89 assertEqual(4, data.pickValue(array)); 90 assertEqual(2, (int) data.pickValue(Arrays.stream(array).boxed().toArray())); 91 assertEqual(3, data.pickValue(Arrays.stream(array).boxed().collect(Collectors.toList()))); 92 assertEqual(2, data.pickValue(Arrays.stream(array).boxed().collect(Collectors.toSet()))); 93 94 // Buffer is almost depleted at this point. 95 assertEqual(7, data.remainingBytes()); 96 assertEqual(true, Arrays.equals(new long[0], data.consumeLongs(3))); 97 assertEqual(7, data.remainingBytes()); 98 assertEqual(true, Arrays.equals(new int[] {0x12345678}, data.consumeInts(3))); 99 assertEqual(3, data.remainingBytes()); 100 assertEqual(0x123456L, data.consumeLong()); 101 102 // Buffer has been fully consumed at this point 103 assertEqual(0, data.remainingBytes()); 104 assertEqual(0, data.consumeInt()); 105 assertEqual(0.0, data.consumeDouble()); 106 assertEqual(-13.37, data.consumeRegularDouble(-13.37, 31.337)); 107 assertEqual(true, Arrays.equals(new byte[0], data.consumeBytes(4))); 108 assertEqual(true, Arrays.equals(new long[0], data.consumeLongs(4))); 109 assertEqual("", data.consumeRemainingAsAsciiString()); 110 assertEqual("", data.consumeRemainingAsString()); 111 assertEqual("", data.consumeAsciiString(100)); 112 assertEqual("", data.consumeString(100)); 113 } 114 } 115