services/network/content_security_policy_fuzzer.cc

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

#include "services/network/public/cpp/content_security_policy/content_security_policy.h"

#include <string>

#include "base/at_exit.h"
#include "base/command_line.h"
#include "base/i18n/icu_util.h"
#include "net/http/http_response_headers.h"
#include "testing/libfuzzer/libfuzzer_exports.h"
#include "url/gurl.h"

namespace {

struct Environment {
  Environment() {
    // This is a workaround for https://crbug.com/778929.
    CHECK(base::i18n::InitializeICU());

    base::CommandLine::Init(0, nullptr);
  }

  // used by ICU integration.
  base::AtExitManager at_exit_manager;
};

}  // namespace

namespace network {

int LLVMFuzzerInitialize(int* argc, char*** argv) {
  static Environment env;

  return 0;
}

int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size) {
  ParseContentSecurityPolicies(
      std::string(reinterpret_cast<const char*>(data), size),
      network::mojom::ContentSecurityPolicyType::kEnforce,
      network::mojom::ContentSecurityPolicySource::kHTTP,
      GURL("https://example.com/"));

  return 0;
}

}  // namespace network

extern "C" int LLVMFuzzerInitialize(int* argc, char*** argv) {
  return network::LLVMFuzzerInitialize(argc, argv);
}

extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size) {
  return network::LLVMFuzzerTestOneInput(data, size);
}