/*
 * Copyright 2018 Red Hat Inc.
 *
 * Permission is hereby granted, free of charge, to any person obtaining a
 * copy of this software and associated documentation files (the "Software"),
 * to deal in the Software without restriction, including without limitation
 * the rights to use, copy, modify, merge, publish, distribute, sublicense,
 * and/or sell copies of the Software, and to permit persons to whom the
 * Software is furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
 * THE COPYRIGHT HOLDER(S) OR AUTHOR(S) BE LIABLE FOR ANY CLAIM, DAMAGES OR
 * OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
 * ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
 * OTHER DEALINGS IN THE SOFTWARE.
 */
#include "priv.h"

#include <core/memory.h>
#include <subdev/mmu.h>
#include <engine/fifo.h>

#include <nvif/class.h>

void
gv100_fault_buffer_process(struct work_struct *work)
{
        struct nvkm_fault *fault = container_of(work, typeof(*fault), nrpfb_work);
        struct nvkm_fault_buffer *buffer = fault->buffer[0];
        struct nvkm_device *device = fault->subdev.device;
        struct nvkm_memory *mem = buffer->mem;
        u32 get = nvkm_rd32(device, buffer->get);
        u32 put = nvkm_rd32(device, buffer->put);
        if (put == get)
                return;

        nvkm_kmap(mem);
        while (get != put) {
                const u32   base = get * buffer->fault->func->buffer.entry_size;
                const u32 instlo = nvkm_ro32(mem, base + 0x00);
                const u32 insthi = nvkm_ro32(mem, base + 0x04);
                const u32 addrlo = nvkm_ro32(mem, base + 0x08);
                const u32 addrhi = nvkm_ro32(mem, base + 0x0c);
                const u32 timelo = nvkm_ro32(mem, base + 0x10);
                const u32 timehi = nvkm_ro32(mem, base + 0x14);
                const u32  info0 = nvkm_ro32(mem, base + 0x18);
                const u32  info1 = nvkm_ro32(mem, base + 0x1c);
                struct nvkm_fault_data info;

                if (++get == buffer->entries)
                        get = 0;
                nvkm_wr32(device, buffer->get, get);

                info.addr   = ((u64)addrhi << 32) | addrlo;
                info.inst   = ((u64)insthi << 32) | instlo;
                info.time   = ((u64)timehi << 32) | timelo;
                info.engine = (info0 & 0x000000ff);
                info.valid  = (info1 & 0x80000000) >> 31;
                info.gpc    = (info1 & 0x1f000000) >> 24;
                info.hub    = (info1 & 0x00100000) >> 20;
                info.access = (info1 & 0x000f0000) >> 16;
                info.client = (info1 & 0x00007f00) >> 8;
                info.reason = (info1 & 0x0000001f);

                nvkm_fifo_fault(device->fifo, &info);
        }
        nvkm_done(mem);
}

static void
gv100_fault_buffer_intr(struct nvkm_fault_buffer *buffer, bool enable)
{
        struct nvkm_device *device = buffer->fault->subdev.device;
        const u32 intr = buffer->id ? 0x08000000 : 0x20000000;
        if (enable)
                nvkm_mask(device, 0x100a2c, intr, intr);
        else
                nvkm_mask(device, 0x100a34, intr, intr);
}

static void
gv100_fault_buffer_fini(struct nvkm_fault_buffer *buffer)
{
        struct nvkm_device *device = buffer->fault->subdev.device;
        const u32 foff = buffer->id * 0x14;
        nvkm_mask(device, 0x100e34 + foff, 0x80000000, 0x00000000);
}

static void
gv100_fault_buffer_init(struct nvkm_fault_buffer *buffer)
{
        struct nvkm_device *device = buffer->fault->subdev.device;
        const u32 foff = buffer->id * 0x14;

        nvkm_mask(device, 0x100e34 + foff, 0xc0000000, 0x40000000);
        nvkm_wr32(device, 0x100e28 + foff, upper_32_bits(buffer->addr));
        nvkm_wr32(device, 0x100e24 + foff, lower_32_bits(buffer->addr));
        nvkm_mask(device, 0x100e34 + foff, 0x80000000, 0x80000000);
}

static void
gv100_fault_buffer_info(struct nvkm_fault_buffer *buffer)
{
        struct nvkm_device *device = buffer->fault->subdev.device;
        const u32 foff = buffer->id * 0x14;

        nvkm_mask(device, 0x100e34 + foff, 0x40000000, 0x40000000);

        buffer->entries = nvkm_rd32(device, 0x100e34 + foff) & 0x000fffff;
        buffer->get = 0x100e2c + foff;
        buffer->put = 0x100e30 + foff;
}

