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