Skip to content

🔒 Fix potential SQL Injection in SQLite schema modification#27

Open
Gunnarguy wants to merge 1 commit into
mainfrom
fix/sql-injection-ensurecolumnexists-5969573126713650346
Open

🔒 Fix potential SQL Injection in SQLite schema modification#27
Gunnarguy wants to merge 1 commit into
mainfrom
fix/sql-injection-ensurecolumnexists-5969573126713650346

Conversation

@Gunnarguy

@Gunnarguy Gunnarguy commented Jun 24, 2026

Copy link
Copy Markdown
Owner

🎯 What: Fixed a potential SQL injection vulnerability in ensureColumnExists within SQLiteFullTextService.swift where table and column names were unsafely interpolated into SQL strings.

⚠️ Risk: Although currently called with hardcoded values, the function's signature allowed strings to be passed and interpolated directly into PRAGMA table_info(...) and ALTER TABLE ... ADD COLUMN ... queries. If used dynamically in the future, it could allow an attacker to inject arbitrary SQL statements or alter schema in unintended ways.

🛡️ Solution: Implemented early validation of the table and column parameters against a strict regular expression ^[a-zA-Z_][a-zA-Z0-9_]*$. This ensures that only valid SQL identifiers can be passed. If validation fails, an error is logged and the function returns early, preventing the query from being executed.

(Note: Tests could not be run directly due to the lack of Swift toolchain in the environment, but the changes were carefully reviewed manually and follow standard Swift regex validation patterns and logging practices.)


PR created automatically by Jules for task 5969573126713650346 started by @Gunnarguy

Summary by Sourcery

Bug Fixes:

  • Harden SQLite schema modification by rejecting invalid table and column identifiers before constructing PRAGMA and ALTER TABLE queries.

Co-authored-by: Gunnarguy <110250624+Gunnarguy@users.noreply.github.com>
@google-labs-jules

Copy link
Copy Markdown

👋 Jules, reporting for duty! I'm here to lend a hand with this pull request.

When you start a review, I'll add a 👀 emoji to each comment to let you know I've read it. I'll focus on feedback directed at me and will do my best to stay out of conversations between you and other bots or reviewers to keep the noise down.

I'll push a commit with your requested changes shortly after. Please note there might be a delay between these steps, but rest assured I'm on the job!

For more direct control, you can switch me to Reactive Mode. When this mode is on, I will only act on comments where you specifically mention me with @jules. You can find this option in the Pull Request section of your global Jules UI settings. You can always switch back!

New to Jules? Learn more at jules.google/docs.


For security, I will only act on instructions from the user who triggered this task.

Copilot AI review requested due to automatic review settings June 24, 2026 23:08
@sourcery-ai

sourcery-ai Bot commented Jun 24, 2026

Copy link
Copy Markdown
Reviewer's guide (collapsed on small PRs)

Reviewer's Guide

Adds strict identifier validation to SQLiteFullTextService.ensureColumnExists to prevent potential SQL injection when constructing schema-modifying SQL statements from table/column parameters.

Sequence diagram for updated ensureColumnExists validation flow

sequenceDiagram
    participant Caller
    participant SQLiteFullTextService
    participant Log
    participant SQLite

    Caller->>SQLiteFullTextService: ensureColumnExists(table, column, definition)
    SQLiteFullTextService->>SQLiteFullTextService: check database
    alt invalid table identifier
        SQLiteFullTextService->>Log: Log.error("[SQLiteFTS5] Invalid table name for schema modification")
        SQLiteFullTextService-->>Caller: return
    else invalid column identifier
        SQLiteFullTextService->>Log: Log.error("[SQLiteFTS5] Invalid column name for schema modification")
        SQLiteFullTextService-->>Caller: return
    else identifiers valid
        SQLiteFullTextService->>SQLite: sqlite3_prepare_v2(db, "PRAGMA table_info(table)")
        SQLite-->>SQLiteFullTextService: statement
        SQLiteFullTextService->>SQLite: sqlite3_step / sqlite3_finalize
        SQLite-->>SQLiteFullTextService: result
        SQLiteFullTextService-->>Caller: schema ensured
    end
Loading

File-Level Changes

