main.c

   1 /*****************************************************************************
   2   Copyright (c) 2006 EMC Corporation.
   3
   4   This program is free software; you can redistribute it and/or modify it
   5   under the terms of the GNU General Public License as published by the Free
   6   Software Foundation; either version 2 of the License, or (at your option)
   7   any later version.
   8
   9   This program is distributed in the hope that it will be useful, but WITHOUT
  10   ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  11   FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
  12   more details.
  13
  14   You should have received a copy of the GNU General Public License along with
  15   this program; if not, write to the Free Software Foundation, Inc., 59
  16   Temple Place - Suite 330, Boston, MA  02111-1307, USA.
  17
  18   The full GNU General Public License is included in this distribution in the
  19   file called LICENSE.
  20
  21   Authors: Srinivas Aji <Aji_Srinivas@emc.com>
  22
  23 ******************************************************************************/
  24
  25 #include "epoll_loop.h"
  26 #include "bridge_ctl.h"
  27 #include "ctl_socket_server.h"
  28 #include "netif_utils.h"
  29 #include "log.h"
  30
  31 #include <stdio.h>
  32 #include <stdlib.h>
  33 #include <unistd.h>
  34 #include <getopt.h>
  35 #include <syslog.h>
  36
  37 static int become_daemon = 1;
  38 static int is_daemon = 0;
  39 int log_level = LOG_LEVEL_DEFAULT;
  40
  41 int main(int argc, char *argv[])
  42 {
  43         int c;
  44         while ((c = getopt(argc, argv, "dv:")) != -1) {
  45                 switch (c) {
  46                 case 'd':
  47                         become_daemon = 0;
  48                         break;
  49                 case 'v':
  50                         {
  51                                 char *end;
  52                                 long l;
  53                                 l = strtoul(optarg, &end, 0);
  54                                 if (*optarg == 0 || *end != 0
  55                                     || l > LOG_LEVEL_MAX) {
  56                                         ERROR("Invalid loglevel %s", optarg);
  57                                         exit(1);
  58                                 }
  59                                 log_level = l;
  60                         }
  61                         break;
  62                 default:
  63                         return -1;
  64                 }
  65         }
  66
  67         TST(init_epoll() == 0, -1);
  68         TST(ctl_socket_init() == 0, -1);
  69
  70         TST(netsock_init() == 0, -1);
  71         TST(init_bridge_ops() == 0, -1);
  72         if (become_daemon) {
  73                 openlog("rstpd", 0, LOG_DAEMON);
  74                 daemon(0, 0);
  75                 is_daemon = 1;
  76         }
  77         return epoll_main_loop();
  78 }
  79
  80 /*********************** Logging *********************/
  81
  82 #include <stdarg.h>
  83 #include <time.h>
  84
  85 void vDprintf(int level, const char *fmt, va_list ap)
  86 {
  87         if (level > log_level)
  88                 return;
  89
  90         if (!is_daemon) {
  91                 char logbuf[256];
  92                 logbuf[255] = 0;
  93                 time_t clock;
  94                 struct tm *local_tm;
  95                 time(&clock);
  96                 local_tm = localtime(&clock);
  97                 int l =
  98                     strftime(logbuf, sizeof(logbuf) - 1, "%F %T ", local_tm);
  99                 vsnprintf(logbuf + l, sizeof(logbuf) - l - 1, fmt, ap);
 100                 printf("%s\n", logbuf);
 101         } else {
 102                 vsyslog((level <= LOG_LEVEL_INFO) ? LOG_INFO : LOG_DEBUG, fmt,
 103                         ap);
 104         }
 105 }
 106
 107 void Dprintf(int level, const char *fmt, ...)
 108 {
 109         va_list ap;
 110         va_start(ap, fmt);
 111         vDprintf(level, fmt, ap);
 112         va_end(ap);
 113 }