daemon: Define daemon options enums the same way as other option enums.
[cascardo/ovs.git] / vswitchd / ovs-vswitchd.c
1 /* Copyright (c) 2008, 2009, 2010 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 "dpif.h"
34 #include "dummy.h"
35 #include "leak-checker.h"
36 #include "netdev.h"
37 #include "ovsdb-idl.h"
38 #include "poll-loop.h"
39 #include "proc-net-compat.h"
40 #include "process.h"
41 #include "signals.h"
42 #include "stream-ssl.h"
43 #include "stream.h"
44 #include "stress.h"
45 #include "svec.h"
46 #include "timeval.h"
47 #include "unixctl.h"
48 #include "util.h"
49 #include "vconn.h"
50 #include "vlog.h"
51 #include "vswitchd/vswitch-idl.h"
52
53 VLOG_DEFINE_THIS_MODULE(vswitchd);
54
55 static unixctl_cb_func ovs_vswitchd_exit;
56
57 static const char *parse_options(int argc, char *argv[]);
58 static void usage(void) NO_RETURN;
59
60 int
61 main(int argc, char *argv[])
62 {
63     struct unixctl_server *unixctl;
64     struct signal *sighup;
65     const char *remote;
66     bool exiting;
67     int retval;
68
69     proctitle_init(argc, argv);
70     set_program_name(argv[0]);
71     stress_init_command();
72     remote = parse_options(argc, argv);
73     signal(SIGPIPE, SIG_IGN);
74     sighup = signal_register(SIGHUP);
75     process_init();
76     ovsrec_init();
77
78     die_if_already_running();
79     daemonize_start();
80
81     retval = unixctl_server_create(NULL, &unixctl);
82     if (retval) {
83         exit(EXIT_FAILURE);
84     }
85     unixctl_command_register("exit", ovs_vswitchd_exit, &exiting);
86
87     daemonize_complete();
88
89     bridge_init(remote);
90     exiting = false;
91     while (!exiting) {
92         if (signal_poll(sighup)) {
93             vlog_reopen_log_file();
94         }
95         bridge_run();
96         unixctl_server_run(unixctl);
97         dp_run();
98         netdev_run();
99
100         signal_wait(sighup);
101         bridge_wait();
102         unixctl_server_wait(unixctl);
103         dp_wait();
104         netdev_wait();
105         if (exiting) {
106             poll_immediate_wake();
107         }
108         poll_block();
109     }
110     bridge_exit();
111     unixctl_server_destroy(unixctl);
112
113     return 0;
114 }
115
116 static const char *
117 parse_options(int argc, char *argv[])
118 {
119     enum {
120         OPT_PEER_CA_CERT = UCHAR_MAX + 1,
121         OPT_MLOCKALL,
122         OPT_FAKE_PROC_NET,
123         VLOG_OPTION_ENUMS,
124         LEAK_CHECKER_OPTION_ENUMS,
125         OPT_BOOTSTRAP_CA_CERT,
126         OPT_ENABLE_DUMMY,
127         DAEMON_OPTION_ENUMS
128     };
129     static struct option long_options[] = {
130         {"help",        no_argument, 0, 'h'},
131         {"version",     no_argument, 0, 'V'},
132         {"mlockall",    no_argument, 0, OPT_MLOCKALL},
133         {"fake-proc-net", no_argument, 0, OPT_FAKE_PROC_NET},
134         DAEMON_LONG_OPTIONS,
135         VLOG_LONG_OPTIONS,
136         LEAK_CHECKER_LONG_OPTIONS,
137 #ifdef HAVE_OPENSSL
138         STREAM_SSL_LONG_OPTIONS
139         {"peer-ca-cert", required_argument, 0, OPT_PEER_CA_CERT},
140         {"bootstrap-ca-cert", required_argument, 0, OPT_BOOTSTRAP_CA_CERT},
141 #endif
142         {"enable-dummy", no_argument, 0, OPT_ENABLE_DUMMY},
143         {0, 0, 0, 0},
144     };
145     char *short_options = long_options_to_short_options(long_options);
146     int error;
147
148     for (;;) {
149         int c;
150
151         c = getopt_long(argc, argv, short_options, long_options, NULL);
152         if (c == -1) {
153             break;
154         }
155
156         switch (c) {
157         case 'H':
158         case 'h':
159             usage();
160
161         case 'V':
162             OVS_PRINT_VERSION(OFP_VERSION, OFP_VERSION);
163             exit(EXIT_SUCCESS);
164
165         case OPT_MLOCKALL:
166 #ifdef HAVE_MLOCKALL
167             if (mlockall(MCL_CURRENT | MCL_FUTURE)) {
168                 VLOG_ERR("mlockall failed: %s", strerror(errno));
169             }
170 #else
171             VLOG_ERR("mlockall not supported on this system");
172 #endif
173             break;
174
175         case OPT_FAKE_PROC_NET:
176             error = proc_net_compat_init();
177             if (error) {
178                 ovs_fatal(error, "failed to initialize /proc/net "
179                           "compatibility");
180             }
181             break;
182
183         VLOG_OPTION_HANDLERS
184         DAEMON_OPTION_HANDLERS
185         LEAK_CHECKER_OPTION_HANDLERS
186
187 #ifdef HAVE_OPENSSL
188         STREAM_SSL_OPTION_HANDLERS
189
190         case OPT_PEER_CA_CERT:
191             stream_ssl_set_peer_ca_cert_file(optarg);
192             break;
193
194         case OPT_BOOTSTRAP_CA_CERT:
195             stream_ssl_set_ca_cert_file(optarg, true);
196             break;
197 #endif
198
199         case OPT_ENABLE_DUMMY:
200             dummy_enable();
201             break;
202
203         case '?':
204             exit(EXIT_FAILURE);
205
206         default:
207             abort();
208         }
209     }
210     free(short_options);
211
212     argc -= optind;
213     argv += optind;
214
215     if (argc != 1) {
216         ovs_fatal(0, "database socket is only non-option argument; "
217                 "use --help for usage");
218     }
219
220     return argv[0];
221 }
222
223 static void
224 usage(void)
225 {
226     printf("%s: Open vSwitch daemon\n"
227            "usage: %s [OPTIONS] DATABASE\n"
228            "where DATABASE is a socket on which ovsdb-server is listening.\n",
229            program_name, program_name);
230     stream_usage("DATABASE", true, false, true);
231     daemon_usage();
232     vlog_usage();
233     printf("\nLegacy compatibility options:\n"
234            " --fake-proc-net          simulate some files in /proc/net\n"
235            "\nOther options:\n"
236            "  -h, --help              display this help message\n"
237            "  -V, --version           display version information\n");
238     leak_checker_usage();
239     exit(EXIT_SUCCESS);
240 }
241
242 static void
243 ovs_vswitchd_exit(struct unixctl_conn *conn, const char *args OVS_UNUSED,
244                   void *exiting_)
245 {
246     bool *exiting = exiting_;
247     *exiting = true;
248     unixctl_command_reply(conn, 200, NULL);
249 }