@aztec/simulator 0.0.0-test.1 → 0.0.1-commit.023c3e5

package/src/private/acvm_wasm.ts

@@ -0,0 +1,76 @@
+import { type Logger, type LoggerBindings, resolveLogger } from '@aztec/foundation/log';
+import { Timer } from '@aztec/foundation/timer';
+import initACVM, { type ExecutionError, type ForeignCallHandler, executeCircuit } from '@aztec/noir-acvm_js';
+import initAbi from '@aztec/noir-noirc_abi';
+import type { FunctionArtifactWithContractName } from '@aztec/stdlib/abi';
+import type { NoirCompiledCircuitWithName } from '@aztec/stdlib/noir';
+
+import { type ACIRCallback, type ACIRExecutionResult, acvm } from './acvm/acvm.js';
+import type { ACVMWitness } from './acvm/acvm_types.js';
+import type { ACVMSuccess } from './acvm_native.js';
+import { type CircuitSimulator, enrichNoirError } from './circuit_simulator.js';
+
+export class WASMSimulator implements CircuitSimulator {
+  protected log: Logger;
+
+  constructor(loggerOrBindings?: Logger | LoggerBindings) {
+    this.log = resolveLogger('wasm-simulator', loggerOrBindings);
+  }
+
+  async init(): Promise<void> {
+    // If these are available, then we are in the
+    // web environment. For the node environment, this
+    // is a no-op.
+    if (typeof initAbi === 'function') {
+      /** @ts-expect-error The node bundle doesn't include these default imports, so TS complains */
+      await Promise.all([initAbi(), initACVM()]);
+    }
+  }
+
+  async executeProtocolCircuit(
+    input: ACVMWitness,
+    artifact: NoirCompiledCircuitWithName,
+    callback: ForeignCallHandler,
+  ): Promise<ACVMSuccess> {
+    this.log.debug('init', { hash: artifact.hash });
+    await this.init();
+
+    // Decode the bytecode from base64 since the acvm does not know about base64 encoding
+    const decodedBytecode = Buffer.from(artifact.bytecode, 'base64');
+    //
+    // Execute the circuit
+    try {
+      const timer = new Timer();
+      const result = await executeCircuit(
+        decodedBytecode,
+        input,
+        callback, // handle calls to debug_log
+      );
+      this.log.debug('execution successful', { hash: artifact.hash });
+      return { witness: result, duration: timer.ms() } as ACVMSuccess;
+    } catch (err) {
+      // Typescript types caught errors as unknown or any, so we need to narrow its type to check if it has raw
+      // assertion payload.
+      if (typeof err === 'object' && err !== null && 'rawAssertionPayload' in err) {
+        const parsed = enrichNoirError(artifact, err as ExecutionError);
+        this.log.debug('execution failed', {
+          hash: artifact.hash,
+          error: parsed,
+          message: parsed.message,
+        });
+        throw parsed;
+      }
+      this.log.debug('execution failed', { hash: artifact.hash, error: err });
+      throw new Error(`Circuit execution failed: ${err}`);
+    }
+  }
+
+  async executeUserCircuit(
+    input: ACVMWitness,
+    artifact: FunctionArtifactWithContractName,
+    callback: ACIRCallback,
+  ): Promise<ACIRExecutionResult> {
+    await this.init();
+    return acvm(artifact.bytecode, input, callback, this.log.createChild('acvm'));
+  }
+}

package/src/private/acvm_wasm_with_blobs.ts

