X-Git-Url: https://git.netwichtig.de/gitweb/?a=blobdiff_plain;f=src%2Fmodules%2Fextra%2Fm_sqlite3.cpp;h=1e3a65a18fdea27e4483daff11f05fd366534116;hb=571714e28b26cc59cbc8d27098a5ba981240ee2d;hp=cc54b90bf195df44b2f5684725a7645b5857846d;hpb=4e0997924d4052dede401fa1c0a1a91ae81c9aa3;p=user%2Fhenk%2Fcode%2Finspircd.git diff --git a/src/modules/extra/m_sqlite3.cpp b/src/modules/extra/m_sqlite3.cpp index cc54b90bf..1e3a65a18 100644 --- a/src/modules/extra/m_sqlite3.cpp +++ b/src/modules/extra/m_sqlite3.cpp @@ -1,27 +1,40 @@ -/* +------------------------------------+ - * | Inspire Internet Relay Chat Daemon | - * +------------------------------------+ +/* + * InspIRCd -- Internet Relay Chat Daemon * - * InspIRCd: (C) 2002-2010 InspIRCd Development Team - * See: http://wiki.inspircd.org/Credits + * Copyright (C) 2009-2010 Daniel De Graaf + * Copyright (C) 2007-2009 Dennis Friis + * Copyright (C) 2007, 2009 Craig Edwards + * Copyright (C) 2008 Pippijn van Steenhoven * - * This program is free but copyrighted software; see - * the file COPYING for details. + * This file is part of InspIRCd. InspIRCd is free software: you can + * redistribute it and/or modify it under the terms of the GNU General Public + * License as published by the Free Software Foundation, version 2. * - * --------------------------------------------------- + * This program is distributed in the hope that it will be useful, but WITHOUT + * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS + * FOR A PARTICULAR PURPOSE. See the GNU General Public License for more + * details. + * + * You should have received a copy of the GNU General Public License + * along with this program. If not, see . */ + #include "inspircd.h" #include #include "sql.h" +#ifdef _WIN32 +# pragma comment(lib, "sqlite3.lib") +#endif + /* $ModDesc: sqlite3 provider */ /* $CompileFlags: pkgconfversion("sqlite3","3.3") pkgconfincludes("sqlite3","/sqlite3.h","") */ /* $LinkerFlags: pkgconflibs("sqlite3","/libsqlite3.so","-lsqlite3") */ /* $NoPedantic */ class SQLConn; -typedef std::map > ConnMap; +typedef std::map ConnMap; class SQLite3Result : public SQLResult { @@ -58,20 +71,21 @@ class SQLite3Result : public SQLResult return false; } } + virtual void GetCols(std::vector& result) { result.assign(columns.begin(), columns.end()); } }; -class SQLConn : public refcountbase +class SQLConn : public SQLProvider { private: sqlite3* conn; reference config; public: - SQLConn(ConfigTag* tag) : config(tag) + SQLConn(Module* Parent, ConfigTag* tag) : SQLProvider(Parent, "SQL/" + tag->getString("id")), config(tag) { std::string host = tag->getString("hostname"); if (sqlite3_open_v2(host.c_str(), &conn, SQLITE_OPEN_READWRITE, 0) != SQLITE_OK) @@ -83,15 +97,18 @@ class SQLConn : public refcountbase ~SQLConn() { - sqlite3_interrupt(conn); - sqlite3_close(conn); + if (conn) + { + sqlite3_interrupt(conn); + sqlite3_close(conn); + } } - void Query(SQLQuery* query) + void Query(SQLQuery* query, const std::string& q) { SQLite3Result res; sqlite3_stmt *stmt; - int err = sqlite3_prepare_v2(conn, query->query.c_str(), query->query.length(), &stmt, NULL); + int err = sqlite3_prepare_v2(conn, q.c_str(), q.length(), &stmt, NULL); if (err != SQLITE_OK) { SQLerror error(SQL_QSEND_FAIL, sqlite3_errmsg(conn)); @@ -134,16 +151,14 @@ class SQLConn : public refcountbase } sqlite3_finalize(stmt); } -}; -class SQLiteProvider : public SQLProvider -{ - public: - ConnMap hosts; - - SQLiteProvider(Module* Parent) : SQLProvider(Parent, "SQL/SQLite") {} + virtual void submit(SQLQuery* query, const std::string& q) + { + Query(query, q); + delete query; + } - std::string FormatQuery(const std::string& q, const ParamL& p) + virtual void submit(SQLQuery* query, const std::string& q, const ParamL& p) { std::string res; unsigned int param = 0; @@ -153,7 +168,6 @@ class SQLiteProvider : public SQLProvider res.push_back(q[i]); else { - // TODO numbered parameter support ('?1') if (param < p.size()) { char* escaped = sqlite3_mprintf("%q", p[param++].c_str()); @@ -162,10 +176,10 @@ class SQLiteProvider : public SQLProvider } } } - return res; + submit(query, res); } - std::string FormatQuery(const std::string& q, const ParamM& p) + virtual void submit(SQLQuery* query, const std::string& q, const ParamM& p) { std::string res; for(std::string::size_type i = 0; i < q.length(); i++) @@ -176,7 +190,7 @@ class SQLiteProvider : public SQLProvider { std::string field; i++; - while (i < q.length() && isalpha(q[i])) + while (i < q.length() && isalnum(q[i])) field.push_back(q[i++]); i--; @@ -189,57 +203,55 @@ class SQLiteProvider : public SQLProvider } } } - return res; - } - - void submit(SQLQuery* query) - { - ConnMap::iterator iter = hosts.find(query->dbid); - if (iter == hosts.end()) - { - SQLerror err(SQL_BAD_DBID); - query->OnError(err); - } - else - { - iter->second->Query(query); - } - delete query; + submit(query, res); } }; class ModuleSQLite3 : public Module { private: - SQLiteProvider sqlserv; + ConnMap conns; public: ModuleSQLite3() - : sqlserv(this) { } void init() { - ServerInstance->Modules->AddService(sqlserv); - ReadConf(); Implementation eventlist[] = { I_OnRehash }; - ServerInstance->Modules->Attach(eventlist, this, 1); + ServerInstance->Modules->Attach(eventlist, this, sizeof(eventlist)/sizeof(Implementation)); } virtual ~ModuleSQLite3() { + ClearConns(); + } + + void ClearConns() + { + for(ConnMap::iterator i = conns.begin(); i != conns.end(); i++) + { + SQLConn* conn = i->second; + ServerInstance->Modules->DelService(*conn); + delete conn; + } + conns.clear(); } void ReadConf() { - sqlserv.hosts.clear(); + ClearConns(); ConfigTagList tags = ServerInstance->Config->ConfTags("database"); for(ConfigIter i = tags.first; i != tags.second; i++) { - sqlserv.hosts.insert(std::make_pair(i->second->getString("id"), new SQLConn(i->second))); + if (i->second->getString("module", "sqlite") != "sqlite") + continue; + SQLConn* conn = new SQLConn(this, i->second); + conns.insert(std::make_pair(i->second->getString("id"), conn)); + ServerInstance->Modules->AddService(*conn); } }