main.c 12.6 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12
/* Icecast
 *
 * This program is distributed under the GNU General Public License, version 2.
 * A copy of this license is included with this source.
 *
 * Copyright 2000-2004, Jack Moffitt <jack@xiph.org, 
 *                      Michael Smith <msmith@xiph.org>,
 *                      oddsock <oddsock@xiph.org>,
 *                      Karl Heyes <karl@xiph.org>
 *                      and others (see AUTHORS for details).
 */

13
/* -*- c-basic-offset: 4; indent-tabs-mode: nil; -*- */
14 15 16 17
#ifdef HAVE_CONFIG_H
#include <config.h>
#endif

Jack Moffitt's avatar
Jack Moffitt committed
18 19 20
#include <stdio.h>
#include <string.h>

21 22 23 24
#ifdef HAVE_UNISTD_H
# include <unistd.h>
#endif

Karl Heyes's avatar
Karl Heyes committed
25 26 27 28 29
#include "thread/thread.h"
#include "avl/avl.h"
#include "net/sock.h"
#include "net/resolver.h"
#include "httpp/httpp.h"
Jack Moffitt's avatar
Jack Moffitt committed
30

31 32 33 34 35 36
#ifdef CHUID
#include <sys/types.h>
#include <grp.h>
#include <pwd.h>
#include <errno.h>
#endif
Jack Moffitt's avatar
Jack Moffitt committed
37

38
#include "cfgfile.h"
Jack Moffitt's avatar
Jack Moffitt committed
39 40 41 42 43 44 45
#include "sighandler.h"

#include "global.h"
#include "os.h"
#include "connection.h"
#include "refbuf.h"
#include "client.h"
46
#include "slave.h"
Jack Moffitt's avatar
Jack Moffitt committed
47 48
#include "stats.h"
#include "logging.h"
49
#include "xslt.h"
50
#include "fserve.h"
51
#include "yp.h"
52
#include "auth.h"
Jack Moffitt's avatar
Jack Moffitt committed
53

54 55
#include <libxml/xmlmemory.h>

56
#ifdef _WIN32
57 58
/* For getpid() */
#include <process.h>
59
#define snprintf _snprintf
60
#define getpid _getpid
61
#endif
62

Jack Moffitt's avatar
Jack Moffitt committed
63 64 65
#undef CATMODULE
#define CATMODULE "main"

66 67 68 69 70 71 72 73 74
static void _fatal_error(char *perr)
{
#ifdef WIN32
    MessageBox(NULL, perr, "Error", MB_OK);
#else
    fprintf(stdout, "%s\n", perr);
#endif
}

75
static void _print_usage()
Jack Moffitt's avatar
Jack Moffitt committed
76
{
77
    printf(ICECAST_VERSION_STRING "\n\n");
78
    printf("usage: icecast [-b -v] -c <file>\n");
79
    printf("options:\n");
80 81 82
    printf("\t-c <file>\tSpecify configuration file\n");
    printf("\t-v\t\tDisplay version info\n");
    printf("\t-b\t\tRun icecast in the background\n");
83
    printf("\n");
Jack Moffitt's avatar
Jack Moffitt committed
84 85
}

86 87
static void _stop_logging(void)
{
88 89
    log_close(errorlog);
    log_close(accesslog);
90
    log_close(playlistlog);
91 92
}

93
static void _initialize_subsystems(void)
Jack Moffitt's avatar
Jack Moffitt committed
94
{
95 96 97 98 99 100 101 102
    log_initialize();
    thread_initialize();
    sock_initialize();
    resolver_initialize();
    config_initialize();
    connection_initialize();
    global_initialize();
    refbuf_initialize();
103
    xslt_initialize();
Jack Moffitt's avatar
Jack Moffitt committed
104 105
}

106
static void _shutdown_subsystems(void)
Jack Moffitt's avatar
Jack Moffitt committed
107
{
108
    fserve_shutdown();
109
    xslt_shutdown();
110 111
    refbuf_shutdown();
    slave_shutdown();
112
    auth_shutdown();
113
    yp_shutdown();
114
    stats_shutdown();
115

116 117 118 119 120 121
    global_shutdown();
    connection_shutdown();
    config_shutdown();
    resolver_shutdown();
    sock_shutdown();
    thread_shutdown();
122 123 124

    /* Now that these are done, we can stop the loggers. */
    _stop_logging();
125
    log_shutdown();
126 127

    xmlCleanupParser();
Jack Moffitt's avatar
Jack Moffitt committed
128 129
}

