Skip to content

Sanitize table names for kusto query#2070

Merged
xiangyan99 merged 3 commits intomainfrom
kusto_sanitize_table_name
Mar 16, 2026
Merged

Sanitize table names for kusto query#2070
xiangyan99 merged 3 commits intomainfrom
kusto_sanitize_table_name

Conversation

@xiangyan99
Copy link
Copy Markdown
Member

What does this PR do?

[Provide a clear, concise description of the changes]

Sanitize table names for kusto query

[Any additional context, screenshots, or information that helps reviewers]

GitHub issue number?

[Link to the GitHub issue this PR addresses]

Pre-merge Checklist

  • Required for All PRs
    • Read contribution guidelines
    • PR title clearly describes the change
    • Commit history is clean with descriptive messages (cleanup guide)
    • Added comprehensive tests for new/modified functionality
    • Updated servers/Azure.Mcp.Server/CHANGELOG.md and/or servers/Fabric.Mcp.Server/CHANGELOG.md for product changes (features, bug fixes, UI/UX, updated dependencies)
  • For MCP tool changes:
    • One tool per PR: This PR adds or modifies only one MCP tool for faster review cycles
    • Updated servers/Azure.Mcp.Server/README.md and/or servers/Fabric.Mcp.Server/README.md documentation
    • Validate README.md changes using script at eng/scripts/Process-PackageReadMe.ps1. See Package README
    • Updated command list in /servers/Azure.Mcp.Server/docs/azmcp-commands.md and/or /docs/fabric-commands.md
    • Run .\eng\scripts\Update-AzCommandsMetadata.ps1 to update tool metadata in azmcp-commands.md (required for CI)
    • For new or modified tool descriptions, ran ToolDescriptionEvaluator and obtained a score of 0.4 or more and a top 3 ranking for all related test prompts
    • For tools with new names, including new tools or renamed tools, update consolidated-tools.json
    • For renamed tools, follow the Tool Rename Checklist and tag the PR with the breaking-change label
    • For new tools associated with Azure services or publicly available tools/APIs/products, add URL to documentation in the PR description
  • Extra steps for Azure MCP Server tool changes:
    • Updated test prompts in /servers/Azure.Mcp.Server/docs/e2eTestPrompts.md
    • 👉 For Community (non-Microsoft team member) PRs:
      • Security review: Reviewed code for security vulnerabilities, malicious code, or suspicious activities before running tests (crypto mining, spam, data exfiltration, etc.)
      • Manual tests run: added comment /azp run mcp - pullrequest - live to run Live Test Pipeline

Copy link
Copy Markdown
Contributor

Copilot AI left a comment

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 mitigates KQL injection risks in the Kusto tool by escaping user-controlled table identifiers before interpolating them into KQL/management commands.

Changes:

  • Added KustoService.EscapeKqlIdentifier to escape table identifiers using KQL bracket notation and reject certain control characters.
  • Updated the sample query construction and corresponding unit tests to use escaped identifiers.
  • Added unit tests covering escaping behavior and control-character rejection; added a changelog entry documenting the security fix.

Reviewed changes

Copilot reviewed 5 out of 5 changed files in this pull request and generated 1 comment.

Show a summary per file
File Description
tools/Azure.Mcp.Tools.Kusto/tests/Azure.Mcp.Tools.Kusto.UnitTests/SampleCommandTests.cs Updates expected query strings to match escaped/bracketed table identifiers.
tools/Azure.Mcp.Tools.Kusto/tests/Azure.Mcp.Tools.Kusto.UnitTests/EscapeKqlIdentifierTests.cs Adds new unit tests for identifier escaping and injection prevention.
tools/Azure.Mcp.Tools.Kusto/src/Services/KustoService.cs Introduces identifier escaping helper and applies it to .show table ... schema query.
tools/Azure.Mcp.Tools.Kusto/src/Commands/SampleCommand.cs Escapes the user-provided table name when building the sample query.
servers/Azure.Mcp.Server/changelog-entries/1773696709347.yaml Documents the bug/security fix in the Azure MCP Server changelog entries.

@xiangyan99 xiangyan99 merged commit 509e8fe into main Mar 16, 2026
15 checks passed
@xiangyan99 xiangyan99 deleted the kusto_sanitize_table_name branch March 16, 2026 23:26
@github-project-automation github-project-automation Bot moved this from Untriaged to Done in Azure MCP Server Mar 16, 2026
colbytimm pushed a commit to colbytimm/microsoft-mcp that referenced this pull request Apr 20, 2026
* Sanitize table names for kusto query

* dotnet format

* address feedback
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

Status: Done

Development

Successfully merging this pull request may close these issues.

3 participants