#include "modcmd.h"
#include "IRCEvents.h"
+#include "IRCParser.h"
#include "ClientSocket.h"
#include "UserNode.h"
#include "ChanNode.h"
#include "ChanUser.h"
+#include "WHOHandler.h"
+#include "lang.h"
+#include "mysqlConn.h"
+#include "DBHelper.h"
struct trigger_callback {
int botid;
trigger_callback_t *func;
struct trigger_callback *next;
-}
+};
+
+struct command_check_user_cache {
+ struct ClientSocket *client, *textclient;
+ struct UserNode *user;
+ struct ChanNode *chan, *sent_chan;
+ char **argv;
+ int argc;
+ char *message;
+ struct cmd_binding *cbind;
+};
static struct cmd_binding **cmd_binds;
static struct cmd_function *cmd_functions = NULL;
static struct trigger_callback *trigger_callbacks = NULL;
+static struct ClientSocket *tmp_text_client;
+
+static const struct default_language_entry msgtab[] = {
+ {"MODCMD_LESS_PARAM_COUNT", "This command requires more parameters."},
+ {"MODCMD_CHAN_REQUIRED", "You must provide the name of a channel that exists."},
+ {"MODCMD_AUTH_REQUIRED", "You need to be authenticated with AuthServ to use this command."},
+ {"MODCMD_PRIVILEGED", "\002%s\002 is a privileged command."},
+ {"MODCMD_PUBCMD", "Public commands in \002%s\002 are restricted."},
+ {"MODCMD_ACCESS_DENIED", "Access denied."},
+ {NULL, NULL}
+};
static int get_binds_index(char first_char) {
if(tolower(first_char) >= 'a' && tolower(first_char) <= 'z') {
return 26;
}
-struct* ClientSocket get_prefered_bot(int botid) {
- struct ClientSocket *client, *source = NULL;
+struct ClientSocket* get_prefered_bot(int botid) {
+ struct ClientSocket *client;
for(client = getBots(SOCKET_FLAG_READY, NULL); client; client = getBots(SOCKET_FLAG_READY, client)) {
- if(client->botid == botid && (client->flags & SOCKET_FLAG_PREFERED))
+ if(client->botid == botid && (client->flags & SOCKET_FLAG_PREFERRED))
return client;
}
return NULL;
return trigger->trigger;
}
+static void handle_command_async(struct ClientSocket *client, struct UserNode *user, struct ChanNode *chan, struct ChanNode *sent_chan struct cmd_binding *cbind, char **argv, int argc);
+
+static USERAUTH_CALLBACK(command_checked_auth) {
+ struct command_check_user_cache *cache = data;
+ tmp_text_client = cache->textclient;
+ handle_command_async(cache->client, user, cache->chan, cache->sent_chan, cache->cbind, cache->argv, cache->argc)
+ free(cache->message);
+ free(cache);
+}
+
static void handle_command(struct ClientSocket *client, struct UserNode *user, struct ChanNode *chan, char *message) {
- if(!strcmp(message, "users")) {
- struct ChanUser *chanuser;
- putsock(client, "PRIVMSG %s :[BOT JOIN] Users on this Channel:", chan->name);
- for(chanuser = getChannelUsers(chan, NULL); chanuser; chanuser = getChannelUsers(chan, chanuser)) {
- putsock(client, "PRIVMSG %s : %s!%s@%s [%s] rights: %d", chan->name, chanuser->user->nick, chanuser->user->ident, chanuser->user->host, ((chanuser->user->flags & USERFLAG_ISAUTHED) ? chanuser->user->auth : "*"), chanuser->flags);
+ struct ChanNode *sent_chan = chan;
+ if(message[0] == '#') {
+ char *chanName = message;
+ message = strstr(message, " ");
+ if(!message) return;
+ *message = '\0';
+ message++;
+ struct ChanNode *chan2 = getChanByName(chanName);
+ if(chan2)
+ chan = chan2;
+ }
+ message = strdup(message);
+ int bind_index = get_binds_index(message[0]);
+ char *args = strstr(message, " ");
+ if(args) {
+ *args = '\0';
+ args++;
+ }
+ struct cmd_binding *cbind;
+ for(cbind = cmd_binds[bind_index]; cbind; cbind = cbind->next) {
+ if(cbind->botid == client->botid && strcmp(cbind->cmd, message) == 0) {
+ //get a text bot
+ tmp_text_client = get_prefered_bot(client->botid);
+ //parse the arguments...
+ char *arga[MAXNUMPARAMS];
+ char **argv;
+ int argc = 0;
+ if(args) {
+ while(*args) {
+ //skip leading spaces
+ while (*args == ' ')
+ *args++ = 0;
+ arga[argc++] = args;
+ if (argc >= MAXNUMPARAMS)
+ break;
+ while (*args != ' ' && *args)
+ args++;
+ }
+ }
+ argv = arga;
+ if(argc != 0 && argv[0][0] == '#') {
+ struct ChanNode *chan2 = getChanByName(argv[0]);
+ if(chan2) {
+ argv += 1;
+ argc -= 1;
+ chan = chan2;
+ }
+ }
+ if(cbind->parameters) {
+ //userdefined parameters...
+ char *uarga[MAXNUMPARAMS];
+ char params[strlen(cbind->parameters)+1];
+ strcpy(params, cbind->parameters);
+ int uargpos = 0, argi, allargs = 0;
+ char *ppos = params;
+ char *prev_ppos = params;
+ while((ppos = strstr(ppos, " "))) {
+ *ppos = '\0';
+ if(prev_ppos[0] == '%') {
+ prev_ppos++;
+ if(prev_ppos[strlen(prev_ppos)-1] == '-') {
+ allargs = 1;
+ prev_ppos[strlen(prev_ppos)-1] = '\0';
+ } else
+ allargs = 0;
+ argi = atoi(prev_ppos);
+ if(argi > 0) {
+ if(argi <= argc) continue;
+ uarga[uargpos++] = argv[argi-1];
+ if(allargs) {
+ for(;argi < argc; argi++)
+ uarga[uargpos++] = argv[argi-1];
+ }
+ } else if(!strcmp(prev_ppos, "c"))
+ uarga[uargpos++] = (chan ? chan->name : NULL);
+ else if(!strcmp(prev_ppos, "n"))
+ uarga[uargpos++] = user->nick;
+ } else {
+ uarga[uargpos++] = prev_ppos;
+ }
+ ppos++;
+ prev_ppos = ppos;
+ }
+ argv = uarga;
+ argc = uargpos;
+ }
+ if(argc < cbind->func->paramcount) {
+ reply(tmp_text_client, user, "MODCMD_LESS_PARAM_COUNT");
+ break;
+ }
+ if((cbind->func->flags & CMDFLAG_REQUIRE_CHAN) && !chan) {
+ reply(tmp_text_client, user, "MODCMD_CHAN_REQUIRED");
+ break;
+ }
+ if((cbind->func->flags & CMDFLAG_CHECK_AUTH) && !(user->flags & USERFLAG_ISAUTHED)) {
+ //check auth...
+ struct command_check_user_cache *data = malloc(sizeof(*data));
+ char **temp_argv = malloc(argc*sizeof(*temp_argv));
+ if (!data || !temp_argv) {
+ perror("malloc() failed");
+ break;
+ }
+ memcpy(temp_argv, argv, argc*sizeof(*temp_argv));
+ data->argv = temp_argv;
+ data->argc = argc;
+ data->client = client;
+ data->user = user;
+ data->chan = chan;
+ data->sent_chan = sent_chan;
+ data->message = message;
+ data->cbind = cbind;
+ data->textclient = tmp_text_client;
+ get_userauth(user, command_checked_auth, data);
+ return;
+ } else
+ handle_command_async(client, user, chan, sent_chan, cbind, argv, argc);
+ break;
+ }
+ }
+ free(message);
+}
+
+static void handle_command_async(struct ClientSocket *client, struct UserNode *user, struct ChanNode *chan, struct ChanNode *sent_chan, struct cmd_binding *cbind, char **argv, int argc) {
+ MYSQL_RES *res;
+ MYSQL_ROW row;
+ int uaccess;
+ if((cbind->func->flags & CMDFLAG_REQUIRE_AUTH) && !(user->flags & USERFLAG_ISAUTHED)) {
+ reply(tmp_text_client, user, "MODCMD_AUTH_REQUIRED");
+ return;
+ }
+ if(sent_chan && sent_chan != chan) {
+ //check pubcmd of this channel
+ printf_mysql_query("SELECT `channel_pubcmd` FROM `channels` WHERE `channel_name` = '%s'", escape_string(sent_chan->name));
+ res = mysql_use();
+ if ((row = mysql_fetch_row(res)) != NULL) {
+ uaccess = getChannelAccess(user, sent_chan, 1);
+ if(row[0] && uaccess < atoi(row[0])) { //NOTE: HARDCODED DEFAULT: pubcmd = 0
+ reply(tmp_text_client, user, "MODCMD_PUBCMD", sent_chan->name);
+ return;
+ }
+ }
+ }
+ int global_access = ((cbind->flags & CMDFLAG_OVERRIDE_GLOBAL_ACCESS) ? cbind->global_access : cbind->func->global_access);
+ if(global_access > 0) {
+ int user_global_access = 0;
+ printf_mysql_query("SELECT `user_access` FROM `users` WHERE `user_user` = '%s'", escape_string(user->auth));
+ res = mysql_use();
+ if ((row = mysql_fetch_row(res)) != NULL) {
+ user_global_access = atoi(row[0]);
+ }
+ if(user_global_access < global_access) {
+ if(!user_global_access)
+ reply(tmp_text_client, user, "MODCMD_PRIVILEGED", cbind->cmd);
+ else
+ reply(tmp_text_client, user, "MODCMD_ACCESS_DENIED");
+ return;
+ }
+ }
+ if((cbind->func->flags & CMDFLAG_REGISTERED_CHAN)) {
+ check_mysql();
+ MYSQL_ROW defaults = NULL;
+ char access_list[256];
+ int access_pos = 0;
+ int access_count = 0;
+ int minaccess = 0;
+ char *str_a, *str_b = cbind->func->channel_access;
+ if(cbind->flags & CMDFLAG_OVERRIDE_CHANNEL_ACCESS)
+ str_b = cbind->channel_access;
+ if(str_b) {
+ while((str_a = strstr(str_b, ","))) {
+ *str_a = '\0';
+ if(*str_b[0] == '#') {
+ access_pos += sprintf(access_list+access_pos, ", `%s`", str_b);
+ access_count++;
+ } else {
+ if(atoi(str_b) > minaccess)
+ minaccess = atoi(str_b);
+ }
+ *str_a = ',';
+ str_b = str_a+1;
+ }
+ access_pos += sprintf(access_list+access_pos, ", `%s`", str_b);
+ access_count++;
+ } else
+ access_list[0] = '\0';
+ if(!(chan->flags & CHANFLAG_REQUESTED_CHANINFO) || (sent_chan && sent_chan == chan) || access_count || minaccess) {
+ printf_mysql_query("SELECT `channel_id`, `channel_pubcmd` %s FROM `channels` WHERE `channel_name` = '%s'", access_list, escape_string(chan->name));
+ res = mysql_use();
+ if ((row = mysql_fetch_row(res)) != NULL) {
+ chan->flags |= CHANFLAG_CHAN_REGISTERED;
+ chan->channel_id = atoi(row[0]);
+ if((sent_chan && sent_chan == chan) || access_count || minaccess) {
+ uaccess = getChannelAccess(user, chan, 1);
+ if(uaccess < minaccess) {
+ //ACCESS DENIED
+ reply(tmp_text_client, user, "MODCMD_ACCESS_DENIED");
+ return;
+ }
+ if(!row[1] && !defaults) {
+ printf_mysql_query("SELECT `channel_id`, `channel_pubcmd` %s FROM `channels` WHERE `channel_name` = 'defaults'", access_list);
+ defaults = mysql_fetch_row(mysql_use());
+ }
+ if(sent_chan && (sent_chan == chan) && uaccess < (row[1] ? atoi(row[1]) : atoi(defaults[1]))) {
+ //PUBCMD
+ reply(tmp_text_client, user, "MODCMD_PUBCMD", chan->name);
+ return;
+ }
+ int i;
+ for(i = 0; i < access_count; i++) {
+ if(!row[2+i] && !defaults) {
+ printf_mysql_query("SELECT `channel_id`, `channel_pubcmd` %s FROM `channels` WHERE `channel_name` = 'defaults'", access_list);
+ defaults = mysql_fetch_row(mysql_use());
+ }
+ if(uaccess < (row[2+i] ? atoi(row[2+i]) : atoi(defaults[2+i]))) {
+ reply(tmp_text_client, user, "MODCMD_ACCESS_DENIED");
+ return;
+ }
+ }
+ }
+ }
+ chan->flags |= CHANFLAG_REQUESTED_CHANINFO;
+ }
+ if(!(chan->flags & CHANFLAG_CHAN_REGISTERED)) {
+ reply(tmp_text_client, user, "MODCMD_CHAN_REQUIRED");
+ return;
+ }
+ printf_mysql_query("SELECT `botid` FROM `bot_channels` LEFT JOIN `bots` ON `bot_channels`.`botid` = `bots`.`id` WHERE `chanid` = '%d' AND `botclass` = '%d'", chan->channel_id, client->botid);
+ res = mysql_use();
+ if ((row = mysql_fetch_row(res)) == NULL) {
+ reply(tmp_text_client, user, "MODCMD_CHAN_REQUIRED");
+ return;
}
}
- if(!strcmp(message, "modes")) {
- char modeBuf[MAXLEN];
- getModeString(chan, modeBuf);
- putsock(client, "PRIVMSG %s :Modes: %s", chan->name, modeBuf);
+ if((cbind->func->flags & CMDFLAG_REQUIRE_GOD) && !isGodMode(user)) {
+ reply(tmp_text_client, user, "MODCMD_PRIVILEGED", cbind->cmd);
+ return;
}
+ cbind->func->func(client, user, chan, argv, argc);
}
static void got_chanmsg(struct UserNode *user, struct ChanNode *chan, char *message) {
struct ClientSocket *client;
FD_ZERO(&fds);
for(client = getBots(SOCKET_FLAG_READY, NULL); client; client = getBots(SOCKET_FLAG_READY, client)) {
- if(isUserOnChan(client->user, chan) && (client->flags & SOCKET_FLAG_PREFERED) && !FD_ISSET(client->botid, &fds)) {
+ if(isUserOnChan(client->user, chan) && (client->flags & SOCKET_FLAG_PREFERRED) && !FD_ISSET(client->botid, &fds)) {
FD_SET(client->botid, &fds);
trigger = get_channel_trigger(client->botid, chan);
if(stricmplen(message, trigger, strlen(trigger)) == 0) {
}
}
-int register_command(int botid, char *name, cmd_bind_t *func) {
+int register_command(int botid, char *name, cmd_bind_t *func, int paramcount, unsigned int flags, char *channel_access, int global_access) {
struct cmd_function *cmdfunc;
for(cmdfunc = cmd_functions; cmdfunc; cmdfunc = cmdfunc->next) {
if(cmdfunc->botid == botid && strcmp(cmdfunc->name, name) == 0)
cmdfunc->botid = botid;
cmdfunc->name = strdup(name);
cmdfunc->func = func;
+ cmdfunc->flags = flags;
+ cmdfunc->paramcount = paramcount;
+ cmdfunc->channel_access = channel_access;
+ cmdfunc->global_access = global_access;
cmdfunc->next = cmd_functions;
cmd_functions = cmdfunc;
return 1;
cbind->botid = botid;
cbind->cmd = strdup(cmd);
cbind->func = func;
+ cbind->parameters = NULL;
+ cbind->global_access = 0;
+ cbind->channel_access = NULL;
+ cbind->flags = 0;
cbind->next = cmd_binds[bind_index];
cmd_binds[bind_index] = cbind;
return 1;
cbind->cmd = strdup(cmd);
cbind->func = cmdfunc;
cbind->next = cmd_binds[bind_index];
+ cbind->parameters = NULL;
+ cbind->global_access = 0;
+ cbind->channel_access = NULL;
+ cbind->flags = 0;
cmd_binds[bind_index] = cbind;
return 1;
}
else
cmd_binds[bind_index] = cbind->next;
free(cbind->cmd);
+ if(cbind->parameters)
+ free(cbind->parameters);
free(cbind);
return 1;
} else
return 0;
}
+struct ClientSocket *getTextBot() {
+ return tmp_text_client;
+}
+
void init_modcmd() {
cmd_binds = calloc(27, sizeof(*cmd_binds));
bind_chanmsg(got_chanmsg);
bind_privmsg(got_privmsg);
+ register_default_language_table(msgtab);
+}
+
+void free_modcmd() {
+ int i;
+ for(i = 0; i < 27; i++) {
+ struct cmd_binding *cbind, *next;
+ for(cbind = cmd_binds[i]; cbind; cbind = next) {
+ next = cbind->next;
+ free(cbind->cmd);
+ if(cbind->parameters)
+ free(cbind->parameters);
+ if(cbind->channel_access)
+ free(cbind->channel_access);
+ free(cbind);
+ }
+ }
+ free(cmd_binds);
+ struct cmd_function *cmdfunct, *next;
+ for(cmdfunct = cmd_functions; cmdfunct; cmdfunct = next) {
+ next = cmdfunct->next;
+ free(cmdfunct->name);
+ free(cmdfunct);
+ }
+ struct trigger_callback *cb, *next_cb;
+ for(cb = trigger_callbacks; cb; cb = next_cb) {
+ next_cb = cb->next;
+ free(next_cb);
+ }
+ cmd_functions = NULL;
+ trigger_callbacks = NULL;
+}
+
+void bind_set_parameters(int botid, char *cmd, char *parameters) {
+ int bind_index = get_binds_index(cmd[0]);
+ struct cmd_binding *cbind;
+ for(cbind = cmd_binds[bind_index]; cbind; cbind = cbind->next) {
+ if(cbind->botid == botid && strcmp(cbind->cmd, cmd) == 0) {
+ if(cbind->parameters)
+ free(cbind->parameters);
+ cbind->parameters = strdup(parameters);
+ return;
+ }
+ }
}
+void bind_set_global_access(int botid, char *cmd, int gaccess) {
+ int bind_index = get_binds_index(cmd[0]);
+ struct cmd_binding *cbind;
+ for(cbind = cmd_binds[bind_index]; cbind; cbind = cbind->next) {
+ if(cbind->botid == botid && strcmp(cbind->cmd, cmd) == 0) {
+ if(gaccess > -1) {
+ cbind->global_access = gaccess;
+ cbind->flags |= CMDFLAG_OVERRIDE_GLOBAL_ACCESS;
+ } else {
+ cbind->flags &= ~CMDFLAG_OVERRIDE_GLOBAL_ACCESS;
+ }
+ return;
+ }
+ }
+}
+
+void bind_set_channel_access(int botid, char *cmd, char *chanaccess) {
+ int bind_index = get_binds_index(cmd[0]);
+ struct cmd_binding *cbind;
+ for(cbind = cmd_binds[bind_index]; cbind; cbind = cbind->next) {
+ if(cbind->botid == botid && strcmp(cbind->cmd, cmd) == 0) {
+ if(cbind->channel_access)
+ free(cbind->channel_access);
+ if(chanaccess) {
+ cbind->channel_access = strdup(chanaccess);
+ cbind->flags |= CMDFLAG_OVERRIDE_CHANNEL_ACCESS;
+ } else {
+ cbind->channel_access = NULL;
+ cbind->flags &= ~CMDFLAG_OVERRIDE_CHANNEL_ACCESS;
+ }
+ return;
+ }
+ }
+}
projects / NeonServV5.git / blobdiff