diff options
| -rw-r--r-- | src/modules/extra/m_sqllog.cpp | 231 |
1 files changed, 231 insertions, 0 deletions
diff --git a/src/modules/extra/m_sqllog.cpp b/src/modules/extra/m_sqllog.cpp new file mode 100644 index 000000000..54fd41798 --- /dev/null +++ b/src/modules/extra/m_sqllog.cpp @@ -0,0 +1,231 @@ +/* +------------------------------------+ + * | Inspire Internet Relay Chat Daemon | + * +------------------------------------+ + * + * Inspire is copyright (C) 2002-2004 ChatSpike-Dev. + * E-mail: + * <brain@chatspike.net> + * <Craig@chatspike.net> + * + * Written by Craig Edwards, Craig McLure, and others. + * This program is free but copyrighted software; see + * the file COPYING for details. + * + * --------------------------------------------------- + */ + +#include <stdio.h> +#include <string> +#include <stdlib.h> +#include <time.h> +#include <sys/types.h> +#include <sys/socket.h> +#include <sys/time.h> +#include <string.h> +#include <unistd.h> +#include <errno.h> +#include <fcntl.h> +#include <poll.h> +#include "users.h" +#include "channels.h" +#include "modules.h" +#include "inspircd.h" +#include "m_sql.h" + +#define LT_OPER 1 +#define LT_KILL 2 +#define LT_SERVLINK 3 +#define LT_XLINE 4 +#define LT_CONNECT 5 +#define LT_DISCONNECT 6 +#define LT_FLOOD 7 +#define LT_LOADMODULE 8 + +/* $ModDesc: Allow/Deny connections based upon an arbitary SQL table */ + +Server *Srv; + +class ModuleSQLLog : public Module +{ + ConfigReader* Conf; + unsigned long dbid; + Module* SQLModule; + + public: + bool ReadConfig() + { + Conf = new ConfigReader(); + dbid = Conf->ReadInteger("sqllog","dbid",0,true); // database id of a database configured in m_sql (see m_sql config) + delete Conf; + SQLModule = Srv->FindModule("m_sql.so"); + if (!SQLModule) + Srv->Log(DEFAULT,"WARNING: m_SQLLog.so could not initialize because m_sql.so is not loaded. Load the module and rehash your server."); + return (SQLModule); + } + + ModuleSQLLog() + { + Srv = new Server; + ReadConfig(); + } + + virtual void OnRehash() + { + ReadConfig(); + } + + long InsertNick(std::string nick) + { + long nid = -1; + SQLRequest* query = new SQLRequest(SQL_RESULT,dbid,"SELECT id,nick FROM ircd_log_actors WHERE actor='"+nick+"'"); + Request queryrequest((char*)query, this, SQLModule); + SQLResult* result = (SQLResult*)queryrequest.Send(); + if (result->GetType() == SQL_OK) + { + SQLRequest* rowrequest = new SQLRequest(SQL_ROW,dbid,""); + Request rowquery((char*)rowrequest, this, SQLModule); + SQLResult* rowresult = (SQLResult*)rowquery.Send(); + if (rowresult->GetType() == SQL_ROW) + { + nid = atoi(rowresult->GetField("id")); + delete rowresult; + } + delete rowrequest; + delete result; + } + query->SetQueryType(SQL_DONE); + query->SetConnID(dbid); + Request donerequest((char*)query, this, SQLModule); + donerequest.Send(); + delete query; + if (nid < 1) + { + SQLRequest* query = new SQLRequest(SQL_COUNT,dbid,"INSERT INTO ircd_log_actors VALUES('','"+nick+"')"); + Request queryrequest((char*)query, this, SQLModule); + SQLResult* result = (SQLResult*)queryrequest.Send(); + if (result->GetType() == SQL_ERROR) + { + Srv->Log(DEFAULT,"SQL log error: " + result->GetError()); + } + if (result) + delete result; + if (query) + delete query; + nid = InsertNick(nick); + } + return nid; + } + + void InsertEntry(long category,long nickid,long hostid,long sourceid,unsigned long date); + { + char querybuffer[MAXBUF]; + snprintf(querybuffer,MAXBUF,"INSERT INTO ircd_log VALUES('',%d,%d,%d,%d,%lu)",category,nickid,hostid,sourceid,date); + SQLRequest* query = new SQLRequest(SQL_COUNT,dbid,querybuffer); + Request queryrequest((char*)query, this, SQLModule); + SQLResult* result = (SQLResult*)queryrequest.Send(); + if (result->GetType() == SQL_ERROR) + { + Srv->Log(DEFAULT,"SQL log error: " + result->GetError()); + } + if (result) + delete result; + if (query) + delete query; + return; + } + + long InsertHost(std::string host) + { + long hid = -1; + SQLRequest* query = new SQLRequest(SQL_RESULT,dbid,"SELECT id,hostname FROM ircd_log_hosts WHERE hostname='"+host+"'"); + Request queryrequest((char*)query, this, SQLModule); + SQLResult* result = (SQLResult*)queryrequest.Send(); + if (result->GetType() == SQL_OK) + { + SQLRequest* rowrequest = new SQLRequest(SQL_ROW,dbid,""); + Request rowquery((char*)rowrequest, this, SQLModule); + SQLResult* rowresult = (SQLResult*)rowquery.Send(); + if (rowresult->GetType() == SQL_ROW) + { + hid = atoi(rowresult->GetField("id")); + delete rowresult; + } + delete rowrequest; + delete result; + } + query->SetQueryType(SQL_DONE); + query->SetConnID(dbid); + Request donerequest((char*)query, this, SQLModule); + donerequest.Send(); + delete query; + if (hid < 1) + { + SQLRequest* query = new SQLRequest(SQL_COUNT,dbid,"INSERT INTO ircd_log_hosts VALUES('','"+host+"')"); + Request queryrequest((char*)query, this, SQLModule); + SQLResult* result = (SQLResult*)queryrequest.Send(); + if (result->GetType() == SQL_ERROR) + { + Srv->Log(DEFAULT,"SQL log error: " + result->GetError()); + } + if (result) + delete result; + if (query) + delete query; + hid = InsertHost(host); + } + return hid; + } + + void AddLogEntry(int category, std::string nick, std::string host, std::string source) + { + // is the sql module loaded? If not, we don't attempt to do anything. + if (!SQLModule) + return false; + + long nickid = InsertNick(nick); + long sourceid = InsertNick(source); + long hostid = InsertHost(host); + InsertEntry(category,nickid,hostid,sourceid,time(NULL)); + } + + virtual void OnOper(userrec* user) + { + AddLogEntry(LT_OPER,user->nick,user->host,user->server); + } + + virtual ~ModuleSQLLog() + { + delete Srv; + } + + virtual Version GetVersion() + { + return Version(1,0,0,1,VF_VENDOR); + } + +}; + +class ModuleSQLLogFactory : public ModuleFactory +{ + public: + ModuleSQLLogFactory() + { + } + + ~ModuleSQLLogFactory() + { + } + + virtual Module * CreateModule() + { + return new ModuleSQLLog; + } + +}; + + +extern "C" void * init_module( void ) +{ + return new ModuleSQLLogFactory; +} + |