summaryrefslogtreecommitdiffstats
path: root/tools/fuzzing/libfuzzer/FuzzerCustomMain.cpp
blob: 2293efd95d8ee1decf07b25004c7e0bc02925097 (plain)
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
/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*-
 * * This Source Code Form is subject to the terms of the Mozilla Public
 * * License, v. 2.0. If a copy of the MPL was not distributed with this
 * * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

#include <cstdlib>

#include "FuzzerInterface.h"
#include "FuzzerInternal.h"
#include "harness/LibFuzzerRegistry.h"

/* This is a wrapper defined in browser/app/nsBrowserApp.cpp,
 * encapsulating the XRE_ equivalent defined in libxul */
extern void libFuzzerGetFuncs(const char*, LibFuzzerInitFunc*,
                                 LibFuzzerTestingFunc*);

int libfuzzer_main(int argc, char **argv) {
  LibFuzzerInitFunc initFunc = nullptr;
  LibFuzzerTestingFunc testingFunc = nullptr;

  libFuzzerGetFuncs(getenv("LIBFUZZER"), &initFunc, &testingFunc);

  if (initFunc) {
    int ret = initFunc(&argc, &argv);
    if (ret) {
      fprintf(stderr, "LibFuzzer: Error: Initialize callback failed\n");
      return ret;
    }
  }

  if (!testingFunc) {
      fprintf(stderr, "LibFuzzer: Error: No testing callback found\n");
      return 1;
  }

  return fuzzer::FuzzerDriver(&argc, &argv, testingFunc);
}