Skip to content

Fix Triton HMAC security vulnerabilities (v2)#5643

Open
pravali96 wants to merge 2 commits intoaws:master-v2from
pravali96:fix/triton-hmac-security-v2-pr
Open

Fix Triton HMAC security vulnerabilities (v2)#5643
pravali96 wants to merge 2 commits intoaws:master-v2from
pravali96:fix/triton-hmac-security-v2-pr

Conversation

@pravali96
Copy link
Collaborator

  • Bug 1: Add HMAC integrity check before pickle deserialization in Triton handler initialize() method (model.py)
  • Bug 2: Replace hardcoded secret key with generate_secret_key() and add _hmac_signing() after ONNX exports (triton_builder.py)
  • Bug 3: Add secret key validation in _start_triton_server() to reject empty/None keys before passing to container (server.py)

Aligns Triton code path with existing HMAC verification patterns used by TorchServe, MMS, TF Serving, and SMD handlers.

Ticket: P400136088

Issue #, if available:

Description of changes:

By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.

@pravali96 pravali96 requested a review from a team as a code owner March 18, 2026 19:59
@pravali96 pravali96 requested a review from nargokul March 18, 2026 19:59
@pravali96 pravali96 force-pushed the fix/triton-hmac-security-v2-pr branch from 3bf5912 to 56bc527 Compare March 18, 2026 22:39
@pravali96 pravali96 force-pushed the fix/triton-hmac-security-v2-pr branch from e6d5423 to 30fdef8 Compare March 19, 2026 17:49
@pravali96 pravali96 force-pushed the fix/triton-hmac-security-v2-pr branch from 5fdbd27 to c241a28 Compare March 19, 2026 18:23
- Bug 1: Add HMAC integrity check before pickle deserialization in
  Triton handler initialize() method (model.py)
- Bug 2: Replace hardcoded secret key with generate_secret_key() and
  add _hmac_signing() after ONNX exports (triton_builder.py)
- Bug 3: Add secret key validation in _start_triton_server() to reject
  empty/None keys before passing to container (server.py)

Aligns Triton code path with existing HMAC verification patterns used
by TorchServe, MMS, TF Serving, and SMD handlers.

Ticket: P400136088
@pravali96 pravali96 force-pushed the fix/triton-hmac-security-v2-pr branch from c241a28 to b4c7ba3 Compare March 19, 2026 18:29
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant