/*
 * Copyright 2008-2010, François Revol, revol@free.fr. All rights reserved.
 * Distributed under the terms of the MIT License.
 */


#include <boot/platform.h>
#include <boot/stage2.h>
#include <boot/stdio.h>
#include <stdarg.h>

#include <Errors.h>

#include "toscalls.h"


void *gXHDIEntryPoint = NULL;
uint32 gXHDIVersion = 0;

NatFeatCookie *gNatFeatCookie = NULL;
uint32 gDebugPrintfNatFeatID = 0;
uint32 gBootstrapNatFeatID = 0;


/*! Maps TOS error codes to native errors
 */
extern "C" status_t
toserror(int32 err)
{
        // generated from:
        // http://www.fortunecity.com/skyscraper/apple/308/html/appendd.htm
        // with:
        // while read N; do read V; read L; echo -e "\tcase $V: /* $N - $L */\n\t\treturn B_BAD_VALUE;"; done >> errs
        switch (err) {
        /* BIOS errors */
        case 0: /* E_OK - No error */
                return B_OK;
        case -1: /* ERROR - Generic error */
                return B_ERROR;
        case -2: /* EDRVNR - Drive not ready */
                return B_DEV_NOT_READY;
        case -3: /* EUNCMD - Unknown command */
                return B_BAD_VALUE;     //XXX
        case -4: /* E_CRC - CRC error */
                return B_DEV_CRC_ERROR;
        case -5: /* EBADRQ - Bad request */
                return B_BAD_VALUE;     //XXX
        case -6: /* E_SEEK - Seek error */
                return B_DEV_SEEK_ERROR;
        case -7: /* EMEDIA - Unknown media */
                return B_DEV_UNREADABLE;
        case -8: /* ESECNF - Sector not found */
                return B_DEV_FORMAT_ERROR;
        case -9: /* EPAPER - Out of paper */
                return B_DEVICE_NOT_FOUND;
        case -10: /* EWRITF - Write fault */
                return B_DEV_WRITE_ERROR;
        case -11: /* EREADF - Read fault */
                return B_DEV_READ_ERROR;
        case -12: /* EWRPRO - Device is write protected */
                return B_READ_ONLY_DEVICE;
        case -14: /* E_CHNG - Media change detected */
                return B_DEV_MEDIA_CHANGED;
        case -15: /* EUNDEV - Unknown device */
                return B_DEV_BAD_DRIVE_NUM;
        case -16: /* EBADSF - Bad sectors on format */
                return B_DEV_UNREADABLE;
        case -17: /* EOTHER - Insert other disk (request) */
                return B_DEV_MEDIA_CHANGE_REQUESTED;
        /* GEMDOS errors */
        case -32: /* EINVFN - Invalid function */
                return B_BAD_VALUE;
        case -33: /* EFILNF - File not found */
        case -34: /* EPTHNF - Path not found */
                return B_ENTRY_NOT_FOUND;
        case -35: /* ENHNDL - No more handles */
                return B_NO_MORE_FDS;
        case -36: /* EACCDN - Access denied */
                return B_PERMISSION_DENIED;
        case -37: /* EIHNDL - Invalid handle */
                return B_FILE_ERROR;
        case -39: /* ENSMEM - Insufficient memory */
                return B_NO_MEMORY;
        case -40: /* EIMBA - Invalid memory block address */
                return B_BAD_ADDRESS;
        case -46: /* EDRIVE - Invalid drive specification */
                return B_DEV_BAD_DRIVE_NUM;
        case -48: /* ENSAME - Cross device rename */
                return B_CROSS_DEVICE_LINK;
        case -49: /* ENMFIL - No more files */
                return B_NO_MORE_FDS;
        case -58: /* ELOCKED - Record is already locked */
                return B_BAD_VALUE;     //XXX
        case -59: /* ENSLOCK - Invalid lock removal request */
                return B_BAD_VALUE;     //XXX
        case -64: /* ERANGE or ENAMETOOLONG - Range error */
                return B_NAME_TOO_LONG;
        case -65: /* EINTRN - Internal error */
                return B_ERROR;
        case -66: /* EPLFMT - Invalid program load format */
                return B_NOT_AN_EXECUTABLE;
        case -67: /* EGSBF - Memory block growth failure */
                return B_BAD_VALUE;
        case -80: /* ELOOP - Too many symbolic links */
                return B_LINK_LIMIT;
        case -200: /* EMOUNT - Mount point crossed (indicator) */
                return B_BAD_VALUE;     
        default:
                return B_ERROR;
        }
}


/*! Maps XHDI error codes to native errors
 * cf. http://toshyp.atari.org/010008.htm#XHDI_20error_20codes
 */