@@ -0,0 +1,54 @@
+import { Timer } from '@aztec/foundation/timer';
+import { type ExecutionError, type ForeignCallHandler, executeCircuit } from '@aztec/noir-acvm_js';
+import type { WitnessMap } from '@aztec/noir-types';
+import type { FunctionArtifactWithContractName } from '@aztec/stdlib/abi';
+import type { NoirCompiledCircuitWithName } from '@aztec/stdlib/noir';
+
+import type { ACIRCallback, ACIRExecutionResult } from './acvm/acvm.js';
+import type { ACVMWitness } from './acvm/acvm_types.js';
+import type { ACVMSuccess } from './acvm_native.js';
+import { type CircuitSimulator, enrichNoirError } from './circuit_simulator.js';
+
+/**
+ * A circuit simulator that uses the WASM simulator with the ability to handle blobs via the foreign call handler.
+ * This class is temporary while brillig cannot handle the blob math, and it is kept separate
+ * because the zkg commitment library used in the blob code is not browser compatible.
+ *
+ * It is only used in the context of server-side code executing simulated protocol circuits.
+ */
+export class WASMSimulatorWithBlobs implements CircuitSimulator {
+  async executeProtocolCircuit(
+    input: WitnessMap,
+    artifact: NoirCompiledCircuitWithName,
+    callback: ForeignCallHandler,
+  ): Promise<ACVMSuccess> {
+    // Decode the bytecode from base64 since the acvm does not know about base64 encoding
+    const decodedBytecode = Buffer.from(artifact.bytecode, 'base64');
+    //
+    // Execute the circuit
+    try {
+      const timer = new Timer();
+      const _witnessMap = await executeCircuit(
+        decodedBytecode,
+        input,
+        callback, // handle calls to debug_log and evaluate_blobs mock
+      );
+      return { witness: _witnessMap, duration: timer.ms() } as ACVMSuccess;
+    } catch (err) {
+      // Typescript types caught errors as unknown or any, so we need to narrow its type to check if it has raw
+      // assertion payload.
+      if (typeof err === 'object' && err !== null && 'rawAssertionPayload' in err) {
+        throw enrichNoirError(artifact, err as ExecutionError);
+      }
+      throw new Error(`Circuit execution failed: ${err}`);
+    }
+  }
+
+  executeUserCircuit(
+    _input: ACVMWitness,
+    _artifact: FunctionArtifactWithContractName,
+    _callback: ACIRCallback,
+  ): Promise<ACIRExecutionResult> {
+    throw new Error('Not implemented');
+  }
+}

package/src/private/circuit_recording/circuit_recorder.ts

