X-Git-Url: http://git.archive.openwrt.org/?p=project%2Fubus.git;a=blobdiff_plain;f=cli.c;h=3be02aefe7d93c9793bc537024782d40a0e576f5;hp=bc884cd2b346eb68d40d215e2a2e0d2097b196fd;hb=87bb1cb17112b1473079c9ae76117f57554e7153;hpb=8321f8c523c9bf5f476f9c5b06c955d0ce3a0f80 diff --git a/cli.c b/cli.c index bc884cd..3be02ae 100644 --- a/cli.c +++ b/cli.c @@ -1,82 +1,205 @@ +#include + +#include #include "libubus.h" static struct blob_buf b; -static struct ubus_context *ctx; -static void receive_lookup(struct ubus_request *req, int type, struct blob_attr *msg) +static const char * const attr_types[] = { + [BLOBMSG_TYPE_INT32] = "\"Integer\"", + [BLOBMSG_TYPE_STRING] = "\"String\"", +}; + +static const char *format_type(void *priv, struct blob_attr *attr) { - struct blob_attr **attr, *cur; + const char *type = NULL; + int typeid; + + if (blob_id(attr) != BLOBMSG_TYPE_INT32) + return NULL; + + typeid = blobmsg_get_u32(attr); + if (typeid < ARRAY_SIZE(attr_types)) + type = attr_types[typeid]; + if (!type) + type = "\"(unknown)\""; + + return type; +} + +static void receive_lookup(struct ubus_context *ctx, struct ubus_object_data *obj, void *priv) +{ + struct blob_attr *cur; char *s; int rem; - attr = ubus_parse_msg(msg); - if (!attr[UBUS_ATTR_OBJID] || !attr[UBUS_ATTR_OBJPATH]) + printf("'%s' @%08x\n", obj->path, obj->id); + + if (!obj->signature) return; - fprintf(stderr, "'%s' @%08x\n", - (char *) blob_data(attr[UBUS_ATTR_OBJPATH]), - blob_get_int32(attr[UBUS_ATTR_OBJID])); + blob_for_each_attr(cur, obj->signature, rem) { + s = blobmsg_format_json_with_cb(cur, false, format_type, NULL); + printf("\t%s\n", s); + free(s); + } +} - if (!attr[UBUS_ATTR_SIGNATURE]) +static void receive_data(struct ubus_request *req, int type, struct blob_attr *msg) +{ + char *str; + if (!msg) return; - blob_for_each_attr(cur, attr[UBUS_ATTR_SIGNATURE], rem) { - s = blobmsg_format_json(cur, false); - fprintf(stderr, "\t%s\n", s); - free(s); + str = blobmsg_format_json(msg, true); + printf("%s\n", str); + free(str); +} + + +static void receive_event(struct ubus_context *ctx, struct ubus_event_handler *ev, + const char *type, struct blob_attr *msg) +{ + char *str; + + str = blobmsg_format_json(msg, true); + printf("\"%s\": %s\n", type, str); + free(str); +} + +static int ubus_cli_listen(struct ubus_context *ctx, int argc, char **argv) +{ + static struct ubus_event_handler listener; + const char *event; + int ret = 0; + + memset(&listener, 0, sizeof(listener)); + listener.cb = receive_event; + + if (!argc) { + event = "*"; + ret = ubus_register_event_handler(ctx, &listener, NULL); + } + + for (;argc;argv++, argc--) { + event = argv[0]; + ret = ubus_register_event_handler(ctx, &listener, argv[0]); + if (ret) + break; + } + + if (ret) { + fprintf(stderr, "Error while registering for event '%s': %s\n", + event, ubus_strerror(ret)); } + + uloop_init(); + ubus_add_uloop(ctx); + uloop_run(); + uloop_done(); + + return 0; } -static int usage(char *prog) +static int ubus_cli_send(struct ubus_context *ctx, int argc, char **argv) +{ + blob_buf_init(&b, 0); + if (argc == 2 && !blobmsg_add_json_from_string(&b, argv[1])) { + fprintf(stderr, "Failed to parse message data\n"); + return UBUS_STATUS_INVALID_ARGUMENT; + } + + return ubus_send_event(ctx, argv[0], b.head); +} + +static int usage(const char *prog) { fprintf(stderr, - "Usage: %s [arguments...]\n" + "Usage: %s [] [arguments...]\n" + "Options:\n" + " -s : Set the unix domain socket to connect to\n" + "\n" "Commands:\n" " - list [] List objects\n" " - call [] Call an object method\n" + " - listen [...] Listen for events\n" + " - send [] Send an event\n" "\n", prog); return 1; } int main(int argc, char **argv) { - struct ubus_request req; + const char *progname, *ubus_socket = NULL; + static struct ubus_context *ctx; char *cmd; - int ret; + int ret = 0; + int ch; + + progname = argv[0]; + + while ((ch = getopt(argc, argv, "s:")) != -1) { + switch (ch) { + case 's': + ubus_socket = optarg; + break; + default: + return usage(progname); + } + } - ctx = ubus_connect(NULL); + argc -= optind; + argv += optind; + + ctx = ubus_connect(ubus_socket); if (!ctx) { fprintf(stderr, "Failed to connect to ubus\n"); return -1; } - cmd = argv[1]; - if (argc < 2) - return usage(argv[0]); + cmd = argv[0]; + if (argc < 1) + return usage(progname); - blob_buf_init(&b, 0); + argv++; + argc--; if (!strcmp(cmd, "list")) { - if (argc == 3) - blob_put_string(&b, UBUS_ATTR_OBJPATH, argv[2]); + const char *path = NULL; - ubus_start_request(ctx, &req, b.head, UBUS_MSG_LOOKUP, 0); - req.data_cb = receive_lookup; - } else if (!strcmp(cmd, "call")) { - if (argc < 4 || argc > 5) - return usage(argv[0]); + if (argc == 1) + path = argv[0]; - blob_put_string(&b, UBUS_ATTR_OBJPATH, argv[2]); - blob_put_string(&b, UBUS_ATTR_METHOD, argv[3]); - ubus_start_request(ctx, &req, b.head, UBUS_MSG_INVOKE, 0); + ret = ubus_lookup(ctx, path, receive_lookup, NULL); + } else if (!strcmp(cmd, "call")) { + uint32_t id; + + if (argc < 2 || argc > 3) + return usage(progname); + + blob_buf_init(&b, 0); + if (argc == 3 && !blobmsg_add_json_from_string(&b, argv[2])) { + fprintf(stderr, "Failed to parse message data\n"); + goto out; + } + + ret = ubus_lookup_id(ctx, argv[0], &id); + if (!ret) + ret = ubus_invoke(ctx, id, argv[1], b.head, receive_data, NULL); + } else if (!strcmp(cmd, "listen")) { + ret = ubus_cli_listen(ctx, argc, argv); + } else if (!strcmp(cmd, "send")) { + if (argc < 1 || argc > 2) + return usage(progname); + ret = ubus_cli_send(ctx, argc, argv); } else { - return usage(argv[0]); + return usage(progname); } - ret = ubus_complete_request(ctx, &req); if (ret) fprintf(stderr, "Failed: %s\n", ubus_strerror(ret)); +out: ubus_free(ctx); - return 0; + return ret; }