/*
 * Copyright 2008 Advanced Micro Devices, 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.
 *
 * Author: Stanislaw Skowronek
 */

#include <linux/module.h>
#include <linux/sched.h>
#include <linux/slab.h>
#include <linux/string_helpers.h>

#include <linux/unaligned.h>

#include <drm/drm_device.h>
#include <drm/drm_util.h>

#define ATOM_DEBUG

#include "atom.h"
#include "atom-names.h"
#include "atom-bits.h"
#include "radeon.h"

#define ATOM_COND_ABOVE         0
#define ATOM_COND_ABOVEOREQUAL  1
#define ATOM_COND_ALWAYS        2
#define ATOM_COND_BELOW         3
#define ATOM_COND_BELOWOREQUAL  4
#define ATOM_COND_EQUAL         5
#define ATOM_COND_NOTEQUAL      6

#define ATOM_PORT_ATI   0
#define ATOM_PORT_PCI   1
#define ATOM_PORT_SYSIO 2

#define ATOM_UNIT_MICROSEC      0
#define ATOM_UNIT_MILLISEC      1

#define PLL_INDEX       2
#define PLL_DATA        3

typedef struct {
        struct atom_context *ctx;
        uint32_t *ps, *ws;
        int ps_size, ws_size;
        int ps_shift;
        uint16_t start;
        unsigned last_jump;
        unsigned long last_jump_jiffies;
        bool abort;
} atom_exec_context;

int atom_debug = 0;
static int atom_execute_table_locked(struct atom_context *ctx, int index, uint32_t *params, int params_size);
int atom_execute_table(struct atom_context *ctx, int index, uint32_t *params, int params_size);

static uint32_t atom_arg_mask[8] = {
        0xFFFFFFFF, 0x0000FFFF, 0x00FFFF00, 0xFFFF0000,
        0x000000FF, 0x0000FF00, 0x00FF0000, 0xFF000000
};
static int atom_arg_shift[8] = { 0, 0, 8, 16, 0, 8, 16, 24 };

static int atom_dst_to_src[8][4] = {
        /* translate destination alignment field to the source alignment encoding */
        {0, 0, 0, 0},
        {1, 2, 3, 0},
        {1, 2, 3, 0},
        {1, 2, 3, 0},
        {4, 5, 6, 7},
        {4, 5, 6, 7},
        {4, 5, 6, 7},
        {4, 5, 6, 7},
};
static int atom_def_dst[8] = { 0, 0, 1, 2, 0, 1, 2, 3 };

static int debug_depth = 0;
#ifdef ATOM_DEBUG
static void debug_print_spaces(int n)
{
        while (n--)
                printk("   ");
}

#ifdef DEBUG
#undef DEBUG
#endif

#define DEBUG(...) do if (atom_debug) { printk(KERN_DEBUG __VA_ARGS__); } while (0)
#define SDEBUG(...) do if (atom_debug) { printk(KERN_DEBUG); debug_print_spaces(debug_depth); printk(__VA_ARGS__); } while (0)
#else
#define DEBUG(...) do { } while (0)
#define SDEBUG(...) do { } while (0)
#endif

static uint32_t atom_iio_execute(struct atom_context *ctx, int base,
                                 uint32_t index, uint32_t data)
{
        struct radeon_device *rdev = ctx->card->dev->dev_private;
        uint32_t temp = 0xCDCDCDCD;

        while (1)
                switch (CU8(base)) {
                case ATOM_IIO_NOP:
                        base++;
                        break;
                case ATOM_IIO_READ:
                        temp = ctx->card->ioreg_read(ctx->card, CU16(base + 1));
                        base += 3;
                        break;
                case ATOM_IIO_WRITE:
                        if (rdev->family == CHIP_RV515)
                                (void)ctx->card->ioreg_read(ctx->card, CU16(base + 1));
                        ctx->card->ioreg_write(ctx->card, CU16(base + 1), temp);
                        base += 3;
                        break;
                case ATOM_IIO_CLEAR:
                        temp &=
                            ~((0xFFFFFFFF >> (32 - CU8(base + 1))) <<
                              CU8(base + 2));
                        base += 3;
                        break;
                case ATOM_IIO_SET:
                        temp |=
                            (0xFFFFFFFF >> (32 - CU8(base + 1))) << CU8(base +
                                                                        2);
                        base += 3;
                        break;
                case ATOM_IIO_MOVE_INDEX:
                        temp &=
                            ~((0xFFFFFFFF >> (32 - CU8(base + 1))) <<
                              CU8(base + 3));
                        temp |=
                            ((index >> CU8(base + 2)) &
                             (0xFFFFFFFF >> (32 - CU8(base + 1)))) << CU8(base +
                                                                          3);
                        base += 4;
                        break;
                case ATOM_IIO_MOVE_DATA:
                        temp &=
                            ~((0xFFFFFFFF >> (32 - CU8(base + 1))) <<
                              CU8(base + 3));
                        temp |=
                            ((data >> CU8(base + 2)) &
                             (0xFFFFFFFF >> (32 - CU8(base + 1)))) << CU8(base +
                                                                          3);
                        base += 4;
                        break;
                case ATOM_IIO_MOVE_ATTR:
                        temp &=
                            ~((0xFFFFFFFF >> (32 - CU8(base + 1))) <<
                              CU8(base + 3));
                        temp |=
                            ((ctx->io_attr >> CU8(base + 2)) &
                             (0xFFFFFFFF >> (32 - CU8(base + 1)))) <<
                             CU8(base + 3);
                        base += 4;
                        break;
                case ATOM_IIO_END:
                        return temp;
                default:
                        pr_info("Unknown IIO opcode\n");
                        return 0;
                }
}

