diff options
Diffstat (limited to 'src/modules')
-rw-r--r-- | src/modules/m_cgiirc.cpp | 177 |
1 files changed, 109 insertions, 68 deletions
diff --git a/src/modules/m_cgiirc.cpp b/src/modules/m_cgiirc.cpp index ecb07f96e..8d911a5b4 100644 --- a/src/modules/m_cgiirc.cpp +++ b/src/modules/m_cgiirc.cpp @@ -34,13 +34,56 @@ enum }; // We need this method up here so that it can be accessed from anywhere -static void ChangeIP(User* user, const std::string& newip) +static void ChangeIP(LocalUser* user, const irc::sockets::sockaddrs& sa) { + // Set the users IP address and make sure they are in the right clone pool. ServerInstance->Users->RemoveCloneCounts(user); - user->SetClientIP(newip); + user->SetClientIP(sa); ServerInstance->Users->AddClone(user); + if (user->quitting) + return; + + // Recheck the connect class. + user->MyClass = NULL; + user->SetClass(); + user->CheckClass(); + if (user->quitting) + return; + + // Check if this user matches any XLines. + user->CheckLines(true); + if (user->quitting) + return; } +// Encapsulates information about an ident host. +class IdentHost +{ + private: + std::string hostmask; + std::string newident; + + public: + IdentHost(const std::string& mask, const std::string& ident) + : hostmask(mask) + , newident(ident) + { + } + + const std::string& GetIdent() const + { + return newident; + } + + bool Matches(LocalUser* user) const + { + if (!InspIRCd::Match(user->GetRealHost(), hostmask, ascii_case_insensitive_map)) + return false; + + return InspIRCd::MatchCIDR(user->GetIPString(), hostmask, ascii_case_insensitive_map); + } +}; + // Encapsulates information about a WebIRC host. class WebIRCHost { @@ -112,11 +155,11 @@ class CommandWebIRC : public SplitCommand CmdResult HandleLocal(LocalUser* user, const Params& parameters) CXX11_OVERRIDE { - if (user->registered == REG_ALL) + if (user->registered == REG_ALL || realhost.get(user)) return CMD_FAILURE; irc::sockets::sockaddrs ipaddr; - if (!irc::sockets::aptosa(parameters[3], 0, ipaddr)) + if (!irc::sockets::aptosa(parameters[3], user->client_sa.port(), ipaddr)) { user->CommandFloodPenalty += 5000; WriteLog("Connecting user %s (%s) tried to use WEBIRC but gave an invalid IP address.", @@ -140,7 +183,7 @@ class CommandWebIRC : public SplitCommand // Set the IP address sent via WEBIRC. We ignore the hostname and lookup // instead do our own DNS lookups because of unreliable gateways. - ChangeIP(user, parameters[3]); + ChangeIP(user, ipaddr); return CMD_SUCCESS; } @@ -166,28 +209,47 @@ class CommandWebIRC : public SplitCommand class ModuleCgiIRC : public Module, public Whois::EventListener { + private: CommandWebIRC cmd; - std::vector<std::string> hosts; + std::vector<IdentHost> hosts; - static void RecheckClass(LocalUser* user) + static bool ParseIdent(LocalUser* user, irc::sockets::sockaddrs& out) { - user->MyClass = NULL; - user->SetClass(); - user->CheckClass(); - } + const char* ident = NULL; + if (user->ident.length() == 8) + { + // The ident is an IPv4 address encoded in hexadecimal with two characters + // per address segment. + ident = user->ident.c_str(); + } + else if (user->ident.length() == 9 && user->ident[0] == '~') + { + // The same as above but m_ident got to this user before we did. Strip the + // ident prefix and continue as normal. + ident = user->ident.c_str() + 1; + } + else + { + // The user either does not have an IPv4 in their ident or the gateway server + // is also running an identd. In the latter case there isn't really a lot we + // can do so we just assume that the client in question is not connecting via + // an ident gateway. + return false; + } - void HandleIdent(LocalUser* user, const std::string& newip) - { - cmd.realhost.set(user, user->GetRealHost()); - cmd.realip.set(user, user->GetIPString()); + // Try to convert the IP address to a string. If this fails then the user + // does not have an IPv4 address in their ident. + errno = 0; + unsigned long address = strtoul(ident, NULL, 16); + if (errno) + return false; - cmd.WriteLog("Connecting user %s is using an ident gateway; changing their IP from %s to %s.", - user->uuid.c_str(), user->GetIPString().c_str(), newip.c_str()); - ChangeIP(user, newip); - RecheckClass(user); + out.in4.sin_family = AF_INET; + out.in4.sin_addr.s_addr = htonl(address); + return true; } -public: + public: ModuleCgiIRC() : Whois::EventListener(this) , cmd(this) @@ -201,7 +263,7 @@ public: void ReadConfig(ConfigStatus& status) CXX11_OVERRIDE { - std::vector<std::string> identhosts; + std::vector<IdentHost> identhosts; std::vector<WebIRCHost> webirchosts; ConfigTagList tags = ServerInstance->Config->ConfTags("cgihost"); @@ -219,7 +281,8 @@ public: if (stdalgo::string::equalsci(type, "ident")) { // The IP address should be looked up from the hex IP address. - identhosts.push_back(mask); + const std::string newident = tag->getString("newident", "gateway", ServerInstance->IsIdent); + identhosts.push_back(IdentHost(mask, newident)); } else if (stdalgo::string::equalsci(type, "webirc")) { @@ -247,22 +310,6 @@ public: cmd.notify = ServerInstance->Config->ConfValue("cgiirc")->getBool("opernotice", true); } - ModResult OnCheckReady(LocalUser *user) CXX11_OVERRIDE - { - if (!cmd.realip.get(user)) - return MOD_RES_PASSTHRU; - - RecheckClass(user); - if (user->quitting) - return MOD_RES_DENY; - - user->CheckLines(true); - if (user->quitting) - return MOD_RES_DENY; - - return MOD_RES_PASSTHRU; - } - ModResult OnSetConnectClass(LocalUser* user, ConnectClass* myclass) CXX11_OVERRIDE { // If <connect:webirc> is not set then we have nothing to do. @@ -283,14 +330,33 @@ public: ModResult OnUserRegister(LocalUser* user) CXX11_OVERRIDE { - for (std::vector<std::string>::const_iterator iter = hosts.begin(); iter != hosts.end(); ++iter) + // There is no need to check for gateways if one is already being used. + if (cmd.realhost.get(user)) + return MOD_RES_PASSTHRU; + + for (std::vector<IdentHost>::const_iterator iter = hosts.begin(); iter != hosts.end(); ++iter) { - if (!InspIRCd::Match(user->GetRealHost(), *iter, ascii_case_insensitive_map) && !InspIRCd::MatchCIDR(user->GetIPString(), *iter, ascii_case_insensitive_map)) + // If we don't match the host then skip to the next host. + if (!iter->Matches(user)) continue; - CheckIdent(user); // Nothing on failure. - user->CheckLines(true); - break; + // We have matched an <cgihost> block! Try to parse the encoded IPv4 address + // out of the ident. + irc::sockets::sockaddrs address(user->client_sa); + if (!ParseIdent(user, address)) + return MOD_RES_PASSTHRU; + + // Store the hostname and IP of the gateway for later use. + cmd.realhost.set(user, user->GetRealHost()); + cmd.realip.set(user, user->GetIPString()); + + const std::string& newident = iter->GetIdent(); + cmd.WriteLog("Connecting user %s is using an ident gateway; changing their IP from %s to %s and their ident from %s to %s.", + user->uuid.c_str(), user->GetIPString().c_str(), address.addr().c_str(), user->ident.c_str(), newident.c_str()); + + user->ChangeIdent(newident); + ChangeIP(user, address); + break; } return MOD_RES_PASSTHRU; } @@ -313,31 +379,6 @@ public: whois.SendLine(RPL_WHOISGATEWAY, *realhost, *realip, "is connected via an ident gateway"); } - bool CheckIdent(LocalUser* user) - { - const char* ident; - in_addr newip; - - if (user->ident.length() == 8) - ident = user->ident.c_str(); - else if (user->ident.length() == 9 && user->ident[0] == '~') - ident = user->ident.c_str() + 1; - else - return false; - - errno = 0; - unsigned long ipaddr = strtoul(ident, NULL, 16); - if (errno) - return false; - newip.s_addr = htonl(ipaddr); - std::string newipstr(inet_ntoa(newip)); - - user->ident = "~cgiirc"; - HandleIdent(user, newipstr); - - return true; - } - Version GetVersion() CXX11_OVERRIDE { return Version("Enables forwarding the real IP address of a user from a gateway to the IRC server", VF_VENDOR); |