130
static int _parse_config_opts(int argc, char **argv, char *filename, int size)
Jack Moffitt's avatar
Jack Moffitt committed
131
{
132
    int i = 1;
133 134
    int config_ok = 0;

Jack Moffitt's avatar
Jack Moffitt committed
135

136
    if (argc < 2) return -1;
Jack Moffitt's avatar
Jack Moffitt committed
137

138 139
    while (i < argc) {
        if (strcmp(argv[i], "-b") == 0) {
140
#ifndef WIN32
Michael Smith's avatar
Michael Smith committed
141
            pid_t pid;
142
            fprintf(stdout, "Starting icecast2\nDetaching from the console\n");
Michael Smith's avatar
Michael Smith committed
143 144 145 146

            pid = fork();

            if (pid > 0) {
147
                /* exit the parent */
148 149
                exit(0);
            }
Michael Smith's avatar
Michael Smith committed
150
            else if(pid < 0) {
151 152
                fprintf(stderr, "FATAL: Unable to fork child!");
                exit(1);
153
            }
154
#endif
155
        }
156 157
        if (strcmp(argv[i], "-v") == 0) {
            fprintf(stdout, "%s\n", ICECAST_VERSION_STRING);
158 159 160
            exit(0);
        }

161 162 163
        if (strcmp(argv[i], "-c") == 0) {
            if (i + 1 < argc) {
                strncpy(filename, argv[i + 1], size-1);
164
                filename[size-1] = 0;
165
                config_ok = 1;
166 167 168 169 170 171 172
            } else {
                return -1;
            }
        }
        i++;
    }

173 174 175 176
    if(config_ok)
        return 1;
    else
        return -1;
Jack Moffitt's avatar
Jack Moffitt committed
177 178
}

179
static int _start_logging(void)
Jack Moffitt's avatar
Jack Moffitt committed
180
{
181 182
    char fn_error[FILENAME_MAX];
    char fn_access[FILENAME_MAX];
183
    char fn_playlist[FILENAME_MAX];
184
    char buf[1024];
185
    int log_to_stderr;
186

187
    ice_config_t *config = config_get_config_unlocked();
Jack Moffitt's avatar
Jack Moffitt committed
188

189
    if(strcmp(config->error_log, "-")) {
190
        snprintf(fn_error, FILENAME_MAX, "%s%s%s", config->log_dir, PATH_SEPARATOR, config->error_log);
191
        errorlog = log_open(fn_error);
192
        log_to_stderr = 0;
193 194
    } else {
        errorlog = log_open_file(stderr);
195
        log_to_stderr = 1;
196
    }
197 198 199

    if (errorlog < 0) {
        buf[sizeof(buf)-1] = 0;
200 201 202
        snprintf(buf, sizeof(buf)-1, 
                "FATAL: could not open error logging (%s): %s",
                log_to_stderr?"standard error":fn_error,
203 204 205 206 207
                strerror(errno));
        _fatal_error(buf);
    }
    log_set_level(errorlog, config->loglevel);

208
    if(strcmp(config->access_log, "-")) {
209
        snprintf(fn_access, FILENAME_MAX, "%s%s%s", config->log_dir, PATH_SEPARATOR, config->access_log);
210
        accesslog = log_open(fn_access);
211
        log_to_stderr = 0;
212 213
    } else {
        accesslog = log_open_file(stderr);
214
        log_to_stderr = 1;
215
    }
216 217 218

    if (accesslog < 0) {
        buf[sizeof(buf)-1] = 0;
219 220 221
        snprintf(buf, sizeof(buf)-1, 
                "FATAL: could not open access logging (%s): %s",
                log_to_stderr?"standard error":fn_access,
222 223 224
                strerror(errno));
        _fatal_error(buf);
    }
225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240

    if(config->playlist_log) {
        snprintf(fn_playlist, FILENAME_MAX, "%s%s%s", config->log_dir, PATH_SEPARATOR, config->playlist_log);
        playlistlog = log_open(fn_playlist);
        if (playlistlog < 0) {
            buf[sizeof(buf)-1] = 0;
            snprintf(buf, sizeof(buf)-1, 
                "FATAL: could not open playlist logging (%s): %s",
                log_to_stderr?"standard error":fn_playlist,
                strerror(errno));
            _fatal_error(buf);
        }
        log_to_stderr = 0;
    } else {
        playlistlog = -1;
    }
241

242 243
    log_set_level(errorlog, config->loglevel);
    log_set_level(accesslog, 4);
244
    log_set_level(playlistlog, 4);
245 246 247 248

    if (errorlog >= 0 && accesslog >= 0) return 1;
    
    return 0;
Jack Moffitt's avatar
Jack Moffitt committed
249 250
}