static uint32_t atom_get_src_int(atom_exec_context *ctx, uint8_t attr,
                                 int *ptr, uint32_t *saved, int print)
{
        uint32_t idx, val = 0xCDCDCDCD, align, arg;
        struct atom_context *gctx = ctx->ctx;
        arg = attr & 7;
        align = (attr >> 3) & 7;
        switch (arg) {
        case ATOM_ARG_REG:
                idx = U16(*ptr);
                (*ptr) += 2;
                if (print)
                        DEBUG("REG[0x%04X]", idx);
                idx += gctx->reg_block;
                switch (gctx->io_mode) {
                case ATOM_IO_MM:
                        val = gctx->card->reg_read(gctx->card, idx);
                        break;
                case ATOM_IO_PCI:
                        pr_info("PCI registers are not implemented\n");
                        return 0;
                case ATOM_IO_SYSIO:
                        pr_info("SYSIO registers are not implemented\n");
                        return 0;
                default:
                        if (!(gctx->io_mode & 0x80)) {
                                pr_info("Bad IO mode\n");
                                return 0;
                        }
                        if (!gctx->iio[gctx->io_mode & 0x7F]) {
                                pr_info("Undefined indirect IO read method %d\n",
                                        gctx->io_mode & 0x7F);
                                return 0;
                        }
                        val =
                            atom_iio_execute(gctx,
                                             gctx->iio[gctx->io_mode & 0x7F],
                                             idx, 0);
                }
                break;
        case ATOM_ARG_PS:
                idx = U8(*ptr);
                (*ptr)++;
                /* get_unaligned_le32 avoids unaligned accesses from atombios
                 * tables, noticed on a DEC Alpha. */
                if (idx < ctx->ps_size)
                        val = get_unaligned_le32((u32 *)&ctx->ps[idx]);
                else
                        pr_info("PS index out of range: %i > %i\n", idx, ctx->ps_size);
                if (print)
                        DEBUG("PS[0x%02X,0x%04X]", idx, val);
                break;
        case ATOM_ARG_WS:
                idx = U8(*ptr);
                (*ptr)++;
                if (print)
                        DEBUG("WS[0x%02X]", idx);
                switch (idx) {
                case ATOM_WS_QUOTIENT:
                        val = gctx->divmul[0];
                        break;
                case ATOM_WS_REMAINDER:
                        val = gctx->divmul[1];
                        break;
                case ATOM_WS_DATAPTR:
                        val = gctx->data_block;
                        break;
                case ATOM_WS_SHIFT:
                        val = gctx->shift;
                        break;
                case ATOM_WS_OR_MASK:
                        val = 1 << gctx->shift;
                        break;
                case ATOM_WS_AND_MASK:
                        val = ~(1 << gctx->shift);
                        break;
                case ATOM_WS_FB_WINDOW:
                        val = gctx->fb_base;
                        break;
                case ATOM_WS_ATTRIBUTES:
                        val = gctx->io_attr;
                        break;
                case ATOM_WS_REGPTR:
                        val = gctx->reg_block;
                        break;
                default:
                        if (idx < ctx->ws_size)
                                val = ctx->ws[idx];
                        else
                                pr_info("WS index out of range: %i > %i\n", idx, ctx->ws_size);
                }
                break;
        case ATOM_ARG_ID:
                idx = U16(*ptr);
                (*ptr) += 2;
                if (print) {
                        if (gctx->data_block)
                                DEBUG("ID[0x%04X+%04X]", idx, gctx->data_block);
                        else
                                DEBUG("ID[0x%04X]", idx);
                }
                val = U32(idx + gctx->data_block);
                break;
        case ATOM_ARG_FB:
                idx = U8(*ptr);
                (*ptr)++;
                if ((gctx->fb_base + (idx * 4)) > gctx->scratch_size_bytes) {
                        DRM_ERROR("ATOM: fb read beyond scratch region: %d vs. %d\n",
                                  gctx->fb_base + (idx * 4), gctx->scratch_size_bytes);
                        val = 0;
                } else
                        val = gctx->scratch[(gctx->fb_base / 4) + idx];
                if (print)
                        DEBUG("FB[0x%02X]", idx);
                break;
        case ATOM_ARG_IMM:
                switch (align) {
                case ATOM_SRC_DWORD:
                        val = U32(*ptr);
                        (*ptr) += 4;
                        if (print)
                                DEBUG("IMM 0x%08X\n", val);
                        return val;
                case ATOM_SRC_WORD0:
                case ATOM_SRC_WORD8:
                case ATOM_SRC_WORD16:
                        val = U16(*ptr);
                        (*ptr) += 2;
                        if (print)
                                DEBUG("IMM 0x%04X\n", val);
                        return val;
                case ATOM_SRC_BYTE0:
                case ATOM_SRC_BYTE8:
                case ATOM_SRC_BYTE16:
                case ATOM_SRC_BYTE24:
                        val = U8(*ptr);
                        (*ptr)++;
                        if (print)
                                DEBUG("IMM 0x%02X\n", val);
                        return val;
                }
                return 0;
        case ATOM_ARG_PLL:
                idx = U8(*ptr);
                (*ptr)++;
                if (print)
                        DEBUG("PLL[0x%02X]", idx);
                val = gctx->card->pll_read(gctx->card, idx);
                break;
        case ATOM_ARG_MC:
                idx = U8(*ptr);
                (*ptr)++;
                if (print)
                        DEBUG("MC[0x%02X]", idx);
                val = gctx->card->mc_read(gctx->card, idx);
                break;
        }
        if (saved)
                *saved = val;
        val &= atom_arg_mask[align];
        val >>= atom_arg_shift[align];
        if (print)
                switch (align) {
                case ATOM_SRC_DWORD:
                        DEBUG(".[31:0] -> 0x%08X\n", val);
                        break;
                case ATOM_SRC_WORD0:
                        DEBUG(".[15:0] -> 0x%04X\n", val);
                        break;
                case ATOM_SRC_WORD8:
                        DEBUG(".[23:8] -> 0x%04X\n", val);
                        break;
                case ATOM_SRC_WORD16:
                        DEBUG(".[31:16] -> 0x%04X\n", val);
                        break;
                case ATOM_SRC_BYTE0:
                        DEBUG(".[7:0] -> 0x%02X\n", val);
                        break;
                case ATOM_SRC_BYTE8:
                        DEBUG(".[15:8] -> 0x%02X\n", val);
                        break;
                case ATOM_SRC_BYTE16:
                        DEBUG(".[23:16] -> 0x%02X\n", val);
                        break;
                case ATOM_SRC_BYTE24:
                        DEBUG(".[31:24] -> 0x%02X\n", val);
                        break;
                }
        return val;
}

static void atom_skip_src_int(atom_exec_context *ctx, uint8_t attr, int *ptr)
{
        uint32_t align = (attr >> 3) & 7, arg = attr & 7;
        switch (arg) {
        case ATOM_ARG_REG:
        case ATOM_ARG_ID:
                (*ptr) += 2;
                break;
        case ATOM_ARG_PLL:
        case ATOM_ARG_MC:
        case ATOM_ARG_PS:
        case ATOM_ARG_WS:
        case ATOM_ARG_FB:
                (*ptr)++;
                break;
        case ATOM_ARG_IMM:
                switch (align) {
                case ATOM_SRC_DWORD:
                        (*ptr) += 4;
                        return;
                case ATOM_SRC_WORD0:
                case ATOM_SRC_WORD8:
                case ATOM_SRC_WORD16:
                        (*ptr) += 2;
                        return;
                case ATOM_SRC_BYTE0:
                case ATOM_SRC_BYTE8:
                case ATOM_SRC_BYTE16:
                case ATOM_SRC_BYTE24:
                        (*ptr)++;
                        return;
                }
                return;
        }
}

