fix: 解决潜在的 SQL 注入漏洞 (#5906)

影响范围:操作日志
This commit is contained in:
zhengkunwang 2024-07-22 22:41:31 +08:00 committed by GitHub
parent 68922676fc
commit 3339ba9bad
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -96,10 +96,9 @@ func OperationLog() gin.HandlerFunc {
if funcs.InputValue == key {
var names []string
if funcs.IsList {
sql := fmt.Sprintf("SELECT %s FROM %s where %s in (?);", funcs.OutputColumn, funcs.DB, funcs.InputColumn)
_ = global.DB.Raw(sql, value).Scan(&names)
_ = global.DB.Raw("select ? from ? where ? in (?);", funcs.OutputColumn, funcs.DB, funcs.InputColumn, value).Scan(&names)
} else {
_ = global.DB.Raw(fmt.Sprintf("select %s from %s where %s = ?;", funcs.OutputColumn, funcs.DB, funcs.InputColumn), value).Scan(&names)
_ = global.DB.Raw("select ? from ? where ? = ?;", funcs.OutputColumn, funcs.DB, funcs.InputColumn, value).Scan(&names)
}
formatMap[funcs.OutputValue] = strings.Join(names, ",")
break