X-Git-Url: http://git.ozlabs.org/?p=petitboot;a=blobdiff_plain;f=lib%2Fpb-config%2Fstorage-test.c;h=48cb238bd383bcaabb1efc2e571d3033d1c749d4;hp=bdb1d0dec2153c46d7c5a49f25dcdef13a7a6ad0;hb=dea7842fb28ff055b4e0f43a6a1fdaf3c4b5ba89;hpb=2b41985dde73e71daad90c36fa5d06c199da3ae1

diff --git a/lib/pb-config/storage-test.c b/lib/pb-config/storage-test.c
index bdb1d0d..48cb238 100644
--- a/lib/pb-config/storage-test.c
+++ b/lib/pb-config/storage-test.c
@@ -4,6 +4,8 @@
 #include <sys/types.h>
 #include <sys/wait.h>
 
+#include <util/util.h>
+
 #include "pb-config.h"
 #include "storage.h"
 
@@ -22,14 +24,15 @@ struct interface_config net2 = {
 };
 
 struct interface_config *interface_configs[] = { &net1, &net2 };
-
-#define ARRAY_SIZE(a) (sizeof(a) / sizeof(a[0]))
+const char *dns_servers[] = { "192.168.1.1", "192.168.1.2" };
 
 struct config test_config = {
 	.autoboot_enabled = true,
 	.network = {
 		.interfaces = interface_configs,
 		.n_interfaces = ARRAY_SIZE(interface_configs),
+		.dns_servers = dns_servers,
+		.n_dns_servers = ARRAY_SIZE(dns_servers),
 	}
 };
 
@@ -37,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;
 }