Change lockserv emergency unlock procedure, and add one for jumpserver [jackmcbarn]

[user/henk/code/inspircd.git] / src / modules / m_sqloper.cpp

diff --git a/src/modules/m_sqloper.cpp b/src/modules/m_sqloper.cpp
index 281593cfa6b8a587ecb0e6a5b8c90ba85b6255b4..307f72f3c034e70194ff67edac07b129c8699ce9 100644 (file)
--- a/src/modules/m_sqloper.cpp
+++ b/src/modules/m_sqloper.cpp
@@ -35,28 +35,47 @@ class OpMeQuery : public SQLQuery
 {
  public:
        const std::string uid, username, password;
-       OpMeQuery(Module* me, const std::string& db, const std::string& q, const std::string& u, const std::string& un, const std::string& pw)
-               : SQLQuery(me, db, q), uid(u), username(un), password(pw) {}
+       OpMeQuery(Module* me, const std::string& q, const std::string& u, const std::string& un, const std::string& pw)
+               : SQLQuery(me, q), uid(u), username(un), password(pw)
+       {
+               ServerInstance->Logs->Log("m_sqloper",DEBUG, "SQLOPER: query=\"%s\"", q.c_str());
+       }
 
        void OnResult(SQLResult& res)
        {
+               ServerInstance->Logs->Log("m_sqloper",DEBUG, "SQLOPER: result for %s", uid.c_str());
                User* user = ServerInstance->FindNick(uid);
                if (!user)
                        return;
 
-               // multiple rows may exist for multiple hosts
-               parameterlist row;
+               // multiple rows may exist
+               SQLEntries row;
                while (res.GetRow(row))
                {
-                       if (OperUser(user, row[2], row[3]))
+#if 0
+                       parameterlist cols;
+                       res.GetCols(cols);
+
+                       std::vector<KeyVal>* items;
+                       reference<ConfigTag> tag = ConfigTag::create("oper", "<m_sqloper>", 0, items);
+                       for(unsigned int i=0; i < cols.size(); i++)
+                       {
+                               if (!row[i].nul)
+                                       items->insert(std::make_pair(cols[i], row[i]));
+                       }
+#else
+                       if (OperUser(user, row[0], row[1]))
                                return;
+#endif
                }
+               ServerInstance->Logs->Log("m_sqloper",DEBUG, "SQLOPER: no matches for %s (checked %d rows)", uid.c_str(), res.Rows());
                // nobody succeeded... fall back to OPER
                fallback();
        }
 
        void OnError(SQLerror& error)
        {
+               ServerInstance->Logs->Log("m_sqloper",DEFAULT, "SQLOPER: query failed (%s)", error.Str());
                fallback();
        }
 
@@ -85,7 +104,10 @@ class OpMeQuery : public SQLQuery
        {
                OperIndex::iterator iter = ServerInstance->Config->oper_blocks.find(" " + type);
                if (iter == ServerInstance->Config->oper_blocks.end())
+               {
+                       ServerInstance->Logs->Log("m_sqloper",DEFAULT, "SQLOPER: bad type '%s' in returned row for oper %s", type.c_str(), username.c_str());
                        return false;
+               }
                OperInfo* ifo = iter->second;
 
                std::string hostname(user->ident);
@@ -106,7 +128,7 @@ class OpMeQuery : public SQLQuery
 
 class ModuleSQLOper : public Module
 {
-       std::string databaseid;
+       std::string query;
        std::string hashtype;
        dynamic_reference<SQLProvider> SQL;
 
@@ -123,10 +145,12 @@ public:
 
        void OnRehash(User* user)
        {
-               ConfigReader Conf;
+               ConfigTag* tag = ServerInstance->Config->ConfValue("sqloper");
 
-               databaseid = Conf.ReadValue("sqloper", "dbid", 0); /* Database ID of a database configured for the service provider module */
-               hashtype = Conf.ReadValue("sqloper", "hash", 0);
+               SQL.SetProvider("SQL/" + tag->getString("dbid"));
+               SQL.lookup();
+               hashtype = tag->getString("hash");
+               query = tag->getString("query", "SELECT hostname as host, type FROM ircd_opers WHERE username='$username' AND password='$password'");
        }
 
        ModResult OnPreCommand(std::string &command, std::vector<std::string> &parameters, LocalUser *user, bool validated, const std::string &original_line)
@@ -144,13 +168,12 @@ public:
        {
                HashProvider* hash = ServerInstance->Modules->FindDataService<HashProvider>("hash/" + hashtype);
 
-               parameterlist params;
-               params.push_back(username);
-               params.push_back(hash ? hash->hexsum(password) : password);
+               ParamM userinfo;
+               SQL->PopulateUserInfo(user, userinfo);
+               userinfo["username"] = username;
+               userinfo["password"] = hash ? hash->hexsum(password) : password;
 
-               SQL->submit(new OpMeQuery(this, databaseid, SQL->FormatQuery(
-                       "SELECT username, password, hostname, type FROM ircd_opers WHERE username = '?' AND password='?'", params
-                       ), user->uuid, username, password));
+               SQL->submit(new OpMeQuery(this, SQL->FormatQuery(query, userinfo), user->uuid, username, password));
        }
 
        Version GetVersion()