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

#include <PR/os.h>
#include <PR/os_bbfs.h>
#include <PR/os_bbatb.h>
#include <PR/os_bbexec.h>
#include <PR/os_bbsa.h>
#include <sha1.h>


#include "signverify.h"

#include "nu64sys.h"
#include "graph.h"

u32 gMiSecModeRegEntryVal;


/*
 * buffer for FS must be dcache aligned. use macro below.
 */
#define ALIGN_DCACHE __attribute__((aligned(DCACHE_LINESIZE)))

/*
 * debug print support
 */

#undef PRINTF

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


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

/* 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 gSignFname[] = "sig.sys";

static const char gIdFname[] =  "cert.sys";
static const char gSaRlFname[] = "crl.sys";
static const char gEccFname[] = "ecc.sys";

static const char bbChainFname[] = "ida.sys";

#define BB_SYSAPP_RL_BUF_SIZE (16*1024*4)

static u8 gRlBuf[BB_SYSAPP_RL_BUF_SIZE] ALIGN_DCACHE;

#define BB_SYSAPP_SERIAL_HEAP_SIZE (16*1024*8)
static u8 gSerialHeap[BB_SYSAPP_SERIAL_HEAP_SIZE] ALIGN_DCACHE;

#define BB_SIG_HEAP_SIZE (16*1024*8)
static u8 gSigHeap[BB_SIG_HEAP_SIZE] ALIGN_DCACHE;

#define BB_ECC_HEAP_SIZE (16*1024)
static u8 gEccHeap[BB_ECC_HEAP_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[SCREEN_WD*SCREEN_HT] __attribute__((aligned(64)));
static void clear(u16 bg) {
    int i;
    for (i = 0; i < SCREEN_WD*SCREEN_HT; ++i) {
            cfb[i] = bg;
    }
}



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);
#if	SCREEN_LOW
    osViSetMode(&osViModeTable[OS_VI_NTSC_LPN1]);
#else
    osViSetMode(&osViModeTable[OS_VI_NTSC_HPF1]);
#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(;;) {
#if 0
	osBbSetErrorLed(1);
	delay();
	osBbSetErrorLed(0);
	delay();
#endif
    }
}

/* TODO: error handling in place of "goto exit" lines in code below. */


BbShaHash hash_data;
BbShaHash hash_copy;
    
SHA1Context sha;
BbGenericSig sign;


#define STATE_SIZE 1024
#define SIGN_SIZE 256 /* short cutting here, for rsa sign */


