netbsd: Add support for HAX_VCPU_IOCTL_SET_CPUID

Signed-off-by: Kamil Rytarowski <[email protected]>

Author: krytarowski

include/netbsd/hax_interface_netbsd.h

@@ -74,6 +74,11 @@
 
 #define HAX_IOCTL_VCPU_DEBUG _IOW(0, 0xc9, struct hax_debug_t)
 
+// `hax_cpuid *` is specified as the size of data buffer because `hax_cpuid` is
+// a variable-length type. When ioctl() is invoked, the argument of user data
+// should pass the address of the pointer to `hax_cpuid`.
+#define HAX_VCPU_IOCTL_SET_CPUID _IOW(0, 0xca, struct hax_cpuid *)
+
 #ifdef _KERNEL
 #define HAX_KERNEL64_CS 0x80
 #define HAX_KERNEL32_CS 0x08

platforms/netbsd/hax_entry_vcpu.c

@@ -58,6 +58,40 @@ struct cdevsw hax_vcpu_cdevsw = {
     .d_flag = D_OTHER | D_MPSAFE
 };
 
+#define load_user_data(dest, src, body_len, body_max, arg_t, body_t)          \
+        void *uaddr = (void *)(*(arg_t **)(src));                             \
+        size_t size;                                                          \
+        arg_t header;                                                         \
+        (dest) = NULL;                                                        \
+        if (copyin(uaddr, &header, sizeof(arg_t))) {                          \
+            hax_log(HAX_LOGE, "%s: argument header read error.\n", __func__); \
+            ret = -EFAULT;                                                    \
+            break;                                                            \
+        }                                                                     \
+        if (header.body_len > (body_max)) {                                   \
+            hax_log(HAX_LOGW, "%s: %d exceeds argument body maximum %d.\n",   \
+                    __func__, header.body_len, (body_max));                   \
+            ret = -E2BIG;                                                     \
+            break;                                                            \
+        }                                                                     \
+        size = sizeof(arg_t) + header.body_len * sizeof(body_t);               \
+        (dest) = hax_vmalloc(size, HAX_MEM_NONPAGE);                          \
+        if ((dest) == NULL) {                                                 \
+            hax_log(HAX_LOGE, "%s: failed to allocate memory.\n", __func__);  \
+            ret = -ENOMEM;                                                    \
+            break;                                                            \
+        }                                                                     \
+        if (copyin(uaddr, (dest), size)) {                                    \
+            hax_log(HAX_LOGE, "%s: argument read error.\n", __func__);        \
+            unload_user_data(dest);                                           \
+            ret = -EFAULT;                                                    \
+            break;                                                            \
+        }
+
+#define unload_user_data(dest)       \
+        if ((dest) != NULL)          \
+            hax_vfree((dest), size);
+
 /* VCPU operations */
 
 int hax_vcpu_open(dev_t self, int flag __unused, int mode __unused,
@@ -243,6 +277,14 @@ int hax_vcpu_ioctl(dev_t self, u_long cmd, void *data, int flag,
         vcpu_debug(cvcpu, hax_debug);
         break;
     }
+    case HAX_VCPU_IOCTL_SET_CPUID: {
+        struct hax_cpuid *cpuid;
+        load_user_data(cpuid, data, total, HAX_MAX_CPUID_ENTRIES, hax_cpuid,
+                       hax_cpuid_entry);
+        ret = vcpu_set_cpuid(cvcpu, cpuid);
+        unload_user_data(cpuid);
+        break;
+    }
     default:
         // TODO: Print information about the process that sent the ioctl.
         hax_log(HAX_LOGE, "Unknown VCPU IOCTL %#lx, pid=%d ('%s')\n", cmd,