diff --git a/index.d.ts b/index.d.ts index 3294dd1..cb4661c 100644 --- a/index.d.ts +++ b/index.d.ts @@ -27,6 +27,7 @@ declare module 'datadog-iast-taint-tracking' { export interface TaintedUtils { createTransaction(transactionId: string): string; newTaintedString(transactionId: string, original: string, paramName: string, type: string): string; + newTaintedObject(transactionId: string, original: any, paramName: string, type: string): any; addSecureMarksToTaintedString(transactionId: string, taintedString: string, secureMarks: number): string; isTainted(transactionId: string, ...args: string[]): boolean; getMetrics(transactionId: string, telemetryVerbosity: number): Metrics; diff --git a/src/api/taint_methods.cc b/src/api/taint_methods.cc index 56385d5..709c5f2 100644 --- a/src/api/taint_methods.cc +++ b/src/api/taint_methods.cc @@ -289,7 +289,7 @@ void NewTaintedObject(const FunctionCallbackInfo& args) { return; } - if ((args[1]->IsString())) { + if (args[1]->IsString()) { args.GetReturnValue().Set(args[1]); return; }