251
static int _setup_sockets(void)
Jack Moffitt's avatar
Jack Moffitt committed
252
{
253
    ice_config_t *config;
254 255 256
    int i = 0;
    int ret = 0;
    int successful = 0;
257
    char pbuf[1024];
Jack Moffitt's avatar
Jack Moffitt committed
258

259
    config = config_get_config_unlocked();
Jack Moffitt's avatar
Jack Moffitt committed
260

261 262 263 264 265 266 267
    for(i = 0; i < MAX_LISTEN_SOCKETS; i++) {
        if(config->listeners[i].port <= 0)
            break;

        global.serversock[i] = sock_get_server_socket(
                config->listeners[i].port, config->listeners[i].bind_address);

268
        if (global.serversock[i] == SOCK_ERROR) {
269 270 271 272 273
            memset(pbuf, '\000', sizeof(pbuf));
            snprintf(pbuf, sizeof(pbuf)-1, 
                "Could not create listener socket on port %d", 
                config->listeners[i].port);
            _fatal_error(pbuf);
274 275 276 277 278 279 280
            return 0;
        }
        else {
            ret = 1;
            successful++;
        }
    }
Michael Smith's avatar
Michael Smith committed
281

282
    global.server_sockets = successful;
283 284
    
    return ret;
Jack Moffitt's avatar
Jack Moffitt committed
285 286
}

287
static int _start_listening(void)
Jack Moffitt's avatar
Jack Moffitt committed
288
{
289 290
    int i;
    for(i=0; i < global.server_sockets; i++) {
291 292
        if (sock_listen(global.serversock[i], ICE_LISTEN_QUEUE) == SOCK_ERROR)
            return 0;
Jack Moffitt's avatar
Jack Moffitt committed
293

294
        sock_set_blocking(global.serversock[i], SOCK_NONBLOCK);
295
    }
Jack Moffitt's avatar
Jack Moffitt committed
296

297
    return 1;
Jack Moffitt's avatar
Jack Moffitt committed
298 299
}

300
/* bind the socket and start listening */
301
static int _server_proc_init(void)
Jack Moffitt's avatar
Jack Moffitt committed
302
{
303 304
    if (!_setup_sockets())
        return 0;
Jack Moffitt's avatar
Jack Moffitt committed
305

306
    if (!_start_listening()) {
307
        _fatal_error("Failed trying to listen on server socket");
308 309
        return 0;
    }
310 311

    return 1;
312
}
Jack Moffitt's avatar
Jack Moffitt committed
313

314 315 316
/* this is the heart of the beast */
static void _server_proc(void)
{
317 318
    int i;

319
    connection_accept_loop();
Jack Moffitt's avatar
Jack Moffitt committed
320

321
    for(i=0; i < MAX_LISTEN_SOCKETS; i++)
322
        sock_close(global.serversock[i]);
Jack Moffitt's avatar
Jack Moffitt committed
323 324
}

325
/* chroot the process. Watch out - we need to do this before starting other
326
 * threads. Change uid as well, after figuring out uid _first_ */
327

Michael Smith's avatar
Michael Smith committed
328
static void _ch_root_uid_setup(void)
329
{
Michael Smith's avatar
Michael Smith committed
330
   ice_config_t *conf = config_get_config_unlocked();
331 332 333 334 335 336 337 338
#ifdef CHUID
   struct passwd *user;
   struct group *group;
   uid_t uid=-1;
   gid_t gid=-1;

   if(conf->chuid)
   {
339 340 341 342 343 344 345 346 347
       if(conf->user) {
           user = getpwnam(conf->user);
           if(user)
               uid = user->pw_uid;
           else
               fprintf(stderr, "Couldn't find user \"%s\" in password file\n", conf->user);
       }
       if(conf->group) {
           group = getgrnam(conf->group);
348

349 350 351 352 353
           if(group)
               gid = group->gr_gid;
           else
               fprintf(stderr, "Couldn't find group \"%s\" in groups file\n", conf->group);
       }
354 355 356 357
   }
#endif

#ifdef CHROOT
358 359 360 361 362 363 364 365 366 367 368 369 370 371 372 373 374 375
   if (conf->chroot)
   {
       if(getuid()) /* root check */
       {
           fprintf(stderr, "WARNING: Cannot change server root unless running as root.\n");
           return;
       }
       if(chroot(conf->base_dir))
       {
           fprintf(stderr,"WARNING: Couldn't change server root: %s\n", strerror(errno));
           return;
       }
       else
           fprintf(stdout, "Changed root successfully to \"%s\".\n", conf->base_dir);

   }   
#endif
#ifdef CHUID
376

377 378 379 380 381 382 383 384
   if(conf->chuid)
   {
       if(getuid()) /* root check */
       {
           fprintf(stderr, "WARNING: Can't change user id unless you are root.\n");
           return;
       }

385 386
       if(gid != -1) {
           if(!setgid(gid))
387
               fprintf(stdout, "Changed groupid to %i.\n", (int)gid);
388 389 390
           else
               fprintf(stdout, "Error changing groupid: %s.\n", strerror(errno));
       }
391

392 393
       if(uid != -1) {
           if(!setuid(uid))
394
               fprintf(stdout, "Changed userid to %i.\n", (int)uid);
395 396 397
           else
               fprintf(stdout, "Error changing userid: %s.\n", strerror(errno));
       }
398 399
   }
#endif
400
}
401

