fluencelabs/redis
1
0
Fork 0
mirror of https://github.com/fluencelabs/redis synced 2025-06-16 02:31:20 +00:00
Code Issues Projects Releases Wiki Activity

Deny commands flagged as REDIS_CMD_NOSCRIPT from Lua scripts

Browse Source
This commit is contained in:
antirez
2011-09-27 13:57:10 +02:00
parent b60ed6e812
commit 15ef605340
2 changed files with 13 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
src/scripting.c
View File

@ -158,24 +158,29 @@ int luaRedisCommand(lua_State *lua) {
return 1;
}
/* Setup our fake client for command execution */
c->argv = argv;
c->argc = argc;
/* Command lookup */
cmd = lookupCommand(argv[0]->ptr);
if (!cmd || ((cmd->arity > 0 && cmd->arity != argc) ||
(argc < -cmd->arity)))
{
for (j = 0; j < argc; j++) decrRefCount(argv[j]);
zfree(argv);
if (cmd)
luaPushError(lua,
"Wrong number of args calling Redis command From Lua script");
else
luaPushError(lua,"Unknown Redis command called from Lua script");
return 1;
goto cleanup;
}
/* Run the command in the context of a fake client */
c->argv = argv;
c->argc = argc;
if (cmd->flags & REDIS_CMD_NOSCRIPT) {
luaPushError(lua, "This Redis command is not allowed from scripts");
goto cleanup;
}
/* Run the command */
cmd->proc(c);
/* Convert the result of the Redis command into a suitable Lua type.
@ -195,6 +200,7 @@ int luaRedisCommand(lua_State *lua) {
redisProtocolToLuaType(lua,reply);
sdsfree(reply);
cleanup:
/* Clean up. Command code may have changed argv/argc so we use the
* argv/argc of the client instead of the local variables. */
for (j = 0; j < c->argc; j++)