File: HTMLPreloadScannerFuzzer.cpp

package info (click to toggle)
chromium-browser 57.0.2987.98-1~deb8u1
  • links: PTS, VCS
  • area: main
  • in suites: jessie
  • size: 2,637,852 kB
  • ctags: 2,544,394
  • sloc: cpp: 12,815,961; ansic: 3,676,222; python: 1,147,112; asm: 526,608; java: 523,212; xml: 286,794; perl: 92,654; sh: 86,408; objc: 73,271; makefile: 27,698; cs: 18,487; yacc: 13,031; tcl: 12,957; pascal: 4,875; ml: 4,716; lex: 3,904; sql: 3,862; ruby: 1,982; lisp: 1,508; php: 1,368; exp: 404; awk: 325; csh: 117; jsp: 39; sed: 37
file content (85 lines) | stat: -rw-r--r-- 3,228 bytes parent folder | download
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
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
// Copyright 2016 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 "core/MediaTypeNames.h"
#include "core/css/MediaValuesCached.h"
#include "core/html/HTMLDocument.h"
#include "core/html/parser/HTMLDocumentParser.h"
#include "core/html/parser/ResourcePreloader.h"
#include "core/html/parser/TextResourceDecoderForFuzzing.h"
#include "platform/testing/BlinkFuzzerTestSupport.h"
#include "platform/testing/FuzzedDataProvider.h"

namespace blink {

std::unique_ptr<CachedDocumentParameters> cachedDocumentParametersForFuzzing(
    FuzzedDataProvider& fuzzedData) {
  std::unique_ptr<CachedDocumentParameters> documentParameters =
      CachedDocumentParameters::create();
  documentParameters->doHtmlPreloadScanning = fuzzedData.ConsumeBool();
  documentParameters->doDocumentWritePreloadScanning = fuzzedData.ConsumeBool();
  // TODO(csharrison): How should this be fuzzed?
  documentParameters->defaultViewportMinWidth = Length();
  documentParameters->viewportMetaZeroValuesQuirk = fuzzedData.ConsumeBool();
  documentParameters->viewportMetaEnabled = fuzzedData.ConsumeBool();
  return documentParameters;
}

class MockResourcePreloader : public ResourcePreloader {
  void preload(std::unique_ptr<PreloadRequest>,
               const NetworkHintsInterface&) override {}
};

int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size) {
  FuzzedDataProvider fuzzedData(data, size);

  HTMLParserOptions options;
  options.scriptEnabled = fuzzedData.ConsumeBool();
  options.pluginsEnabled = fuzzedData.ConsumeBool();

  std::unique_ptr<CachedDocumentParameters> documentParameters =
      cachedDocumentParametersForFuzzing(fuzzedData);

  KURL documentURL(ParsedURLString, "http://whatever.test/");

  // Copied from HTMLPreloadScannerTest. May be worthwhile to fuzz.
  MediaValuesCached::MediaValuesCachedData mediaData;
  mediaData.viewportWidth = 500;
  mediaData.viewportHeight = 600;
  mediaData.deviceWidth = 700;
  mediaData.deviceHeight = 800;
  mediaData.devicePixelRatio = 2.0;
  mediaData.colorBitsPerComponent = 24;
  mediaData.monochromeBitsPerComponent = 0;
  mediaData.primaryPointerType = PointerTypeFine;
  mediaData.defaultFontSize = 16;
  mediaData.threeDEnabled = true;
  mediaData.mediaType = MediaTypeNames::screen;
  mediaData.strictMode = true;
  mediaData.displayMode = WebDisplayModeBrowser;

  MockResourcePreloader preloader;

  std::unique_ptr<HTMLPreloadScanner> scanner = HTMLPreloadScanner::create(
      options, documentURL, std::move(documentParameters), mediaData);

  TextResourceDecoderForFuzzing decoder(fuzzedData);
  CString bytes = fuzzedData.ConsumeRemainingBytes();
  String decodedBytes = decoder.decode(bytes.data(), bytes.length());
  scanner->appendToEnd(decodedBytes);
  PreloadRequestStream requests = scanner->scan(documentURL, nullptr);
  preloader.takeAndPreload(requests);
  return 0;
}

}  // namespace blink

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

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