static uint32_t atom_get_src(atom_exec_context *ctx, uint8_t attr, int *ptr)
{
        return atom_get_src_int(ctx, attr, ptr, NULL, 1);
}

static uint32_t atom_get_src_direct(atom_exec_context *ctx, uint8_t align, int *ptr)
{
        uint32_t val = 0xCDCDCDCD;

        switch (align) {
        case ATOM_SRC_DWORD:
                val = U32(*ptr);
                (*ptr) += 4;
                break;
        case ATOM_SRC_WORD0:
        case ATOM_SRC_WORD8:
        case ATOM_SRC_WORD16:
                val = U16(*ptr);
                (*ptr) += 2;
                break;
        case ATOM_SRC_BYTE0:
        case ATOM_SRC_BYTE8:
        case ATOM_SRC_BYTE16:
        case ATOM_SRC_BYTE24:
                val = U8(*ptr);
                (*ptr)++;
                break;
        }
        return val;
}

static uint32_t atom_get_dst(atom_exec_context *ctx, int arg, uint8_t attr,
                             int *ptr, uint32_t *saved, int print)
{
        return atom_get_src_int(ctx,
                                arg | atom_dst_to_src[(attr >> 3) &
                                                      7][(attr >> 6) & 3] << 3,
                                ptr, saved, print);
}

static void atom_skip_dst(atom_exec_context *ctx, int arg, uint8_t attr, int *ptr)
{
        atom_skip_src_int(ctx,
                          arg | atom_dst_to_src[(attr >> 3) & 7][(attr >> 6) &
                                                                 3] << 3, ptr);
}

static void atom_put_dst(atom_exec_context *ctx, int arg, uint8_t attr,
                         int *ptr, uint32_t val, uint32_t saved)
{
        uint32_t align =
            atom_dst_to_src[(attr >> 3) & 7][(attr >> 6) & 3], old_val =
            val, idx;
        struct atom_context *gctx = ctx->ctx;
        old_val &= atom_arg_mask[align] >> atom_arg_shift[align];
        val <<= atom_arg_shift[align];
        val &= atom_arg_mask[align];
        saved &= ~atom_arg_mask[align];
        val |= saved;
        switch (arg) {
        case ATOM_ARG_REG:
                idx = U16(*ptr);
                (*ptr) += 2;
                DEBUG("REG[0x%04X]", idx);
                idx += gctx->reg_block;
                switch (gctx->io_mode) {
                case ATOM_IO_MM:
                        if (idx == 0)
                                gctx->card->reg_write(gctx->card, idx,
                                                      val << 2);
                        else
                                gctx->card->reg_write(gctx->card, idx, val);
                        break;
                case ATOM_IO_PCI:
                        pr_info("PCI registers are not implemented\n");
                        return;
                case ATOM_IO_SYSIO:
                        pr_info("SYSIO registers are not implemented\n");
                        return;
                default:
                        if (!(gctx->io_mode & 0x80)) {
                                pr_info("Bad IO mode\n");
                                return;
                        }
                        if (!gctx->iio[gctx->io_mode & 0xFF]) {
                                pr_info("Undefined indirect IO write method %d\n",
                                        gctx->io_mode & 0x7F);
                                return;
                        }
                        atom_iio_execute(gctx, gctx->iio[gctx->io_mode & 0xFF],
                                         idx, val);
                }
                break;
        case ATOM_ARG_PS:
                idx = U8(*ptr);
                (*ptr)++;
                DEBUG("PS[0x%02X]", idx);
                if (idx >= ctx->ps_size) {
                        pr_info("PS index out of range: %i > %i\n", idx, ctx->ps_size);
                        return;
                }
                ctx->ps[idx] = cpu_to_le32(val);
                break;
        case ATOM_ARG_WS:
                idx = U8(*ptr);
                (*ptr)++;
                DEBUG("WS[0x%02X]", idx);
                switch (idx) {
                case ATOM_WS_QUOTIENT:
                        gctx->divmul[0] = val;
                        break;
                case ATOM_WS_REMAINDER:
                        gctx->divmul[1] = val;
                        break;
                case ATOM_WS_DATAPTR:
                        gctx->data_block = val;
                        break;
                case ATOM_WS_SHIFT:
                        gctx->shift = val;
                        break;
                case ATOM_WS_OR_MASK:
                case ATOM_WS_AND_MASK:
                        break;
                case ATOM_WS_FB_WINDOW:
                        gctx->fb_base = val;
                        break;
                case ATOM_WS_ATTRIBUTES:
                        gctx->io_attr = val;
                        break;
                case ATOM_WS_REGPTR:
                        gctx->reg_block = val;
                        break;
                default:
                        if (idx >= ctx->ws_size) {
                                pr_info("WS index out of range: %i > %i\n", idx, ctx->ws_size);
                                return;
                        }
                        ctx->ws[idx] = val;
                }
                break;
        case ATOM_ARG_FB:
                idx = U8(*ptr);
                (*ptr)++;
                if ((gctx->fb_base + (idx * 4)) > gctx->scratch_size_bytes) {
                        DRM_ERROR("ATOM: fb write beyond scratch region: %d vs. %d\n",
                                  gctx->fb_base + (idx * 4), gctx->scratch_size_bytes);
                } else
                        gctx->scratch[(gctx->fb_base / 4) + idx] = val;
                DEBUG("FB[0x%02X]", idx);
                break;
        case ATOM_ARG_PLL:
                idx = U8(*ptr);
                (*ptr)++;
                DEBUG("PLL[0x%02X]", idx);
                gctx->card->pll_write(gctx->card, idx, val);
                break;
        case ATOM_ARG_MC:
                idx = U8(*ptr);
                (*ptr)++;
                DEBUG("MC[0x%02X]", idx);
                gctx->card->mc_write(gctx->card, idx, val);
                return;
        }
        switch (align) {
        case ATOM_SRC_DWORD:
                DEBUG(".[31:0] <- 0x%08X\n", old_val);
                break;
        case ATOM_SRC_WORD0:
                DEBUG(".[15:0] <- 0x%04X\n", old_val);
                break;
        case ATOM_SRC_WORD8:
                DEBUG(".[23:8] <- 0x%04X\n", old_val);
                break;
        case ATOM_SRC_WORD16:
                DEBUG(".[31:16] <- 0x%04X\n", old_val);
                break;
        case ATOM_SRC_BYTE0:
                DEBUG(".[7:0] <- 0x%02X\n", old_val);
                break;
        case ATOM_SRC_BYTE8:
                DEBUG(".[15:8] <- 0x%02X\n", old_val);
                break;
        case ATOM_SRC_BYTE16:
                DEBUG(".[23:16] <- 0x%02X\n", old_val);
                break;
        case ATOM_SRC_BYTE24:
                DEBUG(".[31:24] <- 0x%02X\n", old_val);
                break;
        }
}

