Redis 6.0 权限控制基于 Bitmap 实现

2020-03-14 22:03:31 +08:00
 RedisMasterNode

Redis 6.0 在 4 月 30 日就要和大家正式见面了，现在redis.io上已经提供了 RC 版本。在之前的博客中，已经介绍过权限控制新功能的一些用法，主要来源于作者 Antirez 在 Redis Day 上的一些演示。Antirez 在最后提到，ACL 的主要实现是基于 Bitmap，因此对性能影响是可以忽略不计的。当时大致猜想了一下实现的思路，那么现在离发布已经很近了，作者也对 ACL Logging 进行了一些补充，不妨一起来看一下。

user 结构

server.h 中定义了对应的user结构保存用户的 ACL 信息，包括：

typedef struct user {
    sds name;
    uint64_t flags;
    uint64_t allowed_commands[USER_COMMAND_BITS_COUNT/64];
    sds **allowed_subcommands;
    list *passwords;
    list *patterns;
} user;

补充一下一些新鲜的字段描述，allowed_commands实际上是一个（默认）长度为 1024 的位图，它的 index 对应各个命令的 ID，在历史版本中命令结构redisCommand是通过名字（name）来查找的，id为这个版本中新增的属性，专门用于 ACL 功能。

struct redisCommand {
    ...
    int id;
};

user这个结构对应的是client结构的"user"字段，熟悉 Redis 的同学应该对client也有所了解，就不再赘述了。

ACL 操作选读

ACL 的命令很多，总体而言都是围绕着user对象展开的，因此从中挑选了几个函数来看一下具体是如何操作user对象。

一个需要铺垫的通用方法就是ACLGetUserCommandBit，ACL 操作中都会涉及到获取用户的命令位图，ACLGetUserCommandBit()接收一个user结构和命令 ID，根据 ID 定位出命令在allowed_commands中的位置，通过位运算返回用户是否有该命令权限。

int ACLGetUserCommandBit(user *u, unsigned long id) {
    uint64_t word, bit;
    if (ACLGetCommandBitCoordinates(id,&word,&bit) == C_ERR) return 0;
    return (u->allowed_commands[word] & bit) != 0;
}

当用户进行 Redis 操作时，例如set操作，操作的命令会保存在client结构的*cmd字段中，*cmd字段就是一个redisCommand结构的指针，redisCommand结构包含了命令的id，因此在使用时通过ACLGetUserCommandBit(u, cmd->id)传入。

创建用户

创建用户分为两步，首先需要创建一个user，通过调用ACLCreateUser(const char *name, size_t namelen)实现，返回的是一个user对象的指针。在创建时，会在server.h定义的Users中查找是否有同名用户，也是本次功能新增的，因为旧版本中只有"default"用户。此时这个用户拥有名称，flag 被初始化为禁用用户，其余的属性均为 Null 或空 list 等。

然后，通过调用ACLSetUser(user *u, const char *op, ssize_t oplen)，调整传入用户u的对应属性，调整内容放在名为op操作的参数中。这个函数非常长，主要是针对各种不同的“操作” switch case 处理，节选部分如下：

