src/command_parse.cpp

   1 /*       +------------------------------------+
   2  *       | Inspire Internet Relay Chat Daemon |
   3  *       +------------------------------------+
   4  *
   5  *  InspIRCd: (C) 2002-2007 InspIRCd Development Team
   6  * See: http://www.inspircd.org/wiki/index.php/Credits
   7  *
   8  * This program is free but copyrighted software; see
   9  *            the file COPYING for details.
  10  *
  11  * ---------------------------------------------------
  12  */
  13
  14 #include "inspircd.h"
  15 #include "wildcard.h"
  16 #include "xline.h"
  17 #include "socketengine.h"
  18 #include "socket.h"
  19 #include "command_parse.h"
  20 #include "exitcodes.h"
  21
  22 /* Directory Searching for Unix-Only */
  23 #ifndef WIN32
  24 #include <dirent.h>
  25 #include <dlfcn.h>
  26 #endif
  27
  28 int InspIRCd::OperPassCompare(const char* data,const char* input, int tagnumber)
  29 {
  30         int MOD_RESULT = 0;
  31         FOREACH_RESULT_I(this,I_OnOperCompare,OnOperCompare(data, input, tagnumber))
  32         if (MOD_RESULT == 1)
  33                 return 0;
  34         if (MOD_RESULT == -1)
  35                 return 1;
  36         return strcmp(data,input);
  37 }
  38
  39 /* LoopCall is used to call a command classes handler repeatedly based on the contents of a comma seperated list.
  40  * There are two overriden versions of this method, one of which takes two potential lists and the other takes one.
  41  * We need a version which takes two potential lists for JOIN, because a JOIN may contain two lists of items at once,
  42  * the channel names and their keys as follows:
  43  * JOIN #chan1,#chan2,#chan3 key1,,key3
  44  * Therefore, we need to deal with both lists concurrently. The first instance of this method does that by creating
  45  * two instances of irc::commasepstream and reading them both together until the first runs out of tokens.
  46  * The second version is much simpler and just has the one stream to read, and is used in NAMES, WHOIS, PRIVMSG etc.
  47  * Both will only parse until they reach ServerInstance->Config->MaxTargets number of targets, to stop abuse via spam.
  48  */
  49 int CommandParser::LoopCall(User* user, Command* CommandObj, const char** parameters, int pcnt, unsigned int splithere, unsigned int extra)
  50 {
  51         /* First check if we have more than one item in the list, if we don't we return zero here and the handler
  52          * which called us just carries on as it was.
  53          */
  54         if (!strchr(parameters[splithere],','))
  55                 return 0;
  56
  57         /** Some lame ircds will weed out dupes using some shitty O(n^2) algorithm.
  58          * By using std::map (thanks for the idea w00t) we can cut this down a ton.
  59          * ...VOOODOOOO!
  60          */
  61         std::map<irc::string, bool> dupes;
  62
  63         /* Create two lists, one for channel names, one for keys
  64          */
  65         irc::commasepstream items1(parameters[splithere]);
  66         irc::commasepstream items2(parameters[extra]);
  67         std::string extrastuff;
  68         std::string item;
  69         unsigned int max = 0;
  70
  71         /* Attempt to iterate these lists and call the command objech
  72          * which called us, for every parameter pair until there are
  73          * no more left to parse.
  74          */
  75         while (items1.GetToken(item) && (max++ < ServerInstance->Config->MaxTargets))
  76         {
  77                 if (dupes.find(item.c_str()) == dupes.end())
  78                 {
  79                         const char* new_parameters[MAXPARAMETERS];
  80
  81                         for (int t = 0; (t < pcnt) && (t < MAXPARAMETERS); t++)
  82                                 new_parameters[t] = parameters[t];
  83
  84                         if (!items2.GetToken(extrastuff))
  85                                 extrastuff = "";
  86
  87                         new_parameters[splithere] = item.c_str();
  88                         new_parameters[extra] = extrastuff.c_str();
  89
  90                         CommandObj->Handle(new_parameters,pcnt,user);
  91
  92                         dupes[item.c_str()] = true;
  93                 }
  94         }
  95         return 1;
  96 }
  97
  98 int CommandParser::LoopCall(User* user, Command* CommandObj, const char** parameters, int pcnt, unsigned int splithere)
  99 {
 100         /* First check if we have more than one item in the list, if we don't we return zero here and the handler
 101          * which called us just carries on as it was.
 102          */
 103         if (!strchr(parameters[splithere],','))
 104                 return 0;
 105
 106         std::map<irc::string, bool> dupes;
 107
 108         /* Only one commasepstream here */
 109         irc::commasepstream items1(parameters[splithere]);
 110         std::string item;
 111         unsigned int max = 0;
 112
 113         /* Parse the commasepstream until there are no tokens remaining.
 114          * Each token we parse out, call the command handler that called us
 115          * with it
 116          */
 117         while (items1.GetToken(item) && (max++ < ServerInstance->Config->MaxTargets))
 118         {
 119                 if (dupes.find(item.c_str()) == dupes.end())
 120                 {
 121                         const char* new_parameters[MAXPARAMETERS];
 122
 123                         for (int t = 0; (t < pcnt) && (t < MAXPARAMETERS); t++)
 124                                 new_parameters[t] = parameters[t];
 125
 126                         new_parameters[splithere] = item.c_str();
 127
 128                         parameters[splithere] = item.c_str();
 129
 130                         /* Execute the command handler over and over. If someone pulls our user
 131                          * record out from under us (e.g. if we /kill a comma sep list, and we're
 132                          * in that list ourselves) abort if we're gone.
 133                          */
 134                         CommandObj->Handle(new_parameters,pcnt,user);
 135
 136                         dupes[item.c_str()] = true;
 137                 }
 138         }
 139         /* By returning 1 we tell our caller that nothing is to be done,
 140          * as all the previous calls handled the data. This makes the parent
 141          * return without doing any processing.
 142          */
 143         return 1;
 144 }
 145
 146 bool CommandParser::IsValidCommand(const std::string &commandname, int pcnt, User * user)
 147 {
 148         Commandable::iterator n = cmdlist.find(commandname);
 149
 150         if (n != cmdlist.end())
 151         {
 152                 if ((pcnt>=n->second->min_params) && (n->second->source != "<core>"))
 153                 {
 154                         if (IS_LOCAL(user) && n->second->flags_needed)
 155                         {
 156                                 if (user->IsModeSet(n->second->flags_needed))
 157                                 {
 158                                         return (user->HasPermission(commandname));
 159                                 }
 160                         }
 161                         else
 162                         {
 163                                 return true;
 164                         }
 165                 }
 166         }
 167         return false;
 168 }
 169
 170 Command* CommandParser::GetHandler(const std::string &commandname)
 171 {
 172         Commandable::iterator n = cmdlist.find(commandname);
 173         if (n != cmdlist.end())
 174                 return n->second;
 175
 176         return NULL;
 177 }
 178
 179 // calls a handler function for a command
 180
 181 CmdResult CommandParser::CallHandler(const std::string &commandname,const char** parameters, int pcnt, User *user)
 182 {
 183         Commandable::iterator n = cmdlist.find(commandname);
 184
 185         if (n != cmdlist.end())
 186         {
 187                 if (pcnt >= n->second->min_params)
 188                 {
 189                         bool bOkay = false;
 190
 191                         if (IS_LOCAL(user) && n->second->flags_needed)
 192                         {
 193                                 /* if user is local, and flags are needed .. */
 194
 195                                 if (user->IsModeSet(n->second->flags_needed))
 196                                 {
 197                                         /* if user has the flags, and now has the permissions, go ahead */
 198                                         if (user->HasPermission(commandname))
 199                                                 bOkay = true;
 200                                 }
 201                         }
 202                         else
 203                         {
 204                                 /* remote or no flags required anyway */
 205                                 bOkay = true;
 206                         }
 207
 208                         if (bOkay)
 209                         {
 210                                 return n->second->Handle(parameters,pcnt,user);
 211                         }
 212                 }
 213         }
 214         return CMD_INVALID;
 215 }
 216
 217 void CommandParser::ProcessCommand(User *user, std::string &cmd)
 218 {
 219         const char *command_p[MAXPARAMETERS];
 220         int items = 0;
 221         irc::tokenstream tokens(cmd);
 222         std::string command;
 223         tokens.GetToken(command);
 224
 225         /* A client sent a nick prefix on their command (ick)
 226          * rhapsody and some braindead bouncers do this --
 227          * the rfc says they shouldnt but also says the ircd should
 228          * discard it if they do.
 229          */
 230         if (*command.c_str() == ':')
 231                 tokens.GetToken(command);
 232
 233         while (tokens.GetToken(para[items]) && (items < MAXPARAMETERS))
 234         {
 235                 command_p[items] = para[items].c_str();
 236                 items++;
 237         }
 238
 239         std::transform(command.begin(), command.end(), command.begin(), ::toupper);
 240
 241         int MOD_RESULT = 0;
 242         FOREACH_RESULT(I_OnPreCommand,OnPreCommand(command,command_p,items,user,false,cmd));
 243         if (MOD_RESULT == 1) {
 244                 return;
 245         }
 246
 247         if (!user)
 248         {
 249                 /*
 250                  * before, we went and found the command even with no user.. seems nonsensical.
 251                  * I'm not entirely sure when we would be passed NULL, but let's handle it
 252                  * anyway, by dropping it like a hot potato. -- w00t
 253                  */
 254                 return;
 255         }
 256
 257         /* find the command, check it exists */
 258         Commandable::iterator cm = cmdlist.find(command);
 259
 260         if (cm == cmdlist.end())
 261         {
 262                 ServerInstance->stats->statsUnknown++;
 263                 user->WriteServ("421 %s %s :Unknown command",user->nick,command.c_str());
 264                 return;
 265         }
 266
 267         /* activity resets the ping pending timer */
 268         user->nping = ServerInstance->Time() + user->pingmax;
 269         if (cm->second->flags_needed)
 270         {
 271                 if (!user->IsModeSet(cm->second->flags_needed))
 272                 {
 273                         user->WriteServ("481 %s :Permission Denied - You do not have the required operator privileges",user->nick);
 274                         return;
 275                 }
 276                 if (!user->HasPermission(command))
 277                 {
 278                         user->WriteServ("481 %s :Permission Denied - Oper type %s does not have access to command %s",user->nick,user->oper,command.c_str());
 279                         return;
 280                 }
 281         }
 282         if ((user->registered == REG_ALL) && (!IS_OPER(user)) && (cm->second->IsDisabled()))
 283         {
 284                 /* command is disabled! */
 285                 user->WriteServ("421 %s %s :This command has been disabled.",user->nick,command.c_str());
 286                 ServerInstance->SNO->WriteToSnoMask('d', "%s denied for %s (%s@%s)",
 287                                 command.c_str(), user->nick, user->ident, user->host);
 288                 return;
 289         }
 290         if (items < cm->second->min_params)
 291         {
 292                 user->WriteServ("461 %s %s :Not enough parameters.", user->nick, command.c_str());
 293                 if ((ServerInstance->Config->SyntaxHints) && (user->registered == REG_ALL) && (cm->second->syntax.length()))
 294                         user->WriteServ("304 %s :SYNTAX %s %s", user->nick, cm->second->command.c_str(), cm->second->syntax.c_str());
 295                 return;
 296         }
 297         if ((user->registered != REG_ALL) && (!cm->second->WorksBeforeReg()))
 298         {
 299                 user->WriteServ("451 %s :You have not registered",command.c_str());
 300                 return;
 301         }
 302         else
 303         {
 304                 /* passed all checks.. first, do the (ugly) stats counters. */
 305                 cm->second->use_count++;
 306                 cm->second->total_bytes += cmd.length();
 307
 308                 /* module calls too */
 309                 int MOD_RESULT = 0;
 310                 FOREACH_RESULT(I_OnPreCommand,OnPreCommand(command,command_p,items,user,true,cmd));
 311                 if (MOD_RESULT == 1)
 312                         return;
 313
 314                 /*
 315                  * WARNING: be careful, the user may be deleted soon
 316                  */
 317                 CmdResult result = cm->second->Handle(command_p,items,user);
 318
 319                 FOREACH_MOD(I_OnPostCommand,OnPostCommand(command, command_p, items, user, result,cmd));
 320                 return;
 321         }
 322 }
 323
 324 bool CommandParser::RemoveCommands(const char* source)
 325 {
 326         Commandable::iterator i,safei;
 327         for (i = cmdlist.begin(); i != cmdlist.end(); i++)
 328         {
 329                 safei = i;
 330                 safei++;
 331                 if (safei != cmdlist.end())
 332                 {
 333                         RemoveCommand(safei, source);
 334                 }
 335         }
 336         safei = cmdlist.begin();
 337         if (safei != cmdlist.end())
 338         {
 339                 RemoveCommand(safei, source);
 340         }
 341         return true;
 342 }
 343
 344 void CommandParser::RemoveCommand(Commandable::iterator safei, const char* source)
 345 {
 346         Command* x = safei->second;
 347         if (x->source == std::string(source))
 348         {
 349                 cmdlist.erase(safei);
 350                 delete x;
 351         }
 352 }
 353
 354 void CommandParser::ProcessBuffer(std::string &buffer,User *user)
 355 {
 356         std::string::size_type a;
 357
 358         if (!user)
 359                 return;
 360
 361         while ((a = buffer.rfind("\n")) != std::string::npos)
 362                 buffer.erase(a);
 363         while ((a = buffer.rfind("\r")) != std::string::npos)
 364                 buffer.erase(a);
 365
 366         if (buffer.length())
 367         {
 368                 if (!user->muted)
 369                 {
 370                         ServerInstance->Log(DEBUG,"C[%d] I :%s %s",user->GetFd(), user->nick, buffer.c_str());
 371                         this->ProcessCommand(user,buffer);
 372                 }
 373         }
 374 }
 375
 376 bool CommandParser::CreateCommand(Command *f, void* so_handle)
 377 {
 378         if (so_handle)
 379         {
 380                 if (RFCCommands.find(f->command) == RFCCommands.end())
 381                         RFCCommands[f->command] = so_handle;
 382                 else
 383                 {
 384                         ServerInstance->Log(DEFAULT,"ERK! Somehow, we loaded a cmd_*.so file twice! Only the first instance is being recorded.");
 385                         return false;
 386                 }
 387         }
 388
 389         /* create the command and push it onto the table */
 390         if (cmdlist.find(f->command) == cmdlist.end())
 391         {
 392                 cmdlist[f->command] = f;
 393                 return true;
 394         }
 395         else return false;
 396 }
 397
 398 CommandParser::CommandParser(InspIRCd* Instance) : ServerInstance(Instance)
 399 {
 400         para.resize(128);
 401 }
 402
 403 bool CommandParser::FindSym(void** v, void* h, const std::string &name)
 404 {
 405         *v = dlsym(h, "init_command");
 406         const char* err = dlerror();
 407         if (err && !(*v))
 408         {
 409                 ServerInstance->Log(SPARSE, "Error loading core command %s: %s\n", name.c_str(), err);
 410                 return false;
 411         }
 412         return true;
 413 }
 414
 415 bool CommandParser::ReloadCommand(const char* cmd, User* user)
 416 {
 417         char filename[MAXBUF];
 418         char commandname[MAXBUF];
 419         int y = 0;
 420
 421         for (const char* x = cmd; *x; x++, y++)
 422                 commandname[y] = toupper(*x);
 423
 424         commandname[y] = 0;
 425
 426         SharedObjectList::iterator command = RFCCommands.find(commandname);
 427
 428         if (command != RFCCommands.end())
 429         {
 430                 Command* cmdptr = cmdlist.find(commandname)->second;
 431                 cmdlist.erase(cmdlist.find(commandname));
 432
 433                 for (char* x = commandname; *x; x++)
 434                         *x = tolower(*x);
 435
 436
 437                 delete cmdptr;
 438                 dlclose(command->second);
 439                 RFCCommands.erase(command);
 440
 441                 snprintf(filename, MAXBUF, "cmd_%s.so", commandname);
 442                 const char* err = this->LoadCommand(filename);
 443                 if (err)
 444                 {
 445                         if (user)
 446                                 user->WriteServ("NOTICE %s :*** Error loading 'cmd_%s.so': %s", user->nick, cmd, err);
 447                         return false;
 448                 }
 449
 450                 return true;
 451         }
 452
 453         return false;
 454 }
 455
 456 CmdResult cmd_reload::Handle(const char** parameters, int pcnt, User *user)
 457 {
 458         user->WriteServ("NOTICE %s :*** Reloading command '%s'",user->nick, parameters[0]);
 459         if (ServerInstance->Parser->ReloadCommand(parameters[0], user))
 460         {
 461                 user->WriteServ("NOTICE %s :*** Successfully reloaded command '%s'", user->nick, parameters[0]);
 462                 ServerInstance->WriteOpers("*** RELOAD: %s reloaded the '%s' command.", user->nick, parameters[0]);
 463                 return CMD_SUCCESS;
 464         }
 465         else
 466         {
 467                 user->WriteServ("NOTICE %s :*** Could not reload command '%s' -- fix this problem, then /REHASH as soon as possible!", user->nick, parameters[0]);
 468                 return CMD_FAILURE;
 469         }
 470 }
 471
 472 const char* CommandParser::LoadCommand(const char* name)
 473 {
 474         char filename[MAXBUF];
 475         void* h;
 476         Command* (*cmd_factory_func)(InspIRCd*);
 477
 478         /* Command already exists? Succeed silently - this is needed for REHASH */
 479         if (RFCCommands.find(name) != RFCCommands.end())
 480         {
 481                 ServerInstance->Log(DEBUG,"Not reloading command %s/%s, it already exists", LIBRARYDIR, name);
 482                 return NULL;
 483         }
 484
 485         snprintf(filename, MAXBUF, "%s/%s", LIBRARYDIR, name);
 486         h = dlopen(filename, RTLD_NOW | RTLD_GLOBAL);
 487
 488         if (!h)
 489         {
 490                 const char* n = dlerror();
 491                 ServerInstance->Log(SPARSE, "Error loading core command %s: %s", name, n);
 492                 return n;
 493         }
 494
 495         if (this->FindSym((void **)&cmd_factory_func, h, name))
 496         {
 497                 Command* newcommand = cmd_factory_func(ServerInstance);
 498                 this->CreateCommand(newcommand, h);
 499         }
 500         return NULL;
 501 }
 502
 503 void CommandParser::SetupCommandTable(User* user)
 504 {
 505         RFCCommands.clear();
 506
 507         if (!user)
 508         {
 509                 printf("\nLoading core commands");
 510                 fflush(stdout);
 511         }
 512
 513         DIR* library = opendir(LIBRARYDIR);
 514         if (library)
 515         {
 516                 dirent* entry = NULL;
 517                 while ((entry = readdir(library)))
 518                 {
 519                         if (match(entry->d_name, "cmd_*.so"))
 520                         {
 521                                 if (!user)
 522                                 {
 523                                         printf(".");
 524                                         fflush(stdout);
 525                                 }
 526                                 const char* err = this->LoadCommand(entry->d_name);
 527                                 if (err)
 528                                 {
 529                                         if (user)
 530                                         {
 531                                                 user->WriteServ("NOTICE %s :*** Failed to load core command %s: %s", user->nick, entry->d_name, err);
 532                                         }
 533                                         else
 534                                         {
 535                                                 printf("Error loading %s: %s", entry->d_name, err);
 536                                                 exit(EXIT_STATUS_BADHANDLER);
 537                                         }
 538                                 }
 539                         }
 540                 }
 541                 closedir(library);
 542                 if (!user)
 543                         printf("\n");
 544         }
 545
 546         if (cmdlist.find("RELOAD") == cmdlist.end())
 547                 this->CreateCommand(new cmd_reload(ServerInstance));
 548 }
 549
 550 int CommandParser::TranslateUIDs(TranslateType to, const std::string &source, std::string &dest)
 551 {
 552         User* user = NULL;
 553         std::string item;
 554         int translations = 0;
 555         dest.clear();
 556
 557         switch (to)
 558         {
 559                 case TR_NICK:
 560                         /* Translate single nickname */
 561                         user = ServerInstance->FindNick(source);
 562                         if (user)
 563                         {
 564                                 dest = user->uuid;
 565                                 translations++;
 566                         }
 567                         else
 568                                 dest = source;
 569                 break;
 570                 case TR_NICKLIST:
 571                 {
 572                         /* Translate comma seperated list of nicknames */
 573                         irc::commasepstream items(source);
 574                         while (items.GetToken(item))
 575                         {
 576                                 user = ServerInstance->FindNick(item);
 577                                 if (user)
 578                                 {
 579                                         dest.append(user->uuid);
 580                                         translations++;
 581                                 }
 582                                 else
 583                                         dest.append(item);
 584                                 dest.append(",");
 585                         }
 586                         if (!dest.empty())
 587                                 dest.erase(dest.end() - 1);
 588                 }
 589                 break;
 590                 case TR_SPACENICKLIST:
 591                 {
 592                         /* Translate space seperated list of nicknames */
 593                         irc::spacesepstream items(source);
 594                         while (items.GetToken(item))
 595                         {
 596                                 user = ServerInstance->FindNick(item);
 597                                 if (user)
 598                                 {
 599                                         dest.append(user->uuid);
 600                                         translations++;
 601                                 }
 602                                 else
 603                                         dest.append(item);
 604                                 dest.append(" ");
 605                         }
 606                         if (!dest.empty())
 607                                 dest.erase(dest.end() - 1);
 608                 }
 609                 break;
 610                 case TR_END:
 611                 case TR_TEXT:
 612                 default:
 613                         /* Do nothing */
 614                         dest = source;
 615                 break;
 616         }
 617
 618         return translations;
 619 }
 620