static void atom_op_add(atom_exec_context *ctx, int *ptr, int arg)
{
        uint8_t attr = U8((*ptr)++);
        uint32_t dst, src, saved;
        int dptr = *ptr;
        SDEBUG("   dst: ");
        dst = atom_get_dst(ctx, arg, attr, ptr, &saved, 1);
        SDEBUG("   src: ");
        src = atom_get_src(ctx, attr, ptr);
        dst += src;
        SDEBUG("   dst: ");
        atom_put_dst(ctx, arg, attr, &dptr, dst, saved);
}

static void atom_op_and(atom_exec_context *ctx, int *ptr, int arg)
{
        uint8_t attr = U8((*ptr)++);
        uint32_t dst, src, saved;
        int dptr = *ptr;
        SDEBUG("   dst: ");
        dst = atom_get_dst(ctx, arg, attr, ptr, &saved, 1);
        SDEBUG("   src: ");
        src = atom_get_src(ctx, attr, ptr);
        dst &= src;
        SDEBUG("   dst: ");
        atom_put_dst(ctx, arg, attr, &dptr, dst, saved);
}

static void atom_op_beep(atom_exec_context *ctx, int *ptr, int arg)
{
        printk("ATOM BIOS beeped!\n");
}

static void atom_op_calltable(atom_exec_context *ctx, int *ptr, int arg)
{
        int idx = U8((*ptr)++);
        int r = 0;

        if (idx < ATOM_TABLE_NAMES_CNT)
                SDEBUG("   table: %d (%s)\n", idx, atom_table_names[idx]);
        else
                SDEBUG("   table: %d\n", idx);
        if (U16(ctx->ctx->cmd_table + 4 + 2 * idx))
                r = atom_execute_table_locked(ctx->ctx, idx, ctx->ps + ctx->ps_shift, ctx->ps_size - ctx->ps_shift);
        if (r) {
                ctx->abort = true;
        }
}

static void atom_op_clear(atom_exec_context *ctx, int *ptr, int arg)
{
        uint8_t attr = U8((*ptr)++);
        uint32_t saved;
        int dptr = *ptr;
        attr &= 0x38;
        attr |= atom_def_dst[attr >> 3] << 6;
        atom_get_dst(ctx, arg, attr, ptr, &saved, 0);
        SDEBUG("   dst: ");
        atom_put_dst(ctx, arg, attr, &dptr, 0, saved);
}

static void atom_op_compare(atom_exec_context *ctx, int *ptr, int arg)
{
        uint8_t attr = U8((*ptr)++);
        uint32_t dst, src;
        SDEBUG("   src1: ");
        dst = atom_get_dst(ctx, arg, attr, ptr, NULL, 1);
        SDEBUG("   src2: ");
        src = atom_get_src(ctx, attr, ptr);
        ctx->ctx->cs_equal = (dst == src);
        ctx->ctx->cs_above = (dst > src);
        SDEBUG("   result: %s %s\n", ctx->ctx->cs_equal ? "EQ" : "NE",
               ctx->ctx->cs_above ? "GT" : "LE");
}

static void atom_op_delay(atom_exec_context *ctx, int *ptr, int arg)
{
        unsigned count = U8((*ptr)++);
        SDEBUG("   count: %d\n", count);
        if (arg == ATOM_UNIT_MICROSEC)
                udelay(count);
        else if (!drm_can_sleep())
                mdelay(count);
        else
                drm_msleep(count);
}

static void atom_op_div(atom_exec_context *ctx, int *ptr, int arg)
{
        uint8_t attr = U8((*ptr)++);
        uint32_t dst, src;
        SDEBUG("   src1: ");
        dst = atom_get_dst(ctx, arg, attr, ptr, NULL, 1);
        SDEBUG("   src2: ");
        src = atom_get_src(ctx, attr, ptr);
        if (src != 0) {
                ctx->ctx->divmul[0] = dst / src;
                ctx->ctx->divmul[1] = dst % src;
        } else {
                ctx->ctx->divmul[0] = 0;
                ctx->ctx->divmul[1] = 0;
        }
}

static void atom_op_eot(atom_exec_context *ctx, int *ptr, int arg)
{
        /* functionally, a nop */
}

static void atom_op_jump(atom_exec_context *ctx, int *ptr, int arg)
{
        int execute = 0, target = U16(*ptr);
        unsigned long cjiffies;

        (*ptr) += 2;
        switch (arg) {
        case ATOM_COND_ABOVE:
                execute = ctx->ctx->cs_above;
                break;
        case ATOM_COND_ABOVEOREQUAL:
                execute = ctx->ctx->cs_above || ctx->ctx->cs_equal;
                break;
        case ATOM_COND_ALWAYS:
                execute = 1;
                break;
        case ATOM_COND_BELOW:
                execute = !(ctx->ctx->cs_above || ctx->ctx->cs_equal);
                break;
        case ATOM_COND_BELOWOREQUAL:
                execute = !ctx->ctx->cs_above;
                break;
        case ATOM_COND_EQUAL:
                execute = ctx->ctx->cs_equal;
                break;
        case ATOM_COND_NOTEQUAL:
                execute = !ctx->ctx->cs_equal;
                break;
        }
        if (arg != ATOM_COND_ALWAYS)
                SDEBUG("   taken: %s\n", str_yes_no(execute));
        SDEBUG("   target: 0x%04X\n", target);
        if (execute) {
                if (ctx->last_jump == (ctx->start + target)) {
                        cjiffies = jiffies;
                        if (time_after(cjiffies, ctx->last_jump_jiffies)) {
                                cjiffies -= ctx->last_jump_jiffies;
                                if ((jiffies_to_msecs(cjiffies) > 5000)) {
                                        DRM_ERROR("atombios stuck in loop for more than 5secs aborting\n");
                                        ctx->abort = true;
                                }
                        } else {
                                /* jiffies wrap around we will just wait a little longer */
                                ctx->last_jump_jiffies = jiffies;
                        }
                } else {
                        ctx->last_jump = ctx->start + target;
                        ctx->last_jump_jiffies = jiffies;
                }
                *ptr = ctx->start + target;
        }
}

static void atom_op_mask(atom_exec_context *ctx, int *ptr, int arg)
{
        uint8_t attr = U8((*ptr)++);
        uint32_t dst, mask, src, saved;
        int dptr = *ptr;
        SDEBUG("   dst: ");
        dst = atom_get_dst(ctx, arg, attr, ptr, &saved, 1);
        mask = atom_get_src_direct(ctx, ((attr >> 3) & 7), ptr);
        SDEBUG("   mask: 0x%08x", mask);
        SDEBUG("   src: ");
        src = atom_get_src(ctx, attr, ptr);
        dst &= mask;
        dst |= src;
        SDEBUG("   dst: ");
        atom_put_dst(ctx, arg, attr, &dptr, dst, saved);
}

