gardel/lightrag
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Potential fix for code scanning alert no. 13: Clear-text logging of sensitive information

Browse Source 
Co-authored-by: Copilot Autofix powered by AI <62310815+github-advanced-security[bot]@users.noreply.github.com>
This commit is contained in:
VenkateshPabbati
2025-04-09 20:56:12 +05:30
committed by GitHub
parent c02e437104
commit 3c62489a18
1 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
lightrag/kg/tidb_impl.py
View File

@@ -25,8 +25,10 @@ from sqlalchemy import create_engine, text # type: ignore
def sanitize_sensitive_info(data: dict) -> dict:
sanitized_data = data.copy()
if 'password' in sanitized_data:
sanitized_data['password'] = '***'
sensitive_fields = ['password', 'user', 'host', 'database']
for field in sensitive_fields:
if field in sanitized_data:
sanitized_data[field] = '***'
return sanitized_data
class TiDB:
@@ -76,7 +78,7 @@ class TiDB:
result = conn.execute(text(sql), params)
except Exception as e:
sanitized_params = sanitize_sensitive_info(params)
logger.error(f"Tidb database,\nsql:{sql},\nparams:{sanitized_params},\nerror:{e}")
logger.error(f"Tidb database,\nsql:{sql},\nparams:{sanitize_sensitive_info(sanitized_params)},\nerror:{e}")
raise
if multirows:
rows = result.all()
@@ -102,7 +104,7 @@ class TiDB:
conn.execute(text(sql), parameters=data)
except Exception as e:
sanitized_data = sanitize_sensitive_info(data) if data else None
logger.error(f"Tidb database,\nsql:{sql},\ndata:{sanitized_data},\nerror:{e}")
logger.error(f"Tidb database,\nsql:{sql},\ndata:{sanitize_sensitive_info(sanitized_data)},\nerror:{e}")
raise