]> git.netwichtig.de Git - user/henk/code/inspircd.git/blob - src/inspircd_io.cpp
Attempted config fix for end of line missing
[user/henk/code/inspircd.git] / 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 (unsigned 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
188         // empty lines are syntactically valid, as are comments
189         if (!(*buffer) || buffer[0] == '#')
190                 return "";
191
192         for (unsigned 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 (unsigned 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 /* A special version of fgets that will read \r, \n, \r\n or even \n\r, it doesnt care :-) */
290
291 int fgets_safe(char* buffer, size_t maxsize, FILE* &file)
292 {
293         char c_read = '\0';
294         unsigned int bufptr = 0;
295         while ((!feof(file)) && (c_read != '\n') && (c_read != '\r') && (bufptr < maxsize))
296         {
297                 c_read = fgetc(file);
298                 if ((c_read != '\n') && (c_read != '\r'))
299                         buffer[bufptr++] = c_read;
300         }
301         buffer[bufptr] = '\0';
302         return bufptr;
303 }
304
305 bool LoadConf(const char* filename, std::stringstream *target, std::stringstream* errorstream)
306 {
307         target->str("");
308         errorstream->str("");
309         long linenumber = 1;
310         // first, check that the file exists before we try to do anything with it
311         if (!FileExists(filename))
312         {
313                 *errorstream << "File " << filename << " not found." << endl;
314                 return false;
315         }
316         // Fix the chmod of the file to restrict it to the current user and group
317         chmod(filename,0600);
318         for (unsigned int t = 0; t < include_stack.size(); t++)
319         {
320                 if (std::string(filename) == include_stack[t])
321                 {
322                         *errorstream << "File " << filename << " is included recursively (looped inclusion)." << endl;
323                         return false;
324                 }
325         }
326         include_stack.push_back(filename);
327         // now open it
328         FILE* conf = fopen(filename,"r");
329         char buffer[MAXBUF];
330         if (conf)
331         {
332                 while (!feof(conf))
333                 {
334                         if (fgets_safe(buffer, MAXBUF, conf))
335                         {
336                                 if ((!feof(conf)) && (buffer) && (strlen(buffer)))
337                                 {
338                                         if ((buffer[0] != '#') && (buffer[0] != '\r')  && (buffer[0] != '\n'))
339                                         {
340                                                 if (!strncmp(buffer,"<include file=\"",15))
341                                                 {
342                                                         char* buf = buffer;
343                                                         char confpath[10240],newconf[10240];
344                                                         // include file directive
345                                                         buf += 15;      // advance to filename
346                                                         for (unsigned int j = 0; j < strlen(buf); j++)
347                                                         {
348                                                                 if (buf[j] == '\\')
349                                                                         buf[j] = '/';
350                                                                 if (buf[j] == '"')
351                                                                 {
352                                                                         buf[j] = '\0';
353                                                                         break;
354                                                                 }
355                                                         }
356                                                         log(DEFAULT,"Opening included file '%s'",buf);
357                                                         if (*buf != '/')
358                                                         {
359                                                                 strlcpy(confpath,CONFIG_FILE,10240);
360                                                                 if (strstr(confpath,"/inspircd.conf"))
361                                                                 {
362                                                                         // leaves us with just the path
363                                                                         *(strstr(confpath,"/inspircd.conf")) = '\0';
364                                                                 }
365                                                                 snprintf(newconf,10240,"%s/%s",confpath,buf);
366                                                         }
367                                                         else snprintf(newconf,10240,"%s",buf);
368                                                         std::stringstream merge(stringstream::in | stringstream::out);
369                                                         // recursively call LoadConf and get the new data, use the same errorstream
370                                                         if (LoadConf(newconf, &merge, errorstream))
371                                                         {
372                                                                 // append to the end of the file
373                                                                 std::string newstuff = merge.str();
374                                                                 *target << newstuff;
375                                                         }
376                                                         else
377                                                         {
378                                                                 // the error propogates up to its parent recursively
379                                                                 // causing the config reader to bail at the top level.
380                                                                 fclose(conf);
381                                                                 return false;
382                                                         }
383                                                 }
384                                                 else
385                                                 {
386                                                         bool error = false;
387                                                         std::string data = ConfProcess(buffer,linenumber++,errorstream,error,filename);
388                                                         if (error)
389                                                         {
390                                                                 return false;
391                                                         }
392                                                         *target << data;
393                                                 }
394                                         }
395                                         else linenumber++;
396                                 }
397                         }
398                 }
399                 if (*buffer)
400                 {
401                         bool error = false;
402                         std::string data = ConfProcess(buffer,linenumber++,errorstream,error,filename);
403                         if (error)
404                         {
405                                 return false;
406                         }
407                         *target << data;
408                 }
409                 fclose(conf);
410         }
411         target->seekg(0);
412         return true;
413 }
414
415 /* Counts the number of tags of a certain type within the config file, e.g. to enumerate opers */
416
417 int EnumConf(std::stringstream *config, const char* tag)
418 {
419         int ptr = 0;
420         char buffer[MAXBUF], c_tag[MAXBUF], c, lastc;
421         int in_token, in_quotes, tptr, idx = 0;
422
423         const char* buf = config->str().c_str();
424         long bptr = 0;
425         long len = strlen(buf);
426         
427         ptr = 0;
428         in_token = 0;
429         in_quotes = 0;
430         lastc = '\0';
431         while (bptr<len)
432         {
433                 lastc = c;
434                 c = buf[bptr++];
435                 if ((c == '#') && (lastc == '\n'))
436                 {
437                         while ((c != '\n') && (bptr<len))
438                         {
439                                 lastc = c;
440                                 c = buf[bptr++];
441                         }
442                 }
443                 if ((c == '<') && (!in_quotes))
444                 {
445                         tptr = 0;
446                         in_token = 1;
447                         do {
448                                 c = buf[bptr++];
449                                 if (c != ' ')
450                                 {
451                                         c_tag[tptr++] = c;
452                                         c_tag[tptr] = '\0';
453                                 }
454                         } while (c != ' ');
455                 }
456                 if (c == '"')
457                 {
458                         in_quotes = (!in_quotes);
459                 }
460                 if ((c == '>') && (!in_quotes))
461                 {
462                         in_token = 0;
463                         if (!strcmp(c_tag,tag))
464                         {
465                                 /* correct tag, but wrong index */
466                                 idx++;
467                         }
468                         c_tag[0] = '\0';
469                         buffer[0] = '\0';
470                         ptr = 0;
471                         tptr = 0;
472                 }
473                 if (c != '>')
474                 {
475                         if ((in_token) && (c != '\n') && (c != '\r'))
476                         {
477                                 buffer[ptr++] = c;
478                                 buffer[ptr] = '\0';
479                         }
480                 }
481         }
482         return idx;
483 }
484
485 /* Counts the number of values within a certain tag */
486
487 int EnumValues(std::stringstream *config, const char* tag, int index)
488 {
489         int ptr = 0;
490         char buffer[MAXBUF], c_tag[MAXBUF], c, lastc;
491         int in_token, in_quotes, tptr, idx = 0;
492         
493         bool correct_tag = false;
494         int num_items = 0;
495
496         const char* buf = config->str().c_str();
497         long bptr = 0;
498         long len = strlen(buf);
499         
500         ptr = 0;
501         in_token = 0;
502         in_quotes = 0;
503         lastc = '\0';
504         while (bptr<len)
505         {
506                 lastc = c;
507                 c = buf[bptr++];
508                 if ((c == '#') && (lastc == '\n'))
509                 {
510                         while ((c != '\n') && (bptr<len))
511                         {
512                                 lastc = c;
513                                 c = buf[bptr++];
514                         }
515                 }
516                 if ((c == '<') && (!in_quotes))
517                 {
518                         tptr = 0;
519                         in_token = 1;
520                         do {
521                                 c = buf[bptr++];
522                                 if (c != ' ')
523                                 {
524                                         c_tag[tptr++] = c;
525                                         c_tag[tptr] = '\0';
526                                         
527                                         if ((!strcmp(c_tag,tag)) && (idx == index))
528                                         {
529                                                 correct_tag = true;
530                                         }
531                                 }
532                         } while (c != ' ');
533                 }
534                 if (c == '"')
535                 {
536                         in_quotes = (!in_quotes);
537                 }
538                 
539                 if ( (correct_tag) && (!in_quotes) && ( (c == ' ') || (c == '\n') || (c == '\r') ) )
540                 {
541                         num_items++;
542                 }
543                 if ((c == '>') && (!in_quotes))
544                 {
545                         in_token = 0;
546                         if (correct_tag)
547                                 correct_tag = false;
548                         if (!strcmp(c_tag,tag))
549                         {
550                                 /* correct tag, but wrong index */
551                                 idx++;
552                         }
553                         c_tag[0] = '\0';
554                         buffer[0] = '\0';
555                         ptr = 0;
556                         tptr = 0;
557                 }
558                 if (c != '>')
559                 {
560                         if ((in_token) && (c != '\n') && (c != '\r'))
561                         {
562                                 buffer[ptr++] = c;
563                                 buffer[ptr] = '\0';
564                         }
565                 }
566         }
567         return num_items+1;
568 }
569
570
571
572 int ConfValueEnum(char* tag, std::stringstream* config)
573 {
574         return EnumConf(config,tag);
575 }
576
577
578
579 /* Retrieves a value from the config file. If there is more than one value of the specified
580  * key and section (e.g. for opers etc) then the index value specifies which to retreive, e.g.
581  *
582  * ConfValue("oper","name",2,result);
583  */
584
585 int ReadConf(std::stringstream *config, const char* tag, const char* var, int index, char *result)
586 {
587         int ptr = 0;
588         char buffer[65535], c_tag[MAXBUF], c, lastc;
589         int in_token, in_quotes, tptr, idx = 0;
590         char* key;
591
592         const char* buf = config->str().c_str();
593         long bptr = 0;
594         long len = config->str().length();
595         
596         ptr = 0;
597         in_token = 0;
598         in_quotes = 0;
599         lastc = '\0';
600         c_tag[0] = '\0';
601         buffer[0] = '\0';
602         while (bptr<len)
603         {
604                 lastc = c;
605                 c = buf[bptr++];
606                 // FIX: Treat tabs as spaces
607                 if (c == 9)
608                         c = 32;
609                 if ((c == '<') && (!in_quotes))
610                 {
611                         tptr = 0;
612                         in_token = 1;
613                         do {
614                                 c = buf[bptr++];
615                                 if (c != ' ')
616                                 {
617                                         c_tag[tptr++] = c;
618                                         c_tag[tptr] = '\0';
619                                 }
620                         // FIX: Tab can follow a tagname as well as space.
621                         } while ((c != ' ') && (c != 9));
622                 }
623                 if (c == '"')
624                 {
625                         in_quotes = (!in_quotes);
626                 }
627                 if ((c == '>') && (!in_quotes))
628                 {
629                         in_token = 0;
630                         if (idx == index)
631                         {
632                                 if (!strcmp(c_tag,tag))
633                                 {
634                                         if ((buffer) && (c_tag) && (var))
635                                         {
636                                                 key = strstr(buffer,var);
637                                                 if (!key)
638                                                 {
639                                                         /* value not found in tag */
640                                                         strcpy(result,"");
641                                                         return 0;
642                                                 }
643                                                 else
644                                                 {
645                                                         key+=strlen(var);
646                                                         while (key[0] !='"')
647                                                         {
648                                                                 if (!strlen(key))
649                                                                 {
650                                                                         /* missing quote */
651                                                                         strcpy(result,"");
652                                                                         return 0;
653                                                                 }
654                                                                 key++;
655                                                         }
656                                                         key++;
657                                                         for (unsigned j = 0; j < strlen(key); j++)
658                                                         {
659                                                                 if (key[j] == '"')
660                                                                 {
661                                                                         key[j] = '\0';
662                                                                 }
663                                                         }
664                                                         strlcpy(result,key,MAXBUF);
665                                                         return 1;
666                                                 }
667                                         }
668                                 }
669                         }
670                         if (!strcmp(c_tag,tag))
671                         {
672                                 /* correct tag, but wrong index */
673                                 idx++;
674                         }
675                         c_tag[0] = '\0';
676                         buffer[0] = '\0';
677                         ptr = 0;
678                         tptr = 0;
679                 }
680                 if (c != '>')
681                 {
682                         if ((in_token) && (c != '\n') && (c != '\r'))
683                         {
684                                 buffer[ptr++] = c;
685                                 buffer[ptr] = '\0';
686                         }
687                 }
688         }
689         strcpy(result,""); // value or its tag not found at all
690         return 0;
691 }
692
693
694
695 int ConfValue(char* tag, char* var, int index, char *result,std::stringstream *config)
696 {
697         ReadConf(config, tag, var, index, result);
698         return 0;
699 }
700
701
702
703 // This will bind a socket to a port. It works for UDP/TCP
704 int BindSocket (int sockfd, struct sockaddr_in client, struct sockaddr_in server, int port, char* addr)
705 {
706         memset((char *)&server,0,sizeof(server));
707         struct in_addr addy;
708         inet_aton(addr,&addy);
709         server.sin_family = AF_INET;
710         if (!strcmp(addr,""))
711         {
712                 server.sin_addr.s_addr = htonl(INADDR_ANY);
713         }
714         else
715         {
716                 server.sin_addr = addy;
717         }
718         server.sin_port = htons(port);
719         if (bind(sockfd,(struct sockaddr*)&server,sizeof(server))<0)
720         {
721                 return(ERROR);
722         }
723         else
724         {
725                 listen(sockfd, MaxConn);
726                 return(TRUE);
727         }
728 }
729
730
731 // Open a TCP Socket
732 int OpenTCPSocket (void)
733 {
734         int sockfd;
735         int on = 1;
736         struct linger linger = { 0 };
737   
738         if ((sockfd = socket (AF_INET, SOCK_STREAM, 0)) < 0)
739                 return (ERROR);
740         else
741         {
742                 setsockopt(sockfd, SOL_SOCKET, SO_REUSEADDR, (const char*)&on, sizeof(on));
743                 /* This is BSD compatible, setting l_onoff to 0 is *NOT* http://web.irc.org/mla/ircd-dev/msg02259.html */
744                 linger.l_onoff = 1;
745                 linger.l_linger = 1;
746                 setsockopt(sockfd, SOL_SOCKET, SO_LINGER, (const char*)&linger,sizeof(linger));
747                 return (sockfd);
748         }
749 }
750