X-Git-Url: http://git.ozlabs.org/?p=petitboot;a=blobdiff_plain;f=discover%2Fparser.c;h=54787201553010d808297eaf8cc8c1730540d0db;hp=a304f0c989e32f4e124d1f62647547b067d952b5;hb=c3dfc9b7b91045172d460651c0309b09b0ab121a;hpb=227620704107a19c03824ae146249fff4a939839 diff --git a/discover/parser.c b/discover/parser.c index a304f0c..5478720 100644 --- a/discover/parser.c +++ b/discover/parser.c @@ -1,10 +1,9 @@ #include #include -#include -#include #include "types/types.h" +#include #include #include @@ -12,7 +11,6 @@ #include "parser.h" #include "parser-utils.h" #include "paths.h" -#include "file.h" struct p_item { struct list_item list; @@ -22,114 +20,193 @@ struct p_item { STATIC_LIST(parsers); static char *local_path(struct discover_context *ctx, + struct discover_device *dev, const char *filename) { - return join_paths(ctx, ctx->device->mount_path, filename); + return join_paths(ctx, dev->root_path, filename); } -static int download_config(struct discover_context *ctx, char **buf, int *len) +int parser_request_file(struct discover_context *ctx, + struct discover_device *dev, const char *filename, + char **buf, int *len) + +{ + char *path; + int rc; + + /* we only support local files at present */ + if (!dev->mount_path) + return -1; + + path = local_path(ctx, dev, filename); + + rc = read_file(ctx, path, buf, len); + + talloc_free(path); + + return rc; +} + +int parser_stat_path(struct discover_context *ctx, + struct discover_device *dev, const char *path, + struct stat *statbuf) +{ + int rc = -1; + char *full_path; + + /* we only support local files at present */ + if (!dev->mount_path) + return -1; + + full_path = local_path(ctx, dev, path); + + rc = stat(full_path, statbuf); + if (rc) { + rc = -1; + goto out; + } + + rc = 0; +out: + talloc_free(full_path); + + return rc; +} + +bool parser_is_unique(struct discover_context *ctx, struct discover_device *dev, + const char *filename, struct list *found_list) +{ + struct stat stat; + struct parser_found_file *found_file; + const struct parser_found_file *entry; + + if (parser_stat_path(ctx, dev, filename, &stat)) { + pb_debug("%s: Not found: '%s'\n", __func__, filename); + return false; + } + + list_for_each_entry(found_list, entry, list) { + if (entry->ino == stat.st_ino) { + pb_log("%s: Duplicate: '%s' = '%s'\n", + __func__, filename, entry->filename); + return false; + } + } + + pb_debug("%s: Found: '%s'\n", __func__, filename); + + found_file = talloc_zero(found_list, struct parser_found_file); + found_file->filename = talloc_strdup(found_file, filename); + found_file->ino = stat.st_ino; + list_add(found_list, &found_file->list); + + return true; +} + +int parser_replace_file(struct discover_context *ctx, + struct discover_device *dev, const char *filename, + char *buf, int len) +{ + 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; +} + +int parser_request_url(struct discover_context *ctx, struct pb_url *url, + char **buf, int *len) { struct load_url_result *result; int rc; - result = load_url(ctx, ctx->conf_url); + result = load_url(ctx, url); if (!result) - return -1; + goto out; rc = read_file(ctx, result->local, buf, len); - if (rc) + 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 (result->cleanup_local) unlink(result->local); +out: return -1; } -static void iterate_parser_files(struct discover_context *ctx, - const struct parser *parser) +int parser_scandir(struct discover_context *ctx, const char *dirname, + struct dirent ***files, int (*filter)(const struct dirent *), + int (*comp)(const struct dirent **, const struct dirent **)) { - 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); - } - } + char *path; + int n; + + path = talloc_asprintf(ctx, "%s%s", ctx->device->mount_path, dirname); + if (!path) + return -1; + + n = scandir(path, files, filter, comp); + talloc_free(path); + return n; } -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; - printf("%s: %s\n", __func__, parser->name); + 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) { }