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 "packet.h"
  30 #include "log.h"
  31
  32 #include <stdio.h>
  33 #include <stdlib.h>
  34 #include <unistd.h>
  35 #include <getopt.h>
  36 #include <syslog.h>
  37 #include <errno.h>
  38
  39 static int become_daemon = 1;
  40 static int is_daemon = 0;
  41 int log_level = LOG_LEVEL_DEFAULT;
  42
  43 int main(int argc, char *argv[])
  44 {
  45         int c;
  46         while ((c = getopt(argc, argv, "dv:")) != -1) {
  47                 switch (c) {
  48                 case 'd':
  49                         become_daemon = 0;
  50                         break;
  51                 case 'v':
  52                         {
  53                                 char *end;
  54                                 long l;
  55                                 l = strtoul(optarg, &end, 0);
  56                                 if (*optarg == 0 || *end != 0
  57                                     || l > LOG_LEVEL_MAX) {
  58                                         ERROR("Invalid loglevel %s", optarg);
  59                                         exit(1);
  60                                 }
  61                                 log_level = l;
  62                         }
  63                         break;
  64                 default:
  65                         return -1;
  66                 }
  67         }
  68
  69         TST(init_epoll() == 0, -1);
  70         TST(ctl_socket_init() == 0, -1);
  71         TST(packet_sock_init() == 0, -1);
  72         TST(netsock_init() == 0, -1);
  73         TST(init_bridge_ops() == 0, -1);
  74         if (become_daemon) {
  75                 FILE *f = fopen("/var/run/rstpd.pid", "w");
  76                 if (!f) {
  77                         ERROR("can't open /var/run/rstp.pid");
  78                         return -1;
  79                 }
  80                 openlog("rstpd", 0, LOG_DAEMON);
  81                 daemon(0, 0);
  82                 is_daemon = 1;
  83                 fprintf(f, "%d", getpid());
  84                 fclose(f);
  85         }
  86         return epoll_main_loop();
  87 }
  88
  89 /*********************** Logging *********************/
  90
  91 #include <stdarg.h>
  92 #include <time.h>
  93
  94 void vDprintf(int level, const char *fmt, va_list ap)
  95 {
  96         if (level > log_level)
  97                 return;
  98
  99         if (!is_daemon) {
 100                 char logbuf[256];
 101                 logbuf[255] = 0;
 102                 time_t clock;
 103                 struct tm *local_tm;
 104                 time(&clock);
 105                 local_tm = localtime(&clock);
 106                 int l =
 107                     strftime(logbuf, sizeof(logbuf) - 1, "%F %T ", local_tm);
 108                 vsnprintf(logbuf + l, sizeof(logbuf) - l - 1, fmt, ap);
 109                 printf("%s\n", logbuf);
 110         } else {
 111                 vsyslog((level <= LOG_LEVEL_INFO) ? LOG_INFO : LOG_DEBUG, fmt,
 112                         ap);
 113         }
 114 }
 115
 116 void Dprintf(int level, const char *fmt, ...)
 117 {
 118         va_list ap;
 119         va_start(ap, fmt);
 120         vDprintf(level, fmt, ap);
 121         va_end(ap);
 122 }