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