/*

 *

 *	   This application program demonstrates the user program

 *	   interface to the User Flash PROM driver.

 *

 *	   One can read or write a number of bytes up to the size of

 *	   the user PROM by means of pread() and pwrite() calls.

 *	   All other functions of the PROM can be reached by the means

 *	   of ioctl() calls such as:

 *	   	-) identify the chip,

 *	   	-) query block,

 *		-) lock block/unlock block,

 *		-) master lock,

 *		-) erase block, erase all unlocked blocks, and 

 *		   erase whole PROM

 *	   Please note that not all of the above ioctl calls are 

 *	   available for all flash PROMs. It is the user's responsibility

 *	   to find out the features of a given PROM. The type, block size,

 *	   and number of blocks of the PROM are returned by 

 *	   "identify" ioctl().

 *

 *	   The pwrite() erases the block[s] and then does the

 *	   writing.

 *	   The driver uses the buffered write. If the buffered write

 * 	   is not supported in a particular PROM, the non-buffered

 *	   writes are used instead. The buffered write is 15 folds

 *	   faster than the non-buffered write. 

 *

 *	   Use the following line to compile your custom application

 *	   programs:

 *	        make uflash_test

 */

#pragma ident   "@(#)uflash_test.c 1.3     99/08/03 SMI"

#include <stdio.h>

#include <sys/signal.h>

#include <stdio.h>

#include <sys/time.h>

#include <errno.h>

#include <sys/types.h>

#include <sys/fcntl.h>

#include <sys/stream.h>

#include "uflash_if.h"

/*

      * PROM size: 4 or 8 MBytes

      * Uncomment the right block

      */

     #if 1

     #define PROM_SIZE 0x400000 /* 4 MBytes */

     #endif

     #if 0

     #define PROM_SIZE 0x800000 /* 8 MBytes */

     #endif

static char *help[14] = {

	"0 -- read     user flash PROM",

	"1 -- write    user flash PROM",

	"2 -- identify user flash PROM",

	"3 -- query    blocks",

	"4 -- lock     blocks",

	"5 -- master   lock",

	"6 -- clear    all locks",

	"7 -- erase    blocks",

	"8 -- erase    all unlocked blocks",

	"9 -- erase    whole PROM",

	"a -- switch   PROMs",

	"q -- quit",

	"?/h -- display this menu",

	""

};

/*char            get_cmd(); */

static char

get_cmd()

{

		char    buf[10];

		gets(buf);

		return (buf[0]);

}

/*

 * Main 

 */

main(int argc, char *argv[]) 

