lib/libvmmapi/vmmapi.c

   1 /*-
   2  * Copyright (c) 2011 NetApp, Inc.
   3  * All rights reserved.
   4  *
   5  * Redistribution and use in source and binary forms, with or without
   6  * modification, are permitted provided that the following conditions
   7  * are met:
   8  * 1. Redistributions of source code must retain the above copyright
   9  *    notice, this list of conditions and the following disclaimer.
  10  * 2. Redistributions in binary form must reproduce the above copyright
  11  *    notice, this list of conditions and the following disclaimer in the
  12  *    documentation and/or other materials provided with the distribution.
  13  *
  14  * THIS SOFTWARE IS PROVIDED BY NETAPP, INC ``AS IS'' AND
  15  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  16  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  17  * ARE DISCLAIMED.  IN NO EVENT SHALL NETAPP, INC OR CONTRIBUTORS BE LIABLE
  18  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  19  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  20  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  21  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  22  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  23  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  24  * SUCH DAMAGE.
  25  *
  26  * $FreeBSD$
  27  */
  28
  29 #include <sys/cdefs.h>
  30 __FBSDID("$FreeBSD$");
  31
  32 #include <sys/types.h>
  33 #include <sys/sysctl.h>
  34 #include <sys/ioctl.h>
  35 #include <sys/mman.h>
  36
  37 #include <machine/specialreg.h>
  38
  39 #include <stdio.h>
  40 #include <stdlib.h>
  41 #include <assert.h>
  42 #include <string.h>
  43 #include <fcntl.h>
  44 #include <unistd.h>
  45
  46 #include <libutil.h>
  47
  48 #include <machine/vmm.h>
  49 #include <machine/vmm_dev.h>
  50
  51 #include "vmmapi.h"
  52
  53 #define MB      (1024 * 1024UL)
  54 #define GB      (1024 * 1024 * 1024UL)
  55
  56 struct vmctx {
  57         int     fd;
  58         uint32_t lowmem_limit;
  59         enum vm_mmap_style vms;
  60         size_t  lowmem;
  61         char    *lowmem_addr;
  62         size_t  highmem;
  63         char    *highmem_addr;
  64         char    *name;
  65 };
  66
  67 #define CREATE(x)  sysctlbyname("hw.vmm.create", NULL, NULL, (x), strlen((x)))
  68 #define DESTROY(x) sysctlbyname("hw.vmm.destroy", NULL, NULL, (x), strlen((x)))
  69
  70 static int
  71 vm_device_open(const char *name)
  72 {
  73         int fd, len;
  74         char *vmfile;
  75
  76         len = strlen("/dev/vmm/") + strlen(name) + 1;
  77         vmfile = malloc(len);
  78         assert(vmfile != NULL);
  79         snprintf(vmfile, len, "/dev/vmm/%s", name);
  80
  81         /* Open the device file */
  82         fd = open(vmfile, O_RDWR, 0);
  83
  84         free(vmfile);
  85         return (fd);
  86 }
  87
  88 int
  89 vm_create(const char *name)
  90 {
  91
  92         return (CREATE((char *)name));
  93 }
  94
  95 struct vmctx *
  96 vm_open(const char *name)
  97 {
  98         struct vmctx *vm;
  99
 100         vm = malloc(sizeof(struct vmctx) + strlen(name) + 1);
 101         assert(vm != NULL);
 102
 103         vm->fd = -1;
 104         vm->lowmem_limit = 3 * GB;
 105         vm->name = (char *)(vm + 1);
 106         strcpy(vm->name, name);
 107
 108         if ((vm->fd = vm_device_open(vm->name)) < 0)
 109                 goto err;
 110
 111         return (vm);
 112 err:
 113         vm_destroy(vm);
 114         return (NULL);
 115 }
 116
 117 void
 118 vm_destroy(struct vmctx *vm)
 119 {
 120         assert(vm != NULL);
 121
 122         if (vm->fd >= 0)
 123                 close(vm->fd);
 124         DESTROY(vm->name);
 125
 126         free(vm);
 127 }
 128
 129 int
 130 vm_parse_memsize(const char *optarg, size_t *ret_memsize)
 131 {
 132         char *endptr;
 133         size_t optval;
 134         int error;
 135
 136         optval = strtoul(optarg, &endptr, 0);
 137         if (*optarg != '\0' && *endptr == '\0') {
 138                 /*
 139                  * For the sake of backward compatibility if the memory size
 140                  * specified on the command line is less than a megabyte then
 141                  * it is interpreted as being in units of MB.
 142                  */
 143                 if (optval < MB)
 144                         optval *= MB;
 145                 *ret_memsize = optval;
 146                 error = 0;
 147         } else
 148                 error = expand_number(optarg, ret_memsize);
 149
 150         return (error);
 151 }
 152
 153 int
 154 vm_get_memory_seg(struct vmctx *ctx, vm_paddr_t gpa, size_t *ret_len,
 155                   int *wired)
 156 {
 157         int error;
 158         struct vm_memory_segment seg;
 159
 160         bzero(&seg, sizeof(seg));
 161         seg.gpa = gpa;
 162         error = ioctl(ctx->fd, VM_GET_MEMORY_SEG, &seg);
 163         *ret_len = seg.len;
 164         if (wired != NULL)
 165                 *wired = seg.wired;
 166         return (error);
 167 }
 168
 169 uint32_t
 170 vm_get_lowmem_limit(struct vmctx *ctx)
 171 {
 172
 173         return (ctx->lowmem_limit);
 174 }
 175
 176 void
 177 vm_set_lowmem_limit(struct vmctx *ctx, uint32_t limit)
 178 {
 179
 180         ctx->lowmem_limit = limit;
 181 }
 182
 183 static int
 184 setup_memory_segment(struct vmctx *ctx, vm_paddr_t gpa, size_t len, char **addr)
 185 {
 186         int error;
 187         struct vm_memory_segment seg;
 188
 189         /*
 190          * Create and optionally map 'len' bytes of memory at guest
 191          * physical address 'gpa'
 192          */
 193         bzero(&seg, sizeof(seg));
 194         seg.gpa = gpa;
 195         seg.len = len;
 196         error = ioctl(ctx->fd, VM_MAP_MEMORY, &seg);
 197         if (error == 0 && addr != NULL) {
 198                 *addr = mmap(NULL, len, PROT_READ | PROT_WRITE, MAP_SHARED,
 199                                 ctx->fd, gpa);
 200         }
 201         return (error);
 202 }
 203
 204 int
 205 vm_setup_memory(struct vmctx *ctx, size_t memsize, enum vm_mmap_style vms)
 206 {
 207         char **addr;
 208         int error;
 209
 210         /* XXX VM_MMAP_SPARSE not implemented yet */
 211         assert(vms == VM_MMAP_NONE || vms == VM_MMAP_ALL);
 212         ctx->vms = vms;
 213
 214         /*
 215          * If 'memsize' cannot fit entirely in the 'lowmem' segment then
 216          * create another 'highmem' segment above 4GB for the remainder.
 217          */
 218         if (memsize > ctx->lowmem_limit) {
 219                 ctx->lowmem = ctx->lowmem_limit;
 220                 ctx->highmem = memsize - ctx->lowmem;
 221         } else {
 222                 ctx->lowmem = memsize;
 223                 ctx->highmem = 0;
 224         }
 225
 226         if (ctx->lowmem > 0) {
 227                 addr = (vms == VM_MMAP_ALL) ? &ctx->lowmem_addr : NULL;
 228                 error = setup_memory_segment(ctx, 0, ctx->lowmem, addr);
 229                 if (error)
 230                         return (error);
 231         }
 232
 233         if (ctx->highmem > 0) {
 234                 addr = (vms == VM_MMAP_ALL) ? &ctx->highmem_addr : NULL;
 235                 error = setup_memory_segment(ctx, 4*GB, ctx->highmem, addr);
 236                 if (error)
 237                         return (error);
 238         }
 239
 240         return (0);
 241 }
 242
 243 void *
 244 vm_map_gpa(struct vmctx *ctx, vm_paddr_t gaddr, size_t len)
 245 {
 246
 247         /* XXX VM_MMAP_SPARSE not implemented yet */
 248         assert(ctx->vms == VM_MMAP_ALL);
 249
 250         if (gaddr < ctx->lowmem && gaddr + len <= ctx->lowmem)
 251                 return ((void *)(ctx->lowmem_addr + gaddr));
 252
 253         if (gaddr >= 4*GB) {
 254                 gaddr -= 4*GB;
 255                 if (gaddr < ctx->highmem && gaddr + len <= ctx->highmem)
 256                         return ((void *)(ctx->highmem_addr + gaddr));
 257         }
 258
 259         return (NULL);
 260 }
 261
 262 int
 263 vm_set_desc(struct vmctx *ctx, int vcpu, int reg,
 264             uint64_t base, uint32_t limit, uint32_t access)
 265 {
 266         int error;
 267         struct vm_seg_desc vmsegdesc;
 268
 269         bzero(&vmsegdesc, sizeof(vmsegdesc));
 270         vmsegdesc.cpuid = vcpu;
 271         vmsegdesc.regnum = reg;
 272         vmsegdesc.desc.base = base;
 273         vmsegdesc.desc.limit = limit;
 274         vmsegdesc.desc.access = access;
 275
 276         error = ioctl(ctx->fd, VM_SET_SEGMENT_DESCRIPTOR, &vmsegdesc);
 277         return (error);
 278 }
 279
 280 int
 281 vm_get_desc(struct vmctx *ctx, int vcpu, int reg,
 282             uint64_t *base, uint32_t *limit, uint32_t *access)
 283 {
 284         int error;
 285         struct vm_seg_desc vmsegdesc;
 286
 287         bzero(&vmsegdesc, sizeof(vmsegdesc));
 288         vmsegdesc.cpuid = vcpu;
 289         vmsegdesc.regnum = reg;
 290
 291         error = ioctl(ctx->fd, VM_GET_SEGMENT_DESCRIPTOR, &vmsegdesc);
 292         if (error == 0) {
 293                 *base = vmsegdesc.desc.base;
 294                 *limit = vmsegdesc.desc.limit;
 295                 *access = vmsegdesc.desc.access;
 296         }
 297         return (error);
 298 }
 299
 300 int
 301 vm_set_register(struct vmctx *ctx, int vcpu, int reg, uint64_t val)
 302 {
 303         int error;
 304         struct vm_register vmreg;
 305
 306         bzero(&vmreg, sizeof(vmreg));
 307         vmreg.cpuid = vcpu;
 308         vmreg.regnum = reg;
 309         vmreg.regval = val;
 310
 311         error = ioctl(ctx->fd, VM_SET_REGISTER, &vmreg);
 312         return (error);
 313 }
 314
 315 int
 316 vm_get_register(struct vmctx *ctx, int vcpu, int reg, uint64_t *ret_val)
 317 {
 318         int error;
 319         struct vm_register vmreg;
 320
 321         bzero(&vmreg, sizeof(vmreg));
 322         vmreg.cpuid = vcpu;
 323         vmreg.regnum = reg;
 324
 325         error = ioctl(ctx->fd, VM_GET_REGISTER, &vmreg);
 326         *ret_val = vmreg.regval;
 327         return (error);
 328 }
 329
 330 int
 331 vm_run(struct vmctx *ctx, int vcpu, uint64_t rip, struct vm_exit *vmexit)
 332 {
 333         int error;
 334         struct vm_run vmrun;
 335
 336         bzero(&vmrun, sizeof(vmrun));
 337         vmrun.cpuid = vcpu;
 338         vmrun.rip = rip;
 339
 340         error = ioctl(ctx->fd, VM_RUN, &vmrun);
 341         bcopy(&vmrun.vm_exit, vmexit, sizeof(struct vm_exit));
 342         return (error);
 343 }
 344
 345 static int
 346 vm_inject_event_real(struct vmctx *ctx, int vcpu, enum vm_event_type type,
 347                      int vector, int error_code, int error_code_valid)
 348 {
 349         struct vm_event ev;
 350
 351         bzero(&ev, sizeof(ev));
 352         ev.cpuid = vcpu;
 353         ev.type = type;
 354         ev.vector = vector;
 355         ev.error_code = error_code;
 356         ev.error_code_valid = error_code_valid;
 357
 358         return (ioctl(ctx->fd, VM_INJECT_EVENT, &ev));
 359 }
 360
 361 int
 362 vm_inject_event(struct vmctx *ctx, int vcpu, enum vm_event_type type,
 363                 int vector)
 364 {
 365
 366         return (vm_inject_event_real(ctx, vcpu, type, vector, 0, 0));
 367 }
 368
 369 int
 370 vm_inject_event2(struct vmctx *ctx, int vcpu, enum vm_event_type type,
 371                  int vector, int error_code)
 372 {
 373
 374         return (vm_inject_event_real(ctx, vcpu, type, vector, error_code, 1));
 375 }
 376
 377 int
 378 vm_apicid2vcpu(struct vmctx *ctx, int apicid)
 379 {
 380         /*
 381          * The apic id associated with the 'vcpu' has the same numerical value
 382          * as the 'vcpu' itself.
 383          */
 384         return (apicid);
 385 }
 386
 387 int
 388 vm_lapic_irq(struct vmctx *ctx, int vcpu, int vector)
 389 {
 390         struct vm_lapic_irq vmirq;
 391
 392         bzero(&vmirq, sizeof(vmirq));
 393         vmirq.cpuid = vcpu;
 394         vmirq.vector = vector;
 395
 396         return (ioctl(ctx->fd, VM_LAPIC_IRQ, &vmirq));
 397 }
 398
 399 int
 400 vm_inject_nmi(struct vmctx *ctx, int vcpu)
 401 {
 402         struct vm_nmi vmnmi;
 403
 404         bzero(&vmnmi, sizeof(vmnmi));
 405         vmnmi.cpuid = vcpu;
 406
 407         return (ioctl(ctx->fd, VM_INJECT_NMI, &vmnmi));
 408 }
 409
 410 static struct {
 411         const char      *name;
 412         int             type;
 413 } capstrmap[] = {
 414         { "hlt_exit",           VM_CAP_HALT_EXIT },
 415         { "mtrap_exit",         VM_CAP_MTRAP_EXIT },
 416         { "pause_exit",         VM_CAP_PAUSE_EXIT },
 417         { "unrestricted_guest", VM_CAP_UNRESTRICTED_GUEST },
 418         { "enable_invpcid",     VM_CAP_ENABLE_INVPCID },
 419         { 0 }
 420 };
 421
 422 int
 423 vm_capability_name2type(const char *capname)
 424 {
 425         int i;
 426
 427         for (i = 0; capstrmap[i].name != NULL && capname != NULL; i++) {
 428                 if (strcmp(capstrmap[i].name, capname) == 0)
 429                         return (capstrmap[i].type);
 430         }
 431
 432         return (-1);
 433 }
 434
 435 const char *
 436 vm_capability_type2name(int type)
 437 {
 438         int i;
 439
 440         for (i = 0; capstrmap[i].name != NULL; i++) {
 441                 if (capstrmap[i].type == type)
 442                         return (capstrmap[i].name);
 443         }
 444
 445         return (NULL);
 446 }
 447
 448 int
 449 vm_get_capability(struct vmctx *ctx, int vcpu, enum vm_cap_type cap,
 450                   int *retval)
 451 {
 452         int error;
 453         struct vm_capability vmcap;
 454
 455         bzero(&vmcap, sizeof(vmcap));
 456         vmcap.cpuid = vcpu;
 457         vmcap.captype = cap;
 458
 459         error = ioctl(ctx->fd, VM_GET_CAPABILITY, &vmcap);
 460         *retval = vmcap.capval;
 461         return (error);
 462 }
 463
 464 int
 465 vm_set_capability(struct vmctx *ctx, int vcpu, enum vm_cap_type cap, int val)
 466 {
 467         struct vm_capability vmcap;
 468
 469         bzero(&vmcap, sizeof(vmcap));
 470         vmcap.cpuid = vcpu;
 471         vmcap.captype = cap;
 472         vmcap.capval = val;
 473
 474         return (ioctl(ctx->fd, VM_SET_CAPABILITY, &vmcap));
 475 }
 476
 477 int
 478 vm_assign_pptdev(struct vmctx *ctx, int bus, int slot, int func)
 479 {
 480         struct vm_pptdev pptdev;
 481
 482         bzero(&pptdev, sizeof(pptdev));
 483         pptdev.bus = bus;
 484         pptdev.slot = slot;
 485         pptdev.func = func;
 486
 487         return (ioctl(ctx->fd, VM_BIND_PPTDEV, &pptdev));
 488 }
 489
 490 int
 491 vm_unassign_pptdev(struct vmctx *ctx, int bus, int slot, int func)
 492 {
 493         struct vm_pptdev pptdev;
 494
 495         bzero(&pptdev, sizeof(pptdev));
 496         pptdev.bus = bus;
 497         pptdev.slot = slot;
 498         pptdev.func = func;
 499
 500         return (ioctl(ctx->fd, VM_UNBIND_PPTDEV, &pptdev));
 501 }
 502
 503 int
 504 vm_map_pptdev_mmio(struct vmctx *ctx, int bus, int slot, int func,
 505                    vm_paddr_t gpa, size_t len, vm_paddr_t hpa)
 506 {
 507         struct vm_pptdev_mmio pptmmio;
 508
 509         bzero(&pptmmio, sizeof(pptmmio));
 510         pptmmio.bus = bus;
 511         pptmmio.slot = slot;
 512         pptmmio.func = func;
 513         pptmmio.gpa = gpa;
 514         pptmmio.len = len;
 515         pptmmio.hpa = hpa;
 516
 517         return (ioctl(ctx->fd, VM_MAP_PPTDEV_MMIO, &pptmmio));
 518 }
 519
 520 int
 521 vm_setup_msi(struct vmctx *ctx, int vcpu, int bus, int slot, int func,
 522              int destcpu, int vector, int numvec)
 523 {
 524         struct vm_pptdev_msi pptmsi;
 525
 526         bzero(&pptmsi, sizeof(pptmsi));
 527         pptmsi.vcpu = vcpu;
 528         pptmsi.bus = bus;
 529         pptmsi.slot = slot;
 530         pptmsi.func = func;
 531         pptmsi.destcpu = destcpu;
 532         pptmsi.vector = vector;
 533         pptmsi.numvec = numvec;
 534
 535         return (ioctl(ctx->fd, VM_PPTDEV_MSI, &pptmsi));
 536 }
 537
 538 int
 539 vm_setup_msix(struct vmctx *ctx, int vcpu, int bus, int slot, int func,
 540               int idx, uint32_t msg, uint32_t vector_control, uint64_t addr)
 541 {
 542         struct vm_pptdev_msix pptmsix;
 543
 544         bzero(&pptmsix, sizeof(pptmsix));
 545         pptmsix.vcpu = vcpu;
 546         pptmsix.bus = bus;
 547         pptmsix.slot = slot;
 548         pptmsix.func = func;
 549         pptmsix.idx = idx;
 550         pptmsix.msg = msg;
 551         pptmsix.addr = addr;
 552         pptmsix.vector_control = vector_control;
 553
 554         return ioctl(ctx->fd, VM_PPTDEV_MSIX, &pptmsix);
 555 }
 556
 557 uint64_t *
 558 vm_get_stats(struct vmctx *ctx, int vcpu, struct timeval *ret_tv,
 559              int *ret_entries)
 560 {
 561         int error;
 562
 563         static struct vm_stats vmstats;
 564
 565         vmstats.cpuid = vcpu;
 566
 567         error = ioctl(ctx->fd, VM_STATS, &vmstats);
 568         if (error == 0) {
 569                 if (ret_entries)
 570                         *ret_entries = vmstats.num_entries;
 571                 if (ret_tv)
 572                         *ret_tv = vmstats.tv;
 573                 return (vmstats.statbuf);
 574         } else
 575                 return (NULL);
 576 }
 577
 578 const char *
 579 vm_get_stat_desc(struct vmctx *ctx, int index)
 580 {
 581         static struct vm_stat_desc statdesc;
 582
 583         statdesc.index = index;
 584         if (ioctl(ctx->fd, VM_STAT_DESC, &statdesc) == 0)
 585                 return (statdesc.desc);
 586         else
 587                 return (NULL);
 588 }
 589
 590 int
 591 vm_get_x2apic_state(struct vmctx *ctx, int vcpu, enum x2apic_state *state)
 592 {
 593         int error;
 594         struct vm_x2apic x2apic;
 595
 596         bzero(&x2apic, sizeof(x2apic));
 597         x2apic.cpuid = vcpu;
 598
 599         error = ioctl(ctx->fd, VM_GET_X2APIC_STATE, &x2apic);
 600         *state = x2apic.state;
 601         return (error);
 602 }
 603
 604 int
 605 vm_set_x2apic_state(struct vmctx *ctx, int vcpu, enum x2apic_state state)
 606 {
 607         int error;
 608         struct vm_x2apic x2apic;
 609
 610         bzero(&x2apic, sizeof(x2apic));
 611         x2apic.cpuid = vcpu;
 612         x2apic.state = state;
 613
 614         error = ioctl(ctx->fd, VM_SET_X2APIC_STATE, &x2apic);
 615
 616         return (error);
 617 }
 618
 619 /*
 620  * From Intel Vol 3a:
 621  * Table 9-1. IA-32 Processor States Following Power-up, Reset or INIT
 622  */
 623 int
 624 vcpu_reset(struct vmctx *vmctx, int vcpu)
 625 {
 626         int error;
 627         uint64_t rflags, rip, cr0, cr4, zero, desc_base, rdx;
 628         uint32_t desc_access, desc_limit;
 629         uint16_t sel;
 630
 631         zero = 0;
 632
 633         rflags = 0x2;
 634         error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_RFLAGS, rflags);
 635         if (error)
 636                 goto done;
 637
 638         rip = 0xfff0;
 639         if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_RIP, rip)) != 0)
 640                 goto done;
 641
 642         cr0 = CR0_NE;
 643         if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_CR0, cr0)) != 0)
 644                 goto done;
 645
 646         if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_CR3, zero)) != 0)
 647                 goto done;
 648
 649         cr4 = 0;
 650         if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_CR4, cr4)) != 0)
 651                 goto done;
 652
 653         /*
 654          * CS: present, r/w, accessed, 16-bit, byte granularity, usable
 655          */
 656         desc_base = 0xffff0000;
 657         desc_limit = 0xffff;
 658         desc_access = 0x0093;
 659         error = vm_set_desc(vmctx, vcpu, VM_REG_GUEST_CS,
 660                             desc_base, desc_limit, desc_access);
 661         if (error)
 662                 goto done;
 663
 664         sel = 0xf000;
 665         if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_CS, sel)) != 0)
 666                 goto done;
 667
 668         /*
 669          * SS,DS,ES,FS,GS: present, r/w, accessed, 16-bit, byte granularity
 670          */
 671         desc_base = 0;
 672         desc_limit = 0xffff;
 673         desc_access = 0x0093;
 674         error = vm_set_desc(vmctx, vcpu, VM_REG_GUEST_SS,
 675                             desc_base, desc_limit, desc_access);
 676         if (error)
 677                 goto done;
 678
 679         error = vm_set_desc(vmctx, vcpu, VM_REG_GUEST_DS,
 680                             desc_base, desc_limit, desc_access);
 681         if (error)
 682                 goto done;
 683
 684         error = vm_set_desc(vmctx, vcpu, VM_REG_GUEST_ES,
 685                             desc_base, desc_limit, desc_access);
 686         if (error)
 687                 goto done;
 688
 689         error = vm_set_desc(vmctx, vcpu, VM_REG_GUEST_FS,
 690                             desc_base, desc_limit, desc_access);
 691         if (error)
 692                 goto done;
 693
 694         error = vm_set_desc(vmctx, vcpu, VM_REG_GUEST_GS,
 695                             desc_base, desc_limit, desc_access);
 696         if (error)
 697                 goto done;
 698
 699         sel = 0;
 700         if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_SS, sel)) != 0)
 701                 goto done;
 702         if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_DS, sel)) != 0)
 703                 goto done;
 704         if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_ES, sel)) != 0)
 705                 goto done;
 706         if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_FS, sel)) != 0)
 707                 goto done;
 708         if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_GS, sel)) != 0)
 709                 goto done;
 710
 711         /* General purpose registers */
 712         rdx = 0xf00;
 713         if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_RAX, zero)) != 0)
 714                 goto done;
 715         if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_RBX, zero)) != 0)
 716                 goto done;
 717         if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_RCX, zero)) != 0)
 718                 goto done;
 719         if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_RDX, rdx)) != 0)
 720                 goto done;
 721         if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_RSI, zero)) != 0)
 722                 goto done;
 723         if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_RDI, zero)) != 0)
 724                 goto done;
 725         if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_RBP, zero)) != 0)
 726                 goto done;
 727         if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_RSP, zero)) != 0)
 728                 goto done;
 729
 730         /* GDTR, IDTR */
 731         desc_base = 0;
 732         desc_limit = 0xffff;
 733         desc_access = 0;
 734         error = vm_set_desc(vmctx, vcpu, VM_REG_GUEST_GDTR,
 735                             desc_base, desc_limit, desc_access);
 736         if (error != 0)
 737                 goto done;
 738
 739         error = vm_set_desc(vmctx, vcpu, VM_REG_GUEST_IDTR,
 740                             desc_base, desc_limit, desc_access);
 741         if (error != 0)
 742                 goto done;
 743
 744         /* TR */
 745         desc_base = 0;
 746         desc_limit = 0xffff;
 747         desc_access = 0x0000008b;
 748         error = vm_set_desc(vmctx, vcpu, VM_REG_GUEST_TR, 0, 0, desc_access);
 749         if (error)
 750                 goto done;
 751
 752         sel = 0;
 753         if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_TR, sel)) != 0)
 754                 goto done;
 755
 756         /* LDTR */
 757         desc_base = 0;
 758         desc_limit = 0xffff;
 759         desc_access = 0x00000082;
 760         error = vm_set_desc(vmctx, vcpu, VM_REG_GUEST_LDTR, desc_base,
 761                             desc_limit, desc_access);
 762         if (error)
 763                 goto done;
 764
 765         sel = 0;
 766         if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_LDTR, 0)) != 0)
 767                 goto done;
 768
 769         /* XXX cr2, debug registers */
 770
 771         error = 0;
 772 done:
 773         return (error);
 774 }
 775
 776 int
 777 vm_get_gpa_pmap(struct vmctx *ctx, uint64_t gpa, uint64_t *pte, int *num)
 778 {
 779         int error, i;
 780         struct vm_gpa_pte gpapte;
 781
 782         bzero(&gpapte, sizeof(gpapte));
 783         gpapte.gpa = gpa;
 784
 785         error = ioctl(ctx->fd, VM_GET_GPA_PMAP, &gpapte);
 786
 787         if (error == 0) {
 788                 *num = gpapte.ptenum;
 789                 for (i = 0; i < gpapte.ptenum; i++)
 790                         pte[i] = gpapte.pte[i];
 791         }
 792
 793         return (error);
 794 }