zircon/kernel/object/guest_dispatcher.cc

// Copyright 2017 The Fuchsia Authors
//
// Use of this source code is governed by a MIT-style
// license that can be found in the LICENSE file or at
// https://opensource.org/licenses/MIT

#include "object/guest_dispatcher.h"

#include <lib/counters.h>
#include <zircon/rights.h>

#include <arch/hypervisor.h>
#include <fbl/alloc_checker.h>
#include <object/vm_address_region_dispatcher.h>

KCOUNTER(dispatcher_guest_create_count, "dispatcher.guest.create")
KCOUNTER(dispatcher_guest_destroy_count, "dispatcher.guest.destroy")

// static
zx_status_t GuestDispatcher::Create(uint32_t options, KernelHandle<GuestDispatcher>* guest_handle,
                                    zx_rights_t* guest_rights,
                                    KernelHandle<VmAddressRegionDispatcher>* vmar_handle,
                                    zx_rights_t* vmar_rights) {
  if (options != 0) {
    return ZX_ERR_INVALID_ARGS;
  }

  auto guest = Guest::Create();
  if (guest.is_error()) {
    return guest.status_value();
  }
  fbl::RefPtr<VmAddressRegion> vmar = (*guest)->RootVmar();

  fbl::AllocChecker ac;
  KernelHandle new_guest_handle(fbl::AdoptRef(new (&ac) GuestDispatcher(ktl::move(*guest))));
  if (!ac.check()) {
    return ZX_ERR_NO_MEMORY;
  }

  zx_status_t status =
      VmAddressRegionDispatcher::Create(std::move(vmar), 0, vmar_handle, vmar_rights);
  if (status != ZX_OK) {
    return status;
  }

  *guest_rights = default_rights();
  *guest_handle = ktl::move(new_guest_handle);
  return ZX_OK;
}

GuestDispatcher::GuestDispatcher(ktl::unique_ptr<Guest> guest) : guest_(ktl::move(guest)) {
  kcounter_add(dispatcher_guest_create_count, 1);
}

GuestDispatcher::~GuestDispatcher() { kcounter_add(dispatcher_guest_destroy_count, 1); }

zx_status_t GuestDispatcher::SetTrap(uint32_t kind, zx_vaddr_t addr, size_t len,
                                     fbl::RefPtr<PortDispatcher> port, uint64_t key) {
  canary_.Assert();
  return guest_->SetTrap(kind, addr, len, ktl::move(port), key).status_value();
}