pintos-os.org Git - openvswitch/blob - vswitchd/ovs-vswitchd.c

   1 /* Copyright (c) 2008, 2009, 2010, 2011 Nicira Networks
   2  *
   3  * Licensed under the Apache License, Version 2.0 (the "License");
   4  * you may not use this file except in compliance with the License.
   5  * You may obtain a copy of the License at:
   6  *
   7  *     http://www.apache.org/licenses/LICENSE-2.0
   8  *
   9  * Unless required by applicable law or agreed to in writing, software
  10  * distributed under the License is distributed on an "AS IS" BASIS,
  11  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12  * See the License for the specific language governing permissions and
  13  * limitations under the License.
  14  */
  15
  16 #include <config.h>
  17
  18 #include <assert.h>
  19 #include <errno.h>
  20 #include <getopt.h>
  21 #include <limits.h>
  22 #include <signal.h>
  23 #include <stdlib.h>
  24 #include <string.h>
  25 #ifdef HAVE_MLOCKALL
  26 #include <sys/mman.h>
  27 #endif
  28
  29 #include "bridge.h"
  30 #include "command-line.h"
  31 #include "compiler.h"
  32 #include "daemon.h"
  33 #include "dummy.h"
  34 #include "leak-checker.h"
  35 #include "netdev.h"
  36 #include "ovsdb-idl.h"
  37 #include "poll-loop.h"
  38 #include "process.h"
  39 #include "signals.h"
  40 #include "stream-ssl.h"
  41 #include "stream.h"
  42 #include "stress.h"
  43 #include "svec.h"
  44 #include "timeval.h"
  45 #include "unixctl.h"
  46 #include "util.h"
  47 #include "vconn.h"
  48 #include "vlog.h"
  49 #include "vswitchd/vswitch-idl.h"
  50
  51 VLOG_DEFINE_THIS_MODULE(vswitchd);
  52
  53 static unixctl_cb_func ovs_vswitchd_exit;
  54
  55 static const char *parse_options(int argc, char *argv[]);
  56 static void usage(void) NO_RETURN;
  57
  58 int
  59 main(int argc, char *argv[])
  60 {
  61     struct unixctl_server *unixctl;
  62     struct signal *sighup;
  63     const char *remote;
  64     bool exiting;
  65     int retval;
  66
  67     proctitle_init(argc, argv);
  68     set_program_name(argv[0]);
  69     stress_init_command();
  70     remote = parse_options(argc, argv);
  71     signal(SIGPIPE, SIG_IGN);
  72     sighup = signal_register(SIGHUP);
  73     process_init();
  74     ovsrec_init();
  75
  76     daemonize_start();
  77
  78     retval = unixctl_server_create(NULL, &unixctl);
  79     if (retval) {
  80         exit(EXIT_FAILURE);
  81     }
  82     unixctl_command_register("exit", ovs_vswitchd_exit, &exiting);
  83
  84     bridge_init(remote);
  85     exiting = false;
  86     while (!exiting) {
  87         if (signal_poll(sighup)) {
  88             vlog_reopen_log_file();
  89         }
  90         bridge_run();
  91         unixctl_server_run(unixctl);
  92         netdev_run();
  93
  94         signal_wait(sighup);
  95         bridge_wait();
  96         unixctl_server_wait(unixctl);
  97         netdev_wait();
  98         if (exiting) {
  99             poll_immediate_wake();
 100         }
 101         poll_block();
 102     }
 103     bridge_exit();
 104     unixctl_server_destroy(unixctl);
 105     signal_unregister(sighup);
 106
 107     return 0;
 108 }
 109
 110 static const char *
 111 parse_options(int argc, char *argv[])
 112 {
 113     enum {
 114         OPT_PEER_CA_CERT = UCHAR_MAX + 1,
 115         OPT_MLOCKALL,
 116         VLOG_OPTION_ENUMS,
 117         LEAK_CHECKER_OPTION_ENUMS,
 118         OPT_BOOTSTRAP_CA_CERT,
 119         OPT_ENABLE_DUMMY,
 120         DAEMON_OPTION_ENUMS
 121     };
 122     static struct option long_options[] = {
 123         {"help",        no_argument, NULL, 'h'},
 124         {"version",     no_argument, NULL, 'V'},
 125         {"mlockall",    no_argument, NULL, OPT_MLOCKALL},
 126         DAEMON_LONG_OPTIONS,
 127         VLOG_LONG_OPTIONS,
 128         LEAK_CHECKER_LONG_OPTIONS,
 129         STREAM_SSL_LONG_OPTIONS,
 130         {"peer-ca-cert", required_argument, NULL, OPT_PEER_CA_CERT},
 131         {"bootstrap-ca-cert", required_argument, NULL, OPT_BOOTSTRAP_CA_CERT},
 132         {"enable-dummy", no_argument, NULL, OPT_ENABLE_DUMMY},
 133         {NULL, 0, NULL, 0},
 134     };
 135     char *short_options = long_options_to_short_options(long_options);
 136
 137     for (;;) {
 138         int c;
 139
 140         c = getopt_long(argc, argv, short_options, long_options, NULL);
 141         if (c == -1) {
 142             break;
 143         }
 144
 145         switch (c) {
 146         case 'H':
 147         case 'h':
 148             usage();
 149
 150         case 'V':
 151             OVS_PRINT_VERSION(OFP_VERSION, OFP_VERSION);
 152             exit(EXIT_SUCCESS);
 153
 154         case OPT_MLOCKALL:
 155 #ifdef HAVE_MLOCKALL
 156             if (mlockall(MCL_CURRENT | MCL_FUTURE)) {
 157                 VLOG_ERR("mlockall failed: %s", strerror(errno));
 158             }
 159 #else
 160             VLOG_ERR("mlockall not supported on this system");
 161 #endif
 162             break;
 163
 164         VLOG_OPTION_HANDLERS
 165         DAEMON_OPTION_HANDLERS
 166         LEAK_CHECKER_OPTION_HANDLERS
 167         STREAM_SSL_OPTION_HANDLERS
 168
 169         case OPT_PEER_CA_CERT:
 170             stream_ssl_set_peer_ca_cert_file(optarg);
 171             break;
 172
 173         case OPT_BOOTSTRAP_CA_CERT:
 174             stream_ssl_set_ca_cert_file(optarg, true);
 175             break;
 176
 177         case OPT_ENABLE_DUMMY:
 178             dummy_enable();
 179             break;
 180
 181         case '?':
 182             exit(EXIT_FAILURE);
 183
 184         default:
 185             abort();
 186         }
 187     }
 188     free(short_options);
 189
 190     argc -= optind;
 191     argv += optind;
 192
 193     if (argc != 1) {
 194         VLOG_FATAL("database socket is only non-option argument; "
 195                    "use --help for usage");
 196     }
 197
 198     return argv[0];
 199 }
 200
 201 static void
 202 usage(void)
 203 {
 204     printf("%s: Open vSwitch daemon\n"
 205            "usage: %s [OPTIONS] DATABASE\n"
 206            "where DATABASE is a socket on which ovsdb-server is listening.\n",
 207            program_name, program_name);
 208     stream_usage("DATABASE", true, false, true);
 209     daemon_usage();
 210     vlog_usage();
 211     printf("\nOther options:\n"
 212            "  -h, --help              display this help message\n"
 213            "  -V, --version           display version information\n");
 214     leak_checker_usage();
 215     exit(EXIT_SUCCESS);
 216 }
 217
 218 static void
 219 ovs_vswitchd_exit(struct unixctl_conn *conn, const char *args OVS_UNUSED,
 220                   void *exiting_)
 221 {
 222     bool *exiting = exiting_;
 223     *exiting = true;
 224     unixctl_command_reply(conn, 200, NULL);
 225 }