+#include <json/json.h>
#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 bool blobmsg_add_element(struct blob_buf *b, const char *name, json_object *obj);
+
+static bool blobmsg_add_object(struct blob_buf *b, json_object *obj)
+{
+ json_object_object_foreach(obj, key, val) {
+ if (!blobmsg_add_element(b, key, val))
+ return false;
+ }
+ return true;
+}
+
+static bool blobmsg_add_array(struct blob_buf *b, struct array_list *a)
+{
+ int i, len;
+
+ for (i = 0, len = array_list_length(a); i < len; i++) {
+ if (!blobmsg_add_element(b, NULL, array_list_get_idx(a, i)))
+ return false;
+ }
+
+ return true;
+}
+
+static bool blobmsg_add_element(struct blob_buf *b, const char *name, json_object *obj)
+{
+ bool ret = true;
+ void *c;
+
+ if (!obj)
+ return false;
+
+ switch (json_object_get_type(obj)) {
+ case json_type_object:
+ c = blobmsg_open_table(b, name);
+ ret = blobmsg_add_object(b, obj);
+ blobmsg_close_table(b, c);
+ break;
+ case json_type_array:
+ c = blobmsg_open_array(b, name);
+ ret = blobmsg_add_array(b, json_object_get_array(obj));
+ blobmsg_close_array(b, c);
+ break;
+ case json_type_string:
+ blobmsg_add_string(b, name, json_object_get_string(obj));
+ break;
+ case json_type_boolean:
+ blobmsg_add_u8(b, name, json_object_get_boolean(obj));
+ break;
+ case json_type_int:
+ blobmsg_add_u32(b, name, json_object_get_int(obj));
+ break;
+ default:
+ return false;
+ }
+ return ret;
+}
+
+static bool blobmsg_from_json(struct blob_buf *b, const char *str)
{
- struct blob_attr **attr, *cur;
+ json_object *obj;
+ bool ret = false;
+
+ obj = json_tokener_parse(str);
+ if (is_error(obj))
+ return false;
+
+ if (json_object_get_type(obj) != json_type_object)
+ goto out;
+
+ ret = blobmsg_add_object(b, obj);
+
+out:
+ json_object_put(obj);
+ return ret;
+}
+
+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])
- return;
-
- fprintf(stderr, "'%s' @%08x\n",
- (char *) blob_data(attr[UBUS_ATTR_OBJPATH]),
- blob_get_int32(attr[UBUS_ATTR_OBJID]));
+ fprintf(stderr, "'%s' @%08x\n", obj->path, obj->id);
- if (!attr[UBUS_ATTR_SIGNATURE])
+ if (!obj->signature)
return;
- blob_for_each_attr(cur, attr[UBUS_ATTR_SIGNATURE], rem) {
+ blob_for_each_attr(cur, obj->signature, rem) {
s = blobmsg_format_json(cur, false);
fprintf(stderr, "\t%s\n", s);
free(s);
}
}
+static void receive_data(struct ubus_request *req, int type, struct blob_attr *msg)
+{
+ char *str;
+ if (!msg)
+ return;
+
+ str = blobmsg_format_json(msg, true);
+ fprintf(stderr, "%s\n", str);
+ free(str);
+}
+
+
static int usage(char *prog)
{
fprintf(stderr,
"Usage: %s <command> [arguments...]\n"
"Commands:\n"
- " - list [<path>] List objects\n"
+ " - list [<path>] List objects\n"
+ " - call <path> <method> [<message>] Call an object method\n"
+ " - listen [<path>...] Listen for events\n"
"\n", prog);
return 1;
}
+static void receive_event(struct ubus_context *ctx, struct ubus_event_handler *ev,
+ const char *type, struct blob_attr *msg)
+{
+ char *str;
+
+ if (msg)
+ str = blobmsg_format_json(msg, true);
+ else
+ str = "";
+
+ fprintf(stderr, "\"%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;
+}
+
int main(int argc, char **argv)
{
- struct ubus_request req;
+ static struct ubus_context *ctx;
char *cmd;
- int ret;
+ int ret = 0;
ctx = ubus_connect(NULL);
if (!ctx) {
if (argc < 2)
return usage(argv[0]);
- blob_buf_init(&b, 0);
-
if (!strcmp(cmd, "list")) {
+ const char *path = NULL;
+
if (argc == 3)
- blob_put_string(&b, UBUS_ATTR_OBJPATH, argv[2]);
+ path = argv[2];
- ubus_start_request(ctx, &req, b.head, UBUS_MSG_LOOKUP, 0);
- req.data_cb = receive_lookup;
- ret = ubus_complete_request(ctx, &req);
- if (ret)
- fprintf(stderr, "Failed: %d\n", ret);
+ ret = ubus_lookup(ctx, path, receive_lookup, NULL);
+ } else if (!strcmp(cmd, "call")) {
+ uint32_t id;
+
+ if (argc < 4 || argc > 5)
+ return usage(argv[0]);
+
+ blob_buf_init(&b, 0);
+ if (argc == 5 && !blobmsg_from_json(&b, argv[4])) {
+ fprintf(stderr, "Failed to parse message data\n");
+ goto out;
+ }
+
+ ret = ubus_lookup_id(ctx, argv[2], &id);
+ if (!ret)
+ ret = ubus_invoke(ctx, id, argv[3], b.head, receive_data, NULL);
+ } else if (!strcmp(cmd, "listen")) {
+ ret = ubus_cli_listen(ctx, argc - 2, argv + 2);
} else {
return usage(argv[0]);
}
+ if (ret)
+ fprintf(stderr, "Failed: %s\n", ubus_strerror(ret));
+
+out:
ubus_free(ctx);
- return 0;
+ return ret;
}