Line | Count | Source |
1 | | /* Copyright 2025 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 | | |
13 | | #include <stdint.h> |
14 | | #include <stdio.h> |
15 | | #include <stdlib.h> |
16 | | |
17 | | uint8_t *global_fuzz_data; |
18 | | size_t global_size; |
19 | | |
20 | | char *callback1 = NULL; |
21 | | char *callback2 = NULL; |
22 | | char *callback3 = NULL; |
23 | | char *callback4 = NULL; |
24 | | |
25 | 118 | void fuzz_init(const uint8_t *data, size_t size) { |
26 | 118 | global_fuzz_data = (uint8_t *)data; |
27 | 118 | global_size = size; |
28 | | |
29 | 118 | callback1 = NULL; |
30 | 118 | callback2 = NULL; |
31 | 118 | callback3 = NULL; |
32 | 118 | callback4 = NULL; |
33 | 118 | } |
34 | | |
35 | | #ifdef ENABLE_FILE_CALLBACK |
36 | | void create_callback_file(const char *filename) { |
37 | | fprintf(stderr, "create_callback_file: Creating file %s\n", filename); |
38 | | FILE *fp = fopen(filename, "wb"); |
39 | | if (!fp) { |
40 | | return; |
41 | | } |
42 | | fwrite(global_fuzz_data, global_size, 1, fp); |
43 | | fclose(fp); |
44 | | |
45 | | if (callback1 == NULL) { |
46 | | callback1 = strdup(filename); |
47 | | } else if (callback2 == NULL) { |
48 | | callback2 = strdup(filename); |
49 | | } else if (callback3 == NULL) { |
50 | | callback3 = strdup(filename); |
51 | | } else if (callback4 == NULL) { |
52 | | callback4 = strdup(filename); |
53 | | } else { |
54 | | fprintf(stderr, "create_callback_file: Too many callbacks created, ignoring %s\n", filename); |
55 | | } |
56 | | } |
57 | | |
58 | | #else |
59 | 9.44k | void create_callback_file(const char *filename) {} |
60 | | #endif |
61 | | |
62 | 118 | void fuzz_cleanup() { |
63 | 118 | if (callback1) { |
64 | 94 | unlink(callback1); |
65 | 94 | free(callback1); |
66 | | |
67 | 94 | callback1 = NULL; |
68 | 94 | } |
69 | 118 | if (callback2) { |
70 | 67 | unlink(callback2); |
71 | 67 | free(callback2); |
72 | 67 | callback2 = NULL; |
73 | 67 | } |
74 | 118 | if (callback3) { |
75 | 49 | unlink(callback3); |
76 | 49 | free(callback3); |
77 | 49 | callback3 = NULL; |
78 | 49 | } |
79 | 118 | if (callback4) { |
80 | 9 | unlink(callback4); |
81 | 9 | free(callback4); |
82 | 9 | callback4 = NULL; |
83 | 9 | } |
84 | 118 | } |