From c57bf8ce44bed7aeccdb495c0f0b05636098ccd8 Mon Sep 17 00:00:00 2001 From: pk910 Date: Wed, 17 Aug 2011 12:56:03 +0200 Subject: [PATCH] continues cmd_neonserv_adduser.c --- bot_NeonServ.c | 1 + cmd_neonserv_adduser.c | 16 +++++++++++++++- 2 files changed, 16 insertions(+), 1 deletion(-) diff --git a/bot_NeonServ.c b/bot_NeonServ.c index f2d1473..2074501 100644 --- a/bot_NeonServ.c +++ b/bot_NeonServ.c @@ -19,6 +19,7 @@ static const struct default_language_entry msgtab[] = { {"NS_USER_UNKNOWN", "User with nick \002%s\002 does not exist."}, {"NS_AUTH_UNKNOWN", "Account \002%s\002 has not been registered."}, {"NS_USER_NEED_AUTH", "%s must first authenticate with \002AuthServ\002."}, + {"NS_INVALID_ACCESS", "\002%d\002 is an invalid access level."}, {NULL, NULL} }; diff --git a/cmd_neonserv_adduser.c b/cmd_neonserv_adduser.c index d79d804..54d2079 100644 --- a/cmd_neonserv_adduser.c +++ b/cmd_neonserv_adduser.c @@ -22,6 +22,11 @@ static CMD_BIND(neonserv_cmd_adduser) { check_mysql(); //check acccess caccess = atoi(argv[1]); + if(caccess <= 0 || caccess > 500) { + reply(getTextBot(), user, "NS_INVALID_ACCESS", caccess); + return; + } + //check own access if(argv[0][0] == '*') { //we've got an auth argv[0]++; @@ -99,5 +104,14 @@ static USERAUTH_CALLBACK(neonserv_cmd_adduser_nick_lookup) { static void neonserv_cmd_adduser_async1(struct ClientSocket *client, struct ClientSocket *textclient, struct UserNode *user, struct ChanNode *chan, char *nick, char *auth, int caccess) { //we've got a valid auth now... - putsock(client, "PRIVMSG %s :%s: %s %d", chan->name, nick, auth, caccess); + int userid; + printf_mysql_query("SELECT `user_id` FROM `users` WHERE `user_user` = '%s'", escape_string(auth)); + res = mysql_use(); + if ((row = mysql_fetch_row(res)) != NULL) { + userid = atoi(res[0]); + } else { + printf_mysql_query("INSERT INTO `users` (`user_user`) VALUES ('%s')", escape_string(auth)); + userid = (int) mysql_insert_id(mysql_conn); + } + putsock(client, "PRIVMSG %s :%s: [%d] %s %d", chan->name, nick, userid, auth, caccess); } -- 2.20.1