@@ -0,0 +1,263 @@
+import { sha512 } from '@aztec/foundation/crypto/sha512';
+import { type Logger, type LoggerBindings, resolveLogger } from '@aztec/foundation/log';
+import { Timer } from '@aztec/foundation/timer';
+import type { ForeignCallHandler, ForeignCallInput, ForeignCallOutput } from '@aztec/noir-acvm_js';
+
+import type { ACIRCallback } from '../acvm/acvm.js';
+import type { ACVMWitness } from '../acvm/acvm_types.js';
+
+export type OracleCall = {
+  name: string;
+  inputs: unknown[];
+  outputs: unknown;
+  time: number;
+  // Due to the recursive nature of the simulator, we might have
+  // oracle calls performed after a foreign call (which is itself an oracle call)
+  // We keep track of the stack depth in this variable to ensure the recorded oracle
+  // calls are correctly associated with the right circuit.
+  // This is only use as a debugging tool
+  stackDepth: number;
+};
+
+export class CircuitRecording {
+  circuitName: string;
+  functionName: string;
+  bytecodeSHA512Hash: string;
+  timestamp: number;
+  inputs: Record<string, string>;
+  oracleCalls: OracleCall[];
+  error?: string;
+  parent?: CircuitRecording;
+
+  constructor(circuitName: string, functionName: string, bytecodeSHA512Hash: string, inputs: Record<string, string>) {
+    this.circuitName = circuitName;
+    this.functionName = functionName;
+    this.bytecodeSHA512Hash = bytecodeSHA512Hash;
+    this.timestamp = Date.now();
+    this.inputs = inputs;
+    this.oracleCalls = [];
+  }
+
+  setParent(recording?: CircuitRecording): void {
+    this.parent = recording;
+  }
+}
+
+/**
+ * Class responsible for recording circuit inputs necessary to replay the circuit. These inputs are the initial witness
+ * map and the oracle calls made during the circuit execution/witness generation.
+ *
+ * Example recording object:
+ * ```json
+ * {
+ *   "circuitName": "AMM",
+ *   "functionName": "add_liquidity",
+ *   "bytecodeSHA512Hash": "b46c640ed38f20eac5f61a5e41d8dd1e",
+ *   "timestamp": 1740691464360,
+ *   "inputs": {
+ *     "0": "0x1e89de1f0ad5204263733b7ddf65bec45b8f44714a4da85a46474dad677679ef",
+ *     "1": "0x00f4d59c0ff773427bb0fed5b422557ca4dc5655abe53d31fa9408cb3c5a672f",
+ *     "5": "0x000000000000000000000000000000000000000000000000000000000000000f"
+ *   },
+ *   "oracleCalls": [
+ *     {
+ *       "name": "loadCapsule",
+ *       "inputs": [
+ *         [
+ *           "0x102422483bad6abd385948435667e144ac4c272576e325e7563608876cd446fd"
+ *         ],
+ *         [
+ *           "0x000000000000000000000000000000000000000000000000000000000000004d"
+ *         ],
+ *         [
+ *           "0x0000000000000000000000000000000000000000000000000000000000000001"
+ *         ]
+ *       ],
+ *       "outputs": [
+ *         "0x0000000000000000000000000000000000000000000000000000000000000000",
+ *         [
+ *           "0x0000000000000000000000000000000000000000000000000000000000000000"
+ *         ]
+ *       ]
+ *     },
+ *     {
+ *       "name": "fetchTaggedLogs",
+ *       "inputs": []
+ *     }
+ *   ]
+ * }
+ * ```
+ */
+export class CircuitRecorder {
+  protected readonly logger: Logger;
+
+  protected recording?: CircuitRecording;
+
+  private stackDepth: number = 0;
+  private newCircuit: boolean = true;
+
+  protected constructor(loggerOrBindings?: Logger | LoggerBindings) {
+    this.logger = resolveLogger('simulator:acvm:recording', loggerOrBindings);
+  }
+
+  /**
+   * Initializes a new circuit recording session.
+   * @param recordDir - Directory to store the recording
+   * @param input - Circuit input witness
+   * @param circuitBytecode - Compiled circuit bytecode
+   * @param circuitName - Name of the circuit
+   * @param functionName - Name of the circuit function (defaults to 'main'). This is meaningful only for
+   * contracts as protocol circuits artifacts always contain a single entrypoint function called 'main'.
+   */
+  start(input: ACVMWitness, circuitBytecode: Buffer, circuitName: string, functionName: string): Promise<void> {
+    if (this.newCircuit) {
+      const parentRef = this.recording;
+      this.recording = new CircuitRecording(
+        circuitName,
+        functionName,
+        sha512(circuitBytecode).toString('hex'),
+        Object.fromEntries(input),
+      );
+      this.recording.setParent(parentRef);
+    }
+
+    return Promise.resolve();
+  }
+
+  /**
+   * Wraps a callback to record all oracle/foreign calls.
+   * @param callback - The original callback to wrap, either a user circuit callback or protocol circuit callback.
+   * @returns A wrapped callback that records all oracle interactions.
+   */
+  wrapCallback(callback: ACIRCallback | ForeignCallHandler | undefined): ACIRCallback | ForeignCallHandler | undefined {
+    if (!callback) {
+      return undefined;
+    }
+    if (this.#isACIRCallback(callback)) {
+      return this.#wrapUserCircuitCallback(callback);
+    }
+    return this.#wrapProtocolCircuitCallback(callback);
+  }
+
+  /**
+   * Type guard to check if a callback is an ACIRCallback.
+   */
+  #isACIRCallback(callback: ACIRCallback | ForeignCallHandler): callback is ACIRCallback {
+    return typeof callback === 'object' && callback !== null && !('call' in callback);
+  }
+
+  /**
+   * Wraps a user circuit callback to record all oracle calls.
+   * @param callback - The original circuit callback.
+   * @returns A wrapped callback that records all oracle interactions which is to be provided to the ACVM.
+   */
+  #wrapUserCircuitCallback(callback: ACIRCallback): ACIRCallback {
+    const recordingCallback: ACIRCallback = {} as ACIRCallback;
+    const oracleMethods = Object.keys(callback);
+
+    for (const name of oracleMethods) {
+      const fn = callback[name as keyof ACIRCallback];
+      if (!fn || typeof fn !== 'function') {
+        throw new Error(`Oracle method ${name} not found when setting up recording callback`);
+      }
+
+      const isExternalCall = (name as keyof ACIRCallback) === 'privateCallPrivateFunction';
+
+      recordingCallback[name as keyof ACIRCallback] = (...args: ForeignCallInput[]): ReturnType<typeof fn> => {
+        const timer = new Timer();
+        // If we're entering another circuit via `privateCallPrivateFunction`, we increase the stack depth and set the
+        // newCircuit variable to ensure we are creating a new recording object.
+        if (isExternalCall) {
+          this.stackDepth++;
+          this.newCircuit = true;
+        }
+        const result = fn.call(callback, ...args);
+        if (result instanceof Promise) {
+          return result.then(async r => {
+            // Once we leave the nested circuit, we decrease the stack depth and set newCircuit to false
+            // so that the parent circuit continues with its existing recording
+            // Note: recording restoration is handled by finish()
+            if (isExternalCall) {
+              this.stackDepth--;
+              this.newCircuit = false;
+            }
+            await this.recordCall(name, args, r, timer.ms(), this.stackDepth);
+            return r;
+          }) as ReturnType<typeof fn>;
+        }
+        // Once we leave the nested circuit, we decrease the stack depth and set newCircuit to false
+        // so that the parent circuit continues with its existing recording
+        // Note: recording restoration is handled by finish()
+        if (isExternalCall) {
+          this.stackDepth--;
+          this.newCircuit = false;
+        }
+        void this.recordCall(name, args, result, timer.ms(), this.stackDepth);
+        return result;
+      };
+    }
+
+    return recordingCallback;
+  }
+
+  /**
+   * Wraps a protocol circuit callback to record all oracle calls.
+   * @param callback - The original oracle circuit callback.
+   * @returns A wrapped handler that records all oracle interactions which is to be provided to the ACVM.
+   */
+  #wrapProtocolCircuitCallback(callback: ForeignCallHandler): ForeignCallHandler {
+    return async (name: string, inputs: ForeignCallInput[]): Promise<ForeignCallOutput[]> => {
+      const timer = new Timer();
+      const result = await callback(name, inputs);
+      await this.recordCall(name, inputs, result, timer.ms(), 0);
+      return result;
+    };
+  }
+
+  /**
+   * Records a single oracle/foreign call with its inputs and outputs.
+   * @param name - Name of the call
+   * @param inputs - Input arguments
+   * @param outputs - Output results
+   */
+  recordCall(name: string, inputs: unknown[], outputs: unknown, time: number, stackDepth: number): Promise<OracleCall> {
+    const entry = {
+      name,
+      inputs,
+      outputs,
+      time,
+      stackDepth,
+    };
+    this.recording!.oracleCalls.push(entry);
+    return Promise.resolve(entry);
+  }
+
+  /**
+   * Finalizes the recording by resetting the state and returning the recording object.
+   */
+  finish(): Promise<CircuitRecording> {
+    const result = this.recording;
+    // If this is the top-level circuit recording, we reset the state for the next simulator call
+    if (!result!.parent) {
+      this.newCircuit = true;
+      this.recording = undefined;
+    } else {
+      // For nested circuits (utility calls, nested contract calls), restore to parent recording
+      // Note: we don't set newCircuit=false here because:
+      // - For privateCallPrivateFunction, the callback wrapper will set it to false
+      // - For utility calls, we want newCircuit to remain true so the next circuit creates its own recording
+      this.recording = result!.parent;
+    }
+    return Promise.resolve(result!);
+  }
+
+  /**
+   * Finalizes the recording by resetting the state and returning the recording object with an attached error.
+   * @param error - The error that occurred during circuit execution
+   */
+  async finishWithError(error: unknown): Promise<CircuitRecording> {
+    const result = await this.finish();
+    result.error = JSON.stringify(error);
+    return result;
+  }
+}

