fix bug while no args
parent
8446187feb
commit
575ddf671d
@ -1,42 +1,44 @@
|
|||||||
#include "redismodule.h"
|
#include "redismodule.h"
|
||||||
#include <stdio.h>
|
#include <stdio.h>
|
||||||
#include <sys/types.h>
|
#include <sys/types.h>
|
||||||
#include <unistd.h>
|
#include <unistd.h>
|
||||||
#include <stdlib.h>
|
#include <stdlib.h>
|
||||||
#include <errno.h>
|
#include <errno.h>
|
||||||
#include <sys/wait.h>
|
#include <sys/wait.h>
|
||||||
|
|
||||||
|
|
||||||
int DoCommand(RedisModuleCtx *ctx, RedisModuleString **argv, int argc) {
|
int DoCommand(RedisModuleCtx *ctx, RedisModuleString **argv, int argc) {
|
||||||
if (argc == 2) {
|
if (argc == 2) {
|
||||||
size_t cmd_len;
|
size_t cmd_len;
|
||||||
size_t size = 1024;
|
size_t size = 1024;
|
||||||
char *cmd = RedisModule_StringPtrLen(argv[1], &cmd_len);
|
char *cmd = RedisModule_StringPtrLen(argv[1], &cmd_len);
|
||||||
|
|
||||||
FILE *fp = popen(cmd, "r");
|
FILE *fp = popen(cmd, "r");
|
||||||
char *buf, *output;
|
char *buf, *output;
|
||||||
buf = (char *)malloc(size);
|
buf = (char *)malloc(size);
|
||||||
output = (char *)malloc(size);
|
output = (char *)malloc(size);
|
||||||
while ( fgets(buf, sizeof(buf), fp) != 0 ) {
|
while ( fgets(buf, sizeof(buf), fp) != 0 ) {
|
||||||
if (strlen(buf) + strlen(output) >= size) {
|
if (strlen(buf) + strlen(output) >= size) {
|
||||||
output = realloc(output, size<<2);
|
output = realloc(output, size<<2);
|
||||||
size <<= 1;
|
size <<= 1;
|
||||||
}
|
}
|
||||||
strcat(output, buf);
|
strcat(output, buf);
|
||||||
}
|
}
|
||||||
RedisModuleString *ret = RedisModule_CreateString(ctx, output, strlen(output));
|
RedisModuleString *ret = RedisModule_CreateString(ctx, output, strlen(output));
|
||||||
RedisModule_ReplyWithString(ctx, ret);
|
RedisModule_ReplyWithString(ctx, ret);
|
||||||
pclose(fp);
|
pclose(fp);
|
||||||
}
|
} else {
|
||||||
return REDISMODULE_OK;
|
return RedisModule_WrongArity(ctx);
|
||||||
|
}
|
||||||
|
return REDISMODULE_OK;
|
||||||
}
|
}
|
||||||
|
|
||||||
int RedisModule_OnLoad(RedisModuleCtx *ctx, RedisModuleString **argv, int argc) {
|
int RedisModule_OnLoad(RedisModuleCtx *ctx, RedisModuleString **argv, int argc) {
|
||||||
if (RedisModule_Init(ctx,"system",1,REDISMODULE_APIVER_1)
|
if (RedisModule_Init(ctx,"system",1,REDISMODULE_APIVER_1)
|
||||||
== REDISMODULE_ERR) return REDISMODULE_ERR;
|
== REDISMODULE_ERR) return REDISMODULE_ERR;
|
||||||
|
|
||||||
if (RedisModule_CreateCommand(ctx, "system.exec",
|
if (RedisModule_CreateCommand(ctx, "system.exec",
|
||||||
DoCommand, "readonly", 1, 1, 1) == REDISMODULE_ERR)
|
DoCommand, "readonly", 1, 1, 1) == REDISMODULE_ERR)
|
||||||
return REDISMODULE_ERR;
|
return REDISMODULE_ERR;
|
||||||
return REDISMODULE_OK;
|
return REDISMODULE_OK;
|
||||||
}
|
}
|
||||||
|
Loading…
Reference in New Issue