static void atom_op_move(atom_exec_context *ctx, int *ptr, int arg)
{
        uint8_t attr = U8((*ptr)++);
        uint32_t src, saved;
        int dptr = *ptr;
        if (((attr >> 3) & 7) != ATOM_SRC_DWORD)
                atom_get_dst(ctx, arg, attr, ptr, &saved, 0);
        else {
                atom_skip_dst(ctx, arg, attr, ptr);
                saved = 0xCDCDCDCD;
        }
        SDEBUG("   src: ");
        src = atom_get_src(ctx, attr, ptr);
        SDEBUG("   dst: ");
        atom_put_dst(ctx, arg, attr, &dptr, src, saved);
}

static void atom_op_mul(atom_exec_context *ctx, int *ptr, int arg)
{
        uint8_t attr = U8((*ptr)++);
        uint32_t dst, src;
        SDEBUG("   src1: ");
        dst = atom_get_dst(ctx, arg, attr, ptr, NULL, 1);
        SDEBUG("   src2: ");
        src = atom_get_src(ctx, attr, ptr);
        ctx->ctx->divmul[0] = dst * src;
}

static void atom_op_nop(atom_exec_context *ctx, int *ptr, int arg)
{
        /* nothing */
}

static void atom_op_or(atom_exec_context *ctx, int *ptr, int arg)
{
        uint8_t attr = U8((*ptr)++);
        uint32_t dst, src, saved;
        int dptr = *ptr;
        SDEBUG("   dst: ");
        dst = atom_get_dst(ctx, arg, attr, ptr, &saved, 1);
        SDEBUG("   src: ");
        src = atom_get_src(ctx, attr, ptr);
        dst |= src;
        SDEBUG("   dst: ");
        atom_put_dst(ctx, arg, attr, &dptr, dst, saved);
}

static void atom_op_postcard(atom_exec_context *ctx, int *ptr, int arg)
{
        uint8_t val = U8((*ptr)++);
        SDEBUG("POST card output: 0x%02X\n", val);
}

static void atom_op_repeat(atom_exec_context *ctx, int *ptr, int arg)
{
        pr_info("unimplemented!\n");
}

static void atom_op_restorereg(atom_exec_context *ctx, int *ptr, int arg)
{
        pr_info("unimplemented!\n");
}

static void atom_op_savereg(atom_exec_context *ctx, int *ptr, int arg)
{
        pr_info("unimplemented!\n");
}

static void atom_op_setdatablock(atom_exec_context *ctx, int *ptr, int arg)
{
        int idx = U8(*ptr);
        (*ptr)++;
        SDEBUG("   block: %d\n", idx);
        if (!idx)
                ctx->ctx->data_block = 0;
        else if (idx == 255)
                ctx->ctx->data_block = ctx->start;
        else
                ctx->ctx->data_block = U16(ctx->ctx->data_table + 4 + 2 * idx);
        SDEBUG("   base: 0x%04X\n", ctx->ctx->data_block);
}

static void atom_op_setfbbase(atom_exec_context *ctx, int *ptr, int arg)
{
        uint8_t attr = U8((*ptr)++);
        SDEBUG("   fb_base: ");
        ctx->ctx->fb_base = atom_get_src(ctx, attr, ptr);
}

static void atom_op_setport(atom_exec_context *ctx, int *ptr, int arg)
{
        int port;
        switch (arg) {
        case ATOM_PORT_ATI:
                port = U16(*ptr);
                if (port < ATOM_IO_NAMES_CNT)
                        SDEBUG("   port: %d (%s)\n", port, atom_io_names[port]);
                else
                        SDEBUG("   port: %d\n", port);
                if (!port)
                        ctx->ctx->io_mode = ATOM_IO_MM;
                else
                        ctx->ctx->io_mode = ATOM_IO_IIO | port;
                (*ptr) += 2;
                break;
        case ATOM_PORT_PCI:
                ctx->ctx->io_mode = ATOM_IO_PCI;
                (*ptr)++;
                break;
        case ATOM_PORT_SYSIO:
                ctx->ctx->io_mode = ATOM_IO_SYSIO;
                (*ptr)++;
                break;
        }
}

static void atom_op_setregblock(atom_exec_context *ctx, int *ptr, int arg)
{
        ctx->ctx->reg_block = U16(*ptr);
        (*ptr) += 2;
        SDEBUG("   base: 0x%04X\n", ctx->ctx->reg_block);
}

static void atom_op_shift_left(atom_exec_context *ctx, int *ptr, int arg)
{
        uint8_t attr = U8((*ptr)++), shift;
        uint32_t saved, dst;
        int dptr = *ptr;
        attr &= 0x38;
        attr |= atom_def_dst[attr >> 3] << 6;
        SDEBUG("   dst: ");
        dst = atom_get_dst(ctx, arg, attr, ptr, &saved, 1);
        shift = atom_get_src_direct(ctx, ATOM_SRC_BYTE0, ptr);
        SDEBUG("   shift: %d\n", shift);
        dst <<= shift;
        SDEBUG("   dst: ");
        atom_put_dst(ctx, arg, attr, &dptr, dst, saved);
}

static void atom_op_shift_right(atom_exec_context *ctx, int *ptr, int arg)
{
        uint8_t attr = U8((*ptr)++), shift;
        uint32_t saved, dst;
        int dptr = *ptr;
        attr &= 0x38;
        attr |= atom_def_dst[attr >> 3] << 6;
        SDEBUG("   dst: ");
        dst = atom_get_dst(ctx, arg, attr, ptr, &saved, 1);
        shift = atom_get_src_direct(ctx, ATOM_SRC_BYTE0, ptr);
        SDEBUG("   shift: %d\n", shift);
        dst >>= shift;
        SDEBUG("   dst: ");
        atom_put_dst(ctx, arg, attr, &dptr, dst, saved);
}

static void atom_op_shl(atom_exec_context *ctx, int *ptr, int arg)
{
        uint8_t attr = U8((*ptr)++), shift;
        uint32_t saved, dst;
        int dptr = *ptr;
        uint32_t dst_align = atom_dst_to_src[(attr >> 3) & 7][(attr >> 6) & 3];
        SDEBUG("   dst: ");
        dst = atom_get_dst(ctx, arg, attr, ptr, &saved, 1);
        /* op needs to full dst value */
        dst = saved;
        shift = atom_get_src(ctx, attr, ptr);
        SDEBUG("   shift: %d\n", shift);
        dst <<= shift;
        dst &= atom_arg_mask[dst_align];
        dst >>= atom_arg_shift[dst_align];
        SDEBUG("   dst: ");
        atom_put_dst(ctx, arg, attr, &dptr, dst, saved);
}