Jack Moffitt's avatar
Jack Moffitt committed
402 403
int main(int argc, char **argv)
{
404
    int res, ret;
405 406
    ice_config_t *config;
    char *pidfile = NULL;
407
    char filename[512];
408
    char pbuf[1024];
409 410 411 412

    /* parse the '-c icecast.xml' option
    ** only, so that we can read a configfile
    */
413
    res = _parse_config_opts(argc, argv, filename, 512);
414 415 416 417 418
    if (res == 1) {
        /* startup all the modules */
        _initialize_subsystems();

        /* parse the config file */
Michael Smith's avatar
Michael Smith committed
419
        config_get_config();
420
        ret = config_initial_parse_file(filename);
Michael Smith's avatar
Michael Smith committed
421
        config_release_config();
422
        if (ret < 0) {
423 424 425 426
            memset(pbuf, '\000', sizeof(pbuf));
            snprintf(pbuf, sizeof(pbuf)-1, 
                "FATAL: error parsing config file (%s)", filename);
            _fatal_error(pbuf);
427 428
            switch (ret) {
            case CONFIG_EINSANE:
429
                _fatal_error("filename was null or blank");
430 431
                break;
            case CONFIG_ENOROOT:
432
                _fatal_error("no root element found");
433 434
                break;
            case CONFIG_EBADROOT:
435
                _fatal_error("root element is not <icecast>");
436 437
                break;
            default:
438
                _fatal_error("XML config parsing error");
439 440
                break;
            }
441 442
            _shutdown_subsystems();
            return 1;
443 444 445 446 447 448 449 450
        }
    } else if (res == -1) {
        _print_usage();
        return 1;
    }
    
    /* override config file options with commandline options */
    config_parse_cmdline(argc, argv);
Jack Moffitt's avatar
Jack Moffitt committed
451

452 453
    /* Bind socket, before we change userid */
    if(!_server_proc_init()) {
454
        _fatal_error("Server startup failed. Exiting");
455 456 457
        _shutdown_subsystems();
        return 1;
    }
458

Michael Smith's avatar
Michael Smith committed
459
    _ch_root_uid_setup(); /* Change user id and root if requested/possible */
460 461

    stats_initialize(); /* We have to do this later on because of threading */
462
    fserve_initialize(); /* This too */
463 464 465 466 467 468

#ifdef CHUID 
    /* We'll only have getuid() if we also have setuid(), it's reasonable to
     * assume */
    if(!getuid()) /* Running as root! Don't allow this */
    {
469
        fprintf(stderr, "ERROR: You should not run icecast2 as root\n");
470 471 472 473 474 475 476 477 478
        fprintf(stderr, "Use the changeowner directive in the config file\n");
        _shutdown_subsystems();
        return 1;
    }
#endif

    /* setup default signal handlers */
    sighandler_initialize();

479
    if (!_start_logging()) {
480
        _fatal_error("FATAL: Could not start logging");
481 482 483
        _shutdown_subsystems();
        return 1;
    }
Jack Moffitt's avatar
Jack Moffitt committed
484

485 486 487 488 489 490 491 492
    config = config_get_config_unlocked();
    /* recreate the pid file */
    if (config->pidfile)
    {
        FILE *f;
        pidfile = strdup (config->pidfile);
        if (pidfile && (f = fopen (config->pidfile, "w")) != NULL)
        {
493
            fprintf (f, "%d\n", (int)getpid());
494 495 496
            fclose (f);
        }
    }
497

498
    INFO0 (ICECAST_VERSION_STRING " server started");
Jack Moffitt's avatar
Jack Moffitt committed
499

500
    /* REM 3D Graphics */
Jack Moffitt's avatar
Jack Moffitt committed
501

502 503
    /* let her rip */
    global.running = ICE_RUNNING;
504 505 506 507

    /* Startup yp thread */
    yp_initialize();

508 509
    /* Do this after logging init */
    slave_initialize();
510
    auth_initialise ();
511

512
    _server_proc();
Jack Moffitt's avatar
Jack Moffitt committed
513

514
    INFO0("Shutting down");
Jack Moffitt's avatar
Jack Moffitt committed
515

516
    _shutdown_subsystems();
Jack Moffitt's avatar
Jack Moffitt committed
517

518 519 520 521 522 523
    if (pidfile)
    {
        remove (pidfile);
        free (pidfile);
    }

524
    return 0;
Jack Moffitt's avatar
Jack Moffitt committed
525 526 527
}