#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 "crls.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";
static const char gSaRlFname[] = "crl.sys";
static const char gCertFname[] = "cert.sys";

#define BB_CERT_BUF_SIZE (16*1024*4)

static u8 gCertBuf[BB_CERT_BUF_SIZE] ALIGN_DCACHE;

#define BB_SYSAPP_RL_BUF_SIZE (16*1024*4)

static u8 gRlBuf[BB_SYSAPP_RL_BUF_SIZE] ALIGN_DCACHE;

/*
static u8 gShuffleHeap[RECRYPT_CHUNK_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;
/*
static int gLaunchTime = 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;
        }
    }
}


void outputCid()
{
    sprintf(gDispStr,"CID: %08x",
            (unsigned int)(gTickets.ticket[gTicketIndex].cmd.head.id));
    printstr(fbWhite, 3, 6, gDispStr);
    osWritebackDCacheAll();
}
/*
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;
}

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

static void
delay(void) {
    int k;
    for(k = 0; k < 1000000; k++)
	;
}

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)
{
    
    char outstring[256];
    s32 fsret, fd;
    OSBbStatBuf fsStat;
    BbAppLaunchCrls appRls;
    int iter;
    int ret, expectedret;
#if 0
    u8 bigv[4096]; /* enough for three virages */
    BbVirage01 *v0, *v1; 