package/src/private/circuit_recording/file_circuit_recorder.ts

@@ -0,0 +1,163 @@
+import type { Logger } from '@aztec/foundation/log';
+
+import fs from 'fs/promises';
+import path from 'path';
+
+import type { ACVMWitness } from '../acvm/acvm_types.js';
+import { CircuitRecorder, type CircuitRecording } from './circuit_recorder.js';
+
+export class FileCircuitRecorder extends CircuitRecorder {
+  declare recording?: CircuitRecording & { filePath: string; isFirstCall: boolean };
+
+  constructor(
+    private readonly recordDir: string,
+    logger?: Logger,
+  ) {
+    super(logger);
+  }
+
+  override async start(
+    input: ACVMWitness,
+    circuitBytecode: Buffer,
+    circuitName: string,
+    functionName: string = 'main',
+  ) {
+    await super.start(input, circuitBytecode, circuitName, functionName);
+
+    const recordingStringWithoutClosingBracket = JSON.stringify(
+      { ...this.recording, isFirstCall: undefined, parent: undefined, oracleCalls: undefined, filePath: undefined },
+      null,
+      2,
+    ).slice(0, -2);
+
+    try {
+      // Check if the recording directory exists and is a directory
+      const stats = await fs.stat(this.recordDir);
+      if (!stats.isDirectory()) {
+        throw new Error(`Recording path ${this.recordDir} exists but is not a directory`);
+      }
+    } catch (err) {
+      if ((err as NodeJS.ErrnoException).code === 'ENOENT') {
+        // The directory does not exist so we create it
+        await fs.mkdir(this.recordDir, { recursive: true });
+      } else {
+        throw err;
+      }
+    }
+
+    this.recording!.isFirstCall = true;
+    this.recording!.filePath = await FileCircuitRecorder.#computeFilePathAndStoreInitialRecording(
+      this.recordDir,
+      this.recording!.circuitName,
+      this.recording!.functionName,
+      recordingStringWithoutClosingBracket,
+    );
+  }
+
+  /**
+   * Computes a unique file path for the recording by trying different counter values.
+   * This is needed because multiple recordings of the same circuit could be happening simultaneously or an older
+   * recording might be present.
+   * @param recordDir - Directory to store the recording
+   * @param circuitName - Name of the circuit
+   * @param functionName - Name of the circuit function
+   * @param recordingContent - Initial recording content
+   * @returns A unique file path for the recording
+   */
+  static async #computeFilePathAndStoreInitialRecording(
+    recordDir: string,
+    circuitName: string,
+    functionName: string,
+    recordingContent: string,
+  ): Promise<string> {
+    let counter = 0;
+    while (true) {
+      try {
+        const filePath = getFilePath(recordDir, circuitName, functionName, counter);
+        // Write the initial recording content to the file
+        await fs.writeFile(filePath, recordingContent + ',\n  "oracleCalls": [\n', {
+          flag: 'wx', // wx flag fails if file exists
+        });
+        return filePath;
+      } catch (err) {
+        if ((err as NodeJS.ErrnoException).code === 'EEXIST') {
+          counter++;
+          continue;
+        }
+        throw err;
+      }
+    }
+  }
+
+  /**
+   * Records a single oracle/foreign call with its inputs and outputs.
+   * @param name - Name of the call
+   * @param inputs - Input arguments
+   * @param outputs - Output results
+   */
+  override async recordCall(name: string, inputs: unknown[], outputs: unknown, time: number, stackDepth: number) {
+    const entry = await super.recordCall(name, inputs, outputs, time, stackDepth);
+    try {
+      const prefix = this.recording!.isFirstCall ? '    ' : '    ,';
+      this.recording!.isFirstCall = false;
+      await fs.appendFile(this.recording!.filePath, prefix + JSON.stringify(entry) + '\n');
+    } catch (err) {
+      this.logger.error('Failed to log circuit call', { error: err });
+    }
+    return entry;
+  }
+
+  /**
+   * Finalizes the recording file by adding closing brackets. Without calling this method, the recording file is
+   * incomplete and it fails to parse.
+   */
+  override async finish(): Promise<CircuitRecording> {
+    // Finish sets the recording to undefined if we are at the topmost circuit,
+    // so we save the current file path before that
+    const filePath = this.recording!.filePath;
+    const result = await super.finish();
+    try {
+      await fs.appendFile(filePath, '  ]\n}\n');
+    } catch (err) {
+      this.logger.error('Failed to finalize recording file', { error: err });
+    }
+    return result!;
+  }
+
+  /**
+   * Finalizes the recording file by adding the error and closing brackets. Without calling this method or `finish`,
+   * the recording file is incomplete and it fails to parse.
+   * @param error - The error that occurred during circuit execution
+   */
+  override async finishWithError(error: unknown): Promise<CircuitRecording> {
+    // Finish sets the recording to undefined if we are at the topmost circuit,
+    // so we save the current file path before that
+    const filePath = this.recording!.filePath;
+    const result = await super.finishWithError(error);
+    try {
+      await fs.appendFile(filePath, '  ],\n');
+      await fs.appendFile(filePath, `  "error": ${JSON.stringify(error)}\n`);
+      await fs.appendFile(filePath, '}\n');
+    } catch (err) {
+      this.logger.error('Failed to finalize recording file with error', { error: err });
+    }
+    return result!;
+  }
+}
+
+/**
+ * Generates a file path for storing circuit recordings. The format of the filename is:
+ * `circuit_name_circuit_function_name_YYYY-MM-DD_N.json` where N is a counter to ensure unique filenames.
+ * @param recordDir - Base directory for recordings
+ * @param circuitName - Name of the circuit
+ * @param functionName - Name of the circuit function
+ * @param counter - Counter to ensure unique filenames. This is expected to be incremented in a loop until there is no
+ * existing file with the same name.
+ * @returns A file path for the recording.
+ */
+function getFilePath(recordDir: string, circuitName: string, functionName: string, counter: number): string {
+  const date = new Date();
+  const formattedDate = date.toISOString().split('T')[0];
+  const filename = `${circuitName}_${functionName}_${formattedDate}_${counter}.json`;
+  return path.join(recordDir, filename);
+}