{

	int 			n_byte;			/* returned from pread/pwrite */

	int 			size, offset, pat;

	int			fd0, fd1, h, i;

	int			fd, prom_id; 	

	uflash_if_t			uflash_if;

	caddr_t 	r_buf, 	w_buf;

	char 		*devname0 = "/dev/uflash0";

	char 		*devname1 = "/dev/uflash1";

	char		c;

	/*

	 * Assume that the PROM size is 4 MB.

	 */

	r_buf = (caddr_t)malloc(PROM_SIZE);

	w_buf = (caddr_t)malloc(PROM_SIZE);

	/*

	 * Open the user flash PROM #0.

	 */

	if ((fd0 = open(devname0, O_RDWR)) < 0) {

		fprintf(stderr, "couldn't open device: %s\n", devname0);

		exit(1);

	}

	/*

	 * Open the user flash PROM #1.

	 */

	if ((fd1 = open(devname1, O_RDWR)) < 0) {

		fprintf(stderr, "couldn't open device: %s\n", devname1);

		exit(1);

	}

	/* set the default PROM */

	prom_id = 0;

	fd = fd0;

	/* let them know about the help menu */

        fprintf(stderr, "Enter <h> or <?> for help on commands\n");

	while (1) {

		fprintf(stderr, "[%d]command> ", prom_id);

		switch(get_cmd()) {

		case 'q':

			goto getout;

		case 'h':

		case '?':

			h = 0;

			while (*help[h]){

				fprintf(stderr, "	%s\n", help[h]);

				h++;

			}

			break;

		case 'a':		/* switch PROM */

			fd = (fd == fd0)? fd1: fd0;

			prom_id = (prom_id == 1)? 0: 1;

			break;

		case '9':       /* erase the whole flash PROM */

			fprintf(stderr,

                                "Are you sure?[y/n]");

                        scanf ("%c", &c);

			if (c != 'y')

				continue;

			if (ioctl(fd, UIOCEFUL, &uflash_if) == -1)

                                goto getout;

			break;

		case '8':       /* erase all unlocked flash PROM blocks */

			/*

			 * This ioctl is valid only for those

			 * chips that have query command.

			 */

			if (ioctl(fd, UIOCEALL, &uflash_if) == -1)

                                goto getout;

			break;

		case '7':       /* erase flash PROM block */

			fprintf(stderr,

                                "Enter PROM block number[0, 31]> ");

                        scanf ("%d", &uflash_if.blk_num);

			fprintf(stderr, 

				"Enter number of block> ");

			scanf ("%d", &uflash_if.num_of_blks);

			if (ioctl(fd, UIOCEBLK, &uflash_if) == -1)

                                goto getout;

			break;

		case '6':       /* clear all locks */

			/* on certain PROMs */

			if (ioctl(fd, UIOCCLCK, &uflash_if) == -1)

                                goto getout;

			break;

		case '5':       /* master lock */

			/* on certain PROMs */

			if (ioctl(fd, UIOCMLCK, &uflash_if) == -1)

                                goto getout;

			break;

		case '4':       /* lock flash PROM block */

			/* on certain PROMs */

			fprintf(stderr,

                                "Enter PROM block number[0, 31]> ");

                        scanf ("%d", &uflash_if.blk_num);

			fprintf(stderr, 

				"Enter number of block> ");

			scanf ("%d", &uflash_if.num_of_blks);

			if (ioctl(fd, UIOCLBLK, &uflash_if) == -1)

                                goto getout;

			break;

		case '3':       /* query flash PROM */

			/* on certain PROMs */

			fprintf(stderr,

                                "Enter PROM block number[0, 31]> ");

                        scanf ("%d", &uflash_if.blk_num);

			fprintf(stderr, 

				"Enter number of block> ");

			scanf ("%d", &uflash_if.num_of_blks);

			if (ioctl(fd, UIOCQBLK, &uflash_if) == -1)

                                goto getout;

			for (i = uflash_if.blk_num; 

				i < (uflash_if.blk_num+uflash_if.num_of_blks); 

				i++) 

			{

				fprintf(stderr, "block[%d] status = %x\n", 

					i, uflash_if.info.blk_status[i] & 0xF);

			}

			break;

		case '2':       /* identify flash PROM */

			if (ioctl(fd, UIOCIBLK, &uflash_if) == -1)

                                goto getout;

			fprintf(stderr, "manufacturer id = 0x%x, device id =\

						0x%x\n# of blks = %d, blk size = 0x%x\n",

					uflash_if.info.mfr_id & 0xFF,

					uflash_if.info.dev_id & 0xFF,

					uflash_if.info.blk_num,

					uflash_if.info.blk_size);

			break;

		case '1':       /* write to user flash PROM */

			fprintf(stderr,

                                "Enter PROM offset[0, 0xXX,XXXX]> ");

                        scanf ("%x", &offset);

			fprintf(stderr, 

				"Enter number of bytes[hex]> ");

			scanf ("%x", &size);

			fprintf(stderr,

                                "Enter data pattern[0, 0xFF]> ");

                        scanf ("%x", &pat);

			/*

	 		 * init write buffer.

	 		 */

			for (i = 0; i < size; i++) {

				w_buf[i] = pat;

			}

			n_byte = pwrite (fd, w_buf, size, offset);

			if (n_byte != size) {

				/* the write failed */

				printf ("Write process was failed at byte 0x%x \n",

					n_byte);

			}

			break;

		case '0':	/* read from user flash PROM */

			fprintf(stderr,

                                "Enter PROM offset[0, 0xXX,XXXX]> ");

                        scanf ("%x", &offset);

			fprintf(stderr, 

				"Enter number of bytes[hex]> ");

			scanf ("%x", &size);

			getchar();	/* clean up the char buf */

			n_byte = pread (fd, r_buf, size, offset);

			if (n_byte != size) {

                                /* the read failed */

                                printf ("Read process was failed at \

							byte 0x%x \n",

                                        n_byte);

				continue;

                        }

			printf ("\nuser data buffer:\n");

			for (i = 0; i < size; i++) {

				printf("%2x ", r_buf[i] & 0xff);

			}

		printf("\n");

		default:

			continue;

		}

	}

	/* exit */

getout:

	close(fd0);

	close(fd1);

	return;

} /* end of main() */