// Copyright 2018 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 "domain_allocator.h"

#include <assert.h>

namespace intel_iommu {

DomainAllocator::DomainAllocator()
    // Note that next_domain_id_ starts at 1, since under some conditions 0 is
    // an invalid domain ID (i.e. if CM is set in the capability register).
    : num_domains_(0), next_domain_id_(1) {}

zx_status_t DomainAllocator::Allocate(uint32_t* domain_id) {
  if (next_domain_id_ >= num_domains_) {
    return ZX_ERR_NO_RESOURCES;
  }

  // This allocator should be enough, since the hardware should have enough
  // domain IDs for each device hanging off of it. If we start deallocating
  // Context Entries, we'll need to make this allocator more sophisticated to
  // manage the ID reuse.
  *domain_id = next_domain_id_++;
  return ZX_OK;
}

void DomainAllocator::set_num_domains(uint32_t num) {
  ASSERT(num >= next_domain_id_);
  num_domains_ = num;
}

}  // namespace intel_iommu