#include <PR/bcp.h>
#include <PR/R4300.h>
#include <PR/bbsim.h>
#include <PR/bbskapi.h>

#include <PR/os.h>
#include <PR/os_bb.h>
#include <PR/os_bbfs.h>
#include <PR/os_bbatb.h>
#include <PR/os_bbexec.h>
#include <PR/ramrom.h>

#include <PR/os_bbsa.h>
#include <libfb.h>
#include "launch.h"

#define ALIGN_DCACHE __attribute__((aligned(DCACHE_LINESIZE)))
#define printstr fbPrintf

static u32 gMiSecModeRegEntryVal;
OSBbSaTickets      gTickets ALIGN_DCACHE;
OSBbSaGameMetaData gGameMeta[32];
extern u32  __osBbHackFlags;
u8 certBuf[16384] ALIGN_DCACHE;
u8 rlBuf[16384] ALIGN_DCACHE;


/*
 * debug print support
 */

#undef PRINTF

#ifdef DEBUG
#define	PRINTF	osSyncPrintf
#else
#define	PRINTF(format, args...)
#endif

#define fbhBlue         GPACK_RGBA5551(0,0,80,1)
#define fbhGreen        GPACK_RGBA5551(0,80,0,1)
#define fbhRed          GPACK_RGBA5551(80,0,0,1)
#define fbhYellow       GPACK_RGBA5551(80,80,0,1)
#define fbhMagenta      GPACK_RGBA5551(80,0,80,1)
#define fbhCyan         GPACK_RGBA5551(0,80,80,1)

#define fbLemonChiffon          GPACK_RGBA5551(255,250,205,1)
#define fbLightSlateBlue        GPACK_RGBA5551(132,112,255,1)
#define fbLightPink1            GPACK_RGBA5551(255,174,185,1)
#define fbThistle2              GPACK_RGBA5551(238,210,238,1)
#define fbSeaGreen4             GPACK_RGBA5551(105,139,105,1)
#define fbTomato1               GPACK_RGBA5551(255,99,71,1)
#define fbTurquoise2            GPACK_RGBA5551(0,229,238,1)
#define fbSienna4               GPACK_RGBA5551(139,71,38,1)


/*
 * Thread and stack structures
 */

char   bootStack[STACKSIZE] __attribute__ ((aligned (8)));

static OSThread idleThread;
static char     idleThreadStack[STACKSIZE] __attribute__ ((aligned (8)));

static OSThread mainThread;
static char     mainThreadStack[STACKSIZE] __attribute__ ((aligned (8)));

/* 
 * FS specific definitions
 */

static void	idleproc(char *);
static void	mainproc(char *);
static OSBbFs   gFs; /* too big to plop on the stack */

/* XXX: hard-coded size requires more consideration */
#define BB_SYSAPP_NUM_DIR_ENTRIES 256
static OSBbDirEnt gFsDirs[BB_SYSAPP_NUM_DIR_ENTRIES];

static const char gTicketFname[] = "ticket.sys";

#define BB_SYSAPP_CERT_BUF_SIZE (16*1024*4)
static u8 gCertBuf[BB_SYSAPP_CERT_BUF_SIZE] ALIGN_DCACHE;

#define BB_SYSAPP_RL_BUF_SIZE (16*1024*4)
static u8 gRlBuf[BB_SYSAPP_RL_BUF_SIZE] ALIGN_DCACHE;

#define DMA_QUEUE_SIZE	200

static OSMesg           PiMessages[DMA_QUEUE_SIZE];
static OSMesgQueue      PiMessageQ;

static OSMesg           SiMessages[DMA_QUEUE_SIZE];
static OSMesgQueue      SiMessageQ;


/*
 * simple graphics
 */

static OSMesgQueue      retraceMessageQ;
static OSMesg           dummyMessage, retraceMessageBuf;
static u16 cfb[320*240] __attribute__((aligned(64)));

/*
 * controller
 */
