cli.c

   1 #include <json/json.h>
   2 #include "libubus.h"
   3
   4 static struct blob_buf b;
   5
   6
   7 static bool blobmsg_add_element(struct blob_buf *b, const char *name, json_object *obj);
   8
   9 static bool blobmsg_add_object(struct blob_buf *b, json_object *obj)
  10 {
  11         json_object_object_foreach(obj, key, val) {
  12                 if (!blobmsg_add_element(b, key, val))
  13                         return false;
  14         }
  15         return true;
  16 }
  17
  18 static bool blobmsg_add_array(struct blob_buf *b, struct array_list *a)
  19 {
  20         int i, len;
  21
  22         for (i = 0, len = array_list_length(a); i < len; i++) {
  23                 if (!blobmsg_add_element(b, NULL, array_list_get_idx(a, i)))
  24                         return false;
  25         }
  26
  27         return true;
  28 }
  29
  30 static bool blobmsg_add_element(struct blob_buf *b, const char *name, json_object *obj)
  31 {
  32         bool ret = true;
  33         void *c;
  34
  35         if (!obj)
  36                 return false;
  37
  38         switch (json_object_get_type(obj)) {
  39         case json_type_object:
  40                 c = blobmsg_open_table(b, name);
  41                 ret = blobmsg_add_object(b, obj);
  42                 blobmsg_close_table(b, c);
  43                 break;
  44         case json_type_array:
  45                 c = blobmsg_open_array(b, name);
  46                 ret = blobmsg_add_array(b, json_object_get_array(obj));
  47                 blobmsg_close_array(b, c);
  48                 break;
  49         case json_type_string:
  50                 blobmsg_add_string(b, name, json_object_get_string(obj));
  51                 break;
  52         case json_type_boolean:
  53                 blobmsg_add_u8(b, name, json_object_get_boolean(obj));
  54                 break;
  55         case json_type_int:
  56                 blobmsg_add_u32(b, name, json_object_get_int(obj));
  57                 break;
  58         default:
  59                 return false;
  60         }
  61         return ret;
  62 }
  63
  64 static bool blobmsg_from_json(struct blob_buf *b, const char *str)
  65 {
  66         json_object *obj;
  67         bool ret = false;
  68
  69         obj = json_tokener_parse(str);
  70         if (is_error(obj))
  71                 return false;
  72
  73         if (json_object_get_type(obj) != json_type_object)
  74                 goto out;
  75
  76         ret = blobmsg_add_object(b, obj);
  77
  78 out:
  79         json_object_put(obj);
  80         return ret;
  81 }
  82
  83 static void receive_lookup(struct ubus_context *ctx, struct ubus_object_data *obj, void *priv)
  84 {
  85         struct blob_attr *cur;
  86         char *s;
  87         int rem;
  88
  89         fprintf(stderr, "'%s' @%08x\n", obj->path, obj->id);
  90
  91         if (!obj->signature)
  92                 return;
  93
  94         blob_for_each_attr(cur, obj->signature, rem) {
  95                 s = blobmsg_format_json(cur, false);
  96                 fprintf(stderr, "\t%s\n", s);
  97                 free(s);
  98         }
  99 }
 100
 101 static void receive_data(struct ubus_request *req, int type, struct blob_attr *msg)
 102 {
 103         char *str;
 104         if (!msg)
 105                 return;
 106
 107         str = blobmsg_format_json(msg, true);
 108         fprintf(stderr, "%s\n", str);
 109         free(str);
 110 }
 111
 112
 113 static int usage(char *prog)
 114 {
 115         fprintf(stderr,
 116                 "Usage: %s <command> [arguments...]\n"
 117                 "Commands:\n"
 118                 " - list [<path>]                       List objects\n"
 119                 " - call <path> <method> [<message>]    Call an object method\n"
 120                 " - listen [<path>...]                  Listen for events\n"
 121                 "\n", prog);
 122         return 1;
 123 }
 124
 125 static void receive_event(struct ubus_context *ctx, struct ubus_event_handler *ev,
 126                           const char *type, struct blob_attr *msg)
 127 {
 128         char *str;
 129
 130         if (msg)
 131                 str = blobmsg_format_json(msg, true);
 132         else
 133                 str = "";
 134
 135         fprintf(stderr, "\"%s\":{ %s }\n", type, str);
 136         free(str);
 137 }
 138
 139 static int ubus_cli_listen(struct ubus_context *ctx, int argc, char **argv)
 140 {
 141         static struct ubus_event_handler listener;
 142         const char *event;
 143         int ret = 0;
 144
 145         memset(&listener, 0, sizeof(listener));
 146         listener.cb = receive_event;
 147
 148         if (!argc) {
 149                 event = "*";
 150                 ret = ubus_register_event_handler(ctx, &listener, NULL);
 151         }
 152
 153         for (;argc;argv++, argc--) {
 154                 event = argv[0];
 155                 ret = ubus_register_event_handler(ctx, &listener, argv[0]);
 156                 if (ret)
 157                         break;
 158         }
 159
 160         if (ret) {
 161                 fprintf(stderr, "Error while registering for event '%s': %s\n",
 162                         event, ubus_strerror(ret));
 163         }
 164
 165         uloop_init();
 166         ubus_add_uloop(ctx);
 167         uloop_run();
 168         uloop_done();
 169
 170         return 0;
 171 }
 172
 173 int main(int argc, char **argv)
 174 {
 175         static struct ubus_context *ctx;
 176         char *cmd;
 177         int ret = 0;
 178
 179         ctx = ubus_connect(NULL);
 180         if (!ctx) {
 181                 fprintf(stderr, "Failed to connect to ubus\n");
 182                 return -1;
 183         }
 184
 185         cmd = argv[1];
 186         if (argc < 2)
 187                 return usage(argv[0]);
 188
 189         if (!strcmp(cmd, "list")) {
 190                 const char *path = NULL;
 191
 192                 if (argc == 3)
 193                         path = argv[2];
 194
 195                 ret = ubus_lookup(ctx, path, receive_lookup, NULL);
 196         } else if (!strcmp(cmd, "call")) {
 197                 uint32_t id;
 198
 199                 if (argc < 4 || argc > 5)
 200                         return usage(argv[0]);
 201
 202                 blob_buf_init(&b, 0);
 203                 if (argc == 5 && !blobmsg_from_json(&b, argv[4])) {
 204                         fprintf(stderr, "Failed to parse message data\n");
 205                         goto out;
 206                 }
 207
 208                 ret = ubus_lookup_id(ctx, argv[2], &id);
 209                 if (!ret)
 210                         ret = ubus_invoke(ctx, id, argv[3], b.head, receive_data, NULL);
 211         } else if (!strcmp(cmd, "listen")) {
 212                 ret = ubus_cli_listen(ctx, argc - 2, argv + 2);
 213         } else {
 214                 return usage(argv[0]);
 215         }
 216
 217         if (ret)
 218                 fprintf(stderr, "Failed: %s\n", ubus_strerror(ret));
 219
 220 out:
 221         ubus_free(ctx);
 222         return ret;
 223 }