X-Git-Url: https://git.netwichtig.de/gitweb/?a=blobdiff_plain;f=include%2Fnumericbuilder.h;h=a077c666deaa9ff67e1d1261281a487d2b492d1e;hb=635cb9d65f6d7f6758ae8ed874da00c8d94b6e39;hp=371f275a91b44afc694204afd708d74259cfe9b7;hpb=28dcc1f9e017152f03b0d9bfbcc494260b015a0a;p=user%2Fhenk%2Fcode%2Finspircd.git diff --git a/include/numericbuilder.h b/include/numericbuilder.h index 371f275a9..a077c666d 100644 --- a/include/numericbuilder.h +++ b/include/numericbuilder.h @@ -1,7 +1,8 @@ /* * InspIRCd -- Internet Relay Chat Daemon * - * Copyright (C) 2015 Attila Molnar + * Copyright (C) 2018-2020 Sadie Powell + * Copyright (C) 2015-2016 Attila Molnar * * 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 @@ -22,12 +23,19 @@ namespace Numeric { class WriteNumericSink; + class WriteRemoteNumericSink; template class GenericBuilder; template class Builder; + + template + class GenericParamBuilder; + + template + class ParamBuilder; } class Numeric::WriteNumericSink @@ -40,9 +48,25 @@ class Numeric::WriteNumericSink { } - void operator()(unsigned int numeric, const std::string& params) const + void operator()(Numeric& numeric) const + { + user->WriteNumeric(numeric); + } +}; + +class Numeric::WriteRemoteNumericSink +{ + User* const user; + + public: + WriteRemoteNumericSink(User* u) + : user(u) + { + } + + void operator()(Numeric& numeric) const { - user->WriteNumeric(numeric, params); + user->WriteRemoteNumeric(numeric); } }; @@ -50,45 +74,43 @@ template class Numeric::GenericBuilder { Sink sink; - std::string data; - const unsigned int numeric; + Numeric numeric; const std::string::size_type max; - std::string::size_type beginpos; bool HasRoom(const std::string::size_type additional) const { - return (data.size() + additional <= max); + return (numeric.GetParams().back().size() + additional <= max); } public: GenericBuilder(Sink s, unsigned int num, bool addparam = true, size_t additionalsize = 0) : sink(s) , numeric(num) - , max(ServerInstance->Config->Limits.MaxLine - ServerInstance->Config->ServerName.size() - additionalsize - 9) + , max(ServerInstance->Config->Limits.MaxLine - ServerInstance->Config->GetServerName().size() - additionalsize - 10) { if (addparam) - data.push_back(':'); - SaveBeginPos(); + numeric.push(std::string()); } - std::string& GetNumeric() { return data; } + Numeric& GetNumeric() { return numeric; } void Add(const std::string& entry) { if (!HasRoom(entry.size())) Flush(); - data.append(entry).push_back(Sep); + numeric.GetParams().back().append(entry).push_back(Sep); } void Add(const std::string& entry1, const std::string& entry2) { if (!HasRoom(entry1.size() + entry2.size())) Flush(); - data.append(entry1).append(entry2).push_back(Sep); + numeric.GetParams().back().append(entry1).append(entry2).push_back(Sep); } void Flush() { + std::string& data = numeric.GetParams().back(); if (IsEmpty()) { if (!SendEmpty) @@ -99,13 +121,11 @@ class Numeric::GenericBuilder data.erase(data.size()-1); } - sink(numeric, data); - if (data.size() > beginpos) - data.erase(beginpos); + sink(numeric); + data.clear(); } - bool IsEmpty() const { return (data.size() == beginpos); } - void SaveBeginPos() { beginpos = data.size(); } + bool IsEmpty() const { return (numeric.GetParams().back().empty()); } }; template @@ -117,3 +137,184 @@ class Numeric::Builder : public GenericBuilder { } }; + +template +class Numeric::GenericParamBuilder +{ + Sink sink; + Numeric numeric; + std::string::size_type currlen; + std::string::size_type max; + + bool HasRoom(const std::string::size_type additional) const + { + return (currlen + additional <= max); + } + + public: + GenericParamBuilder(Sink s, unsigned int num, size_t additionalsize) + : sink(s) + , numeric(num) + , currlen(0) + , max(ServerInstance->Config->Limits.MaxLine - ServerInstance->Config->GetServerName().size() - additionalsize - 10) + { + } + + void AddStatic(const std::string& entry) + { + max -= (entry.length() + 1); + numeric.GetParams().push_back(entry); + } + + void Add(const std::string& entry) + { + if (!HasRoom(entry.size())) + Flush(); + + currlen += entry.size() + 1; + numeric.GetParams().push_back(entry); + } + + void Flush() + { + if ((!SendEmpty) && (IsEmpty())) + return; + + sink(numeric); + currlen = 0; + numeric.GetParams().erase(numeric.GetParams().begin() + NumStaticParams, numeric.GetParams().end()); + } + + bool IsEmpty() const { return (numeric.GetParams().size() <= NumStaticParams); } +}; + +template +class Numeric::ParamBuilder : public GenericParamBuilder +{ + public: + ParamBuilder(LocalUser* user, unsigned int num) + : ::Numeric::GenericParamBuilder(WriteNumericSink(user), num, user->nick.size()) + { + } +}; + +namespace Numerics +{ + class CannotSendTo; + class InvalidModeParameter; + class NoSuchChannel; + class NoSuchNick; +} + +/** Builder for the ERR_CANNOTSENDTOCHAN and ERR_CANTSENDTOUSER numerics. */ +class Numerics::CannotSendTo : public Numeric::Numeric +{ + public: + CannotSendTo(Channel* chan, const std::string& message) + : Numeric(ERR_CANNOTSENDTOCHAN) + { + push(chan->name); + push(message); + } + + CannotSendTo(Channel* chan, const std::string& what, ModeHandler* mh) + : Numeric(ERR_CANNOTSENDTOCHAN) + { + push(chan->name); + push(InspIRCd::Format("You cannot send %s to this channel whilst the +%c (%s) mode is set.", + what.c_str(), mh->GetModeChar(), mh->name.c_str())); + } + + CannotSendTo(Channel* chan, const std::string& what, char extban, const std::string& extbandesc) + : Numeric(ERR_CANNOTSENDTOCHAN) + { + push(chan->name); + push(InspIRCd::Format("You cannot send %s to this channel whilst %s %c: (%s) extban is set matching you.", + what.c_str(), strchr("AEIOUaeiou", extban) ? "an" : "a", extban, extbandesc.c_str())); + } + + CannotSendTo(User* user, const std::string& message) + : Numeric(ERR_CANTSENDTOUSER) + { + push(user->registered & REG_NICK ? user->nick : "*"); + push(message); + } + + CannotSendTo(User* user, const std::string& what, ModeHandler* mh, bool self = false) + : Numeric(ERR_CANTSENDTOUSER) + { + push(user->registered & REG_NICK ? user->nick : "*"); + push(InspIRCd::Format("You cannot send %s to this user whilst %s have the +%c (%s) mode set.", + what.c_str(), self ? "you" : "they", mh->GetModeChar(), mh->name.c_str())); + } +}; + +/* Builder for the ERR_INVALIDMODEPARAM numeric. */ +class Numerics::InvalidModeParameter : public Numeric::Numeric +{ + private: + void push_message(ModeHandler* mode, const std::string& message) + { + if (!message.empty()) + { + // The caller has specified their own message. + push(message); + return; + } + + const std::string& syntax = mode->GetSyntax(); + if (!syntax.empty()) + { + // If the mode has a syntax hint we include it in the message. + push(InspIRCd::Format("Invalid %s mode parameter. Syntax: %s.", mode->name.c_str(), syntax.c_str())); + } + else + { + // Otherwise, send it without. + push(InspIRCd::Format("Invalid %s mode parameter.", mode->name.c_str())); + } + } + + public: + InvalidModeParameter(Channel* chan, ModeHandler* mode, const std::string& parameter, const std::string& message = "") + : Numeric(ERR_INVALIDMODEPARAM) + { + push(chan->name); + push(mode->GetModeChar()); + push(parameter); + push_message(mode, message); + } + + InvalidModeParameter(User* user, ModeHandler* mode, const std::string& parameter, const std::string& message = "") + : Numeric(ERR_INVALIDMODEPARAM) + { + push(user->registered & REG_NICK ? user->nick : "*"); + push(mode->GetModeChar()); + push(parameter); + push_message(mode, message); + } +}; + +/** Builder for the ERR_NOSUCHCHANNEL numeric. */ +class Numerics::NoSuchChannel : public Numeric::Numeric +{ + public: + NoSuchChannel(const std::string& chan) + : Numeric(ERR_NOSUCHCHANNEL) + { + push(chan.empty() ? "*" : chan); + push("No such channel"); + } +}; + +/** Builder for the ERR_NOSUCHNICK numeric. */ +class Numerics::NoSuchNick : public Numeric::Numeric +{ + public: + NoSuchNick(const std::string& nick) + : Numeric(ERR_NOSUCHNICK) + { + push(nick.empty() ? "*" : nick); + push("No such nick"); + } +};