]>
Commit | Line | Data |
---|---|---|
064af421 | 1 | /* |
13a233f7 | 2 | * Copyright (c) 2008, 2009, 2010, 2011, 2012, 2013, 2014 Nicira, Inc. |
064af421 | 3 | * |
a14bc59f BP |
4 | * Licensed under the Apache License, Version 2.0 (the "License"); |
5 | * you may not use this file except in compliance with the License. | |
6 | * You may obtain a copy of the License at: | |
064af421 | 7 | * |
a14bc59f BP |
8 | * http://www.apache.org/licenses/LICENSE-2.0 |
9 | * | |
10 | * Unless required by applicable law or agreed to in writing, software | |
11 | * distributed under the License is distributed on an "AS IS" BASIS, | |
12 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |
13 | * See the License for the specific language governing permissions and | |
14 | * limitations under the License. | |
064af421 BP |
15 | */ |
16 | ||
17 | #include <config.h> | |
18 | #include "unixctl.h" | |
064af421 | 19 | #include <errno.h> |
064af421 BP |
20 | #include <unistd.h> |
21 | #include "coverage.h" | |
22 | #include "dirs.h" | |
23 | #include "dynamic-string.h" | |
bde9f75d EJ |
24 | #include "json.h" |
25 | #include "jsonrpc.h" | |
064af421 | 26 | #include "list.h" |
064af421 BP |
27 | #include "poll-loop.h" |
28 | #include "shash.h" | |
bde9f75d | 29 | #include "stream.h" |
cb54a8c5 | 30 | #include "stream-provider.h" |
3c442619 | 31 | #include "svec.h" |
5136ce49 | 32 | #include "vlog.h" |
064af421 | 33 | |
d98e6007 | 34 | VLOG_DEFINE_THIS_MODULE(unixctl); |
d76f09ea BP |
35 | |
36 | COVERAGE_DEFINE(unixctl_received); | |
37 | COVERAGE_DEFINE(unixctl_replied); | |
064af421 BP |
38 | \f |
39 | struct unixctl_command { | |
0e15264f BP |
40 | const char *usage; |
41 | int min_args, max_args; | |
8ca79daa BP |
42 | unixctl_cb_func *cb; |
43 | void *aux; | |
064af421 BP |
44 | }; |
45 | ||
46 | struct unixctl_conn { | |
47 | struct list node; | |
bde9f75d | 48 | struct jsonrpc *rpc; |
064af421 | 49 | |
bde9f75d EJ |
50 | /* Only one request can be in progress at a time. While the request is |
51 | * being processed, 'request_id' is populated, otherwise it is null. */ | |
52 | struct json *request_id; /* ID of the currently active request. */ | |
064af421 BP |
53 | }; |
54 | ||
55 | /* Server for control connection. */ | |
56 | struct unixctl_server { | |
bde9f75d | 57 | struct pstream *listener; |
064af421 BP |
58 | struct list conns; |
59 | }; | |
60 | ||
064af421 BP |
61 | static struct vlog_rate_limit rl = VLOG_RATE_LIMIT_INIT(5, 5); |
62 | ||
63 | static struct shash commands = SHASH_INITIALIZER(&commands); | |
64 | ||
65 | static void | |
0e15264f BP |
66 | unixctl_help(struct unixctl_conn *conn, int argc OVS_UNUSED, |
67 | const char *argv[] OVS_UNUSED, void *aux OVS_UNUSED) | |
064af421 BP |
68 | { |
69 | struct ds ds = DS_EMPTY_INITIALIZER; | |
7ff2009a | 70 | const struct shash_node **nodes = shash_sort(&commands); |
3c442619 | 71 | size_t i; |
064af421 BP |
72 | |
73 | ds_put_cstr(&ds, "The available commands are:\n"); | |
3c442619 | 74 | |
7ff2009a JP |
75 | for (i = 0; i < shash_count(&commands); i++) { |
76 | const struct shash_node *node = nodes[i]; | |
77 | const struct unixctl_command *command = node->data; | |
bde9f75d | 78 | |
e7b5947a | 79 | ds_put_format(&ds, " %-23s %s\n", node->name, command->usage); |
3c442619 | 80 | } |
7ff2009a | 81 | free(nodes); |
3c442619 | 82 | |
bde9f75d | 83 | unixctl_command_reply(conn, ds_cstr(&ds)); |
064af421 BP |
84 | ds_destroy(&ds); |
85 | } | |
86 | ||
d5e1e5ed | 87 | static void |
0e15264f BP |
88 | unixctl_version(struct unixctl_conn *conn, int argc OVS_UNUSED, |
89 | const char *argv[] OVS_UNUSED, void *aux OVS_UNUSED) | |
d5e1e5ed | 90 | { |
bde9f75d | 91 | unixctl_command_reply(conn, get_program_version()); |
d5e1e5ed JP |
92 | } |
93 | ||
0e15264f BP |
94 | /* Registers a unixctl command with the given 'name'. 'usage' describes the |
95 | * arguments to the command; it is used only for presentation to the user in | |
96 | * "help" output. | |
97 | * | |
b558fd90 BP |
98 | * 'cb' is called when the command is received. It is passed an array |
99 | * containing the command name and arguments, plus a copy of 'aux'. Normally | |
100 | * 'cb' should reply by calling unixctl_command_reply() or | |
101 | * unixctl_command_reply_error() before it returns, but if the command cannot | |
102 | * be handled immediately then it can defer the reply until later. A given | |
103 | * connection can only process a single request at a time, so a reply must be | |
104 | * made eventually to avoid blocking that connection. */ | |
064af421 | 105 | void |
0e15264f BP |
106 | unixctl_command_register(const char *name, const char *usage, |
107 | int min_args, int max_args, | |
108 | unixctl_cb_func *cb, void *aux) | |
064af421 BP |
109 | { |
110 | struct unixctl_command *command; | |
4895c701 SH |
111 | struct unixctl_command *lookup = shash_find_data(&commands, name); |
112 | ||
cb22974d | 113 | ovs_assert(!lookup || lookup->cb == cb); |
4895c701 SH |
114 | |
115 | if (lookup) { | |
116 | return; | |
117 | } | |
064af421 | 118 | |
064af421 | 119 | command = xmalloc(sizeof *command); |
0e15264f BP |
120 | command->usage = usage; |
121 | command->min_args = min_args; | |
122 | command->max_args = max_args; | |
064af421 | 123 | command->cb = cb; |
8ca79daa | 124 | command->aux = aux; |
064af421 BP |
125 | shash_add(&commands, name, command); |
126 | } | |
127 | ||
bde9f75d EJ |
128 | static void |
129 | unixctl_command_reply__(struct unixctl_conn *conn, | |
130 | bool success, const char *body) | |
064af421 | 131 | { |
bde9f75d EJ |
132 | struct json *body_json; |
133 | struct jsonrpc_msg *reply; | |
134 | ||
135 | COVERAGE_INC(unixctl_replied); | |
cb22974d | 136 | ovs_assert(conn->request_id); |
bde9f75d EJ |
137 | |
138 | if (!body) { | |
139 | body = ""; | |
064af421 | 140 | } |
bde9f75d EJ |
141 | |
142 | if (body[0] && body[strlen(body) - 1] != '\n') { | |
143 | body_json = json_string_create_nocopy(xasprintf("%s\n", body)); | |
144 | } else { | |
145 | body_json = json_string_create(body); | |
146 | } | |
147 | ||
148 | if (success) { | |
149 | reply = jsonrpc_create_reply(body_json, conn->request_id); | |
150 | } else { | |
151 | reply = jsonrpc_create_error(body_json, conn->request_id); | |
152 | } | |
153 | ||
154 | /* If jsonrpc_send() returns an error, the run loop will take care of the | |
155 | * problem eventually. */ | |
156 | jsonrpc_send(conn->rpc, reply); | |
157 | json_destroy(conn->request_id); | |
158 | conn->request_id = NULL; | |
064af421 BP |
159 | } |
160 | ||
bde9f75d EJ |
161 | /* Replies to the active unixctl connection 'conn'. 'result' is sent to the |
162 | * client indicating the command was processed successfully. Only one call to | |
163 | * unixctl_command_reply() or unixctl_command_reply_error() may be made per | |
164 | * request. */ | |
064af421 | 165 | void |
bde9f75d | 166 | unixctl_command_reply(struct unixctl_conn *conn, const char *result) |
064af421 | 167 | { |
bde9f75d EJ |
168 | unixctl_command_reply__(conn, true, result); |
169 | } | |
064af421 | 170 | |
bde9f75d | 171 | /* Replies to the active unixctl connection 'conn'. 'error' is sent to the |
ec9f40dc | 172 | * client indicating an error occurred processing the command. Only one call to |
bde9f75d EJ |
173 | * unixctl_command_reply() or unixctl_command_reply_error() may be made per |
174 | * request. */ | |
175 | void | |
176 | unixctl_command_reply_error(struct unixctl_conn *conn, const char *error) | |
177 | { | |
178 | unixctl_command_reply__(conn, false, error); | |
064af421 BP |
179 | } |
180 | ||
cb54a8c5 | 181 | /* Creates a unixctl server listening on 'path', which for POSIX may be: |
064af421 BP |
182 | * |
183 | * - NULL, in which case <rundir>/<program>.<pid>.ctl is used. | |
184 | * | |
185 | * - A name that does not start with '/', in which case it is put in | |
186 | * <rundir>. | |
187 | * | |
188 | * - An absolute path (starting with '/') that gives the exact name of | |
189 | * the Unix domain socket to listen on. | |
190 | * | |
cb54a8c5 GS |
191 | * For Windows, a kernel assigned TCP port is used and written in 'path' |
192 | * which may be: | |
193 | * | |
194 | * - NULL, in which case <rundir>/<program>.ctl is used. | |
195 | * | |
196 | * - An absolute path that gives the name of the file. | |
197 | * | |
198 | * For both POSIX and Windows, if the path is "none", the function will | |
199 | * return successfully but no socket will actually be created. | |
200 | * | |
064af421 BP |
201 | * A program that (optionally) daemonizes itself should call this function |
202 | * *after* daemonization, so that the socket name contains the pid of the | |
203 | * daemon instead of the pid of the program that exited. (Otherwise, | |
3fbe1d30 | 204 | * "ovs-appctl --target=<program>" will fail.) |
064af421 BP |
205 | * |
206 | * Returns 0 if successful, otherwise a positive errno value. If successful, | |
614c4892 BP |
207 | * sets '*serverp' to the new unixctl_server (or to NULL if 'path' was "none"), |
208 | * otherwise to NULL. */ | |
064af421 BP |
209 | int |
210 | unixctl_server_create(const char *path, struct unixctl_server **serverp) | |
211 | { | |
212 | struct unixctl_server *server; | |
bde9f75d | 213 | struct pstream *listener; |
e3f512b0 | 214 | char *punix_path; |
064af421 BP |
215 | int error; |
216 | ||
bde9f75d | 217 | *serverp = NULL; |
614c4892 | 218 | if (path && !strcmp(path, "none")) { |
614c4892 BP |
219 | return 0; |
220 | } | |
221 | ||
064af421 | 222 | if (path) { |
e3f512b0 GS |
223 | char *abs_path; |
224 | #ifndef _WIN32 | |
cb54a8c5 | 225 | abs_path = abs_file_name(ovs_rundir(), path); |
e3f512b0 | 226 | #else |
13a233f7 | 227 | abs_path = xstrdup(path); |
e3f512b0 | 228 | #endif |
bde9f75d | 229 | punix_path = xasprintf("punix:%s", abs_path); |
e3f512b0 | 230 | free(abs_path); |
064af421 | 231 | } else { |
e3f512b0 | 232 | #ifndef _WIN32 |
bde9f75d EJ |
233 | punix_path = xasprintf("punix:%s/%s.%ld.ctl", ovs_rundir(), |
234 | program_name, (long int) getpid()); | |
cb54a8c5 | 235 | #else |
e3f512b0 | 236 | punix_path = xasprintf("punix:%s/%s.ctl", ovs_rundir(), program_name); |
cb54a8c5 | 237 | #endif |
e3f512b0 | 238 | } |
064af421 | 239 | |
ef8a3d14 | 240 | error = pstream_open(punix_path, &listener, 0); |
bde9f75d | 241 | if (error) { |
324f0c59 BP |
242 | ovs_error(error, "could not initialize control socket %s", punix_path); |
243 | goto exit; | |
064af421 BP |
244 | } |
245 | ||
bde9f75d EJ |
246 | unixctl_command_register("help", "", 0, 0, unixctl_help, NULL); |
247 | unixctl_command_register("version", "", 0, 0, unixctl_version, NULL); | |
064af421 | 248 | |
bde9f75d EJ |
249 | server = xmalloc(sizeof *server); |
250 | server->listener = listener; | |
251 | list_init(&server->conns); | |
064af421 | 252 | *serverp = server; |
324f0c59 BP |
253 | |
254 | exit: | |
255 | free(punix_path); | |
256 | return error; | |
064af421 BP |
257 | } |
258 | ||
259 | static void | |
bde9f75d | 260 | process_command(struct unixctl_conn *conn, struct jsonrpc_msg *request) |
064af421 | 261 | { |
bde9f75d | 262 | char *error = NULL; |
064af421 | 263 | |
064af421 | 264 | struct unixctl_command *command; |
bde9f75d | 265 | struct json_array *params; |
064af421 BP |
266 | |
267 | COVERAGE_INC(unixctl_received); | |
bde9f75d EJ |
268 | conn->request_id = json_clone(request->id); |
269 | ||
270 | params = json_array(request->params); | |
271 | command = shash_find_data(&commands, request->method); | |
272 | if (!command) { | |
273 | error = xasprintf("\"%s\" is not a valid command", request->method); | |
274 | } else if (params->n < command->min_args) { | |
275 | error = xasprintf("\"%s\" command requires at least %d arguments", | |
276 | request->method, command->min_args); | |
277 | } else if (params->n > command->max_args) { | |
278 | error = xasprintf("\"%s\" command takes at most %d arguments", | |
279 | request->method, command->max_args); | |
064af421 | 280 | } else { |
bde9f75d EJ |
281 | struct svec argv = SVEC_EMPTY_INITIALIZER; |
282 | int i; | |
283 | ||
284 | svec_add(&argv, request->method); | |
285 | for (i = 0; i < params->n; i++) { | |
286 | if (params->elems[i]->type != JSON_STRING) { | |
287 | error = xasprintf("\"%s\" command has non-string argument", | |
288 | request->method); | |
289 | break; | |
290 | } | |
291 | svec_add(&argv, json_string(params->elems[i])); | |
292 | } | |
293 | svec_terminate(&argv); | |
294 | ||
295 | if (!error) { | |
0e15264f BP |
296 | command->cb(conn, argv.n, (const char **) argv.names, |
297 | command->aux); | |
298 | } | |
299 | ||
bde9f75d | 300 | svec_destroy(&argv); |
064af421 | 301 | } |
0e15264f | 302 | |
bde9f75d EJ |
303 | if (error) { |
304 | unixctl_command_reply_error(conn, error); | |
305 | free(error); | |
306 | } | |
064af421 BP |
307 | } |
308 | ||
309 | static int | |
bde9f75d | 310 | run_connection(struct unixctl_conn *conn) |
064af421 | 311 | { |
bde9f75d | 312 | int error, i; |
064af421 | 313 | |
bde9f75d EJ |
314 | jsonrpc_run(conn->rpc); |
315 | error = jsonrpc_get_status(conn->rpc); | |
316 | if (error || jsonrpc_get_backlog(conn->rpc)) { | |
317 | return error; | |
318 | } | |
064af421 | 319 | |
bde9f75d EJ |
320 | for (i = 0; i < 10; i++) { |
321 | struct jsonrpc_msg *msg; | |
064af421 | 322 | |
bde9f75d EJ |
323 | if (error || conn->request_id) { |
324 | break; | |
064af421 BP |
325 | } |
326 | ||
bde9f75d EJ |
327 | jsonrpc_recv(conn->rpc, &msg); |
328 | if (msg) { | |
329 | if (msg->type == JSONRPC_REQUEST) { | |
330 | process_command(conn, msg); | |
064af421 | 331 | } else { |
bde9f75d EJ |
332 | VLOG_WARN_RL(&rl, "%s: received unexpected %s message", |
333 | jsonrpc_get_name(conn->rpc), | |
334 | jsonrpc_msg_type_to_string(msg->type)); | |
335 | error = EINVAL; | |
064af421 | 336 | } |
bde9f75d | 337 | jsonrpc_msg_destroy(msg); |
064af421 | 338 | } |
bde9f75d | 339 | error = error ? error : jsonrpc_get_status(conn->rpc); |
064af421 | 340 | } |
064af421 | 341 | |
bde9f75d | 342 | return error; |
064af421 BP |
343 | } |
344 | ||
345 | static void | |
346 | kill_connection(struct unixctl_conn *conn) | |
347 | { | |
348 | list_remove(&conn->node); | |
bde9f75d EJ |
349 | jsonrpc_close(conn->rpc); |
350 | json_destroy(conn->request_id); | |
064af421 BP |
351 | free(conn); |
352 | } | |
353 | ||
354 | void | |
355 | unixctl_server_run(struct unixctl_server *server) | |
356 | { | |
357 | struct unixctl_conn *conn, *next; | |
358 | int i; | |
359 | ||
614c4892 BP |
360 | if (!server) { |
361 | return; | |
362 | } | |
363 | ||
064af421 | 364 | for (i = 0; i < 10; i++) { |
bde9f75d EJ |
365 | struct stream *stream; |
366 | int error; | |
367 | ||
368 | error = pstream_accept(server->listener, &stream); | |
369 | if (!error) { | |
370 | struct unixctl_conn *conn = xzalloc(sizeof *conn); | |
371 | list_push_back(&server->conns, &conn->node); | |
372 | conn->rpc = jsonrpc_open(stream); | |
373 | } else if (error == EAGAIN) { | |
064af421 | 374 | break; |
bde9f75d EJ |
375 | } else { |
376 | VLOG_WARN_RL(&rl, "%s: accept failed: %s", | |
377 | pstream_get_name(server->listener), | |
10a89ef0 | 378 | ovs_strerror(error)); |
064af421 | 379 | } |
064af421 BP |
380 | } |
381 | ||
4e8e4213 | 382 | LIST_FOR_EACH_SAFE (conn, next, node, &server->conns) { |
064af421 BP |
383 | int error = run_connection(conn); |
384 | if (error && error != EAGAIN) { | |
385 | kill_connection(conn); | |
386 | } | |
387 | } | |
388 | } | |
389 | ||
390 | void | |
391 | unixctl_server_wait(struct unixctl_server *server) | |
392 | { | |
393 | struct unixctl_conn *conn; | |
394 | ||
614c4892 BP |
395 | if (!server) { |
396 | return; | |
397 | } | |
398 | ||
bde9f75d | 399 | pstream_wait(server->listener); |
4e8e4213 | 400 | LIST_FOR_EACH (conn, node, &server->conns) { |
bde9f75d EJ |
401 | jsonrpc_wait(conn->rpc); |
402 | if (!jsonrpc_get_backlog(conn->rpc)) { | |
403 | jsonrpc_recv_wait(conn->rpc); | |
064af421 BP |
404 | } |
405 | } | |
406 | } | |
407 | ||
408 | /* Destroys 'server' and stops listening for connections. */ | |
409 | void | |
410 | unixctl_server_destroy(struct unixctl_server *server) | |
411 | { | |
412 | if (server) { | |
413 | struct unixctl_conn *conn, *next; | |
414 | ||
4e8e4213 | 415 | LIST_FOR_EACH_SAFE (conn, next, node, &server->conns) { |
064af421 BP |
416 | kill_connection(conn); |
417 | } | |
418 | ||
bde9f75d | 419 | pstream_close(server->listener); |
064af421 BP |
420 | free(server); |
421 | } | |
422 | } | |
423 | \f | |
cb54a8c5 GS |
424 | /* On POSIX based systems, connects to a unixctl server socket. 'path' should |
425 | * be the name of a unixctl server socket. If it does not start with '/', it | |
426 | * will be prefixed with the rundir (e.g. /usr/local/var/run/openvswitch). | |
427 | * | |
428 | * On Windows, connects to a localhost TCP port as written inside 'path'. | |
429 | * 'path' should be an absolute path of the file. | |
064af421 BP |
430 | * |
431 | * Returns 0 if successful, otherwise a positive errno value. If successful, | |
bde9f75d | 432 | * sets '*client' to the new jsonrpc, otherwise to NULL. */ |
064af421 | 433 | int |
bde9f75d | 434 | unixctl_client_create(const char *path, struct jsonrpc **client) |
064af421 | 435 | { |
bde9f75d EJ |
436 | char *abs_path, *unix_path; |
437 | struct stream *stream; | |
064af421 | 438 | int error; |
cb54a8c5 | 439 | |
e3f512b0 | 440 | #ifdef _WIN32 |
13a233f7 | 441 | abs_path = xstrdup(path); |
cb54a8c5 | 442 | #else |
bde9f75d | 443 | abs_path = abs_file_name(ovs_rundir(), path); |
cb54a8c5 | 444 | #endif |
e3f512b0 | 445 | unix_path = xasprintf("unix:%s", abs_path); |
cb54a8c5 GS |
446 | |
447 | *client = NULL; | |
448 | ||
f125905c MM |
449 | error = stream_open_block(stream_open(unix_path, &stream, DSCP_DEFAULT), |
450 | &stream); | |
bde9f75d EJ |
451 | free(unix_path); |
452 | free(abs_path); | |
064af421 | 453 | |
bde9f75d EJ |
454 | if (error) { |
455 | VLOG_WARN("failed to connect to %s", path); | |
456 | return error; | |
064af421 | 457 | } |
bde9f75d EJ |
458 | |
459 | *client = jsonrpc_open(stream); | |
460 | return 0; | |
064af421 BP |
461 | } |
462 | ||
bde9f75d EJ |
463 | /* Executes 'command' on the server with an argument vector 'argv' containing |
464 | * 'argc' elements. If successfully communicated with the server, returns 0 | |
465 | * and sets '*result', or '*err' (not both) to the result or error the server | |
466 | * returned. Otherwise, sets '*result' and '*err' to NULL and returns a | |
467 | * positive errno value. The caller is responsible for freeing '*result' or | |
468 | * '*err' if not NULL. */ | |
064af421 | 469 | int |
bde9f75d EJ |
470 | unixctl_client_transact(struct jsonrpc *client, const char *command, int argc, |
471 | char *argv[], char **result, char **err) | |
064af421 | 472 | { |
bde9f75d EJ |
473 | struct jsonrpc_msg *request, *reply; |
474 | struct json **json_args, *params; | |
475 | int error, i; | |
064af421 | 476 | |
bde9f75d EJ |
477 | *result = NULL; |
478 | *err = NULL; | |
479 | ||
480 | json_args = xmalloc(argc * sizeof *json_args); | |
481 | for (i = 0; i < argc; i++) { | |
482 | json_args[i] = json_string_create(argv[i]); | |
064af421 | 483 | } |
bde9f75d EJ |
484 | params = json_array_create(json_args, argc); |
485 | request = jsonrpc_create_request(command, params, NULL); | |
486 | ||
487 | error = jsonrpc_transact_block(client, request, &reply); | |
488 | if (error) { | |
489 | VLOG_WARN("error communicating with %s: %s", jsonrpc_get_name(client), | |
2bf1d3cc | 490 | ovs_retval_to_string(error)); |
bde9f75d | 491 | return error; |
064af421 BP |
492 | } |
493 | ||
bde9f75d EJ |
494 | if (reply->error) { |
495 | if (reply->error->type == JSON_STRING) { | |
496 | *err = xstrdup(json_string(reply->error)); | |
497 | } else { | |
498 | VLOG_WARN("%s: unexpected error type in JSON RPC reply: %s", | |
499 | jsonrpc_get_name(client), | |
500 | json_type_to_string(reply->error->type)); | |
501 | error = EINVAL; | |
064af421 | 502 | } |
bde9f75d EJ |
503 | } else if (reply->result) { |
504 | if (reply->result->type == JSON_STRING) { | |
505 | *result = xstrdup(json_string(reply->result)); | |
064af421 | 506 | } else { |
bde9f75d EJ |
507 | VLOG_WARN("%s: unexpected result type in JSON rpc reply: %s", |
508 | jsonrpc_get_name(client), | |
509 | json_type_to_string(reply->result->type)); | |
510 | error = EINVAL; | |
064af421 BP |
511 | } |
512 | } | |
064af421 | 513 | |
bde9f75d EJ |
514 | jsonrpc_msg_destroy(reply); |
515 | return error; | |
064af421 | 516 | } |