device.c

   1 #include <string.h>
   2 #include <stdlib.h>
   3 #include <stdio.h>
   4 #include <assert.h>
   5
   6 #include <sys/types.h>
   7 #include <sys/socket.h>
   8 #include <net/ethernet.h>
   9
  10 #ifdef linux
  11 #include <netinet/ether.h>
  12 #endif
  13
  14 #include "netifd.h"
  15 #include "system.h"
  16 #include "config.h"
  17
  18 static struct avl_tree devices;
  19 static struct avl_tree aliases;
  20
  21 struct alias_device {
  22         struct avl_node avl;
  23         struct device dev;
  24         struct device_user dep;
  25         bool cleanup;
  26         char name[];
  27 };
  28
  29 static const struct device_type alias_device_type;
  30
  31 static const struct blobmsg_policy dev_attrs[__DEV_ATTR_MAX] = {
  32         [DEV_ATTR_TYPE] = { "type", BLOBMSG_TYPE_STRING },
  33         [DEV_ATTR_IFNAME] = { "ifname", BLOBMSG_TYPE_ARRAY },
  34         [DEV_ATTR_MTU] = { "mtu", BLOBMSG_TYPE_INT32 },
  35         [DEV_ATTR_MACADDR] = { "macaddr", BLOBMSG_TYPE_STRING },
  36         [DEV_ATTR_TXQUEUELEN] = { "txqueuelen", BLOBMSG_TYPE_INT32 },
  37 };
  38
  39 const struct config_param_list device_attr_list = {
  40         .n_params = __DEV_ATTR_MAX,
  41         .params = dev_attrs,
  42 };
  43
  44 static struct device *
  45 simple_device_create(const char *name, struct blob_attr *attr)
  46 {
  47         struct blob_attr *tb[__DEV_ATTR_MAX];
  48         struct device *dev = NULL;
  49
  50         blobmsg_parse(dev_attrs, __DEV_ATTR_MAX, tb, blob_data(attr), blob_len(attr));
  51         dev = device_get(name, true);
  52         if (!dev)
  53                 return NULL;
  54
  55         device_init_settings(dev, tb);
  56
  57         return dev;
  58 }
  59
  60 static void simple_device_free(struct device *dev)
  61 {
  62         device_cleanup(dev);
  63         free(dev);
  64 }
  65
  66 const struct device_type simple_device_type = {
  67         .name = "Network device",
  68         .config_params = &device_attr_list,
  69
  70         .create = simple_device_create,
  71         .check_state = system_if_check,
  72         .free = simple_device_free,
  73 };
  74
  75 static int
  76 alias_device_set_state(struct device *dev, bool state)
  77 {
  78         struct alias_device *alias;
  79
  80         alias = container_of(dev, struct alias_device, dev);
  81         if (!alias->dep.dev)
  82                 return -1;
  83
  84         if (state)
  85                 return device_claim(&alias->dep);
  86
  87         device_release(&alias->dep);
  88         if (alias->cleanup)
  89                 device_remove_user(&alias->dep);
  90         return 0;
  91 }
  92
  93 static struct device *
  94 alias_device_create(const char *name, struct blob_attr *attr)
  95 {
  96         struct alias_device *alias;
  97
  98         alias = calloc(1, sizeof(*alias) + strlen(name) + 1);
  99         strcpy(alias->name, name);
 100         alias->dev.set_state = alias_device_set_state;
 101         device_init_virtual(&alias->dev, &alias_device_type, NULL);
 102         alias->avl.key = alias->name;
 103         avl_insert(&aliases, &alias->avl);
 104
 105         return &alias->dev;
 106 }
 107
 108 static void alias_device_free(struct device *dev)
 109 {
 110         struct alias_device *alias;
 111
 112         device_cleanup(dev);
 113
 114         alias = container_of(dev, struct alias_device, dev);
 115         avl_delete(&aliases, &alias->avl);
 116         free(alias);
 117 }
 118
 119 static const struct device_type alias_device_type = {
 120         .name = "Network alias",
 121         .create = alias_device_create,
 122         .free = alias_device_free,
 123 };
 124
 125 void
 126 device_init_settings(struct device *dev, struct blob_attr **tb)
 127 {
 128         struct blob_attr *cur;
 129         struct ether_addr *ea;
 130
 131         dev->flags = 0;
 132
 133         if ((cur = tb[DEV_ATTR_MTU])) {
 134                 dev->mtu = blobmsg_get_u32(cur);
 135                 dev->flags |= DEV_OPT_MTU;
 136         }
 137
 138         if ((cur = tb[DEV_ATTR_TXQUEUELEN])) {
 139                 dev->txqueuelen = blobmsg_get_u32(cur);
 140                 dev->flags |= DEV_OPT_TXQUEUELEN;
 141         }
 142
 143         if ((cur = tb[DEV_ATTR_MACADDR])) {
 144                 ea = ether_aton(blob_data(cur));
 145                 if (ea) {
 146                         memcpy(dev->macaddr, ea, sizeof(dev->macaddr));
 147                         dev->flags |= DEV_OPT_MACADDR;
 148                 }
 149         }
 150 }
 151
 152 static void __init dev_init(void)
 153 {
 154         avl_init(&devices, avl_strcmp, true, NULL);
 155         avl_init(&aliases, avl_strcmp, false, NULL);
 156 }
 157
 158 static void device_broadcast_event(struct device *dev, enum device_event ev)
 159 {
 160         struct device_user *dep, *tmp;
 161
 162         list_for_each_entry_safe(dep, tmp, &dev->users, list) {
 163                 if (!dep->cb)
 164                         continue;
 165
 166                 dep->cb(dep, ev);
 167         }
 168 }
 169
 170 void
 171 alias_notify_device(const char *name, struct device *dev)
 172 {
 173         struct alias_device *alias;
 174
 175         alias = avl_find_element(&aliases, name, alias, avl);
 176         if (!alias)
 177                 return;
 178
 179         alias->cleanup = !dev;
 180         if (dev) {
 181                 if (dev != alias->dep.dev) {
 182                         if (alias->dep.dev)
 183                                 device_remove_user(&alias->dep);
 184                         strcpy(alias->dev.ifname, dev->ifname);
 185                         device_add_user(&alias->dep, dev);
 186                 }
 187         }
 188
 189         device_set_present(&alias->dev, !!dev);
 190
 191         if (!dev && alias->dep.dev && !alias->dep.dev->active)
 192                 device_remove_user(&alias->dep);
 193 }
 194
 195 static int set_device_state(struct device *dev, bool state)
 196 {
 197         if (state)
 198                 system_if_up(dev);
 199         else
 200                 system_if_down(dev);
 201
 202         return 0;
 203 }
 204
 205 int device_claim(struct device_user *dep)
 206 {
 207         struct device *dev = dep->dev;
 208         int ret;
 209
 210         if (dep->claimed)
 211                 return 0;
 212
 213         dep->claimed = true;
 214         D(DEVICE, "Claim %s %s, new refcount: %d\n", dev->type->name, dev->ifname, dev->active + 1);
 215         if (++dev->active != 1)
 216                 return 0;
 217
 218         device_broadcast_event(dev, DEV_EVENT_SETUP);
 219         ret = dev->set_state(dev, true);
 220         if (ret == 0)
 221                 device_broadcast_event(dev, DEV_EVENT_UP);
 222         else {
 223                 D(DEVICE, "claim device %s failed: %d\n", dev->ifname, ret);
 224                 dev->active = 0;
 225                 dep->claimed = false;
 226         }
 227
 228         return ret;
 229 }
 230
 231 void device_release(struct device_user *dep)
 232 {
 233         struct device *dev = dep->dev;
 234
 235         if (!dep->claimed)
 236                 return;
 237
 238         dep->claimed = false;
 239         dev->active--;
 240         D(DEVICE, "Release %s %s, new refcount: %d\n", dev->type->name, dev->ifname, dev->active);
 241         assert(dev->active >= 0);
 242
 243         if (dev->active)
 244                 return;
 245
 246         device_broadcast_event(dev, DEV_EVENT_TEARDOWN);
 247         dev->set_state(dev, false);
 248         device_broadcast_event(dev, DEV_EVENT_DOWN);
 249 }
 250
 251 int device_check_state(struct device *dev)
 252 {
 253         if (!dev->type->check_state)
 254                 return 0;
 255
 256         return dev->type->check_state(dev);
 257 }
 258
 259 void device_init_virtual(struct device *dev, const struct device_type *type, const char *name)
 260 {
 261         assert(dev);
 262         assert(type);
 263
 264         if (name)
 265                 strncpy(dev->ifname, name, IFNAMSIZ);
 266
 267         D(DEVICE, "Initialize device '%s'\n", dev->ifname);
 268         INIT_LIST_HEAD(&dev->users);
 269         dev->type = type;
 270 }
 271
 272 int device_init(struct device *dev, const struct device_type *type, const char *ifname)
 273 {
 274         int ret;
 275
 276         device_init_virtual(dev, type, ifname);
 277
 278         if (!dev->set_state)
 279                 dev->set_state = set_device_state;
 280
 281         dev->avl.key = dev->ifname;
 282
 283         ret = avl_insert(&devices, &dev->avl);
 284         if (ret < 0)
 285                 return ret;
 286
 287         system_if_clear_state(dev);
 288         device_check_state(dev);
 289
 290         return 0;
 291 }
 292
 293 static struct device *
 294 device_create_default(const char *name)
 295 {
 296         struct device *dev;
 297
 298         D(DEVICE, "Create simple device '%s'\n", name);
 299         dev = calloc(1, sizeof(*dev));
 300         device_init(dev, &simple_device_type, name);
 301         dev->default_config = true;
 302         return dev;
 303 }
 304
 305 static struct device *
 306 device_alias_get(const char *name)
 307 {
 308         struct alias_device *alias;
 309
 310         alias = avl_find_element(&aliases, name, alias, avl);
 311         if (alias)
 312                 return &alias->dev;
 313
 314         return alias_device_create(name, NULL);
 315 }
 316
 317 struct device *
 318 device_get(const char *name, bool create)
 319 {
 320         struct device *dev;
 321
 322         if (strchr(name, '.'))
 323                 return get_vlan_device_chain(name, create);
 324
 325         if (name[0] == '@')
 326                 return device_alias_get(name + 1);
 327
 328         dev = avl_find_element(&devices, name, dev, avl);
 329         if (dev)
 330                 return dev;
 331
 332         if (!create)
 333                 return NULL;
 334
 335         return device_create_default(name);
 336 }
 337
 338 static void
 339 device_delete(struct device *dev)
 340 {
 341         if (!dev->avl.key)
 342                 return;
 343
 344         D(DEVICE, "Delete device '%s' from list\n", dev->ifname);
 345         avl_delete(&devices, &dev->avl);
 346         dev->avl.key = NULL;
 347 }
 348
 349 void device_cleanup(struct device *dev)
 350 {
 351         struct device_user *dep, *tmp;
 352
 353         D(DEVICE, "Clean up device '%s'\n", dev->ifname);
 354         list_for_each_entry_safe(dep, tmp, &dev->users, list) {
 355                 if (!dep->cb)
 356                         continue;
 357
 358                 dep->cb(dep, DEV_EVENT_REMOVE);
 359                 device_release(dep);
 360         }
 361
 362         device_delete(dev);
 363 }
 364
 365 void device_set_present(struct device *dev, bool state)
 366 {
 367         if (dev->present == state)
 368                 return;
 369
 370         D(DEVICE, "%s '%s' %s present\n", dev->type->name, dev->ifname, state ? "is now" : "is no longer" );
 371         dev->present = state;
 372         device_broadcast_event(dev, state ? DEV_EVENT_ADD : DEV_EVENT_REMOVE);
 373 }
 374
 375 void device_add_user(struct device_user *dep, struct device *dev)
 376 {
 377         dep->dev = dev;
 378         list_add_tail(&dep->list, &dev->users);
 379         if (dep->cb && dev->present) {
 380                 dep->cb(dep, DEV_EVENT_ADD);
 381                 if (dev->active)
 382                         dep->cb(dep, DEV_EVENT_UP);
 383         }
 384 }
 385
 386 static void
 387 __device_free_unused(struct device *dev)
 388 {
 389         if (!list_empty(&dev->users) || dev->current_config || config_init)
 390                 return;
 391
 392         device_free(dev);
 393 }
 394
 395 void device_remove_user(struct device_user *dep)
 396 {
 397         struct device *dev = dep->dev;
 398
 399         if (dep->claimed)
 400                 device_release(dep);
 401
 402         list_del(&dep->list);
 403         dep->dev = NULL;
 404         __device_free_unused(dev);
 405 }
 406
 407 void
 408 device_free_unused(struct device *dev)
 409 {
 410         struct device *tmp;
 411
 412         if (dev)
 413                 return __device_free_unused(dev);
 414
 415         avl_for_each_element_safe(&devices, dev, avl, tmp)
 416                 __device_free_unused(dev);
 417 }
 418
 419 void
 420 device_init_pending(void)
 421 {
 422         struct device *dev, *tmp;
 423
 424         avl_for_each_element_safe(&devices, dev, avl, tmp) {
 425                 if (!dev->config_pending)
 426                         continue;
 427
 428                 dev->type->config_init(dev);
 429                 dev->config_pending = false;
 430         }
 431 }
 432
 433 static enum dev_change_type
 434 device_reload_config(struct device *dev, struct blob_attr *attr)
 435 {
 436         struct blob_attr *tb[__DEV_ATTR_MAX];
 437         const struct config_param_list *cfg = dev->type->config_params;
 438
 439         if (config_check_equal(dev->config, attr, cfg))
 440                 return DEV_CONFIG_NO_CHANGE;
 441
 442         if (cfg == &device_attr_list) {
 443                 memset(tb, 0, sizeof(tb));
 444
 445                 if (attr)
 446                         blobmsg_parse(dev_attrs, __DEV_ATTR_MAX, tb,
 447                                 blob_data(attr), blob_len(attr));
 448
 449                 device_init_settings(dev, tb);
 450                 return DEV_CONFIG_APPLIED;
 451         } else
 452                 return DEV_CONFIG_RECREATE;
 453 }
 454
 455 enum dev_change_type
 456 device_set_config(struct device *dev, const struct device_type *type,
 457                   struct blob_attr *attr)
 458 {
 459         if (type != dev->type)
 460                 return DEV_CONFIG_RECREATE;
 461
 462         if (dev->type->reload)
 463                 return dev->type->reload(dev, attr);
 464
 465         return device_reload_config(dev, attr);
 466 }
 467
 468 static void
 469 device_replace(struct device *dev, struct device *odev)
 470 {
 471         struct device_user *dep, *tmp;
 472         bool present = odev->present;
 473
 474         if (present)
 475                 device_set_present(odev, false);
 476
 477         list_for_each_entry_safe(dep, tmp, &odev->users, list) {
 478                 device_release(dep);
 479                 list_move_tail(&dep->list, &dev->users);
 480                 dep->dev = dev;
 481         }
 482         device_free(odev);
 483
 484         if (present)
 485                 device_set_present(dev, true);
 486 }
 487
 488 void
 489 device_reset_config(void)
 490 {
 491         struct device *dev;
 492
 493         avl_for_each_element(&devices, dev, avl)
 494                 dev->current_config = false;
 495 }
 496
 497 void
 498 device_reset_old(void)
 499 {
 500         struct device *dev, *tmp, *ndev;
 501
 502         avl_for_each_element_safe(&devices, dev, avl, tmp) {
 503                 if (dev->current_config || dev->default_config)
 504                         continue;
 505
 506                 if (dev->type != &simple_device_type)
 507                         continue;
 508
 509                 ndev = device_create_default(dev->ifname);
 510                 device_replace(ndev, dev);
 511         }
 512 }
 513
 514 struct device *
 515 device_create(const char *name, const struct device_type *type,
 516               struct blob_attr *config)
 517 {
 518         struct device *odev = NULL, *dev;
 519         enum dev_change_type change;
 520
 521         config = config_memdup(config);
 522         if (!config)
 523                 return NULL;
 524
 525         odev = device_get(name, false);
 526         if (odev) {
 527                 odev->current_config = true;
 528                 change = device_set_config(odev, type, config);
 529                 switch (change) {
 530                 case DEV_CONFIG_APPLIED:
 531                         D(DEVICE, "Device '%s': config applied\n", odev->ifname);
 532                         free(odev->config);
 533                         odev->config = config;
 534                         if (odev->present) {
 535                                 device_set_present(odev, false);
 536                                 device_set_present(odev, true);
 537                         }
 538                         return odev;
 539                 case DEV_CONFIG_NO_CHANGE:
 540                         D(DEVICE, "Device '%s': no configuration change\n", odev->ifname);
 541                         free(config);
 542                         return odev;
 543                 case DEV_CONFIG_RECREATE:
 544                         D(DEVICE, "Device '%s': recreate device\n", odev->ifname);
 545                         device_delete(odev);
 546                         break;
 547                 }
 548         } else
 549                 D(DEVICE, "Create new device '%s' (%s)\n", name, type->name);
 550
 551         dev = type->create(name, config);
 552         if (!dev)
 553                 return NULL;
 554
 555         dev->current_config = true;
 556         dev->config = config;
 557         if (odev)
 558                 device_replace(dev, odev);
 559
 560         if (!config_init && dev->config_pending)
 561                 type->config_init(dev);
 562
 563         return dev;
 564 }
 565
 566 void
 567 device_dump_status(struct blob_buf *b, struct device *dev)
 568 {
 569         void *c, *s;
 570
 571         if (!dev) {
 572                 avl_for_each_element(&devices, dev, avl) {
 573                         if (!dev->present)
 574                                 continue;
 575                         c = blobmsg_open_table(b, dev->ifname);
 576                         device_dump_status(b, dev);
 577                         blobmsg_close_table(b, c);
 578                 }
 579
 580                 return;
 581         }
 582
 583         if (!dev->present)
 584                 return;
 585
 586         blobmsg_add_string(b, "type", dev->type->name);
 587         blobmsg_add_u8(b, "up", !!dev->active);
 588         if (dev->type->dump_info)
 589                 dev->type->dump_info(dev, b);
 590
 591         s = blobmsg_open_table(b, "statistics");
 592         if (dev->type->dump_stats)
 593                 dev->type->dump_stats(dev, b);
 594         else
 595                 system_if_dump_stats(dev, b);
 596         blobmsg_close_table(b, s);
 597 }