1 /* +------------------------------------+
2 * | Inspire Internet Relay Chat Daemon |
3 * +------------------------------------+
5 * InspIRCd is copyright (C) 2002-2004 ChatSpike-Dev.
7 * <brain@chatspike.net>
8 * <Craig@chatspike.net>
10 * Written by Craig Edwards, Craig McLure, and others.
11 * This program is free but copyrighted software; see
12 * the file COPYING for details.
14 * ---------------------------------------------------
23 #include <sys/types.h>
24 #include <sys/socket.h>
34 #include "configreader.h"
39 static Module* SQLModule;
41 static std::string dbid;
44 extern InspIRCd* ServerInstance;
46 enum LogTypes { LT_OPER = 1, LT_KILL, LT_SERVLINK, LT_XLINE, LT_CONNECT, LT_DISCONNECT, LT_FLOOD, LT_LOADMODULE };
48 enum QueryState { FIND_SOURCE, INSERT_SOURCE, FIND_NICK, INSERT_NICK, FIND_HOST, INSERT_HOST, INSERT_LOGENTRY, DONE };
52 std::map<unsigned long,QueryInfo*> active_queries;
66 std::string lastquery;
68 QueryInfo(const std::string &n, const std::string &h, unsigned long i, int cat)
75 sourceid = nickid = hostid = -1;
80 void Go(SQLresult* res)
82 // Nothing here and not sent yet
83 SQLrequest req = SQLreq(MyMod, SQLModule, dbid, "", "");
85 log(DEBUG,"State: %d",qs);
90 // "SELECT id,actor FROM ircd_log_actors WHERE actor='"+nick+"'"
91 // If we find it, advance to FIND_NICK state, otherwise go to INSERT_SOURCE
96 sourceid = atoi(res->GetValue(0,0).d.c_str());
99 else qs = INSERT_SOURCE;
103 if (lastquery == "SELECT id,actor FROM ircd_log_actors WHERE actor='?'")
109 lastquery = "SELECT id,actor FROM ircd_log_actors WHERE actor='?'";
110 req = SQLreq(MyMod, SQLModule, dbid, "SELECT id,actor FROM ircd_log_actors WHERE actor='?'", nick);
114 active_queries[req.id] = this;
118 log(DEBUG, "SQLrequest failed: %s", req.error.Str());
124 // "INSERT INTO ircd_log_actors VALUES('','"+nick+"')")
125 // after we've done this, go back to FIND_SOURCE
126 lastquery = "INSERT INTO ircd_log_actors VALUES('','?')";
127 req = SQLreq(MyMod, SQLModule, dbid, "INSERT INTO ircd_log_actors VALUES('','?')", nick);
131 active_queries[req.id] = this;
135 log(DEBUG, "SQLrequest failed: %s", req.error.Str());
140 // "SELECT id,actor FROM ircd_log_actors WHERE actor='"+nick+"'"
141 // If we find it, advance to FIND_HOST state, otherwise go to INSERT_NICK
146 nickid = atoi(res->GetValue(0,0).d.c_str());
149 else qs = INSERT_NICK;
153 if (lastquery == "SELECT id,actor FROM ircd_log_actors WHERE actor='?'")
159 req = SQLreq(MyMod, SQLModule, dbid, "SELECT id,actor FROM ircd_log_actors WHERE actor='?'", nick);
163 active_queries[req.id] = this;
167 log(DEBUG, "SQLrequest failed: %s", req.error.Str());
173 // "INSERT INTO ircd_log_actors VALUES('','"+nick+"')")
174 // after we've done this, go back to FIND_NICK
175 req = SQLreq(MyMod, SQLModule, dbid, "INSERT INTO ircd_log_actors VALUES('','?')",nick);
179 active_queries[req.id] = this;
183 log(DEBUG, "SQLrequest failed: %s", req.error.Str());
187 // "SELECT id,hostname FROM ircd_log_hosts WHERE hostname='"+host+"'"
188 // If we find it, advance to INSERT_LOGENTRY state, otherwise go to INSERT_HOST
193 hostid = atoi(res->GetValue(0,0).d.c_str());
194 qs = INSERT_LOGENTRY;
196 else qs = INSERT_HOST;
200 if (lastquery == "SELECT id,hostname FROM ircd_log_hosts WHERE hostname='?'")
206 lastquery = "SELECT id,hostname FROM ircd_log_hosts WHERE hostname='?'";
207 req = SQLreq(MyMod, SQLModule, dbid, "SELECT id,hostname FROM ircd_log_hosts WHERE hostname='?'",hostname);
211 active_queries[req.id] = this;
215 log(DEBUG, "SQLrequest failed: %s", req.error.Str());
221 // "INSERT INTO ircd_log_hosts VALUES('','"+host+"')"
222 // after we've done this, go back to FIND_HOST
223 lastquery = "INSERT INTO ircd_log_hosts VALUES('','?')";
224 req = SQLreq(MyMod, SQLModule, dbid, "INSERT INTO ircd_log_hosts VALUES('','?')", hostname);
227 qs = INSERT_LOGENTRY;
228 active_queries[req.id] = this;
232 log(DEBUG, "SQLrequest failed: %s", req.error.Str());
235 case INSERT_LOGENTRY:
236 // INSERT INTO ircd_log VALUES('',%lu,%lu,%lu,%lu,%lu)",(unsigned long)category,(unsigned long)nickid,(unsigned long)hostid,(unsigned long)sourceid,(unsigned long)date
237 // aaand done! (discard result)
238 if ((nickid == -1) || (hostid == -1) || (sourceid == -1))
245 lastquery = "INSERT INTO ircd_log VALUES()";
246 req = SQLreq(MyMod, SQLModule, dbid, "INSERT INTO ircd_log VALUES('',"+ConvToStr(category)+","+ConvToStr(nickid)+","+ConvToStr(hostid)+","+ConvToStr(sourceid)+","+ConvToStr(date)+")");
258 log(DEBUG, "SQLrequest failed: %s", req.error.Str());
263 active_queries[req.id] = NULL;
269 /* $ModDesc: Logs network-wide data to an SQL database */
271 class ModuleSQLLog : public Module
279 Conf = new ConfigReader();
280 dbid = Conf->ReadValue("sqllog","dbid",0); // database id of a database configured in sql module
282 SQLModule = Srv->FindFeature("SQL");
284 Srv->Log(DEFAULT,"WARNING: m_sqllog.so could not initialize because an SQL module is not loaded. Load the module and rehash your server.");
288 ModuleSQLLog(Server* Me) : Module::Module(Me)
293 active_queries.clear();
296 void Implements(char* List)
298 List[I_OnRehash] = List[I_OnOper] = List[I_OnGlobalOper] = List[I_OnKill] = 1;
299 List[I_OnPreCommand] = List[I_OnUserConnect] = List[I_OnGlobalConnect] = 1;
300 List[I_OnUserQuit] = List[I_OnLoadModule] = List[I_OnRequest] = 1;
303 virtual void OnRehash(const std::string ¶meter)
308 virtual char* OnRequest(Request* request)
310 log(DEBUG,"OnRequest in m_sqllog.so");
311 if(strcmp(SQLRESID, request->GetId()) == 0)
314 std::map<unsigned long, QueryInfo*>::iterator n;
316 res = static_cast<SQLresult*>(request);
317 log(DEBUG, "Got SQL result (%s) with ID %lu", res->GetId(), res->id);
319 n = active_queries.find(res->id);
321 if (n != active_queries.end())
323 log(DEBUG,"This is an active query");
326 std::map<unsigned long, QueryInfo*>::iterator n = active_queries.find(res->id);
327 active_queries.erase(n);
333 void AddLogEntry(int category, const std::string &nick, const std::string &host, const std::string &source)
335 // is the sql module loaded? If not, we don't attempt to do anything.
339 SQLrequest req = SQLreq(this, SQLModule, dbid, "SELECT id,actor FROM ircd_log_actors WHERE actor='?'", nick);
342 QueryInfo* i = new QueryInfo(nick, host, req.id, category);
344 active_queries[req.id] = i;
345 log(DEBUG,"Active query id %d",req.id);
349 log(DEBUG, "SQLrequest failed: %s", req.error.Str());
352 /*long nickid = InsertNick(nick);
353 long sourceid = InsertNick(source);
354 long hostid = InsertHost(host);
355 InsertEntry((unsigned)category,(unsigned)nickid,(unsigned)hostid,(unsigned)sourceid,(unsigned long)time(NULL));*/
358 virtual void OnOper(userrec* user, const std::string &opertype)
360 AddLogEntry(LT_OPER,user->nick,user->host,user->server);
363 virtual void OnGlobalOper(userrec* user)
365 AddLogEntry(LT_OPER,user->nick,user->host,user->server);
368 virtual int OnKill(userrec* source, userrec* dest, const std::string &reason)
370 AddLogEntry(LT_KILL,dest->nick,dest->host,source->nick);
374 virtual int OnPreCommand(const std::string &command, const char** parameters, int pcnt, userrec *user, bool validated)
376 if ((command == "GLINE") || (command == "KLINE") || (command == "ELINE") || (command == "ZLINE"))
378 AddLogEntry(LT_XLINE,user->nick,command[0]+std::string(":")+std::string(parameters[0]),user->server);
383 virtual void OnUserConnect(userrec* user)
385 AddLogEntry(LT_CONNECT,user->nick,user->host,user->server);
388 virtual void OnGlobalConnect(userrec* user)
390 AddLogEntry(LT_CONNECT,user->nick,user->host,user->server);
393 virtual void OnUserQuit(userrec* user, const std::string &reason)
395 AddLogEntry(LT_DISCONNECT,user->nick,user->host,user->server);
398 virtual void OnLoadModule(Module* mod, const std::string &name)
400 AddLogEntry(LT_LOADMODULE,name,ServerInstance->Config->ServerName,ServerInstance->Config->ServerName);
403 virtual ~ModuleSQLLog()
407 virtual Version GetVersion()
409 return Version(1,0,0,1,VF_VENDOR);
414 class ModuleSQLLogFactory : public ModuleFactory
417 ModuleSQLLogFactory()
421 ~ModuleSQLLogFactory()
425 virtual Module * CreateModule(Server* Me)
427 return new ModuleSQLLog(Me);
433 extern "C" void * init_module( void )
435 return new ModuleSQLLogFactory;