1 /* +------------------------------------+
\r * | Inspire Internet Relay Chat Daemon |
\r * +------------------------------------+
\r *
\r * InspIRCd: (C) 2002-2007 InspIRCd Development Team
\r * See: http://www.inspircd.org/wiki/index.php/Credits
\r *
\r * This program is free but copyrighted software; see
\r * the file COPYING for details.
\r *
\r * ---------------------------------------------------
\r */
\r\r#include "inspircd.h"
\r#include "users.h"
\r#include "channels.h"
\r#include "modules.h"
\r#include "configreader.h"
\r#include "m_sqlv2.h"
\r\rstatic Module* SQLModule;
\rstatic Module* MyMod;
\rstatic std::string dbid;
\r\renum LogTypes { LT_OPER = 1, LT_KILL, LT_SERVLINK, LT_XLINE, LT_CONNECT, LT_DISCONNECT, LT_FLOOD, LT_LOADMODULE };
\r\renum QueryState { FIND_SOURCE, FIND_NICK, FIND_HOST, DONE};
\r\rclass QueryInfo;
\r\rstd::map<unsigned long,QueryInfo*> active_queries;
\r\rclass QueryInfo
\r{
\rpublic:
\r QueryState qs;
\r unsigned long id;
\r std::string nick;
\r std::string source;
\r std::string hostname;
\r int sourceid;
\r int nickid;
\r int hostid;
\r int category;
\r time_t date;
\r bool insert;
\r\r QueryInfo(const std::string &n, const std::string &s, const std::string &h, unsigned long i, int cat)
\r {
\r qs = FIND_SOURCE;
\r nick = n;
\r source = s;
\r hostname = h;
\r id = i;
\r category = cat;
\r sourceid = nickid = hostid = -1;
\r date = time(NULL);
\r insert = false;
\r }
\r\r void Go(SQLresult* res)
\r {
\r SQLrequest req = SQLreq(MyMod, SQLModule, dbid, "", "");
\r switch (qs)
\r {
\r case FIND_SOURCE:
\r if (res->Rows() && sourceid == -1 && !insert)
\r {
\r sourceid = atoi(res->GetValue(0,0).d.c_str());
\r req = SQLreq(MyMod, SQLModule, dbid, "SELECT id,actor FROM ircd_log_actors WHERE actor='?'", nick);
\r if(req.Send())
\r {
\r insert = false;
\r qs = FIND_NICK;
\r active_queries[req.id] = this;
\r }
\r }
\r else if (res->Rows() && sourceid == -1 && insert)
\r {
\r req = SQLreq(MyMod, SQLModule, dbid, "SELECT id,actor FROM ircd_log_actors WHERE actor='?'", source);
\r if(req.Send())
\r {
\r insert = false;
\r qs = FIND_SOURCE;
\r active_queries[req.id] = this;
\r }
\r }
\r else
\r {
\r req = SQLreq(MyMod, SQLModule, dbid, "INSERT INTO ircd_log_actors (actor) VALUES('?')", source);
\r if(req.Send())
\r {
\r insert = true;
\r qs = FIND_SOURCE;
\r active_queries[req.id] = this;
\r }
\r }
\r break;
\r\r case FIND_NICK:
\r if (res->Rows() && nickid == -1 && !insert)
\r {
\r nickid = atoi(res->GetValue(0,0).d.c_str());
\r req = SQLreq(MyMod, SQLModule, dbid, "SELECT id,hostname FROM ircd_log_hosts WHERE hostname='?'", hostname);
\r if(req.Send())
\r {
\r insert = false;
\r qs = FIND_HOST;
\r active_queries[req.id] = this;
\r }
\r }
\r else if (res->Rows() && nickid == -1 && insert)
\r {
\r req = SQLreq(MyMod, SQLModule, dbid, "SELECT id,actor FROM ircd_log_actors WHERE actor='?'", nick);
\r if(req.Send())
\r {
\r insert = false;
\r qs = FIND_NICK;
\r active_queries[req.id] = this;
\r }
\r }
\r else
\r {
\r req = SQLreq(MyMod, SQLModule, dbid, "INSERT INTO ircd_log_actors (actor) VALUES('?')",nick);
\r if(req.Send())
\r {
\r insert = true;
\r qs = FIND_NICK;
\r active_queries[req.id] = this;
\r }
\r }
\r break;
\r\r case FIND_HOST:
\r if (res->Rows() && hostid == -1 && !insert)
\r {
\r hostid = atoi(res->GetValue(0,0).d.c_str());
\r req = SQLreq(MyMod, SQLModule, dbid, "INSERT INTO ircd_log (category_id,nick,host,source,dtime) VALUES("+ConvToStr(category)+","+ConvToStr(nickid)+","+ConvToStr(hostid)+","+ConvToStr(sourceid)+","+ConvToStr(date)+")");
\r if(req.Send())
\r {
\r insert = true;
\r qs = DONE;
\r active_queries[req.id] = this;
\r }
\r }
\r else if (res->Rows() && hostid == -1 && insert)
\r {
\r req = SQLreq(MyMod, SQLModule, dbid, "SELECT id,hostname FROM ircd_log_hosts WHERE hostname='?'", hostname);
\r if(req.Send())
\r {
\r insert = false;
\r qs = FIND_HOST;
\r active_queries[req.id] = this;
\r }
\r }
\r else
\r {
\r req = SQLreq(MyMod, SQLModule, dbid, "INSERT INTO ircd_log_hosts (hostname) VALUES('?')", hostname);
\r if(req.Send())
\r {
\r insert = true;
\r qs = FIND_HOST;
\r active_queries[req.id] = this;
\r }
\r }
\r break;
\r\r case DONE:
\r delete active_queries[req.id];
\r active_queries[req.id] = NULL;
\r break;
\r }
\r }
\r};
\r\r/* $ModDesc: Logs network-wide data to an SQL database */
\r\rclass ModuleSQLLog : public Module
\r{
\r ConfigReader* Conf;
\r\r public:
\r ModuleSQLLog(InspIRCd* Me)
\r : Module::Module(Me)
\r {
\r ServerInstance->UseInterface("SQLutils");
\r ServerInstance->UseInterface("SQL");
\r\r Module* SQLutils = ServerInstance->FindModule("m_sqlutils.so");
\r if (!SQLutils)
\r throw ModuleException("Can't find m_sqlutils.so. Please load m_sqlutils.so before m_sqlauth.so.");
\r\r SQLModule = ServerInstance->FindFeature("SQL");
\r\r OnRehash(NULL,"");
\r MyMod = this;
\r active_queries.clear();
\r }
\r\r virtual ~ModuleSQLLog()
\r {
\r ServerInstance->DoneWithInterface("SQL");
\r ServerInstance->DoneWithInterface("SQLutils");
\r }
\r\r void Implements(char* List)
\r {
\r List[I_OnRehash] = List[I_OnOper] = List[I_OnGlobalOper] = List[I_OnKill] = 1;
\r List[I_OnPreCommand] = List[I_OnUserConnect] = 1;
\r List[I_OnUserQuit] = List[I_OnLoadModule] = List[I_OnRequest] = 1;
\r }
\r\r void ReadConfig()
\r {
\r ConfigReader Conf(ServerInstance);
\r dbid = Conf.ReadValue("sqllog","dbid",0); // database id of a database configured in sql module
\r }
\r\r virtual void OnRehash(userrec* user, const std::string ¶meter)
\r {
\r ReadConfig();
\r }
\r\r virtual char* OnRequest(Request* request)
\r {
\r if(strcmp(SQLRESID, request->GetId()) == 0)
\r {
\r SQLresult* res;
\r std::map<unsigned long, QueryInfo*>::iterator n;
\r\r res = static_cast<SQLresult*>(request);
\r n = active_queries.find(res->id);
\r\r if (n != active_queries.end())
\r {
\r n->second->Go(res);
\r std::map<unsigned long, QueryInfo*>::iterator n = active_queries.find(res->id);
\r active_queries.erase(n);
\r }
\r\r return SQLSUCCESS;
\r }
\r\r return NULL;
\r }
\r\r void AddLogEntry(int category, const std::string &nick, const std::string &host, const std::string &source)
\r {
\r // is the sql module loaded? If not, we don't attempt to do anything.
\r if (!SQLModule)
\r return;
\r\r SQLrequest req = SQLreq(this, SQLModule, dbid, "SELECT id,actor FROM ircd_log_actors WHERE actor='?'", source);
\r if(req.Send())
\r {
\r QueryInfo* i = new QueryInfo(nick, source, host, req.id, category);
\r i->qs = FIND_SOURCE;
\r active_queries[req.id] = i;
\r }
\r }
\r\r virtual void OnOper(userrec* user, const std::string &opertype)
\r {
\r AddLogEntry(LT_OPER,user->nick,user->host,user->server);
\r }
\r\r virtual void OnGlobalOper(userrec* user)
\r {
\r AddLogEntry(LT_OPER,user->nick,user->host,user->server);
\r }
\r\r virtual int OnKill(userrec* source, userrec* dest, const std::string &reason)
\r {
\r AddLogEntry(LT_KILL,dest->nick,dest->host,source->nick);
\r return 0;
\r }
\r\r virtual int OnPreCommand(const std::string &command, const char** parameters, int pcnt, userrec *user, bool validated, const std::string &original_line)
\r {
\r if ((command == "GLINE" || command == "KLINE" || command == "ELINE" || command == "ZLINE") && validated)
\r {
\r AddLogEntry(LT_XLINE,user->nick,command[0]+std::string(":")+std::string(parameters[0]),user->server);
\r }
\r return 0;
\r }
\r\r virtual void OnUserConnect(userrec* user)
\r {
\r AddLogEntry(LT_CONNECT,user->nick,user->host,user->server);
\r }
\r\r virtual void OnUserQuit(userrec* user, const std::string &reason, const std::string &oper_message)
\r {
\r AddLogEntry(LT_DISCONNECT,user->nick,user->host,user->server);
\r }
\r\r virtual void OnLoadModule(Module* mod, const std::string &name)
\r {
\r AddLogEntry(LT_LOADMODULE,name,ServerInstance->Config->ServerName, ServerInstance->Config->ServerName);
\r }
\r\r virtual Version GetVersion()
\r {
\r return Version(1,1,0,1,VF_VENDOR,API_VERSION);
\r }
\r \r};
\r\rMODULE_INIT(ModuleSQLLog);
\r\r