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         { 0 }
 419 };
 420
 421 int
 422 vm_capability_name2type(const char *capname)
 423 {
 424         int i;
 425
 426         for (i = 0; capstrmap[i].name != NULL && capname != NULL; i++) {
 427                 if (strcmp(capstrmap[i].name, capname) == 0)
 428                         return (capstrmap[i].type);
 429         }
 430
 431         return (-1);
 432 }
 433
 434 const char *
 435 vm_capability_type2name(int type)
 436 {
 437         int i;
 438
 439         for (i = 0; capstrmap[i].name != NULL; i++) {
 440                 if (capstrmap[i].type == type)
 441                         return (capstrmap[i].name);
 442         }
 443
 444         return (NULL);
 445 }
 446
 447 int
 448 vm_get_capability(struct vmctx *ctx, int vcpu, enum vm_cap_type cap,
 449                   int *retval)
 450 {
 451         int error;
 452         struct vm_capability vmcap;
 453
 454         bzero(&vmcap, sizeof(vmcap));
 455         vmcap.cpuid = vcpu;
 456         vmcap.captype = cap;
 457
 458         error = ioctl(ctx->fd, VM_GET_CAPABILITY, &vmcap);
 459         *retval = vmcap.capval;
 460         return (error);
 461 }
 462
 463 int
 464 vm_set_capability(struct vmctx *ctx, int vcpu, enum vm_cap_type cap, int val)
 465 {
 466         struct vm_capability vmcap;
 467
 468         bzero(&vmcap, sizeof(vmcap));
 469         vmcap.cpuid = vcpu;
 470         vmcap.captype = cap;
 471         vmcap.capval = val;
 472
 473         return (ioctl(ctx->fd, VM_SET_CAPABILITY, &vmcap));
 474 }
 475
 476 int
 477 vm_assign_pptdev(struct vmctx *ctx, int bus, int slot, int func)
 478 {
 479         struct vm_pptdev pptdev;
 480
 481         bzero(&pptdev, sizeof(pptdev));
 482         pptdev.bus = bus;
 483         pptdev.slot = slot;
 484         pptdev.func = func;
 485
 486         return (ioctl(ctx->fd, VM_BIND_PPTDEV, &pptdev));
 487 }
 488
 489 int
 490 vm_unassign_pptdev(struct vmctx *ctx, int bus, int slot, int func)
 491 {
 492         struct vm_pptdev pptdev;
 493
 494         bzero(&pptdev, sizeof(pptdev));
 495         pptdev.bus = bus;
 496         pptdev.slot = slot;
 497         pptdev.func = func;
 498
 499         return (ioctl(ctx->fd, VM_UNBIND_PPTDEV, &pptdev));
 500 }
 501
 502 int
 503 vm_map_pptdev_mmio(struct vmctx *ctx, int bus, int slot, int func,
 504                    vm_paddr_t gpa, size_t len, vm_paddr_t hpa)
 505 {
 506         struct vm_pptdev_mmio pptmmio;
 507
 508         bzero(&pptmmio, sizeof(pptmmio));
 509         pptmmio.bus = bus;
 510         pptmmio.slot = slot;
 511         pptmmio.func = func;
 512         pptmmio.gpa = gpa;
 513         pptmmio.len = len;
 514         pptmmio.hpa = hpa;
 515
 516         return (ioctl(ctx->fd, VM_MAP_PPTDEV_MMIO, &pptmmio));
 517 }
 518
 519 int
 520 vm_setup_msi(struct vmctx *ctx, int vcpu, int bus, int slot, int func,
 521              int destcpu, int vector, int numvec)
 522 {
 523         struct vm_pptdev_msi pptmsi;
 524
 525         bzero(&pptmsi, sizeof(pptmsi));
 526         pptmsi.vcpu = vcpu;
 527         pptmsi.bus = bus;
 528         pptmsi.slot = slot;
 529         pptmsi.func = func;
 530         pptmsi.destcpu = destcpu;
 531         pptmsi.vector = vector;
 532         pptmsi.numvec = numvec;
 533
 534         return (ioctl(ctx->fd, VM_PPTDEV_MSI, &pptmsi));
 535 }
 536
 537 int
 538 vm_setup_msix(struct vmctx *ctx, int vcpu, int bus, int slot, int func,
 539               int idx, uint32_t msg, uint32_t vector_control, uint64_t addr)
 540 {
 541         struct vm_pptdev_msix pptmsix;
 542
 543         bzero(&pptmsix, sizeof(pptmsix));
 544         pptmsix.vcpu = vcpu;
 545         pptmsix.bus = bus;
 546         pptmsix.slot = slot;
 547         pptmsix.func = func;
 548         pptmsix.idx = idx;
 549         pptmsix.msg = msg;
 550         pptmsix.addr = addr;
 551         pptmsix.vector_control = vector_control;
 552
 553         return ioctl(ctx->fd, VM_PPTDEV_MSIX, &pptmsix);
 554 }
 555
 556 uint64_t *
 557 vm_get_stats(struct vmctx *ctx, int vcpu, struct timeval *ret_tv,
 558              int *ret_entries)
 559 {
 560         int error;
 561
 562         static struct vm_stats vmstats;
 563
 564         vmstats.cpuid = vcpu;
 565
 566         error = ioctl(ctx->fd, VM_STATS, &vmstats);
 567         if (error == 0) {
 568                 if (ret_entries)
 569                         *ret_entries = vmstats.num_entries;
 570                 if (ret_tv)
 571                         *ret_tv = vmstats.tv;
 572                 return (vmstats.statbuf);
 573         } else
 574                 return (NULL);
 575 }
 576
 577 const char *
 578 vm_get_stat_desc(struct vmctx *ctx, int index)
 579 {
 580         static struct vm_stat_desc statdesc;
 581
 582         statdesc.index = index;
 583         if (ioctl(ctx->fd, VM_STAT_DESC, &statdesc) == 0)
 584                 return (statdesc.desc);
 585         else
 586                 return (NULL);
 587 }
 588
 589 int
 590 vm_get_x2apic_state(struct vmctx *ctx, int vcpu, enum x2apic_state *state)
 591 {
 592         int error;
 593         struct vm_x2apic x2apic;
 594
 595         bzero(&x2apic, sizeof(x2apic));
 596         x2apic.cpuid = vcpu;
 597
 598         error = ioctl(ctx->fd, VM_GET_X2APIC_STATE, &x2apic);
 599         *state = x2apic.state;
 600         return (error);
 601 }
 602
 603 int
 604 vm_set_x2apic_state(struct vmctx *ctx, int vcpu, enum x2apic_state state)
 605 {
 606         int error;
 607         struct vm_x2apic x2apic;
 608
 609         bzero(&x2apic, sizeof(x2apic));
 610         x2apic.cpuid = vcpu;
 611         x2apic.state = state;
 612
 613         error = ioctl(ctx->fd, VM_SET_X2APIC_STATE, &x2apic);
 614
 615         return (error);
 616 }
 617
 618 /*
 619  * From Intel Vol 3a:
 620  * Table 9-1. IA-32 Processor States Following Power-up, Reset or INIT
 621  */
 622 int
 623 vcpu_reset(struct vmctx *vmctx, int vcpu)
 624 {
 625         int error;
 626         uint64_t rflags, rip, cr0, cr4, zero, desc_base, rdx;
 627         uint32_t desc_access, desc_limit;
 628         uint16_t sel;
 629
 630         zero = 0;
 631
 632         rflags = 0x2;
 633         error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_RFLAGS, rflags);
 634         if (error)
 635                 goto done;
 636
 637         rip = 0xfff0;
 638         if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_RIP, rip)) != 0)
 639                 goto done;
 640
 641         cr0 = CR0_NE;
 642         if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_CR0, cr0)) != 0)
 643                 goto done;
 644
 645         if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_CR3, zero)) != 0)
 646                 goto done;
 647
 648         cr4 = 0;
 649         if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_CR4, cr4)) != 0)
 650                 goto done;
 651
 652         /*
 653          * CS: present, r/w, accessed, 16-bit, byte granularity, usable
 654          */
 655         desc_base = 0xffff0000;
 656         desc_limit = 0xffff;
 657         desc_access = 0x0093;
 658         error = vm_set_desc(vmctx, vcpu, VM_REG_GUEST_CS,
 659                             desc_base, desc_limit, desc_access);
 660         if (error)
 661                 goto done;
 662
 663         sel = 0xf000;
 664         if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_CS, sel)) != 0)
 665                 goto done;
 666
 667         /*
 668          * SS,DS,ES,FS,GS: present, r/w, accessed, 16-bit, byte granularity
 669          */
 670         desc_base = 0;
 671         desc_limit = 0xffff;
 672         desc_access = 0x0093;
 673         error = vm_set_desc(vmctx, vcpu, VM_REG_GUEST_SS,
 674                             desc_base, desc_limit, desc_access);
 675         if (error)
 676                 goto done;
 677
 678         error = vm_set_desc(vmctx, vcpu, VM_REG_GUEST_DS,
 679                             desc_base, desc_limit, desc_access);
 680         if (error)
 681                 goto done;
 682
 683         error = vm_set_desc(vmctx, vcpu, VM_REG_GUEST_ES,
 684                             desc_base, desc_limit, desc_access);
 685         if (error)
 686                 goto done;
 687
 688         error = vm_set_desc(vmctx, vcpu, VM_REG_GUEST_FS,
 689                             desc_base, desc_limit, desc_access);
 690         if (error)
 691                 goto done;
 692
 693         error = vm_set_desc(vmctx, vcpu, VM_REG_GUEST_GS,
 694                             desc_base, desc_limit, desc_access);
 695         if (error)
 696                 goto done;
 697
 698         sel = 0;
 699         if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_SS, sel)) != 0)
 700                 goto done;
 701         if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_DS, sel)) != 0)
 702                 goto done;
 703         if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_ES, sel)) != 0)
 704                 goto done;
 705         if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_FS, sel)) != 0)
 706                 goto done;
 707         if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_GS, sel)) != 0)
 708                 goto done;
 709
 710         /* General purpose registers */
 711         rdx = 0xf00;
 712         if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_RAX, zero)) != 0)
 713                 goto done;
 714         if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_RBX, zero)) != 0)
 715                 goto done;
 716         if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_RCX, zero)) != 0)
 717                 goto done;
 718         if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_RDX, rdx)) != 0)
 719                 goto done;
 720         if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_RSI, zero)) != 0)
 721                 goto done;
 722         if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_RDI, zero)) != 0)
 723                 goto done;
 724         if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_RBP, zero)) != 0)
 725                 goto done;
 726         if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_RSP, zero)) != 0)
 727                 goto done;
 728
 729         /* GDTR, IDTR */
 730         desc_base = 0;
 731         desc_limit = 0xffff;
 732         desc_access = 0;
 733         error = vm_set_desc(vmctx, vcpu, VM_REG_GUEST_GDTR,
 734                             desc_base, desc_limit, desc_access);
 735         if (error != 0)
 736                 goto done;
 737
 738         error = vm_set_desc(vmctx, vcpu, VM_REG_GUEST_IDTR,
 739                             desc_base, desc_limit, desc_access);
 740         if (error != 0)
 741                 goto done;
 742
 743         /* TR */
 744         desc_base = 0;
 745         desc_limit = 0xffff;
 746         desc_access = 0x0000008b;
 747         error = vm_set_desc(vmctx, vcpu, VM_REG_GUEST_TR, 0, 0, desc_access);
 748         if (error)
 749                 goto done;
 750
 751         sel = 0;
 752         if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_TR, sel)) != 0)
 753                 goto done;
 754
 755         /* LDTR */
 756         desc_base = 0;
 757         desc_limit = 0xffff;
 758         desc_access = 0x00000082;
 759         error = vm_set_desc(vmctx, vcpu, VM_REG_GUEST_LDTR, desc_base,
 760                             desc_limit, desc_access);
 761         if (error)
 762                 goto done;
 763
 764         sel = 0;
 765         if ((error = vm_set_register(vmctx, vcpu, VM_REG_GUEST_LDTR, 0)) != 0)
 766                 goto done;
 767
 768         /* XXX cr2, debug registers */
 769
 770         error = 0;
 771 done:
 772         return (error);
 773 }
 774
 775 int
 776 vm_get_gpa_pmap(struct vmctx *ctx, uint64_t gpa, uint64_t *pte, int *num)
 777 {
 778         int error, i;
 779         struct vm_gpa_pte gpapte;
 780
 781         bzero(&gpapte, sizeof(gpapte));
 782         gpapte.gpa = gpa;
 783
 784         error = ioctl(ctx->fd, VM_GET_GPA_PMAP, &gpapte);
 785
 786         if (error == 0) {
 787                 *num = gpapte.ptenum;
 788                 for (i = 0; i < gpapte.ptenum; i++)
 789                         pte[i] = gpapte.pte[i];
 790         }
 791
 792         return (error);
 793 }