From e5f5d5886af5d3fad07c2e64945236258b3aca23 Mon Sep 17 00:00:00 2001
From: Your Name <you@example.com>
Date: Thu, 26 Aug 2021 18:06:41 +0100
Subject: Add unstable coverage support

---
 frida_mode/ts/lib/afl.ts | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

(limited to 'frida_mode/ts/lib/afl.ts')

diff --git a/frida_mode/ts/lib/afl.ts b/frida_mode/ts/lib/afl.ts
index d7f8b7bc..c722f558 100644
--- a/frida_mode/ts/lib/afl.ts
+++ b/frida_mode/ts/lib/afl.ts
@@ -163,6 +163,15 @@ class Afl {
     Afl.jsApiSetInstrumentTraceUnique();
   }
 
+  /**
+   * See `AFL_FRIDA_INST_UNSTABLE_COVERAGE_FILE`. This function takes a single
+   * `string` as an argument.
+   */
+  public static setInstrumentUnstableCoverageFile(file: string): void {
+    const buf = Memory.allocUtf8String(file);
+    Afl.jsApiSetInstrumentUnstableCoverageFile(buf);
+  }
+
   /**
    * This is equivalent to setting `AFL_FRIDA_PERSISTENT_ADDR`, again a
    * `NativePointer` should be provided as it's argument.
@@ -338,6 +347,11 @@ class Afl {
     "void",
     []);
 
+  private static readonly jsApiSetInstrumentUnstableCoverageFile = Afl.jsApiGetFunction(
+    "js_api_set_instrument_unstable_coverage_file",
+    "void",
+    ["pointer"]);
+
   private static readonly jsApiSetPersistentAddress = Afl.jsApiGetFunction(
     "js_api_set_persistent_address",
     "void",
-- 
cgit 1.4.1