proto-shell.c

   1 #define _GNU_SOURCE
   2
   3 #include <string.h>
   4 #include <stdlib.h>
   5 #include <stdio.h>
   6 #include <glob.h>
   7 #include <unistd.h>
   8 #include <fcntl.h>
   9 #include <signal.h>
  10
  11 #include <arpa/inet.h>
  12 #include <netinet/in.h>
  13
  14 #include <libubox/blobmsg_json.h>
  15
  16 #include "netifd.h"
  17 #include "interface.h"
  18 #include "interface-ip.h"
  19 #include "proto.h"
  20
  21 static struct netifd_fd proto_fd;
  22
  23 struct proto_shell_handler {
  24         struct list_head list;
  25         struct proto_handler proto;
  26         struct config_param_list config;
  27         char *config_buf;
  28         char script_name[];
  29 };
  30
  31 struct proto_shell_state {
  32         struct interface_proto_state proto;
  33         struct proto_shell_handler *handler;
  34         struct blob_attr *config;
  35
  36         struct device_user l3_dev;
  37
  38         struct uloop_timeout setup_timeout;
  39         struct netifd_process setup_task;
  40         struct netifd_process teardown_task;
  41         bool teardown_pending;
  42         bool teardown_wait_task;
  43
  44         struct netifd_process proto_task;
  45         int last_error;
  46 };
  47
  48 static int
  49 proto_shell_handler(struct interface_proto_state *proto,
  50                     enum interface_proto_cmd cmd, bool force)
  51 {
  52         struct proto_shell_state *state;
  53         struct proto_shell_handler *handler;
  54         struct netifd_process *proc;
  55         static char error_buf[32];
  56         const char *argv[7];
  57         char *envp[2];
  58         const char *action;
  59         char *config;
  60         int ret, i = 0, j = 0;
  61
  62         state = container_of(proto, struct proto_shell_state, proto);
  63         handler = state->handler;
  64
  65         if (cmd == PROTO_CMD_SETUP) {
  66                 action = "setup";
  67                 proc = &state->setup_task;
  68                 state->last_error = -1;
  69         } else {
  70                 action = "teardown";
  71                 proc = &state->teardown_task;
  72                 if (state->setup_task.uloop.pending && !state->teardown_wait_task) {
  73                         uloop_timeout_set(&state->setup_timeout, 1000);
  74                         kill(state->setup_task.uloop.pid, SIGTERM);
  75                         state->teardown_pending = true;
  76                         return 0;
  77                 }
  78                 if (state->last_error >= 0) {
  79                         snprintf(error_buf, sizeof(error_buf), "ERROR=%d", state->last_error);
  80                         envp[j++] = error_buf;
  81                 }
  82         }
  83
  84         config = blobmsg_format_json(state->config, true);
  85         if (!config)
  86                 return -1;
  87
  88         argv[i++] = handler->script_name;
  89         argv[i++] = handler->proto.name;
  90         argv[i++] = action;
  91         argv[i++] = proto->iface->name;
  92         argv[i++] = config;
  93         if (proto->iface->main_dev.dev)
  94                 argv[i++] = proto->iface->main_dev.dev->ifname;
  95         argv[i] = NULL;
  96         envp[j] = NULL;
  97
  98         ret = netifd_start_process(argv, envp, proc);
  99         free(config);
 100
 101         return ret;
 102 }
 103
 104 static void
 105 proto_shell_setup_timeout_cb(struct uloop_timeout *timeout)
 106 {
 107         struct proto_shell_state *state;
 108
 109         state = container_of(timeout, struct proto_shell_state, setup_timeout);
 110         kill(state->setup_task.uloop.pid, SIGKILL);
 111 }
 112
 113 static void
 114 proto_shell_setup_cb(struct netifd_process *p, int ret)
 115 {
 116         struct proto_shell_state *state;
 117
 118         state = container_of(p, struct proto_shell_state, setup_task);
 119         uloop_timeout_cancel(&state->setup_timeout);
 120         if (state->teardown_pending) {
 121                 state->teardown_pending = false;
 122                 proto_shell_handler(&state->proto, PROTO_CMD_TEARDOWN, false);
 123         }
 124 }
 125
 126 static void
 127 proto_shell_teardown_cb(struct netifd_process *p, int ret)
 128 {
 129         struct proto_shell_state *state;
 130
 131         state = container_of(p, struct proto_shell_state, teardown_task);
 132
 133         if (state->teardown_wait_task)
 134                 return;
 135
 136         netifd_kill_process(&state->proto_task);
 137         state->proto.proto_event(&state->proto, IFPEV_DOWN);
 138 }
 139
 140 static void
 141 proto_shell_task_cb(struct netifd_process *p, int ret)
 142 {
 143         struct proto_shell_state *state;
 144         bool teardown_wait_task;
 145
 146         state = container_of(p, struct proto_shell_state, proto_task);
 147
 148         teardown_wait_task = state->teardown_wait_task;
 149         state->teardown_wait_task = false;
 150         if (state->teardown_pending || state->teardown_task.uloop.pending)
 151                 return;
 152
 153         if (teardown_wait_task) {
 154                 proto_shell_teardown_cb(&state->teardown_task, 0);
 155                 return;
 156         }
 157
 158         state->last_error = WEXITSTATUS(ret);
 159         state->proto.proto_event(&state->proto, IFPEV_LINK_LOST);
 160         proto_shell_handler(&state->proto, PROTO_CMD_TEARDOWN, false);
 161 }
 162
 163 static void
 164 proto_shell_free(struct interface_proto_state *proto)
 165 {
 166         struct proto_shell_state *state;
 167
 168         state = container_of(proto, struct proto_shell_state, proto);
 169         free(state->config);
 170         free(state);
 171 }
 172
 173 static void
 174 proto_shell_parse_addr_list(struct interface_ip_settings *ip, struct blob_attr *attr,
 175                             bool v6, bool external)
 176 {
 177         struct device_addr *addr;
 178         struct blob_attr *cur;
 179         int rem;
 180
 181         blobmsg_for_each_attr(cur, attr, rem) {
 182                 if (blobmsg_type(cur) != BLOBMSG_TYPE_STRING) {
 183                         DPRINTF("Ignore wrong address type: %d\n", blobmsg_type(cur));
 184                         continue;
 185                 }
 186
 187                 addr = proto_parse_ip_addr_string(blobmsg_data(cur), v6, v6 ? 32 : 128);
 188                 if (!addr) {
 189                         DPRINTF("Failed to parse IP address string: %s\n", (char *) blobmsg_data(cur));
 190                         continue;
 191                 }
 192
 193                 if (external)
 194                         addr->flags |= DEVADDR_EXTERNAL;
 195
 196                 vlist_add(&ip->addr, &addr->node);
 197         }
 198 }
 199
 200 static void
 201 proto_shell_parse_route_list(struct interface *iface, struct blob_attr *attr,
 202                              bool v6)
 203 {
 204         struct blob_attr *cur;
 205         int rem;
 206
 207         blobmsg_for_each_attr(cur, attr, rem) {
 208                 if (blobmsg_type(cur) != BLOBMSG_TYPE_TABLE) {
 209                         DPRINTF("Ignore wrong route type: %d\n", blobmsg_type(cur));
 210                         continue;
 211                 }
 212
 213                 interface_ip_add_route(iface, cur, v6);
 214         }
 215 }
 216
 217 enum {
 218         NOTIFY_ACTION,
 219         NOTIFY_ERROR,
 220         NOTIFY_COMMAND,
 221         NOTIFY_ENV,
 222         NOTIFY_SIGNAL,
 223         NOTIFY_LINK_UP,
 224         NOTIFY_IFNAME,
 225         NOTIFY_ADDR_EXT,
 226         NOTIFY_IPADDR,
 227         NOTIFY_IP6ADDR,
 228         NOTIFY_ROUTES,
 229         NOTIFY_ROUTES6,
 230         NOTIFY_DNS,
 231         NOTIFY_DNS_SEARCH,
 232         __NOTIFY_LAST
 233 };
 234
 235 static const struct blobmsg_policy notify_attr[__NOTIFY_LAST] = {
 236         [NOTIFY_ACTION] = { .name = "action", .type = BLOBMSG_TYPE_INT32 },
 237         [NOTIFY_ERROR] = { .name = "error", .type = BLOBMSG_TYPE_ARRAY },
 238         [NOTIFY_COMMAND] = { .name = "command", .type = BLOBMSG_TYPE_ARRAY },
 239         [NOTIFY_ENV] = { .name = "env", .type = BLOBMSG_TYPE_ARRAY },
 240         [NOTIFY_SIGNAL] = { .name = "signal", .type = BLOBMSG_TYPE_INT32 },
 241         [NOTIFY_LINK_UP] = { .name = "link-up", .type = BLOBMSG_TYPE_BOOL },
 242         [NOTIFY_IFNAME] = { .name = "ifname", .type = BLOBMSG_TYPE_STRING },
 243         [NOTIFY_ADDR_EXT] = { .name = "address-external", .type = BLOBMSG_TYPE_BOOL },
 244         [NOTIFY_IPADDR] = { .name = "ipaddr", .type = BLOBMSG_TYPE_ARRAY },
 245         [NOTIFY_IP6ADDR] = { .name = "ip6addr", .type = BLOBMSG_TYPE_ARRAY },
 246         [NOTIFY_ROUTES] = { .name = "routes", .type = BLOBMSG_TYPE_ARRAY },
 247         [NOTIFY_ROUTES6] = { .name = "routes6", .type = BLOBMSG_TYPE_ARRAY },
 248         [NOTIFY_DNS] = { .name = "dns", .type = BLOBMSG_TYPE_ARRAY },
 249         [NOTIFY_DNS_SEARCH] = { .name = "dns_search", .type = BLOBMSG_TYPE_ARRAY },
 250 };
 251
 252 static int
 253 proto_shell_update_link(struct proto_shell_state *state, struct blob_attr **tb)
 254 {
 255         struct interface_ip_settings *ip;
 256         struct blob_attr *cur;
 257         int dev_create = 1;
 258         bool addr_ext = false;
 259         bool up;
 260
 261         if (!tb[NOTIFY_LINK_UP])
 262                 return UBUS_STATUS_INVALID_ARGUMENT;
 263
 264         up = blobmsg_get_bool(tb[NOTIFY_LINK_UP]);
 265         if (!up) {
 266                 state->proto.proto_event(&state->proto, IFPEV_LINK_LOST);
 267                 return 0;
 268         }
 269
 270         if ((cur = tb[NOTIFY_ADDR_EXT]) != NULL) {
 271                 addr_ext = blobmsg_get_bool(cur);
 272                 if (addr_ext)
 273                         dev_create = 2;
 274         }
 275
 276         if (!tb[NOTIFY_IFNAME]) {
 277                 if (!state->proto.iface->main_dev.dev)
 278                         return UBUS_STATUS_INVALID_ARGUMENT;
 279         } else {
 280                 if (state->l3_dev.dev)
 281                         device_remove_user(&state->l3_dev);
 282
 283                 device_add_user(&state->l3_dev,
 284                         device_get(blobmsg_data(tb[NOTIFY_IFNAME]), dev_create));
 285                 state->proto.iface->l3_dev = &state->l3_dev;
 286                 device_claim(&state->l3_dev);
 287         }
 288
 289         ip = &state->proto.iface->proto_ip;
 290         interface_update_start(state->proto.iface);
 291
 292         if ((cur = tb[NOTIFY_IPADDR]) != NULL)
 293                 proto_shell_parse_addr_list(ip, cur, false, addr_ext);
 294
 295         if ((cur = tb[NOTIFY_IP6ADDR]) != NULL)
 296                 proto_shell_parse_addr_list(ip, cur, true, addr_ext);
 297
 298         if ((cur = tb[NOTIFY_ROUTES]) != NULL)
 299                 proto_shell_parse_route_list(state->proto.iface, cur, false);
 300
 301         if ((cur = tb[NOTIFY_ROUTES6]) != NULL)
 302                 proto_shell_parse_route_list(state->proto.iface, cur, true);
 303
 304         if ((cur = tb[NOTIFY_DNS]) != NULL)
 305                 interface_add_dns_server_list(ip, cur);
 306
 307         if ((cur = tb[NOTIFY_DNS_SEARCH]) != NULL)
 308                 interface_add_dns_search_list(ip, cur);
 309
 310         interface_update_complete(state->proto.iface);
 311
 312         state->proto.proto_event(&state->proto, IFPEV_UP);
 313
 314         return 0;
 315 }
 316
 317 static bool
 318 fill_string_list(struct blob_attr *attr, char **argv, int max)
 319 {
 320         struct blob_attr *cur;
 321         int argc = 0;
 322         int rem;
 323
 324         if (!attr)
 325                 goto out;
 326
 327         blobmsg_for_each_attr(cur, attr, rem) {
 328                 if (blobmsg_type(cur) != BLOBMSG_TYPE_STRING)
 329                         return false;
 330
 331                 if (!blobmsg_check_attr(cur, NULL))
 332                         return false;
 333
 334                 argv[argc++] = blobmsg_data(cur);
 335                 if (argc == max - 1)
 336                         return false;
 337         }
 338
 339 out:
 340         argv[argc] = NULL;
 341         return true;
 342 }
 343
 344 static int
 345 proto_shell_run_command(struct proto_shell_state *state, struct blob_attr **tb)
 346 {
 347         static char *argv[64];
 348         static char *env[32];
 349
 350         if (!tb[NOTIFY_COMMAND])
 351                 goto error;
 352
 353         if (!fill_string_list(tb[NOTIFY_COMMAND], argv, ARRAY_SIZE(argv)))
 354                 goto error;
 355
 356         if (!fill_string_list(tb[NOTIFY_ENV], env, ARRAY_SIZE(env)))
 357                 goto error;
 358
 359         netifd_start_process((const char **) argv, (char **) env, &state->proto_task);
 360
 361         return 0;
 362
 363 error:
 364         return UBUS_STATUS_INVALID_ARGUMENT;
 365 }
 366
 367 static int
 368 proto_shell_kill_command(struct proto_shell_state *state, struct blob_attr **tb)
 369 {
 370         unsigned int signal = ~0;
 371
 372         if (tb[NOTIFY_SIGNAL])
 373                 signal = blobmsg_get_u32(tb[NOTIFY_SIGNAL]);
 374
 375         if (signal > 31)
 376                 signal = SIGTERM;
 377
 378         if (state->proto_task.uloop.pending) {
 379                 kill(state->proto_task.uloop.pid, signal);
 380                 state->teardown_wait_task = true;
 381         }
 382
 383         return 0;
 384 }
 385
 386 static int
 387 proto_shell_notify_error(struct proto_shell_state *state, struct blob_attr **tb)
 388 {
 389         struct blob_attr *cur;
 390         char *data[16];
 391         int n_data = 0;
 392         int rem;
 393
 394         if (!tb[NOTIFY_ERROR])
 395                 return UBUS_STATUS_INVALID_ARGUMENT;
 396
 397         blobmsg_for_each_attr(cur, tb[NOTIFY_ERROR], rem) {
 398                 if (n_data + 1 == ARRAY_SIZE(data))
 399                         goto error;
 400
 401                 if (blobmsg_type(cur) != BLOBMSG_TYPE_STRING)
 402                         goto error;
 403
 404                 if (!blobmsg_check_attr(cur, NULL))
 405                         goto error;
 406
 407                 data[n_data++] = blobmsg_data(cur);
 408         }
 409
 410         if (!n_data)
 411                 goto error;
 412
 413         interface_add_error(state->proto.iface, state->handler->proto.name,
 414                         data[0], (const char **) &data[1], n_data - 1);
 415
 416         return 0;
 417
 418 error:
 419         return UBUS_STATUS_INVALID_ARGUMENT;
 420 }
 421
 422 static int
 423 proto_shell_block_restart(struct proto_shell_state *state, struct blob_attr **tb)
 424 {
 425         state->proto.iface->autostart = false;
 426         return 0;
 427 }
 428
 429 static int
 430 proto_shell_notify(struct interface_proto_state *proto, struct blob_attr *attr)
 431 {
 432         struct proto_shell_state *state;
 433         struct blob_attr *tb[__NOTIFY_LAST];
 434
 435         state = container_of(proto, struct proto_shell_state, proto);
 436
 437         blobmsg_parse(notify_attr, __NOTIFY_LAST, tb, blob_data(attr), blob_len(attr));
 438         if (!tb[NOTIFY_ACTION])
 439                 return UBUS_STATUS_INVALID_ARGUMENT;
 440
 441         switch(blobmsg_get_u32(tb[NOTIFY_ACTION])) {
 442         case 0:
 443                 return proto_shell_update_link(state, tb);
 444         case 1:
 445                 return proto_shell_run_command(state, tb);
 446         case 2:
 447                 return proto_shell_kill_command(state, tb);
 448         case 3:
 449                 return proto_shell_notify_error(state, tb);
 450         case 4:
 451                 return proto_shell_block_restart(state, tb);
 452         default:
 453                 return UBUS_STATUS_INVALID_ARGUMENT;
 454         }
 455 }
 456
 457 static struct interface_proto_state *
 458 proto_shell_attach(const struct proto_handler *h, struct interface *iface,
 459                    struct blob_attr *attr)
 460 {
 461         struct proto_shell_state *state;
 462
 463         state = calloc(1, sizeof(*state));
 464         state->config = malloc(blob_pad_len(attr));
 465         if (!state->config)
 466                 goto error;
 467
 468         memcpy(state->config, attr, blob_pad_len(attr));
 469         state->proto.free = proto_shell_free;
 470         state->proto.notify = proto_shell_notify;
 471         state->proto.cb = proto_shell_handler;
 472         state->setup_timeout.cb = proto_shell_setup_timeout_cb;
 473         state->setup_task.cb = proto_shell_setup_cb;
 474         state->setup_task.dir_fd = proto_fd.fd;
 475         state->setup_task.log_prefix = iface->name;
 476         state->teardown_task.cb = proto_shell_teardown_cb;
 477         state->teardown_task.dir_fd = proto_fd.fd;
 478         state->teardown_task.log_prefix = iface->name;
 479         state->proto_task.cb = proto_shell_task_cb;
 480         state->proto_task.dir_fd = proto_fd.fd;
 481         state->proto_task.log_prefix = iface->name;
 482         state->handler = container_of(h, struct proto_shell_handler, proto);
 483
 484         return &state->proto;
 485
 486 error:
 487         free(state);
 488         return NULL;
 489 }
 490
 491 static json_object *
 492 check_type(json_object *obj, json_type type)
 493 {
 494         if (!obj)
 495                 return NULL;
 496
 497         if (json_object_get_type(obj) != type)
 498                 return NULL;
 499
 500         return obj;
 501 }
 502
 503 static inline json_object *
 504 get_field(json_object *obj, const char *name, json_type type)
 505 {
 506         return check_type(json_object_object_get(obj, name), type);
 507 }
 508
 509 static char *
 510 proto_shell_parse_config(struct config_param_list *config, json_object *obj)
 511 {
 512         struct blobmsg_policy *attrs;
 513         char *str_buf, *str_cur;
 514         int str_len = 0;
 515         int i;
 516
 517         config->n_params = json_object_array_length(obj);
 518         attrs = calloc(1, sizeof(*attrs) * config->n_params);
 519         if (!attrs)
 520                 return NULL;
 521
 522         config->params = attrs;
 523         for (i = 0; i < config->n_params; i++) {
 524                 json_object *cur, *name, *type;
 525
 526                 cur = check_type(json_object_array_get_idx(obj, i), json_type_array);
 527                 if (!cur)
 528                         goto error;
 529
 530                 name = check_type(json_object_array_get_idx(cur, 0), json_type_string);
 531                 if (!name)
 532                         goto error;
 533
 534                 type = check_type(json_object_array_get_idx(cur, 1), json_type_int);
 535                 if (!type)
 536                         goto error;
 537
 538                 attrs[i].name = json_object_get_string(name);
 539                 attrs[i].type = json_object_get_int(type);
 540                 if (attrs[i].type > BLOBMSG_TYPE_LAST)
 541                         goto error;
 542
 543                 str_len += strlen(attrs[i].name) + 1;
 544         }
 545
 546         str_buf = malloc(str_len);
 547         if (!str_buf)
 548                 goto error;
 549
 550         str_cur = str_buf;
 551         for (i = 0; i < config->n_params; i++) {
 552                 const char *name = attrs[i].name;
 553
 554                 attrs[i].name = str_cur;
 555                 str_cur += sprintf(str_cur, "%s", name) + 1;
 556         }
 557
 558         return str_buf;
 559
 560 error:
 561         free(attrs);
 562         config->n_params = 0;
 563         return NULL;
 564 }
 565
 566 static void
 567 proto_shell_add_handler(const char *script, json_object *obj)
 568 {
 569         struct proto_shell_handler *handler;
 570         struct proto_handler *proto;
 571         json_object *config, *tmp;
 572         const char *name;
 573         char *str;
 574
 575         if (!check_type(obj, json_type_object))
 576                 return;
 577
 578         tmp = get_field(obj, "name", json_type_string);
 579         if (!tmp)
 580                 return;
 581
 582         name = json_object_get_string(tmp);
 583
 584         handler = calloc(1, sizeof(*handler) +
 585                          strlen(script) + 1 +
 586                          strlen(name) + 1);
 587         if (!handler)
 588                 return;
 589
 590         strcpy(handler->script_name, script);
 591
 592         str = handler->script_name + strlen(handler->script_name) + 1;
 593         strcpy(str, name);
 594
 595         proto = &handler->proto;
 596         proto->name = str;
 597         proto->config_params = &handler->config;
 598         proto->attach = proto_shell_attach;
 599
 600         tmp = get_field(obj, "no-device", json_type_boolean);
 601         if (tmp && json_object_get_boolean(tmp))
 602                 handler->proto.flags |= PROTO_FLAG_NODEV;
 603
 604         config = get_field(obj, "config", json_type_array);
 605         if (config)
 606                 handler->config_buf = proto_shell_parse_config(&handler->config, config);
 607
 608         DPRINTF("Add handler for script %s: %s\n", script, proto->name);
 609         add_proto_handler(proto);
 610 }
 611
 612 static void proto_shell_add_script(const char *name)
 613 {
 614         struct json_tokener *tok = NULL;
 615         json_object *obj;
 616         static char buf[512];
 617         char *start, *cmd;
 618         FILE *f;
 619         int len;
 620
 621 #define DUMP_SUFFIX     " '' dump"
 622
 623         cmd = alloca(strlen(name) + 1 + sizeof(DUMP_SUFFIX));
 624         sprintf(cmd, "%s" DUMP_SUFFIX, name);
 625
 626         f = popen(cmd, "r");
 627         if (!f)
 628                 return;
 629
 630         do {
 631                 start = fgets(buf, sizeof(buf), f);
 632                 if (!start)
 633                         continue;
 634
 635                 len = strlen(start);
 636
 637                 if (!tok)
 638                         tok = json_tokener_new();
 639
 640                 obj = json_tokener_parse_ex(tok, start, len);
 641                 if (!is_error(obj)) {
 642                         proto_shell_add_handler(name, obj);
 643                         json_object_put(obj);
 644                         json_tokener_free(tok);
 645                         tok = NULL;
 646                 } else if (start[len - 1] == '\n') {
 647                         json_tokener_free(tok);
 648                         tok = NULL;
 649                 }
 650         } while (!feof(f) && !ferror(f));
 651
 652         if (tok)
 653                 json_tokener_free(tok);
 654
 655         pclose(f);
 656 }
 657
 658 static void __init proto_shell_init(void)
 659 {
 660         glob_t g;
 661         int main_fd;
 662         int i;
 663
 664         main_fd = open(".", O_RDONLY | O_DIRECTORY);
 665         if (main_fd < 0)
 666                 return;
 667
 668         if (chdir(main_path)) {
 669                 perror("chdir(main path)");
 670                 goto close_cur;
 671         }
 672
 673         if (chdir("./proto"))
 674                 goto close_cur;
 675
 676         proto_fd.fd = open(".", O_RDONLY | O_DIRECTORY);
 677         if (proto_fd.fd < 0)
 678                 goto close_cur;
 679
 680         netifd_fd_add(&proto_fd);
 681         glob("./*.sh", 0, NULL, &g);
 682         for (i = 0; i < g.gl_pathc; i++)
 683                 proto_shell_add_script(g.gl_pathv[i]);
 684
 685 close_cur:
 686         fchdir(main_fd);
 687         close(main_fd);
 688 }