/* 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.
         */
        this_cpu_write(cpu_tss_rw.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