TenderAgent/TenderAgent完整代码/review/demo_1.py

def build_filter(tenant_id: str, tender_document_id=None):
    safe_tenant = tenant_id.replace('"', '\\"')
    print(f'safe_tenant: {safe_tenant}')
    expr = f'tenant_id == "{safe_tenant}"'
    if tender_document_id:
        safe_tender_document = tender_document_id.replace('"', '\\"')
        print(f'safe_tender_document: {safe_tender_document}')
        expr += f' and tender_document_id == "{safe_tender_document}"'
    return expr

# 例子 1：正常值（看不出变化）
print("【例1】正常值")
print("输入 tenant=abc, tender_document=123")
print("输出:", build_filter("abc", "123"))
print("-" * 40)
#
# 例子 2：值里面带了双引号（重点看转义效果）
print("【例2】tenant 里带双引号")
print('输入 tenant=te"nt, tender_document=co"urse')
print("输出:", build_filter('te"nt', 'co"urse'))
print("-" * 40)

# 例子 3：恶意注入尝试（双引号被转义成了字符，失去注入能力）
'''
如果没有转义代码（没有 replace），拼出来的字符串会变成：
tenant_id == "123" and tender_document_id == "" or "1"="1"
'''
print("【例3】模拟注入攻击")
print('输入 tenant=tenant, tender_document=" or "1"="1')
print("输出:", build_filter("tenant", '" or "1"="1'))