src/devices/block/drivers/ahci/pci-bus.h

// Copyright 2019 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_DEVICES_BLOCK_DRIVERS_AHCI_PCI_BUS_H_
#define SRC_DEVICES_BLOCK_DRIVERS_AHCI_PCI_BUS_H_

#include <fuchsia/hardware/pci/c/banjo.h>
#include <lib/device-protocol/pci.h>
#include <lib/mmio/mmio.h>
#include <lib/zx/bti.h>
#include <lib/zx/interrupt.h>
#include <zircon/types.h>

#include "bus.h"

namespace ahci {

class PciBus : public Bus {
 public:
  explicit PciBus(zx_device_t* parent) : pci_(parent, ddk::Pci::kFragmentName) {}
  ~PciBus() override;
  zx_status_t Configure(zx_device_t* parent) override;
  zx_status_t IoBufferInit(io_buffer_t* buffer_, size_t size, uint32_t flags, zx_paddr_t* phys_out,
                           void** virt_out) override;
  zx_status_t BtiPin(uint32_t options, const zx::unowned_vmo& vmo, uint64_t offset, uint64_t size,
                     zx_paddr_t* addrs, size_t addrs_count, zx::pmt* pmt_out) override;

  zx_status_t RegRead(size_t offset, uint32_t* val_out) override;
  zx_status_t RegWrite(size_t offset, uint32_t val) override;

  zx_status_t InterruptWait() override;
  void InterruptCancel() override;

 private:
  ddk::Pci pci_;
  pci_interrupt_mode_t irq_mode_;
  std::optional<fdf::MmioBuffer> mmio_;
  zx::bti bti_;
  zx::interrupt irq_;
};

}  // namespace ahci

#endif  // SRC_DEVICES_BLOCK_DRIVERS_AHCI_PCI_BUS_H_