cli.c

   1 #include <unistd.h>
   2
   3 #include <libubox/blobmsg_json.h>
   4 #include "libubus.h"
   5
   6 static struct blob_buf b;
   7 static int timeout = 30;
   8
   9 static const char *format_type(void *priv, struct blob_attr *attr)
  10 {
  11         static const char * const attr_types[] = {
  12                 [BLOBMSG_TYPE_INT32] = "\"Integer\"",
  13                 [BLOBMSG_TYPE_STRING] = "\"String\"",
  14         };
  15         const char *type = NULL;
  16         int typeid;
  17
  18         if (blob_id(attr) != BLOBMSG_TYPE_INT32)
  19                 return NULL;
  20
  21         typeid = blobmsg_get_u32(attr);
  22         if (typeid < ARRAY_SIZE(attr_types))
  23                 type = attr_types[typeid];
  24         if (!type)
  25                 type = "\"(unknown)\"";
  26
  27         return type;
  28 }
  29
  30 static void receive_list_result(struct ubus_context *ctx, struct ubus_object_data *obj, void *priv)
  31 {
  32         struct blob_attr *cur;
  33         char *s;
  34         int rem;
  35
  36         printf("'%s' @%08x\n", obj->path, obj->id);
  37
  38         if (!obj->signature)
  39                 return;
  40
  41         blob_for_each_attr(cur, obj->signature, rem) {
  42                 s = blobmsg_format_json_with_cb(cur, false, format_type, NULL);
  43                 printf("\t%s\n", s);
  44                 free(s);
  45         }
  46 }
  47
  48 static void receive_call_result_data(struct ubus_request *req, int type, struct blob_attr *msg)
  49 {
  50         char *str;
  51         if (!msg)
  52                 return;
  53
  54         str = blobmsg_format_json(msg, true);
  55         printf("%s\n", str);
  56         free(str);
  57 }
  58
  59 static void receive_event(struct ubus_context *ctx, struct ubus_event_handler *ev,
  60                           const char *type, struct blob_attr *msg)
  61 {
  62         char *str;
  63
  64         str = blobmsg_format_json(msg, true);
  65         printf("\"%s\": %s\n", type, str);
  66         free(str);
  67 }
  68
  69 static int ubus_cli_list(struct ubus_context *ctx, int argc, char **argv)
  70 {
  71         const char *path = NULL;
  72
  73         if (argc > 1)
  74                 return -2;
  75
  76         if (argc == 1)
  77                 path = argv[0];
  78
  79         return ubus_lookup(ctx, path, receive_list_result, NULL);
  80 }
  81
  82 static int ubus_cli_call(struct ubus_context *ctx, int argc, char **argv)
  83 {
  84         uint32_t id;
  85         int ret;
  86
  87         if (argc < 2 || argc > 3)
  88                 return -2;
  89
  90         blob_buf_init(&b, 0);
  91         if (argc == 3 && !blobmsg_add_json_from_string(&b, argv[2])) {
  92                 fprintf(stderr, "Failed to parse message data\n");
  93                 return -1;
  94         }
  95
  96         ret = ubus_lookup_id(ctx, argv[0], &id);
  97         if (ret)
  98                 return ret;
  99
 100         return ubus_invoke(ctx, id, argv[1], b.head, receive_call_result_data, NULL, timeout * 1000);
 101 }
 102
 103 static int ubus_cli_listen(struct ubus_context *ctx, int argc, char **argv)
 104 {
 105         static struct ubus_event_handler listener;
 106         const char *event;
 107         int ret = 0;
 108
 109         memset(&listener, 0, sizeof(listener));
 110         listener.cb = receive_event;
 111
 112         if (argc > 0) {
 113                 event = argv[0];
 114         } else {
 115                 event = "*";
 116                 argc = 1;
 117         }
 118
 119         do {
 120                 ret = ubus_register_event_handler(ctx, &listener, event);
 121                 if (ret)
 122                         break;
 123
 124                 argv++;
 125                 argc--;
 126                 if (argc <= 0)
 127                         break;
 128
 129                 event = argv[0];
 130         } while (1);
 131
 132         if (ret) {
 133                 fprintf(stderr, "Error while registering for event '%s': %s\n",
 134                         event, ubus_strerror(ret));
 135                 return -1;
 136         }
 137
 138         uloop_init();
 139         ubus_add_uloop(ctx);
 140         uloop_run();
 141         uloop_done();
 142
 143         return 0;
 144 }
 145
 146 static int ubus_cli_send(struct ubus_context *ctx, int argc, char **argv)
 147 {
 148         if (argc < 1 || argc > 2)
 149                 return -2;
 150
 151         blob_buf_init(&b, 0);
 152
 153         if (argc == 2 && !blobmsg_add_json_from_string(&b, argv[1])) {
 154                 fprintf(stderr, "Failed to parse message data\n");
 155                 return -1;
 156         }
 157
 158         return ubus_send_event(ctx, argv[0], b.head);
 159 }
 160
 161 static int usage(const char *prog)
 162 {
 163         fprintf(stderr,
 164                 "Usage: %s [<options>] <command> [arguments...]\n"
 165                 "Options:\n"
 166                 " -s <socket>:          Set the unix domain socket to connect to\n"
 167                 "\n"
 168                 "Commands:\n"
 169                 " - list [<path>]                       List objects\n"
 170                 " - call <path> <method> [<message>]    Call an object method\n"
 171                 " - listen [<path>...]                  Listen for events\n"
 172                 " - send <type> [<message>]             Send an event\n"
 173                 "\n", prog);
 174         return 1;
 175 }
 176
 177
 178 struct {
 179         const char *name;
 180         int (*cb)(struct ubus_context *ctx, int argc, char **argv);
 181 } commands[] = {
 182         { "list", ubus_cli_list },
 183         { "call", ubus_cli_call },
 184         { "listen", ubus_cli_listen },
 185         { "send", ubus_cli_send },
 186 };
 187
 188 int main(int argc, char **argv)
 189 {
 190         const char *progname, *ubus_socket = NULL;
 191         static struct ubus_context *ctx;
 192         char *cmd;
 193         int ret = 0;
 194         int i, ch;
 195
 196         progname = argv[0];
 197
 198         while ((ch = getopt(argc, argv, "s:t:")) != -1) {
 199                 switch (ch) {
 200                 case 's':
 201                         ubus_socket = optarg;
 202                         break;
 203                 case 't':
 204                         timeout = atoi(optarg);
 205                         break;
 206                 default:
 207                         return usage(progname);
 208                 }
 209         }
 210
 211         argc -= optind;
 212         argv += optind;
 213
 214         cmd = argv[0];
 215         if (argc < 1)
 216                 return usage(progname);
 217
 218         ctx = ubus_connect(ubus_socket);
 219         if (!ctx) {
 220                 fprintf(stderr, "Failed to connect to ubus\n");
 221                 return -1;
 222         }
 223
 224         argv++;
 225         argc--;
 226
 227         ret = -2;
 228         for (i = 0; i < ARRAY_SIZE(commands); i++) {
 229                 if (strcmp(commands[i].name, cmd) != 0)
 230                         continue;
 231
 232                 ret = commands[i].cb(ctx, argc, argv);
 233                 break;
 234         }
 235
 236         if (ret > 0)
 237                 fprintf(stderr, "Command failed: %s\n", ubus_strerror(ret));
 238         else if (ret == -2)
 239                 usage(progname);
 240
 241         ubus_free(ctx);
 242         return ret;
 243 }