int ACLSetUser(user *u, const char *op, ssize_t oplen) {
    if (oplen == -1) oplen = strlen(op);
    /* Part1 - 处理用户状态(flag)操作 */
    // 控制用户启用状态
    if (!strcasecmp(op,"on")) {
        u->flags |= USER_FLAG_ENABLED;
        u->flags &= ~USER_FLAG_DISABLED;
    } else if (!strcasecmp(op,"off")) {
        u->flags |= USER_FLAG_DISABLED;
        u->flags &= ~USER_FLAG_ENABLED;
    // 控制全局键、命令等可用与否
    } else if (!strcasecmp(op,"allkeys") ||
               !strcasecmp(op,"~*"))
    {
        u->flags |= USER_FLAG_ALLKEYS;
        listEmpty(u->patterns);
    }
    ...


    /* Part2 - 操作用户密码增删改查 */
    // > 和 < 等控制密码的改动删除等
    else if (op[0] == '>' || op[0] == '#') {
        sds newpass;
        if (op[0] == '>') {
            newpass = ACLHashPassword((unsigned char*)op+1,oplen-1);
        }


    /* Part3 - 操作用户可用命令的范围 */
    else if (op[0] == '+' && op[1] != '@') {
        if (strchr(op,'|') == NULL) {
            if (ACLLookupCommand(op+1) == NULL) {
                errno = ENOENT;
                return C_ERR;
            }
            unsigned long id = ACLGetCommandID(op+1);
            // 根据传入的 id 参数设置对应 allowed_commands 位图的值
            ACLSetUserCommandBit(u,id,1);
            // 新调整的命令的子命令数组会被重置
            ACLResetSubcommandsForCommand(u,id);
        }
    }

补充一下具体调用例子，其实 Redis 的默认用户就是按照这套流程创建的：初始化名为“default”的空白无权限用户，然后为这个用户设置上所有权限：

DefaultUser = ACLCreateUser("default",7);
ACLSetUser(DefaultUser,"+@all",-1);
ACLSetUser(DefaultUser,"~*",-1);
ACLSetUser(DefaultUser,"on",-1);
ACLSetUser(DefaultUser,"nopass",-1);

拦截不可用命令 /键

命令 /键拦截操作非常简单：

ACL 判断

我们先看一下“不可用”的判断逻辑，然后再回到命令执行流程中看判断方法的调用。

判断函数同样非常长，展示完后会进行总结：

int ACLCheckCommandPerm(client *c, int *keyidxptr) {
    user *u = c->user;
    uint64_t id = c->cmd->id;
    // 命令相关的全局 flag 的检查，若满足则跳过后续部分
    if (!(u->flags & USER_FLAG_ALLCOMMANDS) &&
        c->cmd->proc != authCommand)
    {
        // 即使当前命令没有在 allowed_commands 中，还要检查子命令是否可用
        // 以免出现仅开放了部分子命令权限的情况
        if (ACLGetUserCommandBit(u,id) == 0) {
            ...
            // 遍历子命令
            long subid = 0;
            while (1) {
                if (u->allowed_subcommands[id][subid] == NULL)
                    return ACL_DENIED_CMD;
                if (!strcasecmp(c->argv[1]->ptr,
                                u->allowed_subcommands[id][subid]))
                    break; // 子命令可用，跳出循环
                subid++;
            }
        }
    }

    // 键相关的全局 flag 检查，若满足则跳过后续部分
    if (!(c->user->flags & USER_FLAG_ALLKEYS) &&
        (c->cmd->getkeys_proc || c->cmd->firstkey))
    {
        int numkeys;
        // 先拿到当前要进行操作的 Key
        int *keyidx = getKeysFromCommand(c->cmd,c->argv,c->argc,&numkeys);
        for (int j = 0; j < numkeys; j++) {
            listIter li;
            listNode *ln;
            listRewind(u->patterns,&li);

            // 检查当前 user 所有的关于 Key 的匹配 Pattern
            // 如果有任意命中则跳出，否则判定不可用
            int match = 0;
            while((ln = listNext(&li))) {
                sds pattern = listNodeValue(ln);
                size_t plen = sdslen(pattern);
                int idx = keyidx[j];
                if (stringmatchlen(pattern,plen,c->argv[idx]->ptr,
                                   sdslen(c->argv[idx]->ptr),0))
                {
                    match = 1;
                    break;
                }
            }
            if (!match) {
                if (keyidxptr) *keyidxptr = keyidx[j];
                getKeysFreeResult(keyidx);
                return ACL_DENIED_KEY;
            }
        }
        getKeysFreeResult(keyidx);
    }
    return ACL_OK;
}

那么为了方便喜欢跳过代码的同学看结论：

命令执行流程中的调用

判断逻辑之后到何时调用这套判断。我们先来复习一下 Redis 如何执行命令：

目前执行命令的方法是在server.c中的processCommand(client *c)，传入client对象，执行，返回执行成功与否。我们节选其中关于 ACL 的部分如下：

int processCommand(client *c) {
    ...
    int acl_keypos;
    int acl_retval = ACLCheckCommandPerm(c,&acl_keypos);
    if (acl_retval != ACL_OK) {
        addACLLogEntry(c,acl_retval,acl_keypos,NULL);
        flagTransaction(c);
        if (acl_retval == ACL_DENIED_CMD)
            addReplyErrorFormat(c,
                "-NOPERM this user has no permissions to run "
                "the '%s' command or its subcommand", c->cmd->name);
        else
            addReplyErrorFormat(c,
                "-NOPERM this user has no permissions to access "
                "one of the keys used as arguments");
        return C_OK;
    }
    ...

在命令解析之后，真正执行之前，通过调用ACLCheckCommandPerm获取判断结果，如果判定不通过，进行以下操作：

因此这次 ACL 功能影响的是执行命令前后的操作。

其他功能对 ACL 的调用

通过搜索可以发现一共有 3 处调用了ACLCheckCommandPerm方法：

/home/duck/study/redis/src/multi.c:
  179  
  180          int acl_keypos;
  181:         int acl_retval = ACLCheckCommandPerm(c,&acl_keypos);
  182          if (acl_retval != ACL_OK) {
  183              addACLLogEntry(c,acl_retval,acl_keypos,NULL);

/home/duck/study/redis/src/scripting.c:
  608      /* Check the ACLs. */
  609      int acl_keypos;
  610:     int acl_retval = ACLCheckCommandPerm(c,&acl_keypos);
  611      if (acl_retval != ACL_OK) {
  612          addACLLogEntry(c,acl_retval,acl_keypos,NULL);

/home/duck/study/redis/src/server.c:
 3394       * ACLs. */
 3395      int acl_keypos;
 3396:     int acl_retval = ACLCheckCommandPerm(c,&acl_keypos);
 3397      if (acl_retval != ACL_OK) {
 3398          addACLLogEntry(c,acl_retval,acl_keypos,NULL);

形式都是大同小异，了解一下即可。总结一下需要判定 ACL 的位置：

总结

补充一张图来描述新增的 ACL 功能相关的结构： 图中部分的表达可能与实际的数据结构有所差异，主要原因是代码理解和 C 语言的语法掌握不到位所致。

阅读代码的过程中留意到，对命令的限制是通过 Bitmap 来实现的，而对 Key 的限制是通过特定 Pattern 来实现的。当对 Key 的限制 Pattern 数量特别多时，是否会因为匹配 Pattern 而对性能造成影响，例如超多次的stringmatchlen()执行。当然这一块内容似乎确实没有想到什么提升非常大的判断方式，后续也会继续关注 ACL 的相关改进。

博客： https://blog.2014bduck.com/archives/343
备注：毕业不久多积累一点总是好的 orz，如果解读得不正确或者不恰当欢迎邮件骚扰 2014bduck@gmail.com
6948 次点击
所在节点    Redis
3 条回复
longlong
2020-03-15 12:16:34 +08:00
挺不错，我竟然读完了
AyoCross
2020-04-07 10:26:22 +08:00
redis 为何会考虑用 ACL，而不是 RBAC 或者更复杂的 ABAC 呢
RedisMasterNode
2020-04-07 11:26:07 +08:00
@AyoCross 个人认为因为简单才符合 redis 各项功能设计的理念

这是一个专为移动设备优化的页面（即为了让你能够在 Google 搜索结果里秒开这个页面），如果你希望参与 V2EX 社区的讨论，你可以继续到 V2EX 上打开本讨论主题的完整版本。

https://www.v2ex.com/t/652851

V2EX 是创意工作者们的社区，是一个分享自己正在做的有趣事物、交流想法，可以遇见新朋友甚至新机会的地方。

V2EX is a community of developers, designers and creative people.

© 2021 V2EX