cli.c

   1 #include "libubus.h"
   2
   3 static void receive_lookup(struct ubus_context *ctx, struct ubus_object_data *obj, void *priv)
   4 {
   5         struct blob_attr *cur;
   6         char *s;
   7         int rem;
   8
   9         fprintf(stderr, "'%s' @%08x\n", obj->path, obj->id);
  10
  11         if (!obj->signature)
  12                 return;
  13
  14         blob_for_each_attr(cur, obj->signature, rem) {
  15                 s = blobmsg_format_json(cur, false);
  16                 fprintf(stderr, "\t%s\n", s);
  17                 free(s);
  18         }
  19 }
  20
  21 static void receive_data(struct ubus_request *req, int type, struct blob_attr *msg)
  22 {
  23         if (!msg)
  24                 return;
  25
  26         fprintf(stderr, "%s\n", blobmsg_format_json(msg, true));
  27 }
  28
  29
  30 static int usage(char *prog)
  31 {
  32         fprintf(stderr,
  33                 "Usage: %s <command> [arguments...]\n"
  34                 "Commands:\n"
  35                 " - list [<path>]                       List objects\n"
  36                 " - call <path> <method> [<message>]    Call an object method\n"
  37                 " - listen [<path>...]                  Listen for events\n"
  38                 "\n", prog);
  39         return 1;
  40 }
  41
  42 static int ubus_cli_listen(struct ubus_context *ctx, int argc, char **argv)
  43 {
  44         static struct ubus_object listener;
  45         const char *event;
  46         int ret = 0;
  47
  48         if (!argc) {
  49                 event = "*";
  50                 ret = ubus_register_event_handler(ctx, &listener, NULL);
  51         }
  52
  53         for (;argc;argv++, argc--) {
  54                 event = argv[0];
  55                 ret = ubus_register_event_handler(ctx, &listener, argv[0]);
  56                 if (ret)
  57                         break;
  58         }
  59
  60         if (ret) {
  61                 fprintf(stderr, "Error while registering for event '%s': %s\n",
  62                         event, ubus_strerror(ret));
  63         }
  64
  65         uloop_init();
  66         ubus_add_uloop(ctx);
  67         uloop_run();
  68         uloop_done();
  69
  70         return 0;
  71 }
  72
  73 int main(int argc, char **argv)
  74 {
  75         static struct ubus_context *ctx;
  76         char *cmd;
  77         int ret;
  78
  79         ctx = ubus_connect(NULL);
  80         if (!ctx) {
  81                 fprintf(stderr, "Failed to connect to ubus\n");
  82                 return -1;
  83         }
  84
  85         cmd = argv[1];
  86         if (argc < 2)
  87                 return usage(argv[0]);
  88
  89         if (!strcmp(cmd, "list")) {
  90                 const char *path = NULL;
  91
  92                 if (argc == 3)
  93                         path = argv[2];
  94
  95                 ret = ubus_lookup(ctx, path, receive_lookup, NULL);
  96         } else if (!strcmp(cmd, "call")) {
  97                 uint32_t id;
  98
  99                 if (argc < 4 || argc > 5)
 100                         return usage(argv[0]);
 101
 102                 ret = ubus_lookup_id(ctx, argv[2], &id);
 103                 if (!ret)
 104                         ret = ubus_invoke(ctx, id, argv[3], NULL, receive_data, NULL);
 105         } else if (!strcmp(cmd, "listen")) {
 106                 ret = ubus_cli_listen(ctx, argc - 2, argv + 2);
 107         } else {
 108                 return usage(argv[0]);
 109         }
 110
 111         if (ret)
 112                 fprintf(stderr, "Failed: %s\n", ubus_strerror(ret));
 113
 114         ubus_free(ctx);
 115         return ret;
 116 }