#ifndef ASM_NMI_H
#define ASM_NMI_H

#include <public/nmi.h>

struct cpu_user_regs;
 
typedef int (*nmi_callback_t)(struct cpu_user_regs *regs, int cpu);
 
/** 
 * set_nmi_callback
 *
 * Set a handler for an NMI. Only one handler may be
 * set. Return 1 if the NMI was handled.
 */
void set_nmi_callback(nmi_callback_t callback);
 
/** 
 * unset_nmi_callback
 *
 * Remove the handler previously set.
 */
void unset_nmi_callback(void);
 
/**
 * register_guest_nmi_callback
 *
 * The default NMI handler passes the NMI to a guest callback. This
 * function registers the address of that callback.
 */
long register_guest_nmi_callback(unsigned long address);

/**
 * unregister_guest_nmi_callback
 *
 * Unregister a guest NMI handler.
 */
long unregister_guest_nmi_callback(void);

#endif /* ASM_NMI_H */