/[cvs]/eggdrop1.9/src/main.c
ViewVC logotype

Contents of /eggdrop1.9/src/main.c

Parent Directory Parent Directory | Revision Log Revision Log | View Revision Graph Revision Graph


Revision 1.183 - (show annotations) (download) (as text)
Thu Aug 19 18:39:36 2004 UTC (15 years ago) by darko
Branch: MAIN
Changes since 1.182: +2 -2 lines
File MIME type: text/x-chdr
- Added timeutil.c and timeutil.h - this is where (obviously) time related
  stuff should go.
- Finalized some of the channel mask stuff now that time functions exist.
- Fixed issue where egg_set_params() was called char* instead of char**

1 /* main.c
2 *
3 * Copyright (C) 1997 Robey Pointer
4 * Copyright (C) 1999, 2000, 2001, 2002, 2003, 2004 Eggheads Development Team
5 *
6 * This program is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU General Public License
8 * as published by the Free Software Foundation; either version 2
9 * of the License, or (at your option) any later version.
10 *
11 * This program is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 * GNU General Public License for more details.
15 *
16 * You should have received a copy of the GNU General Public License
17 * along with this program; if not, write to the Free Software
18 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
19 */
20
21 #ifndef lint
22 static const char rcsid[] = "$Id: main.c,v 1.182 2004/07/04 23:55:35 darko Exp $";
23 #endif
24
25 #if HAVE_CONFIG_H
26 # include <config.h>
27 #endif
28
29 #include <ctype.h>
30 #include <unistd.h>
31 #include <fcntl.h>
32 #include <string.h>
33 #include <stdlib.h>
34 #include <sys/stat.h>
35 #include <errno.h>
36 #include <signal.h>
37 #include <netdb.h>
38 #include <setjmp.h>
39 #include <locale.h>
40 #include <ctype.h>
41 #include <ltdl.h>
42
43 #ifdef TIME_WITH_SYS_TIME
44 # include <sys/time.h>
45 # include <time.h>
46 #else
47 # ifdef HAVE_SYS_TIME_H
48 # include <sys/time.h>
49 # else
50 # include <time.h>
51 # endif
52 #endif
53
54 #include <eggdrop/eggdrop.h>
55 #include "lib/compat/compat.h"
56 #include "debug.h"
57 #include "core_config.h"
58 #include "core_party.h"
59 #include "core_binds.h"
60 #include "logfile.h"
61 #include "terminal.h"
62 #include "bg.h"
63 #include "main.h"
64
65 #ifdef STOP_UAC /* OSF/1 complains a lot */
66 # include <sys/sysinfo.h>
67 # define UAC_NOPRINT 0x00000001 /* Don't report unaligned fixups */
68 #endif
69
70 #ifdef CYGWIN_HACKS
71 # include <windows.h>
72 #endif
73
74 #ifndef _POSIX_SOURCE
75 # define _POSIX_SOURCE 1 /* Solaris needs this */
76 #endif
77
78 #define RUNMODE_NORMAL 1
79 #define RUNMODE_SHUTDOWN 2
80 #define RUNMODE_RESTART 3
81
82 int runmode = 1; /* The most important variable ever! */
83 int backgrd = 1; /* Run in the background? */
84 int make_userfile = 0; /* Start bot in make-userfile mode? */
85 int terminal_mode = 0; /* Terminal mode */
86
87 /* XXX: remove this feature (but not yet) */
88 int debug_run = 0; /* Just init and shutdown */
89
90 const char *configfile = NULL; /* Name of the config file */
91 char pid_file[512]; /* Name of Eggdrop's pid file */
92
93 time_t online_since; /* Time the bot was started */
94 time_t now; /* Current time */
95 egg_timeval_t egg_timeval_now; /* Current time in seconds and microseconds. */
96
97 int use_stderr = 1; /* Send stuff to stderr instead of logfiles? */
98
99 char version[81];
100
101 static int lastmin = 99;
102 static struct tm nowtm;
103
104 void fatal(const char *errmsg)
105 {
106 putlog(LOG_MISC, "*", "Fatal: %s", errmsg);
107 flushlogs();
108 exit(1);
109 }
110
111 static void got_bus(int z)
112 {
113 fatal("BUS ERROR.");
114 }
115
116 static void got_segv(int z)
117 {
118 fatal("SEGMENT VIOLATION.");
119 }
120
121 static void got_fpe(int z)
122 {
123 fatal("FLOATING POINT ERROR.");
124 }
125
126 static void got_term(int z)
127 {
128 eggdrop_event("sigterm");
129 if (core_config.die_on_sigterm) core_shutdown(SHUTDOWN_SIGTERM, NULL, NULL);
130 else putlog(LOG_MISC, "*", _("Received TERM signal (ignoring)."));
131 }
132
133 static void got_quit(int z)
134 {
135 eggdrop_event("sigquit");
136 putlog(LOG_MISC, "*", _("Received QUIT signal (ignoring)."));
137 return;
138 }
139
140 static void got_hup(int z)
141 {
142 eggdrop_event("sighup");
143 putlog(LOG_MISC, "*", _("Received HUP signal (ignoring)."));
144 return;
145 }
146
147 static void print_version(void)
148 {
149 printf("%s\n", version);
150 }
151
152 static void print_help(char *const *argv)
153 {
154 print_version();
155 printf(_("\nUsage: %s [OPTIONS]... [FILE]\n"), argv[0]);
156 printf(_("\n\
157 -h, --help Print help and exit\n\
158 -v, --version Print version and exit\n\
159 -n, --foreground Don't go into the background (default=off)\n\
160 -c --channel-stats (with -n) Display channel stats every 10 seconds\n\
161 -t --terminal (with -n) Use terminal to simulate dcc-chat\n\
162 -m --make-userfile Userfile creation mode\n\
163 -d --debug Do a debugging dry-run\n\
164 FILE optional config filename (default 'config.xml')\n"));
165 printf("\n");
166 }
167
168 static void do_args(int argc, char **argv)
169 {
170 int c;
171
172 optarg = 0;
173 optind = 1;
174 opterr = 1;
175 optopt = '?';
176
177 while (1) {
178 int option_index = 0;
179
180 static struct option long_options[] = {
181 { "help", 0, NULL, 'h' },
182 { "version", 0, NULL, 'v' },
183 { "load-module", 1, NULL, 'p' },
184 { "foreground", 0, NULL, 'n' },
185 { "terminal", 0, NULL, 't' },
186 { "make-userfile", 0, NULL, 'm' },
187 { "debug", 0, NULL, 'd' },
188 {NULL, 0, NULL, 0}
189 };
190
191 c = getopt_long(argc, argv, "hvp:nctmd", long_options, &option_index);
192 if (c == -1) break;
193 switch (c) {
194 case 'n':
195 backgrd = 0;
196 break;
197 case 't':
198 terminal_mode = 1;
199 break;
200 case 'm':
201 make_userfile = 1;
202 break;
203 case 'd':
204 backgrd = 0;
205 debug_run = 1;
206 break;
207 case 'v':
208 print_version();
209 exit(0);
210 break; /* this should never be reached */
211 case 'h':
212 print_help(argv);
213 exit(0);
214 break; /* this should never be reached */
215 case 0: /* Long option with no short option */
216 case '?': /* Invalid option. */
217 /* getopt_long() already printed an error message. */
218 exit(1);
219 default: /* bug: option not considered. */
220 fprintf(stderr, _("%s: option unknown: %c\n"), argv[0], c);
221 exit(1);
222 break; /* this should never be reached */
223 }
224 }
225 if (optind < argc) {
226 configfile = argv[optind++];
227 eggdrop_set_params((const char **)&argv[optind], argc-optind);
228 }
229 else configfile = "config.xml";
230 }
231
232 /* Called once a second.
233 */
234 static int core_secondly()
235 {
236 static int cnt = 0;
237 int miltime;
238
239 check_bind_secondly();
240 cnt++;
241 memcpy(&nowtm, localtime(&now), sizeof(struct tm));
242 if (nowtm.tm_min != lastmin) {
243 int i = 0;
244
245 /* Once a minute */
246 lastmin = (lastmin + 1) % 60;
247 eggdrop_event("minutely");
248 check_bind_time(&nowtm);
249 /* In case for some reason more than 1 min has passed: */
250 while (nowtm.tm_min != lastmin) {
251 /* Timer drift, dammit */
252 putlog(LOG_DEBUG, "*", "timer: drift (lastmin=%d, now=%d)", lastmin, nowtm.tm_min);
253 i++;
254 lastmin = (lastmin + 1) % 60;
255 eggdrop_event("minutely");
256 }
257 if (i > 1) putlog(LOG_MISC, "*", _("Warning: timer drift (%d minutes)."), i);
258
259 miltime = (nowtm.tm_hour * 100) + (nowtm.tm_min);
260 if (nowtm.tm_min % 5 == 0) { /* 5 minutes */
261 eggdrop_event("5minutely");
262 if (!miltime) { /* At midnight */
263 char s[25];
264
265 strlcpy(s, ctime(&now), sizeof s);
266 putlog(LOG_ALL, "*", "--- %.11s%s", s, s + 20);
267 eggdrop_event("backup");
268 eggdrop_event("daily");
269 }
270 }
271 }
272 return(0);
273 }
274
275
276 int file_check(const char *filename)
277 {
278 struct stat sbuf = {0};
279 int r;
280
281 r = stat(filename, &sbuf);
282 if (r) return(-1);
283 errno = 0;
284 if (sbuf.st_mode & (S_IRGRP|S_IWGRP|S_IROTH|S_IWOTH)) return(-1);
285 return(0);
286 }
287
288 static int create_userfile(void)
289 {
290 user_t *owner;
291 char handle[512], password[12];
292 int len;
293
294 printf("\n");
295 if (user_count() != 0) {
296 printf("You are trying to create a new userfile, but the old one still exists (%s)!\n", core_config.userfile);
297 printf("Please remove the userfile, or do not pass the -m option.\n\n");
298 return(-1);
299 }
300
301 printf("Hello! I see you are creating a new userfile.\n\n");
302 printf("Let's create the owner account.\n\n");
303 do {
304 printf("Enter the owner handle: ");
305 fflush(stdout);
306 fgets(handle, sizeof(handle), stdin);
307 for (len = 0; handle[len]; len++) {
308 if (!ispunct(handle[len]) && !isalnum(handle[len])) break;
309 }
310 if (len == 0) printf("Come on, enter a real handle.\n\n");
311 } while (len <= 0);
312 handle[len] = 0;
313
314 owner = user_new(handle);
315 if (!owner) {
316 printf("Failed to create user record! Out of memory?\n\n");
317 return(-1);
318 }
319
320 user_rand_pass(password, sizeof(password));
321 user_set_pass(owner, password);
322 user_set_flags_str(owner, NULL, "+n");
323 printf("Your owner handle is '%s' and your password is '%s' (without the quotes).\n\n", handle, password);
324 memset(password, 0, sizeof(password));
325 str_redup(&core_config.owner, handle);
326 core_config_save();
327 user_save(core_config.userfile);
328
329 return(0);
330 }
331
332 static void init_signals(void)
333 {
334 struct sigaction sv;
335
336
337 sv.sa_handler = got_bus;
338 sigemptyset(&sv.sa_mask);
339 #ifdef SA_RESETHAND
340 sv.sa_flags = SA_RESETHAND;
341 #else
342 sv.sa_flags = 0;
343 #endif
344 sigaction(SIGBUS, &sv, NULL);
345
346 sv.sa_handler = got_segv;
347 //sigaction(SIGSEGV, &sv, NULL);
348 #ifdef SA_RESETHAND
349 sv.sa_flags = 0;
350 #endif
351
352 sv.sa_handler = got_fpe;
353 sigaction(SIGFPE, &sv, NULL);
354
355 sv.sa_handler = got_term;
356 sigaction(SIGTERM, &sv, NULL);
357
358 sv.sa_handler = got_hup;
359 sigaction(SIGHUP, &sv, NULL);
360
361 sv.sa_handler = got_quit;
362 sigaction(SIGQUIT, &sv, NULL);
363
364 sv.sa_handler = SIG_IGN;
365 sigaction(SIGPIPE, &sv, NULL);
366 }
367
368 static void core_shutdown_or_restart()
369 {
370 /* notifiy shutdown listeners */
371 check_bind_shutdown();
372
373 putlog(LOG_MISC, "*", _("Saving user file..."));
374 user_save(core_config.userfile);
375
376 putlog(LOG_MISC, "*", _("Saving config file..."));
377 core_config_save();
378
379 /* destroy terminal */
380 if (terminal_mode && runmode != RUNMODE_RESTART)
381 terminal_shutdown();
382
383 /* unload core help */
384 help_unload_by_module("core");
385
386 /* shutdown core binds */
387 core_binds_shutdown();
388
389 /* shutdown logging */
390 logfile_shutdown();
391
392 /* shutdown libeggdrop */
393 eggdrop_shutdown();
394
395 /* force a garbage run */
396 garbage_run();
397
398 /* just remove pid file if didn't restart */
399 if (runmode != RUNMODE_RESTART)
400 unlink(pid_file);
401 }
402
403 int main(int argc, char **argv)
404 {
405 int timeout;
406
407 #ifdef DEBUG
408 {
409 # include <sys/resource.h>
410 struct rlimit cdlim;
411
412 cdlim.rlim_cur = RLIM_INFINITY;
413 cdlim.rlim_max = RLIM_INFINITY;
414 setrlimit(RLIMIT_CORE, &cdlim);
415 }
416 #endif
417
418 init_signals();
419
420 #ifdef ENABLE_NLS
421 setlocale(LC_MESSAGES, "");
422 bindtextdomain(PACKAGE, LOCALEDIR);
423 textdomain(PACKAGE);
424 #endif
425
426 /* Initialize ltdl. */
427 LTDL_SET_PRELOADED_SYMBOLS();
428 if (lt_dlinit()) {
429 printf("Fatal error initializing ltdl: %s\n", lt_dlerror());
430 return(-1);
431 }
432
433 snprintf(version, sizeof version, "Eggdrop v%s (C) 1997 Robey Pointer (C) 2004 Eggheads Development Team", VERSION);
434
435 #ifdef STOP_UAC
436 {
437 int nvpair[2];
438
439 nvpair[0] = SSIN_UACPROC;
440 nvpair[1] = UAC_NOPRINT;
441 setsysinfo(SSI_NVPAIRS, (char *) nvpair, 1, NULL, 0);
442 }
443 #endif
444
445 timer_update_now(&egg_timeval_now);
446 now = egg_timeval_now.sec;
447 memcpy(&nowtm, localtime(&now), sizeof(struct tm));
448 lastmin = nowtm.tm_min;
449
450 srandom(now % (getpid() + getppid()));
451
452 do_args(argc, argv);
453 printf("\n%s\n", version);
454 printf("WARNING: Do NOT run this DEVELOPMENT version for any purpose other than testing.\n\n");
455
456 /* we may not run as root */
457 if (((int) getuid() == 0) || ((int) geteuid() == 0)) {
458 fprintf(stderr, "ERROR\n");
459 fprintf(stderr, "\tEggdrop will not run as root!\n\n");
460 return -1;
461 }
462
463 init_flag_map();
464
465 /* config file may not be world read/writeable */
466 if (file_check(configfile)) {
467 fprintf(stderr, "ERROR\n");
468 if (errno) {
469 fprintf(stderr, "\tFailed to load config file '%s': %s\n\n", configfile, strerror(errno));
470 } else {
471 fprintf(stderr, "\tCheck file permissions on your config file '%s'!\n", configfile);
472 fprintf(stderr, "\tMake sure other groups and users cannot read/write it.\n\n");
473 }
474 return -1;
475 }
476
477 /* set uptime */
478 online_since = now;
479
480 do {
481 int tid;
482 egg_timeval_t howlong;
483
484 /* default select timeout is 1 sec */
485 howlong.sec = 1;
486 howlong.usec = 0;
487
488 tid = timer_create_repeater(&howlong, "main loop", core_secondly);
489
490 /* init core */
491 if (core_init() != 0)
492 break;
493
494 /* set normal running mode */
495 runmode = RUNMODE_NORMAL;
496
497 putlog(LOG_DEBUG, "*", "Entering main loop.");
498
499 /* main loop */
500 while (!debug_run && runmode == RUNMODE_NORMAL) {
501 /* Update the time. */
502 timer_update_now(&egg_timeval_now);
503 now = egg_timeval_now.sec;
504
505 /* Run any expired timers. */
506 timer_run();
507
508 /* Figure out how long to wait for activity. */
509 if (timer_get_shortest(&howlong)) timeout = 1000;
510 else timeout = howlong.sec * 1000 + howlong.usec / 1000;
511
512 /* Wait and monitor sockets. */
513 sockbuf_update_all(timeout);
514
515 /* Run any scheduled cleanups. */
516 garbage_run();
517 }
518
519 /* shutdown secondly timer */
520 timer_destroy(tid);
521
522 /* Save user file, config file, ... */
523 core_shutdown_or_restart();
524
525 /* the only chance to loop again is that running = 2, meaning
526 we have a restart */
527 } while (runmode == RUNMODE_RESTART);
528
529 if (debug_run)
530 mem_dbg_stats();
531
532 return 0;
533 }
534
535 int core_init()
536 {
537 int i;
538 char *name;
539 void *config_root, *entry;
540 char datetime[25];
541 static int lockfd = -1;
542 struct flock lock;
543
544 /* init libeggdrop */
545 eggdrop_init();
546
547 /* load config */
548 if (core_config_init(configfile) != 0)
549 return (-1);
550
551 /* init logging */
552 logfile_init();
553
554 /* did the user specify -m? */
555 if (make_userfile) {
556 if (create_userfile()) return(-1);
557 make_userfile = 0;
558 }
559
560 /* init background mode and pid file */
561 if (runmode != RUNMODE_RESTART) {
562 if (backgrd) bg_begin_split();
563 }
564
565 /* Check lock file. */
566 if (core_config.lockfile && lockfd == -1) {
567 lockfd = open(core_config.lockfile, O_WRONLY | O_CREAT, S_IWUSR);
568 if (lockfd < 0) {
569 fatal("Could not open lock file!");
570 }
571 memset(&lock, 0, sizeof(lock));
572 lock.l_type = F_WRLCK;
573 if (fcntl(lockfd, F_SETLK, &lock) != 0) {
574 fatal("Lock file is already locked! Is eggdrop already running?");
575 }
576 }
577
578 /* just issue this "Loading Eggdrop" message if we are not
579 * restarting */
580 if (runmode != RUNMODE_RESTART) {
581 strlcpy(datetime, ctime(&now), sizeof(datetime));
582 strcpy(&datetime[11], &datetime[20]);
583 putlog(LOG_ALL, "*", "Loading Eggdrop %s (%s)", VERSION, datetime);
584 }
585
586 /* load userlist */
587 if (core_config.userfile)
588 user_load(core_config.userfile);
589
590 /* init core bindings */
591 core_binds_init();
592
593 /* init core partyline */
594 core_party_init();
595
596 /* Load core help */
597 help_load_by_module ("core");
598
599 /* Put the module directory in the ltdl search path. */
600 if (core_config.module_path)
601 module_add_dir(core_config.module_path);
602
603 /* Scan the autoload section of config. */
604 config_root = config_get_root("eggdrop");
605 for (i = 0; (entry = config_exists(config_root, "eggdrop", 0, "autoload", 0, "module", i, NULL)); i++) {
606 name = NULL;
607 config_get_str(&name, entry, NULL);
608 module_load(name);
609 }
610 for (i = 0; (entry = config_exists(config_root, "eggdrop", 0, "autoload", 0, "script", i, NULL)); i++) {
611 name = NULL;
612 config_get_str(&name, entry, NULL);
613 script_load(name);
614 }
615
616 /* notify init listeners */
617 check_bind_init();
618
619 /* start terminal */
620 if (runmode != RUNMODE_RESTART) {
621 if (backgrd) bg_finish_split();
622 if (terminal_mode) terminal_init();
623 }
624
625 return (0);
626 }
627
628 int core_restart(const char *nick)
629 {
630 putlog(LOG_MISC, "*", "Restarting...");
631 runmode = RUNMODE_RESTART;
632 return (0);
633 }
634
635 int core_shutdown(int how, const char *nick, const char *reason)
636 {
637 if (how == SHUTDOWN_SIGTERM)
638 putlog(LOG_MISC, "*", ("Received TERM signal, shutting down."));
639 else if (how != SHUTDOWN_RESTART)
640 putlog(LOG_MISC, "*", ("Shutdown requested by %s: %s"), nick,
641 (reason) ? reason : _("No reason"));
642
643 /* check how to shut down */
644 switch (how) {
645
646 case (SHUTDOWN_RESTART):
647 runmode = RUNMODE_RESTART;
648 break;
649
650 case (SHUTDOWN_GRACEFULL):
651 runmode = RUNMODE_SHUTDOWN;
652 break;
653
654 /* These two cases are treaded special because something
655 * unusual happened and we're better off if we exit() NOW
656 * than doing a gracefull shutdown with freeing all
657 * memory, closing all sockets, ... */
658 case (SHUTDOWN_HARD):
659 case (SHUTDOWN_SIGTERM):
660 runmode = RUNMODE_SHUTDOWN;
661 core_shutdown_or_restart();
662 exit(0);
663 break;
664
665 }
666
667 return BIND_RET_LOG;
668 }

webmaster@eggheads.org
ViewVC Help
Powered by ViewVC 1.1.23