mirror of
https://github.com/fluencelabs/redis
synced 2025-06-18 03:31:21 +00:00
ACL: populate category flags from command table.
This commit is contained in:
56
src/server.c
56
src/server.c
@ -2875,23 +2875,47 @@ int populateCommandTableParseFlags(struct redisCommand *c, char *strflags) {
|
||||
|
||||
for (int j = 0; j < argc; j++) {
|
||||
char *flag = argv[j];
|
||||
if (!strcasecmp(flag,"write")) c->flags |= CMD_WRITE;
|
||||
else if (!strcasecmp(flag,"read-only")) c->flags |= CMD_READONLY;
|
||||
else if (!strcasecmp(flag,"use-memory")) c->flags |= CMD_DENYOOM;
|
||||
else if (!strcasecmp(flag,"admin")) c->flags |= CMD_ADMIN;
|
||||
else if (!strcasecmp(flag,"pub-sub")) c->flags |= CMD_PUBSUB;
|
||||
else if (!strcasecmp(flag,"no-script")) c->flags |= CMD_NOSCRIPT;
|
||||
else if (!strcasecmp(flag,"random")) c->flags |= CMD_RANDOM;
|
||||
else if (!strcasecmp(flag,"to-sort")) c->flags |= CMD_SORT_FOR_SCRIPT;
|
||||
else if (!strcasecmp(flag,"ok-loading")) c->flags |= CMD_LOADING;
|
||||
else if (!strcasecmp(flag,"ok-stale")) c->flags |= CMD_STALE;
|
||||
else if (!strcasecmp(flag,"no-monitor")) c->flags |= CMD_SKIP_MONITOR;
|
||||
else if (!strcasecmp(flag,"cluster-asking")) c->flags |= CMD_ASKING;
|
||||
else if (!strcasecmp(flag,"fast")) c->flags |= CMD_FAST;
|
||||
else {
|
||||
sdsfreesplitres(argv,argc);
|
||||
return C_ERR;
|
||||
if (!strcasecmp(flag,"write")) {
|
||||
c->flags |= CMD_WRITE|CMD_CATEGORY_WRITE;
|
||||
} else if (!strcasecmp(flag,"read-only")) {
|
||||
c->flags |= CMD_READONLY|CMD_CATEGORY_READ;
|
||||
} else if (!strcasecmp(flag,"use-memory")) {
|
||||
c->flags |= CMD_DENYOOM;
|
||||
} else if (!strcasecmp(flag,"admin")) {
|
||||
c->flags |= CMD_ADMIN|CMD_CATEGORY_ADMIN|CMD_CATEGORY_DANGEROUS;
|
||||
} else if (!strcasecmp(flag,"pub-sub")) {
|
||||
c->flags |= CMD_PUBSUB|CMD_CATEGORY_PUBSUB;
|
||||
} else if (!strcasecmp(flag,"no-script")) {
|
||||
c->flags |= CMD_NOSCRIPT;
|
||||
} else if (!strcasecmp(flag,"random")) {
|
||||
c->flags |= CMD_RANDOM;
|
||||
} else if (!strcasecmp(flag,"to-sort")) {
|
||||
c->flags |= CMD_SORT_FOR_SCRIPT;
|
||||
} else if (!strcasecmp(flag,"ok-loading")) {
|
||||
c->flags |= CMD_LOADING;
|
||||
} else if (!strcasecmp(flag,"ok-stale")) {
|
||||
c->flags |= CMD_STALE;
|
||||
} else if (!strcasecmp(flag,"no-monitor")) {
|
||||
c->flags |= CMD_SKIP_MONITOR;
|
||||
} else if (!strcasecmp(flag,"cluster-asking")) {
|
||||
c->flags |= CMD_ASKING;
|
||||
} else if (!strcasecmp(flag,"fast")) {
|
||||
c->flags |= CMD_FAST | CMD_CATEGORY_FAST;
|
||||
} else {
|
||||
/* Parse ACL categories here if the flag name starts with @. */
|
||||
uint64_t catflag;
|
||||
if (flag[0] == '@' &&
|
||||
(catflag = ACLGetCommandCategoryFlagByName(flag+1)) != 0)
|
||||
{
|
||||
c->flags |= catflag;
|
||||
} else {
|
||||
sdsfreesplitres(argv,argc);
|
||||
return C_ERR;
|
||||
}
|
||||
}
|
||||
|
||||
/* If it's not @fast is @slow in this binary world. */
|
||||
if (!(c->flags & CMD_CATEGORY_FAST)) c->flags |= CMD_CATEGORY_SLOW;
|
||||
}
|
||||
sdsfreesplitres(argv,argc);
|
||||
return C_OK;
|
||||
|
Reference in New Issue
Block a user