#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 "launchcrl.h"

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

static u32 gMiSecModeRegEntryVal;
static char gDispStr[1024];
static int gTicketIndex;
OSBbSaTickets      gTickets __attribute__((aligned(16)));
OSBbSaGameMetaData gGameMeta[32];
extern u32  __osBbHackFlags;


/*
 * 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)

#define BB_SYSAPP_RL_BUF_SIZE (16*1024*4)




#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];
#if 0
static u16 lastbutton[MAXCONTROLLERS];
#endif
static int no_controller = 1;


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;
        }
    }
}


void outputCid()
{
    sprintf(gDispStr,"CID: %08x",
            (unsigned int)(gTickets.ticket[gTicketIndex].cmd.head.id));
    printstr(fbWhite, 3, 6, gDispStr);
    osWritebackDCacheAll();
}
#if 0
static void
processButton(u16 button, int lastbutton)
{
    if (button & D_JPAD && !(lastbutton & D_JPAD)) {
        gTicketIndex = (gTicketIndex + 1) % gTickets.numTickets;
    }

    if (button & U_JPAD && !(lastbutton & U_JPAD)) {
        gTicketIndex = (gTicketIndex - 1) < 0? gTickets.numTickets - 1 : 
            gTicketIndex-1;
    }

    outputCid();

    if ((button & A_BUTTON && !(lastbutton & A_BUTTON)) ||
        (button & START_BUTTON && !(lastbutton & START_BUTTON)))
        gLaunchTime = 1;
}

#endif
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 */
    /*
    osUsbInit();
    */
#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 mainproc(char *argv)
{
    int ret, iter;
    s32 fsret;
    u8 outstring[256];
    int expected;
    /* 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, "Test Launch w/CRL App");
    osWritebackDCacheAll();

    skResetWindow();
    skResetWindow();

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

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

    for(iter =0; iter < 8; iter++){
      /* CPCRL no revocation, upgrades to ver =3 */
      if(iter ==0){
	osBbFRename("ticka.sys", "ticket.sys");
	osBbFRename("crla.sys", "crl.sys");
	expected = SALAUNCH_OK;
	skResetWindow();
	skResetWindow();
      }
      
      /* CPCRL actually revokes! */
      else if(iter == 1){
	osBbFRename("tickb.sys", "ticket.sys");
	osBbFRename("crlb.sys", "crl.sys");
	expected = SALAUNCH_ERR_DATA;
	skResetWindow();
	skResetWindow();
      }
      /* TSCRL no revocation */
      else if(iter == 2){
	osBbFRename("tickc.sys", "ticket.sys");
	osBbFRename("crlc.sys", "crl.sys");
	expected = SALAUNCH_OK;
	skResetWindow();
	skResetWindow();
      }
      else if(iter == 3){
	osBbFRename("tickd.sys", "ticket.sys");
	osBbFRename("crld.sys", "crl.sys");
	expected = SALAUNCH_ERR_DATA;
	skResetWindow();
	skResetWindow();
      }
      /* cpca no revocation */
      else if(iter == 4){
	osBbFRename("ticke.sys", "ticket.sys");
	osBbFRename("crle.sys", "crl.sys");
	expected = SALAUNCH_OK;
	skResetWindow();
	skResetWindow();
      }
      /* cpca revoked */
      else if(iter == 5){
	osBbFRename("tickf.sys", "ticket.sys");
	osBbFRename("crlf.sys", "crl.sys");
	expected = SALAUNCH_ERR_DATA;
	skResetWindow();
	skResetWindow();
      }
      /* ts ca no revocation */
      else if(iter == 6){
	osBbFRename("tickg.sys", "ticket.sys");
	osBbFRename("crlg.sys", "crl.sys");
	expected = SALAUNCH_OK;
	skResetWindow();
	skResetWindow();
      }
      /* ts ca revoked */
      else if(iter == 7){
	osBbFRename("tickh.sys", "ticket.sys");
	osBbFRename("crlh.sys", "crl.sys");
	expected = SALAUNCH_ERR_DATA;
	skResetWindow();
	skResetWindow();
      }
      

      
      if(osBbSaRlInit() != BB_SYSAPP_PASS){
#ifdef DEBUG
	PRINTF("RL init error\n");
#endif
      }            
      osBbSaMetaGetTickets(&gTickets);

      gTicketIndex = 0;
      

      osBbSaMetaGetData(&gTickets,gGameMeta);

      
      ret = osBbSaGamePrelaunch(&(gTickets.ticket[gTicketIndex]));
    
      if(ret==expected){
	sprintf(outstring, "Launch CRL PASS iter %d\n", iter);
        printstr(fbWhite, 3, 10, outstring);
        osWritebackDCacheAll();
#ifdef DEBUG
        PRINTF("Launch PASS iter = %d\n", iter);
#endif
      }
      else{
	sprintf(outstring, "Launch CRL FAIL iter %d\n", iter);
	printstr(fbWhite, 3, 12, outstring);
	osWritebackDCacheAll();
#ifdef DEBUG
	PRINTF("Launch FAILED iter = %d\n", iter);
#endif
      }

      /* put back the files */
      /* CPCRL no revocation, upgrades to ver =3 */
      if(iter ==0){
	osBbFRename("ticket.sys", "ticka.sys");
	osBbFRename("crl.sys", "crla.sys");
      }
      /* CPCRL actually revokes! */
      else if(iter == 1){
	osBbFRename("ticket.sys", "tickb.sys");
	osBbFRename("crl.sys", "crlb.sys");
      }
      /* TSCRL no revocation */
      else if(iter == 2){
	osBbFRename("ticket.sys", "tickc.sys");
	osBbFRename("crl.sys", "crlc.sys");
      }
      else if(iter == 3){
	osBbFRename("ticket.sys", "tickd.sys");
	osBbFRename("crl.sys", "crld.sys");
      }
      /* cpca no revocation */
      else if(iter == 4){
	osBbFRename("ticket.sys", "ticke.sys");
	osBbFRename("crl.sys", "crle.sys");
      }
      /* cpca revoked */
      else if(iter == 5){
	osBbFRename("ticket.sys", "tickf.sys");
	osBbFRename("crl.sys", "crlf.sys");
      }
      /* ts ca no revocation */
      else if(iter == 6){
	osBbFRename("ticket.sys", "tickg.sys");
	osBbFRename("crl.sys", "crlg.sys");
      }
      /* ts ca revoked */
      else if(iter == 7){
	osBbFRename("ticket.sys", "tickh.sys");
	osBbFRename("crl.sys", "crlh.sys");
      }

      /* dont forget!! */
      skResetWindow();
      skResetWindow();
      
    }
    exit:
#ifdef DEBUG
      PRINTF("\nAPP FINISHED\n");
#endif
      /* power off (cannot do this since not in secure mode) */
      printstr(fbWhite, 3, 13, "Launch CRL finished");
      osWritebackDCacheAll();
      

      
      return;
}