pi-lens/rules/tree-sitter-queries/python/python-ssrf.yml

Real-time code feedback for pi — LSP, linters, formatters, type-checking, structural analysis & booboo

# Python Security
# Detects user-controlled URL usage in outbound HTTP request sinks.
id: python-ssrf
name: SSRF Risk
severity: warning
category: security
defect_class: injection
inline_tier: warning
language: python

message: "Potential SSRF sink — validate/allowlist outbound URLs"

description: |
  Outbound HTTP calls with untrusted URLs can allow server-side request forgery.

  ✅ FIX: enforce allowlisted hosts/schemes and reject private/internal targets.

query: |
  (call
    function: (attribute
      object: (identifier) @MOD
      attribute: (identifier) @FN)
    arguments: (argument_list
      [(identifier) (subscript) (call)] @URL)
    (#eq? @MOD "requests")
    (#match? @FN "^(get|post|put|patch|delete|request|head|options)$"))

metavars:
  - MOD
  - FN
  - URL

post_filter: py_ssrf_sink

has_fix: false

tags:
  - python
  - security
  - ssrf
  - cwe-918
  - owasp-a10

examples:
  bad: |
    requests.get(user_url)

  good: |
    if host in ALLOWLIST:
        requests.get(url)