Change Details Files
Add defensive validation of table and column identifiers before executing schema-modifying SQLite statements to prevent SQL injection.
  • Introduce a regular-expression-based identifier whitelist for table and column names
  • Guard-return early if the table name does not match the allowed identifier pattern and log an error in the vectorDB category
  • Guard-return early if the column name does not match the allowed identifier pattern and log an error in the vectorDB category
  • Leave existing PRAGMA table_info and ALTER TABLE logic unchanged, now protected by the identifier validation
OpenIntelligence/Services/Storage/SQLiteFullTextService.swift

Tips and commands

Interacting with Sourcery

  • Trigger a new review: Comment @sourcery-ai review on the pull request.
  • Continue discussions: Reply directly to Sourcery's review comments.
  • Generate a GitHub issue from a review comment: Ask Sourcery to create an
    issue from a review comment by replying to it. You can also reply to a
    review comment with @sourcery-ai issue to create an issue from it.
  • Generate a pull request title: Write @sourcery-ai anywhere in the pull
    request title to generate a title at any time. You can also comment
    @sourcery-ai title on the pull request to (re-)generate the title at any time.
  • Generate a pull request summary: Write @sourcery-ai summary anywhere in
    the pull request body to generate a PR summary at any time exactly where you
    want it. You can also comment @sourcery-ai summary on the pull request to
    (re-)generate the summary at any time.
  • Generate reviewer's guide: Comment @sourcery-ai guide on the pull
    request to (re-)generate the reviewer's guide at any time.
  • Resolve all Sourcery comments: Comment @sourcery-ai resolve on the
    pull request to resolve all Sourcery comments. Useful if you've already
    addressed all the comments and don't want to see them anymore.
  • Dismiss all Sourcery reviews: Comment @sourcery-ai dismiss on the pull
    request to dismiss all existing Sourcery reviews. Especially useful if you
    want to start fresh with a new review - don't forget to comment
    @sourcery-ai review to trigger a new review!

Customizing Your Experience

Access your dashboard to:

  • Enable or disable review features such as the Sourcery-generated pull request
    summary, the reviewer's guide, and others.
  • Change the review language.
  • Add, remove or edit custom review instructions.
  • Adjust other review settings.

Getting Help

@sourcery-ai sourcery-ai Bot left a comment

Copy link
Copy Markdown

Choose a reason for hiding this comment

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

Hey - I've left some high level feedback:

  • Consider extracting the identifier regex into a shared constant or helper (e.g., a static on SQLiteFullTextService or a small IdentifierValidator) so that future identifier validation stays consistent and is easier to reuse and update.
Prompt for AI Agents
Please address the comments from this code review:

## Overall Comments
- Consider extracting the identifier regex into a shared constant or helper (e.g., a static on SQLiteFullTextService or a small IdentifierValidator) so that future identifier validation stays consistent and is easier to reuse and update.

Sourcery is free for open source - if you like our reviews please consider sharing them ✨
Help me be more useful! Please click 👍 or 👎 on each comment and I'll use the feedback to improve your reviews.

Copilot AI left a comment

Copy link
Copy Markdown

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 hardens SQLite schema-migration logic in SQLiteFullTextService by validating schema identifiers before constructing dynamic PRAGMA table_info(...) and ALTER TABLE ... ADD COLUMN ... SQL strings, reducing the risk of SQL injection if the helper is ever called with non-hardcoded values.

Changes:

  • Added strict regex validation for table and column identifiers in ensureColumnExists.
  • Logged and returned early when invalid identifiers are provided to prevent executing the constructed SQL.

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

Comment on lines +3061 to +3071
// Security fix: Validate table and column names to prevent SQL injection
let validIdentifierRegex = "^[a-zA-Z_][a-zA-Z0-9_]*$"
guard table.range(of: validIdentifierRegex, options: .regularExpression) != nil else {
Log.error("[SQLiteFTS5] Invalid table name for schema modification: \(table)", category: .vectorDB)
return
}
guard column.range(of: validIdentifierRegex, options: .regularExpression) != nil else {
Log.error("[SQLiteFTS5] Invalid column name for schema modification: \(column)", category: .vectorDB)
return
}

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.

2 participants