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