fix(milvus,baidu): sanitize filter values to prevent expression injection (#5746)

This commit is contained in:
Hrushikesh Yadav
2026-06-23 16:51:05 +05:30
committed by GitHub
parent c0ac9f81fa
commit 565db27121
4 changed files with 85 additions and 4 deletions
+27
View File
@@ -310,6 +310,33 @@ class TestMilvusDB:
with pytest.raises(ValueError, match="no vector data"):
milvus_db.update(vector_id="test_id", vector=None, payload={"data": "test"})
def test_create_filter_rejects_expression_injection(self, milvus_db):
"""Crafted string value must not break out of the quoted expression."""
with pytest.raises(ValueError, match="must be str, int, float, or bool"):
milvus_db._create_filter({"user_id": {"$ne": ""}})
def test_create_filter_rejects_malicious_key(self, milvus_db):
"""Keys with special characters must be rejected."""
with pytest.raises(ValueError, match="Invalid filter key"):
milvus_db._create_filter({'"] == "") or true or ("': "x"})
def test_create_filter_escapes_quotes_in_value(self, milvus_db):
"""Double-quotes inside string values must be escaped."""
result = milvus_db._create_filter({"user_id": 'alice"}'})
assert '\\"' in result
assert 'alice\\"' in result
def test_create_filter_escapes_backslash_and_quote(self, milvus_db):
"""Backslashes and double-quotes in the same value must both be escaped."""
result = milvus_db._create_filter({"user_id": r'alice\path"beta'})
assert result == r'(metadata["user_id"] == "alice\\path\"beta")'
def test_create_filter_renders_boolean(self, milvus_db):
"""Boolean values must be rendered unquoted in the backend's expected format."""
result = milvus_db._create_filter({"active": True, "deleted": False})
assert '(metadata["active"] == True)' in result
assert '(metadata["deleted"] == False)' in result
def test_collection_already_exists(self, mock_milvus_client):
"""Test that existing collection is not recreated."""
mock_milvus_client.has_collection.return_value = True