1 files changed, 22 insertions, 5 deletions
diff --git a/packages/sol-tracing-utils/src/trace_info_subprovider.ts b/packages/sol-tracing-utils/src/trace_info_subprovider.ts
index 635a68f58..43853e152 100644
--- a/packages/sol-tracing-utils/src/trace_info_subprovider.ts
+++ b/packages/sol-tracing-utils/src/trace_info_subprovider.ts
@@ -12,11 +12,28 @@ export abstract class TraceInfoSubprovider extends TraceCollectionSubprovider {
     protected abstract _handleTraceInfoAsync(traceInfo: TraceInfo): Promise<void>;
     protected async _recordTxTraceAsync(address: string, data: string | undefined, txHash: string): Promise<void> {
         await this._web3Wrapper.awaitTransactionMinedAsync(txHash, 0);
-        const trace = await this._web3Wrapper.getTransactionTraceAsync(txHash, {
-            disableMemory: true,
-            disableStack: false,
-            disableStorage: true,
-        });
+        // For very large traces we use a custom tracer that outputs a format compatible with a
+        // regular trace. We only need the 2nd item on the stack when the instruction is a call.
+        // By not including othe stack values, we severly limit the amount of data to be collectd.
+        const tracer = `
+            {
+                data: [],
+                step: function(log) {
+                    const op = log.op.toString();
+                    const opn = 0 | log.op.toNumber();
+                    const pc = 0 | log.getPC();
+                    const depth = 0 | log.getDepth();
+                    const gasCost = 0 | log.getCost();
+                    const gas = 0 | log.getGas();
+                    const isCall = opn == 0xf1 || opn == 0xf2 || opn == 0xf4 || opn == 0xf5;
+                    const stack = isCall ? ['0x'+log.stack.peek(1).toString(16), null] : null;
+                    this.data.push({ pc, gasCost, depth, op, stack, gas });
+                },
+                fault: function() { },
+                result: function() { return {structLogs: this.data}; }
+            }
+        `;
+        const trace = await this._web3Wrapper.getTransactionTraceAsync(txHash, { tracer, timeout: '600s' });
         const tracesByContractAddress = getTracesByContractAddress(trace.structLogs, address);
         const subcallAddresses = _.keys(tracesByContractAddress);
         if (address === constants.NEW_CONTRACT) {