package/src/private/circuit_recording/memory_circuit_recorder.ts

@@ -0,0 +1,13 @@
+import type { Logger } from '@aztec/foundation/log';
+
+import { CircuitRecorder } from './circuit_recorder.js';
+
+/**
+ * In memory circuit recorder uses the default implementation. This is kept
+ * while we decide the fate of the FileCircuitRecorder.
+ */
+export class MemoryCircuitRecorder extends CircuitRecorder {
+  constructor(logger?: Logger) {
+    super(logger);
+  }
+}

package/src/private/circuit_recording/simulator_recorder_wrapper.ts

@@ -0,0 +1,91 @@
+import type { ForeignCallHandler } from '@aztec/noir-acvm_js';
+import type { FunctionArtifactWithContractName } from '@aztec/stdlib/abi';
+import type { NoirCompiledCircuitWithName } from '@aztec/stdlib/noir';
+
+import type { ACIRCallback, ACIRCallbackStats, ACIRExecutionResult } from '../acvm/acvm.js';
+import type { ACVMWitness } from '../acvm/acvm_types.js';
+import type { ACVMSuccess } from '../acvm_native.js';
+import type { CircuitSimulator } from '../circuit_simulator.js';
+import type { CircuitRecorder } from './circuit_recorder.js';
+
+/**
+ * Takes a circuit simulator and wraps it in a circuit recorder. See CircuitRecorder for more details on how circuit
+ * recording works.
+ */
+export class SimulatorRecorderWrapper implements CircuitSimulator {
+  constructor(
+    private simulator: CircuitSimulator,
+    private recorder: CircuitRecorder,
+  ) {}
+
+  executeProtocolCircuit(
+    input: ACVMWitness,
+    artifact: NoirCompiledCircuitWithName,
+    callback: ForeignCallHandler | undefined,
+  ): Promise<ACVMSuccess> {
+    const bytecode = Buffer.from(artifact.bytecode, 'base64');
+
+    return this.#simulate<ForeignCallHandler | undefined, ACVMSuccess>(
+      wrappedCallback => this.simulator.executeProtocolCircuit(input, artifact, wrappedCallback),
+      input,
+      bytecode,
+      artifact.name,
+      'main',
+      callback,
+    );
+  }
+
+  executeUserCircuit(
+    input: ACVMWitness,
+    artifact: FunctionArtifactWithContractName,
+    callback: ACIRCallback,
+  ): Promise<ACIRExecutionResult> {
+    return this.#simulate<ACIRCallback, ACIRExecutionResult>(
+      wrappedCallback => this.simulator.executeUserCircuit(input, artifact, wrappedCallback),
+      input,
+      artifact.bytecode,
+      artifact.contractName,
+      artifact.name,
+      callback,
+    );
+  }
+
+  async #simulate<C extends ACIRCallback | ForeignCallHandler | undefined, T extends ACIRExecutionResult | ACVMSuccess>(
+    simulateFn: (wrappedCallback: C) => Promise<T>,
+    input: ACVMWitness,
+    bytecode: Buffer,
+    contractName: string,
+    functionName: string,
+    callback: C,
+  ): Promise<T> {
+    // Start recording circuit execution
+    await this.recorder.start(input, bytecode, contractName, functionName);
+
+    // If callback was provided, we wrap it in a circuit recorder callback wrapper
+    const wrappedCallback = this.recorder.wrapCallback(callback);
+    let result: T;
+    try {
+      result = await simulateFn(wrappedCallback as C);
+    } catch (error) {
+      // If an error occurs, we finalize the recording file with the error
+      await this.recorder.finishWithError(error);
+      throw error;
+    }
+
+    // Witness generation is complete so we finish the circuit recorder
+    const recording = await this.recorder.finish();
+
+    (result as ACIRExecutionResult).oracles = recording.oracleCalls?.reduce(
+      (acc, { time, name }) => {
+        if (!acc[name]) {
+          acc[name] = { times: [] };
+        }
+        acc[name].times.push(time);
+        return acc;
+      },
+      {} as Record<string, ACIRCallbackStats>,
+    );
+
+    return result;
+  }
+}