File: fuzzer_driver.cpp

package info (click to toggle)
icu 76.1-4
  • links: PTS
  • area: main
  • in suites: forky, sid, trixie
  • size: 121,296 kB
  • sloc: cpp: 522,712; ansic: 113,387; sh: 4,983; makefile: 4,709; perl: 3,198; python: 2,847; xml: 2,652; sed: 36; lisp: 12
file content (58 lines) | stat: -rw-r--r-- 1,861 bytes parent folder | download | duplicates (8)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
// © 2019 and later: Unicode, Inc. and others.
// License & terms of use: http://www.unicode.org/copyright.html

#include <fstream>
#include <iostream>
#include <sstream>
#include <stddef.h>
#include <stdint.h>
#include <string>

#include "cmemory.h"

extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size);

int main(int argc, char* argv[])
{
    bool show_warning = true;
    bool show_error = true;
#if UPRV_HAS_FEATURE(address_sanitizer)
    show_warning = false;
#endif
#if UPRV_HAS_FEATURE(memory_sanitizer)
    show_warning = false;
#endif
    if (argc > 2 && strcmp(argv[2], "-q") == 0) {
        show_warning = false;
        show_error = false;
    }
    if (show_warning) {
        std::cerr << "WARNING: This binary work only under build configure with" << std::endl
                  << " CFLAGS=\"-fsanitize=$SANITIZE\""
                  << " CXXFLAGS=\"-fsanitize=$SANITIZE\""
                  << " ./runConfigureICU ... " << std::endl
                  << "  where $SANITIZE is 'address' or 'memory'" << std::endl
                  << "Please run the above step and make tests to rebuild" << std::endl;
        // Do not return -1 here so we will pass the unit test.
    }
    if (argc < 2) {
        if (show_error) {
            std::cerr << "Usage: " << argv[0] << "  testcasefile [-q]" << std::endl
                      << "  -q : quiet while error" << std::endl;
        }
        return -1;
    }
    const char *path = argv[1];
    std::ifstream file(path, std::ios::binary);
    if (!file.is_open()) {
        if (show_error) {
            std::cerr << "Cannot open testcase file " << path << std::endl;
        }
        return -1;
    }
    std::ostringstream ostrm;
    ostrm << file.rdbuf();
    LLVMFuzzerTestOneInput(reinterpret_cast<const uint8_t*>(ostrm.str().c_str()), ostrm.str().size());

    return 0;
}