o iC @srUddlmZddlmZddlmZddlmZddlmZddlmZddl m Z ddl m Z dd l m Z dd lmZdd lmZdd lmZdd lmZddlmZeeZiZeeefed<ddZ ddZ!e!eed)dedee"e#dfddfddZ$e!eddZ%e!eddZ&e!ed*dd Z'e!ed!d"Z(e!ed#d$Z)e!ed%d&Z*d'd(Z+dS)+)Text)Union)IAST)TELEMETRY_INFORMATION_VERBOSITY)TELEMETRY_MANDATORY_VERBOSITY) deduplication)_num_objects_tainted_in_request) OriginType origin_to_str)_is_iast_debug_enabled) telemetry) get_logger)config)TELEMETRY_NAMESPACE_IAST_SPAN_METRICScOs2tj}d}tjD] \}}||kr|Sq |S)N) asm_config_iast_telemetry_report_lvlupperrMETRICS_REPORT_LVLS)argskwargsreport_lvl_name report_lvllvllvl_namerQ/home/ubuntu/.local/lib/python3.10/site-packages/ddtrace/appsec/_iast/_metrics.pyget_iast_metrics_report_lvls rcsfdd}|S)Ncs@tkrz|WStytjdddYddSwddS)Nz&iast::metrics::error::metric_verbosityT)exc_infocSsdSNrrrrr'sz3metric_verbosity..wrapper..)r Exceptionlogwarning)frrrwrapper!s  z!metric_verbosity..wrapperr)rr(rr'rmetric_verbosity s r)NmsgexcreturncCstr tjj||ddSdS)aUThis was originally implemented to analyze which services were triggering this issue, and we used that insight to refactor how IAST creates and destroys context. However, after that refactor, this information no longer provides value and only adds noise. So now, those telemetry metrics are only emitted if IAST is in debug mode )r+N)r r telemetry_writer add_error_log)r*r+rrr_set_iast_error_metric,sr/cC.ddlm}tjtjddd||ffdS)Nr zinstrumented.source source_type_taint_trackingr r r-add_count_metricrrr2r rrr$_set_metric_iast_instrumented_source7 r7cCstjtjdddS)Nzinstrumented.propagationr1r r-r5rrrrrr)_set_metric_iast_instrumented_propagation@sr:r1cCstjtjd|d|ffdS)Nzinstrumented.sinkvulnerability_typer9)r;counterrrr"_set_metric_iast_instrumented_sinkEr=cCr0)Nr1r zexecuted.sourcer2r3r6rrr _set_metric_iast_executed_sourceLr8r?cCstjtjddd|ffdS)Nz executed.sinkr1r;r9)r;rrr_set_metric_iast_executed_sinkUr>r@cCs(t}|dkrtjtjd|dSdS)Nrzrequest.tainted)rr r-r5rr)total_objects_taintedrrr _set_metric_iast_request_tainted\srBcCs&t|tr t|}|dd}|S)N._) isinstancer r replacelower)keyrrr_metric_key_as_snake_casecs  rIr!)r1),typingrrddtrace.appsec._constantsrrrddtrace.appsec._deduplicationsr/ddtrace.appsec._iast._iast_request_context_baser$ddtrace.appsec._iast._taint_trackingr r ddtrace.appsec._iast._utilsr ddtrace.internalr ddtrace.internal.loggerrddtrace.internal.settings.asmrr$ddtrace.internal.telemetry.constantsr__name__r$rdictstrint__annotations__rr) BaseExceptiontupler/r7r:r=r?r@rBrIrrrrsD               &