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 "fxjs/cfx_v8.h"
#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();
  virtual ~PDFFuzzerPublic();

#ifdef PDF_ENABLE_V8
#ifdef PDF_ENABLE_XFA
  // Virtualize to avoid linker issues in component builds. This results
  // in an indirect function callback to code in a higher layer.
  virtual cppgc::Heap* GetHeap() const;
  virtual void MaybeForceGCAndPump();
#endif  // PDF_ENABLE_XFA
#endif  // PDF_ENABLE_V8

 private:
  FPDF_LIBRARY_CONFIG config_;
  UNSUPPORT_INFO unsupport_info_;
#ifdef PDF_ENABLE_V8
  v8::StartupData snapshot_blob_;
  std::unique_ptr<v8::Platform> platform_;
  std::unique_ptr<v8::ArrayBuffer::Allocator> allocator_;
  std::unique_ptr<v8::Isolate, CFX_V8IsolateDeleter> isolate_;
#ifdef PDF_ENABLE_XFA
  uint32_t iterations_ = 0;
  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_