mirror of
https://github.com/lunchcat/sif.git
synced 2026-07-28 22:40:54 -07:00
add recon modules for unauthenticated gpu serving stacks that leak model info without a key: triton, sglang, and torchserve.
40 lines
815 B
YAML
40 lines
815 B
YAML
# TorchServe Management API Exposure Detection Module
|
|
|
|
id: torchserve-api-exposure
|
|
info:
|
|
name: TorchServe Management API Exposure
|
|
author: sif
|
|
severity: high
|
|
description: Detects an exposed TorchServe management api that lists models and accepts registration of arbitrary model archives without authentication
|
|
tags: [torchserve, pytorch, ai, ml, inference, api, exposure, recon]
|
|
|
|
type: http
|
|
|
|
http:
|
|
method: GET
|
|
paths:
|
|
- "{{BaseURL}}/models"
|
|
|
|
matchers:
|
|
- type: status
|
|
status:
|
|
- 200
|
|
|
|
- type: word
|
|
part: body
|
|
words:
|
|
- "\"nextPageToken\""
|
|
|
|
- type: word
|
|
part: body
|
|
words:
|
|
- "\"modelUrl\""
|
|
|
|
extractors:
|
|
- type: regex
|
|
name: torchserve_model
|
|
part: body
|
|
regex:
|
|
- '"modelName"\s*:\s*"([^"]+)"'
|
|
group: 1
|