/*************************************************************************
 *
 *  File: main.c
 *
 *  $Header: /root/leakn64/depot/rf/sw/bbplayer/iosim/src/mi_test/main.c,v 1.29 2003/02/15 19:40:54 jprincen Exp $
 *
 */
#include <errno.h>
#include <fcntl.h>
#include <getopt.h>
#include <stdio.h>
#include <stdlib.h>
#include <stdarg.h>
#include <string.h>
#include <unistd.h>
#include <ctype.h>

#include "iomap.h"
#include "iotest.h"
#include "sim.h"
#include "simipc.h"
#include "trace.h"
#include "bcp_util.h"
#include "mi_test.h"
#include "nvram_test.h"

#define PASS 0
#define FAIL -1

/***********************************************************************
 * Extern definitions
 */
extern	TstCmd	TstCmdTbl[];
extern	void	IoTestAll(void);


/***********************************************************************
 * Macro definitions
 */
#define	TEST_CMD_FILE	"all.tst"
#define	OPTARG		"v:r:d:l:hs:gq"


/***********************************************************************
 * Global definitions
 */
int		IpcFd = -1;
unsigned long	Dflags;
char		*CmdFile = TEST_CMD_FILE;
FILE		*LogFp, *CmdFp;
int		ConfigRdramV = 0;
int             NumberStallCycles = 0;



/***********************************************************************
 * Local static definitions
 */
static void		usage(void);

#define BROM_SIZE 16384
#define VIRAGE2_SIZE (64*4)