static void atom_op_shr(atom_exec_context *ctx, int *ptr, int arg)
{
        uint8_t attr = U8((*ptr)++), shift;
        uint32_t saved, dst;
        int dptr = *ptr;
        uint32_t dst_align = atom_dst_to_src[(attr >> 3) & 7][(attr >> 6) & 3];
        SDEBUG("   dst: ");
        dst = atom_get_dst(ctx, arg, attr, ptr, &saved, 1);
        /* op needs to full dst value */
        dst = saved;
        shift = atom_get_src(ctx, attr, ptr);
        SDEBUG("   shift: %d\n", shift);
        dst >>= shift;
        dst &= atom_arg_mask[dst_align];
        dst >>= atom_arg_shift[dst_align];
        SDEBUG("   dst: ");
        atom_put_dst(ctx, arg, attr, &dptr, dst, saved);
}

static void atom_op_sub(atom_exec_context *ctx, int *ptr, int arg)
{
        uint8_t attr = U8((*ptr)++);
        uint32_t dst, src, saved;
        int dptr = *ptr;
        SDEBUG("   dst: ");
        dst = atom_get_dst(ctx, arg, attr, ptr, &saved, 1);
        SDEBUG("   src: ");
        src = atom_get_src(ctx, attr, ptr);
        dst -= src;
        SDEBUG("   dst: ");
        atom_put_dst(ctx, arg, attr, &dptr, dst, saved);
}

static void atom_op_switch(atom_exec_context *ctx, int *ptr, int arg)
{
        uint8_t attr = U8((*ptr)++);
        uint32_t src, val, target;
        SDEBUG("   switch: ");
        src = atom_get_src(ctx, attr, ptr);
        while (U16(*ptr) != ATOM_CASE_END)
                if (U8(*ptr) == ATOM_CASE_MAGIC) {
                        (*ptr)++;
                        SDEBUG("   case: ");
                        val =
                            atom_get_src(ctx, (attr & 0x38) | ATOM_ARG_IMM,
                                         ptr);
                        target = U16(*ptr);
                        if (val == src) {
                                SDEBUG("   target: %04X\n", target);
                                *ptr = ctx->start + target;
                                return;
                        }
                        (*ptr) += 2;
                } else {
                        pr_info("Bad case\n");
                        return;
                }
        (*ptr) += 2;
}

static void atom_op_test(atom_exec_context *ctx, int *ptr, int arg)
{
        uint8_t attr = U8((*ptr)++);
        uint32_t dst, src;
        SDEBUG("   src1: ");
        dst = atom_get_dst(ctx, arg, attr, ptr, NULL, 1);
        SDEBUG("   src2: ");
        src = atom_get_src(ctx, attr, ptr);
        ctx->ctx->cs_equal = ((dst & src) == 0);
        SDEBUG("   result: %s\n", ctx->ctx->cs_equal ? "EQ" : "NE");
}

static void atom_op_xor(atom_exec_context *ctx, int *ptr, int arg)
{
        uint8_t attr = U8((*ptr)++);
        uint32_t dst, src, saved;
        int dptr = *ptr;
        SDEBUG("   dst: ");
        dst = atom_get_dst(ctx, arg, attr, ptr, &saved, 1);
        SDEBUG("   src: ");
        src = atom_get_src(ctx, attr, ptr);
        dst ^= src;
        SDEBUG("   dst: ");
        atom_put_dst(ctx, arg, attr, &dptr, dst, saved);
}

static void atom_op_debug(atom_exec_context *ctx, int *ptr, int arg)
{
        pr_info("unimplemented!\n");
}