extern "C" status_t
xhdierror(int32 err)
{
        if (err <= -456) {
                int ide = -(err + 456);
                // ide status reg
                // guessed mapping
                if (ide & (1 << 1)) {   // track 0 not found
                        return B_DEV_FORMAT_ERROR;
                } else if (ide & (1 << 0)) {    // DAM not found
                        return B_DEV_FORMAT_ERROR;
                } else if (ide & (1 << 4)) {    // ID field not found
                        return B_DEV_ID_ERROR;
                } else if (ide & (1 << 7)) {    // bad block mark
                        return B_DEV_FORMAT_ERROR;
                } else if (ide & (1 << 6)) {    // uncorrectable error
                        return B_DEV_UNREADABLE;
                } else if (ide & (1 << 2)) {    // command aborted
                        return B_INTERRUPTED;
                } else if (ide & (1 << 5)) {    // media change
                        return B_DEV_MEDIA_CHANGED;
                } else if (ide & (1 << 3)) {    // media change requested
                        return B_DEV_MEDIA_CHANGE_REQUESTED;
                }
                return B_ERROR;
        } else if (err <= -200) {
                /* SCSI errors */
                int scsi = -(err + 200);
                //XXX:
                switch (scsi) {
                case 0x06:
                        return B_DEV_FORMAT_ERROR;
                case 0x10:
                        return B_DEV_FORMAT_ERROR;
                case 0x11:
                        return B_DEV_UNREADABLE;
                case 0x12:
                        return B_DEV_ID_ERROR;
                case 0x13:
                        return B_DEV_FORMAT_ERROR;
                case 0x20:
                        return B_INTERRUPTED;
                case 0x28:
                        return B_DEV_FORMAT_ERROR;
                case 0x5a:
                        return B_DEV_FORMAT_ERROR;
                }
        }
        return toserror(err);
}


static void
dump_tos_cookie(const struct tos_cookie *c)
{
        if (c != NULL) {
                dprintf("%4.4s: 0x%08lx, %ld\n", (const char *)&c->cookie, c->ivalue,
                        c->ivalue);
        }
}


extern "C" void
dump_tos_cookies(void)
{
        const tos_cookie *c = COOKIE_JAR;
        dprintf("Cookies:\n");
        while (c && (c->cookie)) {
                dump_tos_cookie(c++);
        }
}


extern "C" status_t
init_xhdi(void)
{
        const struct tos_cookie *c;

        if (gXHDIEntryPoint)
                return B_OK;
        
        c = tos_find_cookie(XHDI_COOKIE);
        if (!c)
                return B_ENTRY_NOT_FOUND;
        if (((uint32 *)c->pvalue)[-1] != XHDI_MAGIC)
                return B_BAD_VALUE;
        gXHDIEntryPoint = c->pvalue;
        gXHDIVersion = XHGetVersion();
        return B_OK;
}


extern "C" status_t
init_nat_features(void)
{
        gKernelArgs.arch_args.plat_args.atari.nat_feat.nf_get_id = NULL;
        gKernelArgs.arch_args.plat_args.atari.nat_feat.nf_call = NULL;
        gKernelArgs.arch_args.plat_args.atari.nat_feat.nf_dprintf_id = 0;
        if (nat_features()) {
                // find debugprintf id
                gDebugPrintfNatFeatID = nat_feat_getid("DEBUGPRINTF");
                dprintf("DEBUGPRINTF natfeat id 0x%08lx\n", gDebugPrintfNatFeatID);
                // pass native features infos to the kernel
                gKernelArgs.arch_args.plat_args.atari.nat_feat.nf_get_id =
                        nat_features()->nfGetID;
                gKernelArgs.arch_args.plat_args.atari.nat_feat.nf_call = 
                        nat_features()->nfCall;
                gKernelArgs.arch_args.plat_args.atari.nat_feat.nf_dprintf_id =
                        gDebugPrintfNatFeatID;
                // find other natfeat ids
                gBootstrapNatFeatID = nat_feat_getid("BOOTSTRAP");
                dprintf("BOOTSTRAP natfeat id 0x%08lx\n", gBootstrapNatFeatID);
        }
        return nat_features() ? B_OK : B_ENTRY_NOT_FOUND;
}


extern "C" void
nat_feat_debugprintf(const char *str)
{
        if (gDebugPrintfNatFeatID)
                nat_feat_call(gDebugPrintfNatFeatID, 0, str);
}

extern "C" int
nat_feat_get_bootdrive(void)
{
        if (gBootstrapNatFeatID == 0)
                return -1;
        return nat_feat_call(gBootstrapNatFeatID, 1);
}

extern "C" status_t
nat_feat_get_bootargs(char *str, long size)
{
        status_t err;
        if (gBootstrapNatFeatID == 0)
                return B_ERROR;
        return toserror(nat_feat_call(gBootstrapNatFeatID, 2, str, size));
}