X-Git-Url: https://git.ozlabs.org/?p=petitboot;a=blobdiff_plain;f=lib%2Fpb-config%2Fstorage-test.c;h=48cb238bd383bcaabb1efc2e571d3033d1c749d4;hp=e4a0e5acce3c041534138ad06326fa9fe909507e;hb=52bec18801b226b57f53d9f075addd2c07caa3db;hpb=dd8fe077022138d1f6aca7e6d6cceec61f56ce95

diff --git a/lib/pb-config/storage-test.c b/lib/pb-config/storage-test.c
index e4a0e5a..48cb238 100644
--- a/lib/pb-config/storage-test.c
+++ b/lib/pb-config/storage-test.c
@@ -40,22 +40,34 @@ struct test_storage {
 	struct config_storage	storage;
 	struct param		*params;
 	int			n_params;
+	struct config		*config;
 };
 
-static int load(struct config_storage *st __attribute__((unused)),
-		struct config *config)
+#define to_test_storage(st) container_of(st, struct test_storage, storage)
+
+static int load(struct config_storage *st, struct config *config)
+{
+	struct test_storage *ts = to_test_storage(st);
+	memcpy(config, ts->config, sizeof(test_config));
+	return 0;
+}
+
+static int save(struct config_storage *st, struct config *newconfig)
 {
-	memcpy(config, &test_config, sizeof(test_config));
+	struct test_storage *ts = to_test_storage(st);
+	ts->config = newconfig;
 	return 0;
 }
 
 static struct test_storage st = {
 	.storage = {
 		.load  = load,
+		.save = save,
 	},
 };
 
 struct config_storage *create_test_storage(void *ctx __attribute__((unused)))
 {
+	st.config = &test_config;
 	return &st.storage;
 }