libubus.c

   1 /*
   2  * Copyright (C) 2011 Felix Fietkau <nbd@openwrt.org>
   3  *
   4  * This program is free software; you can redistribute it and/or modify
   5  * it under the terms of the GNU Lesser General Public License version 2.1
   6  * as published by the Free Software Foundation
   7  *
   8  * This program is distributed in the hope that it will be useful,
   9  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  10  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  11  * GNU General Public License for more details.
  12  */
  13
  14 #include <sys/types.h>
  15 #include <sys/uio.h>
  16 #include <sys/socket.h>
  17 #include <fcntl.h>
  18 #include <unistd.h>
  19 #include <poll.h>
  20
  21 #include <libubox/blob.h>
  22 #include <libubox/blobmsg.h>
  23 #include <libubox/usock.h>
  24
  25 #include "libubus.h"
  26 #include "ubusmsg.h"
  27
  28 #define STATIC_IOV(_var) { .iov_base = (char *) &(_var), .iov_len = sizeof(_var) }
  29
  30 const char *__ubus_strerror[__UBUS_STATUS_LAST] = {
  31         [UBUS_STATUS_OK] = "Success",
  32         [UBUS_STATUS_INVALID_COMMAND] = "Invalid command",
  33         [UBUS_STATUS_INVALID_ARGUMENT] = "Invalid argument",
  34         [UBUS_STATUS_METHOD_NOT_FOUND] = "Method not found",
  35         [UBUS_STATUS_NOT_FOUND] = "Not found",
  36         [UBUS_STATUS_NO_DATA] = "No response",
  37         [UBUS_STATUS_PERMISSION_DENIED] = "Permission denied",
  38         [UBUS_STATUS_TIMEOUT] = "Request timed out",
  39         [UBUS_STATUS_NOT_SUPPORTED] = "Operation not supported",
  40         [UBUS_STATUS_UNKNOWN_ERROR] = "Unknown error",
  41 };
  42
  43 static struct blob_buf b;
  44
  45 static const struct blob_attr_info ubus_policy[UBUS_ATTR_MAX] = {
  46         [UBUS_ATTR_STATUS] = { .type = BLOB_ATTR_INT32 },
  47         [UBUS_ATTR_OBJID] = { .type = BLOB_ATTR_INT32 },
  48         [UBUS_ATTR_OBJPATH] = { .type = BLOB_ATTR_STRING },
  49         [UBUS_ATTR_METHOD] = { .type = BLOB_ATTR_STRING },
  50 };
  51 static struct blob_attr *attrbuf[UBUS_ATTR_MAX];
  52
  53 struct ubus_pending_data {
  54         struct list_head list;
  55         int type;
  56         struct blob_attr data[];
  57 };
  58
  59 static int ubus_cmp_id(const void *k1, const void *k2, void *ptr)
  60 {
  61         const uint32_t *id1 = k1, *id2 = k2;
  62
  63         if (*id1 < *id2)
  64                 return -1;
  65         else
  66                 return *id1 > *id2;
  67 }
  68
  69 static struct blob_attr **ubus_parse_msg(struct blob_attr *msg)
  70 {
  71         blob_parse(msg, attrbuf, ubus_policy, UBUS_ATTR_MAX);
  72         return attrbuf;
  73 }
  74
  75 const char *ubus_strerror(int error)
  76 {
  77         static char err[32];
  78
  79         if (error < 0 || error >= __UBUS_STATUS_LAST)
  80                 goto out;
  81
  82         if (!__ubus_strerror[error])
  83                 goto out;
  84
  85         return __ubus_strerror[error];
  86
  87 out:
  88         sprintf(err, "Unknown error: %d", error);
  89         return err;
  90 }
  91
  92 static void wait_data(int fd, bool write)
  93 {
  94         struct pollfd pfd = { .fd = fd };
  95
  96         pfd.events = write ? POLLOUT : POLLIN;
  97         poll(&pfd, 1, 0);
  98 }
  99
 100 static int writev_retry(int fd, struct iovec *iov, int iov_len)
 101 {
 102         int len = 0;
 103
 104         do {
 105                 int cur_len = writev(fd, iov, iov_len);
 106                 if (cur_len < 0) {
 107                         switch(errno) {
 108                         case EAGAIN:
 109                                 wait_data(fd, true);
 110                                 break;
 111                         case EINTR:
 112                                 break;
 113                         default:
 114                                 return -1;
 115                         }
 116                         continue;
 117                 }
 118                 len += cur_len;
 119                 while (cur_len >= iov->iov_len) {
 120                         cur_len -= iov->iov_len;
 121                         iov_len--;
 122                         iov++;
 123                         if (!cur_len || !iov_len)
 124                                 return len;
 125                 }
 126                 iov->iov_len -= cur_len;
 127         } while (1);
 128 }
 129
 130 static int ubus_send_msg(struct ubus_context *ctx, uint32_t seq,
 131                          struct blob_attr *msg, int cmd, uint32_t peer)
 132 {
 133         struct ubus_msghdr hdr;
 134         struct iovec iov[2] = {
 135                 STATIC_IOV(hdr)
 136         };
 137
 138         hdr.version = 0;
 139         hdr.type = cmd;
 140         hdr.seq = seq;
 141         hdr.peer = peer;
 142
 143         if (!msg) {
 144                 blob_buf_init(&b, 0);
 145                 msg = b.head;
 146         }
 147
 148         iov[1].iov_base = (char *) msg;
 149         iov[1].iov_len = blob_raw_len(msg);
 150
 151         return writev_retry(ctx->sock.fd, iov, ARRAY_SIZE(iov));
 152 }
 153
 154 static int ubus_start_request(struct ubus_context *ctx, struct ubus_request *req,
 155                               struct blob_attr *msg, int cmd, uint32_t peer)
 156 {
 157         memset(req, 0, sizeof(*req));
 158
 159         if (msg && blob_pad_len(msg) > UBUS_MAX_MSGLEN)
 160                 return -1;
 161
 162         INIT_LIST_HEAD(&req->list);
 163         INIT_LIST_HEAD(&req->pending);
 164         req->ctx = ctx;
 165         req->peer = peer;
 166         req->seq = ++ctx->request_seq;
 167         return ubus_send_msg(ctx, req->seq, msg, cmd, peer);
 168 }
 169
 170 static bool recv_retry(int fd, struct iovec *iov, bool wait)
 171 {
 172         int bytes;
 173
 174         while (iov->iov_len > 0) {
 175                 if (wait)
 176                         wait_data(fd, false);
 177
 178                 bytes = read(fd, iov->iov_base, iov->iov_len);
 179                 if (bytes < 0) {
 180                         bytes = 0;
 181                         if (uloop_cancelled)
 182                                 return false;
 183                         if (errno == EINTR)
 184                                 continue;
 185
 186                         if (errno != EAGAIN)
 187                                 return false;
 188                 }
 189                 if (!wait && !bytes)
 190                         return false;
 191
 192                 wait = true;
 193                 iov->iov_len -= bytes;
 194                 iov->iov_base += bytes;
 195         }
 196
 197         return true;
 198 }
 199
 200 static bool ubus_validate_hdr(struct ubus_msghdr *hdr)
 201 {
 202         if (hdr->version != 0)
 203                 return false;
 204
 205         if (blob_raw_len(hdr->data) < sizeof(*hdr->data))
 206                 return false;
 207
 208         if (blob_pad_len(hdr->data) > UBUS_MAX_MSGLEN)
 209                 return false;
 210
 211         return true;
 212 }
 213
 214 static bool get_next_msg(struct ubus_context *ctx)
 215 {
 216         struct iovec iov = STATIC_IOV(ctx->msgbuf.hdr);
 217
 218         /* receive header + start attribute */
 219         iov.iov_len += sizeof(struct blob_attr);
 220         if (!recv_retry(ctx->sock.fd, &iov, false))
 221                 return false;
 222
 223         iov.iov_len = blob_len(ctx->msgbuf.hdr.data);
 224         if (iov.iov_len > 0 && !recv_retry(ctx->sock.fd, &iov, true))
 225                 return false;
 226
 227         return ubus_validate_hdr(&ctx->msgbuf.hdr);
 228 }
 229
 230 static bool ubus_get_status(struct ubus_msghdr *hdr, int *ret)
 231 {
 232         ubus_parse_msg(hdr->data);
 233
 234         if (!attrbuf[UBUS_ATTR_STATUS])
 235                 return false;
 236
 237         *ret = blob_get_u32(attrbuf[UBUS_ATTR_STATUS]);
 238         return true;
 239 }
 240
 241 static void req_data_cb(struct ubus_request *req, int type, struct blob_attr *data)
 242 {
 243         struct blob_attr **attr;
 244
 245         if (req->raw_data_cb)
 246                 req->raw_data_cb(req, type, data);
 247
 248         if (!req->data_cb)
 249                 return;
 250
 251         attr = ubus_parse_msg(data);
 252         req->data_cb(req, type, attr[UBUS_ATTR_DATA]);
 253 }
 254
 255 static void ubus_process_req_data(struct ubus_request *req)
 256 {
 257         struct ubus_pending_data *data;
 258
 259         while (!list_empty(&req->pending)) {
 260                 data = list_first_entry(&req->pending,
 261                         struct ubus_pending_data, list);
 262                 list_del(&data->list);
 263                 if (!req->cancelled)
 264                         req_data_cb(req, data->type, data->data);
 265                 free(data);
 266         }
 267 }
 268
 269 static void ubus_req_complete_cb(struct ubus_request *req)
 270 {
 271         ubus_complete_handler_t cb = req->complete_cb;
 272
 273         if (!cb)
 274                 return;
 275
 276         req->complete_cb = NULL;
 277         cb(req, req->status_code);
 278 }
 279
 280 static int ubus_process_req_status(struct ubus_request *req, struct ubus_msghdr *hdr)
 281 {
 282         int ret = UBUS_STATUS_INVALID_ARGUMENT;
 283
 284         if (!list_empty(&req->list))
 285                 list_del_init(&req->list);
 286
 287         ubus_get_status(hdr, &ret);
 288         req->peer = hdr->peer;
 289         req->status_msg = true;
 290         req->status_code = ret;
 291         if (!req->blocked)
 292                 ubus_req_complete_cb(req);
 293
 294         return ret;
 295 }
 296
 297 static void ubus_req_data(struct ubus_request *req, struct ubus_msghdr *hdr)
 298 {
 299         struct ubus_pending_data *data;
 300         int len;
 301
 302         if (!req->blocked) {
 303                 req->blocked = true;
 304                 req_data_cb(req, hdr->type, hdr->data);
 305                 ubus_process_req_data(req);
 306                 req->blocked = false;
 307
 308                 if (req->status_msg)
 309                         ubus_req_complete_cb(req);
 310
 311                 return;
 312         }
 313
 314         len = blob_raw_len(hdr->data);
 315         data = calloc(1, sizeof(*data) + len);
 316         if (!data)
 317                 return;
 318
 319         data->type = hdr->type;
 320         memcpy(data->data, hdr->data, len);
 321         list_add(&data->list, &req->pending);
 322 }
 323
 324 static struct ubus_request *ubus_find_request(struct ubus_context *ctx, uint32_t seq, uint32_t peer)
 325 {
 326         struct ubus_request *req;
 327
 328         list_for_each_entry(req, &ctx->requests, list) {
 329                 if (seq != req->seq || peer != req->peer)
 330                         continue;
 331
 332                 return req;
 333         }
 334         return NULL;
 335 }
 336
 337 static void ubus_process_invoke(struct ubus_context *ctx, struct ubus_msghdr *hdr)
 338 {
 339         struct ubus_request_data req;
 340         struct ubus_object *obj;
 341         uint32_t objid = 0;
 342         int method;
 343         int ret = 0;
 344
 345         req.peer = hdr->peer;
 346         req.seq = hdr->seq;
 347         ubus_parse_msg(hdr->data);
 348
 349         if (!attrbuf[UBUS_ATTR_OBJID])
 350                 return;
 351
 352         objid = blob_get_u32(attrbuf[UBUS_ATTR_OBJID]);
 353
 354         if (!attrbuf[UBUS_ATTR_METHOD]) {
 355                 ret = UBUS_STATUS_INVALID_ARGUMENT;
 356                 goto send;
 357         }
 358
 359         obj = avl_find_element(&ctx->objects, &objid, obj, avl);
 360         if (!obj) {
 361                 ret = UBUS_STATUS_NOT_FOUND;
 362                 goto send;
 363         }
 364
 365         for (method = 0; method < obj->n_methods; method++)
 366                 if (!obj->methods[method].name ||
 367                     !strcmp(obj->methods[method].name,
 368                             blob_data(attrbuf[UBUS_ATTR_METHOD])))
 369                         goto found;
 370
 371         /* not found */
 372         ret = UBUS_STATUS_METHOD_NOT_FOUND;
 373         goto send;
 374
 375 found:
 376         req.object = objid;
 377         ret = obj->methods[method].handler(ctx, obj, &req,
 378                                            blob_data(attrbuf[UBUS_ATTR_METHOD]),
 379                                            attrbuf[UBUS_ATTR_DATA]);
 380
 381 send:
 382         blob_buf_init(&b, 0);
 383         blob_put_int32(&b, UBUS_ATTR_STATUS, ret);
 384         blob_put_int32(&b, UBUS_ATTR_OBJID, objid);
 385         ubus_send_msg(ctx, req.seq, b.head, UBUS_MSG_STATUS, req.peer);
 386 }
 387
 388 static void ubus_process_msg(struct ubus_context *ctx, struct ubus_msghdr *hdr)
 389 {
 390         struct ubus_request *req;
 391
 392         switch(hdr->type) {
 393         case UBUS_MSG_STATUS:
 394                 req = ubus_find_request(ctx, hdr->seq, hdr->peer);
 395                 if (!req)
 396                         break;
 397
 398                 ubus_process_req_status(req, hdr);
 399                 break;
 400
 401         case UBUS_MSG_DATA:
 402                 req = ubus_find_request(ctx, hdr->seq, hdr->peer);
 403                 if (req && (req->data_cb || req->raw_data_cb))
 404                         ubus_req_data(req, hdr);
 405                 break;
 406
 407         case UBUS_MSG_INVOKE:
 408                 ubus_process_invoke(ctx, hdr);
 409                 break;
 410         }
 411 }
 412
 413 void ubus_abort_request(struct ubus_context *ctx, struct ubus_request *req)
 414 {
 415         if (!list_empty(&req->list))
 416                 return;
 417
 418         req->cancelled = true;
 419         ubus_process_req_data(req);
 420         list_del_init(&req->list);
 421 }
 422
 423 void ubus_complete_request_async(struct ubus_context *ctx, struct ubus_request *req)
 424 {
 425         if (!list_empty(&req->list))
 426                 return;
 427
 428         list_add(&req->list, &ctx->requests);
 429 }
 430
 431 static void ubus_handle_data(struct uloop_fd *u, unsigned int events)
 432 {
 433         struct ubus_context *ctx = container_of(u, struct ubus_context, sock);
 434         struct ubus_msghdr *hdr = &ctx->msgbuf.hdr;
 435
 436         while (get_next_msg(ctx)) {
 437                 ubus_process_msg(ctx, hdr);
 438                 if (uloop_cancelled)
 439                         break;
 440         }
 441
 442         if (u->eof)
 443                 ctx->connection_lost(ctx);
 444 }
 445
 446 static void ubus_sync_req_cb(struct ubus_request *req, int ret)
 447 {
 448         req->status_msg = true;
 449         req->status_code = ret;
 450         uloop_end();
 451 }
 452
 453 struct ubus_sync_req_cb {
 454         struct uloop_timeout timeout;
 455         struct ubus_request *req;
 456 };
 457
 458 static void ubus_sync_req_timeout_cb(struct uloop_timeout *timeout)
 459 {
 460         struct ubus_sync_req_cb *cb;
 461
 462         cb = container_of(timeout, struct ubus_sync_req_cb, timeout);
 463         ubus_sync_req_cb(cb->req, UBUS_STATUS_TIMEOUT);
 464 }
 465
 466 int ubus_complete_request(struct ubus_context *ctx, struct ubus_request *req,
 467                           int timeout)
 468 {
 469         struct ubus_sync_req_cb cb;
 470         ubus_complete_handler_t complete_cb = req->complete_cb;
 471         bool registered = ctx->sock.registered;
 472         int status = UBUS_STATUS_NO_DATA;
 473
 474         if (!registered) {
 475                 uloop_init();
 476                 ubus_add_uloop(ctx);
 477         }
 478
 479         if (timeout) {
 480                 memset(&cb, 0, sizeof(cb));
 481                 cb.req = req;
 482                 cb.timeout.cb = ubus_sync_req_timeout_cb;
 483                 uloop_timeout_set(&cb.timeout, timeout);
 484         }
 485
 486         ubus_complete_request_async(ctx, req);
 487         req->complete_cb = ubus_sync_req_cb;
 488
 489         while (!req->status_msg) {
 490                 bool cancelled = uloop_cancelled;
 491                 uloop_cancelled = false;
 492                 uloop_run();
 493                 uloop_cancelled = cancelled;
 494         }
 495
 496         if (timeout)
 497                 uloop_timeout_cancel(&cb.timeout);
 498
 499         if (req->status_msg)
 500                 status = req->status_code;
 501
 502         req->complete_cb = complete_cb;
 503         if (req->complete_cb)
 504                 req->complete_cb(req, status);
 505
 506         if (!registered)
 507                 uloop_fd_delete(&ctx->sock);
 508
 509         return status;
 510 }
 511
 512 struct ubus_lookup_request {
 513         struct ubus_request req;
 514         ubus_lookup_handler_t cb;
 515 };
 516
 517 static void ubus_lookup_cb(struct ubus_request *ureq, int type, struct blob_attr *msg)
 518 {
 519         struct ubus_lookup_request *req;
 520         struct ubus_object_data obj;
 521         struct blob_attr **attr;
 522
 523         req = container_of(ureq, struct ubus_lookup_request, req);
 524         attr = ubus_parse_msg(msg);
 525
 526         if (!attr[UBUS_ATTR_OBJID] || !attr[UBUS_ATTR_OBJPATH] ||
 527             !attr[UBUS_ATTR_OBJTYPE])
 528                 return;
 529
 530         memset(&obj, 0, sizeof(obj));
 531         obj.id = blob_get_u32(attr[UBUS_ATTR_OBJID]);
 532         obj.path = blob_data(attr[UBUS_ATTR_OBJPATH]);
 533         obj.type_id = blob_get_u32(attr[UBUS_ATTR_OBJTYPE]);
 534         obj.signature = attr[UBUS_ATTR_SIGNATURE];
 535         req->cb(ureq->ctx, &obj, ureq->priv);
 536 }
 537
 538 int ubus_lookup(struct ubus_context *ctx, const char *path,
 539                 ubus_lookup_handler_t cb, void *priv)
 540 {
 541         struct ubus_lookup_request lookup;
 542
 543         blob_buf_init(&b, 0);
 544         if (path)
 545                 blob_put_string(&b, UBUS_ATTR_OBJPATH, path);
 546
 547         if (ubus_start_request(ctx, &lookup.req, b.head, UBUS_MSG_LOOKUP, 0) < 0)
 548                 return UBUS_STATUS_INVALID_ARGUMENT;
 549
 550         lookup.req.raw_data_cb = ubus_lookup_cb;
 551         lookup.req.priv = priv;
 552         lookup.cb = cb;
 553         return ubus_complete_request(ctx, &lookup.req, 0);
 554 }
 555
 556 static void ubus_lookup_id_cb(struct ubus_request *req, int type, struct blob_attr *msg)
 557 {
 558         struct blob_attr **attr;
 559         uint32_t *id = req->priv;
 560
 561         attr = ubus_parse_msg(msg);
 562
 563         if (!attr[UBUS_ATTR_OBJID])
 564                 return;
 565
 566         *id = blob_get_u32(attr[UBUS_ATTR_OBJID]);
 567 }
 568
 569 int ubus_lookup_id(struct ubus_context *ctx, const char *path, uint32_t *id)
 570 {
 571         struct ubus_request req;
 572
 573         blob_buf_init(&b, 0);
 574         if (path)
 575                 blob_put_string(&b, UBUS_ATTR_OBJPATH, path);
 576
 577         if (ubus_start_request(ctx, &req, b.head, UBUS_MSG_LOOKUP, 0) < 0)
 578                 return UBUS_STATUS_INVALID_ARGUMENT;
 579
 580         req.raw_data_cb = ubus_lookup_id_cb;
 581         req.priv = id;
 582
 583         return ubus_complete_request(ctx, &req, 0);
 584 }
 585
 586 int ubus_send_reply(struct ubus_context *ctx, struct ubus_request_data *req,
 587                     struct blob_attr *msg)
 588 {
 589         int ret;
 590
 591         blob_buf_init(&b, 0);
 592         blob_put_int32(&b, UBUS_ATTR_OBJID, req->object);
 593         blob_put(&b, UBUS_ATTR_DATA, blob_data(msg), blob_len(msg));
 594         ret = ubus_send_msg(ctx, req->seq, b.head, UBUS_MSG_DATA, req->peer);
 595         if (ret < 0)
 596                 return UBUS_STATUS_NO_DATA;
 597
 598         return 0;
 599 }
 600
 601 int ubus_invoke_async(struct ubus_context *ctx, uint32_t obj, const char *method,
 602                        struct blob_attr *msg, struct ubus_request *req)
 603 {
 604         blob_buf_init(&b, 0);
 605         blob_put_int32(&b, UBUS_ATTR_OBJID, obj);
 606         blob_put_string(&b, UBUS_ATTR_METHOD, method);
 607         if (msg)
 608                 blob_put(&b, UBUS_ATTR_DATA, blob_data(msg), blob_len(msg));
 609
 610         if (ubus_start_request(ctx, req, b.head, UBUS_MSG_INVOKE, obj) < 0)
 611                 return UBUS_STATUS_INVALID_ARGUMENT;
 612
 613         return 0;
 614 }
 615
 616 int ubus_invoke(struct ubus_context *ctx, uint32_t obj, const char *method,
 617                 struct blob_attr *msg, ubus_data_handler_t cb, void *priv,
 618                 int timeout)
 619 {
 620         struct ubus_request req;
 621
 622         ubus_invoke_async(ctx, obj, method, msg, &req);
 623         req.data_cb = cb;
 624         req.priv = priv;
 625         return ubus_complete_request(ctx, &req, timeout);
 626 }
 627
 628 static void ubus_add_object_cb(struct ubus_request *req, int type, struct blob_attr *msg)
 629 {
 630         struct ubus_object *obj = req->priv;
 631
 632         ubus_parse_msg(msg);
 633
 634         if (!attrbuf[UBUS_ATTR_OBJID])
 635                 return;
 636
 637         obj->id = blob_get_u32(attrbuf[UBUS_ATTR_OBJID]);
 638
 639         if (attrbuf[UBUS_ATTR_OBJTYPE])
 640                 obj->type->id = blob_get_u32(attrbuf[UBUS_ATTR_OBJTYPE]);
 641
 642         obj->avl.key = &obj->id;
 643         avl_insert(&req->ctx->objects, &obj->avl);
 644 }
 645
 646 static void ubus_push_method_data(const struct ubus_method *m)
 647 {
 648         void *mtbl;
 649         int i;
 650
 651         mtbl = blobmsg_open_table(&b, m->name);
 652
 653         for (i = 0; i < m->n_policy; i++)
 654                 blobmsg_add_u32(&b, m->policy[i].name, m->policy[i].type);
 655
 656         blobmsg_close_table(&b, mtbl);
 657 }
 658
 659 static bool ubus_push_object_type(const struct ubus_object_type *type)
 660 {
 661         void *s;
 662         int i;
 663
 664         s = blob_nest_start(&b, UBUS_ATTR_SIGNATURE);
 665
 666         for (i = 0; i < type->n_methods; i++)
 667                 ubus_push_method_data(&type->methods[i]);
 668
 669         blob_nest_end(&b, s);
 670
 671         return true;
 672 }
 673
 674 static int __ubus_add_object(struct ubus_context *ctx, struct ubus_object *obj)
 675 {
 676         struct ubus_request req;
 677         int ret;
 678
 679         blob_buf_init(&b, 0);
 680
 681         if (obj->name && obj->type) {
 682                 blob_put_string(&b, UBUS_ATTR_OBJPATH, obj->name);
 683
 684                 if (obj->type->id)
 685                         blob_put_int32(&b, UBUS_ATTR_OBJTYPE, obj->type->id);
 686                 else if (!ubus_push_object_type(obj->type))
 687                         return UBUS_STATUS_INVALID_ARGUMENT;
 688         }
 689
 690         if (ubus_start_request(ctx, &req, b.head, UBUS_MSG_ADD_OBJECT, 0) < 0)
 691                 return UBUS_STATUS_INVALID_ARGUMENT;
 692
 693         req.raw_data_cb = ubus_add_object_cb;
 694         req.priv = obj;
 695         ret = ubus_complete_request(ctx, &req, 0);
 696         if (ret)
 697                 return ret;
 698
 699         if (!obj->id)
 700                 return UBUS_STATUS_NO_DATA;
 701
 702         return 0;
 703 }
 704
 705 int ubus_add_object(struct ubus_context *ctx, struct ubus_object *obj)
 706 {
 707         if (!obj->name || !obj->type)
 708                 return UBUS_STATUS_INVALID_ARGUMENT;
 709
 710         return __ubus_add_object(ctx, obj);
 711 }
 712
 713 static void ubus_remove_object_cb(struct ubus_request *req, int type, struct blob_attr *msg)
 714 {
 715         struct ubus_object *obj = req->priv;
 716
 717         ubus_parse_msg(msg);
 718
 719         if (!attrbuf[UBUS_ATTR_OBJID])
 720                 return;
 721
 722         obj->id = 0;
 723
 724         if (attrbuf[UBUS_ATTR_OBJTYPE] && obj->type)
 725                 obj->type->id = 0;
 726
 727         avl_delete(&req->ctx->objects, &obj->avl);
 728 }
 729
 730 int ubus_remove_object(struct ubus_context *ctx, struct ubus_object *obj)
 731 {
 732         struct ubus_request req;
 733         int ret;
 734
 735         blob_buf_init(&b, 0);
 736         blob_put_int32(&b, UBUS_ATTR_OBJID, obj->id);
 737
 738         if (ubus_start_request(ctx, &req, b.head, UBUS_MSG_REMOVE_OBJECT, 0) < 0)
 739                 return UBUS_STATUS_INVALID_ARGUMENT;
 740
 741         req.raw_data_cb = ubus_remove_object_cb;
 742         req.priv = obj;
 743         ret = ubus_complete_request(ctx, &req, 0);
 744         if (ret)
 745                 return ret;
 746
 747         if (obj->id)
 748                 return UBUS_STATUS_NO_DATA;
 749
 750         return 0;
 751 }
 752
 753 static int ubus_event_cb(struct ubus_context *ctx, struct ubus_object *obj,
 754                          struct ubus_request_data *req,
 755                          const char *method, struct blob_attr *msg)
 756 {
 757         struct ubus_event_handler *ev;
 758
 759         ev = container_of(obj, struct ubus_event_handler, obj);
 760         ev->cb(ctx, ev, method, msg);
 761         return 0;
 762 }
 763
 764 static const struct ubus_method event_method = {
 765         .name = NULL,
 766         .handler = ubus_event_cb,
 767 };
 768
 769 int ubus_register_event_handler(struct ubus_context *ctx,
 770                                 struct ubus_event_handler *ev,
 771                                 const char *pattern)
 772 {
 773         struct ubus_object *obj = &ev->obj;
 774         struct blob_buf b2;
 775         int ret;
 776
 777         if (!obj->id) {
 778                 obj->methods = &event_method;
 779                 obj->n_methods = 1;
 780
 781                 if (!!obj->name ^ !!obj->type)
 782                         return UBUS_STATUS_INVALID_ARGUMENT;
 783
 784                 ret = __ubus_add_object(ctx, obj);
 785                 if (ret)
 786                         return ret;
 787         }
 788
 789         /* use a second buffer, ubus_invoke() overwrites the primary one */
 790         memset(&b2, 0, sizeof(b2));
 791         blob_buf_init(&b2, 0);
 792         blobmsg_add_u32(&b2, "object", obj->id);
 793         if (pattern)
 794                 blobmsg_add_string(&b2, "pattern", pattern);
 795
 796         return ubus_invoke(ctx, UBUS_SYSTEM_OBJECT_EVENT, "register", b2.head,
 797                           NULL, NULL, 0);
 798 }
 799
 800 int ubus_send_event(struct ubus_context *ctx, const char *id,
 801                     struct blob_attr *data)
 802 {
 803         struct ubus_request req;
 804         void *s;
 805
 806         blob_buf_init(&b, 0);
 807         blob_put_int32(&b, UBUS_ATTR_OBJID, UBUS_SYSTEM_OBJECT_EVENT);
 808         blob_put_string(&b, UBUS_ATTR_METHOD, "send");
 809         s = blob_nest_start(&b, UBUS_ATTR_DATA);
 810         blobmsg_add_string(&b, "id", id);
 811         blobmsg_add_field(&b, BLOBMSG_TYPE_TABLE, "data", blob_data(data), blob_len(data));
 812         blob_nest_end(&b, s);
 813
 814         if (ubus_start_request(ctx, &req, b.head, UBUS_MSG_INVOKE, UBUS_SYSTEM_OBJECT_EVENT) < 0)
 815                 return UBUS_STATUS_INVALID_ARGUMENT;
 816
 817         return ubus_complete_request(ctx, &req, 0);
 818 }
 819
 820 static void ubus_default_connection_lost(struct ubus_context *ctx)
 821 {
 822         if (ctx->sock.registered)
 823                 uloop_end();
 824 }
 825
 826 struct ubus_context *ubus_connect(const char *path)
 827 {
 828         struct ubus_context *ctx;
 829         struct {
 830                 struct ubus_msghdr hdr;
 831                 struct blob_attr data;
 832         } hdr;
 833         struct blob_attr *buf;
 834
 835         if (!path)
 836                 path = UBUS_UNIX_SOCKET;
 837
 838         ctx = calloc(1, sizeof(*ctx));
 839         if (!ctx)
 840                 goto error;
 841
 842         ctx->sock.fd = usock(USOCK_UNIX, path, NULL);
 843         if (ctx->sock.fd < 0)
 844                 goto error_free;
 845
 846         ctx->sock.cb = ubus_handle_data;
 847
 848         if (read(ctx->sock.fd, &hdr, sizeof(hdr)) != sizeof(hdr))
 849                 goto error_close;
 850
 851         if (!ubus_validate_hdr(&hdr.hdr))
 852                 goto error_close;
 853
 854         if (hdr.hdr.type != UBUS_MSG_HELLO)
 855                 goto error_close;
 856
 857         buf = calloc(1, blob_raw_len(&hdr.data));
 858         if (!buf)
 859                 goto error_close;
 860
 861         memcpy(buf, &hdr.data, sizeof(hdr.data));
 862         if (read(ctx->sock.fd, blob_data(buf), blob_len(buf)) != blob_len(buf))
 863                 goto error_free_buf;
 864
 865         ctx->local_id = hdr.hdr.peer;
 866         free(buf);
 867
 868         ctx->connection_lost = ubus_default_connection_lost;
 869
 870         INIT_LIST_HEAD(&ctx->requests);
 871         avl_init(&ctx->objects, ubus_cmp_id, false, NULL);
 872
 873         if (!ctx->local_id)
 874                 goto error_close;
 875
 876         fcntl(ctx->sock.fd, F_SETFL, fcntl(ctx->sock.fd, F_GETFL) | O_NONBLOCK);
 877
 878         return ctx;
 879
 880 error_free_buf:
 881         free(buf);
 882 error_close:
 883         close(ctx->sock.fd);
 884 error_free:
 885         free(ctx);
 886 error:
 887         return NULL;
 888 }
 889
 890 void ubus_free(struct ubus_context *ctx)
 891 {
 892         close(ctx->sock.fd);
 893         free(ctx);
 894 }