static int
gv100_fault_ntfy_nrpfb(struct nvkm_event_ntfy *ntfy, u32 bits)
{
        struct nvkm_fault *fault = container_of(ntfy, typeof(*fault), nrpfb);

        schedule_work(&fault->nrpfb_work);
        return NVKM_EVENT_KEEP;
}

static void
gv100_fault_intr_fault(struct nvkm_fault *fault)
{
        struct nvkm_subdev *subdev = &fault->subdev;
        struct nvkm_device *device = subdev->device;
        struct nvkm_fault_data info;
        const u32 addrlo = nvkm_rd32(device, 0x100e4c);
        const u32 addrhi = nvkm_rd32(device, 0x100e50);
        const u32  info0 = nvkm_rd32(device, 0x100e54);
        const u32 insthi = nvkm_rd32(device, 0x100e58);
        const u32  info1 = nvkm_rd32(device, 0x100e5c);

        info.addr = ((u64)addrhi << 32) | addrlo;
        info.inst = ((u64)insthi << 32) | (info0 & 0xfffff000);
        info.time = 0;
        info.engine = (info0 & 0x000000ff);
        info.valid  = (info1 & 0x80000000) >> 31;
        info.gpc    = (info1 & 0x1f000000) >> 24;
        info.hub    = (info1 & 0x00100000) >> 20;
        info.access = (info1 & 0x000f0000) >> 16;
        info.client = (info1 & 0x00007f00) >> 8;
        info.reason = (info1 & 0x0000001f);

        nvkm_fifo_fault(device->fifo, &info);
}

static void
gv100_fault_intr(struct nvkm_fault *fault)
{
        struct nvkm_subdev *subdev = &fault->subdev;
        struct nvkm_device *device = subdev->device;
        u32 stat = nvkm_rd32(device, 0x100a20);

        if (stat & 0x80000000) {
                gv100_fault_intr_fault(fault);
                nvkm_wr32(device, 0x100e60, 0x80000000);
                stat &= ~0x80000000;
        }

        if (stat & 0x20000000) {
                if (fault->buffer[0]) {
                        nvkm_event_ntfy(&fault->event, 0, NVKM_FAULT_BUFFER_EVENT_PENDING);
                        stat &= ~0x20000000;
                }
        }

        if (stat & 0x08000000) {
                if (fault->buffer[1]) {
                        nvkm_event_ntfy(&fault->event, 1, NVKM_FAULT_BUFFER_EVENT_PENDING);
                        stat &= ~0x08000000;
                }
        }

        if (stat) {
                nvkm_debug(subdev, "intr %08x\n", stat);
        }
}

static void
gv100_fault_fini(struct nvkm_fault *fault)
{
        nvkm_event_ntfy_block(&fault->nrpfb);
        flush_work(&fault->nrpfb_work);

        if (fault->buffer[0])
                fault->func->buffer.fini(fault->buffer[0]);

        nvkm_mask(fault->subdev.device, 0x100a34, 0x80000000, 0x80000000);
}

static void
gv100_fault_init(struct nvkm_fault *fault)
{
        nvkm_mask(fault->subdev.device, 0x100a2c, 0x80000000, 0x80000000);
        fault->func->buffer.init(fault->buffer[0]);
        nvkm_event_ntfy_allow(&fault->nrpfb);
}

int
gv100_fault_oneinit(struct nvkm_fault *fault)
{
        nvkm_event_ntfy_add(&fault->event, 0, NVKM_FAULT_BUFFER_EVENT_PENDING, true,
                            gv100_fault_ntfy_nrpfb, &fault->nrpfb);
        return 0;
}

static const struct nvkm_fault_func
gv100_fault = {
        .oneinit = gv100_fault_oneinit,
        .init = gv100_fault_init,
        .fini = gv100_fault_fini,
        .intr = gv100_fault_intr,
        .buffer.nr = 2,
        .buffer.entry_size = 32,
        .buffer.info = gv100_fault_buffer_info,
        .buffer.pin = gp100_fault_buffer_pin,
        .buffer.init = gv100_fault_buffer_init,
        .buffer.fini = gv100_fault_buffer_fini,
        .buffer.intr = gv100_fault_buffer_intr,
        /*TODO: Figure out how to expose non-replayable fault buffer, which,
         *      for some reason, is where recoverable CE faults appear...
         *
         *      It's a bit tricky, as both NVKM and SVM will need access to
         *      the non-replayable fault buffer.
         */
        .user = { { 0, 0, VOLTA_FAULT_BUFFER_A }, 1 },
};

int
gv100_fault_new(struct nvkm_device *device, enum nvkm_subdev_type type, int inst,
                struct nvkm_fault **pfault)
{
        int ret = nvkm_fault_new_(&gv100_fault, device, type, inst, pfault);
        if (ret)
                return ret;

        INIT_WORK(&(*pfault)->nrpfb_work, gv100_fault_buffer_process);
        return 0;
}