static OSMesgQueue      contMessageQ;
OSContStatus statusData[MAXCONTROLLERS];
OSContPad controllerData[MAXCONTROLLERS];
static u16 lastbutton[MAXCONTROLLERS];
static int no_controller = 1;

int gBeginTest = 0;

static void
initController(void)
{
    int i, c = 0;
    u8 pattern;

    if (__osBbIsBb<2)
        __osBbHackFlags = 1;
    else
        __osBbHackFlags = 0;

    osCreateMesgQueue(&contMessageQ, &dummyMessage, 1);
    osSetEventMesg(OS_EVENT_SI, &contMessageQ, (OSMesg) 0);

    osContInit(&contMessageQ, &pattern, &statusData[0]);

    for (i = 0; i < MAXCONTROLLERS; i++) {
        if ((pattern & (1 << i)) &&
            !(statusData[i].errno & CONT_NO_RESPONSE_ERROR)) {
            ++c;
            no_controller = 0;
        }
    }
}


static void
processButton(u16 button, int lastbutton)
{
    if (button & D_JPAD && !(lastbutton & D_JPAD)) {
        // pad down
    }

    if (button & U_JPAD && !(lastbutton & U_JPAD)) {
        // pad up
    }

    // output anything changing here

    if ((button & A_BUTTON && !(lastbutton & A_BUTTON)) ||
        (button & START_BUTTON && !(lastbutton & START_BUTTON))){
        // start or A button pressed
        gBeginTest = 1;
    }
}

static void
readControllers(void)
{
    int i;
    OSContPad *pad;

    osRecvMesg(&contMessageQ, &dummyMessage, OS_MESG_BLOCK);
    osContGetReadData(controllerData);

    for (i = 0; i < MAXCONTROLLERS; ++i) {
        pad = &controllerData[i];
        if (!pad->errno && pad->button != lastbutton[i]) {
            processButton(pad->button, lastbutton[i]);
            lastbutton[i] = pad->button;
        }
    }
}


void  __osBbVideoPllInit(u32);
void boot(u32 miSecModeRegVal)
{
    gMiSecModeRegEntryVal = miSecModeRegVal;

    /* Init the video PLL */
    __osBbVideoPllInit(OS_TV_NTSC);

    osInitialize();

    osCreateThread(&idleThread, 1, (void(*)(void *))idleproc, (void *)0,
                   idleThreadStack+STACKSIZE, 8);
    osStartThread(&idleThread);
}


static void
idleproc(char *argv)		/* priority 8 */
{
    osCreateViManager(OS_PRIORITY_VIMGR);
    fbInit(FB_LOW_RES);

#ifdef DEBUG
    /* should allow usb printing */
    osBbUsbInit();
#endif

    /*
     * Start PI Mgr for access to cartridge - start before the debugger
     */
    osCreatePiManager((OSPri) OS_PRIORITY_PIMGR, &PiMessageQ, PiMessages,
            DMA_QUEUE_SIZE);

    osCreateMesgQueue(&SiMessageQ, SiMessages, DMA_QUEUE_SIZE);
    osSetEventMesg(OS_EVENT_SI, &SiMessageQ, (OSMesg)DMA_QUEUE_SIZE);

    /*
     * The main thread's priority must be the same or lower than the original
     * idle's thread priority. This allows the idle thread to change its
     * priority to 0 before the main thread starts execution.
     */
    osCreateThread(&mainThread, 3, (void(*)(void *))mainproc, argv,
           mainThreadStack+STACKSIZE/8, (OSPri)7);
    osStartThread(&mainThread);

    osSetThreadPri(0, OS_PRIORITY_IDLE);

    for(;;){}
}


void pause()
{
    fbPrintf(fbWhite, 5, 4, "Press A to move to next test");
    osWritebackDCacheAll();

    gBeginTest=0;
    do {
        osRecvMesg(&retraceMessageQ, NULL, OS_MESG_BLOCK);
        osContStartReadData(&contMessageQ);
        readControllers();
    }while(!gBeginTest);
    fbPrintf(fbWhite, 5, 4, "performing test             ");
    fbPrintf(fbhGreen, 10, 8, "    ");
    osWritebackDCacheAll();
}