static struct {
        void (*func) (atom_exec_context *, int *, int);
        int arg;
} opcode_table[ATOM_OP_CNT] = {
        {
        NULL, 0}, {
        atom_op_move, ATOM_ARG_REG}, {
        atom_op_move, ATOM_ARG_PS}, {
        atom_op_move, ATOM_ARG_WS}, {
        atom_op_move, ATOM_ARG_FB}, {
        atom_op_move, ATOM_ARG_PLL}, {
        atom_op_move, ATOM_ARG_MC}, {
        atom_op_and, ATOM_ARG_REG}, {
        atom_op_and, ATOM_ARG_PS}, {
        atom_op_and, ATOM_ARG_WS}, {
        atom_op_and, ATOM_ARG_FB}, {
        atom_op_and, ATOM_ARG_PLL}, {
        atom_op_and, ATOM_ARG_MC}, {
        atom_op_or, ATOM_ARG_REG}, {
        atom_op_or, ATOM_ARG_PS}, {
        atom_op_or, ATOM_ARG_WS}, {
        atom_op_or, ATOM_ARG_FB}, {
        atom_op_or, ATOM_ARG_PLL}, {
        atom_op_or, ATOM_ARG_MC}, {
        atom_op_shift_left, ATOM_ARG_REG}, {
        atom_op_shift_left, ATOM_ARG_PS}, {
        atom_op_shift_left, ATOM_ARG_WS}, {
        atom_op_shift_left, ATOM_ARG_FB}, {
        atom_op_shift_left, ATOM_ARG_PLL}, {
        atom_op_shift_left, ATOM_ARG_MC}, {
        atom_op_shift_right, ATOM_ARG_REG}, {
        atom_op_shift_right, ATOM_ARG_PS}, {
        atom_op_shift_right, ATOM_ARG_WS}, {
        atom_op_shift_right, ATOM_ARG_FB}, {
        atom_op_shift_right, ATOM_ARG_PLL}, {
        atom_op_shift_right, ATOM_ARG_MC}, {
        atom_op_mul, ATOM_ARG_REG}, {
        atom_op_mul, ATOM_ARG_PS}, {
        atom_op_mul, ATOM_ARG_WS}, {
        atom_op_mul, ATOM_ARG_FB}, {
        atom_op_mul, ATOM_ARG_PLL}, {
        atom_op_mul, ATOM_ARG_MC}, {
        atom_op_div, ATOM_ARG_REG}, {
        atom_op_div, ATOM_ARG_PS}, {
        atom_op_div, ATOM_ARG_WS}, {
        atom_op_div, ATOM_ARG_FB}, {
        atom_op_div, ATOM_ARG_PLL}, {
        atom_op_div, ATOM_ARG_MC}, {
        atom_op_add, ATOM_ARG_REG}, {
        atom_op_add, ATOM_ARG_PS}, {
        atom_op_add, ATOM_ARG_WS}, {
        atom_op_add, ATOM_ARG_FB}, {
        atom_op_add, ATOM_ARG_PLL}, {
        atom_op_add, ATOM_ARG_MC}, {
        atom_op_sub, ATOM_ARG_REG}, {
        atom_op_sub, ATOM_ARG_PS}, {
        atom_op_sub, ATOM_ARG_WS}, {
        atom_op_sub, ATOM_ARG_FB}, {
        atom_op_sub, ATOM_ARG_PLL}, {
        atom_op_sub, ATOM_ARG_MC}, {
        atom_op_setport, ATOM_PORT_ATI}, {
        atom_op_setport, ATOM_PORT_PCI}, {
        atom_op_setport, ATOM_PORT_SYSIO}, {
        atom_op_setregblock, 0}, {
        atom_op_setfbbase, 0}, {
        atom_op_compare, ATOM_ARG_REG}, {
        atom_op_compare, ATOM_ARG_PS}, {
        atom_op_compare, ATOM_ARG_WS}, {
        atom_op_compare, ATOM_ARG_FB}, {
        atom_op_compare, ATOM_ARG_PLL}, {
        atom_op_compare, ATOM_ARG_MC}, {
        atom_op_switch, 0}, {
        atom_op_jump, ATOM_COND_ALWAYS}, {
        atom_op_jump, ATOM_COND_EQUAL}, {
        atom_op_jump, ATOM_COND_BELOW}, {
        atom_op_jump, ATOM_COND_ABOVE}, {
        atom_op_jump, ATOM_COND_BELOWOREQUAL}, {
        atom_op_jump, ATOM_COND_ABOVEOREQUAL}, {
        atom_op_jump, ATOM_COND_NOTEQUAL}, {
        atom_op_test, ATOM_ARG_REG}, {
        atom_op_test, ATOM_ARG_PS}, {
        atom_op_test, ATOM_ARG_WS}, {
        atom_op_test, ATOM_ARG_FB}, {
        atom_op_test, ATOM_ARG_PLL}, {
        atom_op_test, ATOM_ARG_MC}, {
        atom_op_delay, ATOM_UNIT_MILLISEC}, {
        atom_op_delay, ATOM_UNIT_MICROSEC}, {
        atom_op_calltable, 0}, {
        atom_op_repeat, 0}, {
        atom_op_clear, ATOM_ARG_REG}, {
        atom_op_clear, ATOM_ARG_PS}, {
        atom_op_clear, ATOM_ARG_WS}, {
        atom_op_clear, ATOM_ARG_FB}, {
        atom_op_clear, ATOM_ARG_PLL}, {
        atom_op_clear, ATOM_ARG_MC}, {
        atom_op_nop, 0}, {
        atom_op_eot, 0}, {
        atom_op_mask, ATOM_ARG_REG}, {
        atom_op_mask, ATOM_ARG_PS}, {
        atom_op_mask, ATOM_ARG_WS}, {
        atom_op_mask, ATOM_ARG_FB}, {
        atom_op_mask, ATOM_ARG_PLL}, {
        atom_op_mask, ATOM_ARG_MC}, {
        atom_op_postcard, 0}, {
        atom_op_beep, 0}, {
        atom_op_savereg, 0}, {
        atom_op_restorereg, 0}, {
        atom_op_setdatablock, 0}, {
        atom_op_xor, ATOM_ARG_REG}, {
        atom_op_xor, ATOM_ARG_PS}, {
        atom_op_xor, ATOM_ARG_WS}, {
        atom_op_xor, ATOM_ARG_FB}, {
        atom_op_xor, ATOM_ARG_PLL}, {
        atom_op_xor, ATOM_ARG_MC}, {
        atom_op_shl, ATOM_ARG_REG}, {
        atom_op_shl, ATOM_ARG_PS}, {
        atom_op_shl, ATOM_ARG_WS}, {
        atom_op_shl, ATOM_ARG_FB}, {
        atom_op_shl, ATOM_ARG_PLL}, {
        atom_op_shl, ATOM_ARG_MC}, {
        atom_op_shr, ATOM_ARG_REG}, {
        atom_op_shr, ATOM_ARG_PS}, {
        atom_op_shr, ATOM_ARG_WS}, {
        atom_op_shr, ATOM_ARG_FB}, {
        atom_op_shr, ATOM_ARG_PLL}, {
        atom_op_shr, ATOM_ARG_MC}, {
atom_op_debug, 0},};

static int atom_execute_table_locked(struct atom_context *ctx, int index, uint32_t *params, int params_size)
{
        int base = CU16(ctx->cmd_table + 4 + 2 * index);
        int len, ws, ps, ptr;
        unsigned char op;
        atom_exec_context ectx;
        int ret = 0;

        if (!base)
                return -EINVAL;

        len = CU16(base + ATOM_CT_SIZE_PTR);
        ws = CU8(base + ATOM_CT_WS_PTR);
        ps = CU8(base + ATOM_CT_PS_PTR) & ATOM_CT_PS_MASK;
        ptr = base + ATOM_CT_CODE_PTR;

        SDEBUG(">> execute %04X (len %d, WS %d, PS %d)\n", base, len, ws, ps);

        ectx.ctx = ctx;
        ectx.ps_shift = ps / 4;
        ectx.start = base;
        ectx.ps = params;
        ectx.ps_size = params_size;
        ectx.abort = false;
        ectx.last_jump = 0;
        if (ws) {
                ectx.ws = kcalloc(4, ws, GFP_KERNEL);
                ectx.ws_size = ws;
        } else {
                ectx.ws = NULL;
                ectx.ws_size = 0;
        }

        debug_depth++;
        while (1) {
                op = CU8(ptr++);
                if (op < ATOM_OP_NAMES_CNT)
                        SDEBUG("%s @ 0x%04X\n", atom_op_names[op], ptr - 1);
                else
                        SDEBUG("[%d] @ 0x%04X\n", op, ptr - 1);
                if (ectx.abort) {
                        DRM_ERROR("atombios stuck executing %04X (len %d, WS %d, PS %d) @ 0x%04X\n",
                                base, len, ws, ps, ptr - 1);
                        ret = -EINVAL;
                        goto free;
                }

                if (op < ATOM_OP_CNT && op > 0)
                        opcode_table[op].func(&ectx, &ptr,
                                              opcode_table[op].arg);
                else
                        break;

                if (op == ATOM_OP_EOT)
                        break;
        }
        debug_depth--;
        SDEBUG("<<\n");

free:
        kfree(ectx.ws);
        return ret;
}

int atom_execute_table_scratch_unlocked(struct atom_context *ctx, int index, uint32_t *params, int params_size)
{
        int r;

        mutex_lock(&ctx->mutex);
        /* reset data block */
        ctx->data_block = 0;
        /* reset reg block */
        ctx->reg_block = 0;
        /* reset fb window */
        ctx->fb_base = 0;
        /* reset io mode */
        ctx->io_mode = ATOM_IO_MM;
        /* reset divmul */
        ctx->divmul[0] = 0;
        ctx->divmul[1] = 0;
        r = atom_execute_table_locked(ctx, index, params, params_size);
        mutex_unlock(&ctx->mutex);
        return r;
}

