fix(oauth): avoid overwriting user state when binding

This commit is contained in:
CaIon
2026-08-11 22:03:45 +08:00
parent 3d5dc36f1d
commit d7992672a6
11 changed files with 125 additions and 22 deletions
+24
View File
@@ -190,6 +190,30 @@ func UpdateUserSetting(userId int, setting dto.UserSetting) error {
return updateUserSettingCache(userId, settingValue)
}
// userBindColumns 允许通过 UpdateUserBindColumn 更新的第三方账号绑定列白名单。
// 列名只可能来自代码内部的 provider 实现,白名单是防御纵深,不依赖调用方自律。
var userBindColumns = map[string]bool{
"github_id": true,
"discord_id": true,
"oidc_id": true,
"linux_do_id": true,
"wechat_id": true,
}
// UpdateUserBindColumn 第三方账号绑定字段的专用更新。
// 绑定操作必须只写绑定列:若改为“读取完整用户 → 改一个字段 → 整体更新”,
// 读快照期间并发发生的封禁、降权或分组变更会被旧快照覆盖恢复。
// 角色、状态、分组只允许通过各自带锁/CAS 的专用方法修改。
func UpdateUserBindColumn(userId int, column string, value string) error {
if userId <= 0 {
return errors.New("id 为空!")
}
if !userBindColumns[column] {
return fmt.Errorf("invalid user bind column: %s", column)
}
return DB.Model(&User{}).Where("id = ?", userId).Update(column, value).Error
}
// 根据用户角色生成默认的边栏配置
func generateDefaultSidebarConfigForRole(userRole int) string {
defaultConfig := map[string]interface{}{}
+60
View File
@@ -27,6 +27,21 @@ func setupUserUpdateTestState(t *testing.T) {
})
}
func createUserBindTestUser(t *testing.T) User {
t.Helper()
user := User{
Username: "bind-test-user",
Password: "unused-password-hash",
Role: common.RoleCommonUser,
Status: common.UserStatusEnabled,
Group: "default",
AuthVersion: 1,
AffCode: "bind-test-aff-code",
}
require.NoError(t, DB.Create(&user).Error)
return user
}
func TestUserUpdateDoesNotOverwriteConcurrentAccountingOrTokenChanges(t *testing.T) {
setupUserUpdateTestState(t)
@@ -218,6 +233,51 @@ func TestInsertKeepsBlankPasswordForPasswordlessUser(t *testing.T) {
assert.Empty(t, stored.Password)
}
func TestUpdateUserBindColumnOnlyTouchesTheBindingColumn(t *testing.T) {
truncateTables(t)
user := createUserBindTestUser(t)
require.NoError(t, DB.Model(&User{}).Where("id = ?", user.Id).Updates(map[string]interface{}{
"role": common.RoleAdminUser,
"status": common.UserStatusEnabled,
"group": "vip",
}).Error)
require.NoError(t, UpdateUserBindColumn(user.Id, "github_id", "gh-12345"))
reloaded, err := GetUserById(user.Id, true)
require.NoError(t, err)
assert.Equal(t, "gh-12345", reloaded.GitHubId)
assert.Equal(t, common.RoleAdminUser, reloaded.Role)
assert.Equal(t, common.UserStatusEnabled, reloaded.Status)
assert.Equal(t, "vip", reloaded.Group)
}
func TestUpdateUserBindColumnPreservesRestrictiveChange(t *testing.T) {
truncateTables(t)
user := createUserBindTestUser(t)
require.NoError(t, DB.Model(&User{}).Where("id = ?", user.Id).
Update("status", common.UserStatusDisabled).Error)
require.NoError(t, UpdateUserBindColumn(user.Id, "wechat_id", "wx-open-id"))
reloaded, err := GetUserById(user.Id, true)
require.NoError(t, err)
assert.Equal(t, "wx-open-id", reloaded.WeChatId)
assert.Equal(t, common.UserStatusDisabled, reloaded.Status)
}
func TestUpdateUserBindColumnRejectsNonWhitelistedColumns(t *testing.T) {
truncateTables(t)
user := createUserBindTestUser(t)
for _, column := range []string{"role", "status", "group", "quota", "username", "password", "id"} {
assert.Error(t, UpdateUserBindColumn(user.Id, column, "1"), "column %s must be rejected", column)
}
assert.Error(t, UpdateUserBindColumn(user.Id, "github_id; DROP TABLE users", "x"))
assert.Error(t, UpdateUserBindColumn(0, "github_id", "x"))
}
func TestValidateAndFillRejectsPasswordlessUser(t *testing.T) {
setupUserUpdateTestState(t)