]> git.ozlabs.org Git - petitboot/blobdiff - discover/sysinfo.c
discover/discover-server: Restrict clients based on uid
[petitboot] / discover / sysinfo.c
index 870e9a122a53a60f316d738770e4cc80daf7bac2..a9d7ff547b11ffb22a40bbdf4e738448de4f6a16 100644 (file)
@@ -3,27 +3,73 @@
 
 #include <talloc/talloc.h>
 #include <process/process.h>
+#include <log/log.h>
+#include <url/url.h>
 
 #include "discover-server.h"
+#include "platform.h"
 #include "sysinfo.h"
 
 static struct system_info *sysinfo;
 static struct discover_server *server;
 
-static const char *sysinfo_helper = PKG_LIBEXEC_DIR "/pb-sysinfo";
-
 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)
+               const char *name, bool link)
 {
        struct interface_info *if_info;
        unsigned int i;
 
        for (i = 0; i < sysinfo->n_interfaces; i++) {
+               bool changed = false;
+
                if_info = sysinfo->interfaces[i];
 
                if (if_info->hwaddr_size != hwaddr_size)
@@ -32,10 +78,22 @@ void system_info_register_interface(unsigned int hwaddr_size, uint8_t *hwaddr,
                if (memcmp(if_info->hwaddr, hwaddr, hwaddr_size))
                        continue;
 
-               /* update the name and we're done */
-               talloc_free(if_info->name);
-               if_info->name = talloc_strdup(if_info, name);
-               discover_server_notify_system_info(server, sysinfo);
+               /* Found an existing interface. Notify clients on any name or
+                * link changes */
+               if (strcmp(if_info->name, name)) {
+                       talloc_free(if_info->name);
+                       if_info->name = talloc_strdup(if_info, name);
+                       changed = true;
+               }
+
+               if (if_info->link != link) {
+                       if_info->link = link;
+                       changed = true;
+               }
+
+               if (changed)
+                       discover_server_notify_system_info(server, sysinfo);
+
                return;
        }
 
@@ -43,6 +101,7 @@ void system_info_register_interface(unsigned int hwaddr_size, uint8_t *hwaddr,
        if_info->hwaddr_size = hwaddr_size;
        if_info->hwaddr = talloc_memdup(if_info, hwaddr, hwaddr_size);
        if_info->name = talloc_strdup(if_info, name);
+       if_info->link = link;
 
        sysinfo->n_interfaces++;
        sysinfo->interfaces = talloc_realloc(sysinfo, sysinfo->interfaces,
@@ -53,41 +112,61 @@ void system_info_register_interface(unsigned int hwaddr_size, uint8_t *hwaddr,
        discover_server_notify_system_info(server, sysinfo);
 }
 
-static void system_info_set_identifier(struct system_info *info)
+void system_info_register_blockdev(const char *name, const char *uuid,
+               const char *mountpoint)
 {
-       struct process *process;
-       int rc;
-       const char *argv[] = {
-               sysinfo_helper, NULL, NULL,
-       };
-
-       process = process_create(info);
-       process->path = sysinfo_helper;
-       process->argv = argv;
-       process->keep_stdout = true;
-
-       argv[1] = "--type";
-       rc = process_run_sync(process);
-
-       if (!rc) {
-               info->type = talloc_strndup(info, process->stdout_buf,
-                               process->stdout_len);
-       }
+       struct blockdev_info *bd_info;
+       unsigned int i;
+
+       for (i = 0; i < sysinfo->n_blockdevs; i++) {
+               bd_info = sysinfo->blockdevs[i];
 
-       argv[1] = "--id";
-       rc = process_run_sync(process);
+               if (strcmp(bd_info->name, name))
+                       continue;
 
-       if (!rc) {
-               info->identifier = talloc_strndup(info, process->stdout_buf,
-                               process->stdout_len);
+               /* 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;
        }
 
-       process_release(process);
+       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);
 }
 
 void system_info_init(struct discover_server *s)
 {
        server = s;
        sysinfo = talloc_zero(server, struct system_info);
-       system_info_set_identifier(sysinfo);
+       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;
 }