src/modules/m_spanningtree/capab.cpp

   1 /*
   2  * InspIRCd -- Internet Relay Chat Daemon
   3  *
   4  *   Copyright (C) 2009-2010 Daniel De Graaf <danieldg@inspircd.org>
   5  *   Copyright (C) 2008 Robin Burchell <robin+git@viroteck.net>
   6  *   Copyright (C) 2008 Craig Edwards <craigedwards@brainbox.cc>
   7  *
   8  * This file is part of InspIRCd.  InspIRCd is free software: you can
   9  * redistribute it and/or modify it under the terms of the GNU General Public
  10  * License as published by the Free Software Foundation, version 2.
  11  *
  12  * This program is distributed in the hope that it will be useful, but WITHOUT
  13  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
  14  * FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
  15  * details.
  16  *
  17  * You should have received a copy of the GNU General Public License
  18  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  19  */
  20
  21
  22 #include "inspircd.h"
  23
  24 #include "treeserver.h"
  25 #include "utils.h"
  26 #include "link.h"
  27 #include "main.h"
  28
  29 std::string TreeSocket::MyModules(int filter)
  30 {
  31         const ModuleManager::ModuleMap& modlist = ServerInstance->Modules->GetModules();
  32
  33         std::string capabilities;
  34         for (ModuleManager::ModuleMap::const_iterator i = modlist.begin(); i != modlist.end(); ++i)
  35         {
  36                 Version v = i->second->GetVersion();
  37                 if (!(v.Flags & filter))
  38                         continue;
  39
  40                 if (i != modlist.begin())
  41                         capabilities.push_back(' ');
  42                 capabilities.append(i->first);
  43                 if (!v.link_data.empty())
  44                 {
  45                         capabilities.push_back('=');
  46                         capabilities.append(v.link_data);
  47                 }
  48         }
  49         return capabilities;
  50 }
  51
  52 static std::string BuildModeList(ModeType type)
  53 {
  54         std::vector<std::string> modes;
  55         for(char c='A'; c <= 'z'; c++)
  56         {
  57                 ModeHandler* mh = ServerInstance->Modes->FindMode(c, type);
  58                 if (mh)
  59                 {
  60                         std::string mdesc = mh->name;
  61                         mdesc.push_back('=');
  62                         PrefixMode* pm = mh->IsPrefixMode();
  63                         if (pm)
  64                         {
  65                                 if (pm->GetPrefix())
  66                                         mdesc.push_back(pm->GetPrefix());
  67                         }
  68                         mdesc.push_back(mh->GetModeChar());
  69                         modes.push_back(mdesc);
  70                 }
  71         }
  72         sort(modes.begin(), modes.end());
  73         return irc::stringjoiner(modes);
  74 }
  75
  76 void TreeSocket::SendCapabilities(int phase)
  77 {
  78         if (capab->capab_phase >= phase)
  79                 return;
  80
  81         if (capab->capab_phase < 1 && phase >= 1)
  82                 WriteLine("CAPAB START " + ConvToStr(ProtocolVersion));
  83
  84         capab->capab_phase = phase;
  85         if (phase < 2)
  86                 return;
  87
  88         const char sep = ' ';
  89         irc::sepstream modulelist(MyModules(VF_COMMON), sep);
  90         irc::sepstream optmodulelist(MyModules(VF_OPTCOMMON), sep);
  91         /* Send module names, split at 509 length */
  92         std::string item;
  93         std::string line = "CAPAB MODULES :";
  94         while (modulelist.GetToken(item))
  95         {
  96                 if (line.length() + item.length() + 1 > 509)
  97                 {
  98                         this->WriteLine(line);
  99                         line = "CAPAB MODULES :";
 100                 }
 101
 102                 if (line != "CAPAB MODULES :")
 103                         line.push_back(sep);
 104
 105                 line.append(item);
 106         }
 107         if (line != "CAPAB MODULES :")
 108                 this->WriteLine(line);
 109
 110         line = "CAPAB MODSUPPORT :";
 111         while (optmodulelist.GetToken(item))
 112         {
 113                 if (line.length() + item.length() + 1 > 509)
 114                 {
 115                         this->WriteLine(line);
 116                         line = "CAPAB MODSUPPORT :";
 117                 }
 118
 119                 if (line != "CAPAB MODSUPPORT :")
 120                         line.push_back(sep);
 121
 122                 line.append(item);
 123         }
 124         if (line != "CAPAB MODSUPPORT :")
 125                 this->WriteLine(line);
 126
 127         WriteLine("CAPAB CHANMODES :" + BuildModeList(MODETYPE_CHANNEL));
 128         WriteLine("CAPAB USERMODES :" + BuildModeList(MODETYPE_USER));
 129
 130         std::string extra;
 131         /* Do we have sha256 available? If so, we send a challenge */
 132         if (ServerInstance->Modules->FindService(SERVICE_DATA, "hash/sha256"))
 133         {
 134                 SetOurChallenge(ServerInstance->GenRandomStr(20));
 135                 extra = " CHALLENGE=" + this->GetOurChallenge();
 136         }
 137
 138         // 2.0 needs this key
 139         if (proto_version == 1202)
 140                 extra.append(" PROTOCOL="+ConvToStr(ProtocolVersion));
 141
 142         this->WriteLine("CAPAB CAPABILITIES " /* Preprocessor does this one. */
 143                         ":NICKMAX="+ConvToStr(ServerInstance->Config->Limits.NickMax)+
 144                         " CHANMAX="+ConvToStr(ServerInstance->Config->Limits.ChanMax)+
 145                         " MAXMODES="+ConvToStr(ServerInstance->Config->Limits.MaxModes)+
 146                         " IDENTMAX="+ConvToStr(ServerInstance->Config->Limits.IdentMax)+
 147                         " MAXQUIT="+ConvToStr(ServerInstance->Config->Limits.MaxQuit)+
 148                         " MAXTOPIC="+ConvToStr(ServerInstance->Config->Limits.MaxTopic)+
 149                         " MAXKICK="+ConvToStr(ServerInstance->Config->Limits.MaxKick)+
 150                         " MAXGECOS="+ConvToStr(ServerInstance->Config->Limits.MaxGecos)+
 151                         " MAXAWAY="+ConvToStr(ServerInstance->Config->Limits.MaxAway)+
 152                         " MAXHOST="+ConvToStr(ServerInstance->Config->Limits.MaxHost)+
 153                         extra+
 154                         " PREFIX="+ServerInstance->Modes->BuildPrefixes()+
 155                         " CHANMODES="+ServerInstance->Modes->GiveModeList(MODETYPE_CHANNEL)+
 156                         " USERMODES="+ServerInstance->Modes->GiveModeList(MODETYPE_USER)
 157                         );
 158
 159         this->WriteLine("CAPAB END");
 160 }
 161
 162 /* Isolate and return the elements that are different between two comma seperated lists */
 163 void TreeSocket::ListDifference(const std::string &one, const std::string &two, char sep,
 164                 std::string& mleft, std::string& mright)
 165 {
 166         std::set<std::string> values;
 167         irc::sepstream sepleft(one, sep);
 168         irc::sepstream sepright(two, sep);
 169         std::string item;
 170         while (sepleft.GetToken(item))
 171         {
 172                 values.insert(item);
 173         }
 174         while (sepright.GetToken(item))
 175         {
 176                 if (!values.erase(item))
 177                 {
 178                         mright.push_back(sep);
 179                         mright.append(item);
 180                 }
 181         }
 182         for(std::set<std::string>::iterator i = values.begin(); i != values.end(); ++i)
 183         {
 184                 mleft.push_back(sep);
 185                 mleft.append(*i);
 186         }
 187 }
 188
 189 bool TreeSocket::Capab(const parameterlist &params)
 190 {
 191         if (params.size() < 1)
 192         {
 193                 this->SendError("Invalid number of parameters for CAPAB - Mismatched version");
 194                 return false;
 195         }
 196         if (params[0] == "START")
 197         {
 198                 capab->ModuleList.clear();
 199                 capab->OptModuleList.clear();
 200                 capab->CapKeys.clear();
 201                 if (params.size() > 1)
 202                         proto_version = ConvToInt(params[1]);
 203
 204                 if (proto_version < MinCompatProtocol)
 205                 {
 206                         SendError("CAPAB negotiation failed: Server is using protocol version " + (proto_version ? ConvToStr(proto_version) : "1201 or older")
 207                                 + " which is too old to link with this server (version " + ConvToStr(ProtocolVersion)
 208                                 + (ProtocolVersion != MinCompatProtocol ? ", links with " + ConvToStr(MinCompatProtocol) + " and above)" : ")"));
 209                         return false;
 210                 }
 211
 212                 // Special case, may be removed in the future
 213                 if (proto_version == 1203 || proto_version == 1204)
 214                 {
 215                         SendError("CAPAB negotiation failed: InspIRCd 2.1 beta is not supported");
 216                         return false;
 217                 }
 218
 219                 SendCapabilities(2);
 220         }
 221         else if (params[0] == "END")
 222         {
 223                 std::string reason;
 224                 /* Compare ModuleList and check CapKeys */
 225                 if ((this->capab->ModuleList != this->MyModules(VF_COMMON)) && (this->capab->ModuleList.length()))
 226                 {
 227                         std::string diffIneed, diffUneed;
 228                         ListDifference(this->capab->ModuleList, this->MyModules(VF_COMMON), ' ', diffIneed, diffUneed);
 229                         if (diffIneed.length() || diffUneed.length())
 230                         {
 231                                 reason = "Modules incorrectly matched on these servers.";
 232                                 if (diffIneed.length())
 233                                         reason += " Not loaded here:" + diffIneed;
 234                                 if (diffUneed.length())
 235                                         reason += " Not loaded there:" + diffUneed;
 236                                 this->SendError("CAPAB negotiation failed: "+reason);
 237                                 return false;
 238                         }
 239                 }
 240                 if (this->capab->OptModuleList != this->MyModules(VF_OPTCOMMON) && this->capab->OptModuleList.length())
 241                 {
 242                         std::string diffIneed, diffUneed;
 243                         ListDifference(this->capab->OptModuleList, this->MyModules(VF_OPTCOMMON), ' ', diffIneed, diffUneed);
 244                         if (diffIneed.length() || diffUneed.length())
 245                         {
 246                                 if (Utils->AllowOptCommon)
 247                                 {
 248                                         ServerInstance->SNO->WriteToSnoMask('l',
 249                                                 "Optional module lists do not match, some commands may not work globally.%s%s%s%s",
 250                                                 diffIneed.length() ? " Not loaded here:" : "", diffIneed.c_str(),
 251                                                 diffUneed.length() ? " Not loaded there:" : "", diffUneed.c_str());
 252                                 }
 253                                 else
 254                                 {
 255                                         reason = "Optional modules incorrectly matched on these servers, and options::allowmismatch not set.";
 256                                         if (diffIneed.length())
 257                                                 reason += " Not loaded here:" + diffIneed;
 258                                         if (diffUneed.length())
 259                                                 reason += " Not loaded there:" + diffUneed;
 260                                         this->SendError("CAPAB negotiation failed: "+reason);
 261                                         return false;
 262                                 }
 263                         }
 264                 }
 265
 266                 if(this->capab->CapKeys.find("PREFIX") != this->capab->CapKeys.end() && this->capab->CapKeys.find("PREFIX")->second != ServerInstance->Modes->BuildPrefixes())
 267                         reason = "One or more of the prefixes on the remote server are invalid on this server.";
 268
 269                 if (!capab->ChanModes.empty())
 270                 {
 271                         if (capab->ChanModes != BuildModeList(MODETYPE_CHANNEL))
 272                         {
 273                                 std::string diffIneed, diffUneed;
 274                                 ListDifference(capab->ChanModes, BuildModeList(MODETYPE_CHANNEL), ' ', diffIneed, diffUneed);
 275                                 if (diffIneed.length() || diffUneed.length())
 276                                 {
 277                                         reason = "Channel modes not matched on these servers.";
 278                                         if (diffIneed.length())
 279                                                 reason += " Not loaded here:" + diffIneed;
 280                                         if (diffUneed.length())
 281                                                 reason += " Not loaded there:" + diffUneed;
 282                                 }
 283                         }
 284                 }
 285                 else if (this->capab->CapKeys.find("CHANMODES") != this->capab->CapKeys.end())
 286                 {
 287                         if (this->capab->CapKeys.find("CHANMODES")->second != ServerInstance->Modes->GiveModeList(MODETYPE_CHANNEL))
 288                                 reason = "One or more of the channel modes on the remote server are invalid on this server.";
 289                 }
 290
 291                 if (!capab->UserModes.empty())
 292                 {
 293                         if (capab->UserModes != BuildModeList(MODETYPE_USER))
 294                         {
 295                                 std::string diffIneed, diffUneed;
 296                                 ListDifference(capab->UserModes, BuildModeList(MODETYPE_USER), ' ', diffIneed, diffUneed);
 297                                 if (diffIneed.length() || diffUneed.length())
 298                                 {
 299                                         reason = "User modes not matched on these servers.";
 300                                         if (diffIneed.length())
 301                                                 reason += " Not loaded here:" + diffIneed;
 302                                         if (diffUneed.length())
 303                                                 reason += " Not loaded there:" + diffUneed;
 304                                 }
 305                         }
 306                 }
 307                 else if (this->capab->CapKeys.find("USERMODES") != this->capab->CapKeys.end())
 308                 {
 309                         if (this->capab->CapKeys.find("USERMODES")->second != ServerInstance->Modes->GiveModeList(MODETYPE_USER))
 310                                 reason = "One or more of the user modes on the remote server are invalid on this server.";
 311                 }
 312
 313                 /* Challenge response, store their challenge for our password */
 314                 std::map<std::string,std::string>::iterator n = this->capab->CapKeys.find("CHALLENGE");
 315                 if ((n != this->capab->CapKeys.end()) && (ServerInstance->Modules->FindService(SERVICE_DATA, "hash/sha256")))
 316                 {
 317                         /* Challenge-response is on now */
 318                         this->SetTheirChallenge(n->second);
 319                         if (!this->GetTheirChallenge().empty() && (this->LinkState == CONNECTING))
 320                         {
 321                                 this->SendCapabilities(2);
 322                                 this->WriteLine("SERVER "+ServerInstance->Config->ServerName+" "+this->MakePass(capab->link->SendPass, capab->theirchallenge)+" 0 "+ServerInstance->Config->GetSID()+" :"+ServerInstance->Config->ServerDesc);
 323                         }
 324                 }
 325                 else
 326                 {
 327                         // They didn't specify a challenge or we don't have sha256, we use plaintext
 328                         if (this->LinkState == CONNECTING)
 329                         {
 330                                 this->SendCapabilities(2);
 331                                 this->WriteLine("SERVER "+ServerInstance->Config->ServerName+" "+capab->link->SendPass+" 0 "+ServerInstance->Config->GetSID()+" :"+ServerInstance->Config->ServerDesc);
 332                         }
 333                 }
 334
 335                 if (reason.length())
 336                 {
 337                         this->SendError("CAPAB negotiation failed: "+reason);
 338                         return false;
 339                 }
 340         }
 341         else if ((params[0] == "MODULES") && (params.size() == 2))
 342         {
 343                 if (!capab->ModuleList.length())
 344                 {
 345                         capab->ModuleList = params[1];
 346                 }
 347                 else
 348                 {
 349                         capab->ModuleList.push_back(' ');
 350                         capab->ModuleList.append(params[1]);
 351                 }
 352         }
 353         else if ((params[0] == "MODSUPPORT") && (params.size() == 2))
 354         {
 355                 if (!capab->OptModuleList.length())
 356                 {
 357                         capab->OptModuleList = params[1];
 358                 }
 359                 else
 360                 {
 361                         capab->OptModuleList.push_back(' ');
 362                         capab->OptModuleList.append(params[1]);
 363                 }
 364         }
 365         else if ((params[0] == "CHANMODES") && (params.size() == 2))
 366         {
 367                 capab->ChanModes = params[1];
 368         }
 369         else if ((params[0] == "USERMODES") && (params.size() == 2))
 370         {
 371                 capab->UserModes = params[1];
 372         }
 373         else if ((params[0] == "CAPABILITIES") && (params.size() == 2))
 374         {
 375                 irc::tokenstream capabs(params[1]);
 376                 std::string item;
 377                 while (capabs.GetToken(item))
 378                 {
 379                         /* Process each key/value pair */
 380                         std::string::size_type equals = item.find('=');
 381                         if (equals != std::string::npos)
 382                         {
 383                                 std::string var = item.substr(0, equals);
 384                                 std::string value = item.substr(equals+1, item.length());
 385                                 capab->CapKeys[var] = value;
 386                         }
 387                 }
 388         }
 389         return true;
 390 }