1 /* Copyright 2021 Google LLC
2 Licensed under the Apache License, Version 2.0 (the "License");
3 you may not use this file except in compliance with the License.
4 You may obtain a copy of the License at
5 http://www.apache.org/licenses/LICENSE-2.0
6 Unless required by applicable law or agreed to in writing, software
7 distributed under the License is distributed on an "AS IS" BASIS,
8 WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
9 See the License for the specific language governing permissions and
10 limitations under the License.
11 */
12 #include <iostream>
13 #include <bitset>
14 #include <cinttypes>
15 #include <unistd.h>
16 #include "parallel_hashmap/phmap_dump.h"
17 #include <fuzzer/FuzzedDataProvider.h>
18
19 using phmap::flat_hash_map;
20 using namespace std;
21
serialise_test(const uint8_t * data,size_t size)22 void serialise_test(const uint8_t *data, size_t size) {
23 phmap::flat_hash_map<unsigned int, int> table;
24 FuzzedDataProvider fuzzed_data(data, size);
25 const int num_items = fuzzed_data.ConsumeIntegral<int16_t>();
26
27 for (int i=0; i < num_items; ++i) {
28 table.insert(typename phmap::flat_hash_map<unsigned int, int>::value_type(
29 fuzzed_data.ConsumeIntegral<uint32_t>(),
30 fuzzed_data.ConsumeIntegral<int32_t>()));
31 }
32
33 phmap::BinaryOutputArchive ar_out("/dump.data");
34 table.dump(ar_out);
35
36 //MapType table_in;
37 phmap::flat_hash_map<unsigned int, int> table_in;
38 phmap::BinaryInputArchive ar_in("/dump.data");
39 table_in.load(ar_in);
40
41 if(table == table_in) {
42 unlink("/dump.data");
43 return;
44 }
45 unlink("/dump.data");
46 }
47
48 void
test_assignments(const uint8_t * data,size_t size)49 test_assignments(const uint8_t *data, size_t size) {
50 phmap::flat_hash_map<std::string, std::string> email;
51 FuzzedDataProvider fuzzed_data(data, size);
52 const int num_items = fuzzed_data.ConsumeIntegral<int16_t>();
53 for (int i=0; i < num_items; ++i) {
54 phmap::flat_hash_map<std::string, std::string>::value_type(
55 fuzzed_data.ConsumeRandomLengthString(),
56 fuzzed_data.ConsumeRandomLengthString());
57 }
58 // Iterate through all of the items.
59 for (const auto& n: email) {}
60 }
61
LLVMFuzzerTestOneInput(const uint8_t * data,size_t size)62 extern "C" int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size) {
63 serialise_test(data, size);
64 test_assignments(data, size);
65 return 0;
66 }
67