#include <asm/interrupt.h>
#include <asm/kprobes.h>

struct soft_mask_table_entry {
        unsigned long start;
        unsigned long end;
};

struct restart_table_entry {
        unsigned long start;
        unsigned long end;
        unsigned long fixup;
};

extern struct soft_mask_table_entry __start___soft_mask_table[];
extern struct soft_mask_table_entry __stop___soft_mask_table[];

extern struct restart_table_entry __start___restart_table[];
extern struct restart_table_entry __stop___restart_table[];

/* Given an address, look for it in the soft mask table */
bool search_kernel_soft_mask_table(unsigned long addr)
{
        struct soft_mask_table_entry *smte = __start___soft_mask_table;

        while (smte < __stop___soft_mask_table) {
                unsigned long start = smte->start;
                unsigned long end = smte->end;

                if (addr >= start && addr < end)
                        return true;

                smte++;
        }
        return false;
}
NOKPROBE_SYMBOL(search_kernel_soft_mask_table);

/* Given an address, look for it in the kernel exception table */
unsigned long search_kernel_restart_table(unsigned long addr)
{
        struct restart_table_entry *rte = __start___restart_table;

        while (rte < __stop___restart_table) {
                unsigned long start = rte->start;
                unsigned long end = rte->end;
                unsigned long fixup = rte->fixup;

                if (addr >= start && addr < end)
                        return fixup;

                rte++;
        }
        return 0;
}
NOKPROBE_SYMBOL(search_kernel_restart_table);