regexp.cc 3.36 KB
Newer Older
1 2 3 4
// Copyright 2016 the V8 project authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

jochen's avatar
jochen committed
5
#include <limits.h>
6 7 8 9
#include <stddef.h>
#include <stdint.h>

#include "include/v8.h"
10
#include "src/heap/factory.h"
11 12 13 14 15 16
#include "src/objects-inl.h"
#include "src/regexp/jsregexp.h"
#include "test/fuzzer/fuzzer-support.h"

namespace i = v8::internal;

17
void Test(v8::Isolate* isolate, i::Handle<i::JSRegExp> regexp,
18 19
          i::Handle<i::String> subject,
          i::Handle<i::RegExpMatchInfo> results_array) {
20
  v8::TryCatch try_catch(isolate);
21 22 23
  i::Isolate* i_isolate = reinterpret_cast<i::Isolate*>(isolate);
  if (i::RegExpImpl::Exec(i_isolate, regexp, subject, 0, results_array)
          .is_null()) {
24 25
    i_isolate->OptionalRescheduleException(true);
  }
26 27
}

28 29 30 31 32 33 34 35 36 37 38 39
extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size) {
  v8_fuzzer::FuzzerSupport* support = v8_fuzzer::FuzzerSupport::Get();
  v8::Isolate* isolate = support->GetIsolate();

  v8::Isolate::Scope isolate_scope(isolate);
  v8::HandleScope handle_scope(isolate);
  v8::Context::Scope context_scope(support->GetContext());
  v8::TryCatch try_catch(isolate);

  i::Isolate* i_isolate = reinterpret_cast<i::Isolate*>(isolate);
  i::Factory* factory = i_isolate->factory();

40
  CHECK(!i_isolate->has_pending_exception());
41 42 43 44
  if (size > INT_MAX) return 0;
  i::MaybeHandle<i::String> maybe_source = factory->NewStringFromOneByte(
      i::Vector<const uint8_t>(data, static_cast<int>(size)));
  i::Handle<i::String> source;
45 46 47 48
  if (!maybe_source.ToHandle(&source)) {
    i_isolate->clear_pending_exception();
    return 0;
  }
49 50 51

  static const int kAllFlags = i::JSRegExp::kGlobal | i::JSRegExp::kIgnoreCase |
                               i::JSRegExp::kMultiline | i::JSRegExp::kSticky |
52
                               i::JSRegExp::kUnicode | i::JSRegExp::kDotAll;
53 54 55 56

  const uint8_t one_byte_array[6] = {'f', 'o', 'o', 'b', 'a', 'r'};
  const i::uc16 two_byte_array[6] = {'f', 0xD83D, 0xDCA9, 'b', 'a', 0x2603};

57
  CHECK(!i_isolate->has_pending_exception());
58
  i::Handle<i::RegExpMatchInfo> results_array = factory->NewRegExpMatchInfo();
59 60 61 62 63 64 65
  i::Handle<i::String> one_byte =
      factory->NewStringFromOneByte(i::Vector<const uint8_t>(one_byte_array, 6))
          .ToHandleChecked();
  i::Handle<i::String> two_byte =
      factory->NewStringFromTwoByte(i::Vector<const i::uc16>(two_byte_array, 6))
          .ToHandleChecked();

66 67
  i::Handle<i::JSRegExp> regexp;
  {
68
    CHECK(!i_isolate->has_pending_exception());
69 70 71 72 73
    v8::TryCatch try_catch(isolate);
    // Create a string so that we can calculate a hash from the input data.
    std::string str = std::string(reinterpret_cast<const char*>(data), size);
    i::JSRegExp::Flags flag = static_cast<i::JSRegExp::Flags>(
        std::hash<std::string>()(str) % (kAllFlags + 1));
74 75
    i::MaybeHandle<i::JSRegExp> maybe_regexp =
        i::JSRegExp::New(i_isolate, source, flag);
76 77 78
    if (!maybe_regexp.ToHandle(&regexp)) {
      i_isolate->clear_pending_exception();
      return 0;
79
    }
80
  }
81 82 83 84
  Test(isolate, regexp, one_byte, results_array);
  Test(isolate, regexp, two_byte, results_array);
  Test(isolate, regexp, factory->empty_string(), results_array);
  Test(isolate, regexp, source, results_array);
85 86
  isolate->RequestGarbageCollectionForTesting(
      v8::Isolate::kFullGarbageCollection);
87
  CHECK(!i_isolate->has_pending_exception());
88 89
  return 0;
}