-/* cmd_neonserv_extscript.c - NeonServ v5.3
- * Copyright (C) 2011-2012 Philipp Kreil (pk910)
- *
- * This program is free software: you can redistribute it and/or modify
- * it under the terms of the GNU General Public License as published by
- * the Free Software Foundation, either version 3 of the License, or
- * (at your option) any later version.
- *
- * This program is distributed in the hope that it will be useful,
- * but WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
- * GNU General Public License for more details.
- *
- * You should have received a copy of the GNU General Public License
- * along with this program. If not, see <http://www.gnu.org/licenses/>.
- */
-
-#include "cmd_neonserv.h"
-#include <fcntl.h>
-
-/*
-* argv[0] 'toys' if it's a toy command (check if toys are enabled)
-* argv[0-*] script name & parameter patterns
-* argv[argc-1] all arguments passed to the command
-*/
-
-static TIMEQ_CALLBACK(neonserv_cmd_extscript_callback);
-
-struct neonserv_cmd_extscript_cache {
- struct ClientSocket *client, *textclient;
- struct Event *event;
- struct UserNode *user;
- struct ChanNode *chan;
- int answere_channel;
- FILE *pipe;
-};
-
-CMD_BIND(neonserv_cmd_extscript) {
- int i, j;
- char *args[MAXNUMPARAMS];
- int argpos = 0;
- char *next, *curr;
- char command[1024];
- int commandpos = 0;
- char part[MAXLEN];
- int partpos;
- int escape_param;
- int answere_channel = 0;
- //check first arg
- if(argc && !stricmp(argv[0], "toys")) {
- if(!chan) return; //toys are not allowed via query
- MYSQL_RES *res;
- MYSQL_ROW row;
- printf_mysql_query("SELECT `channel_toys` FROM `channels` WHERE `channel_name` = '%s'", escape_string(chan->name));
- res = mysql_use();
- row = mysql_fetch_row(res);
- if(!row || !strcmp(row[0], "0")) {
- //disabled
- reply(getTextBot(), user, "NS_FUN_DISABLED", chan->name);
- return;
- } else if(!strcmp(row[0], "2"))
- answere_channel = 1;
- argc--;
- argv++;
- }
- //parse arguments
- if(argc < 2) return;
- curr = argv[argc-1];
- while(curr) {
- next = strstr(curr, " ");
- args[argpos++] = curr;
- if(next) {
- *next = '\0';
- curr = next+1;
- } else
- curr = NULL;
- }
- //parse command pattern and build command
- commandpos = sprintf(command, "%s", argv[0]);
- for(i = 1; i < argc-1; i++) {
- partpos = 0;
- escape_param = 1;
- if(argv[i][0] == '$') {
- argv[i]++;
- if(argv[i][strlen(argv[i])-1] == '-') {
- argv[i][strlen(argv[i])-1] = '\0';
- j = atoi(argv[i]);
- if(j <= argpos)
- partpos = sprintf(part, "%s", merge_argv(args, j-1, argpos));
- } else if((j = atoi(argv[i])) > 0) {
- if(j <= argpos)
- partpos = sprintf(part, "%s", args[j-1]);
- } else if(!strcmp(argv[i], "c")) {
- partpos = sprintf(part, "%s", (chan ? chan->name : ""));
- } else if(!strcmp(argv[i], "n")) {
- partpos = sprintf(part, "%s", user->nick);
- } else if(!strcmp(argv[i], "a")) {
- partpos = sprintf(part, "%s", ((user->flags & USERFLAG_ISAUTHED) ? user->auth : ""));
- } else if(!strcmp(argv[i], "access")) {
- if(chan)
- partpos = sprintf(part, "%d", getChannelAccess(user, chan));
- }
- } else {
- partpos = sprintf(part, "%s", argv[i]);
- escape_param = 0;
- }
- //escape shell argument
- command[commandpos++] = ' ';
- if(escape_param) {
- command[commandpos++] = '\'';
- for(j = 0; j < partpos; j++) {
- if(part[j] == '\'') {
- command[commandpos++] = '\'';
- command[commandpos++] = '\\';
- command[commandpos++] = '\'';
- command[commandpos++] = '\'';
- } else
- command[commandpos++] = part[j];
- }
- command[commandpos++] = '\'';
- } else
- commandpos += sprintf(command + commandpos, " %s", part);
- }
- command[commandpos] = '\0';
- //we should now have a valid command
-
- struct neonserv_cmd_extscript_cache *cache = malloc(sizeof(*cache));
- if (!cache) {
- perror("malloc() failed");
- return;
- }
- cache->client = client;
- cache->textclient = getTextBot();
- cache->event = event;
- cache->user = user;
- cache->chan = chan;
- cache->answere_channel = answere_channel;
- cache->pipe = popen(command, "r");
- #ifndef WIN32
- fcntl(fileno(cache->pipe), F_SETFL, O_NONBLOCK);
- #endif
- timeq_uadd(200, module_id, neonserv_cmd_extscript_callback, cache);
-}
-
-static TIMEQ_CALLBACK(neonserv_cmd_extscript_callback) {
- struct neonserv_cmd_extscript_cache *cache = data;
- char command[512];
- char *a;
- if(feof(cache->pipe)) {
- pclose(cache->pipe);
- free(cache);
- return;
- }
- while (fgets(command, 512, cache->pipe) != NULL) {
- if((a = strchr(command, '\n')))
- *a = '\0';
- if(!stricmp(command, "/log")) {
- logEvent(cache->event);
- continue;
- }
- if(cache->answere_channel)
- putsock(cache->client, "PRIVMSG %s :%s", cache->chan->name, command);
- else
- reply(cache->textclient, cache->user, "%s", command);
- }
- timeq_uadd(200, module_id, neonserv_cmd_extscript_callback, cache);
-}
-
-
-
projects / NeonServV5.git / commitdiff