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
   8 static const char * const attr_types[] = {
   9         [BLOBMSG_TYPE_INT32] = "\"Integer\"",
  10         [BLOBMSG_TYPE_STRING] = "\"String\"",
  11 };
  12
  13 static const char *format_type(void *priv, struct blob_attr *attr)
  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_lookup(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_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
  60 static void receive_event(struct ubus_context *ctx, struct ubus_event_handler *ev,
  61                           const char *type, struct blob_attr *msg)
  62 {
  63         char *str;
  64
  65         str = blobmsg_format_json(msg, true);
  66         printf("\"%s\": %s\n", type, str);
  67         free(str);
  68 }
  69
  70 static int ubus_cli_listen(struct ubus_context *ctx, int argc, char **argv)
  71 {
  72         static struct ubus_event_handler listener;
  73         const char *event;
  74         int ret = 0;
  75
  76         memset(&listener, 0, sizeof(listener));
  77         listener.cb = receive_event;
  78
  79         if (!argc) {
  80                 event = "*";
  81                 ret = ubus_register_event_handler(ctx, &listener, NULL);
  82         }
  83
  84         for (;argc;argv++, argc--) {
  85                 event = argv[0];
  86                 ret = ubus_register_event_handler(ctx, &listener, argv[0]);
  87                 if (ret)
  88                         break;
  89         }
  90
  91         if (ret) {
  92                 fprintf(stderr, "Error while registering for event '%s': %s\n",
  93                         event, ubus_strerror(ret));
  94         }
  95
  96         uloop_init();
  97         ubus_add_uloop(ctx);
  98         uloop_run();
  99         uloop_done();
 100
 101         return 0;
 102 }
 103
 104 static int ubus_cli_send(struct ubus_context *ctx, int argc, char **argv)
 105 {
 106         blob_buf_init(&b, 0);
 107         if (argc == 2 && !blobmsg_add_json_from_string(&b, argv[1])) {
 108                 fprintf(stderr, "Failed to parse message data\n");
 109                 return UBUS_STATUS_INVALID_ARGUMENT;
 110         }
 111
 112         return ubus_send_event(ctx, argv[0], b.head);
 113 }
 114
 115 static int usage(const char *prog)
 116 {
 117         fprintf(stderr,
 118                 "Usage: %s [<options>] <command> [arguments...]\n"
 119                 "Options:\n"
 120                 " -s <socket>:          Set the unix domain socket to connect to\n"
 121                 "\n"
 122                 "Commands:\n"
 123                 " - list [<path>]                       List objects\n"
 124                 " - call <path> <method> [<message>]    Call an object method\n"
 125                 " - listen [<path>...]                  Listen for events\n"
 126                 " - send <type> [<message>]             Send an event\n"
 127                 "\n", prog);
 128         return 1;
 129 }
 130
 131 int main(int argc, char **argv)
 132 {
 133         const char *progname, *ubus_socket = NULL;
 134         static struct ubus_context *ctx;
 135         char *cmd;
 136         int ret = 0;
 137         int ch;
 138
 139         progname = argv[0];
 140
 141         while ((ch = getopt(argc, argv, "s:")) != -1) {
 142                 switch (ch) {
 143                 case 's':
 144                         ubus_socket = optarg;
 145                         break;
 146                 default:
 147                         return usage(progname);
 148                 }
 149         }
 150
 151         argc -= optind;
 152         argv += optind;
 153
 154         ctx = ubus_connect(ubus_socket);
 155         if (!ctx) {
 156                 fprintf(stderr, "Failed to connect to ubus\n");
 157                 return -1;
 158         }
 159
 160         cmd = argv[0];
 161         if (argc < 1)
 162                 return usage(progname);
 163
 164         argv++;
 165         argc--;
 166
 167         if (!strcmp(cmd, "list")) {
 168                 const char *path = NULL;
 169
 170                 if (argc == 1)
 171                         path = argv[0];
 172
 173                 ret = ubus_lookup(ctx, path, receive_lookup, NULL);
 174         } else if (!strcmp(cmd, "call")) {
 175                 uint32_t id;
 176
 177                 if (argc < 2 || argc > 3)
 178                         return usage(progname);
 179
 180                 blob_buf_init(&b, 0);
 181                 if (argc == 3 && !blobmsg_add_json_from_string(&b, argv[2])) {
 182                         fprintf(stderr, "Failed to parse message data\n");
 183                         goto out;
 184                 }
 185
 186                 ret = ubus_lookup_id(ctx, argv[0], &id);
 187                 if (!ret)
 188                         ret = ubus_invoke(ctx, id, argv[1], b.head, receive_data, NULL);
 189         } else if (!strcmp(cmd, "listen")) {
 190                 ret = ubus_cli_listen(ctx, argc, argv);
 191         } else if (!strcmp(cmd, "send")) {
 192                 if (argc < 1 || argc > 2)
 193                         return usage(progname);
 194                 ret = ubus_cli_send(ctx, argc, argv);
 195         } else {
 196                 return usage(progname);
 197         }
 198
 199         if (ret)
 200                 fprintf(stderr, "Failed: %s\n", ubus_strerror(ret));
 201
 202 out:
 203         ubus_free(ctx);
 204         return ret;
 205 }