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 com.code_intelligence.jazzer.driver; 16 17 import com.code_intelligence.jazzer.api.FuzzedDataProvider; 18 import java.math.BigDecimal; 19 import java.math.RoundingMode; 20 import java.util.Arrays; 21 import java.util.stream.Collectors; 22 23 public class FuzzedDataProviderImplTest { main(String[] args)24 public static void main(String[] args) { 25 try (FuzzedDataProviderImpl fuzzedDataProvider = 26 FuzzedDataProviderImpl.withJavaData(INPUT_BYTES)) { 27 verifyFuzzedDataProvider(fuzzedDataProvider); 28 } 29 } 30 verifyFuzzedDataProvider(FuzzedDataProvider data)31 private strictfp static void verifyFuzzedDataProvider(FuzzedDataProvider data) { 32 assertEqual(true, data.consumeBoolean()); 33 34 assertEqual((byte) 0x7F, data.consumeByte()); 35 assertEqual((byte) 0x14, data.consumeByte((byte) 0x12, (byte) 0x22)); 36 37 assertEqual(0x12345678, data.consumeInt()); 38 assertEqual(-0x12345600, data.consumeInt(-0x12345678, -0x12345600)); 39 assertEqual(0x12345679, data.consumeInt(0x12345678, 0x12345679)); 40 41 assertEqual(true, Arrays.equals(new byte[] {0x01, 0x02}, data.consumeBytes(2))); 42 43 assertEqual("jazzer", data.consumeString(6)); 44 assertEqual("ja\u0000zer", data.consumeString(6)); 45 assertEqual("€ß", data.consumeString(2)); 46 47 assertEqual("jazzer", data.consumeAsciiString(6)); 48 assertEqual("ja\u0000zer", data.consumeAsciiString(6)); 49 assertEqual("\u0062\u0002\u002C\u0043\u001F", data.consumeAsciiString(5)); 50 51 assertEqual(true, 52 Arrays.equals(new boolean[] {false, false, true, false, true}, data.consumeBooleans(5))); 53 assertEqual(true, 54 Arrays.equals(new long[] {0x0123456789abdcefL, 0xfedcba9876543210L}, data.consumeLongs(2))); 55 56 assertAtLeastAsPrecise((float) 0.28969181, data.consumeProbabilityFloat()); 57 assertAtLeastAsPrecise(0.086814121166605432, data.consumeProbabilityDouble()); 58 assertAtLeastAsPrecise((float) 0.30104411, data.consumeProbabilityFloat()); 59 assertAtLeastAsPrecise(0.96218831486039413, data.consumeProbabilityDouble()); 60 61 assertAtLeastAsPrecise((float) -2.8546307e+38, data.consumeRegularFloat()); 62 assertAtLeastAsPrecise(8.0940194040236032e+307, data.consumeRegularDouble()); 63 assertAtLeastAsPrecise( 64 (float) 271.49084, data.consumeRegularFloat((float) 123.0, (float) 777.0)); 65 assertAtLeastAsPrecise(30.859126145478349, data.consumeRegularDouble(13.37, 31.337)); 66 67 assertEqual((float) 0.0, data.consumeFloat()); 68 assertEqual((float) -0.0, data.consumeFloat()); 69 assertEqual(Float.POSITIVE_INFINITY, data.consumeFloat()); 70 assertEqual(Float.NEGATIVE_INFINITY, data.consumeFloat()); 71 assertEqual(true, Float.isNaN(data.consumeFloat())); 72 assertEqual(Float.MIN_VALUE, data.consumeFloat()); 73 assertEqual(-Float.MIN_VALUE, data.consumeFloat()); 74 assertEqual(Float.MIN_NORMAL, data.consumeFloat()); 75 assertEqual(-Float.MIN_NORMAL, data.consumeFloat()); 76 assertEqual(Float.MAX_VALUE, data.consumeFloat()); 77 assertEqual(-Float.MAX_VALUE, data.consumeFloat()); 78 79 assertEqual(0.0, data.consumeDouble()); 80 assertEqual(-0.0, data.consumeDouble()); 81 assertEqual(Double.POSITIVE_INFINITY, data.consumeDouble()); 82 assertEqual(Double.NEGATIVE_INFINITY, data.consumeDouble()); 83 assertEqual(true, Double.isNaN(data.consumeDouble())); 84 assertEqual(Double.MIN_VALUE, data.consumeDouble()); 85 assertEqual(-Double.MIN_VALUE, data.consumeDouble()); 86 assertEqual(Double.MIN_NORMAL, data.consumeDouble()); 87 assertEqual(-Double.MIN_NORMAL, data.consumeDouble()); 88 assertEqual(Double.MAX_VALUE, data.consumeDouble()); 89 assertEqual(-Double.MAX_VALUE, data.consumeDouble()); 90 91 int[] array = {0, 1, 2, 3, 4}; 92 assertEqual(4, data.pickValue(array)); 93 assertEqual(2, (int) data.pickValue(Arrays.stream(array).boxed().toArray())); 94 assertEqual(3, data.pickValue(Arrays.stream(array).boxed().collect(Collectors.toList()))); 95 assertEqual(2, data.pickValue(Arrays.stream(array).boxed().collect(Collectors.toSet()))); 96 97 // Buffer is almost depleted at this point. 98 assertEqual(7, data.remainingBytes()); 99 assertEqual(true, Arrays.equals(new long[0], data.consumeLongs(3))); 100 assertEqual(7, data.remainingBytes()); 101 assertEqual(true, Arrays.equals(new int[] {0x12345678}, data.consumeInts(3))); 102 assertEqual(3, data.remainingBytes()); 103 assertEqual(0x123456L, data.consumeLong()); 104 105 // Buffer has been fully consumed at this point 106 assertEqual(0, data.remainingBytes()); 107 assertEqual(0, data.consumeInt()); 108 assertEqual(0.0, data.consumeDouble()); 109 assertEqual(-13.37, data.consumeRegularDouble(-13.37, 31.337)); 110 assertEqual(true, Arrays.equals(new byte[0], data.consumeBytes(4))); 111 assertEqual(true, Arrays.equals(new long[0], data.consumeLongs(4))); 112 assertEqual("", data.consumeRemainingAsAsciiString()); 113 assertEqual("", data.consumeRemainingAsString()); 114 assertEqual("", data.consumeAsciiString(100)); 115 assertEqual("", data.consumeString(100)); 116 } 117 assertAtLeastAsPrecise(double expected, double actual)118 private static void assertAtLeastAsPrecise(double expected, double actual) { 119 BigDecimal exactExpected = BigDecimal.valueOf(expected); 120 BigDecimal roundedActual = 121 BigDecimal.valueOf(actual).setScale(exactExpected.scale(), RoundingMode.HALF_UP); 122 if (!exactExpected.equals(roundedActual)) { 123 throw new IllegalArgumentException( 124 String.format("Expected: %s, got: %s (rounded: %s)", expected, actual, roundedActual)); 125 } 126 } 127 assertEqual(T a, T b)128 private static <T extends Comparable<T>> void assertEqual(T a, T b) { 129 if (a.compareTo(b) != 0) { 130 throw new IllegalArgumentException("Expected: " + a + ", got: " + b); 131 } 132 } 133 134 private static final byte[] INPUT_BYTES = new byte[] { 135 // Bytes read from the start 136 0x01, 0x02, // consumeBytes(2): {0x01, 0x02} 137 138 'j', 'a', 'z', 'z', 'e', 'r', // consumeString(6): "jazzer" 139 'j', 'a', 0x00, 'z', 'e', 'r', // consumeString(6): "ja\u0000zer" 140 (byte) 0xE2, (byte) 0x82, (byte) 0xAC, (byte) 0xC3, (byte) 0x9F, // consumeString(2): "€ẞ" 141 142 'j', 'a', 'z', 'z', 'e', 'r', // consumeAsciiString(6): "jazzer" 143 'j', 'a', 0x00, 'z', 'e', 'r', // consumeAsciiString(6): "ja\u0000zer" 144 (byte) 0xE2, (byte) 0x82, (byte) 0xAC, (byte) 0xC3, 145 (byte) 0x9F, // consumeAsciiString(5): "\u0062\u0002\u002C\u0043\u001F" 146 147 0, 0, 1, 0, 1, // consumeBooleans(5): { false, false, true, false, true } 148 (byte) 0xEF, (byte) 0xDC, (byte) 0xAB, (byte) 0x89, 0x67, 0x45, 0x23, 0x01, 0x10, 0x32, 0x54, 149 0x76, (byte) 0x98, (byte) 0xBA, (byte) 0xDC, (byte) 0xFE, 150 // consumeLongs(2): { 0x0123456789ABCDEF, 0xFEDCBA9876543210 } 151 152 0x78, 0x56, 0x34, 0x12, // consumeInts(3): { 0x12345678 } 153 0x56, 0x34, 0x12, // consumeLong(): 154 155 // Bytes read from the end 156 0x02, 0x03, 0x02, 0x04, // 4x pickValue in array with five elements 157 158 0x12, 0x34, 0x56, 0x78, (byte) 0x90, 0x12, 0x34, 0x56, 159 0x78, // consumed but unused by consumeDouble() 160 10, // -max for next consumeDouble 161 0x12, 0x34, 0x56, 0x78, (byte) 0x90, 0x12, 0x34, 0x56, 162 0x78, // consumed but unused by consumeDouble() 163 9, // max for next consumeDouble 164 0x12, 0x34, 0x56, 0x78, (byte) 0x90, 0x12, 0x34, 0x56, 165 0x78, // consumed but unused by consumeDouble() 166 8, // -min for next consumeDouble 167 0x12, 0x34, 0x56, 0x78, (byte) 0x90, 0x12, 0x34, 0x56, 168 0x78, // consumed but unused by consumeDouble() 169 7, // min for next consumeDouble 170 0x12, 0x34, 0x56, 0x78, (byte) 0x90, 0x12, 0x34, 0x56, 171 0x78, // consumed but unused by consumeDouble() 172 6, // -denorm_min for next consumeDouble 173 0x12, 0x34, 0x56, 0x78, (byte) 0x90, 0x12, 0x34, 0x56, 174 0x78, // consumed but unused by consumeDouble() 175 5, // denorm_min for next consumeDouble 176 0x12, 0x34, 0x56, 0x78, (byte) 0x90, 0x12, 0x34, 0x56, 177 0x78, // consumed but unused by consumeDouble() 178 4, // NaN for next consumeDouble 179 0x12, 0x34, 0x56, 0x78, (byte) 0x90, 0x12, 0x34, 0x56, 180 0x78, // consumed but unused by consumeDouble() 181 3, // -infinity for next consumeDouble 182 0x12, 0x34, 0x56, 0x78, (byte) 0x90, 0x12, 0x34, 0x56, 183 0x78, // consumed but unused by consumeDouble() 184 2, // infinity for next consumeDouble 185 0x12, 0x34, 0x56, 0x78, (byte) 0x90, 0x12, 0x34, 0x56, 186 0x78, // consumed but unused by consumeDouble() 187 1, // -0.0 for next consumeDouble 188 0x12, 0x34, 0x56, 0x78, (byte) 0x90, 0x12, 0x34, 0x56, 189 0x78, // consumed but unused by consumeDouble() 190 0, // 0.0 for next consumeDouble 191 192 0x12, 0x34, 0x56, 0x78, (byte) 0x90, // consumed but unused by consumeFloat() 193 10, // -max for next consumeFloat 194 0x12, 0x34, 0x56, 0x78, (byte) 0x90, // consumed but unused by consumeFloat() 195 9, // max for next consumeFloat 196 0x12, 0x34, 0x56, 0x78, (byte) 0x90, // consumed but unused by consumeFloat() 197 8, // -min for next consumeFloat 198 0x12, 0x34, 0x56, 0x78, (byte) 0x90, // consumed but unused by consumeFloat() 199 7, // min for next consumeFloat 200 0x12, 0x34, 0x56, 0x78, (byte) 0x90, // consumed but unused by consumeFloat() 201 6, // -denorm_min for next consumeFloat 202 0x12, 0x34, 0x56, 0x78, (byte) 0x90, // consumed but unused by consumeFloat() 203 5, // denorm_min for next consumeFloat 204 0x12, 0x34, 0x56, 0x78, (byte) 0x90, // consumed but unused by consumeFloat() 205 4, // NaN for next consumeFloat 206 0x12, 0x34, 0x56, 0x78, (byte) 0x90, // consumed but unused by consumeFloat() 207 3, // -infinity for next consumeFloat 208 0x12, 0x34, 0x56, 0x78, (byte) 0x90, // consumed but unused by consumeFloat() 209 2, // infinity for next consumeFloat 210 0x12, 0x34, 0x56, 0x78, (byte) 0x90, // consumed but unused by consumeFloat() 211 1, // -0.0 for next consumeFloat 212 0x12, 0x34, 0x56, 0x78, (byte) 0x90, // consumed but unused by consumeFloat() 213 0, // 0.0 for next consumeFloat 214 215 (byte) 0x88, (byte) 0xAB, 0x61, (byte) 0xCB, 0x32, (byte) 0xEB, 0x30, (byte) 0xF9, 216 // consumeDouble(13.37, 31.337): 30.859126145478349 (small range) 217 0x51, (byte) 0xF6, 0x1F, 0x3A, // consumeFloat(123.0, 777.0): 271.49084 (small range) 218 0x11, 0x4D, (byte) 0xFD, 0x54, (byte) 0xD6, 0x3D, 0x43, 0x73, 0x39, 219 // consumeRegularDouble(): 8.0940194040236032e+307 220 0x16, (byte) 0xCF, 0x3D, 0x29, 0x4A, // consumeRegularFloat(): -2.8546307e+38 221 222 0x61, (byte) 0xCB, 0x32, (byte) 0xEB, 0x30, (byte) 0xF9, 0x51, (byte) 0xF6, 223 // consumeProbabilityDouble(): 0.96218831486039413 224 0x1F, 0x3A, 0x11, 0x4D, // consumeProbabilityFloat(): 0.30104411 225 (byte) 0xFD, 0x54, (byte) 0xD6, 0x3D, 0x43, 0x73, 0x39, 0x16, 226 // consumeProbabilityDouble(): 0.086814121166605432 227 (byte) 0xCF, 0x3D, 0x29, 0x4A, // consumeProbabilityFloat(): 0.28969181 228 229 0x01, // consumeInt(0x12345678, 0x12345679): 0x12345679 230 0x78, // consumeInt(-0x12345678, -0x12345600): -0x12345600 231 0x78, 0x56, 0x34, 0x12, // consumeInt(): 0x12345678 232 233 0x02, // consumeByte(0x12, 0x22): 0x14 234 0x7F, // consumeByte(): 0x7F 235 236 0x01, // consumeBool(): true 237 }; 238 } 239