src/inspircd_io.cpp

   1 /*       +------------------------------------+
   2  *       | Inspire Internet Relay Chat Daemon |
   3  *       +------------------------------------+
   4  *
   5  *  Inspire is copyright (C) 2002-2004 ChatSpike-Dev.
   6  *                       E-mail:
   7  *                <brain@chatspike.net>
   8  *                <Craig@chatspike.net>
   9  *
  10  * Written by Craig Edwards, Craig McLure, and others.
  11  * This program is free but copyrighted software; see
  12  *            the file COPYING for details.
  13  *
  14  * ---------------------------------------------------
  15  */
  16
  17 using namespace std;
  18
  19 #include "inspircd_config.h"
  20 #include <sys/time.h>
  21 #include <sys/resource.h>
  22 #include <sys/types.h>
  23 #include <string>
  24 #include <unistd.h>
  25 #include <sstream>
  26 #include <iostream>
  27 #include <fstream>
  28 #include "inspircd.h"
  29 #include "inspircd_io.h"
  30 #include "inspircd_util.h"
  31 #include "inspstring.h"
  32 #include "helperfuncs.h"
  33
  34 extern FILE *log_file;
  35 extern int boundPortCount;
  36 extern int openSockfd[MAXSOCKS];
  37 extern time_t TIME;
  38 extern bool unlimitcore;
  39 extern int MaxConn;
  40 std::vector<std::string> include_stack;
  41
  42 void WriteOpers(char* text, ...);
  43
  44 void Exit (int status)
  45 {
  46         if (log_file)
  47                 fclose(log_file);
  48         send_error("Server shutdown.");
  49         exit (status);
  50 }
  51
  52 void Killed(int status)
  53 {
  54         if (log_file)
  55                 fclose(log_file);
  56         send_error("Server terminated.");
  57         exit(status);
  58 }
  59
  60 void Rehash(int status)
  61 {
  62         WriteOpers("Rehashing config file %s due to SIGHUP",CONFIG_FILE);
  63         ReadConfig(false,NULL);
  64 }
  65
  66
  67
  68 void Start (void)
  69 {
  70         printf("\033[1;32mInspire Internet Relay Chat Server, compiled %s at %s\n",__DATE__,__TIME__);
  71         printf("(C) ChatSpike Development team.\033[0m\n\n");
  72         printf("Developers:\033[1;32m     Brain, FrostyCoolSlug\033[0m\n");
  73         printf("Documentation:\033[1;32m  FrostyCoolSlug, w00t\033[0m\n");
  74         printf("Testers:\033[1;32m        typobox43, piggles, Lord_Zathras, CC\033[0m\n");
  75         printf("Name concept:\033[1;32m   Lord_Zathras\033[0m\n\n");
  76 }
  77
  78 void WritePID(std::string filename)
  79 {
  80         ofstream outfile(filename.c_str());
  81         if (outfile.is_open())
  82         {
  83                 outfile << getpid();
  84                 outfile.close();
  85         }
  86         else
  87         {
  88                 printf("Failed to write PID-file '%s', exiting.\n",filename.c_str());
  89                 log(DEFAULT,"Failed to write PID-file '%s', exiting.",filename.c_str());
  90                 Exit(0);
  91         }
  92 }
  93
  94
  95 int DaemonSeed (void)
  96 {
  97         int childpid;
  98         signal (SIGALRM, SIG_IGN);
  99         signal (SIGHUP, Rehash);
 100         signal (SIGPIPE, SIG_IGN);
 101         signal (SIGTERM, Exit);
 102         signal (SIGSEGV, Error);
 103         if ((childpid = fork ()) < 0)
 104                 return (ERROR);
 105         else if (childpid > 0)
 106                 exit (0);
 107         setsid ();
 108         umask (007);
 109         printf("InspIRCd Process ID: \033[1;32m%lu\033[0m\n",(unsigned long)getpid());
 110         freopen("/dev/null","w",stdout);
 111         freopen("/dev/null","w",stderr);
 112
 113         setpriority(PRIO_PROCESS,(int)getpid(),15);
 114
 115         if (unlimitcore)
 116         {
 117                 rlimit rl;
 118                 if (getrlimit(RLIMIT_CORE, &rl) == -1)
 119                 {
 120                         log(DEFAULT,"Failed to getrlimit()!");
 121                         return(FALSE);
 122                 }
 123                 else
 124                 {
 125                         rl.rlim_cur = rl.rlim_max;
 126                         if (setrlimit(RLIMIT_CORE, &rl) == -1)
 127                                 log(DEFAULT,"setrlimit() failed, cannot increase coredump size.");
 128                 }
 129         }
 130
 131         return (TRUE);
 132 }
 133
 134
 135 /* Make Sure Modules Are Avaliable!
 136  * (BugFix By Craig.. See? I do work! :p)
 137  * Modified by brain, requires const char*
 138  * to work with other API functions
 139  */
 140
 141 bool FileExists (const char* file)
 142 {
 143         FILE *input;
 144         if ((input = fopen (file, "r")) == NULL)
 145         {
 146                 return(false);
 147         }
 148         else
 149         {
 150                 fclose (input);
 151                 return(true);
 152         }
 153 }
 154
 155 /* ConfProcess does the following things to a config line in the following order:
 156  *
 157  * Processes the line for syntax errors as shown below
 158  *      (1) Line void of quotes or equals (a malformed, illegal tag format)
 159  *      (2) Odd number of quotes on the line indicating a missing quote
 160  *      (3) number of equals signs not equal to number of quotes / 2 (missing an equals sign)
 161  *      (4) Spaces between the opening bracket (<) and the keyword
 162  *      (5) Spaces between a keyword and an equals sign
 163  *      (6) Spaces between an equals sign and a quote
 164  * Removes trailing spaces
 165  * Removes leading spaces
 166  * Converts tabs to spaces
 167  * Turns multiple spaces that are outside of quotes into single spaces
 168  */
 169
 170 std::string ConfProcess(char* buffer, long linenumber, std::stringstream* errorstream, bool &error, std::string filename)
 171 {
 172         long number_of_quotes = 0;
 173         long number_of_equals = 0;
 174         bool has_open_bracket = false;
 175         bool in_quotes = false;
 176         error = false;
 177         if (!buffer)
 178         {
 179                 return "";
 180         }
 181         // firstly clean up the line by stripping spaces from the start and end and converting tabs to spaces
 182         for (int d = 0; d < strlen(buffer); d++)
 183                 if ((buffer[d]) == 9)
 184                         buffer[d] = ' ';
 185         while ((buffer[0] == ' ') && (strlen(buffer)>0)) buffer++;
 186         while ((buffer[strlen(buffer)-1] == ' ') && (strlen(buffer)>0)) buffer[strlen(buffer)-1] = '\0';
 187         // empty lines are syntactically valid
 188         if (!strcmp(buffer,""))
 189                 return "";
 190         else if (buffer[0] == '#')
 191                 return "";
 192         for (int c = 0; c < strlen(buffer); c++)
 193         {
 194                 // convert all spaces that are OUTSIDE quotes into hardspace (0xA0) as this will make them easier to
 195                 // search and replace later :)
 196                 if ((!in_quotes) && (buffer[c] == ' '))
 197                         buffer[c] = '\xA0';
 198                 if ((buffer[c] == '<') && (!in_quotes))
 199                 {
 200                         has_open_bracket = true;
 201                         if (strlen(buffer) == 1)
 202                         {
 203                                 *errorstream << "Tag without identifier at " << filename << ":" << linenumber << endl;
 204                                 error = true;
 205                                 return "";
 206                         }
 207                         else if ((tolower(buffer[c+1]) < 'a') || (tolower(buffer[c+1]) > 'z'))
 208                         {
 209                                 *errorstream << "Invalid characters in identifier at " << filename << ":" << linenumber << endl;
 210                                 error = true;
 211                                 return "";
 212                         }
 213                 }
 214                 if (buffer[c] == '"')
 215                 {
 216                         number_of_quotes++;
 217                         in_quotes = (!in_quotes);
 218                 }
 219                 if ((buffer[c] == '=') && (!in_quotes))
 220                 {
 221                         number_of_equals++;
 222                         if (strlen(buffer) == c)
 223                         {
 224                                 *errorstream << "Variable without a value at " << filename << ":" << linenumber << endl;
 225                                 error = true;
 226                                 return "";
 227                         }
 228                         else if (buffer[c+1] != '"')
 229                         {
 230                                 *errorstream << "Variable name not followed immediately by its value at " << filename << ":" << linenumber << endl;
 231                                 error = true;
 232                                 return "";
 233                         }
 234                         else if (!c)
 235                         {
 236                                 *errorstream << "Value without a variable (line starts with '=') at " << filename << ":" << linenumber << endl;
 237                                 error = true;
 238                                 return "";
 239                         }
 240                         else if (buffer[c-1] == '\xA0')
 241                         {
 242                                 *errorstream << "Variable name not followed immediately by its value at " << filename << ":" << linenumber << endl;
 243                                 error = true;
 244                                 return "";
 245                         }
 246                 }
 247         }
 248         // no quotes, and no equals. something freaky.
 249         if ((!number_of_quotes) || (!number_of_equals) && (strlen(buffer)>2) && (buffer[0]=='<'))
 250         {
 251                 *errorstream << "Malformed tag at " << filename << ":" << linenumber << endl;
 252                 error = true;
 253                 return "";
 254         }
 255         // odd number of quotes. thats just wrong.
 256         if ((number_of_quotes % 2) != 0)
 257         {
 258                 *errorstream << "Missing \" at " << filename << ":" << linenumber << endl;
 259                 error = true;
 260                 return "";
 261         }
 262         if (number_of_equals < (number_of_quotes/2))
 263         {
 264                 *errorstream << "Missing '=' at " << filename << ":" << linenumber << endl;
 265         }
 266         if (number_of_equals > (number_of_quotes/2))
 267         {
 268                 *errorstream << "Too many '=' at " << filename << ":" << linenumber << endl;
 269         }
 270
 271         std::string parsedata = buffer;
 272         // turn multispace into single space
 273         while (parsedata.find("\xA0\xA0") != std::string::npos)
 274         {
 275                 parsedata.erase(parsedata.find("\xA0\xA0"),1);
 276         }
 277
 278         // turn our hardspace back into softspace
 279         for (int d = 0; d < parsedata.length(); d++)
 280         {
 281                 if (parsedata[d] == '\xA0')
 282                         parsedata[d] = ' ';
 283         }
 284
 285         // and we're done, the line is fine!
 286         return parsedata;
 287 }
 288
 289 bool LoadConf(const char* filename, std::stringstream *target, std::stringstream* errorstream)
 290 {
 291         target->str("");
 292         errorstream->str("");
 293         long linenumber = 1;
 294         // first, check that the file exists before we try to do anything with it
 295         if (!FileExists(filename))
 296         {
 297                 *errorstream << "File " << filename << " not found." << endl;
 298                 return false;
 299         }
 300         // Fix the chmod of the file to restrict it to the current user and group
 301         chmod(filename,0600);
 302         for (int t = 0; t < include_stack.size(); t++)
 303         {
 304                 if (std::string(filename) == include_stack[t])
 305                 {
 306                         *errorstream << "File " << filename << " is included recursively (looped inclusion)." << endl;
 307                         return false;
 308                 }
 309         }
 310         include_stack.push_back(filename);
 311         // now open it
 312         FILE* conf = fopen(filename,"r");
 313         char buffer[MAXBUF];
 314         if (conf)
 315         {
 316                 while (!feof(conf))
 317                 {
 318                         if (fgets(buffer, MAXBUF, conf))
 319                         {
 320                                 if ((!feof(conf)) && (buffer) && (strlen(buffer)))
 321                                 {
 322                                         if ((buffer[0] != '#') && (buffer[0] != '\r')  && (buffer[0] != '\n'))
 323                                         {
 324                                                 if (!strncmp(buffer,"<include file=\"",15))
 325                                                 {
 326                                                         char* buf = buffer;
 327                                                         char confpath[10240],newconf[10240];
 328                                                         // include file directive
 329                                                         buf += 15;      // advance to filename
 330                                                         for (int j = 0; j < strlen(buf); j++)
 331                                                         {
 332                                                                 if (buf[j] == '\\')
 333                                                                         buf[j] = '/';
 334                                                                 if (buf[j] == '"')
 335                                                                 {
 336                                                                         buf[j] = '\0';
 337                                                                         break;
 338                                                                 }
 339                                                         }
 340                                                         log(DEFAULT,"Opening included file '%s'",buf);
 341                                                         if (*buf != '/')
 342                                                         {
 343                                                                 strlcpy(confpath,CONFIG_FILE,10240);
 344                                                                 if (strstr(confpath,"/inspircd.conf"))
 345                                                                 {
 346                                                                         // leaves us with just the path
 347                                                                         *(strstr(confpath,"/inspircd.conf")) = '\0';
 348                                                                 }
 349                                                                 snprintf(newconf,10240,"%s/%s",confpath,buf);
 350                                                         }
 351                                                         else snprintf(newconf,10240,"%s",buf);
 352                                                         std::stringstream merge(stringstream::in | stringstream::out);
 353                                                         // recursively call LoadConf and get the new data, use the same errorstream
 354                                                         if (LoadConf(newconf, &merge, errorstream))
 355                                                         {
 356                                                                 // append to the end of the file
 357                                                                 std::string newstuff = merge.str();
 358                                                                 *target << newstuff;
 359                                                         }
 360                                                         else
 361                                                         {
 362                                                                 // the error propogates up to its parent recursively
 363                                                                 // causing the config reader to bail at the top level.
 364                                                                 fclose(conf);
 365                                                                 return false;
 366                                                         }
 367                                                 }
 368                                                 else
 369                                                 {
 370                                                         bool error = false;
 371                                                         std::string data = ConfProcess(buffer,linenumber++,errorstream,error,filename);
 372                                                         if (error)
 373                                                         {
 374                                                                 return false;
 375                                                         }
 376                                                         *target << data;
 377                                                 }
 378                                         }
 379                                         else linenumber++;
 380                                 }
 381                         }
 382                 }
 383                 fclose(conf);
 384         }
 385         target->seekg(0);
 386         return true;
 387 }
 388
 389 /* Counts the number of tags of a certain type within the config file, e.g. to enumerate opers */
 390
 391 int EnumConf(std::stringstream *config, const char* tag)
 392 {
 393         int ptr = 0;
 394         char buffer[MAXBUF], c_tag[MAXBUF], c, lastc;
 395         int in_token, in_quotes, tptr, idx = 0;
 396
 397         const char* buf = config->str().c_str();
 398         long bptr = 0;
 399         long len = strlen(buf);
 400
 401         ptr = 0;
 402         in_token = 0;
 403         in_quotes = 0;
 404         lastc = '\0';
 405         while (bptr<len)
 406         {
 407                 lastc = c;
 408                 c = buf[bptr++];
 409                 if ((c == '#') && (lastc == '\n'))
 410                 {
 411                         while ((c != '\n') && (bptr<len))
 412                         {
 413                                 lastc = c;
 414                                 c = buf[bptr++];
 415                         }
 416                 }
 417                 if ((c == '<') && (!in_quotes))
 418                 {
 419                         tptr = 0;
 420                         in_token = 1;
 421                         do {
 422                                 c = buf[bptr++];
 423                                 if (c != ' ')
 424                                 {
 425                                         c_tag[tptr++] = c;
 426                                         c_tag[tptr] = '\0';
 427                                 }
 428                         } while (c != ' ');
 429                 }
 430                 if (c == '"')
 431                 {
 432                         in_quotes = (!in_quotes);
 433                 }
 434                 if ((c == '>') && (!in_quotes))
 435                 {
 436                         in_token = 0;
 437                         if (!strcmp(c_tag,tag))
 438                         {
 439                                 /* correct tag, but wrong index */
 440                                 idx++;
 441                         }
 442                         c_tag[0] = '\0';
 443                         buffer[0] = '\0';
 444                         ptr = 0;
 445                         tptr = 0;
 446                 }
 447                 if (c != '>')
 448                 {
 449                         if ((in_token) && (c != '\n') && (c != '\r'))
 450                         {
 451                                 buffer[ptr++] = c;
 452                                 buffer[ptr] = '\0';
 453                         }
 454                 }
 455         }
 456         return idx;
 457 }
 458
 459 /* Counts the number of values within a certain tag */
 460
 461 int EnumValues(std::stringstream *config, const char* tag, int index)
 462 {
 463         int ptr = 0;
 464         char buffer[MAXBUF], c_tag[MAXBUF], c, lastc;
 465         int in_token, in_quotes, tptr, idx = 0;
 466
 467         bool correct_tag = false;
 468         int num_items = 0;
 469
 470         const char* buf = config->str().c_str();
 471         long bptr = 0;
 472         long len = strlen(buf);
 473
 474         ptr = 0;
 475         in_token = 0;
 476         in_quotes = 0;
 477         lastc = '\0';
 478         while (bptr<len)
 479         {
 480                 lastc = c;
 481                 c = buf[bptr++];
 482                 if ((c == '#') && (lastc == '\n'))
 483                 {
 484                         while ((c != '\n') && (bptr<len))
 485                         {
 486                                 lastc = c;
 487                                 c = buf[bptr++];
 488                         }
 489                 }
 490                 if ((c == '<') && (!in_quotes))
 491                 {
 492                         tptr = 0;
 493                         in_token = 1;
 494                         do {
 495                                 c = buf[bptr++];
 496                                 if (c != ' ')
 497                                 {
 498                                         c_tag[tptr++] = c;
 499                                         c_tag[tptr] = '\0';
 500
 501                                         if ((!strcmp(c_tag,tag)) && (idx == index))
 502                                         {
 503                                                 correct_tag = true;
 504                                         }
 505                                 }
 506                         } while (c != ' ');
 507                 }
 508                 if (c == '"')
 509                 {
 510                         in_quotes = (!in_quotes);
 511                 }
 512
 513                 if ( (correct_tag) && (!in_quotes) && ( (c == ' ') || (c == '\n') || (c == '\r') ) )
 514                 {
 515                         num_items++;
 516                 }
 517                 if ((c == '>') && (!in_quotes))
 518                 {
 519                         in_token = 0;
 520                         if (correct_tag)
 521                                 correct_tag = false;
 522                         if (!strcmp(c_tag,tag))
 523                         {
 524                                 /* correct tag, but wrong index */
 525                                 idx++;
 526                         }
 527                         c_tag[0] = '\0';
 528                         buffer[0] = '\0';
 529                         ptr = 0;
 530                         tptr = 0;
 531                 }
 532                 if (c != '>')
 533                 {
 534                         if ((in_token) && (c != '\n') && (c != '\r'))
 535                         {
 536                                 buffer[ptr++] = c;
 537                                 buffer[ptr] = '\0';
 538                         }
 539                 }
 540         }
 541         return num_items+1;
 542 }
 543
 544
 545
 546 int ConfValueEnum(char* tag, std::stringstream* config)
 547 {
 548         return EnumConf(config,tag);
 549 }
 550
 551
 552
 553 /* Retrieves a value from the config file. If there is more than one value of the specified
 554  * key and section (e.g. for opers etc) then the index value specifies which to retreive, e.g.
 555  *
 556  * ConfValue("oper","name",2,result);
 557  */
 558
 559 int ReadConf(std::stringstream *config, const char* tag, const char* var, int index, char *result)
 560 {
 561         int ptr = 0;
 562         char buffer[65535], c_tag[MAXBUF], c, lastc;
 563         int in_token, in_quotes, tptr, j, idx = 0;
 564         char* key;
 565
 566         const char* buf = config->str().c_str();
 567         long bptr = 0;
 568         long len = config->str().length();
 569
 570         ptr = 0;
 571         in_token = 0;
 572         in_quotes = 0;
 573         lastc = '\0';
 574         c_tag[0] = '\0';
 575         buffer[0] = '\0';
 576         while (bptr<len)
 577         {
 578                 lastc = c;
 579                 c = buf[bptr++];
 580                 // FIX: Treat tabs as spaces
 581                 if (c == 9)
 582                         c = 32;
 583                 if ((c == '<') && (!in_quotes))
 584                 {
 585                         tptr = 0;
 586                         in_token = 1;
 587                         do {
 588                                 c = buf[bptr++];
 589                                 if (c != ' ')
 590                                 {
 591                                         c_tag[tptr++] = c;
 592                                         c_tag[tptr] = '\0';
 593                                 }
 594                         // FIX: Tab can follow a tagname as well as space.
 595                         } while ((c != ' ') && (c != 9));
 596                 }
 597                 if (c == '"')
 598                 {
 599                         in_quotes = (!in_quotes);
 600                 }
 601                 if ((c == '>') && (!in_quotes))
 602                 {
 603                         in_token = 0;
 604                         if (idx == index)
 605                         {
 606                                 if (!strcmp(c_tag,tag))
 607                                 {
 608                                         if ((buffer) && (c_tag) && (var))
 609                                         {
 610                                                 key = strstr(buffer,var);
 611                                                 if (!key)
 612                                                 {
 613                                                         /* value not found in tag */
 614                                                         strcpy(result,"");
 615                                                         return 0;
 616                                                 }
 617                                                 else
 618                                                 {
 619                                                         key+=strlen(var);
 620                                                         while (key[0] !='"')
 621                                                         {
 622                                                                 if (!strlen(key))
 623                                                                 {
 624                                                                         /* missing quote */
 625                                                                         strcpy(result,"");
 626                                                                         return 0;
 627                                                                 }
 628                                                                 key++;
 629                                                         }
 630                                                         key++;
 631                                                         for (j = 0; j < strlen(key); j++)
 632                                                         {
 633                                                                 if (key[j] == '"')
 634                                                                 {
 635                                                                         key[j] = '\0';
 636                                                                 }
 637                                                         }
 638                                                         strlcpy(result,key,MAXBUF);
 639                                                         return 1;
 640                                                 }
 641                                         }
 642                                 }
 643                         }
 644                         if (!strcmp(c_tag,tag))
 645                         {
 646                                 /* correct tag, but wrong index */
 647                                 idx++;
 648                         }
 649                         c_tag[0] = '\0';
 650                         buffer[0] = '\0';
 651                         ptr = 0;
 652                         tptr = 0;
 653                 }
 654                 if (c != '>')
 655                 {
 656                         if ((in_token) && (c != '\n') && (c != '\r'))
 657                         {
 658                                 buffer[ptr++] = c;
 659                                 buffer[ptr] = '\0';
 660                         }
 661                 }
 662         }
 663         strcpy(result,""); // value or its tag not found at all
 664         return 0;
 665 }
 666
 667
 668
 669 int ConfValue(char* tag, char* var, int index, char *result,std::stringstream *config)
 670 {
 671         ReadConf(config, tag, var, index, result);
 672         return 0;
 673 }
 674
 675
 676
 677 // This will bind a socket to a port. It works for UDP/TCP
 678 int BindSocket (int sockfd, struct sockaddr_in client, struct sockaddr_in server, int port, char* addr)
 679 {
 680         memset((char *)&server,0,sizeof(server));
 681         struct in_addr addy;
 682         inet_aton(addr,&addy);
 683         server.sin_family = AF_INET;
 684         if (!strcmp(addr,""))
 685         {
 686                 server.sin_addr.s_addr = htonl(INADDR_ANY);
 687         }
 688         else
 689         {
 690                 server.sin_addr = addy;
 691         }
 692         server.sin_port = htons(port);
 693         if (bind(sockfd,(struct sockaddr*)&server,sizeof(server))<0)
 694         {
 695                 return(ERROR);
 696         }
 697         else
 698         {
 699                 listen(sockfd, MaxConn);
 700                 return(TRUE);
 701         }
 702 }
 703
 704
 705 // Open a TCP Socket
 706 int OpenTCPSocket (void)
 707 {
 708         int sockfd;
 709         int on = 1;
 710         struct linger linger = { 0 };
 711
 712         if ((sockfd = socket (AF_INET, SOCK_STREAM, 0)) < 0)
 713                 return (ERROR);
 714         else
 715         {
 716                 setsockopt(sockfd, SOL_SOCKET, SO_REUSEADDR, (const char*)&on, sizeof(on));
 717                 /* This is BSD compatible, setting l_onoff to 0 is *NOT* http://web.irc.org/mla/ircd-dev/msg02259.html */
 718                 linger.l_onoff = 1;
 719                 linger.l_linger = 1;
 720                 setsockopt(sockfd, SOL_SOCKET, SO_LINGER, (const char*)&linger,sizeof(linger));
 721                 return (sockfd);
 722         }
 723 }
 724