void readInCrl(BbAppLaunchCrls *apprls){
  
  /* read entire rl file into serialized heap */
  
#ifdef DEBUG
  PRINTF("create bundle\n");
#endif  
  if(osBbSaRlInit() != BB_SYSAPP_PASS){
#ifdef DEBUG
    PRINTF("crl reading error\n");
#endif
  }
  if(osBbSaRlBundle(apprls)!=BB_SYSAPP_PASS){
#ifdef DEBUG
    PRINTF("bundle creation error\n");
#endif
  }
}
void mainproc(char *argv)
{
    int i;
    s32 fsret, fd, signfd;
    OSBbStatBuf fsStat;
    u8 *certHeap;
    BbCertBase *certChain[BB_CERT_CHAIN_MAXLEN];
    u8 statedata[STATE_SIZE];
    u8 filename[512];
    u32 *signdata;
    BbEccSig sig;
    BbGenericSig signature;
    BbAppLaunchCrls appRls;
    int expected;
    char outstring[512];
    int iter;
    int numiter = 50;
    u32 starttime;
    u32 endtime;
    u32 first, second, third, fourth;
    u32 ret;
    
    
    /* setup text output to screen */
    osCreateMesgQueue(&retraceMessageQ, &retraceMessageBuf, 1);
    osViSetEvent(&retraceMessageQ, dummyMessage, 1);
    osViBlack(1);
    osViSwapBuffer(cfb);
    clear(0);
    osViBlack(0);
    osWritebackDCacheAll();
    osViSwapBuffer(cfb);

    
    printstr(white, 3, 1, "Test Sign/Verify");
    osWritebackDCacheAll();
    delay();

    /* pointer tests first ! */
    for(iter=0; iter< 8; iter++){
      if(iter ==0){ 
	/* starts at end */
	first = (K0BASE+0x800000);
	second = (u32 ) sig;
	expected = SK_API_FAIL;
      }
      else if(iter ==1){
	/* crosses over end */
	first =( K0BASE+0x800000) -  sizeof(BbShaHash)+4;
	second = (u32 ) sig;
	expected = SK_API_FAIL;
      }
      else if(iter ==2){
	/* crosses over beginning */
	first = K0BASE -4;
	second = (u32 )sig;
	expected = SK_API_FAIL;
      }
      else if(iter ==3){
	/* alignment */
	first = K0BASE + 3;
	second = (u32 )sig;
	expected = SK_API_FAIL;
      }
	
      /* repeat for sig pointer */
      else if(iter ==4){
	first = (u32)hash_data;
	second = (K0BASE+0x800000);
	expected = SK_API_FAIL;
      }
      else if(iter ==5){
	first = (u32)hash_data;
	second = ( K0BASE+0x800000) -  sizeof(BbShaHash)+4;
	expected = SK_API_FAIL;
      }
      else if(iter ==6){
	first = (u32)hash_data;
	second = K0BASE -4;
	expected = SK_API_FAIL;
      }
      else if(iter ==7){
	first = (u32)hash_data;
	second = K0BASE +3;
	expected = SK_API_FAIL;
      }
	
      
      if(skSignHash((u32 *)first, (u32 *)second) == expected){
#ifdef DEBUG
	PRINTF("Pointer check pass \n");
#endif
	printstr(white, 3, 2, "Pointer Check PASS");
	osWritebackDCacheAll();
      }
      else{
#ifdef DEBUG
	PRINTF("Pointer check fail  \n");
#endif
	printstr(white, 3, 3, "Pointer Check FAIL");
	osWritebackDCacheAll();
      }
    }


    /* pointer checks on verify */
  
    for(iter =0; iter < 7; iter++){
      if(iter ==0){ 
	/* starts at end */
	first = K0BASE + (0x800000);
	second = (u32 ) sig;
	third = 0;
	fourth = 0;
	expected = SK_API_FAIL;
      }
      else if(iter ==1){
	first = (u32)hash_data;
	second = ( K0BASE+0x800000) -  sizeof(BbShaHash)+4;
	third = 0;
	fourth = 0;
	expected = SK_API_FAIL;
      }
      else if(iter ==2){
	first = (u32)hash_data;
	second = ( K0BASE+0x800000) -  sizeof(BbShaHash)+4;
	third = 0;
	fourth = 0;
	expected = SK_API_FAIL;
      }
      else if(iter ==3){
	first = (u32) hash_data;
	second = (u32) sig;
	third = K0BASE - 4;
	fourth = 0;
	expected = SK_API_FAIL;
      }
      else if(iter ==4){
	first = (u32) hash_data;
	second = (u32) sig;
	third = K0BASE;
	appRls.carl.head = (BbCrlHead *)(K0BASE+0x800000);
	fourth =  (u32) &appRls;
      }
      else if(iter ==5){
	first = (u32) hash_data;
	second = (u32) sig;
	third = K0BASE;
	appRls.carl.list = (BbServerSuffix *) (K0BASE-4);
	fourth =  (u32) &appRls;
      }
      else if(iter ==6){
	/* check for cert chain pointer error */
	first = (u32) hash_data;
	second = (u32) sig;
	third = K0BASE;
	appRls.carl.certChain[0] = (BbCertBase *)(K0BASE-4);
	fourth =  (u32) &appRls;
      }
      if(skVerifyHash((u32 *)first, (BbGenericSig *) second, (BbCertBase **)third,(BbAppLaunchCrls *)fourth)==expected){
#ifdef DEBUG
	  PRINTF("Pointer check verify PASS iter = %d\n", iter);
#endif
	  sprintf(outstring, "Verify ptr PASS %d", iter);
	  printstr(white, 3, 2, outstring);
	  osWritebackDCacheAll();
	}
      else{
#ifdef DEBUG
	  PRINTF("Pointer check verify FAIL iter = %d\n", iter);
#endif
	  sprintf(outstring, "Verify ptr FAIL %d", iter);
	  printstr(white, 3, 3, outstring);
	  osWritebackDCacheAll();
	}
    }




    if(osBbFInit(&gFs)<0){
#ifdef DEBUG
      PRINTF("ERROR osBbFInit()\n");
#endif
        goto exit;
    }
    
    
    fsret = osBbFReadDir(gFsDirs, BB_SYSAPP_NUM_DIR_ENTRIES);
    
    if(fsret<0){
#ifdef DEBUG
        PRINTF("ERROR osBbFReadDir()\n");
#endif
        goto exit;
    }

    
    for(i=0; i< STATE_SIZE; i++){
      statedata[i] = (u8 )i;
    }

 
    SHA1Reset(&sha);
    SHA1Input(&sha, (u8 *) statedata, STATE_SIZE);
    SHA1Result(&sha, (u8 *)hash_data);
    memcpy(hash_copy, hash_data, sizeof(BbShaHash));

    for(iter = 0; iter < 10;iter++ ){      
      sprintf(outstring, "Iteration = %d", iter);
      printstr(white, 3, 4, outstring);
      osWritebackDCacheAll();
      starttime = osGetCount();
      hash_data[0] = hash_data[0] ^ iter;
      if(skSignHash(hash_data, sig) == SK_API_SUCCESS){
#ifdef DEBUG
	PRINTF("Signed data \n");
#endif
	printstr(white, 3, 5, "Sign PASS");
	osWritebackDCacheAll();
      }
      else{
#ifdef DEBUG
	PRINTF("Signed data  \n");
#endif
	printstr(white, 3, 5, "After sign done, FAIL inside");
	osWritebackDCacheAll();
      }
      endtime = osGetCount();
      sprintf(outstring, "Time taken sign = %llu",OS_CYCLES_TO_USEC(endtime - starttime));
      printstr(white, 3, 12, outstring);
      osWritebackDCacheAll();
      
      

      for(i=0; i< sizeof(BbEccSig)/sizeof(u32); i++){
	signature.ecc[i] = sig[i];
	sprintf(outstring, "%08x\n", (unsigned int )sig[i]);
#ifdef DEBUG
	PRINTF(outstring);
#endif
      }

      /*  check verify with cert chain */
      if(iter == 0){
	
	fd = osBbFOpen(bbChainFname,"r");
	if (fd < 0){
#ifdef DEBUG
	  PRINTF("open failed");
#endif
	}
	fsret = osBbFStat(fd, &fsStat, NULL, 0);
#ifdef DEBUG
	PRINTF("%s contains %d bytes\n",gIdFname,fsStat.size);
#endif

	if(fsret !=0){
#ifdef DEBUG
	  PRINTF("stat error");
#endif
	}
   

	certHeap = (u8 *)gSerialHeap;

	/* read entire cert file into serialized heap */
	if(osBbFRead(fd, 0, certHeap, fsStat.size)<0){
	  osBbFClose(fd);
#ifdef DEBUG
	  PRINTF("reading into heap error\n");
#endif
	}

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

	certChain[0] = certHeap + 4;
	certChain[1] = ((u32) (certChain[0])) + sizeof(BbEccCert);
	certChain[2] = ((u32) (certChain[1])) + sizeof(BbRsaCert);
	
        memset((void *)&appRls,0,sizeof(BbAppLaunchCrls));
	expected = SK_API_SUCCESS;
      
	if(skVerifyHash(hash_data, &signature, certChain, &appRls) == expected){
#ifdef DEBUG
	  PRINTF("BB CERT TEST PASS iter = %d\n", iter);
#endif
	  sprintf(outstring, "Verify PASS %d", iter);
	  printstr(white, 3, 8, outstring);
	  osWritebackDCacheAll();
	}
	else{
#ifdef DEBUG
	  PRINTF("BB CERT TEST FAIL = %d\n", iter);
#endif
	  sprintf(outstring, "Verify FAIL %d", iter);
	  printstr(white, 3, 9, outstring);
	  osWritebackDCacheAll();
	}
      }
      else{
    
      starttime = osGetCount();
      if(skVerifyHash(hash_data, &signature, 0x0, 0x0) == SK_API_SUCCESS){
#ifdef DEBUG
	PRINTF("ECC sign verification succeeded\n");
#endif
	sprintf(outstring, "Verify PASS %d", iter);
	printstr(white, 3, 7, outstring);
	osWritebackDCacheAll();
      }
      else{
#ifdef DEBUG
	PRINTF("ECC sign verify failed\n");
#endif
	sprintf(outstring, "After sign: FAIL  %d", iter);
	printstr(white, 3, 9, outstring);
	osWritebackDCacheAll();
      }

      endtime = osGetCount();

      sprintf(outstring, "Time taken verify = %llu",OS_CYCLES_TO_USEC(endtime - starttime));
      printstr(white, 3, 14, outstring);
      osWritebackDCacheAll();
      
      }
    }


    /*
     * load certs
     */

    fd = osBbFOpen(gIdFname,"r");
    if (fd < 0){
#ifdef DEBUG
      PRINTF("open failed");
#endif
    }
    fsret = osBbFStat(fd, &fsStat, NULL, 0);
#ifdef DEBUG
    PRINTF("%s contains %d bytes\n",gIdFname,fsStat.size);
#endif

    if(fsret !=0){
#ifdef DEBUG
      PRINTF("stat error");
#endif
    }
   

    certHeap = (u8 *)gSerialHeap;

    /* read entire cert file into serialized heap */
    if(osBbFRead(fd, 0, certHeap, fsStat.size)<0){
        osBbFClose(fd);
#ifdef DEBUG
        PRINTF("reading into heap error\n");
#endif
    }

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

    /* initialize cert parser */
    if(osBbSaCertInit(certHeap)!=BB_SYSAPP_PASS){
#ifdef DEBUG
      PRINTF("error initialising parser\n");
#endif
    }

    
    if(osBbSaCertCreateChain("Root-XSCA00000002-XS00000001",
                         certChain)<0){
#ifdef DEBUG
      PRINTF("saCertCreateChain() failure for cert chain.\n");
#endif
    }
        
    

    /*
     * open sign file
     */
    signfd = osBbFOpen(gSignFname, "r");
    if(signfd < 0){
#ifdef DEBUG
      PRINTF("open failed");
#endif
    }
    fsret = osBbFStat(signfd, &fsStat, NULL, 0);
    
    if(osBbFRead(signfd, 0, gSigHeap, sizeof(BbRsaSig2048)*numiter)<0){
      osBbFClose(signfd);
#ifdef DEBUG
      PRINTF("reading into sign data  error\n");
#endif
    }

    memcpy(hash_data, hash_copy, sizeof(BbShaHash));
    for(iter=0; iter < numiter; iter++){
     
      /* read part of sign file into serialized heap */
      signdata = (u32 *)(gSigHeap + sizeof(BbRsaSig2048)*iter);
      for(i=0; i< sizeof(BbRsaSig2048)/sizeof(u32); i++){
	sign.rsa2048[i] = signdata[i];
      }
      /* do CRL tests simultaneously. 
       */

      if(iter ==0){
	/* crla.sys revokes nothing being used */
	
	fsret = osBbFRename("crla.sys", "crl.sys");
#ifdef DEBUG
	if(fsret < 0){
	  PRINTF("Rename did not happen!\n");
	}
#endif
	readInCrl(&appRls);

	/*reset virage */
	skResetWindow();
	skResetWindow();
	expected = SK_API_SUCCESS;
      }

      else if(iter == 1){
	/* check if it fails because internal number is upgraded and
	 * no crl is passed
	 */
	memset((void *)&appRls,0,sizeof(BbAppLaunchCrls));
	expected = SK_API_FAIL;
      }

      if(iter ==2){
	fsret  = osBbFRename("crlb.sys", "crl.sys");
#ifdef DEBUG
	if(fsret < 0){
	  PRINTF("Rename did not happen!\n");
	}
#endif
	readInCrl(&appRls);

	/* crlb.sys revokes the ca */
	
	/*reset virage */
	skResetWindow();
	skResetWindow();
	expected = SK_API_REVOKED_SERVER;
      }

      else if (iter ==3){
	/* crlc.sys revokes the XS */
	
	osBbFRename("crlc.sys", "crl.sys");
	readInCrl(&appRls);

	/*reset virage */
	skResetWindow();
	skResetWindow();
	expected = SK_API_REVOKED_SERVER;
      }
      else if (iter ==4){
	/* crld.sys revokes the XS from a cpcrl */
	

	osBbFRename("crld.sys", "crl.sys");
	readInCrl(&appRls);

	/*reset virage */
	skResetWindow();
	skResetWindow();
	expected = SK_API_REVOKED_SERVER;
      }
      else if (iter ==5){
	/* negative test */
	skResetWindow();
	skResetWindow();
	memset((void *)&appRls,0,sizeof(BbAppLaunchCrls));
	sign.rsa2048[0] = sign.rsa2048[0] ^ iter;
	expected = SK_API_FAIL;
      }

      else{ /* no CRL */

	skResetWindow();
	skResetWindow();
	memset((void *)&appRls,0,sizeof(BbAppLaunchCrls));
	expected = SK_API_SUCCESS;

      }

      starttime = osGetCount();
      ret = skVerifyHash(hash_data, &sign, certChain, &appRls);
      if(ret == expected){

#ifdef DEBUG
	PRINTF("RSA sign verify TEST PASS iter = %d\n", iter);
#endif
	sprintf(outstring, "RSA PASS iter = %d\n", iter);
	printstr(white, 3, 8, outstring);
	osWritebackDCacheAll();
      }
      else{
#ifdef DEBUG
	PRINTF("RSA sign verify TEST FAIL = %d\n", iter);
#endif
	sprintf(outstring, "RSA FAIL iter = %d\n", iter);
	printstr(white, 3, 9, outstring);
	osWritebackDCacheAll();
      }
      endtime = osGetCount();
      
      if(iter > 0){
      skResetWindow();
      skResetWindow();
      }

      if(iter ==0){
	osBbFRename("crl.sys", "crla.sys");
      }
      else if(iter == 2){
	osBbFRename("crl.sys", "crlb.sys");
      }

      else if(iter == 3){
	osBbFRename("crl.sys", "crlc.sys");
      }
      else if(iter == 4){
	osBbFRename("crl.sys", "crld.sys");
      }

    }

    if(osBbFClose(signfd)<0){
#ifdef DEBUG
      PRINTF("error closing file\n");
#endif
    }
 exit: 
    return;
}