+ file->data = talloc_memdup(test, buf, len);
+ file->size = len;
+ return 0;
+}
+
+struct load_url_result *load_url_async(void *ctx, struct pb_url *url,
+ load_url_complete async_cb, void *async_data,
+ waiter_cb stdout_cb, void *stdout_data)
+{
+ struct conf_context *conf = async_data;
+ struct parser_test *test = conf->dc->test_data;
+ struct load_url_result *result;
+ char tmp[] = "/tmp/pb-XXXXXX";
+ ssize_t rc = -1, sz = 0;
+ struct test_file *file;
+ int fd;
+
+ /* Ignore the stdout callback for tests */
+ (void)stdout_cb;
+ (void)stdout_data;
+
+ fd = mkstemp(tmp);
+
+ if (fd < 0)
+ return NULL;
+
+ /* Some parsers will expect to need to read a file, so write the
+ * specified file to a temporary file */
+ list_for_each_entry(&test->files, file, list) {
+ if (file->dev)
+ continue;
+
+ if (strcmp(file->name, url->full))
+ continue;
+
+ while (sz < file->size) {
+ rc = write(fd, file->data, file->size);
+ if (rc < 0) {
+ fprintf(stderr,
+ "Failed to write to tmpfile, %m\n");
+ break;
+ }
+ sz += rc;
+ }
+ break;
+ }
+
+ close(fd);
+
+ result = talloc_zero(ctx, struct load_url_result);
+ if (!result)
+ return NULL;
+
+ result->local = talloc_strdup(result, tmp);
+ result->url = url;
+ if (rc < 0)
+ result->status = LOAD_ERROR;
+ else
+ result->status = result->local ? LOAD_OK : LOAD_ERROR;
+ result->cleanup_local = true;
+
+ async_cb(result, conf);
+
+ return result;
+}
+
+int parser_request_url(struct discover_context *ctx, struct pb_url *url,
+ char **buf, int *len)
+{
+ struct parser_test *test = ctx->test_data;
+ struct test_file *file;
+ char *tmp;
+
+ list_for_each_entry(&test->files, file, list) {
+ if (file->dev)
+ continue;
+
+ if (strcmp(file->name, url->full))
+ continue;
+
+ /* the read_file() interface always adds a trailing null
+ * for string-safety; do the same here */
+ tmp = talloc_array(test, char, file->size + 1);
+ memcpy(tmp, file->data, file->size);
+ tmp[file->size] = '\0';
+ *buf = tmp;
+ *len = file->size;
+ return 0;
+ }
+
+ return -1;
+}
+
+int test_run_parser(struct parser_test *test, const char *parser_name)
+{
+ struct p_item* i;
+
+ list_for_each_entry(&parsers, i, list) {
+ if (strcmp(i->parser->name, parser_name))
+ continue;
+ test->ctx->parser = i->parser;
+ return i->parser->parse(test->ctx);
+ }
+
+ errx(EXIT_FAILURE, "%s: parser '%s' not found", __func__, parser_name);
+}
+
+bool resource_resolve(struct device_handler *handler, struct parser *parser,
+ struct resource *resource)
+{
+ if (!resource)
+ return true;
+ if (resource->resolved)
+ return true;
+
+ assert(parser);
+ assert(parser->resolve_resource);
+
+ return parser->resolve_resource(handler, resource);
+}
+
+void boot_option_resolve(struct device_handler *handler,
+ struct discover_boot_option *opt)
+{
+ resource_resolve(handler, opt->source, opt->boot_image);
+ resource_resolve(handler, opt->source, opt->initrd);
+ resource_resolve(handler, opt->source, opt->icon);
+}
+
+void test_hotplug_device(struct parser_test *test, struct discover_device *dev)
+{
+ struct discover_boot_option *opt;
+
+ device_handler_add_device(test->handler, dev);
+
+ list_for_each_entry(&test->ctx->boot_options, opt, list)
+ boot_option_resolve(test->handler, opt);
+}
+
+void test_remove_device(struct parser_test *test, struct discover_device *dev)
+{
+ struct discover_boot_option *opt, *tmp;
+
+ if (dev == test->ctx->device) {
+ list_for_each_entry_safe(&test->ctx->boot_options,
+ opt, tmp, list) {
+ list_remove(&opt->list);
+ talloc_free(opt);
+ }
+ }
+
+ device_handler_remove(test->handler, dev);