blob: 1de128cfbab502ce5f07ee197a526a5f69318226 [file] [log] [blame]
aizatskyc562c132016-01-20 00:00:141// Copyright 2015 The Chromium Authors. All rights reserved.
2// Use of this source code is governed by a BSD-style license that can be
3// found in the LICENSE file.
4
5// A simple unit-test style driver for libfuzzer tests.
6// Usage: <fuzzer_test> <file>...
7
8#include <fstream>
9#include <iostream>
10#include <iterator>
11#include <vector>
12
aizatskye4599ba52016-01-29 23:09:3313// Libfuzzer API.
14extern "C" {
15 // User function.
16 int LLVMFuzzerTestOneInput(const unsigned char* data, size_t size);
17 // Initialization function.
18 __attribute__((weak)) int LLVMFuzzerInitialize(int *argc, char ***argv);
19}
aizatskyc562c132016-01-20 00:00:1420
21std::vector<char> readFile(std::string path) {
22 std::ifstream in(path);
23 return std::vector<char>((std::istreambuf_iterator<char>(in)),
24 std::istreambuf_iterator<char>());
25}
26
27int main(int argc, char **argv) {
28 if (argc == 1) {
29 std::cerr << "Usage: " << argv[0] << " <file>..." << std::endl;
30 exit(1);
31 }
32
aizatskye4599ba52016-01-29 23:09:3333 if (LLVMFuzzerInitialize)
34 LLVMFuzzerInitialize(&argc, &argv);
35
aizatskyc562c132016-01-20 00:00:1436 for (int i = 1; i < argc; ++i) {
37 std::cout << argv[i] << std::endl;
38 auto v = readFile(argv[i]);
39 LLVMFuzzerTestOneInput((const unsigned char *)v.data(), v.size());
40 }
41}