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>
35 #include "helperfuncs.h"
39 /* $ModDesc: Allows storage of oper credentials in an SQL table */
41 /* Required for the FOREACH_MOD alias (OnOper event) */
43 extern ServerConfig* Config;
44 extern std::vector<Module*> modules;
45 extern std::vector<ircd_module*> factory;
49 class ModuleSQLOper : public Module
58 dbid = Conf->ReadInteger("sqloper","dbid",0,true); // database id of a database configured in m_sql (see m_sql config)
59 SQLModule = Srv->FindModule("m_sql.so");
61 Srv->Log(DEFAULT,"WARNING: m_sqloper.so could not initialize because m_sql.so is not loaded. Load the module and rehash your server.");
65 ModuleSQLOper(Server* Me)
69 Conf = new ConfigReader();
73 virtual void OnRehash(std::string parameter)
76 Conf = new ConfigReader();
80 void Implements(char* List)
82 List[I_OnRehash] = List[I_OnPreCommand] = 1;
85 virtual int OnPreCommand(std::string command, char **parameters, int pcnt, userrec *user, bool validated)
87 if ((command == "OPER") && (validated))
89 if (LookupOper(parameters[0],parameters[1],user))
95 bool LookupOper(std::string username, std::string password, userrec* user)
99 // is the sql module loaded? If not, we don't attempt to do anything.
103 // sanitize the password (we dont want any mysql insertion exploits!)
104 std::string temp = "";
105 for (unsigned int q = 0; q < password.length(); q++)
107 if (password[q] == '\'')
111 else if (password[q] == '"')
113 temp = temp + "\\\"";
115 else temp = temp + password[q];
119 for (unsigned int v = 0; v < username.length(); v++)
121 if (username[v] == '\'')
125 if (username[v] == '"')
127 temp = temp + "\\\"";
129 else temp = temp + username[v];
133 // Create a request containing the SQL query and send it to m_sql.so
134 SQLRequest* query = new SQLRequest(SQL_RESULT,dbid,"SELECT username,password,hostname,type FROM ircd_opers WHERE username='"+username+"' AND password=md5('"+password+"')");
135 Request queryrequest((char*)query, this, SQLModule);
136 SQLResult* result = (SQLResult*)queryrequest.Send();
138 // Did we get "OK" as a result?
139 if (result->GetType() == SQL_OK)
141 Srv->Log(DEBUG,"An SQL based oper exists");
142 // if we did, this means we may now request a row... there should be only one row for each user, so,
143 // we don't need to loop to fetch multiple rows.
144 SQLRequest* rowrequest = new SQLRequest(SQL_ROW,dbid,"");
145 Request rowquery((char*)rowrequest, this, SQLModule);
146 SQLResult* rowresult = (SQLResult*)rowquery.Send();
148 // did we get a row? If we did, we can now do something with the fields
149 if (rowresult->GetType() == SQL_ROW)
151 if (rowresult->GetField("username") == username)
156 for (int j =0; j < Conf->Enumerate("type"); j++)
158 std::string TypeName = Conf->ReadValue("type","name",j);
159 Srv->Log(DEBUG,"Scanning opertype: "+TypeName);
160 std::string pattern = std::string(user->ident) + "@" + std::string(user->host);
162 if((TypeName == rowresult->GetField("type")) && OneOfMatches(pattern.c_str(), rowresult->GetField("hostname").c_str()))
164 /* found this oper's opertype */
165 Srv->Log(DEBUG,"Host and type match: "+TypeName+" "+rowresult->GetField("type"));
166 std::string HostName = Conf->ReadValue("type","host",j);
169 Srv->ChangeHost(user,HostName);
171 strlcpy(user->oper,rowresult->GetField("type").c_str(),NICKMAX-1);
172 WriteOpers("*** %s (%s@%s) is now an IRC operator of type %s",user->nick,user->ident,user->host,rowresult->GetField("type").c_str());
173 WriteServ(user->fd,"381 %s :You are now an IRC operator of type %s",user->nick,rowresult->GetField("type").c_str());
174 if(!strchr(user->modes,'o'))
176 strcat(user->modes,"o");
177 WriteServ(user->fd,"MODE %s :+o",user->nick);
178 FOREACH_MOD(I_OnOper,OnOper(user,rowresult->GetField("type")));
180 FOREACH_MOD(I_OnPostOper,OnPostOper(user,rowresult->GetField("type")));
181 log(DEFAULT,"OPER: %s!%s@%s opered as type: %s",user->nick,user->ident,user->host,rowresult->GetField("type").c_str());
193 // we didn't have a row.
205 query->SetQueryType(SQL_DONE);
206 query->SetConnID(dbid);
207 Request donerequest((char*)query, this, SQLModule);
213 virtual ~ModuleSQLOper()
218 virtual Version GetVersion()
220 return Version(1,0,0,1,VF_VENDOR);
225 class ModuleSQLOperFactory : public ModuleFactory
228 ModuleSQLOperFactory()
232 ~ModuleSQLOperFactory()
236 virtual Module * CreateModule(Server* Me)
238 return new ModuleSQLOper(Me);
244 extern "C" void * init_module( void )
246 return new ModuleSQLOperFactory;