Skip to content

ci: Add model name validation for model management requests - #8672

Merged
pskiran1 merged 7 commits into
mainfrom
spolisetty/tri-736-psirt-path-traversal-triton-inference-server-grpc-interface
Mar 3, 2026
Merged

ci: Add model name validation for model management requests#8672
pskiran1 merged 7 commits into
mainfrom
spolisetty/tri-736-psirt-path-traversal-triton-inference-server-grpc-interface

Conversation

@pskiran1

@pskiran1 pskiran1 commented Feb 23, 2026

Copy link
Copy Markdown
Member

What does the PR do?

This PR adds comprehensive test coverage for model name validation in model management operations (load/unload). The tests verify that the server properly rejects malicious model names that could be used for path traversal attacks, as well as empty model names.

Checklist

  • PR title reflects the change and is of format <commit_type>: <Title>
  • Changes are described in the pull request.
  • Related issues are referenced.
  • Populated github labels field
  • Added test plan and verified test passes.
  • Verified that the PR passes existing CI.
  • Verified copyright is correct on all changed files.
  • Added succinct git squash message before merging ref.
  • All template sections are filled out.
  • Optional: Additional screenshots for behavior/output changes with before/after.

Commit Type:

Check the conventional commit type
box here and add the label to the github PR.

  • build
  • ci
  • docs
  • feat
  • fix
  • perf
  • refactor
  • revert
  • style
  • test

Related PRs: triton-inference-server/core#472

Where should the reviewer start?

Test plan:

  • CI Pipeline ID: 44596299

Caveats:

Background

Related Issues: (use one of the action keywords Closes / Fixes / Resolves / Relates to)

  • closes GitHub issue: #xxx

Copilot AI left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Pull request overview

This PR adds comprehensive test coverage for model name validation in model management operations (load/unload). The tests verify that the server properly rejects malicious model names that could be used for path traversal attacks, as well as empty model names.

Changes:

  • Added new test suite ModelNameValidationTest to validate model name security checks
  • Added test execution block in test.sh to run the model name validation tests
  • Updated copyright years to 2024-2026

Reviewed changes

Copilot reviewed 2 out of 2 changed files in this pull request and generated 3 comments.

File Description
qa/L0_input_validation/test.sh Added test orchestration for model name validation tests, including server setup with explicit model control mode
qa/L0_input_validation/input_validation_test.py Added ModelNameValidationTest class with tests for path traversal prevention, empty name rejection, and valid name handling

💡 Add Copilot custom instructions for smarter, more guided reviews. Learn how to get started.

Comment thread qa/L0_input_validation/input_validation_test.py
Comment thread qa/L0_input_validation/input_validation_test.py
Comment thread qa/L0_input_validation/test.sh
pskiran1 and others added 2 commits February 23, 2026 11:24
Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>

@whoisj whoisj left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Looks good overall, left a couple of comments.

Comment thread qa/L0_input_validation/input_validation_test.py Outdated
Comment thread qa/L0_input_validation/input_validation_test.py Outdated
Comment thread qa/L0_input_validation/input_validation_test.py Outdated
Comment thread qa/L0_input_validation/input_validation_test.py Outdated
whoisj
whoisj previously approved these changes Feb 27, 2026
yinggeh
yinggeh previously approved these changes Feb 27, 2026
 into spolisetty/tri-736-psirt-path-traversal-triton-inference-server-grpc-interface
@pskiran1
pskiran1 requested a review from yinggeh March 1, 2026 11:19

@yinggeh yinggeh left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Well done

@pskiran1
pskiran1 merged commit 36f4b17 into main Mar 3, 2026
3 checks passed
@pskiran1
pskiran1 deleted the spolisetty/tri-736-psirt-path-traversal-triton-inference-server-grpc-interface branch March 3, 2026 05:14
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

PR: ci Changes to our CI configuration files and scripts

Development

Successfully merging this pull request may close these issues.

4 participants