fix(milvus,baidu): sanitize filter values to prevent expression injection (#5746)
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user