int main(int argc, char **argv)
{

    int c; 
    int i;
    int rom[BROM_SIZE/4];
    int v2[VIRAGE2_SIZE/4];
    char line[1024];
    int do_quit = 0;
    char	*file;
    FILE	*fp = NULL;
    FILE *v_fp = NULL;
    int sysclk_scale = 1;
    int test =0;
    int g_value = 0;

    extern char *optarg;
    extern int optind, opterr, optopt;

    optind = 1;
    Dflags = 0;
    LogFp  = stdout;
    

    while ((c = getopt(argc, argv, OPTARG)) != EOF) {
        switch (c) {
        case 'd': {	/* Debug flag */
            Dflags = strtoul(optarg, NULL, 16);
            printf("Debug flag=0x%08lx\n", Dflags);
            break;
        }
        case 'l': {	/* Log file */
            file = optarg;
            printf("Log file = %s\n", file);
            if ((fp = fopen(file, "w+")) == NULL) {
                printf("Unable to open log: %s\n",
                       file);
            }
	    
            else{
                LogFp = fp;
	    }
	    
            break;
        }
        case 'h': {	/* Help */
            usage();
            exit(0);
        }
        case 'q': {
            do_quit = 1;
            break;
        }
	case 'g': {
	  /* run just one test for gate level */
	  g_value = 1;
	  break;
        }
	case 's': {
	  sysclk_scale = atoi(optarg);
	  break;
	}
        case 'r': {	/* Rom file */
            file = optarg;
            printf("ROM file = %s\n", file);
            if ((fp = fopen(file, "r")) == NULL) {
                printf("Unable to open ROM: %s\n", file);
                fp = NULL;
            }
            break;
        }

        case 'v': {	/* Virage 2 file */
            file = optarg;
            printf("Virage2 file = %s\n", file);
            if ((v_fp = fopen(file, "r")) == NULL) {
                printf("Unable to open ROM: %s\n", file);
                v_fp = NULL;
            }
            break;
        }

        default:
            break;
        }
    }  /* while */

    /*
     * Read ROM file if it exists
     */
    if (fp) {
        i = 0;
        while (i<BROM_SIZE/4) {
            fgets(line, 1024, fp);
            // Skip comment lines
            if (line[0] == '/') {
                continue;
            }
            if (sscanf(line, "%x", rom+i) > 0) {
                i++;
            }
        }

    }

    if (v_fp) {
        i = 0;
        while (i<VIRAGE2_SIZE/4) {
            fgets(line, 1024, v_fp);
            // Skip comment lines
            if (line[0] == '/') {
                continue;
            }
            if (sscanf(line, "%x", v2+i) > 0) {
                i++;
            }
        }

    }

    ExceptionHandlerInit();
    IpcInit(NULL);
    FindServer();

    do_keep_alive_socket(1);

    if(g_value == 1){
      fprintf(LogFp, "Entered Short Test\n");
      secure_mode_entry_app();
      BCP_STALL(2000);
      if (nms_store_test(sysclk_scale, VIRAGE0_CTRL_REG, 16, 1) != PASS) {
	fprintf(LogFp, "FAIL SHORT TEST\n");
      } else {
	fprintf(LogFp, "PASS SHORT TEST\n");
      }
      fflush(NULL);
      goto finish;
    }
    /* 
     * MI functional tests 
     */

    /*
     * Run this one first, since it depends on contents of 
     * V2.
     * Check that a recall happened on reset. May have to do
     * a pin reset
     */
    secure_mode_entry_app();
    if (v_fp) {
        if (nv_v2_reset_test(v2) != PASS) {
            fprintf(LogFp, "FAIL\n");
        } else {
            fprintf(LogFp, "PASS\n");
        }
    }
    secure_mode_exit();

    /*
     * XXX Need some stall before turning off recall after reset
     */
    BCP_STALL(200);
    if (mi_reg_reset_test(0, 0, 0, 0) != PASS) {
        fprintf(LogFp, "FAIL\n");
    } else {
        fprintf(LogFp, "PASS\n");
    }

    if (mi_sram_access_test(0, 0, 0, 0) != PASS) {
        fprintf(LogFp, "FAIL\n");
    } else {
        fprintf(LogFp, "PASS\n");
    }

    if (mi_pifram_access_test(0, 0, 0, 0) != PASS) {
        fprintf(LogFp, "FAIL\n");
    } else {
        fprintf(LogFp, "PASS\n");
    }
        
    if (mi_block_write_error_test(0, 0, 0, 0) != PASS) {
        fprintf(LogFp, "FAIL\n");
    } else {
        fprintf(LogFp, "PASS\n");
    }
        
    if (mi_register_test(0, 0, 0, 0) != PASS) {
        fprintf(LogFp, "FAIL\n");
    } else {
        fprintf(LogFp, "PASS\n");
    }

    if (mi_secure_timer_test(0, 0, 0, 0) != PASS) {
        fprintf(LogFp, "FAIL\n");
    } else {
        fprintf(LogFp, "PASS\n");
    }

    if (mi_fatal_trap_test(0, 0, 0, 0) != PASS) {
        fprintf(LogFp, "FAIL\n");
    } else {
        fprintf(LogFp, "PASS\n");
    }

    /*
     * Only run read test if file was given
     */
    if (fp != NULL) {
        if (mi_brom_read_test((int) rom, 0, 0, 0) != PASS) {
            fprintf(LogFp, "FAIL\n");
        } else {
            fprintf(LogFp, "PASS\n");
        }
    }

    /*
     * VIRAGE tests - always in secure mode, exit at the end
     */
    


    /* this is to set default values back in registers */
    bd_pin_reset();
    BCP_STALL(10);
    secure_mode_entry_app();
    BCP_STALL(2000);


    if (nv_store_recall_test(sysclk_scale, VIRAGE0_CTRL_REG, 16) != PASS) {
	fprintf(LogFp, "FAIL\n");
    } else {
        fprintf(LogFp, "PASS\n");
    }
    fflush(NULL);

    if (nv_store_recall_test(sysclk_scale, VIRAGE1_CTRL_REG, 16) != PASS) {
	fprintf(LogFp, "FAIL\n");
    } else {
        fprintf(LogFp, "PASS\n");
    }
    fflush(NULL);

    if (nv_store_recall_test(sysclk_scale, VIRAGE2_CTRL_REG, 64) != PASS) {
	fprintf(LogFp, "FAIL\n");
    } else {
        fprintf(LogFp, "PASS\n");
    }
    fflush(NULL);

    if (nv_recall_test(sysclk_scale, VIRAGE0_CTRL_REG) != PASS) {
	fprintf(LogFp, "FAIL\n");
    } else {
        fprintf(LogFp, "PASS\n");
    }
    fflush(NULL);
    if (nv_recall_test(sysclk_scale, VIRAGE1_CTRL_REG) != PASS) {
	fprintf(LogFp, "FAIL\n");
    } else {
        fprintf(LogFp, "PASS\n");
    }
    fflush(NULL);
    if (nv_recall_test(sysclk_scale, VIRAGE2_CTRL_REG) != PASS) {
	fprintf(LogFp, "FAIL\n");
    } else {
        fprintf(LogFp, "PASS\n");
    }
    fflush(NULL);

    if (nv_compare_test(sysclk_scale, VIRAGE0_CTRL_REG, 0, 0, 1) != PASS) {
	fprintf(LogFp, "FAIL\n");
    } else {
        fprintf(LogFp, "PASS\n");
    }
    fflush(NULL);

    if (nv_compare_test(sysclk_scale, VIRAGE1_CTRL_REG, 0, 0, 1) != PASS) {
	fprintf(LogFp, "FAIL\n");
    } else {
        fprintf(LogFp, "PASS\n");
    }
    fflush(NULL);
    if (nv_compare_test(sysclk_scale, VIRAGE2_CTRL_REG, 0, 0, 1) != PASS) {
	fprintf(LogFp, "FAIL\n");
    } else {
        fprintf(LogFp, "PASS\n");
    }
    fflush(NULL);

    if (nms_compare_test(sysclk_scale, VIRAGE0_CTRL_REG) != PASS) {
        fprintf(LogFp, "FAIL: virage 0\n");
    } else {
        fprintf(LogFp, "PASS: virage 0\n");
    }
    fflush(NULL);

    if (nms_compare_test_negative(sysclk_scale, VIRAGE0_CTRL_REG, 16) != PASS) {
        fprintf(LogFp, "FAIL: virage 0 negative\n");
    } else {
        fprintf(LogFp, "PASS: virage 0 negative\n");
    }
    fflush(NULL);

    if (nms_compare_test(sysclk_scale, VIRAGE1_CTRL_REG) != PASS) {
        fprintf(LogFp, "FAIL: virage 1\n");
    } else {
        fprintf(LogFp, "PASS: virage 1\n");
    }
    fflush(NULL);

    if (nms_compare_test_negative(sysclk_scale, VIRAGE1_CTRL_REG, 16) != PASS) {
        fprintf(LogFp, "FAIL: virage 1 negative\n");
    } else {
        fprintf(LogFp, "PASS: virage 1 negative\n");
    }
    fflush(NULL);

    if (nms_compare_test(sysclk_scale, VIRAGE2_CTRL_REG) != PASS) {
      fprintf(LogFp, "FAIL: virage 2\n");
    } else {
      fprintf(LogFp, "PASS: virage 2\n");
    }
    fflush(NULL);

    if (nms_compare_test_negative(sysclk_scale, VIRAGE2_CTRL_REG, 64) != PASS) {
        fprintf(LogFp, "FAIL: virage 2 negative\n");
    } else {
        fprintf(LogFp, "PASS: virage 2 negative\n");
    }
    fflush(NULL);

    if (nms_recall_auto_margin_test(sysclk_scale, VIRAGE0_CTRL_REG, 16) != PASS) {
      fprintf(LogFp, "FAIL\n");
    } else {
      fprintf(LogFp, "PASS\n");
    }
    fflush(NULL);

    if (nms_recall_auto_margin_test(sysclk_scale, VIRAGE1_CTRL_REG, 16) != PASS) {
        fprintf(LogFp, "FAIL\n");
    } else {
        fprintf(LogFp, "PASS\n");
    }
    fflush(NULL);

    if (nms_recall_auto_margin_test(sysclk_scale, VIRAGE2_CTRL_REG, 64) != PASS) {
        fprintf(LogFp, "FAIL\n");
    } else {
        fprintf(LogFp, "PASS\n");
    }
    fflush(NULL);

    /* loop over test number: last argument : runs fast test, default 
     parameter test and random parameter test*/
    for(test =0; test < 3; test++){
      if (nms_store_test(sysclk_scale, VIRAGE0_CTRL_REG, 16, test) != PASS) {
	fprintf(LogFp, "FAIL\n");
      } else {
        fprintf(LogFp, "PASS\n");
      }
      fflush(NULL);

      if (nms_store_test(sysclk_scale, VIRAGE1_CTRL_REG, 16, test) != PASS) {
	fprintf(LogFp, "FAIL\n");
      } else {
        fprintf(LogFp, "PASS\n");
      }
      fflush(NULL);
      if (nms_store_test(sysclk_scale, VIRAGE2_CTRL_REG, 64, test) != PASS) {
	fprintf(LogFp, "FAIL\n");
      } else {
        fprintf(LogFp, "PASS\n");
      }
      fflush(NULL);
    }

    

    if(nms_store_test_interrupt(sysclk_scale, VIRAGE0_CTRL_REG, 16) != PASS){
      fprintf(LogFp, "FAIL\n");
    } else {
      fprintf(LogFp, "PASS\n");
    }
    fflush(NULL);

    if (nms_keep_mode_test(sysclk_scale, VIRAGE0_CTRL_REG, 16) != PASS) {
        fprintf(LogFp, "FAIL\n");
    } else {
        fprintf(LogFp, "PASS\n");
    }
    fflush(NULL);
    if (nms_keep_mode_test(sysclk_scale, VIRAGE1_CTRL_REG, 16) != PASS) {
        fprintf(LogFp, "FAIL\n");
    } else {
        fprintf(LogFp, "PASS\n");
    }
    fflush(NULL);
    if (nms_keep_mode_test(sysclk_scale, VIRAGE2_CTRL_REG, 16) != PASS) {
        fprintf(LogFp, "FAIL\n");
    } else {
        fprintf(LogFp, "PASS\n");
    }
    fflush(NULL);
finish:
    secure_mode_exit();


    do_keep_alive_socket(0);

    /* exit simulator */
    if (do_quit) {
        printf("Quit\n");
        IoCmd(REQ_QUIT, 0);
    }

    if (fp)
        fclose(fp);

    if (IpcFd != -1)
        IpcClose(IpcFd);

    exit(0);

}

static void 
usage(void)
{
printf("Usage: mi_test \n");
printf("       <-l log_file>\n");
printf("       <-d debug_flags in hex>\n");
printf("       <-s scale (example 20)>\n");
printf("       <-g (short test for gate level)>\n");
}