X-Git-Url: https://git.archive.openwrt.org/?p=project%2Fnetifd.git;a=blobdiff_plain;f=proto-shell.c;h=281debc056146d13085376e9b095a084e5589ec0;hp=2b68c354d00483e6d13f072df12b1c2e0ddf858a;hb=e32103ad7e944ccbebef4335333481e565311f5c;hpb=23cfcee20f43d824d61fadef48af84a83d5b8816 diff --git a/proto-shell.c b/proto-shell.c index 2b68c35..281debc 100644 --- a/proto-shell.c +++ b/proto-shell.c @@ -13,35 +13,244 @@ #include "proto.h" static LIST_HEAD(handlers); -static int proto_fd, main_fd; +static int proto_fd; struct proto_shell_handler { struct list_head list; struct proto_handler proto; + struct config_param_list config; + char *config_buf; + char script_name[]; }; -#define DUMP_PREFIX "./" -#define DUMP_SUFFIX " dump" +struct proto_shell_state { + struct interface_proto_state proto; + struct proto_shell_handler *handler; + struct blob_attr *config; +}; + +static int run_script(const char **argv) +{ + int pid, ret; + + if ((pid = fork()) < 0) + return -1; + + if (!pid) { + fchdir(proto_fd); + execvp(argv[0], (char **) argv); + exit(127); + } + + if (waitpid(pid, &ret, 0) == -1) + ret = -1; + + if (ret > 0) + return -ret; + + return 0; +} -static void proto_shell_add_handler(const char *script, struct json_object *obj) +static int +proto_shell_handler(struct interface_proto_state *proto, + enum interface_proto_cmd cmd, bool force) { - if (json_object_get_type(obj) != json_type_object) + struct proto_shell_state *state; + struct proto_shell_handler *handler; + const char *argv[5]; + char *config; + int ret; + + state = container_of(proto, struct proto_shell_state, proto); + handler = state->handler; + + config = blobmsg_format_json(state->config, true); + if (!config) + return -1; + + argv[0] = handler->script_name; + argv[1] = handler->proto.name; + argv[2] = "teardown"; + argv[3] = config; + argv[4] = NULL; + + switch(cmd) { + case PROTO_CMD_SETUP: + argv[2] = "setup"; + /* fall through */ + case PROTO_CMD_TEARDOWN: + ret = run_script(argv); + break; + } + + free(config); + + return ret; +} + +static void +proto_shell_free(struct interface_proto_state *proto) +{ + struct proto_shell_state *state; + + state = container_of(proto, struct proto_shell_state, proto); + free(state->config); + free(state); +} + +struct interface_proto_state * +proto_shell_attach(const struct proto_handler *h, struct interface *iface, + struct blob_attr *attr) +{ + struct proto_shell_state *state; + + state = calloc(1, sizeof(*state)); + state->config = malloc(blob_pad_len(attr)); + if (!state->config) + goto error; + + memcpy(state->config, attr, blob_pad_len(attr)); + state->proto.free = proto_shell_free; + state->proto.handler = proto_shell_handler; + state->handler = container_of(h, struct proto_shell_handler, proto); + + return &state->proto; + +error: + free(state); + return NULL; +} + +static json_object * +check_type(json_object *obj, json_type type) +{ + if (!obj) + return NULL; + + if (json_object_get_type(obj) != type) + return NULL; + + return obj; +} + +static inline json_object * +get_field(json_object *obj, const char *name, json_type type) +{ + return check_type(json_object_object_get(obj, name), type); +} + +static char * +proto_shell_parse_config(struct config_param_list *config, json_object *obj) +{ + struct blobmsg_policy *attrs; + char *str_buf, *str_cur; + int str_len = 0; + int i; + + attrs = calloc(1, sizeof(*attrs)); + if (!attrs) + return NULL; + + config->n_params = json_object_array_length(obj); + config->params = attrs; + for (i = 0; i < config->n_params; i++) { + json_object *cur, *name, *type; + + cur = check_type(json_object_array_get_idx(obj, i), json_type_array); + if (!cur) + goto error; + + name = check_type(json_object_array_get_idx(cur, 0), json_type_string); + if (!name) + goto error; + + type = check_type(json_object_array_get_idx(cur, 1), json_type_int); + if (!type) + goto error; + + attrs[i].name = json_object_get_string(name); + attrs[i].type = json_object_get_int(type); + if (attrs[i].type > BLOBMSG_TYPE_LAST) + goto error; + + str_len += strlen(attrs[i].name + 1); + } + + str_buf = malloc(str_len); + if (!str_buf) + goto error; + + str_cur = str_buf; + for (i = 0; i < config->n_params; i++) { + const char *name = attrs[i].name; + + attrs[i].name = str_cur; + str_cur += sprintf(str_cur, "%s", name) + 1; + } + + return str_buf; + +error: + free(attrs); + config->n_params = 0; + return NULL; +} + +static void +proto_shell_add_handler(const char *script, json_object *obj) +{ + struct proto_shell_handler *handler; + struct proto_handler *proto; + json_object *config, *tmp; + const char *name; + char *str; + + if (!check_type(obj, json_type_object)) + return; + + tmp = get_field(obj, "name", json_type_string); + if (!tmp) return; - fprintf(stderr, "Add handler for script %s: %s\n", script, json_object_to_json_string(obj)); + name = json_object_get_string(tmp); + + handler = calloc(1, sizeof(*handler) + + strlen(script) + 1 + + strlen(name) + 1); + if (!handler) + return; + + strcpy(handler->script_name, script); + + str = handler->script_name + strlen(handler->script_name) + 1; + strcpy(str, name); + + proto = &handler->proto; + proto->name = str; + proto->config_params = &handler->config; + proto->attach = proto_shell_attach; + + config = get_field(obj, "config", json_type_array); + if (config) + handler->config_buf = proto_shell_parse_config(&handler->config, config); + + DPRINTF("Add handler for script %s: %s\n", script, proto->name); + add_proto_handler(proto); } static void proto_shell_add_script(const char *name) { struct json_tokener *tok = NULL; - struct json_object *obj; + json_object *obj; static char buf[512]; char *start, *end, *cmd; FILE *f; int buflen, len; - cmd = alloca(strlen(name) + 1 + sizeof(DUMP_PREFIX) + sizeof(DUMP_SUFFIX)); - sprintf(cmd, DUMP_PREFIX "%s" DUMP_SUFFIX, name); +#define DUMP_SUFFIX " '' dump" + + cmd = alloca(strlen(name) + 1 + sizeof(DUMP_SUFFIX)); + sprintf(cmd, "%s" DUMP_SUFFIX, name); f = popen(cmd, "r"); if (!f) @@ -86,6 +295,7 @@ static void proto_shell_add_script(const char *name) void __init proto_shell_init(void) { glob_t g; + int main_fd; int i; main_fd = open(".", O_RDONLY | O_DIRECTORY); @@ -104,15 +314,11 @@ void __init proto_shell_init(void) if (proto_fd < 0) goto close_cur; - glob("*.sh", 0, NULL, &g); + glob("./*.sh", 0, NULL, &g); for (i = 0; i < g.gl_pathc; i++) proto_shell_add_script(g.gl_pathv[i]); - if (list_empty(&handlers)) - close(proto_fd); - close_cur: fchdir(main_fd); - if (list_empty(&handlers)) - close(main_fd); + close(main_fd); }