X-Git-Url: https://git.netwichtig.de/gitweb/?a=blobdiff_plain;f=src%2Fmodules%2Fm_sasl.cpp;h=3c3b1d2a00112dbd7f780bfb5e375953f0348f2c;hb=04cfaea87edd1b4b742cb607ea69ece760c460aa;hp=3049a929ce613c646eaa04d7bf62c96085bc874b;hpb=287af799be5873ee32319f861fc3e205c6c2089c;p=user%2Fhenk%2Fcode%2Finspircd.git diff --git a/src/modules/m_sasl.cpp b/src/modules/m_sasl.cpp index 3049a929c..3c3b1d2a0 100644 --- a/src/modules/m_sasl.cpp +++ b/src/modules/m_sasl.cpp @@ -2,8 +2,8 @@ * | Inspire Internet Relay Chat Daemon | * +------------------------------------+ * - * InspIRCd: (C) 2002-2008 InspIRCd Development Team - * See: http://www.inspircd.org/wiki/index.php/Credits + * 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. @@ -27,7 +27,6 @@ class SaslAuthenticator { private: InspIRCd *ServerInstance; - Module *Creator; std::string agent; User *user; SaslState state; @@ -35,12 +34,10 @@ class SaslAuthenticator bool state_announced; public: - SaslAuthenticator(User *user, std::string method, InspIRCd *instance, Module *ctor) - : ServerInstance(instance), Creator(ctor), user(user), state(SASL_INIT) + SaslAuthenticator(User *user_, std::string method, Module *ctor) + : user(user_), state(SASL_INIT), state_announced(false) { - this->user->Extend("sasl_authenticator", this); - - std::deque params; + parameterlist params; params.push_back("*"); params.push_back("SASL"); params.push_back(user->uuid); @@ -48,42 +45,40 @@ class SaslAuthenticator params.push_back("S"); params.push_back(method); - Event e((char*)¶ms, Creator, "send_encap"); - e.Send(ServerInstance); + ServerInstance->PI->SendEncapsulatedData(params); } - SaslResult GetSaslResult(std::string &result) + SaslResult GetSaslResult(const std::string &result_) { - if (result == "F") + if (result_ == "F") return SASL_FAIL; - if (result == "A") + if (result_ == "A") return SASL_ABORT; return SASL_OK; } /* checks for and deals with a state change. */ - SaslState ProcessInboundMessage(std::deque &msg) + SaslState ProcessInboundMessage(const std::vector &msg) { switch (this->state) { case SASL_INIT: - this->agent = msg[1]; - this->user->Write("AUTHENTICATE %s", msg[5].c_str()); + this->agent = msg[0]; + this->user->Write("AUTHENTICATE %s", msg[3].c_str()); this->state = SASL_COMM; break; case SASL_COMM: - if (msg[1] != this->agent) + if (msg[0] != this->agent) return this->state; - if (msg[4] != "D") - this->user->Write("AUTHENTICATE %s", msg[5].c_str()); + if (msg[2] != "D") + this->user->Write("AUTHENTICATE %s", msg[3].c_str()); else { this->state = SASL_DONE; - this->result = this->GetSaslResult(msg[5]); - this->AnnounceState(); + this->result = this->GetSaslResult(msg[3]); } break; @@ -97,29 +92,31 @@ class SaslAuthenticator return this->state; } - bool SendClientMessage(const char* const* parameters, int pcnt) + void Abort(void) + { + this->state = SASL_DONE; + this->result = SASL_ABORT; + } + + bool SendClientMessage(const std::vector& parameters) { if (this->state != SASL_COMM) return true; - std::deque params; + parameterlist params; params.push_back("*"); params.push_back("SASL"); params.push_back(this->user->uuid); params.push_back(this->agent); params.push_back("C"); - for (int i = 0; i < pcnt; ++i) - params.push_back(parameters[i]); + params.insert(params.end(), parameters.begin(), parameters.end()); - Event e((char*)¶ms, Creator, "send_encap"); - e.Send(ServerInstance); + ServerInstance->PI->SendEncapsulatedData(params); - if (*parameters[0] == '*') + if (parameters[0][0] == '*') { - this->state = SASL_DONE; - this->result = SASL_ABORT; - + this->Abort(); return false; } @@ -134,137 +131,146 @@ class SaslAuthenticator switch (this->result) { case SASL_OK: - this->user->WriteServ("903 %s :SASL authentication successful", this->user->nick); + this->user->WriteNumeric(903, "%s :SASL authentication successful", this->user->nick.c_str()); break; case SASL_ABORT: - this->user->WriteServ("906 %s :SASL authentication aborted", this->user->nick); + this->user->WriteNumeric(906, "%s :SASL authentication aborted", this->user->nick.c_str()); break; case SASL_FAIL: - this->user->WriteServ("904 %s :SASL authentication failed", this->user->nick); + this->user->WriteNumeric(904, "%s :SASL authentication failed", this->user->nick.c_str()); break; default: break; } - } - ~SaslAuthenticator() - { - this->user->Shrink("sasl_authenticator"); - this->AnnounceState(); + this->state_announced = true; } }; class CommandAuthenticate : public Command { - Module* Creator; public: - CommandAuthenticate (InspIRCd* Instance, Module* creator) : Command(Instance,"AUTHENTICATE", 0, 1, true), Creator(creator) + SimpleExtItem& authExt; + GenericCap& cap; + CommandAuthenticate(Module* Creator, SimpleExtItem& ext, GenericCap& Cap) + : Command(Creator, "AUTHENTICATE", 1), authExt(ext), cap(Cap) { - this->source = "m_sasl.so"; + works_before_reg = true; } - CmdResult Handle (const char* const* parameters, int pcnt, User *user) + CmdResult Handle (const std::vector& parameters, User *user) { /* Only allow AUTHENTICATE on unregistered clients */ if (user->registered != REG_ALL) { - if (!user->GetExt("sasl")) + if (!cap.ext.get(user)) return CMD_FAILURE; - SaslAuthenticator *sasl; - if (!(user->GetExt("sasl_authenticator", sasl))) - sasl = new SaslAuthenticator(user, parameters[0], ServerInstance, Creator); - else if (sasl->SendClientMessage(parameters, pcnt) == false) // IAL abort extension --nenolod - delete sasl; + SaslAuthenticator *sasl = authExt.get(user); + if (!sasl) + authExt.set(user, new SaslAuthenticator(user, parameters[0], creator)); + else if (sasl->SendClientMessage(parameters) == false) // IAL abort extension --nenolod + { + sasl->AnnounceState(); + authExt.unset(user); + } } return CMD_FAILURE; } }; +class CommandSASL : public Command +{ + public: + SimpleExtItem& authExt; + CommandSASL(Module* Creator, SimpleExtItem& ext) : Command(Creator, "SASL", 2), authExt(ext) + { + this->flags_needed = FLAG_SERVERONLY; // should not be called by users + } + + CmdResult Handle(const std::vector& parameters, User *user) + { + User* target = ServerInstance->FindNick(parameters[1]); + if (!target) + { + ServerInstance->Logs->Log("m_sasl", DEBUG,"User not found in sasl ENCAP event: %s", parameters[1].c_str()); + return CMD_FAILURE; + } + + SaslAuthenticator *sasl = authExt.get(target); + if (!sasl) + return CMD_FAILURE; + + SaslState state = sasl->ProcessInboundMessage(parameters); + if (state == SASL_DONE) + { + sasl->AnnounceState(); + authExt.unset(target); + } + return CMD_SUCCESS; + } + + RouteDescriptor GetRouting(User* user, const std::vector& parameters) + { + return ROUTE_BROADCAST; + } +}; class ModuleSASL : public Module { - CommandAuthenticate* sasl; + SimpleExtItem authExt; + GenericCap cap; + CommandAuthenticate auth; + CommandSASL sasl; public: - - ModuleSASL(InspIRCd* Me) - : Module(Me) + ModuleSASL() + : authExt("sasl_auth", this), cap(this, "sasl"), auth(this, authExt, cap), sasl(this, authExt) { - Implementation eventlist[] = { I_OnEvent, I_OnUserRegister, I_OnPostConnect }; + Implementation eventlist[] = { I_OnEvent, I_OnUserRegister, I_OnSetConnectClass }; ServerInstance->Modules->Attach(eventlist, this, 3); - sasl = new CommandAuthenticate(ServerInstance, this); - ServerInstance->AddCommand(sasl); + ServiceProvider* providelist[] = { &auth, &sasl, &authExt }; + ServerInstance->Modules->AddServices(providelist, 3); if (!ServerInstance->Modules->Find("m_services_account.so") || !ServerInstance->Modules->Find("m_cap.so")) ServerInstance->Logs->Log("m_sasl", DEFAULT, "WARNING: m_services_account.so and m_cap.so are not loaded! m_sasl.so will NOT function correctly until these two modules are loaded!"); } - virtual int OnUserRegister(User *user) + ModResult OnUserRegister(LocalUser *user) { - if (user->GetExt("sasl")) + SaslAuthenticator *sasl_ = authExt.get(user); + if (sasl_) { - user->WriteServ("906 %s :SASL authentication aborted", user->nick); - user->Shrink("sasl"); + sasl_->Abort(); + authExt.unset(user); } - return 0; + return MOD_RES_PASSTHRU; } - virtual void OnPostConnect(User* user) + ModResult OnSetConnectClass(LocalUser* user, ConnectClass* myclass) { - if (!IS_LOCAL(user)) - return; - - std::string* str = NULL; - - if (user->GetExt("accountname", str)) + if (myclass->config->getBool("requiresasl")) { - std::deque params; - params.push_back(user->uuid); - params.push_back("accountname"); - params.push_back(*str); - Event e((char*)¶ms, this, "send_metadata"); - e.Send(ServerInstance); + const AccountExtItem* ext = GetAccountExtItem(); + if (ext && !ext->get(user)) + return MOD_RES_DENY; } - return; + return MOD_RES_PASSTHRU; } - virtual ~ModuleSASL() + ~ModuleSASL() { } - virtual Version GetVersion() + Version GetVersion() { - return Version(1,2,0,1,VF_VENDOR,API_VERSION); + return Version("Provides support for IRC Authentication Layer (aka: atheme SASL) via AUTHENTICATE.",VF_VENDOR); } - virtual void OnEvent(Event *ev) + void OnEvent(Event &ev) { - GenericCapHandler(ev, "sasl", "sasl"); - - if (ev->GetEventID() == "encap_received") - { - std::deque* parameters = (std::deque*)ev->GetData(); - - if ((*parameters)[1] != "SASL") - return; - - User* target = ServerInstance->FindNick((*parameters)[2]); - if (!target) - { - ServerInstance->Logs->Log("m_sasl", DEBUG,"User not found in sasl ENCAP event: %s", (*parameters)[2].c_str()); - return; - } - - SaslAuthenticator *sasl; - if (!target->GetExt("sasl_authenticator", sasl)) - return; - - SaslState state = sasl->ProcessInboundMessage(*parameters); - if (state == SASL_DONE) - delete sasl; - } + cap.HandleEvent(ev); } };