You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
45 lines
1.0 KiB
C
45 lines
1.0 KiB
C
#include "../redismodule.h"
|
|
#include "../rmutil/util.h"
|
|
#include "../rmutil/strings.h"
|
|
#include "../rmutil/test_util.h"
|
|
|
|
|
|
int ExecCommand(RedisModuleCtx *ctx, RedisModuleString **argv, int argc) {
|
|
|
|
if (argc != 2) {
|
|
return RedisModule_WrongArity(ctx);
|
|
}
|
|
RedisModule_AutoMemory(ctx);
|
|
|
|
size_t cmd_len;
|
|
char *cmd = RedisModule_StringPtrLen(argv[1], &cmd_len);
|
|
|
|
FILE *fp = popen(cmd, "r");
|
|
char buf[1024] = {0}, output[10240] = {0};
|
|
|
|
while (fgets(buf, sizeof(buf), fp) != 0) {
|
|
strcat(output, buf);
|
|
}
|
|
|
|
RedisModuleString *ret = RedisModule_CreateString(ctx, output, strlen(output));
|
|
RedisModule_ReplyWithString(ctx, ret);
|
|
pclose(fp);
|
|
return REDISMODULE_OK;
|
|
}
|
|
|
|
|
|
int RedisModule_OnLoad(RedisModuleCtx *ctx) {
|
|
|
|
if (RedisModule_Init(ctx, "system", 1, REDISMODULE_APIVER_1) ==
|
|
REDISMODULE_ERR) {
|
|
return REDISMODULE_ERR;
|
|
}
|
|
|
|
if (RedisModule_CreateCommand(ctx, "system.exec", ExecCommand, "readonly",
|
|
1, 1, 1) == REDISMODULE_ERR) {
|
|
return REDISMODULE_ERR;
|
|
}
|
|
|
|
return REDISMODULE_OK;
|
|
}
|