/*
 * This file is subject to the terms and conditions of the GNU General Public
 * License.  See the file "COPYING" in the main directory of this archive
 * for more details.
 *
 * Copyright (C) 2009 Wind River Systems,
 *   written by Ralf Baechle <ralf@linux-mips.org>
 */
#include <linux/init.h>
#include <linux/irqflags.h>
#include <linux/notifier.h>
#include <linux/prefetch.h>
#include <linux/ptrace.h>
#include <linux/sched.h>
#include <linux/sched/task_stack.h>

#include <asm/cop2.h>
#include <asm/current.h>
#include <asm/mipsregs.h>
#include <asm/page.h>
#include <asm/octeon/octeon.h>

static int cnmips_cu2_call(struct notifier_block *nfb, unsigned long action,
        void *data)
{
        unsigned long flags;
        unsigned int status;

        switch (action) {
        case CU2_EXCEPTION:
                prefetch(&current->thread.cp2);
                local_irq_save(flags);
                KSTK_STATUS(current) |= ST0_CU2;
                status = read_c0_status();
                write_c0_status(status | ST0_CU2);
                octeon_cop2_restore(&(current->thread.cp2));
                write_c0_status(status & ~ST0_CU2);
                local_irq_restore(flags);

                return NOTIFY_BAD;      /* Don't call default notifier */
        }

        return NOTIFY_OK;               /* Let default notifier send signals */
}

static int __init cnmips_cu2_setup(void)
{
        return cu2_notifier(cnmips_cu2_call, 0);
}
early_initcall(cnmips_cu2_setup);