Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

executor: cleanup entries from mysql.db on revoke #38370

Merged
merged 10 commits into from
Oct 11, 2022
Merged

executor: cleanup entries from mysql.db on revoke #38370

Show file tree
Hide file tree
Changes from 9 commits
Commits
Show all changes
10 commits
Select commit Hold shift + click to select a range
713eae5
executor: cleanup entries from mysql.db on revoke
dveeden Oct 10, 2022
8522024
Improve builing of sql string for cleanup of mysql.db
dveeden Oct 10, 2022
10fd150
Merge branch 'master' into executor_revoke_cleanup_db
dveeden Oct 10, 2022
0e2e7fa
Merge branch 'master' into executor_revoke_cleanup_db
dveeden Oct 10, 2022
d1da2ec
Update test for db level grant removal
dveeden Oct 10, 2022
22ebdb8
Fix typo
dveeden Oct 10, 2022
60b5a6b
Don't remove mysql.db record if WITH GRANT OPTION is set
dveeden Oct 10, 2022
4ec7ce7
Update based on PR review
dveeden Oct 11, 2022
c8f87bf
Merge branch 'master' into executor_revoke_cleanup_db
dveeden Oct 11, 2022
0646a7f
Merge branch 'master' into executor_revoke_cleanup_db
ti-chi-bot Oct 11, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Jump to
Jump to file
Failed to load files.
Diff view
Diff view
11 changes: 11 additions & 0 deletions executor/revoke.go
View file
Open in desktop
Expand Up @@ -242,6 +242,17 @@ func (e *RevokeExec) revokeDBPriv(internalSession sessionctx.Context, priv *ast.
}
sqlexec.MustFormatSQL(sql, " WHERE User=%? AND Host=%? AND DB=%?", userName, host, dbName)

_, err = internalSession.(sqlexec.SQLExecutor).ExecuteInternal(ctx, sql.String())
if err != nil {
return err
}

sql = new(strings.Builder)
sqlexec.MustFormatSQL(sql, "DELETE FROM %n.%n WHERE User=%? AND Host=%? AND DB=%?", mysql.SystemDB, mysql.DBTable, userName, host, dbName)

for _, v := range append(mysql.AllDBPrivs, mysql.GrantPriv) {
sqlexec.MustFormatSQL(sql, " AND %n='N'", v.ColumnString())
}
_, err = internalSession.(sqlexec.SQLExecutor).ExecuteInternal(ctx, sql.String())
return err
}
Expand Down
8 changes: 7 additions & 1 deletion executor/revoke_test.go
View file
Open in desktop
Expand Up @@ -72,7 +72,13 @@ func TestRevokeDBScope(t *testing.T) {

tk.MustQuery(check).Check(testkit.Rows("Y"))
tk.MustExec(sql)
tk.MustQuery(check).Check(testkit.Rows("N"))
if v == mysql.AllDBPrivs[len(mysql.AllDBPrivs)-1] {
// When all privileges are set to 'N', then the record should be removed as well.
// https://github.com/pingcap/tidb/issues/38363
tk.MustQuery(check).Check(testkit.Rows())
} else {
tk.MustQuery(check).Check(testkit.Rows("N"))
}
}
}

Expand Down