summaryrefslogtreecommitdiff
path: root/src
diff options
context:
space:
mode:
authordanieldg <danieldg@e03df62e-2008-0410-955e-edbf42e46eb7>2010-03-06 16:58:26 +0000
committerdanieldg <danieldg@e03df62e-2008-0410-955e-edbf42e46eb7>2010-03-06 16:58:26 +0000
commitd81c373961a61871899736b27e67c9341ff6231f (patch)
treee2e7f288f066609f1b5f86feaf1099315e2e678d /src
parent410de52634ddeabfe9a57aa15131311d16abb42c (diff)
Convert SQLite driver to SQLv3
git-svn-id: http://svn.inspircd.org/repository/trunk/inspircd@12603 e03df62e-2008-0410-955e-edbf42e46eb7
Diffstat (limited to 'src')
-rw-r--r--src/modules/extra/m_sqlite3.cpp580
-rw-r--r--src/modules/m_sqllog.cpp267
-rw-r--r--src/modules/sql.h5
3 files changed, 109 insertions, 743 deletions
diff --git a/src/modules/extra/m_sqlite3.cpp b/src/modules/extra/m_sqlite3.cpp
index 902355a23..4a0538bc8 100644
--- a/src/modules/extra/m_sqlite3.cpp
+++ b/src/modules/extra/m_sqlite3.cpp
@@ -13,52 +13,24 @@
#include "inspircd.h"
#include <sqlite3.h>
-#include "m_sqlv2.h"
+#include "sql.h"
/* $ModDesc: sqlite3 provider */
/* $CompileFlags: pkgconfversion("sqlite3","3.3") pkgconfincludes("sqlite3","/sqlite3.h","") */
/* $LinkerFlags: pkgconflibs("sqlite3","/libsqlite3.so","-lsqlite3") */
-/* $ModDep: m_sqlv2.h */
/* $NoPedantic */
class SQLConn;
-class SQLite3Result;
-class ResultNotifier;
-class SQLiteListener;
-class ModuleSQLite3;
+typedef std::map<std::string, reference<SQLConn> > ConnMap;
-typedef std::map<std::string, SQLConn*> ConnMap;
-typedef std::deque<classbase*> paramlist;
-typedef std::deque<SQLite3Result*> ResultQueue;
-
-static unsigned long count(const char * const str, char a)
+class SQLite3Result : public SQLResult
{
- unsigned long n = 0;
- for (const char *p = str; *p; ++p)
- {
- if (*p == '?')
- ++n;
- }
- return n;
-}
-
-class SQLite3Result : public SQLresult
-{
- private:
+ public:
int currentrow;
int rows;
- int cols;
+ std::vector<std::vector<std::string> > fieldlists;
- std::vector<std::string> colnames;
- std::vector<SQLfieldList> fieldlists;
- SQLfieldList emptyfieldlist;
-
- SQLfieldList* fieldlist;
- SQLfieldMap* fieldmap;
-
- public:
- SQLite3Result(Module* self, Module* to, unsigned int rid)
- : SQLresult(self, to, rid), currentrow(0), rows(0), cols(0), fieldlist(NULL), fieldmap(NULL)
+ SQLite3Result() : currentrow(0), rows(0)
{
}
@@ -66,391 +38,169 @@ class SQLite3Result : public SQLresult
{
}
- void AddRow(int colsnum, char **dat, char **colname)
- {
- colnames.clear();
- cols = colsnum;
- for (int i = 0; i < colsnum; i++)
- {
- fieldlists.resize(fieldlists.size()+1);
- colnames.push_back(colname[i]);
- SQLfield sf(dat[i] ? dat[i] : "", dat[i] ? false : true);
- fieldlists[rows].push_back(sf);
- }
- rows++;
- }
-
- void UpdateAffectedCount()
- {
- rows++;
- }
-
virtual int Rows()
{
return rows;
}
- virtual int Cols()
+ virtual bool GetRow(std::vector<std::string>& result)
{
- return cols;
- }
-
- virtual std::string ColName(int column)
- {
- if (column < (int)colnames.size())
+ if (currentrow < rows)
{
- return colnames[column];
+ result.assign(fieldlists[currentrow].begin(), fieldlists[currentrow].end());
+ currentrow++;
+ return true;
}
else
{
- throw SQLbadColName();
+ result.clear();
+ return false;
}
- return "";
}
+};
- virtual int ColNum(const std::string &column)
- {
- for (unsigned int i = 0; i < colnames.size(); i++)
- {
- if (column == colnames[i])
- return i;
- }
- throw SQLbadColName();
- return 0;
- }
+class SQLConn : public refcountbase
+{
+ private:
+ sqlite3* conn;
+ reference<ConfigTag> config;
- virtual SQLfield GetValue(int row, int column)
+ public:
+ SQLConn(ConfigTag* tag) : config(tag)
{
- if ((row >= 0) && (row < rows) && (column >= 0) && (column < Cols()))
+ std::string host = tag->getString("hostname");
+ if (sqlite3_open_v2(host.c_str(), &conn, SQLITE_OPEN_READWRITE, 0) != SQLITE_OK)
{
- return fieldlists[row][column];
+ ServerInstance->Logs->Log("m_sqlite3",DEFAULT, "WARNING: Could not open DB with id: " + tag->getString("id"));
+ conn = NULL;
}
-
- throw SQLbadColName();
-
- /* XXX: We never actually get here because of the throw */
- return SQLfield("",true);
}
- virtual SQLfieldList& GetRow()
+ ~SQLConn()
{
- if (currentrow < rows)
- return fieldlists[currentrow];
- else
- return emptyfieldlist;
+ sqlite3_interrupt(conn);
+ sqlite3_close(conn);
}
- virtual SQLfieldMap& GetRowMap()
+ void Query(SQLQuery* query)
{
- /* In an effort to reduce overhead we don't actually allocate the map
- * until the first time it's needed...so...
- */
- if(fieldmap)
+ SQLite3Result res;
+ sqlite3_stmt *stmt;
+ int err = sqlite3_prepare_v2(conn, query->query.c_str(), query->query.length(), &stmt, NULL);
+ if (err != SQLITE_OK)
{
- fieldmap->clear();
+ SQLerror error(SQL_QSEND_FAIL, sqlite3_errmsg(conn));
+ query->OnError(error);
+ return;
}
- else
+ int cols = sqlite3_column_count(stmt);
+ while (1)
{
- fieldmap = new SQLfieldMap;
- }
-
- if (currentrow < rows)
- {
- for (int i = 0; i < Cols(); i++)
+ err = sqlite3_step(stmt);
+ if (err == SQLITE_ROW)
{
- fieldmap->insert(std::make_pair(ColName(i), GetValue(currentrow, i)));
+ // Add the row
+ res.fieldlists.resize(res.rows + 1);
+ res.fieldlists[res.rows].resize(cols);
+ for(int i=0; i < cols; i++)
+ {
+ const char* txt = (const char*)sqlite3_column_text(stmt, i);
+ res.fieldlists[res.rows][i] = txt ? txt : "";
+ }
+ res.rows++;
}
- currentrow++;
- }
-
- return *fieldmap;
- }
-
- virtual SQLfieldList* GetRowPtr()
- {
- fieldlist = new SQLfieldList();
-
- if (currentrow < rows)
- {
- for (int i = 0; i < Rows(); i++)
+ else if (err == SQLITE_DONE)
{
- fieldlist->push_back(fieldlists[currentrow][i]);
+ query->OnResult(res);
+ break;
}
- currentrow++;
- }
- return fieldlist;
- }
-
- virtual SQLfieldMap* GetRowMapPtr()
- {
- fieldmap = new SQLfieldMap();
-
- if (currentrow < rows)
- {
- for (int i = 0; i < Cols(); i++)
+ else
{
- fieldmap->insert(std::make_pair(colnames[i],GetValue(currentrow, i)));
+ SQLerror error(SQL_QREPLY_FAIL, sqlite3_errmsg(conn));
+ query->OnError(error);
+ break;
}
- currentrow++;
}
-
- return fieldmap;
+ sqlite3_finalize(stmt);
}
-
- virtual void Free(SQLfieldMap* fm)
- {
- delete fm;
- }
-
- virtual void Free(SQLfieldList* fl)
- {
- delete fl;
- }
-
-
};
-class SQLConn : public classbase
+class SQLiteProvider : public SQLProvider
{
- private:
- ResultQueue results;
- Module* mod;
- SQLhost host;
- sqlite3* conn;
-
public:
- SQLConn(Module* m, const SQLhost& hi)
- : mod(m), host(hi)
- {
- if (OpenDB() != SQLITE_OK)
- {
- ServerInstance->Logs->Log("m_sqlite3",DEFAULT, "WARNING: Could not open DB with id: " + host.id);
- CloseDB();
- }
- }
+ ConnMap hosts;
- ~SQLConn()
- {
- CloseDB();
- }
+ SQLiteProvider(Module* Parent) : SQLProvider(Parent, "SQL/SQLite") {}
- SQLerror Query(SQLrequest &req)
+ std::string FormatQuery(std::string q, ParamL p)
{
- /* Pointer to the buffer we screw around with substitution in */
- char* query;
-
- /* Pointer to the current end of query, where we append new stuff */
- char* queryend;
-
- /* Total length of the unescaped parameters */
- unsigned long maxparamlen, paramcount;
-
- /* The length of the longest parameter */
- maxparamlen = 0;
-
- for(ParamL::iterator i = req.query.p.begin(); i != req.query.p.end(); i++)
+ std::string res;
+ unsigned int param = 0;
+ for(std::string::size_type i = 0; i < q.length(); i++)
{
- if (i->size() > maxparamlen)
- maxparamlen = i->size();
- }
-
- /* How many params are there in the query? */
- paramcount = count(req.query.q.c_str(), '?');
-
- /* This stores copy of params to be inserted with using numbered params 1;3B*/
- ParamL paramscopy(req.query.p);
-
- /* To avoid a lot of allocations, allocate enough memory for the biggest the escaped query could possibly be.
- * sizeofquery + (maxtotalparamlength*2) + 1
- *
- * The +1 is for null-terminating the string
- */
-
- query = new char[req.query.q.length() + (maxparamlen*paramcount*2) + 1];
- queryend = query;
-
- for(unsigned long i = 0; i < req.query.q.length(); i++)
- {
- if(req.query.q[i] == '?')
+ if (q[i] != '?')
+ res.push_back(q[i]);
+ else
{
- /* We found a place to substitute..what fun.
- * use sqlite calls to escape and write the
- * escaped string onto the end of our query buffer,
- * then we "just" need to make sure queryend is
- * pointing at the right place.
- */
-
- /* Is it numbered parameter?
- */
-
- bool numbered;
- numbered = false;
-
- /* Numbered parameter number :|
- */
- unsigned int paramnum;
- paramnum = 0;
-
- /* Let's check if it's a numbered param. And also calculate it's number.
- */
-
- while ((i < req.query.q.length() - 1) && (req.query.q[i+1] >= '0') && (req.query.q[i+1] <= '9'))
- {
- numbered = true;
- ++i;
- paramnum = paramnum * 10 + req.query.q[i] - '0';
- }
-
- if (paramnum > paramscopy.size() - 1)
+ // TODO numbered parameter support ('?1')
+ if (param < p.size())
{
- /* index is out of range!
- */
- numbered = false;
- }
-
-
- if (numbered)
- {
- char* escaped;
- escaped = sqlite3_mprintf("%q", paramscopy[paramnum].c_str());
- for (char* n = escaped; *n; n++)
- {
- *queryend = *n;
- queryend++;
- }
+ char* escaped = sqlite3_mprintf("%q", p[param++].c_str());
+ res.append(escaped);
sqlite3_free(escaped);
}
- else if (req.query.p.size())
- {
- char* escaped;
- escaped = sqlite3_mprintf("%q", req.query.p.front().c_str());
- for (char* n = escaped; *n; n++)
- {
- *queryend = *n;
- queryend++;
- }
- sqlite3_free(escaped);
- req.query.p.pop_front();
- }
- else
- break;
- }
- else
- {
- *queryend = req.query.q[i];
- queryend++;
}
}
- *queryend = 0;
- req.query.q = query;
-
- SQLite3Result* res = new SQLite3Result(mod, req.source, req.id);
- res->dbid = host.id;
- res->query = req.query.q;
- paramlist params;
- params.push_back(this);
- params.push_back(res);
-
- char *errmsg = 0;
- sqlite3_update_hook(conn, QueryUpdateHook, &params);
- if (sqlite3_exec(conn, req.query.q.data(), QueryResult, &params, &errmsg) != SQLITE_OK)
- {
- std::string error(errmsg);
- sqlite3_free(errmsg);
- delete[] query;
- delete res;
- return SQLerror(SQL_QSEND_FAIL, error);
- }
- delete[] query;
-
- results.push_back(res);
- SendResults();
- return SQLerror();
- }
-
- static int QueryResult(void *params, int argc, char **argv, char **azColName)
- {
- paramlist* p = (paramlist*)params;
- ((SQLConn*)(*p)[0])->ResultReady(((SQLite3Result*)(*p)[1]), argc, argv, azColName);
- return 0;
- }
-
- static void QueryUpdateHook(void *params, int eventid, char const * azSQLite, char const * azColName, sqlite_int64 rowid)
- {
- paramlist* p = (paramlist*)params;
- ((SQLConn*)(*p)[0])->AffectedReady(((SQLite3Result*)(*p)[1]));
- }
-
- void ResultReady(SQLite3Result *res, int cols, char **data, char **colnames)
- {
- res->AddRow(cols, data, colnames);
+ return res;
}
- void AffectedReady(SQLite3Result *res)
+ std::string FormatQuery(std::string q, ParamM p)
{
- res->UpdateAffectedCount();
- }
-
- int OpenDB()
- {
- return sqlite3_open_v2(host.host.c_str(), &conn, SQLITE_OPEN_READWRITE, 0);
- }
-
- void CloseDB()
- {
- sqlite3_interrupt(conn);
- sqlite3_close(conn);
- }
-
- SQLhost GetConfHost()
- {
- return host;
- }
-
- void SendResults()
- {
- while (results.size())
+ std::string res;
+ for(std::string::size_type i = 0; i < q.length(); i++)
{
- SQLite3Result* res = results[0];
- if (res->dest)
- {
- res->Send();
- }
+ if (q[i] != '$')
+ res.push_back(q[i]);
else
{
- /* If the client module is unloaded partway through a query then the provider will set
- * the pointer to NULL. We cannot just cancel the query as the result will still come
- * through at some point...and it could get messy if we play with invalid pointers...
- */
- delete res;
+ std::string field;
+ i++;
+ while (i < q.length() && isalpha(q[i]))
+ field.push_back(q[i++]);
+ i--;
+
+ char* escaped = sqlite3_mprintf("%q", p[field].c_str());
+ res.append(escaped);
+ sqlite3_free(escaped);
}
- results.pop_front();
}
+ return res;
}
-
- void ClearResults()
+
+ void submit(SQLQuery* query)
{
- while (results.size())
+ ConnMap::iterator iter = hosts.find(query->dbid);
+ if (iter == hosts.end())
+ {
+ SQLerror err(SQL_BAD_DBID);
+ query->OnError(err);
+ }
+ else
{
- SQLite3Result* res = results[0];
- delete res;
- results.pop_front();
+ iter->second->Query(query);
}
+ delete query;
}
-
};
-
class ModuleSQLite3 : public Module
{
private:
- ConnMap connections;
- unsigned long currid;
- ServiceProvider sqlserv;
+ SQLiteProvider sqlserv;
public:
ModuleSQLite3()
- : currid(0), sqlserv(this, "SQL/sqlite", SERVICE_DATA)
+ : sqlserv(this)
{
}
@@ -466,145 +216,27 @@ class ModuleSQLite3 : public Module
virtual ~ModuleSQLite3()
{
- ClearQueue();
- ClearAllConnections();
- }
-
- void ClearQueue()
- {
- for (ConnMap::iterator iter = connections.begin(); iter != connections.end(); iter++)
- {
- iter->second->ClearResults();
- }
- }
-
- bool HasHost(const SQLhost &host)
- {
- for (ConnMap::iterator iter = connections.begin(); iter != connections.end(); iter++)
- {
- if (host == iter->second->GetConfHost())
- return true;
- }
- return false;
- }
-
- bool HostInConf(const SQLhost &h)
- {
- ConfigReader conf;
- for(int i = 0; i < conf.Enumerate("database"); i++)
- {
- SQLhost host;
- host.id = conf.ReadValue("database", "id", i);
- host.host = conf.ReadValue("database", "hostname", i);
- host.port = conf.ReadInteger("database", "port", i, true);
- host.name = conf.ReadValue("database", "name", i);
- host.user = conf.ReadValue("database", "username", i);
- host.pass = conf.ReadValue("database", "password", i);
- if (h == host)
- return true;
- }
- return false;
}
void ReadConf()
{
- ClearOldConnections();
-
- ConfigReader conf;
- for(int i = 0; i < conf.Enumerate("database"); i++)
- {
- SQLhost host;
-
- host.id = conf.ReadValue("database", "id", i);
- host.host = conf.ReadValue("database", "hostname", i);
- host.port = conf.ReadInteger("database", "port", i, true);
- host.name = conf.ReadValue("database", "name", i);
- host.user = conf.ReadValue("database", "username", i);
- host.pass = conf.ReadValue("database", "password", i);
-
- if (HasHost(host))
- continue;
-
- this->AddConn(host);
- }
- }
-
- void AddConn(const SQLhost& hi)
- {
- if (HasHost(hi))
+ sqlserv.hosts.clear();
+ ConfigTagList tags = ServerInstance->Config->ConfTags("database");
+ for(ConfigIter i = tags.first; i != tags.second; i++)
{
- ServerInstance->Logs->Log("m_sqlite3",DEFAULT, "WARNING: A sqlite connection with id: %s already exists. Aborting database open attempt.", hi.id.c_str());
- return;
+ sqlserv.hosts.insert(std::make_pair(i->second->getString("id"), new SQLConn(i->second)));
}
-
- SQLConn* newconn;
-
- newconn = new SQLConn(this, hi);
-
- connections.insert(std::make_pair(hi.id, newconn));
}
- void ClearOldConnections()
- {
- ConnMap::iterator iter,safei;
- for (iter = connections.begin(); iter != connections.end(); iter++)
- {
- if (!HostInConf(iter->second->GetConfHost()))
- {
- delete iter->second;
- safei = iter;
- --iter;
- connections.erase(safei);
- }
- }
- }
-
- void ClearAllConnections()
- {
- ConnMap::iterator i;
- while ((i = connections.begin()) != connections.end())
- {
- connections.erase(i);
- delete i->second;
- }
- }
-
- virtual void OnRehash(User* user)
+ void OnRehash(User* user)
{
ReadConf();
}
- void OnRequest(Request& request)
- {
- if(strcmp(SQLREQID, request.id) == 0)
- {
- SQLrequest* req = (SQLrequest*)&request;
- ConnMap::iterator iter;
- if((iter = connections.find(req->dbid)) != connections.end())
- {
- req->id = NewID();
- req->error = iter->second->Query(*req);
- }
- else
- {
- req->error.Id(SQL_BAD_DBID);
- }
- }
- }
-
- unsigned long NewID()
- {
- if (currid+1 == 0)
- currid++;
-
- return ++currid;
- }
-
- virtual Version GetVersion()
+ Version GetVersion()
{
return Version("sqlite3 provider", VF_VENDOR);
}
-
};
MODULE_INIT(ModuleSQLite3)
diff --git a/src/modules/m_sqllog.cpp b/src/modules/m_sqllog.cpp
deleted file mode 100644
index 5e14735ce..000000000
--- a/src/modules/m_sqllog.cpp
+++ /dev/null
@@ -1,267 +0,0 @@
-/* +------------------------------------+
- * | Inspire Internet Relay Chat Daemon |
- * +------------------------------------+
- *
- * InspIRCd: (C) 2002-2010 InspIRCd Development Team
- * See: http://wiki.inspircd.org/Credits
- *
- * This program is free but copyrighted software; see
- * the file COPYING for details.
- *
- * ---------------------------------------------------
- */
-
-#include "inspircd.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 = ServerInstance->Time();
- insert = false;
- }
-
- void Go(SQLresult* res)
- {
- switch (qs)
- {
- case FIND_SOURCE:
- if (res->Rows() && sourceid == -1 && !insert)
- {
- sourceid = atoi(res->GetValue(0,0).d.c_str());
- SQLrequest req = SQLrequest(MyMod, SQLModule, dbid, SQLquery("SELECT id,actor FROM ircd_log_actors WHERE actor='?'") % nick);
- req.Send();
- insert = false;
- qs = FIND_NICK;
- active_queries[req.id] = this;
- }
- else if (res->Rows() && sourceid == -1 && insert)
- {
- SQLrequest req = SQLrequest(MyMod, SQLModule, dbid, SQLquery("SELECT id,actor FROM ircd_log_actors WHERE actor='?'") % source);
- req.Send();
- insert = false;
- qs = FIND_SOURCE;
- active_queries[req.id] = this;
- }
- else
- {
- SQLrequest req = SQLrequest(MyMod, SQLModule, dbid, SQLquery("INSERT INTO ircd_log_actors (actor) VALUES('?')") % source);
- 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());
- SQLrequest req = SQLrequest(MyMod, SQLModule, dbid, SQLquery("SELECT id,hostname FROM ircd_log_hosts WHERE hostname='?'") % hostname);
- req.Send();
- insert = false;
- qs = FIND_HOST;
- active_queries[req.id] = this;
- }
- else if (res->Rows() && nickid == -1 && insert)
- {
- SQLrequest req = SQLrequest(MyMod, SQLModule, dbid, SQLquery("SELECT id,actor FROM ircd_log_actors WHERE actor='?'") % nick);
- req.Send();
- insert = false;
- qs = FIND_NICK;
- active_queries[req.id] = this;
- }
- else
- {
- SQLrequest req = SQLrequest(MyMod, SQLModule, dbid, SQLquery("INSERT INTO ircd_log_actors (actor) VALUES('?')") % nick);
- 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());
- SQLrequest req = SQLrequest(MyMod, SQLModule, dbid,
- SQLquery("INSERT INTO ircd_log (category_id,nick,host,source,dtime) VALUES('?','?','?','?','?')") % category % nickid % hostid % sourceid % date);
- req.Send();
- insert = true;
- qs = DONE;
- active_queries[req.id] = this;
- }
- else if (res->Rows() && hostid == -1 && insert)
- {
- SQLrequest req = SQLrequest(MyMod, SQLModule, dbid, SQLquery("SELECT id,hostname FROM ircd_log_hosts WHERE hostname='?'") % hostname);
- req.Send();
- insert = false;
- qs = FIND_HOST;
- active_queries[req.id] = this;
- }
- else
- {
- SQLrequest req = SQLrequest(MyMod, SQLModule, dbid, SQLquery("INSERT INTO ircd_log_hosts (hostname) VALUES('?')") % hostname);
- req.Send();
- insert = true;
- qs = FIND_HOST;
- active_queries[req.id] = this;
- }
- break;
-
- case DONE:
- break;
- }
- }
-};
-
-/* $ModDesc: Logs network-wide data to an SQL database */
-
-class ModuleSQLLog : public Module
-{
-
- public:
- void init()
- {
- Module* SQLutils = ServerInstance->Modules->Find("m_sqlutils.so");
- if (!SQLutils)
- throw ModuleException("Can't find m_sqlutils.so. Please load m_sqlutils.so before m_sqlauth.so.");
-
- ServiceProvider* prov = ServerInstance->Modules->FindService(SERVICE_DATA, "SQL");
- if (!prov)
- throw ModuleException("Can't find an SQL provider module. Please load one before attempting to load m_sqlauth.");
- SQLModule = prov->creator;
-
- OnRehash(NULL);
- MyMod = this;
- active_queries.clear();
-
- Implementation eventlist[] = { I_OnRehash, I_OnOper, I_OnGlobalOper, I_OnKill,
- I_OnPreCommand, I_OnUserRegister, I_OnUserQuit, I_OnLoadModule };
- ServerInstance->Modules->Attach(eventlist, this, 8);
- }
-
- void ReadConfig()
- {
- ConfigReader Conf;
- dbid = Conf.ReadValue("sqllog","dbid",0); // database id of a database configured in sql module
- }
-
- virtual void OnRehash(User* user)
- {
- ReadConfig();
- }
-
- void OnRequest(Request& request)
- {
- if(strcmp(SQLRESID, request.id) == 0)
- {
- SQLresult* res = static_cast<SQLresult*>(&request);
- std::map<unsigned long, QueryInfo*>::iterator n;
-
- n = active_queries.find(res->id);
-
- if (n != active_queries.end())
- {
- n->second->Go(res);
- active_queries.erase(n);
- }
- }
- }
-
- 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 = SQLrequest(this, SQLModule, dbid, SQLquery("SELECT id,actor FROM ircd_log_actors WHERE actor='?'") % source);
- req.Send();
- QueryInfo* i = new QueryInfo(nick, source, host, req.id, category);
- i->qs = FIND_SOURCE;
- active_queries[req.id] = i;
- }
-
- virtual void OnOper(User* user, const std::string &opertype)
- {
- AddLogEntry(LT_OPER,user->nick,user->host,user->server);
- }
-
- virtual void OnGlobalOper(User* user)
- {
- AddLogEntry(LT_OPER,user->nick,user->host,user->server);
- }
-
- virtual ModResult OnKill(User* source, User* dest, const std::string &reason)
- {
- AddLogEntry(LT_KILL,dest->nick,dest->host,source->nick);
- return MOD_RES_PASSTHRU;
- }
-
- virtual ModResult OnPreCommand(std::string &command, std::vector<std::string> &parameters, LocalUser *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(":")+parameters[0],user->server);
- }
- return MOD_RES_PASSTHRU;
- }
-
- virtual ModResult OnUserRegister(LocalUser* user)
- {
- AddLogEntry(LT_CONNECT,user->nick,user->host,user->server);
- return MOD_RES_PASSTHRU;
- }
-
- virtual void OnUserQuit(User* user, const std::string &reason, const std::string &oper_message)
- {
- AddLogEntry(LT_DISCONNECT,user->nick,user->host,user->server);
- }
-
- virtual void OnLoadModule(Module* mod)
- {
- AddLogEntry(LT_LOADMODULE,mod->ModuleSourceFile,ServerInstance->Config->ServerName.c_str(), ServerInstance->Config->ServerName.c_str());
- }
-
- virtual Version GetVersion()
- {
- return Version("Logs network-wide data to an SQL database", VF_VENDOR);
- }
-
-};
-
-MODULE_INIT(ModuleSQLLog)
diff --git a/src/modules/sql.h b/src/modules/sql.h
index 2d44584fa..9114bea88 100644
--- a/src/modules/sql.h
+++ b/src/modules/sql.h
@@ -131,6 +131,7 @@ class SQLQuery : public classbase
class SQLProvider : public DataProvider
{
public:
+ SQLProvider(Module* Creator, const std::string& Name) : DataProvider(Creator, Name) {}
/** Submit an asynchronous SQL request
* @param dbid The database ID to apply the request to
* @param query The query string
@@ -142,13 +143,13 @@ class SQLProvider : public DataProvider
* @param q The query string, with '?' parameters
* @param p The parameters to fill in in the '?' slots
*/
- virtual std::string FormatQuery(std::string q, ParamL p);
+ virtual std::string FormatQuery(std::string q, ParamL p) = 0;
/** Format a parameterized query string using proper SQL escaping.
* @param q The query string, with '$foo' parameters
* @param p The map to look up parameters in
*/
- virtual std::string FormatQuery(std::string q, ParamM p);
+ virtual std::string FormatQuery(std::string q, ParamM p) = 0;
};
#endif