o i@sddlmZddlmZddlmZddlmZddlmZddl m Z ddl m Z dd l mZdd lmZe eZed d Gd ddeZdS)) dataclass)ClassVar)Optional) SpanProcessor)Span) SpanTypes) get_logger)_iast_end_request)_iast_start_requestF)eqc@sdeZdZUdZeeded<edddZedddZ ddd Z d e fd d Z d e fd dZ dS)AppSecIastSpanProcessorN _instancereturncCs&|jdur|}|_|dSdS)zEnable the IAST span processor.N)rregister)clsinstancerR/home/ubuntu/.local/lib/python3.10/site-packages/ddtrace/appsec/_iast/processor.pyenables   zAppSecIastSpanProcessor.enablecCs"|jdur|jd|_dSdS)z Disable the IAST span processor.N)r unregister)rrrrdisables   zAppSecIastSpanProcessor.disablecCsddlm}|dS)Nr ) load_iast)r)selfrrrr __post_init__#s  z%AppSecIastSpanProcessor.__post_init__spancCs|jtjkrdSt|dS)N) span_typerWEBr rrrrr on_span_start(s  z%AppSecIastSpanProcessor.on_span_startcCs|jtjkrdSt|ddS)aRReport reported vulnerabilities. Span Tags: - `_dd.iast.json`: Only when one or more vulnerabilities have been detected will we include the custom tag. - `_dd.iast.enabled`: Set to 1 when IAST is enabled in a request. If a request is disabled (e.g. by sampling), then it is not set. N)r)rrrr rrrron_span_finish.s z&AppSecIastSpanProcessor.on_span_finish)rN)__name__ __module__ __qualname__rrr__annotations__ classmethodrrrrr r!rrrrr s    r N) dataclassesrtypingrrddtrace._trace.processorrddtrace._trace.spanr ddtrace.extrddtrace.internal.loggerr_iast_request_contextr _iast_request_context_baser r"logr rrrrs