X-Git-Url: http://git.ozlabs.org/?p=petitboot;a=blobdiff_plain;f=discover%2Fparser.c;h=5598f963e236d2c6806aa0ac9166b242c18d266a;hp=e2940e19e1b939a3aaaaf2dca45b28d8a02e418b;hb=03fe06fa71f065e918c46a65e6a45e76063a07eb;hpb=9974f2d82b9450eaccd7661b3bfabb686ab8e161 diff --git a/discover/parser.c b/discover/parser.c index e2940e1..5598f96 100644 --- a/discover/parser.c +++ b/discover/parser.c @@ -1,10 +1,9 @@ #include #include -#include -#include #include "types/types.h" +#include #include #include @@ -20,167 +19,148 @@ struct p_item { STATIC_LIST(parsers); -static const int max_file_size = 1024 * 1024; +static char *local_path(struct discover_context *ctx, + struct discover_device *dev, + const char *filename) +{ + return join_paths(ctx, dev->root_path, filename); +} + +int parser_request_file(struct discover_context *ctx, + struct discover_device *dev, const char *filename, + char **buf, int *len) -static int read_file(struct discover_context *ctx, - const char *filename, char **bufp, int *lenp) { - struct stat statbuf; - int rc, fd, i, len; - char *buf; + char *path; + int rc; - fd = open(filename, O_RDONLY); - if (fd < 0) + /* we only support local files at present */ + if (!dev->mount_path) return -1; - rc = fstat(fd, &statbuf); - if (rc < 0) - goto err_close; + path = local_path(ctx, dev, filename); - len = statbuf.st_size; - if (len > max_file_size) - goto err_close; + rc = read_file(ctx, path, buf, len); - buf = talloc_array(ctx, char, len + 1); - if (!buf) - goto err_close; + talloc_free(path); - for (i = 0; i < len; i += rc) { - rc = read(fd, buf + i, len - i); + return rc; +} - /* unexpected EOF: trim and return */ - if (rc == 0) { - len = i; - break; - } +int parser_stat_path(struct discover_context *ctx, + struct discover_device *dev, const char *path, + struct stat *statbuf) +{ + int rc = -1; + char *full_path; - if (rc < 0) - goto err_free; + /* we only support local files at present */ + if (!dev->mount_path) + return -1; - } + full_path = local_path(ctx, dev, path); - buf[len] = '\0'; + rc = stat(full_path, statbuf); + if (rc) { + rc = -1; + goto out; + } - close(fd); - *bufp = buf; - *lenp = len; - return 0; + rc = 0; +out: + talloc_free(full_path); -err_free: - talloc_free(buf); -err_close: - close(fd); - return -1; + return rc; } -static char *local_path(struct discover_context *ctx, - const char *filename) +int parser_replace_file(struct discover_context *ctx, + struct discover_device *dev, const char *filename, + char *buf, int len) { - return join_paths(ctx, ctx->device->mount_path, filename); + bool release; + char *path; + int rc; + + if (!dev->mounted) + return -1; + + rc = device_request_write(dev, &release); + if (rc) { + pb_log("Can't write file %s: device doesn't allow write\n", + dev->device_path); + return -1; + } + + path = local_path(ctx, dev, filename); + + rc = replace_file(path, buf, len); + + talloc_free(path); + + device_release_write(dev, release); + + return rc; } -static int download_config(struct discover_context *ctx, char **buf, int *len) +int parser_request_url(struct discover_context *ctx, struct pb_url *url, + char **buf, int *len) { - unsigned tempfile; - const char *file; + struct load_url_result *result; int rc; - file = load_url(ctx, ctx->conf_url, &tempfile); - if (!file) - return -1; + result = load_url(ctx, url); + if (!result) + goto out; - rc = read_file(ctx, file, buf, len); - if (rc) + rc = read_file(ctx, result->local, buf, len); + if (rc) { + pb_log("Read failed for the parser %s on file %s\n", + ctx->parser->name, result->local); goto out_clean; + } return 0; out_clean: - if (tempfile) - unlink(file); + if (result->cleanup_local) + unlink(result->local); +out: return -1; } -static void iterate_parser_files(struct discover_context *ctx, - const struct parser *parser) -{ - const char * const *filename; - const char *path; - - if (!parser->filenames) - return; - - for (filename = parser->filenames; *filename; filename++) { - int rc, len; - char *buf; - - path = local_path(ctx, *filename); - if (!path) - continue; - - rc = read_file(ctx, path, &buf, &len); - if (!rc) { - pb_log("Running parser %s on file %s\n", - parser->name, *filename); - parser->parse(ctx, buf, len); - talloc_free(buf); - } - } -} - -void iterate_parsers(struct discover_context *ctx, enum conf_method method) +void iterate_parsers(struct discover_context *ctx) { struct p_item* i; - int rc, len; - char *buf; pb_log("trying parsers for %s\n", ctx->device->device->id); - switch (method) { - case CONF_METHOD_LOCAL_FILE: - list_for_each_entry(&parsers, i, list) { - if (i->parser->method != CONF_METHOD_LOCAL_FILE) - continue; - - pb_log("\ttrying parser '%s'\n", i->parser->name); - ctx->parser = i->parser; - iterate_parser_files(ctx, ctx->parser); - } - ctx->parser = NULL; - break; - - case CONF_METHOD_DHCP: - rc = download_config(ctx, &buf, &len); - if (rc) { - pb_log("\tdownload failed, aborting\n"); - return; - } - - list_for_each_entry(&parsers, i, list) { - if (i->parser->method != method) - continue; - - pb_log("\ttrying parser '%s'\n", i->parser->name); - ctx->parser = i->parser; - i->parser->parse(ctx, buf, len); - } - - break; - - case CONF_METHOD_UNKNOWN: - break; - + list_for_each_entry(&parsers, i, list) { + pb_debug("\ttrying parser '%s'\n", i->parser->name); + ctx->parser = i->parser; + i->parser->parse(ctx); } + ctx->parser = NULL; } +static void *parsers_ctx; + void __register_parser(struct parser *parser) { - struct p_item* i = talloc(NULL, struct p_item); + struct p_item *i; + + if (!parsers_ctx) + parsers_ctx = talloc_new(NULL); + i = talloc(parsers_ctx, struct p_item); i->parser = parser; list_add(&parsers, &i->list); } +static __attribute__((destructor)) void cleanup_parsers(void) +{ + talloc_free(parsers_ctx); +} + void parser_init(void) { }