src/ubus.c

   1 #include <syslog.h>
   2 #include <libubus.h>
   3 #include <libubox/uloop.h>
   4 #include <arpa/inet.h>
   5
   6 #include "odhcpd.h"
   7 #include "dhcpv6.h"
   8 #include "dhcpv4.h"
   9
  10
  11 static struct ubus_context *ubus = NULL;
  12 static struct ubus_subscriber netifd;
  13 static struct blob_buf b;
  14 static struct blob_attr *dump = NULL;
  15 static uint32_t objid = 0;
  16
  17
  18 static int handle_dhcpv4_leases(struct ubus_context *ctx, _unused struct ubus_object *obj,
  19                 struct ubus_request_data *req, _unused const char *method,
  20                 _unused struct blob_attr *msg)
  21 {
  22         blob_buf_init(&b, 0);
  23         void *a = blobmsg_open_table(&b, "device");
  24         time_t now = odhcpd_time();
  25
  26         struct interface *iface;
  27         list_for_each_entry(iface, &interfaces, head) {
  28                 if (iface->dhcpv4 != RELAYD_SERVER)
  29                         continue;
  30
  31                 void *i = blobmsg_open_table(&b, iface->ifname);
  32                 void *j = blobmsg_open_array(&b, "leases");
  33
  34                 struct dhcpv4_assignment *lease;
  35                 list_for_each_entry(lease, &iface->dhcpv4_assignments, head) {
  36                         if (lease->valid_until < now)
  37                                 continue;
  38
  39                         void *l = blobmsg_open_table(&b, NULL);
  40
  41                         char *buf = blobmsg_alloc_string_buffer(&b, "mac", 13);
  42                         odhcpd_hexlify(buf, lease->hwaddr, sizeof(lease->hwaddr));
  43                         blobmsg_add_string_buffer(&b);
  44
  45                         blobmsg_add_string(&b, "hostname", lease->hostname);
  46
  47                         buf = blobmsg_alloc_string_buffer(&b, "ip", INET_ADDRSTRLEN);
  48                         struct in_addr addr = {htonl(lease->addr)};
  49                         inet_ntop(AF_INET, &addr, buf, INET_ADDRSTRLEN);
  50                         blobmsg_add_string_buffer(&b);
  51
  52                         blobmsg_add_u32(&b, "valid", now - lease->valid_until);
  53
  54                         blobmsg_close_table(&b, l);
  55                 }
  56
  57                 blobmsg_close_array(&b, j);
  58                 blobmsg_close_table(&b, i);
  59         }
  60
  61         blobmsg_close_table(&b, a);
  62         ubus_send_reply(ctx, req, b.head);
  63         return 0;
  64 }
  65
  66
  67 static int handle_dhcpv6_leases(_unused struct ubus_context *ctx, _unused struct ubus_object *obj,
  68                 _unused struct ubus_request_data *req, _unused const char *method,
  69                 _unused struct blob_attr *msg)
  70 {
  71         blob_buf_init(&b, 0);
  72         void *a = blobmsg_open_table(&b, "device");
  73         time_t now = odhcpd_time();
  74
  75         struct interface *iface;
  76         list_for_each_entry(iface, &interfaces, head) {
  77                 if (iface->dhcpv6 != RELAYD_SERVER)
  78                         continue;
  79
  80                 void *i = blobmsg_open_table(&b, iface->ifname);
  81                 void *j = blobmsg_open_array(&b, "leases");
  82
  83                 struct dhcpv6_assignment *lease;
  84                 list_for_each_entry(lease, &iface->ia_assignments, head) {
  85                         if (lease->valid_until < now)
  86                                 continue;
  87
  88                         void *l = blobmsg_open_table(&b, NULL);
  89
  90                         char *buf = blobmsg_alloc_string_buffer(&b, "duid", 264);
  91                         odhcpd_hexlify(buf, lease->clid_data, lease->clid_len);
  92                         blobmsg_add_string_buffer(&b);
  93
  94                         blobmsg_add_u32(&b, "iaid", ntohl(lease->iaid));
  95                         blobmsg_add_string(&b, "hostname", (lease->hostname) ? lease->hostname : "");
  96                         blobmsg_add_u32(&b, "assigned", lease->assigned);
  97                         blobmsg_add_u32(&b, "length", lease->length);
  98
  99                         void *m = blobmsg_open_array(&b, "ipv6");
 100                         struct in6_addr addr;
 101                         for (size_t i = 0; i < iface->ia_addr_len; ++i) {
 102                                 if (iface->ia_addr[i].prefix > 64)
 103                                         continue;
 104
 105                                 addr = iface->ia_addr[i].addr;
 106                                 if (lease->length == 128)
 107                                         addr.s6_addr32[3] = htonl(lease->assigned);
 108                                 else
 109                                         addr.s6_addr32[1] |= htonl(lease->assigned);
 110
 111                                 char *c = blobmsg_alloc_string_buffer(&b, NULL, INET6_ADDRSTRLEN);
 112                                 inet_ntop(AF_INET6, &addr, c, INET6_ADDRSTRLEN);
 113                                 blobmsg_add_string_buffer(&b);
 114                         }
 115                         blobmsg_close_table(&b, m);
 116
 117                         blobmsg_add_u32(&b, "valid", now - lease->valid_until);
 118
 119                         blobmsg_close_table(&b, l);
 120                 }
 121
 122                 blobmsg_close_array(&b, j);
 123                 blobmsg_close_table(&b, i);
 124         }
 125
 126         blobmsg_close_table(&b, a);
 127         ubus_send_reply(ctx, req, b.head);
 128         return 0;
 129 }
 130
 131
 132 static struct ubus_method main_object_methods[] = {
 133         {.name = "ipv4leases", .handler = handle_dhcpv4_leases},
 134         {.name = "ipv6leases", .handler = handle_dhcpv6_leases},
 135 };
 136
 137 static struct ubus_object_type main_object_type =
 138                 UBUS_OBJECT_TYPE("dhcp", main_object_methods);
 139
 140 static struct ubus_object main_object = {
 141         .name = "dhcp",
 142         .type = &main_object_type,
 143         .methods = main_object_methods,
 144         .n_methods = ARRAY_SIZE(main_object_methods),
 145 };
 146
 147
 148 enum {
 149         DUMP_ATTR_INTERFACE,
 150         DUMP_ATTR_MAX
 151 };
 152
 153 static const struct blobmsg_policy dump_attrs[DUMP_ATTR_MAX] = {
 154         [DUMP_ATTR_INTERFACE] = { .name = "interface", .type = BLOBMSG_TYPE_ARRAY },
 155 };
 156
 157
 158 enum {
 159         IFACE_ATTR_INTERFACE,
 160         IFACE_ATTR_IFNAME,
 161         IFACE_ATTR_UP,
 162         IFACE_ATTR_DATA,
 163         IFACE_ATTR_PREFIX,
 164         IFACE_ATTR_ADDRESS,
 165         IFACE_ATTR_MAX,
 166 };
 167
 168 static const struct blobmsg_policy iface_attrs[IFACE_ATTR_MAX] = {
 169         [IFACE_ATTR_INTERFACE] = { .name = "interface", .type = BLOBMSG_TYPE_STRING },
 170         [IFACE_ATTR_IFNAME] = { .name = "l3_device", .type = BLOBMSG_TYPE_STRING },
 171         [IFACE_ATTR_UP] = { .name = "up", .type = BLOBMSG_TYPE_BOOL },
 172         [IFACE_ATTR_DATA] = { .name = "data", .type = BLOBMSG_TYPE_TABLE },
 173         [IFACE_ATTR_PREFIX] = { .name = "ipv6-prefix", .type = BLOBMSG_TYPE_ARRAY },
 174         [IFACE_ATTR_ADDRESS] = { .name = "ipv6-address", .type = BLOBMSG_TYPE_ARRAY },
 175 };
 176
 177 static void handle_dump(_unused struct ubus_request *req, _unused int type, struct blob_attr *msg)
 178 {
 179         struct blob_attr *tb[DUMP_ATTR_INTERFACE];
 180         blobmsg_parse(dump_attrs, DUMP_ATTR_MAX, tb, blob_data(msg), blob_len(msg));
 181
 182         if (!tb[DUMP_ATTR_INTERFACE])
 183                 return;
 184
 185         free(dump);
 186         dump = blob_memdup(tb[DUMP_ATTR_INTERFACE]);
 187         odhcpd_reload();
 188 }
 189
 190
 191 static int handle_update(_unused struct ubus_context *ctx, _unused struct ubus_object *obj,
 192                 _unused struct ubus_request_data *req, _unused const char *method,
 193                 struct blob_attr *msg)
 194 {
 195         struct blob_attr *tb[IFACE_ATTR_MAX];
 196         blobmsg_parse(iface_attrs, IFACE_ATTR_MAX, tb, blob_data(msg), blob_len(msg));
 197
 198         const char *interface = (tb[IFACE_ATTR_INTERFACE]) ?
 199                         blobmsg_get_string(tb[IFACE_ATTR_INTERFACE]) : "";
 200         const char *ifname = (tb[IFACE_ATTR_IFNAME]) ?
 201                         blobmsg_get_string(tb[IFACE_ATTR_IFNAME]) : "";
 202
 203         struct interface *c, *iface = NULL;
 204         list_for_each_entry(c, &interfaces, head)
 205                 if (!strcmp(interface, c->name) || !strcmp(ifname, c->ifname))
 206                         iface = c;
 207
 208         if (iface && iface->ignore)
 209                 return 0;
 210
 211         ubus_invoke(ubus, objid, "dump", NULL, handle_dump, NULL, 0);
 212         return 0;
 213 }
 214
 215
 216 static void subscribe_netifd(void)
 217 {
 218         ubus_subscribe(ubus, &netifd, objid);
 219         ubus_invoke(ubus, objid, "dump", NULL, handle_dump, NULL, 0);
 220 }
 221
 222
 223 void ubus_apply_network(void)
 224 {
 225         struct blob_attr *c;
 226         unsigned rem;
 227
 228         if (!dump)
 229                 return;
 230
 231         blobmsg_for_each_attr(c, dump, rem) {
 232                 struct blob_attr *tb[IFACE_ATTR_MAX];
 233                 blobmsg_parse(iface_attrs, IFACE_ATTR_MAX, tb, blobmsg_data(c), blobmsg_data_len(c));
 234
 235                 if (!tb[IFACE_ATTR_INTERFACE] || !tb[IFACE_ATTR_DATA])
 236                         continue;
 237
 238                 const char *interface = (tb[IFACE_ATTR_INTERFACE]) ?
 239                                 blobmsg_get_string(tb[IFACE_ATTR_INTERFACE]) : "";
 240                 const char *ifname = (tb[IFACE_ATTR_IFNAME]) ?
 241                                 blobmsg_get_string(tb[IFACE_ATTR_IFNAME]) : "";
 242
 243                 bool matched = false;
 244                 struct interface *c;
 245                 list_for_each_entry(c, &interfaces, head) {
 246                         char *f = memmem(c->upstream, c->upstream_len,
 247                                         interface, strlen(interface) + 1);
 248                         bool cmatched = !strcmp(interface, c->name) || !strcmp(ifname, c->ifname);
 249                         matched |= cmatched;
 250
 251                         if (!cmatched && (!c->upstream_len || !f || (f != c->upstream && f[-1] != 0)))
 252                                 continue;
 253
 254                         if (!c->ignore)
 255                                 config_parse_interface(blobmsg_data(tb[IFACE_ATTR_DATA]),
 256                                                 blobmsg_data_len(tb[IFACE_ATTR_DATA]), c->name, false);
 257                 }
 258
 259                 if (!matched)
 260                         config_parse_interface(blobmsg_data(tb[IFACE_ATTR_DATA]),
 261                                         blobmsg_data_len(tb[IFACE_ATTR_DATA]), interface, false);
 262         }
 263 }
 264
 265
 266 enum {
 267         OBJ_ATTR_ID,
 268         OBJ_ATTR_PATH,
 269         OBJ_ATTR_MAX
 270 };
 271
 272 static const struct blobmsg_policy obj_attrs[OBJ_ATTR_MAX] = {
 273         [OBJ_ATTR_ID] = { .name = "id", .type = BLOBMSG_TYPE_INT32 },
 274         [OBJ_ATTR_PATH] = { .name = "path", .type = BLOBMSG_TYPE_STRING },
 275 };
 276
 277
 278 static void handle_event(_unused struct ubus_context *ctx, _unused struct ubus_event_handler *ev,
 279                 _unused const char *type, struct blob_attr *msg)
 280 {
 281         struct blob_attr *tb[OBJ_ATTR_MAX];
 282         blobmsg_parse(obj_attrs, OBJ_ATTR_MAX, tb, blob_data(msg), blob_len(msg));
 283
 284         if (!tb[OBJ_ATTR_ID] || !tb[OBJ_ATTR_PATH])
 285                 return;
 286
 287         if (strcmp(blobmsg_get_string(tb[OBJ_ATTR_PATH]), "network.interface"))
 288                 return;
 289
 290         objid = blobmsg_get_u32(tb[OBJ_ATTR_ID]);
 291         subscribe_netifd();
 292 }
 293
 294 static struct ubus_event_handler event_handler = { .cb = handle_event };
 295
 296
 297 const char* ubus_get_ifname(const char *name)
 298 {
 299         struct blob_attr *c;
 300         unsigned rem;
 301
 302         if (!dump)
 303                 return NULL;
 304
 305         blobmsg_for_each_attr(c, dump, rem) {
 306                 struct blob_attr *tb[IFACE_ATTR_MAX];
 307                 blobmsg_parse(iface_attrs, IFACE_ATTR_MAX, tb, blobmsg_data(c), blobmsg_data_len(c));
 308
 309                 if (!tb[IFACE_ATTR_INTERFACE] || strcmp(name,
 310                                 blobmsg_get_string(tb[IFACE_ATTR_INTERFACE])))
 311                         continue;
 312
 313                 if (tb[IFACE_ATTR_IFNAME])
 314                         return blobmsg_get_string(tb[IFACE_ATTR_IFNAME]);
 315         }
 316
 317         return NULL;
 318 }
 319
 320
 321 bool ubus_has_prefix(const char *name, const char *ifname)
 322 {
 323         struct blob_attr *c, *cur;
 324         unsigned rem;
 325
 326         if (!dump)
 327                 return NULL;
 328
 329         blobmsg_for_each_attr(c, dump, rem) {
 330                 struct blob_attr *tb[IFACE_ATTR_MAX];
 331                 blobmsg_parse(iface_attrs, IFACE_ATTR_MAX, tb, blobmsg_data(c), blobmsg_data_len(c));
 332
 333                 if (!tb[IFACE_ATTR_INTERFACE] || !tb[IFACE_ATTR_IFNAME])
 334                         continue;
 335
 336                 if (strcmp(name, blobmsg_get_string(tb[IFACE_ATTR_INTERFACE])) ||
 337                                 strcmp(ifname, blobmsg_get_string(tb[IFACE_ATTR_IFNAME])))
 338                         continue;
 339
 340                 if ((cur = tb[IFACE_ATTR_PREFIX])) {
 341                         if (blobmsg_type(cur) != BLOBMSG_TYPE_ARRAY || !blobmsg_check_attr(cur, NULL))
 342                                 continue;
 343
 344                         struct blob_attr *d;
 345                         unsigned drem;
 346                         blobmsg_for_each_attr(d, cur, drem) {
 347                                 return true;
 348                         }
 349                 }
 350         }
 351
 352         return false;
 353 }
 354
 355
 356 enum {
 357         ADDR_ATTR_ADDR,
 358         ADDR_ATTR_CLASS,
 359         ADDR_ATTR_MAX
 360 };
 361
 362 static const struct blobmsg_policy addr_attrs[ADDR_ATTR_MAX] = {
 363         [ADDR_ATTR_ADDR] = { .name = "address", .type = BLOBMSG_TYPE_STRING },
 364         [ADDR_ATTR_CLASS] = { .name = "class", .type = BLOBMSG_TYPE_STRING },
 365 };
 366
 367 bool ubus_get_class(const char *ifname, const struct in6_addr *addr, uint16_t *pclass)
 368 {
 369         struct blob_attr *c, *cur;
 370         unsigned rem;
 371
 372         if (!dump)
 373                 return false;
 374
 375         blobmsg_for_each_attr(c, dump, rem) {
 376                 struct blob_attr *tb[IFACE_ATTR_MAX];
 377                 blobmsg_parse(iface_attrs, IFACE_ATTR_MAX, tb, blobmsg_data(c), blobmsg_data_len(c));
 378
 379                 if (!tb[IFACE_ATTR_IFNAME])
 380                         continue;
 381
 382                 if (strcmp(ifname, blobmsg_get_string(tb[IFACE_ATTR_IFNAME])))
 383                         continue;
 384
 385                 if ((cur = tb[IFACE_ATTR_ADDRESS])) {
 386                         if (blobmsg_type(cur) != BLOBMSG_TYPE_ARRAY || !blobmsg_check_attr(cur, NULL))
 387                                 continue;
 388
 389                         struct blob_attr *d;
 390                         unsigned drem;
 391                         blobmsg_for_each_attr(d, cur, drem) {
 392                                 struct blob_attr *t[ADDR_ATTR_MAX];
 393                                 blobmsg_parse(addr_attrs, ADDR_ATTR_MAX, t, blobmsg_data(d), blobmsg_data_len(d));
 394
 395                                 if (!t[ADDR_ATTR_ADDR] || !t[ADDR_ATTR_CLASS])
 396                                         continue;
 397
 398                                 const char *addrs = blobmsg_get_string(t[ADDR_ATTR_ADDR]);
 399                                 const char *class = blobmsg_get_string(t[ADDR_ATTR_CLASS]);
 400
 401                                 struct in6_addr ip6addr;
 402                                 inet_pton(AF_INET6, addrs, &ip6addr);
 403
 404                                 if (IN6_ARE_ADDR_EQUAL(&ip6addr, addr)) {
 405                                         *pclass = atoi(class);
 406                                         return true;
 407                                 }
 408                         }
 409                 }
 410
 411                 return false;
 412         }
 413
 414         return false;
 415 }
 416
 417
 418 int init_ubus(void)
 419 {
 420         if (!(ubus = ubus_connect(NULL))) {
 421                 syslog(LOG_ERR, "Unable to connect to ubus: %s", strerror(errno));
 422                 return -1;
 423         }
 424
 425         netifd.cb = handle_update;
 426         ubus_register_subscriber(ubus, &netifd);
 427
 428         ubus_add_uloop(ubus);
 429         ubus_add_object(ubus, &main_object);
 430         ubus_register_event_handler(ubus, &event_handler, "ubus.object.add");
 431         if (!ubus_lookup_id(ubus, "network.interface", &objid))
 432                 subscribe_netifd();
 433
 434         return 0;
 435 }
 436