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