int atom_execute_table(struct atom_context *ctx, int index, uint32_t *params, int params_size)
{
        int r;
        mutex_lock(&ctx->scratch_mutex);
        r = atom_execute_table_scratch_unlocked(ctx, index, params, params_size);
        mutex_unlock(&ctx->scratch_mutex);
        return r;
}

static int atom_iio_len[] = { 1, 2, 3, 3, 3, 3, 4, 4, 4, 3 };

static void atom_index_iio(struct atom_context *ctx, int base)
{
        ctx->iio = kzalloc(2 * 256, GFP_KERNEL);
        if (!ctx->iio)
                return;
        while (CU8(base) == ATOM_IIO_START) {
                ctx->iio[CU8(base + 1)] = base + 2;
                base += 2;
                while (CU8(base) != ATOM_IIO_END)
                        base += atom_iio_len[CU8(base)];
                base += 3;
        }
}

struct atom_context *atom_parse(struct card_info *card, void *bios)
{
        int base;
        struct atom_context *ctx =
            kzalloc(sizeof(struct atom_context), GFP_KERNEL);
        char *str;
        char name[512];
        int i;

        if (!ctx)
                return NULL;

        ctx->card = card;
        ctx->bios = bios;

        if (CU16(0) != ATOM_BIOS_MAGIC) {
                pr_info("Invalid BIOS magic\n");
                kfree(ctx);
                return NULL;
        }
        if (strncmp
            (CSTR(ATOM_ATI_MAGIC_PTR), ATOM_ATI_MAGIC,
             strlen(ATOM_ATI_MAGIC))) {
                pr_info("Invalid ATI magic\n");
                kfree(ctx);
                return NULL;
        }

        base = CU16(ATOM_ROM_TABLE_PTR);
        if (strncmp
            (CSTR(base + ATOM_ROM_MAGIC_PTR), ATOM_ROM_MAGIC,
             strlen(ATOM_ROM_MAGIC))) {
                pr_info("Invalid ATOM magic\n");
                kfree(ctx);
                return NULL;
        }

        ctx->cmd_table = CU16(base + ATOM_ROM_CMD_PTR);
        ctx->data_table = CU16(base + ATOM_ROM_DATA_PTR);
        atom_index_iio(ctx, CU16(ctx->data_table + ATOM_DATA_IIO_PTR) + 4);
        if (!ctx->iio) {
                atom_destroy(ctx);
                return NULL;
        }

        str = CSTR(CU16(base + ATOM_ROM_MSG_PTR));
        while (*str && ((*str == '\n') || (*str == '\r')))
                str++;
        /* name string isn't always 0 terminated */
        for (i = 0; i < 511; i++) {
                name[i] = str[i];
                if (name[i] < '.' || name[i] > 'z') {
                        name[i] = 0;
                        break;
                }
        }
        pr_info("ATOM BIOS: %s\n", name);

        return ctx;
}

int atom_asic_init(struct atom_context *ctx)
{
        struct radeon_device *rdev = ctx->card->dev->dev_private;
        int hwi = CU16(ctx->data_table + ATOM_DATA_FWI_PTR);
        uint32_t ps[16];
        int ret;

        memset(ps, 0, 64);

        ps[0] = cpu_to_le32(CU32(hwi + ATOM_FWI_DEFSCLK_PTR));
        ps[1] = cpu_to_le32(CU32(hwi + ATOM_FWI_DEFMCLK_PTR));
        if (!ps[0] || !ps[1])
                return 1;

        if (!CU16(ctx->cmd_table + 4 + 2 * ATOM_CMD_INIT))
                return 1;
        ret = atom_execute_table(ctx, ATOM_CMD_INIT, ps, 16);
        if (ret)
                return ret;

        memset(ps, 0, 64);

        if (rdev->family < CHIP_R600) {
                if (CU16(ctx->cmd_table + 4 + 2 * ATOM_CMD_SPDFANCNTL))
                        atom_execute_table(ctx, ATOM_CMD_SPDFANCNTL, ps, 16);
        }
        return ret;
}

void atom_destroy(struct atom_context *ctx)
{
        kfree(ctx->iio);
        kfree(ctx);
}

bool atom_parse_data_header(struct atom_context *ctx, int index,
                            uint16_t *size, uint8_t *frev, uint8_t *crev,
                            uint16_t *data_start)
{
        int offset = index * 2 + 4;
        int idx = CU16(ctx->data_table + offset);
        u16 *mdt = (u16 *)(ctx->bios + ctx->data_table + 4);

        if (!mdt[index])
                return false;

        if (size)
                *size = CU16(idx);
        if (frev)
                *frev = CU8(idx + 2);
        if (crev)
                *crev = CU8(idx + 3);
        *data_start = idx;
        return true;
}

bool atom_parse_cmd_header(struct atom_context *ctx, int index, uint8_t *frev,
                           uint8_t *crev)
{
        int offset = index * 2 + 4;
        int idx = CU16(ctx->cmd_table + offset);
        u16 *mct = (u16 *)(ctx->bios + ctx->cmd_table + 4);

        if (!mct[index])
                return false;

        if (frev)
                *frev = CU8(idx + 2);
        if (crev)
                *crev = CU8(idx + 3);
        return true;
}

int atom_allocate_fb_scratch(struct atom_context *ctx)
{
        int index = GetIndexIntoMasterTable(DATA, VRAM_UsageByFirmware);
        uint16_t data_offset;
        int usage_bytes = 0;
        struct _ATOM_VRAM_USAGE_BY_FIRMWARE *firmware_usage;

        if (atom_parse_data_header(ctx, index, NULL, NULL, NULL, &data_offset)) {
                firmware_usage = (struct _ATOM_VRAM_USAGE_BY_FIRMWARE *)(ctx->bios + data_offset);

                DRM_DEBUG("atom firmware requested %08x %dkb\n",
                          le32_to_cpu(firmware_usage->asFirmwareVramReserveInfo[0].ulStartAddrUsedByFirmware),
                          le16_to_cpu(firmware_usage->asFirmwareVramReserveInfo[0].usFirmwareUseInKb));

                usage_bytes = le16_to_cpu(firmware_usage->asFirmwareVramReserveInfo[0].usFirmwareUseInKb) * 1024;
        }
        ctx->scratch_size_bytes = 0;
        if (usage_bytes == 0)
                usage_bytes = 20 * 1024;
        /* allocate some scratch memory */
        ctx->scratch = kzalloc(usage_bytes, GFP_KERNEL);
        if (!ctx->scratch)
                return -ENOMEM;
        ctx->scratch_size_bytes = usage_bytes;
        return 0;
}