From b958a9aa60a683a4e4cc1882cb13fd00c8622be6 Mon Sep 17 00:00:00 2001 From: "Zhao, Gang" Date: Fri, 26 Feb 2016 17:18:39 +0800 Subject: [PATCH] cli: add timeout support for listen command Currently 'ubus listen' command does not respect the timeout parameter, i.e., whether timeout parameter is provided, 'ubus listen' will always run infinitely. This patch let 'ubus listen' command support the timeout parameter. For example, 'ubus listen -t 60' will cause the command to exit after 60 seconds. Signed-off-by: Zhao, Gang --- cli.c | 25 ++++++++++++++++++++----- 1 file changed, 20 insertions(+), 5 deletions(-) diff --git a/cli.c b/cli.c index eea43e3..c5cbfc3 100644 --- a/cli.c +++ b/cli.c @@ -140,15 +140,29 @@ static int ubus_cli_call(struct ubus_context *ctx, int argc, char **argv) return ubus_invoke(ctx, id, argv[1], b.head, receive_call_result_data, NULL, timeout * 1000); } +struct cli_listen_data { + struct uloop_timeout timeout; + struct ubus_event_handler ev; + bool timed_out; +}; + +static void listen_timeout(struct uloop_timeout *timeout) +{ + struct cli_listen_data *data = container_of(timeout, struct cli_listen_data, timeout); + data->timed_out = true; + uloop_end(); +} + static int ubus_cli_listen(struct ubus_context *ctx, int argc, char **argv) { - struct ubus_event_handler listener; + struct cli_listen_data data = { + .timeout.cb = listen_timeout, + .ev.cb = receive_event, + .timed_out = false, + }; const char *event; int ret = 0; - memset(&listener, 0, sizeof(listener)); - listener.cb = receive_event; - if (argc > 0) { event = argv[0]; } else { @@ -157,7 +171,7 @@ static int ubus_cli_listen(struct ubus_context *ctx, int argc, char **argv) } do { - ret = ubus_register_event_handler(ctx, &listener, event); + ret = ubus_register_event_handler(ctx, &data.ev, event); if (ret) break; @@ -178,6 +192,7 @@ static int ubus_cli_listen(struct ubus_context *ctx, int argc, char **argv) uloop_init(); ubus_add_uloop(ctx); + uloop_timeout_set(&data.timeout, timeout * 1000); uloop_run(); uloop_done(); -- 2.25.1