summaryrefslogtreecommitdiffstats
path: root/fluent-bit/tests/internal/fuzzers/local_test.c
diff options
context:
space:
mode:
Diffstat (limited to 'fluent-bit/tests/internal/fuzzers/local_test.c')
-rw-r--r--fluent-bit/tests/internal/fuzzers/local_test.c85
1 files changed, 85 insertions, 0 deletions
diff --git a/fluent-bit/tests/internal/fuzzers/local_test.c b/fluent-bit/tests/internal/fuzzers/local_test.c
new file mode 100644
index 00000000..309f66b7
--- /dev/null
+++ b/fluent-bit/tests/internal/fuzzers/local_test.c
@@ -0,0 +1,85 @@
+/* -*- Mode: C; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
+
+/* Fluent Bit
+ * ==========
+ * Copyright (C) 2019-2020 The Fluent Bit Authors
+ * Copyright (C) 2015-2018 Treasure Data Inc.
+ *
+ * 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.
+ */
+
+#include <fluent-bit/flb_info.h>
+#include <fluent-bit/flb_log.h>
+#include <fluent-bit/flb_mem.h>
+#include <monkey/mk_core.h>
+
+#include <sys/types.h>
+#include <sys/stat.h>
+#include <unistd.h>
+#include <fcntl.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <errno.h>
+
+/* declare external function test */
+int LLVMFuzzerTestOneInput(unsigned char *data, size_t size);
+
+int main(int argc, char **argv)
+{
+ int i;
+ int ret;
+ FILE *fp;
+ char *buffer;
+ long bytes;
+ struct stat st;
+
+ if (argc < 2) {
+ flb_error("usage: %s TESTCASE_FILE", argv[0]);
+ exit(EXIT_FAILURE);
+ }
+
+ /* Validate the file */
+ ret = stat(argv[1], &st);
+ if (ret == -1) {
+ flb_errno();
+ flb_error("cannot stat(2) testcase file '%s'", argv[1]);
+ exit(EXIT_FAILURE);
+ }
+
+ if (!(fp = fopen(argv[1], "rb"))) {
+ flb_errno();
+ flb_error("cannot fopen(2) testcase file '%s'", argv[1]);
+ return -1;
+ }
+
+ buffer = flb_malloc(st.st_size);
+ if (!buffer) {
+ flb_errno();
+ return -1;
+ }
+
+ bytes = fread(buffer, st.st_size, 1, fp);
+ if (bytes < 1) {
+ fclose(fp);
+ flb_free(buffer);
+ return -1;
+ }
+ fclose(fp);
+
+ /* Invoke the fuzzer entry-point function */
+ for (i = 0; i < 100; i++) {
+ ret = LLVMFuzzerTestOneInput((unsigned char *) buffer, st.st_size);
+ }
+ flb_free(buffer);
+ return 0;
+}