]>
git.proxmox.com Git - ovs.git/blob - lib/unixctl.c
2 * Copyright (c) 2008, 2009, 2010, 2011, 2012 Nicira Networks.
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:
8 * http://www.apache.org/licenses/LICENSE-2.0
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.
24 #include "dynamic-string.h"
28 #include "poll-loop.h"
34 VLOG_DEFINE_THIS_MODULE(unixctl
);
36 COVERAGE_DEFINE(unixctl_received
);
37 COVERAGE_DEFINE(unixctl_replied
);
39 struct unixctl_command
{
41 int min_args
, max_args
;
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. */
55 /* Server for control connection. */
56 struct unixctl_server
{
57 struct pstream
*listener
;
61 static struct vlog_rate_limit rl
= VLOG_RATE_LIMIT_INIT(5, 5);
63 static struct shash commands
= SHASH_INITIALIZER(&commands
);
66 unixctl_help(struct unixctl_conn
*conn
, int argc OVS_UNUSED
,
67 const char *argv
[] OVS_UNUSED
, void *aux OVS_UNUSED
)
69 struct ds ds
= DS_EMPTY_INITIALIZER
;
70 const struct shash_node
**nodes
= shash_sort(&commands
);
73 ds_put_cstr(&ds
, "The available commands are:\n");
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
;
79 ds_put_format(&ds
, " %-23s %s\n", node
->name
, command
->usage
);
83 unixctl_command_reply(conn
, ds_cstr(&ds
));
88 unixctl_version(struct unixctl_conn
*conn
, int argc OVS_UNUSED
,
89 const char *argv
[] OVS_UNUSED
, void *aux OVS_UNUSED
)
91 unixctl_command_reply(conn
, get_program_version());
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
98 * 'cb' is called when the command is received. It is passed the actual set of
99 * arguments, as a text string, plus a copy of 'aux'. Normally 'cb' should
100 * reply by calling unixctl_command_reply() or unixctl_command_reply_error()
101 * before it returns, but if the command cannot be handled immediately then it
102 * can defer the reply until later. A given connection can only process a
103 * single request at a time, so a reply must be made eventually to avoid
104 * blocking that connection. */
106 unixctl_command_register(const char *name
, const char *usage
,
107 int min_args
, int max_args
,
108 unixctl_cb_func
*cb
, void *aux
)
110 struct unixctl_command
*command
;
111 struct unixctl_command
*lookup
= shash_find_data(&commands
, name
);
113 assert(!lookup
|| lookup
->cb
== cb
);
119 command
= xmalloc(sizeof *command
);
120 command
->usage
= usage
;
121 command
->min_args
= min_args
;
122 command
->max_args
= max_args
;
125 shash_add(&commands
, name
, command
);
129 unixctl_command_reply__(struct unixctl_conn
*conn
,
130 bool success
, const char *body
)
132 struct json
*body_json
;
133 struct jsonrpc_msg
*reply
;
135 COVERAGE_INC(unixctl_replied
);
136 assert(conn
->request_id
);
142 if (body
[0] && body
[strlen(body
) - 1] != '\n') {
143 body_json
= json_string_create_nocopy(xasprintf("%s\n", body
));
145 body_json
= json_string_create(body
);
149 reply
= jsonrpc_create_reply(body_json
, conn
->request_id
);
151 reply
= jsonrpc_create_error(body_json
, conn
->request_id
);
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
;
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
166 unixctl_command_reply(struct unixctl_conn
*conn
, const char *result
)
168 unixctl_command_reply__(conn
, true, result
);
171 /* Replies to the active unixctl connection 'conn'. 'error' is sent to the
172 * client indicating an error occured processing the command. Only one call to
173 * unixctl_command_reply() or unixctl_command_reply_error() may be made per
176 unixctl_command_reply_error(struct unixctl_conn
*conn
, const char *error
)
178 unixctl_command_reply__(conn
, false, error
);
181 /* Creates a unixctl server listening on 'path', which may be:
183 * - NULL, in which case <rundir>/<program>.<pid>.ctl is used.
185 * - "none", in which case the function will return successfully but
186 * no socket will actually be created.
188 * - A name that does not start with '/', in which case it is put in
191 * - An absolute path (starting with '/') that gives the exact name of
192 * the Unix domain socket to listen on.
194 * A program that (optionally) daemonizes itself should call this function
195 * *after* daemonization, so that the socket name contains the pid of the
196 * daemon instead of the pid of the program that exited. (Otherwise,
197 * "ovs-appctl --target=<program>" will fail.)
199 * Returns 0 if successful, otherwise a positive errno value. If successful,
200 * sets '*serverp' to the new unixctl_server (or to NULL if 'path' was "none"),
201 * otherwise to NULL. */
203 unixctl_server_create(const char *path
, struct unixctl_server
**serverp
)
205 struct unixctl_server
*server
;
206 struct pstream
*listener
;
211 if (path
&& !strcmp(path
, "none")) {
216 char *abs_path
= abs_file_name(ovs_rundir(), path
);
217 punix_path
= xasprintf("punix:%s", abs_path
);
220 punix_path
= xasprintf("punix:%s/%s.%ld.ctl", ovs_rundir(),
221 program_name
, (long int) getpid());
224 error
= pstream_open(punix_path
, &listener
, DSCP_INVALID
);
226 ovs_error(error
, "could not initialize control socket %s", punix_path
);
230 unixctl_command_register("help", "", 0, 0, unixctl_help
, NULL
);
231 unixctl_command_register("version", "", 0, 0, unixctl_version
, NULL
);
233 server
= xmalloc(sizeof *server
);
234 server
->listener
= listener
;
235 list_init(&server
->conns
);
244 process_command(struct unixctl_conn
*conn
, struct jsonrpc_msg
*request
)
248 struct unixctl_command
*command
;
249 struct json_array
*params
;
251 COVERAGE_INC(unixctl_received
);
252 conn
->request_id
= json_clone(request
->id
);
254 params
= json_array(request
->params
);
255 command
= shash_find_data(&commands
, request
->method
);
257 error
= xasprintf("\"%s\" is not a valid command", request
->method
);
258 } else if (params
->n
< command
->min_args
) {
259 error
= xasprintf("\"%s\" command requires at least %d arguments",
260 request
->method
, command
->min_args
);
261 } else if (params
->n
> command
->max_args
) {
262 error
= xasprintf("\"%s\" command takes at most %d arguments",
263 request
->method
, command
->max_args
);
265 struct svec argv
= SVEC_EMPTY_INITIALIZER
;
268 svec_add(&argv
, request
->method
);
269 for (i
= 0; i
< params
->n
; i
++) {
270 if (params
->elems
[i
]->type
!= JSON_STRING
) {
271 error
= xasprintf("\"%s\" command has non-string argument",
275 svec_add(&argv
, json_string(params
->elems
[i
]));
277 svec_terminate(&argv
);
280 command
->cb(conn
, argv
.n
, (const char **) argv
.names
,
288 unixctl_command_reply_error(conn
, error
);
294 run_connection(struct unixctl_conn
*conn
)
298 jsonrpc_run(conn
->rpc
);
299 error
= jsonrpc_get_status(conn
->rpc
);
300 if (error
|| jsonrpc_get_backlog(conn
->rpc
)) {
304 for (i
= 0; i
< 10; i
++) {
305 struct jsonrpc_msg
*msg
;
307 if (error
|| conn
->request_id
) {
311 jsonrpc_recv(conn
->rpc
, &msg
);
313 if (msg
->type
== JSONRPC_REQUEST
) {
314 process_command(conn
, msg
);
316 VLOG_WARN_RL(&rl
, "%s: received unexpected %s message",
317 jsonrpc_get_name(conn
->rpc
),
318 jsonrpc_msg_type_to_string(msg
->type
));
321 jsonrpc_msg_destroy(msg
);
323 error
= error
? error
: jsonrpc_get_status(conn
->rpc
);
330 kill_connection(struct unixctl_conn
*conn
)
332 list_remove(&conn
->node
);
333 jsonrpc_close(conn
->rpc
);
334 json_destroy(conn
->request_id
);
339 unixctl_server_run(struct unixctl_server
*server
)
341 struct unixctl_conn
*conn
, *next
;
348 for (i
= 0; i
< 10; i
++) {
349 struct stream
*stream
;
352 error
= pstream_accept(server
->listener
, &stream
);
354 struct unixctl_conn
*conn
= xzalloc(sizeof *conn
);
355 list_push_back(&server
->conns
, &conn
->node
);
356 conn
->rpc
= jsonrpc_open(stream
);
357 } else if (error
== EAGAIN
) {
360 VLOG_WARN_RL(&rl
, "%s: accept failed: %s",
361 pstream_get_name(server
->listener
),
366 LIST_FOR_EACH_SAFE (conn
, next
, node
, &server
->conns
) {
367 int error
= run_connection(conn
);
368 if (error
&& error
!= EAGAIN
) {
369 kill_connection(conn
);
375 unixctl_server_wait(struct unixctl_server
*server
)
377 struct unixctl_conn
*conn
;
383 pstream_wait(server
->listener
);
384 LIST_FOR_EACH (conn
, node
, &server
->conns
) {
385 jsonrpc_wait(conn
->rpc
);
386 if (!jsonrpc_get_backlog(conn
->rpc
)) {
387 jsonrpc_recv_wait(conn
->rpc
);
392 /* Destroys 'server' and stops listening for connections. */
394 unixctl_server_destroy(struct unixctl_server
*server
)
397 struct unixctl_conn
*conn
, *next
;
399 LIST_FOR_EACH_SAFE (conn
, next
, node
, &server
->conns
) {
400 kill_connection(conn
);
403 pstream_close(server
->listener
);
408 /* Connects to a unixctl server socket. 'path' should be the name of a unixctl
409 * server socket. If it does not start with '/', it will be prefixed with the
410 * rundir (e.g. /usr/local/var/run/openvswitch).
412 * Returns 0 if successful, otherwise a positive errno value. If successful,
413 * sets '*client' to the new jsonrpc, otherwise to NULL. */
415 unixctl_client_create(const char *path
, struct jsonrpc
**client
)
417 char *abs_path
, *unix_path
;
418 struct stream
*stream
;
423 abs_path
= abs_file_name(ovs_rundir(), path
);
424 unix_path
= xasprintf("unix:%s", abs_path
);
425 error
= stream_open_block(stream_open(unix_path
, &stream
, DSCP_DEFAULT
),
431 VLOG_WARN("failed to connect to %s", path
);
435 *client
= jsonrpc_open(stream
);
439 /* Executes 'command' on the server with an argument vector 'argv' containing
440 * 'argc' elements. If successfully communicated with the server, returns 0
441 * and sets '*result', or '*err' (not both) to the result or error the server
442 * returned. Otherwise, sets '*result' and '*err' to NULL and returns a
443 * positive errno value. The caller is responsible for freeing '*result' or
444 * '*err' if not NULL. */
446 unixctl_client_transact(struct jsonrpc
*client
, const char *command
, int argc
,
447 char *argv
[], char **result
, char **err
)
449 struct jsonrpc_msg
*request
, *reply
;
450 struct json
**json_args
, *params
;
456 json_args
= xmalloc(argc
* sizeof *json_args
);
457 for (i
= 0; i
< argc
; i
++) {
458 json_args
[i
] = json_string_create(argv
[i
]);
460 params
= json_array_create(json_args
, argc
);
461 request
= jsonrpc_create_request(command
, params
, NULL
);
463 error
= jsonrpc_transact_block(client
, request
, &reply
);
465 VLOG_WARN("error communicating with %s: %s", jsonrpc_get_name(client
),
471 if (reply
->error
->type
== JSON_STRING
) {
472 *err
= xstrdup(json_string(reply
->error
));
474 VLOG_WARN("%s: unexpected error type in JSON RPC reply: %s",
475 jsonrpc_get_name(client
),
476 json_type_to_string(reply
->error
->type
));
479 } else if (reply
->result
) {
480 if (reply
->result
->type
== JSON_STRING
) {
481 *result
= xstrdup(json_string(reply
->result
));
483 VLOG_WARN("%s: unexpected result type in JSON rpc reply: %s",
484 jsonrpc_get_name(client
),
485 json_type_to_string(reply
->result
->type
));
490 jsonrpc_msg_destroy(reply
);