X-Git-Url: http://git.ozlabs.org/?p=petitboot;a=blobdiff_plain;f=discover%2Fsysinfo.c;h=a9d7ff547b11ffb22a40bbdf4e738448de4f6a16;hp=219369a0d5456f5bda27fed2ba7a534a4db7b64b;hb=c3dfc9b7b91045172d460651c0309b09b0ab121a;hpb=6897abaa97a02e0ab8ac07209a5e4966bfe101c5

diff --git a/discover/sysinfo.c b/discover/sysinfo.c
index 219369a..a9d7ff5 100644
--- a/discover/sysinfo.c
+++ b/discover/sysinfo.c
@@ -3,6 +3,8 @@
 
 #include <talloc/talloc.h>
 #include <process/process.h>
+#include <log/log.h>
+#include <url/url.h>
 
 #include "discover-server.h"
 #include "platform.h"
@@ -16,6 +18,49 @@ const struct system_info *system_info_get(void)
 	return sysinfo;
 }
 
+void system_info_set_interface_address(unsigned int hwaddr_size,
+		uint8_t *hwaddr, const char *address)
+{
+	struct interface_info *if_info;
+	unsigned int i;
+	char mac[20], **if_addr, *new_addr, *subnet;
+
+	for (i = 0; i < sysinfo->n_interfaces; i++) {
+		if_info = sysinfo->interfaces[i];
+
+		if (if_info->hwaddr_size != hwaddr_size)
+			continue;
+
+		if (memcmp(if_info->hwaddr, hwaddr, hwaddr_size))
+			continue;
+
+		/*
+		 * Don't include the subnet from a static config, and check if
+		 * we're updating the IPv4 or IPv6 address.
+		 * */
+		if ((subnet = strchr(address, '/')))
+			new_addr = talloc_strndup(if_info, address, subnet - address);
+		else
+			new_addr = talloc_strdup(if_info, address);
+		if (addr_scheme(new_addr) == AF_INET)
+			if_addr = &if_info->address;
+		else
+			if_addr = &if_info->address_v6;
+
+		/* Found an existing interface. Notify clients if a new address
+		 * is set */
+		if (!*if_addr || strcmp(*if_addr, address)) {
+			talloc_free(*if_addr);
+			*if_addr = new_addr;
+			discover_server_notify_system_info(server, sysinfo);
+			return;
+		}
+	}
+
+	mac_str(hwaddr, hwaddr_size, mac, sizeof(mac));
+	pb_log("Couldn't find interface matching %s\n", mac);
+}
+
 void system_info_register_interface(unsigned int hwaddr_size, uint8_t *hwaddr,
 		const char *name, bool link)
 {
@@ -107,3 +152,21 @@ void system_info_init(struct discover_server *s)
 	sysinfo = talloc_zero(server, struct system_info);
 	platform_get_sysinfo(sysinfo);
 }
+
+/* Only reset device information. Platform information is static */
+void system_info_reinit(void)
+{
+	unsigned int i;
+
+	for (i = 0; i < sysinfo->n_blockdevs; i++)
+		talloc_free(sysinfo->blockdevs[i]);
+	talloc_free(sysinfo->blockdevs);
+	sysinfo->blockdevs = NULL;
+	sysinfo->n_blockdevs = 0;
+
+	for (i = 0; i < sysinfo->n_interfaces; i++)
+		talloc_free(sysinfo->interfaces[i]);
+	talloc_free(sysinfo->interfaces);
+	sysinfo->interfaces = NULL;
+	sysinfo->n_interfaces = 0;
+}