diff options
author | 2016-11-29 10:58:16 -0800 | |
---|---|---|
committer | 2016-11-29 10:58:31 -0800 | |
commit | c03c92cce0b04ba2be95a8ac421c9d41777e8e7d (patch) | |
tree | a53a87674514234812f01a4c95208cc4554c6c84 /projects/file/magic_fuzzer.cc | |
parent | ea87305a11ab9654ea315c666f581785a862dcb6 (diff) | |
parent | ef765503cb3bbf7d2f82cdf01ccc033f6008ac91 (diff) |
Merge branch 'master' of https://github.com/google/oss-fuzz
Diffstat (limited to 'projects/file/magic_fuzzer.cc')
-rw-r--r-- | projects/file/magic_fuzzer.cc | 51 |
1 files changed, 51 insertions, 0 deletions
diff --git a/projects/file/magic_fuzzer.cc b/projects/file/magic_fuzzer.cc new file mode 100644 index 00000000..1f5b5f09 --- /dev/null +++ b/projects/file/magic_fuzzer.cc @@ -0,0 +1,51 @@ +// Copyright 2016 Google 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 <libgen.h> +#include <stddef.h> +#include <stdint.h> +#include <stdio.h> +#include <stdlib.h> +#include <string> + +#include <magic.h> + +struct Environment { + Environment(std::string data_dir) { + magic = magic_open(MAGIC_NONE); + std::string magic_path = data_dir + "/magic"; + if (magic_load(magic, magic_path.c_str())) { + fprintf(stderr, "error loading magic file: %s\n", magic_error(magic)); + exit(1); + } + } + + magic_t magic; +}; + +static Environment* env; + +extern "C" int LLVMFuzzerInitialize(int* argc, char*** argv) { + char* exe_path = (*argv)[0]; + char* dir = dirname(exe_path); + env = new Environment(dir); + return 0; +} + +extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size) { + if (size < 1) + return 0; + magic_buffer(env->magic, data, size); + return 0; +} |