void reportResult(int pass)
{
    if(pass){
        fbPrintf(fbhGreen, 10, 8, "PASS");
    }
    else{
        fbPrintf(fbhRed, 10, 8, "FAIL");
    }
    osWritebackDCacheAll();
}

#define LAUNCH_TEST_ERROR -1
#define LAUNCH_TEST_FAIL   0
#define LAUNCH_TEST_INJECT_ERR_CMD_SIG 1
#define LAUNCH_TEST_INJECT_ERR_TIK_SIG 2
int launchTest(u32 cid,int injectErr,int tryLaunch)
{
    int ticketIndx, ret;

    /* ticket of interest */
    for(ticketIndx=0; ticketIndx<gTickets.numTickets; ticketIndx++)
        if(cid == (u32)gTickets.ticket[ticketIndx].cmd.head.id)
            break;
    if(ticketIndx>=gTickets.numTickets)
        return LAUNCH_TEST_ERROR;

    if(injectErr==LAUNCH_TEST_INJECT_ERR_CMD_SIG){
        gTickets.ticket[ticketIndx].cmd.head.contentMetaDataSign[0]++;
    }
    else if(injectErr==LAUNCH_TEST_INJECT_ERR_TIK_SIG){
        gTickets.ticket[ticketIndx].head.ticketSign[0]++;
    }

    osBbSaCertInit();
    ret = osBbSaGamePrelaunch(&(gTickets.ticket[ticketIndx]));

    if(injectErr==LAUNCH_TEST_INJECT_ERR_CMD_SIG){
        gTickets.ticket[ticketIndx].cmd.head.contentMetaDataSign[0]--;
    }
    else if(injectErr==LAUNCH_TEST_INJECT_ERR_TIK_SIG){
        gTickets.ticket[ticketIndx].head.ticketSign[0]--;
    }

    if(tryLaunch){
        if(ret!=SALAUNCH_OK)
            return LAUNCH_TEST_ERROR;
        osBbSaGameLaunch(&gGameMeta[ticketIndx].launch);
        return LAUNCH_TEST_FAIL;
    }
    else{
        if(ret==SALAUNCH_OK || ret==SALAUNCH_PERSONALIZE)
            return LAUNCH_TEST_ERROR;
        return LAUNCH_TEST_FAIL;
    }
    return LAUNCH_TEST_ERROR;
}


int createCertSigError(char *certName)
{
    s32 fsret,fd;
    OSBbStatBuf fsStat;
    BbRsaCert *cert;
    int i,numCerts;

    /* alter sigs for failure test */
    osBbFRename("cert.sys","_cert.sys");
    fd = osBbFOpen("_cert.sys","r");
    if(fd<0){
        return -1;
    }
    fsret = osBbFStat(fd, &fsStat, NULL, 0);
    if(fsret !=0)
        return -1;
    if(fsStat.size >= sizeof gCertBuf)
        return -1;
    if(osBbFRead(fd, 0, gCertBuf, fsStat.size)<0){
        osBbFClose(fd);
        return -1;
    }
    osBbFClose(fd);
    numCerts=*(u32 *)gCertBuf;
    for(i=0, cert=(BbRsaCert *)(gCertBuf+4);
        i<numCerts && strcmp(cert->certId.name.server,certName);
        i++, cert++){}
    if(i==numCerts)
        return -1;
    (cert->signature.rsa2048)[0] ++;
    fd=osBbFCreate("cert.sys",1,fsStat.size);
    if(fd<0){
        return -1;
    }
    osBbFWrite(fd,0,gCertBuf,fsStat.size);
    osBbFClose(fd);
    return 0;
}

