drivers/msd-vsi-vip/src/page_table_arrays.h

// Copyright 2018 The Fuchsia Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef SRC_GRAPHICS_DRIVERS_MSD_VSI_VIP_SRC_PAGE_TABLE_ARRAYS_H_
#define SRC_GRAPHICS_DRIVERS_MSD_VSI_VIP_SRC_PAGE_TABLE_ARRAYS_H_

#include <lib/magma/platform/platform_buffer.h>
#include <lib/magma/platform/platform_bus_mapper.h>
#include <lib/magma/util/short_macros.h>
#include <lib/magma_service/util/register_io.h>

#include "address_space.h"
#include "registers.h"

class PageTableArrays {
 public:
  static std::unique_ptr<PageTableArrays> Create(magma::PlatformBusMapper* bus_mapper);

  PageTableArrays() = default;

  uint64_t bus_addr() {
    DASSERT(bus_mapping_);
    return bus_mapping_->Get()[0];
  }

  static uint32_t size() { return kPageTableArrayEntries; }

  void HardwareInit(magma::RegisterIo* register_io);
  void Enable(magma::RegisterIo* register_io, bool enable);
  void AssignAddressSpace(uint32_t index, AddressSpace* address_space);

  bool IsEnabled(magma::RegisterIo* register_io) {
    return registers::MmuSecureControl::Get().ReadFrom(register_io).enable();
  }

 private:
  static constexpr uint32_t kPageTableArraySizeInPages = 1;
  static constexpr uint32_t kPageTableArrayEntries =
      kPageTableArraySizeInPages * PAGE_SIZE / sizeof(uint64_t);

  bool Init(magma::PlatformBusMapper* bus_mapper);

  std::unique_ptr<magma::PlatformBuffer> page_table_array_;
  std::unique_ptr<magma::PlatformBusMapper::BusMapping> bus_mapping_;
  uint64_t* master_tlb_bus_addr_ = nullptr;

  std::unique_ptr<magma::PlatformBuffer> security_safe_page_;
  std::unique_ptr<magma::PlatformBusMapper::BusMapping> security_safe_page_bus_mapping_;

  std::unique_ptr<magma::PlatformBuffer> non_security_safe_page_;
  std::unique_ptr<magma::PlatformBusMapper::BusMapping> non_security_safe_page_bus_mapping_;

  PageTableArrays(const PageTableArrays&) = delete;
  void operator=(const PageTableArrays&) = delete;

  friend class TestMsdVsiDevice;
};

#endif  // SRC_GRAPHICS_DRIVERS_MSD_VSI_VIP_SRC_PAGE_TABLE_ARRAYS_H_