Stops SPLICE from accepting negative counts

This commit is contained in:
Itamar Haber 2016-04-09 13:25:50 -07:00 committed by antirez
parent 2e464bf0b1
commit 6054089fa8

View File

@ -97,7 +97,8 @@ int HelloListSplice_RedisCommand(RedisModuleCtx *ctx, RedisModuleString **argv,
} }
long long count; long long count;
if (RedisModule_StringToLongLong(argv[3],&count) != REDISMODULE_OK) { if ((RedisModule_StringToLongLong(argv[3],&count) != REDISMODULE_OK) ||
(count < 0)) {
RedisModule_CloseKey(srckey); RedisModule_CloseKey(srckey);
RedisModule_CloseKey(dstkey); RedisModule_CloseKey(dstkey);
return RedisModule_ReplyWithError(ctx,"ERR invalid count"); return RedisModule_ReplyWithError(ctx,"ERR invalid count");
@ -141,8 +142,11 @@ int HelloListSpliceAuto_RedisCommand(RedisModuleCtx *ctx, RedisModuleString **ar
} }
long long count; long long count;
if (RedisModule_StringToLongLong(argv[3],&count) != REDISMODULE_OK) if ((RedisModule_StringToLongLong(argv[3],&count) != REDISMODULE_OK) ||
(count < 0))
{
return RedisModule_ReplyWithError(ctx,"ERR invalid count"); return RedisModule_ReplyWithError(ctx,"ERR invalid count");
}
while(count-- > 0) { while(count-- > 0) {
RedisModuleString *ele; RedisModuleString *ele;