#endif
    u32 versions[3]; /* to hold ticket version numbers */
    u32 expectedversions[3];
    /* 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 CRLs App");
    osWritebackDCacheAll();

    if(osBbFInit(&gFs)<0){
        goto exit;
    }
    
    fsret = osBbFReadDir(gFsDirs, BB_SYSAPP_NUM_DIR_ENTRIES);
    if(fsret<0){
        goto exit;
    }
    /* load cert file */
    fd = osBbFOpen(gCertFname,"r");
    if(fd<0){
#ifdef DEBUG
      PRINTF("open error for cert file \n");
#endif
    }
    fsret = osBbFStat(fd, &fsStat, NULL, 0);
        
    /* read entire cert file into serialized heap */
    if(osBbFRead(fd, 0, gCertBuf, fsStat.size)<0){
        osBbFClose(fd);
#ifdef DEBUG
        PRINTF("reading to heap error\n");
#endif
    }

    if(osBbFClose(fd)<0){
#ifdef DEBUG
      PRINTF("cert file close error\n");
#endif
    }

    /* initialize cert parser */
    if(osBbSaCertInit(gCertBuf)!=BB_SYSAPP_PASS){
#ifdef DEBUG
      PRINTF("cert init error\n");
#endif
    }
    skResetWindow();
    skResetWindow();
    for(iter =0; iter < 11; iter++){
      if(iter ==0){
	/* ca revoke head ver1, ticket ver=2 FAIL */
	osBbFRename("crla.sys", "crl.sys");
	expectedret = SK_API_INVALID_CARL;
	expectedversions[0] =0;
	expectedversions[1] =0;
	expectedversions[2] =0;
	
	versions[0]=2; versions[1] =0; versions[2]=0;
      }
      else if(iter ==1){
	/*ca revoke head ver1, ticket ver 1 update */
	expectedret = SK_API_SUCCESS;
	expectedversions[0] =1;
	expectedversions[1] =0;
	expectedversions[2] =0;
	versions[0]=1; versions[1] =0; versions[2]=0;
      }
      else if(iter ==2){
	/*ts revoke head ver3, ticket ver 3 update */
	osBbFRename("crlb.sys", "crl.sys");
	expectedret = SK_API_SUCCESS;
	expectedversions[0] =0;
	expectedversions[1] =0;
	expectedversions[2] =3;
	versions[0]=0; versions[1] =0; versions[2]=3;
      }
      else if(iter ==3){
	/*ts revoke head ver3, ticket ver 3 update */
	osBbFRename("crlc.sys", "crl.sys");
	expectedret = SK_API_SUCCESS;
	expectedversions[0] =0;
	expectedversions[1] =3;
	expectedversions[2] =0;
	versions[0]=0; versions[1] =3; versions[2]=0;
      }
      else if(iter ==4){
	/*ca revoke head ver3, all update */
	osBbFRename("crld.sys", "crl.sys");
	expectedret = SK_API_SUCCESS;
	expectedversions[0] =3;
	expectedversions[1] =0;
	expectedversions[2] =0;
	versions[0]=3; versions[1] =0; versions[2]=0;
      }
      else if(iter ==5){
	/*name check fail */
	osBbFRename("crle.sys", "crl.sys");
	expectedret = SK_API_INVALID_CPRL;
	expectedversions[0] =0;
	expectedversions[1] =0;
	expectedversions[2] =0;
	versions[0]=0; versions[1] =0; versions[2]=0;
      }
      else if(iter ==6){
	/* everything ok: nothing revoked  */
	osBbFRename("crlf.sys", "crl.sys");
	expectedret = SK_API_SUCCESS;
	expectedversions[0] =3;
	expectedversions[1] =3;
	expectedversions[2] =3;
	versions[0]=3; versions[1] =3; versions[2]=3;
      }
      else if(iter ==7){
	/* crl->head has smaller ver =1 than virage */
	osBbFRename("crlaa.sys", "crl.sys");
	expectedret = SK_API_INVALID_CARL;
	expectedversions[0] =3;
	expectedversions[1] =3;
	expectedversions[2] =3;
	versions[0]=1; versions[1] =3; versions[2]=3;
      }
      else if(iter ==8){
	/* everything ok: root revokes cp ca in bundle  */
	osBbFRename("crlg.sys", "crl.sys");
	expectedret = SK_API_REVOKED_SERVER;
	expectedversions[0] =3;
	expectedversions[1] =0;
	expectedversions[2] =0;
	versions[0]=3; versions[1] =3; versions[2]=0;
      }
      else if(iter ==9){
	/* revoked server : root revokes cp ca in bundle  */
	osBbFRename("crlh.sys", "crl.sys");
	expectedret = SK_API_REVOKED_SERVER;
	expectedversions[0] =3;
	expectedversions[1] =0;
	expectedversions[2] =0;
	versions[0]=3; versions[1] =0; versions[2]=3;
      }
      else if(iter ==10){
	/* everything ok, hand changed sign */
	osBbFRename("crli.sys", "crl.sys");
	expectedret = SK_API_INVALID_CARL;
	expectedversions[0] =0;
	expectedversions[1] =0;
	expectedversions[2] =0;
	versions[0]=3; versions[1] =0; versions[2]=0;
      }

      if(osBbSaRlInit() != BB_SYSAPP_PASS){
#ifdef DEBUG
	PRINTF("bundle creation error\n");
#endif
      }
      if(osBbSaRlBundle(&appRls)!=BB_SYSAPP_PASS){
#ifdef DEBUG
	PRINTF("bundle creation error\n");
#endif
      }
    /* reset virage */
      ret = skValidateRls(&(appRls.carl), &(appRls.cprl),&(appRls.tsrl), versions);
      if((ret == expectedret)
	 &&(versions[0] == expectedversions[0])
	 &&(versions[1] == expectedversions[1])
	 &&(versions[2] == expectedversions[2])){
	sprintf(outstring,"iter %d PASS\n", iter);
	printstr(fbWhite, 3, 6, outstring);
	osWritebackDCacheAll();
#ifdef DEBUG
	PRINTF(outstring);
	PRINTF("validate ret = %d\n", ret);
#endif
      }
      else{
	sprintf(outstring,"iter %d FAIL\n", iter);
	
	printstr(fbWhite, 3, 7, outstring);
	osWritebackDCacheAll();
#ifdef DEBUG
	PRINTF(outstring);
	PRINTF("validate ret = %d\n", ret);
#endif
      }

    
#if 0 
      skDumpVirage(bigv);
      v0 = (BbVirage01 *) bigv + sizeof(BbVirage2);
      v1 = (BbVirage01 *) bigv + sizeof(BbVirage2) + sizeof(BbVirage01);
      sprintf(outstring, "tsCrlVersion = %02x\n", v0->tsCrlVersion);
#endif
      sprintf(outstring, "tsCrlVer = %02x\n", (unsigned int) versions[2]);
      printstr(fbWhite, 3, 12, outstring);
      osWritebackDCacheAll();
#ifdef DEBUG
      PRINTF(outstring);
#endif
#if 0
      sprintf(outstring, "caCrlVersion = %02x\n", v0->caCrlVersion);
#endif
      sprintf(outstring, "caCrlVer = %02x\n", (unsigned int) versions[0]);
      printstr(fbWhite, 3, 13, outstring);
      osWritebackDCacheAll();
#ifdef DEBUG
      PRINTF(outstring);
#endif
#if 0    
      sprintf(outstring, "cpCrlVersion = %02x\n", v0->cpCrlVersion);
#endif
      sprintf(outstring, "cpCrlVersion = %02x\n", (unsigned int) versions[1]);
      printstr(fbWhite, 3, 14, outstring);
      osWritebackDCacheAll();
#ifdef DEBUG
      PRINTF(outstring);
#endif

      if(iter ==0){
	/* ca revoke head ver1, ticket ver=2 FAIL */
	/*no rename */
      }
      else if(iter ==1){
	osBbFRename("crl.sys", "crla.sys");
	skResetWindow();
	skResetWindow();
      }
      else if(iter ==2){
	osBbFRename("crl.sys", "crlb.sys");
	skResetWindow();
	skResetWindow();
      }
      else if(iter ==3){
	osBbFRename("crl.sys", "crlc.sys");
	skResetWindow();
	skResetWindow();
      }
      else if(iter ==4){
	osBbFRename("crl.sys", "crld.sys");
	skResetWindow();
	skResetWindow();
      }
      else if(iter ==5){
	osBbFRename("crl.sys", "crle.sys");
	skResetWindow();
	skResetWindow();
      }
      else if(iter ==6){
	osBbFRename("crl.sys", "crlf.sys");
	/* dont reset virage, next test needs it at 3. */
      }
      else if(iter ==7){
	osBbFRename("crl.sys", "crlaa.sys");
	skResetWindow();
	skResetWindow();
      }
      else if(iter ==8){
	osBbFRename("crl.sys", "crlg.sys");
	skResetWindow();
	skResetWindow();
      }
      else if(iter ==9){
	osBbFRename("crl.sys", "crlh.sys");
	skResetWindow();
	skResetWindow();
      }
      else if(iter ==10){
	osBbFRename("crl.sys", "crli.sys");
	skResetWindow();
	skResetWindow();
      }
      delay();
    }
 exit:
    /* dont forget this*/
    skResetWindow();
    skResetWindow();
    return;
}