/* SPDX-License-Identifier: GPL-2.0 */
#ifndef _ASM_X86_IOBITMAP_H
#define _ASM_X86_IOBITMAP_H

#include <linux/refcount.h>
#include <asm/processor.h>

struct io_bitmap {
    u64        sequence;
    refcount_t    refcnt;
    /* The maximum number of bytes to copy so all zero bits are covered */
    unsigned int    max;
    unsigned long    bitmap[IO_BITMAP_LONGS];
};

struct task_struct;

#ifdef CONFIG_X86_IOPL_IOPERM
void io_bitmap_share(struct task_struct *tsk);
void io_bitmap_exit(struct task_struct *tsk);

static inline void native_tss_invalidate_io_bitmap(void)
{
    /*
     * Invalidate the I/O bitmap by moving io_bitmap_base outside the
     * TSS limit so any subsequent I/O access from user space will
     * trigger a #GP.
     *
     * This is correct even when VMEXIT rewrites the TSS limit
     * to 0x67 as the only requirement is that the base points
     * outside the limit.
     */
    cpu_tss_rw[smp_processor_id()].x86_tss.io_bitmap_base = IO_BITMAP_OFFSET_INVALID;
}

void native_tss_update_io_bitmap(void);

#ifdef CONFIG_PARAVIRT_XXL
#include <asm/paravirt.h>
#else
#define tss_update_io_bitmap native_tss_update_io_bitmap
#define tss_invalidate_io_bitmap native_tss_invalidate_io_bitmap
#endif

#else
static inline void io_bitmap_share(struct task_struct *tsk) { }
static inline void io_bitmap_exit(struct task_struct *tsk) { }
static inline void tss_update_io_bitmap(void) { }
#endif

#endif