testing/fuzzers/pdf_fuzzer_init_public.h

// Copyright 2020 The PDFium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef TESTING_FUZZERS_PDF_FUZZER_INIT_PUBLIC_H_
#define TESTING_FUZZERS_PDF_FUZZER_INIT_PUBLIC_H_

#include <memory>

#include "public/fpdf_ext.h"
#include "public/fpdfview.h"

#ifdef PDF_ENABLE_V8
#include "v8/include/v8-platform.h"
#include "v8/include/v8.h"
#ifdef PDF_ENABLE_XFA
#include "fxjs/gc/heap.h"
#endif  // PDF_ENABLE_XFA
#endif  // PDF_ENABLE_V8

// Context for all runs of the fuzzer.
class PDFFuzzerPublic {
 public:
  PDFFuzzerPublic();
  ~PDFFuzzerPublic();

#ifdef PDF_ENABLE_V8
#ifdef PDF_ENABLE_XFA
  cppgc::Heap* GetHeap() { return heap_.get(); }
#endif  // PDF_ENABLE_XFA
#endif  // PDF_ENABLE_V8

 private:
  FPDF_LIBRARY_CONFIG config_;
  UNSUPPORT_INFO unsupport_info_;
#ifdef PDF_ENABLE_V8
  std::unique_ptr<v8::Platform> platform_;
  v8::StartupData snapshot_blob_;
#ifdef PDF_ENABLE_XFA
  FXGCScopedHeap heap_;
#endif  // PDF_ENABLE_XFA
#endif  // PDF_ENABLE_V8
};

// Initializes the library once for all runs of the fuzzer.
class PDFFuzzerInitPublic {
 public:
  PDFFuzzerInitPublic();
  ~PDFFuzzerInitPublic();

 private:
  std::unique_ptr<PDFFuzzerPublic> context_;
};

#endif  // TESTING_FUZZERS_PDF_FUZZER_INIT_PUBLIC_H_