test/fuzzilli/main.cc

// Copyright 2020 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.

#include <stdio.h>
#include <stdlib.h>
#include <string.h>

// Libreprl is a .c file so the header needs to be in an 'extern "C"' block.
extern "C" {
#include "libreprl.h"
}  // extern "C"

struct reprl_context* ctx;

bool execute(const char* code) {
  uint64_t exec_time;
  const uint64_t SECONDS = 1000000;  // Timeout is in microseconds.
  int status =
      reprl_execute(ctx, code, strlen(code), 1 * SECONDS, &exec_time, 0);
  return RIFEXITED(status) && REXITSTATUS(status) == 0;
}

void expect_success(const char* code) {
  if (!execute(code)) {
    printf("Execution of \"%s\" failed\n", code);
    exit(1);
  }
}

void expect_failure(const char* code) {
  if (execute(code)) {
    printf("Execution of \"%s\" unexpectedly succeeded\n", code);
    exit(1);
  }
}

int main(int argc, char** argv) {
  ctx = reprl_create_context();

  const char* env[] = {nullptr};
  const char* d8_path = argc > 1 ? argv[1] : "./out.gn/x64.debug/d8";
  const char* args[] = {d8_path, nullptr};
  if (reprl_initialize_context(ctx, args, env, 1, 1) != 0) {
    printf("REPRL initialization failed\n");
    return -1;
  }

  // Basic functionality test
  if (!execute("let greeting = \"Hello World!\";")) {
    printf(
        "Script execution failed, is %s the path to d8 built with "
        "v8_fuzzilli=true?\n",
        d8_path);
    return -1;
  }

  // Verify that runtime exceptions can be detected
  expect_failure("throw 'failure';");

  // Verify that existing state is property reset between executions
  expect_success("globalProp = 42; Object.prototype.foo = \"bar\";");
  expect_success("if (typeof(globalProp) !== 'undefined') throw 'failure'");
  expect_success("if (typeof(({}).foo) !== 'undefined') throw 'failure'");

  // Verify that rejected promises are properly reset between executions
  expect_failure("async function fail() { throw 42; }; fail()");
  expect_success("42");
  expect_failure("async function fail() { throw 42; }; fail()");

  puts("OK");
  return 0;
}