/[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.170 - (show annotations) (download) (as text)
Sat Jun 19 18:07:01 2004 UTC (15 years, 2 months ago) by wingman
Branch: MAIN
Changes since 1.169: +49 -12 lines
File MIME type: text/x-chdr
* Added new (debugging) command: .binds
* Added "framework" for init/shutdown/restart. We should really try to avoid
  use of exit system call. More cleanups to come later.

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.169 2004/06/17 13:32:44 wingman Exp $";
23 #endif
24
25 #if HAVE_CONFIG_H
26 # include <config.h>
27 #endif
28
29 #include <eggdrop/eggdrop.h>
30 #include <ctype.h>
31 #include <unistd.h>
32 #include <fcntl.h>
33 #include <string.h>
34 #include <stdlib.h>
35 #include <sys/stat.h>
36 #include <errno.h>
37 #include <signal.h>
38 #include <netdb.h>
39 #include <setjmp.h>
40 #include <locale.h>
41 #include <ctype.h>
42 #include <ltdl.h>
43
44 #include <eggdrop/stat.h>
45
46 #ifdef TIME_WITH_SYS_TIME
47 # include <sys/time.h>
48 # include <time.h>
49 #else
50 # ifdef HAVE_SYS_TIME_H
51 # include <sys/time.h>
52 # else
53 # include <time.h>
54 # endif
55 #endif
56
57 #include "lib/compat/compat.h"
58 #include "debug.h"
59 #include "core_config.h"
60 #include "core_party.h"
61 #include "core_binds.h"
62 #include "logfile.h"
63 #include "bg.h"
64 #include "main.h"
65
66 #ifdef STOP_UAC /* OSF/1 complains a lot */
67 # include <sys/sysinfo.h>
68 # define UAC_NOPRINT 0x00000001 /* Don't report unaligned fixups */
69 #endif
70
71 #ifdef CYGWIN_HACKS
72 # include <windows.h>
73 #endif
74
75 #ifndef _POSIX_SOURCE
76 # define _POSIX_SOURCE 1 /* Solaris needs this */
77 #endif
78
79 int running = 1; /* The most important variable ever! */
80 int backgrd = 1; /* Run in the background? */
81 int make_userfile = 0; /* Start bot in make-userfile mode? */
82
83 const char *configfile = NULL; /* Name of the config file */
84 char pid_file[512]; /* Name of Eggdrop's pid file */
85
86 time_t online_since; /* Time the bot was started */
87 time_t now; /* Current time */
88 egg_timeval_t egg_timeval_now; /* Current time in seconds and microseconds. */
89
90 int use_stderr = 1; /* Send stuff to stderr instead of logfiles? */
91
92 char version[81];
93
94 static int lastmin = 99;
95 static struct tm nowtm;
96
97 void core_config_init();
98
99 void fatal(const char *s, int recoverable)
100 {
101 putlog(LOG_MISC, "*", "Fatal: %s", s);
102 flushlogs();
103 unlink(pid_file);
104 if (!recoverable) {
105 bg_send_quit(BG_ABORT);
106 exit(1);
107 }
108 }
109
110 static void got_bus(int z)
111 {
112 fatal("BUS ERROR.", 1);
113 #ifdef SA_RESETHAND
114 kill(getpid(), SIGBUS);
115 #else
116 bg_send_quit(BG_ABORT);
117 exit(1);
118 #endif
119 }
120
121 static void got_segv(int z)
122 {
123 fatal("SEGMENT VIOLATION.", 1);
124 #ifdef SA_RESETHAND
125 kill(getpid(), SIGSEGV);
126 #else
127 bg_send_quit(BG_ABORT);
128 exit(1);
129 #endif
130 }
131
132 static void got_fpe(int z)
133 {
134 fatal("FLOATING POINT ERROR.", 0);
135 }
136
137 static void got_term(int z)
138 {
139 eggdrop_event("sigterm");
140 if (core_config.die_on_sigterm) {
141 putlog(LOG_MISC, "*", _("Saving user file..."));
142 user_save(core_config.userfile);
143 putlog(LOG_MISC, "*", _("Saving config file..."));
144 core_config_save();
145 putlog(LOG_MISC, "*", _("Bot shutting down: received TERMINATE signal."));
146 flushlogs();
147 unlink(pid_file);
148 exit(0);
149 }
150 else putlog(LOG_MISC, "*", _("Received TERMINATE signal (ignoring)."));
151 }
152
153 static void got_quit(int z)
154 {
155 eggdrop_event("sigquit");
156 putlog(LOG_MISC, "*", _("Received QUIT signal (ignoring)."));
157 return;
158 }
159
160 static void got_hup(int z)
161 {
162 eggdrop_event("sighup");
163 putlog(LOG_MISC, "*", _("Received HUP signal (ignoring)."));
164 return;
165 }
166
167 #if 0
168 static void got_ill(int z)
169 {
170 eggdrop_event("sigill");
171 putlog(LOG_MISC, "*", _("Received ILL signal (ignoring)."));
172 }
173 #endif
174
175
176 static void print_version(void)
177 {
178 printf("%s\n", version);
179 }
180
181 static void print_help(char *const *argv)
182 {
183 print_version();
184 printf(_("\nUsage: %s [OPTIONS]... [FILE]\n"), argv[0]);
185 printf(_("\n\
186 -h, --help Print help and exit\n\
187 -v, --version Print version and exit\n\
188 -n, --foreground Don't go into the background (default=off)\n\
189 -c --channel-stats (with -n) Display channel stats every 10 seconds\n\
190 -t --terminal (with -n) Use terminal to simulate dcc-chat\n\
191 -m --make-userfile Userfile creation mode\n\
192 FILE optional config filename (default 'config.xml')\n"));
193 printf("\n");
194 }
195
196 static void do_args(int argc, char *const *argv)
197 {
198 int c;
199
200 optarg = 0;
201 optind = 1;
202 opterr = 1;
203 optopt = '?';
204
205 while (1) {
206 int option_index = 0;
207
208 static struct option long_options[] = {
209 {"help", 0, NULL, 'h'},
210 {"version", 0, NULL, 'v'},
211 {"load-module", 1, NULL, 'p'},
212 {"foreground", 0, NULL, 'n'},
213 {"terminal", 0, NULL, 't'},
214 {"make-userfile", 0, NULL, 'm'},
215 {NULL, 0, NULL, 0}
216 };
217
218 c = getopt_long(argc, argv, "hvp:nctm", long_options, &option_index);
219 if (c == -1) break;
220 switch (c) {
221 case 'n':
222 backgrd = 0;
223 break;
224 case 't':
225 partyline_terminal_mode = 1;
226 break;
227 case 'm':
228 make_userfile = 1;
229 break;
230 case 'v':
231 print_version();
232 bg_send_quit(BG_ABORT);
233 exit(0);
234 break; /* this should never be reached */
235 case 'h':
236 print_help(argv);
237 bg_send_quit(BG_ABORT);
238 exit(0);
239 break; /* this should never be reached */
240 case 0: /* Long option with no short option */
241 case '?': /* Invalid option. */
242 /* getopt_long() already printed an error message. */
243 bg_send_quit(BG_ABORT);
244 exit(1);
245 default: /* bug: option not considered. */
246 fprintf(stderr, _("%s: option unknown: %c\n"), argv[0], c);
247 bg_send_quit(BG_ABORT);
248 exit(1);
249 break; /* this should never be reached */
250 }
251 }
252 if (optind < argc) configfile = argv[optind];
253 else configfile = "config.xml";
254 }
255
256 /* Called once a second.
257 */
258 static int core_secondly()
259 {
260 static int cnt = 0;
261 int miltime;
262
263 check_bind_secondly();
264 cnt++;
265 memcpy(&nowtm, localtime(&now), sizeof(struct tm));
266 if (nowtm.tm_min != lastmin) {
267 int i = 0;
268
269 /* Once a minute */
270 lastmin = (lastmin + 1) % 60;
271 eggdrop_event("minutely");
272 check_bind_time(&nowtm);
273 /* In case for some reason more than 1 min has passed: */
274 while (nowtm.tm_min != lastmin) {
275 /* Timer drift, dammit */
276 putlog(LOG_DEBUG, "*", "timer: drift (lastmin=%d, now=%d)", lastmin, nowtm.tm_min);
277 i++;
278 lastmin = (lastmin + 1) % 60;
279 eggdrop_event("minutely");
280 }
281 if (i > 1) putlog(LOG_MISC, "*", _("Warning: timer drift (%d minutes)."), i);
282
283 miltime = (nowtm.tm_hour * 100) + (nowtm.tm_min);
284 if (((int) (nowtm.tm_min / 5) * 5) == (nowtm.tm_min)) { /* 5 minutes */
285 eggdrop_event("5minutely");
286 if (!miltime) { /* At midnight */
287 char s[25];
288
289 strlcpy(s, ctime(&now), sizeof s);
290 putlog(LOG_ALL, "*", "--- %.11s%s", s, s + 20);
291 eggdrop_event("backup");
292 eggdrop_event("daily");
293 }
294 }
295 }
296 return(0);
297 }
298
299
300 int file_check(const char *filename)
301 {
302 struct stat sbuf = {0};
303 int r;
304
305 r = stat(filename, &sbuf);
306 if (r) return(-1);
307 errno = 0;
308 if (sbuf.st_mode & (S_IRGRP|S_IWGRP|S_IROTH|S_IWOTH)) return(-1);
309 return(0);
310 }
311
312 int main(int argc, char **argv)
313 {
314 int xx, i, timeout;
315 char s[25], *modname, *scriptname;
316 FILE *f;
317 struct sigaction sv;
318 egg_timeval_t howlong;
319 void *config_root, *entry;
320
321 #ifdef DEBUG
322 {
323 # include <sys/resource.h>
324 struct rlimit cdlim;
325
326 cdlim.rlim_cur = RLIM_INFINITY;
327 cdlim.rlim_max = RLIM_INFINITY;
328 setrlimit(RLIMIT_CORE, &cdlim);
329 }
330 #endif
331
332 #ifdef ENABLE_NLS
333 setlocale(LC_MESSAGES, "");
334 bindtextdomain(PACKAGE, LOCALEDIR);
335 textdomain(PACKAGE);
336 #endif
337
338 /* Initialize ltdl. */
339 LTDL_SET_PRELOADED_SYMBOLS();
340 if (lt_dlinit()) {
341 printf("Fatal error initializing ltdl: %s\n", lt_dlerror());
342 return(-1);
343 }
344
345 snprintf(version, sizeof version, "Eggdrop v%s (C) 1997 Robey Pointer (C) 2004 Eggheads Development Team", VERSION);
346
347 #ifdef STOP_UAC
348 {
349 int nvpair[2];
350
351 nvpair[0] = SSIN_UACPROC;
352 nvpair[1] = UAC_NOPRINT;
353 setsysinfo(SSI_NVPAIRS, (char *) nvpair, 1, NULL, 0);
354 }
355 #endif
356
357 sv.sa_handler = got_bus;
358 sigemptyset(&sv.sa_mask);
359 #ifdef SA_RESETHAND
360 sv.sa_flags = SA_RESETHAND;
361 #else
362 sv.sa_flags = 0;
363 #endif
364 sigaction(SIGBUS, &sv, NULL);
365
366 sv.sa_handler = got_segv;
367 sigaction(SIGSEGV, &sv, NULL);
368 #ifdef SA_RESETHAND
369 sv.sa_flags = 0;
370 #endif
371
372 sv.sa_handler = got_fpe;
373 sigaction(SIGFPE, &sv, NULL);
374
375 sv.sa_handler = got_term;
376 sigaction(SIGTERM, &sv, NULL);
377
378 sv.sa_handler = got_hup;
379 sigaction(SIGHUP, &sv, NULL);
380
381 sv.sa_handler = got_quit;
382 sigaction(SIGQUIT, &sv, NULL);
383
384 sv.sa_handler = SIG_IGN;
385 sigaction(SIGPIPE, &sv, NULL);
386
387 /* XXX: why is this disabled? */
388 #if 0
389 sv.sa_handler = got_ill;
390 sigaction(SIGILL, &sv, NULL);
391 #endif
392
393 timer_update_now(&egg_timeval_now);
394 now = egg_timeval_now.sec;
395 memcpy(&nowtm, localtime(&now), sizeof(struct tm));
396 lastmin = nowtm.tm_min;
397
398 srandom(now % (getpid() + getppid()));
399
400 do_args(argc, argv);
401 printf("\n%s\n", version);
402 printf("WARNING: Do NOT run this DEVELOPMENT version for any purpose other than testing.\n\n");
403
404 if (((int) getuid() == 0) || ((int) geteuid() == 0)) fatal("Eggdrop will not run as root!", 0);
405
406 if (file_check(configfile)) {
407 if (errno) perror(configfile);
408 else fprintf(stderr, "ERROR\n\nCheck file permissions on your config file!\nMake sure other groups and users cannot read/write it.\n");
409 exit(-1);
410 }
411
412 eggdrop_init();
413 core_config_init(configfile);
414 logfile_init();
415 if (core_config.userfile)
416 user_load(core_config.userfile);
417 core_binds_init();
418 core_party_init();
419
420 if (make_userfile) {
421 user_t *owner;
422 char handle[512], password[12];
423 int len;
424
425 if (user_count() != 0) {
426 printf("\n\n\nYou are trying to create a new userfile, but the old one still exists (%s)!\n\nPlease remove the userfile, or do not pass the -m option.\n\n", core_config.userfile);
427 exit(1);
428 }
429
430 printf("\n\n\nHello! I see you are creating a new userfile.\n\n");
431 printf("Let's create the owner account.\n\n");
432 do {
433 printf("Enter the owner handle: ");
434 fflush(stdout);
435 fgets(handle, sizeof(handle), stdin);
436 for (len = 0; handle[len]; len++) {
437 if (!ispunct(handle[len]) && !isalnum(handle[len])) break;
438 }
439 if (len == 0) printf("Come on, enter a real handle.\n\n");
440 } while (len <= 0);
441 handle[len] = 0;
442 owner = user_new(handle);
443 if (!owner) {
444 printf("Failed to create user record! Out of memory?\n\n");
445 exit(1);
446 }
447 user_rand_pass(password, sizeof(password));
448 user_set_pass(owner, password);
449 user_set_flags_str(owner, NULL, "+n");
450 printf("Your owner handle is '%s' and your password is '%s' (without the quotes).\n\n", handle, password);
451 memset(password, 0, sizeof(password));
452 str_redup(&core_config.owner, handle);
453 core_config_save();
454 user_save(core_config.userfile);
455 }
456
457 if (backgrd) bg_prepare_split();
458
459 strlcpy(s, ctime(&now), sizeof s);
460 strcpy(&s[11], &s[20]);
461 putlog(LOG_ALL, "*", "Loading Eggdrop %s (%s)", VERSION, s);
462 snprintf(pid_file, sizeof pid_file, "pid.%s", core_config.botname);
463
464 f = fopen(pid_file, "r");
465 if (f != NULL) {
466 fgets(s, 10, f);
467 xx = atoi(s);
468 kill(xx, SIGCHLD); /* Meaningless kill to determine if pid is used */
469 if (errno != ESRCH) {
470 printf("I detect %s already running from this directory.\n", core_config.botname);
471 printf("If this is incorrect, please erase '%s'.\n", pid_file);
472 bg_send_quit(BG_ABORT);
473 exit(1);
474 }
475 }
476
477 /* Load core help */
478 help_load_by_module ("core");
479
480 /* Put the module directory in the ltdl search path. */
481 if (core_config.module_path)
482 module_add_dir(core_config.module_path);
483
484 /* Scan the autoload section of config. */
485 config_root = config_get_root("eggdrop");
486 for (i = 0; (entry = config_exists(config_root, "eggdrop", 0, "autoload", 0, "module", i, NULL)); i++) {
487 modname = NULL;
488 config_get_str(&modname, entry, NULL);
489 module_load(modname);
490 }
491 for (i = 0; (entry = config_exists(config_root, "eggdrop", 0, "autoload", 0, "script", i, NULL)); i++) {
492 scriptname = NULL;
493 config_get_str(&scriptname, entry, NULL);
494 script_load(scriptname);
495 }
496
497 if (backgrd) { /* Move into background? */
498 #ifndef CYGWIN_HACKS
499 bg_do_split();
500 }
501 else {
502 #endif
503 xx = getpid();
504 if (xx != 0) {
505 FILE *fp;
506
507 unlink(pid_file);
508 fp = fopen(pid_file, "w");
509 if (fp != NULL) {
510 fprintf(fp, "%u\n", xx);
511 if (fflush(fp)) {
512 /* Let the bot live since this doesn't appear to be a botchk */
513 printf("WARNING: Could not write pid file '%s'.\n", pid_file);
514 fclose(fp);
515 unlink(pid_file);
516 }
517 else fclose(fp);
518 }
519 else printf("WARNING: Could not write pid file '%s'.\n", pid_file);
520 #ifdef CYGWIN_HACKS
521 printf("Launched into the background (pid: %d).\n\n", xx);
522 #endif
523 }
524 }
525
526 use_stderr = 0; /* Stop writing to stderr now */
527
528 if (backgrd) {
529 /* Try to disassociate from controlling terminal (finger cross) */
530 #if HAVE_SETPGID && !defined(CYGWIN_HACKS)
531 setpgid(0, 0);
532 #endif
533 /* Tcl wants the stdin, stdout and stderr file handles kept open. */
534 freopen("/dev/null", "r", stdin);
535 freopen("/dev/null", "w", stdout);
536 freopen("/dev/null", "w", stderr);
537 #ifdef CYGWIN_HACKS
538 FreeConsole();
539 #endif
540 }
541
542 online_since = now;
543 howlong.sec = 1;
544 howlong.usec = 0;
545 timer_create_repeater(&howlong, "main loop", core_secondly);
546
547 putlog(LOG_DEBUG, "*", "Entering main loop.");
548
549 core_init();
550
551 putlog(LOG_DEBUG, "*", "Entering main loop.");
552
553 while (running) {
554 timer_update_now(&egg_timeval_now);
555 now = egg_timeval_now.sec;
556 random(); /* Woop, lets really jumble things */
557 timer_run();
558 if (timer_get_shortest(&howlong)) timeout = 1000;
559 else timeout = howlong.sec * 1000 + howlong.usec / 1000;
560 sockbuf_update_all(timeout);
561 garbage_run();
562 }
563
564 /* No work need to be done here ever since if we can only reach this
565 * if someone called core_shutdown. So do your shutdown work THERE! */
566
567 return 1;
568 }
569
570 int core_init()
571 {
572 check_bind_init();
573
574 return 1;
575 }
576
577 int core_shutdown(int how, const char *nick, const char *reason)
578 {
579 putlog(LOG_MISC, "*", "Shutdown requested by %s: %s", nick, (reason) ? reason : "No reason");
580
581 /* notify that we go down NOW */
582 check_bind_shutdown();
583
584 /* XXX: move this to users.c */
585 putlog(LOG_MISC, "*", _("Saving user file..."));
586 user_save(core_config.userfile);
587
588 /* flush logs */
589 flushlogs();
590
591 /* remove pid file */
592 unlink(pid_file);
593
594 switch (how) {
595
596 case (SHUTDOWN_GRACEFULL):
597 running = 0;
598 break;
599
600 case (SHUTDOWN_HARD):
601 exit(0);
602 break;
603 }
604
605 return BIND_RET_LOG;
606 }

webmaster@eggheads.org
ViewVC Help
Powered by ViewVC 1.1.23