redis如何检测lua脚本的合理性。

最新推荐文章于 2024-05-08 09:15:41 发布

开心小熊

最新推荐文章于 2024-05-08 09:15:41 发布

阅读量2.3k

点赞数

分类专栏： lua nosql redis

本文链接：https://blog.csdn.net/kaixinfelix/article/details/38234363

版权

redis 同时被 3 个专栏收录

3 篇文章 0 订阅

订阅专栏

nosql

2 篇文章 0 订阅

订阅专栏

lua

1 篇文章 0 订阅

订阅专栏

char *s[32];
    sds code = sdsempty();
    int j = 0;

    /* strict.lua from: http://metalua.luaforge.net/src/lib/strict.lua.html.
     * Modified to be adapted to Redis. */
    s[j++]="local mt = {}\n";
    s[j++]="setmetatable(_G, mt)\n";
    s[j++]="mt.__newindex = function (t, n, v)\n";
    s[j++]="  if debug.getinfo(2) then\n";
    s[j++]="    local w = debug.getinfo(2, \"S\").what\n";
    s[j++]="    if w ~= \"main\" and w ~= \"C\" then\n";
    s[j++]="      error(\"Script attempted to create global variable '\"..tostring(n)..\"'\", 2)\n";
    s[j++]="    end\n";
    s[j++]="  end\n";
    s[j++]="  rawset(t, n, v)\n";
    s[j++]="end\n";
    s[j++]="mt.__index = function (t, n)\n";
    s[j++]="  if debug.getinfo(2) and debug.getinfo(2, \"S\").what ~= \"C\" then\n";
    s[j++]="    error(\"Script attempted to access unexisting global variable '\"..tostring(n)..\"'\", 2)\n";
    s[j++]="  end\n";
    s[j++]="  return rawget(t, n)\n";
    s[j++]="end\n";
    s[j++]=NULL;

    for (j = 0; s[j] != NULL; j++) code = sdscatlen(code,s[j],strlen(s[j]));
    luaL_loadbuffer(lua,code,sdslen(code),"@enable_strict_lua");
    lua_pcall(lua,0,0,0);
    sdsfree(code);

主要是两点：

1、禁止设置新的全局变量

2、禁止访问不存在的全局变量

另外并没有限制：

1、修改已存在的全局变量

redis 参考了以下代码：

http://metalua.luaforge.net/src/lib/strict.lua.html