115 lines
		
	
	
		
			5.2 KiB
		
	
	
	
		
			Java
		
	
	
	
			
		
		
	
	
			115 lines
		
	
	
		
			5.2 KiB
		
	
	
	
		
			Java
		
	
	
	
| // Copyright 2021 Code Intelligence GmbH
 | |
| //
 | |
| // Licensed under the Apache License, Version 2.0 (the "License");
 | |
| // you may not use this file except in compliance with the License.
 | |
| // You may obtain a copy of the License at
 | |
| //
 | |
| //      http://www.apache.org/licenses/LICENSE-2.0
 | |
| //
 | |
| // Unless required by applicable law or agreed to in writing, software
 | |
| // distributed under the License is distributed on an "AS IS" BASIS,
 | |
| // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 | |
| // See the License for the specific language governing permissions and
 | |
| // limitations under the License.
 | |
| 
 | |
| package test;
 | |
| 
 | |
| import com.code_intelligence.jazzer.api.FuzzedDataProvider;
 | |
| import java.util.ArrayList;
 | |
| import java.util.Arrays;
 | |
| import java.util.stream.Collectors;
 | |
| 
 | |
| class FuzzTargetWithDataProvider {
 | |
|   public static <T extends Comparable<T>> void assertEqual(T a, T b) {
 | |
|     if (a.compareTo(b) != 0) {
 | |
|       throw new IllegalArgumentException("Expected: " + a + ", got: " + b);
 | |
|     }
 | |
|   }
 | |
| 
 | |
|   public strictfp static void fuzzerTestOneInput(FuzzedDataProvider data) {
 | |
|     assertEqual(true, data.consumeBoolean());
 | |
| 
 | |
|     assertEqual((byte) 0x7F, data.consumeByte());
 | |
|     assertEqual((byte) 0x14, data.consumeByte((byte) 0x12, (byte) 0x22));
 | |
| 
 | |
|     assertEqual(0x12345678, data.consumeInt());
 | |
|     assertEqual(-0x12345600, data.consumeInt(-0x12345678, -0x12345600));
 | |
|     assertEqual(0x12345679, data.consumeInt(0x12345678, 0x12345679));
 | |
| 
 | |
|     assertEqual(true, Arrays.equals(new byte[] {0x01, 0x02}, data.consumeBytes(2)));
 | |
| 
 | |
|     assertEqual("jazzer", data.consumeString(6));
 | |
|     assertEqual("ja\u0000zer", data.consumeString(6));
 | |
|     assertEqual("ۧ", data.consumeString(2));
 | |
| 
 | |
|     assertEqual("jazzer", data.consumeAsciiString(6));
 | |
|     assertEqual("ja\u0000zer", data.consumeAsciiString(6));
 | |
|     assertEqual("\u0062\u0002\u002C\u0043\u001F", data.consumeAsciiString(5));
 | |
| 
 | |
|     assertEqual(true,
 | |
|         Arrays.equals(new boolean[] {false, false, true, false, true}, data.consumeBooleans(5)));
 | |
|     assertEqual(true,
 | |
|         Arrays.equals(new long[] {0x0123456789abdcefL, 0xfedcba9876543210L}, data.consumeLongs(2)));
 | |
| 
 | |
|     assertEqual((float) 0.28969181, data.consumeProbabilityFloat());
 | |
|     assertEqual(0.086814121166605432, data.consumeProbabilityDouble());
 | |
|     assertEqual((float) 0.30104411, data.consumeProbabilityFloat());
 | |
|     assertEqual(0.96218831486039413, data.consumeProbabilityDouble());
 | |
| 
 | |
|     assertEqual((float) -2.8546307e+38, data.consumeRegularFloat());
 | |
|     assertEqual(8.0940194040236032e+307, data.consumeRegularDouble());
 | |
|     assertEqual((float) 271.49084, data.consumeRegularFloat((float) 123.0, (float) 777.0));
 | |
|     assertEqual(30.859126145478349, data.consumeRegularDouble(13.37, 31.337));
 | |
| 
 | |
|     assertEqual((float) 0.0, data.consumeFloat());
 | |
|     assertEqual((float) -0.0, data.consumeFloat());
 | |
|     assertEqual(Float.POSITIVE_INFINITY, data.consumeFloat());
 | |
|     assertEqual(Float.NEGATIVE_INFINITY, data.consumeFloat());
 | |
|     assertEqual(true, Float.isNaN(data.consumeFloat()));
 | |
|     assertEqual(Float.MIN_VALUE, data.consumeFloat());
 | |
|     assertEqual(-Float.MIN_VALUE, data.consumeFloat());
 | |
|     assertEqual(Float.MIN_NORMAL, data.consumeFloat());
 | |
|     assertEqual(-Float.MIN_NORMAL, data.consumeFloat());
 | |
|     assertEqual(Float.MAX_VALUE, data.consumeFloat());
 | |
|     assertEqual(-Float.MAX_VALUE, data.consumeFloat());
 | |
| 
 | |
|     assertEqual(0.0, data.consumeDouble());
 | |
|     assertEqual(-0.0, data.consumeDouble());
 | |
|     assertEqual(Double.POSITIVE_INFINITY, data.consumeDouble());
 | |
|     assertEqual(Double.NEGATIVE_INFINITY, data.consumeDouble());
 | |
|     assertEqual(true, Double.isNaN(data.consumeDouble()));
 | |
|     assertEqual(Double.MIN_VALUE, data.consumeDouble());
 | |
|     assertEqual(-Double.MIN_VALUE, data.consumeDouble());
 | |
|     assertEqual(Double.MIN_NORMAL, data.consumeDouble());
 | |
|     assertEqual(-Double.MIN_NORMAL, data.consumeDouble());
 | |
|     assertEqual(Double.MAX_VALUE, data.consumeDouble());
 | |
|     assertEqual(-Double.MAX_VALUE, data.consumeDouble());
 | |
| 
 | |
|     int[] array = {0, 1, 2, 3, 4};
 | |
|     assertEqual(4, data.pickValue(array));
 | |
|     assertEqual(2, (int) data.pickValue(Arrays.stream(array).boxed().toArray()));
 | |
|     assertEqual(3, data.pickValue(Arrays.stream(array).boxed().collect(Collectors.toList())));
 | |
|     assertEqual(2, data.pickValue(Arrays.stream(array).boxed().collect(Collectors.toSet())));
 | |
| 
 | |
|     // Buffer is almost depleted at this point.
 | |
|     assertEqual(7, data.remainingBytes());
 | |
|     assertEqual(true, Arrays.equals(new long[0], data.consumeLongs(3)));
 | |
|     assertEqual(7, data.remainingBytes());
 | |
|     assertEqual(true, Arrays.equals(new int[] {0x12345678}, data.consumeInts(3)));
 | |
|     assertEqual(3, data.remainingBytes());
 | |
|     assertEqual(0x123456L, data.consumeLong());
 | |
| 
 | |
|     // Buffer has been fully consumed at this point
 | |
|     assertEqual(0, data.remainingBytes());
 | |
|     assertEqual(0, data.consumeInt());
 | |
|     assertEqual(0.0, data.consumeDouble());
 | |
|     assertEqual(-13.37, data.consumeRegularDouble(-13.37, 31.337));
 | |
|     assertEqual(true, Arrays.equals(new byte[0], data.consumeBytes(4)));
 | |
|     assertEqual(true, Arrays.equals(new long[0], data.consumeLongs(4)));
 | |
|     assertEqual("", data.consumeRemainingAsAsciiString());
 | |
|     assertEqual("", data.consumeRemainingAsString());
 | |
|     assertEqual("", data.consumeAsciiString(100));
 | |
|     assertEqual("", data.consumeString(100));
 | |
|   }
 | |
| }
 |