xref: /aosp_15_r20/external/llvm/lib/Fuzzer/test/SimpleHashTest.cpp (revision 9880d6810fe72a1726cb53787c6711e909410d58)
1*9880d681SAndroid Build Coastguard Worker // This file is distributed under the University of Illinois Open Source
2*9880d681SAndroid Build Coastguard Worker // License. See LICENSE.TXT for details.
3*9880d681SAndroid Build Coastguard Worker 
4*9880d681SAndroid Build Coastguard Worker // This test computes a checksum of the data (all but the last 4 bytes),
5*9880d681SAndroid Build Coastguard Worker // and then compares the last 4 bytes with the computed value.
6*9880d681SAndroid Build Coastguard Worker // A fuzzer with cmp traces is expected to defeat this check.
7*9880d681SAndroid Build Coastguard Worker #include <cstdint>
8*9880d681SAndroid Build Coastguard Worker #include <cstdlib>
9*9880d681SAndroid Build Coastguard Worker #include <cstring>
10*9880d681SAndroid Build Coastguard Worker #include <cstdio>
11*9880d681SAndroid Build Coastguard Worker 
12*9880d681SAndroid Build Coastguard Worker // A modified jenkins_one_at_a_time_hash initialized by non-zero,
13*9880d681SAndroid Build Coastguard Worker // so that simple_hash(0) != 0. See also
14*9880d681SAndroid Build Coastguard Worker // https://en.wikipedia.org/wiki/Jenkins_hash_function
simple_hash(const uint8_t * Data,size_t Size)15*9880d681SAndroid Build Coastguard Worker static uint32_t simple_hash(const uint8_t *Data, size_t Size) {
16*9880d681SAndroid Build Coastguard Worker   uint32_t Hash = 0x12039854;
17*9880d681SAndroid Build Coastguard Worker   for (uint32_t i = 0; i < Size; i++) {
18*9880d681SAndroid Build Coastguard Worker     Hash += Data[i];
19*9880d681SAndroid Build Coastguard Worker     Hash += (Hash << 10);
20*9880d681SAndroid Build Coastguard Worker     Hash ^= (Hash >> 6);
21*9880d681SAndroid Build Coastguard Worker   }
22*9880d681SAndroid Build Coastguard Worker   Hash += (Hash << 3);
23*9880d681SAndroid Build Coastguard Worker   Hash ^= (Hash >> 11);
24*9880d681SAndroid Build Coastguard Worker   Hash += (Hash << 15);
25*9880d681SAndroid Build Coastguard Worker   return Hash;
26*9880d681SAndroid Build Coastguard Worker }
27*9880d681SAndroid Build Coastguard Worker 
LLVMFuzzerTestOneInput(const uint8_t * Data,size_t Size)28*9880d681SAndroid Build Coastguard Worker extern "C" int LLVMFuzzerTestOneInput(const uint8_t *Data, size_t Size) {
29*9880d681SAndroid Build Coastguard Worker   if (Size < 14)
30*9880d681SAndroid Build Coastguard Worker     return 0;
31*9880d681SAndroid Build Coastguard Worker 
32*9880d681SAndroid Build Coastguard Worker   uint32_t Hash = simple_hash(&Data[0], Size - 4);
33*9880d681SAndroid Build Coastguard Worker   uint32_t Want = reinterpret_cast<const uint32_t *>(&Data[Size - 4])[0];
34*9880d681SAndroid Build Coastguard Worker   if (Hash != Want)
35*9880d681SAndroid Build Coastguard Worker     return 0;
36*9880d681SAndroid Build Coastguard Worker   fprintf(stderr, "BINGO; simple_hash defeated: %x == %x\n", (unsigned int)Hash,
37*9880d681SAndroid Build Coastguard Worker           (unsigned int)Want);
38*9880d681SAndroid Build Coastguard Worker   exit(1);
39*9880d681SAndroid Build Coastguard Worker   return 0;
40*9880d681SAndroid Build Coastguard Worker }
41