X-Git-Url: http://git.ozlabs.org/?p=petitboot;a=blobdiff_plain;f=discover%2Fsysinfo.c;h=3c132774e21b3599e0c61e45b6ceaba5ebb847c3;hp=f75819bd96ac02adc563b30afd046b457d97bd67;hb=8b2ccd525b328994d3e62c01092b20859a23e36d;hpb=8ec7ea724de780a6a428a8485b2760bf1d9e0a4e

diff --git a/discover/sysinfo.c b/discover/sysinfo.c
index f75819b..3c13277 100644
--- a/discover/sysinfo.c
+++ b/discover/sysinfo.c
@@ -53,6 +53,40 @@ void system_info_register_interface(unsigned int hwaddr_size, uint8_t *hwaddr,
 	discover_server_notify_system_info(server, sysinfo);
 }
 
+void system_info_register_blockdev(const char *name, const char *uuid,
+		const char *mountpoint)
+{
+	struct blockdev_info *bd_info;
+	unsigned int i;
+
+	for (i = 0; i < sysinfo->n_blockdevs; i++) {
+		bd_info = sysinfo->blockdevs[i];
+
+		if (strcmp(bd_info->name, name))
+			continue;
+
+		/* update the mountpoint and UUID, and we're done */
+		talloc_free(bd_info->mountpoint);
+		bd_info->uuid = talloc_strdup(bd_info, uuid);
+		bd_info->mountpoint = talloc_strdup(bd_info, mountpoint);
+		discover_server_notify_system_info(server, sysinfo);
+		return;
+	}
+
+	bd_info = talloc_zero(sysinfo, struct blockdev_info);
+	bd_info->name = talloc_strdup(bd_info, name);
+	bd_info->uuid = talloc_strdup(bd_info, uuid);
+	bd_info->mountpoint = talloc_strdup(bd_info, mountpoint);
+
+	sysinfo->n_blockdevs++;
+	sysinfo->blockdevs = talloc_realloc(sysinfo, sysinfo->blockdevs,
+						struct blockdev_info *,
+						sysinfo->n_blockdevs);
+	sysinfo->blockdevs[sysinfo->n_blockdevs - 1] = bd_info;
+
+	discover_server_notify_system_info(server, sysinfo);
+}
+
 static void system_info_set_identifier(struct system_info *info)
 {
 	struct process *process;
@@ -87,7 +121,7 @@ static void system_info_set_identifier(struct system_info *info)
 
 void system_info_init(struct discover_server *s)
 {
-	sysinfo = talloc_zero(server, struct system_info);
 	server = s;
+	sysinfo = talloc_zero(server, struct system_info);
 	system_info_set_identifier(sysinfo);
 }