int twiddleAppByte(char *app,u32 offset)
{
    s32 fsret,fd;
    OSBbStatBuf fsStat;

    fd = osBbFOpen(app,"rw");
    if(fd<0){
        return -1;
    }
    fsret = osBbFStat(fd, &fsStat, NULL, 0);
    if(fsStat.size<=offset)
        return -1;
    if(osBbFRead(fd, offset&(~(16384-1)), gCertBuf, 16384)<0){
        osBbFClose(fd);
        return -1;
    }
    gCertBuf[offset&(16384-1)]^=1;
    if(osBbFWrite(fd, offset&(~(16384-1)), gCertBuf, 16384)<0){
        osBbFClose(fd);
        return -1;
    }

    osBbFClose(fd);
    return 0;
}

void mainproc(char *argv)
{
    int ret, ticketIndx;
    u32 cid;
    s32 fsret;
    BbTicketBundle ticketBundle;
    BbAppLaunchCrls appRls;


    /* setup text output to screen */
    osCreateMesgQueue(&retraceMessageQ, &retraceMessageBuf, 1);
    osViSetEvent(&retraceMessageQ, dummyMessage, 1);
    osViBlack(1);
    osViSwapBuffer(cfb);
    fbSetBg(fbBlack);
    fbClear();
    osViBlack(0);
    osWritebackDCacheAll();
    osViSwapBuffer(cfb);
    fbClear();
    initController();

    printstr(fbWhite, 3, 2, "Launch Test");
    osWritebackDCacheAll();

    if(osBbFInit(&gFs)<0){
        goto exit;
    }

    fsret = osBbFReadDir(gFsDirs, BB_SYSAPP_NUM_DIR_ENTRIES);
    if(fsret<0){
        goto exit;
    }

    osBbSaMetaGetTickets(&gTickets);

    osBbSaMetaGetData(&gTickets,gGameMeta);

    pause();

    /* ptr checks for skLaunchSetup() */
    ret=0;
    fbPrintf(fbWhite, 5, 6, "Conducting ptr checks   ");
    osWritebackDCacheAll();
    /* checks go here. set ret=0 if one fails */
    cid = 0x00020004;
    for(ticketIndx=0; ticketIndx<gTickets.numTickets; ticketIndx++)
        if(cid == (u32)gTickets.ticket[ticketIndx].cmd.head.id)
            break;
    if((ticketIndx < gTickets.numTickets) &&
       (BB_SYSAPP_PASS == osBbSaBundleTicket(&(gTickets.ticket[ticketIndx]),
                           &ticketBundle, &appRls))){
        do{ 
            void *tmp;
#define UPPER_MEM_LIMIT(a) (0x80800000 - (sizeof (a) + 4)) 
#define LOWER_MEM_LIMIT(a) (0x80000000 - (sizeof (a) + 4))
            /* initial sanity check */
            if(SK_API_SUCCESS != skLaunchSetup(&ticketBundle, &appRls, rlBuf))
                break;

            /* ticket bundle ptr */
            if(SK_API_SUCCESS == skLaunchSetup(
                      (void *) UPPER_MEM_LIMIT(ticketBundle), &appRls, rlBuf))
                break;
            if(SK_API_SUCCESS == skLaunchSetup(
                      (void *) LOWER_MEM_LIMIT(ticketBundle), &appRls, rlBuf))
                break;

            /* the ticket */
            tmp = ticketBundle.ticket;
            ticketBundle.ticket = 
                        (void *)(UPPER_MEM_LIMIT(ticketBundle.ticket));
            if(SK_API_SUCCESS == skLaunchSetup(&ticketBundle, &appRls, rlBuf)){
                ticketBundle.ticket = tmp;
                break;
            }
            ticketBundle.ticket = 
                        (void *)(LOWER_MEM_LIMIT(ticketBundle.ticket));
            if(SK_API_SUCCESS == skLaunchSetup(&ticketBundle, &appRls, rlBuf)){
                ticketBundle.ticket = tmp;
                break;
            }
            ticketBundle.ticket = tmp;

            /* the ticket chain */
            tmp = ticketBundle.ticketChain[0];
            ticketBundle.ticketChain[0] = 
                        (void *)(UPPER_MEM_LIMIT(ticketBundle.ticketChain[0]));
            if(SK_API_SUCCESS == skLaunchSetup(&ticketBundle, &appRls, rlBuf)){
                ticketBundle.ticketChain[0] = tmp;
                break;
            }
            ticketBundle.ticketChain[0] = 
                        (void *)(LOWER_MEM_LIMIT(ticketBundle.ticketChain[0]));
            if(SK_API_SUCCESS == skLaunchSetup(&ticketBundle, &appRls, rlBuf)){
                ticketBundle.ticketChain[0] = tmp;
                break;
            }
            ticketBundle.ticketChain[0] = tmp;
            tmp = ticketBundle.ticketChain[1];
            ticketBundle.ticketChain[1] = 
                        (void *)(UPPER_MEM_LIMIT(ticketBundle.ticketChain[1]));
            if(SK_API_SUCCESS == skLaunchSetup(&ticketBundle, &appRls, rlBuf)){
                ticketBundle.ticketChain[1] = tmp;
                break;
            }
            ticketBundle.ticketChain[1] = 
                        (void *)(LOWER_MEM_LIMIT(ticketBundle.ticketChain[1]));
            if(SK_API_SUCCESS == skLaunchSetup(&ticketBundle, &appRls, rlBuf)){
                ticketBundle.ticketChain[1] = tmp;
                break;
            }
            ticketBundle.ticketChain[1] = tmp;

            /* the cmd chain */
            tmp = ticketBundle.cmdChain[0];
            ticketBundle.cmdChain[0] = 
                        (void *)(UPPER_MEM_LIMIT(ticketBundle.cmdChain[0]));
            if(SK_API_SUCCESS == skLaunchSetup(&ticketBundle, &appRls, rlBuf)){
                ticketBundle.cmdChain[0] = tmp;
                break;
            }
            ticketBundle.cmdChain[0] = 
                        (void *)(LOWER_MEM_LIMIT(ticketBundle.cmdChain[0]));
            if(SK_API_SUCCESS == skLaunchSetup(&ticketBundle, &appRls, rlBuf)){
                ticketBundle.cmdChain[0] = tmp;
                break;
            }
            ticketBundle.cmdChain[0] = tmp;
            tmp = ticketBundle.cmdChain[1];
            ticketBundle.cmdChain[1] = 
                        (void *)(UPPER_MEM_LIMIT(ticketBundle.cmdChain[1]));
            if(SK_API_SUCCESS == skLaunchSetup(&ticketBundle, &appRls, rlBuf)){
                ticketBundle.cmdChain[1] = tmp;
                break;
            }
            ticketBundle.cmdChain[1] = 
                        (void *)(LOWER_MEM_LIMIT(ticketBundle.cmdChain[1]));
            if(SK_API_SUCCESS == skLaunchSetup(&ticketBundle, &appRls, rlBuf)){
                ticketBundle.cmdChain[1] = tmp;
                break;
            }
            ticketBundle.cmdChain[1] = tmp;

            /* appRls ptr */
            if(SK_API_SUCCESS == skLaunchSetup(
                      &ticketBundle, (void *) UPPER_MEM_LIMIT(appRls), rlBuf))
                break;
            if(SK_API_SUCCESS == skLaunchSetup(
                      &ticketBundle, (void *) LOWER_MEM_LIMIT(appRls), rlBuf))
                break;

            /* Keylist ptr */
            if(SK_API_SUCCESS == skLaunchSetup(
                      &ticketBundle, &appRls, (void *)(0x80000000-4)))
                break;
            if(SK_API_SUCCESS == skLaunchSetup(
                      &ticketBundle, &appRls, (void *)(0x80800000-16380)))
                break;

            /* final sanity check */
            if(SK_API_SUCCESS != skLaunchSetup(&ticketBundle, &appRls, rlBuf))
                break;

            ret = 1;
        }while(0);
    }
    reportResult(ret);

    pause();

    fbPrintf(fbWhite, 5, 6, "cmd sign by xs chain    ");
    osWritebackDCacheAll();
    ret = launchTest(0x00020001,0,0);
    reportResult(ret==LAUNCH_TEST_FAIL);

    pause();

    fbPrintf(fbWhite, 5, 6, "xs sign by cp chain     ");
    osWritebackDCacheAll();
    ret = launchTest(0x00020002,0,0);
    reportResult(ret==LAUNCH_TEST_FAIL);

    pause();

    /* bbid bad */
    fbPrintf(fbWhite, 5, 6, "bbid bad                ");
    osWritebackDCacheAll();
    ret = launchTest(0x00020003,0,0);
    reportResult(ret==LAUNCH_TEST_FAIL);

    pause();

    /* app hash bad */
    fbPrintf(fbWhite, 5, 6, "app hash bad            ");
    osWritebackDCacheAll();
    twiddleAppByte("00020004.app",0);
    ret = launchTest(0x00020004,0,1);
    twiddleAppByte("00020004.app",0);
    if(ret==LAUNCH_TEST_FAIL){
        twiddleAppByte("00020004.app",16384);
        ret = launchTest(0x00020004,0,1);
        twiddleAppByte("00020004.app",16384);
    }
    reportResult(ret==LAUNCH_TEST_FAIL);

    pause();

    /* cert sig errors */

    if(createCertSigError("CP00000001")<0)
        goto exit;
    fbPrintf(fbWhite, 5, 6, "cp sig bad              ");
    osWritebackDCacheAll();
    ret = launchTest(0x00020000,0,0);
    reportResult(ret==LAUNCH_TEST_FAIL);

    pause();

    osBbFRename("_cert.sys","cert.sys");
    if(createCertSigError("CPCA00000001")<0)
        goto exit;
    fbPrintf(fbWhite, 5, 6, "cpca sig bad            ");
    osWritebackDCacheAll();
    ret = launchTest(0x00020000,0,0);
    reportResult(ret==LAUNCH_TEST_FAIL);

    pause();

    osBbFRename("_cert.sys","cert.sys");
    if(createCertSigError("XS00000001")<0)
        goto exit;
    fbPrintf(fbWhite, 5, 6, "xs sig bad              ");
    osWritebackDCacheAll();
    ret = launchTest(0x00020000,0,0);
    reportResult(ret==LAUNCH_TEST_FAIL);

    pause();

    osBbFRename("_cert.sys","cert.sys");
    if(createCertSigError("XSCA00000001")<0)
        goto exit;
    fbPrintf(fbWhite, 5, 6, "xsca sig bad            ");
    osWritebackDCacheAll();
    ret = launchTest(0x00020000,0,0);
    reportResult(ret==LAUNCH_TEST_FAIL);

    /* ticket sig errors */
    osBbFRename("_cert.sys","cert.sys");
    pause();

    fbPrintf(fbWhite, 5, 6, "cmd sig bad             ");
    osWritebackDCacheAll();
    ret = launchTest(0x00020000,LAUNCH_TEST_INJECT_ERR_CMD_SIG,0);
    reportResult(ret==LAUNCH_TEST_FAIL);

    pause();

    fbPrintf(fbWhite, 5, 6, "ticket sig bad          ");
    osWritebackDCacheAll();
    ret = launchTest(0x00020000,LAUNCH_TEST_INJECT_ERR_TIK_SIG,0 );
    reportResult(ret==LAUNCH_TEST_FAIL);

    osBbFRename("_cert.sys","cert.sys");


    fbPrintf(fbWhite, 5, 4, "                            ");
    fbPrintf(fbWhite, 3, 11, "All tests complete          ");
    osWritebackDCacheAll();
    while(1){}
    return;

 exit:
    /* power off (cannot do this since not in secure mode) */
    printstr(fbWhite, 3, 12, "TEST ERROR!!!");
    osWritebackDCacheAll();
    while(1){}
    return;
}