source: unix.c @ 33528bd

Last change on this file since 33528bd was 5ebff60, checked in by dequis <dx@…>, at 2015-02-20T22:50:54Z

Reindent everything to K&R style with tabs

Used uncrustify, with the configuration file in ./doc/uncrustify.cfg

Commit author set to "Indent <please@…>" so that it's easier to
skip while doing git blame.

  • Property mode set to 100644
File size: 8.4 KB
Line 
1/********************************************************************\
2  * BitlBee -- An IRC to other IM-networks gateway                     *
3  *                                                                    *
4  * Copyright 2002-2012 Wilmer van der Gaast and others                *
5  \********************************************************************/
6
7/* Main file (Unix specific part)                                       */
8
9/*
10  This program is free software; you can redistribute it and/or modify
11  it under the terms of the GNU General Public License as published by
12  the Free Software Foundation; either version 2 of the License, or
13  (at your option) any later version.
14
15  This program is distributed in the hope that it will be useful,
16  but WITHOUT ANY WARRANTY; without even the implied warranty of
17  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
18  GNU General Public License for more details.
19
20  You should have received a copy of the GNU General Public License with
21  the Debian GNU/Linux distribution in /usr/share/common-licenses/GPL;
22  if not, write to the Free Software Foundation, Inc., 51 Franklin St.,
23  Fifth Floor, Boston, MA  02110-1301  USA
24*/
25
26#include "bitlbee.h"
27
28#include "arc.h"
29#include "base64.h"
30#include "commands.h"
31#include "protocols/nogaim.h"
32#include "help.h"
33#include "ipc.h"
34#include "md5.h"
35#include "misc.h"
36#include <signal.h>
37#include <unistd.h>
38#include <sys/time.h>
39#include <sys/wait.h>
40#include <pwd.h>
41#include <locale.h>
42#include <grp.h>
43
44#if defined(OTR_BI) || defined(OTR_PI)
45#include "otr.h"
46#endif
47
48global_t global;        /* Against global namespace pollution */
49
50static int signal_shutdown_pipe[2] = { -1, -1 };
51static void sighandler_shutdown(int signal);
52static void sighandler_crash(int signal);
53
54static int crypt_main(int argc, char *argv[]);
55
56int main(int argc, char *argv[])
57{
58        int i = 0;
59        char *old_cwd = NULL;
60        struct sigaction sig, old;
61
62        /* Required to make iconv to ASCII//TRANSLIT work. This makes BitlBee
63           system-locale-sensitive. :-( */
64        setlocale(LC_CTYPE, "");
65
66        if (argc > 1 && strcmp(argv[1], "-x") == 0) {
67                return crypt_main(argc, argv);
68        }
69
70        log_init();
71
72        global.conf_file = g_strdup(CONF_FILE_DEF);
73        global.conf = conf_load(argc, argv);
74        if (global.conf == NULL) {
75                return(1);
76        }
77
78        b_main_init();
79
80        /* libpurple doesn't like fork()s after initializing itself, so if
81           we use it, do this init a little later (in case we're running in
82           ForkDaemon mode). */
83#ifndef WITH_PURPLE
84        nogaim_init();
85#endif
86
87#ifdef OTR_BI
88        otr_init();
89#endif
90
91        global.helpfile = g_strdup(HELP_FILE);
92        if (help_init(&global.help, global.helpfile) == NULL) {
93                log_message(LOGLVL_WARNING, "Error opening helpfile %s.", HELP_FILE);
94        }
95
96        global.storage = storage_init(global.conf->primary_storage, global.conf->migrate_storage);
97        if (global.storage == NULL) {
98                log_message(LOGLVL_ERROR, "Unable to load storage backend '%s'", global.conf->primary_storage);
99                return(1);
100        }
101
102        if (global.conf->runmode == RUNMODE_INETD) {
103                log_link(LOGLVL_ERROR, LOGOUTPUT_IRC);
104                log_link(LOGLVL_WARNING, LOGOUTPUT_IRC);
105
106                i = bitlbee_inetd_init();
107                log_message(LOGLVL_INFO, "%s %s starting in inetd mode.", PACKAGE, BITLBEE_VERSION);
108
109        } else if (global.conf->runmode == RUNMODE_DAEMON) {
110                log_link(LOGLVL_ERROR, LOGOUTPUT_CONSOLE);
111                log_link(LOGLVL_WARNING, LOGOUTPUT_CONSOLE);
112
113                i = bitlbee_daemon_init();
114                log_message(LOGLVL_INFO, "%s %s starting in daemon mode.", PACKAGE, BITLBEE_VERSION);
115        } else if (global.conf->runmode == RUNMODE_FORKDAEMON) {
116                log_link(LOGLVL_ERROR, LOGOUTPUT_CONSOLE);
117                log_link(LOGLVL_WARNING, LOGOUTPUT_CONSOLE);
118
119                /* In case the operator requests a restart, we need this. */
120                old_cwd = g_malloc(256);
121                if (getcwd(old_cwd, 255) == NULL) {
122                        log_message(LOGLVL_WARNING, "Could not save current directory: %s", strerror(errno));
123                        g_free(old_cwd);
124                        old_cwd = NULL;
125                }
126
127                i = bitlbee_daemon_init();
128                log_message(LOGLVL_INFO, "%s %s starting in forking daemon mode.", PACKAGE, BITLBEE_VERSION);
129        }
130        if (i != 0) {
131                return(i);
132        }
133
134        if ((global.conf->user && *global.conf->user) &&
135            (global.conf->runmode == RUNMODE_DAEMON ||
136             global.conf->runmode == RUNMODE_FORKDAEMON) &&
137            (!getuid() || !geteuid())) {
138                struct passwd *pw = NULL;
139                pw = getpwnam(global.conf->user);
140                if (pw) {
141                        initgroups(global.conf->user, pw->pw_gid);
142                        setgid(pw->pw_gid);
143                        setuid(pw->pw_uid);
144                } else {
145                        log_message(LOGLVL_WARNING, "Failed to look up user %s.", global.conf->user);
146                }
147        }
148
149        /* Catch some signals to tell the user what's happening before quitting */
150        memset(&sig, 0, sizeof(sig));
151        sig.sa_handler = SIG_IGN;
152        sigaction(SIGCHLD, &sig, &old);
153        sigaction(SIGPIPE, &sig, &old);
154        sig.sa_flags = SA_RESETHAND;
155        sig.sa_handler = sighandler_crash;
156        sigaction(SIGSEGV, &sig, &old);
157
158        /* Use a pipe for SIGTERM/SIGINT so the actual signal handler doesn't do anything unsafe */
159        if (pipe(signal_shutdown_pipe) == 0) {
160                b_input_add(signal_shutdown_pipe[0], B_EV_IO_READ, bitlbee_shutdown, NULL);
161                sig.sa_handler = sighandler_shutdown;
162                sigaction(SIGINT, &sig, &old);
163                sigaction(SIGTERM, &sig, &old);
164        }
165
166        if (!getuid() || !geteuid()) {
167                log_message(LOGLVL_WARNING, "BitlBee is running with root privileges. Why?");
168        }
169
170        b_main_run();
171
172        /* Mainly good for restarting, to make sure we close the help.txt fd. */
173        help_free(&global.help);
174
175        if (global.restart) {
176                char *fn = ipc_master_save_state();
177                char *env;
178
179                env = g_strdup_printf("_BITLBEE_RESTART_STATE=%s", fn);
180                putenv(env);
181                g_free(fn);
182                /* Looks like env should *not* be freed here as putenv
183                   doesn't make a copy. Odd. */
184
185                i = chdir(old_cwd);
186                close(global.listen_socket);
187
188                if (execv(argv[0], argv) == -1) {
189                        /* Apparently the execve() failed, so let's just
190                           jump back into our own/current main(). */
191                        /* Need more cleanup code to make this work. */
192                        return 1; /* main( argc, argv ); */
193                }
194        }
195
196        return(0);
197}
198
199static int crypt_main(int argc, char *argv[])
200{
201        int pass_len;
202        unsigned char *pass_cr, *pass_cl;
203
204        if (argc < 4 || (strcmp(argv[2], "hash") != 0 &&
205                         strcmp(argv[2], "unhash") != 0 && argc < 5)) {
206                printf("Supported:\n"
207                       "  %s -x enc <key> <cleartext password>\n"
208                       "  %s -x dec <key> <encrypted password>\n"
209                       "  %s -x hash <cleartext password>\n"
210                       "  %s -x unhash <hashed password>\n"
211                       "  %s -x chkhash <hashed password> <cleartext password>\n",
212                       argv[0], argv[0], argv[0], argv[0], argv[0]);
213        } else if (strcmp(argv[2], "enc") == 0) {
214                pass_len = arc_encode(argv[4], strlen(argv[4]), (unsigned char **) &pass_cr, argv[3], 12);
215                printf("%s\n", base64_encode(pass_cr, pass_len));
216        } else if (strcmp(argv[2], "dec") == 0) {
217                pass_len = base64_decode(argv[4], (unsigned char **) &pass_cr);
218                arc_decode(pass_cr, pass_len, (char **) &pass_cl, argv[3]);
219                printf("%s\n", pass_cl);
220        } else if (strcmp(argv[2], "hash") == 0) {
221                md5_byte_t pass_md5[21];
222                md5_state_t md5_state;
223
224                random_bytes(pass_md5 + 16, 5);
225                md5_init(&md5_state);
226                md5_append(&md5_state, (md5_byte_t *) argv[3], strlen(argv[3]));
227                md5_append(&md5_state, pass_md5 + 16, 5);   /* Add the salt. */
228                md5_finish(&md5_state, pass_md5);
229
230                printf("%s\n", base64_encode(pass_md5, 21));
231        } else if (strcmp(argv[2], "unhash") == 0) {
232                printf("Hash %s submitted to a massive Beowulf cluster of\n"
233                       "overclocked 486s. Expect your answer next year somewhere around this time. :-)\n", argv[3]);
234        } else if (strcmp(argv[2], "chkhash") == 0) {
235                char *hash = strncmp(argv[3], "md5:", 4) == 0 ? argv[3] + 4 : argv[3];
236                int st = md5_verify_password(argv[4], hash);
237
238                printf("Hash %s given password.\n", st == 0 ? "matches" : "does not match");
239
240                return st;
241        }
242
243        return 0;
244}
245
246/* Signal handler for SIGTERM and SIGINT */
247static void sighandler_shutdown(int signal)
248{
249        /* Write a single null byte to the pipe, just to send a message to the main loop.
250         * This gets handled by bitlbee_shutdown (the b_input_add callback for this pipe) */
251        write(signal_shutdown_pipe[1], "", 1);
252}
253
254/* Signal handler for SIGSEGV
255 * A desperate attempt to tell the user that everything is wrong in the world.
256 * Avoids using irc_abort() because it has several unsafe calls to malloc */
257static void sighandler_crash(int signal)
258{
259        GSList *l;
260        const char *message = "ERROR :BitlBee crashed! (SIGSEGV received)\r\n";
261        int len = strlen(message);
262
263        for (l = irc_connection_list; l; l = l->next) {
264                irc_t *irc = l->data;
265                write(irc->fd, message, len);
266        }
267
268        raise(signal);
269}
270
271double gettime()
272{
273        struct timeval time[1];
274
275        gettimeofday(time, 0);
276        return((double) time->tv_sec + (double) time->tv_usec / 1000000);
277}
Note: See TracBrowser for help on using the repository browser.