/*
   * (C) Copyright 2007-2008 Semihalf
   *
   * Written by: Rafal Jaworowski <raj@semihalf.com>
   *
   * SPDX-License-Identifier:	GPL-2.0+
   */
  
  #include <common.h>
  #include <linux/types.h>
  #include <api_public.h>
  
  #include "glue.h"
  
  #define errf(fmt, args...) do { printf("ERROR @ %s(): ", __func__); printf(fmt, ##args); } while (0)
  
  #define BUF_SZ		2048
  #define WAIT_SECS	5
  
  void	test_dump_buf(void *, int);
  void	test_dump_di(int);
  void	test_dump_si(struct sys_info *);
  void	test_dump_sig(struct api_signature *);
  
  static char buf[BUF_SZ];
  
  int main(int argc, char * const argv[])
  {
  	int rv = 0, h, i, j, devs_no;
  	struct api_signature *sig = NULL;
  	ulong start, now;
  	struct device_info *di;
  	lbasize_t rlen;
  	struct display_info disinfo;
  
  	if (!api_search_sig(&sig))
  		return -1;
  
  	syscall_ptr = sig->syscall;
  	if (syscall_ptr == NULL)
  		return -2;
  
  	if (sig->version > API_SIG_VERSION)
  		return -3;
  
  	printf("API signature found @%x
  ", (unsigned int)sig);
  	test_dump_sig(sig);
  
  	printf("
  *** Consumer API test ***
  ");
  	printf("syscall ptr 0x%08x@%08x
  ", (unsigned int)syscall_ptr,
  		(unsigned int)&syscall_ptr);
  
  	/* console activities */
  	ub_putc('B');
  
  	printf("*** Press any key to continue ***
  ");
  	printf("got char 0x%x
  ", ub_getc());
  
  	/* system info */
  	test_dump_si(ub_get_sys_info());
  
  	/* timing */
  	printf("
  *** Timing - wait a couple of secs ***
  ");
  	start = ub_get_timer(0);
  	printf("
  time: start %lu
  
  ", start);
  	for (i = 0; i < WAIT_SECS; i++)
  		for (j = 0; j < 1000; j++)
  			ub_udelay(1000);	/* wait 1 ms */
  
  	/* this is the number of milliseconds that passed from ub_get_timer(0) */
  	now = ub_get_timer(start);
  	printf("
  time: now %lu
  
  ", now);
  
  	/* enumerate devices */
  	printf("
  *** Enumerate devices ***
  ");
  	devs_no = ub_dev_enum();
  
  	printf("Number of devices found: %d
  ", devs_no);
  	if (devs_no == 0)
  		return -1;
  
  	printf("
  *** Show devices ***
  ");
  	for (i = 0; i < devs_no; i++) {
  		test_dump_di(i);
  		printf("
  ");
  	}
  
  	printf("
  *** Operations on devices ***
  ");
  
  	/* test opening a device already opened */
  	h = 0;
  	if ((rv = ub_dev_open(h)) != 0) {
  		errf("open device %d error %d
  ", h, rv);
  		return -1;
  	}
  	if ((rv = ub_dev_open(h)) != 0)
  		errf("open device %d error %d
  ", h, rv);
  
  	ub_dev_close(h);
  
  	/* test storage */
  	printf("Trying storage devices...
  ");
  	for (i = 0; i < devs_no; i++) {
  		di = ub_dev_get(i);
  
  		if (di->type & DEV_TYP_STOR)
  			break;
  
  	}
  	if (i == devs_no)
  		printf("No storage devices available
  ");
  	else {
  		memset(buf, 0, BUF_SZ);
  
  		if ((rv = ub_dev_open(i)) != 0)
  			errf("open device %d error %d
  ", i, rv);
  
  		else if ((rv = ub_dev_read(i, buf, 1, 0, &rlen)) != 0)
  			errf("could not read from device %d, error %d
  ", i, rv);
  		else {
  			printf("Sector 0 dump (512B):
  ");
  			test_dump_buf(buf, 512);
  		}
  
  		ub_dev_close(i);
  	}
  
  	/* test networking */
  	printf("Trying network devices...
  ");
  	for (i = 0; i < devs_no; i++) {
  		di = ub_dev_get(i);
  
  		if (di->type == DEV_TYP_NET)
  			break;
  
  	}
  	if (i == devs_no)
  		printf("No network devices available
  ");
  	else {
  		if ((rv = ub_dev_open(i)) != 0)
  			errf("open device %d error %d
  ", i, rv);
  		else if ((rv = ub_dev_send(i, &buf, 2048)) != 0)
  			errf("could not send to device %d, error %d
  ", i, rv);
  
  		ub_dev_close(i);
  	}
  
  	if (ub_dev_close(h) != 0)
  		errf("could not close device %d
  ", h);
  
  	printf("
  *** Env vars ***
  ");
  
  	printf("ethact = %s
  ", ub_env_get("ethact"));
  	printf("old fileaddr = %s
  ", ub_env_get("fileaddr"));
  	ub_env_set("fileaddr", "deadbeef");
  	printf("new fileaddr = %s
  ", ub_env_get("fileaddr"));
  
  	const char *env = NULL;
  
  	while ((env = ub_env_enum(env)) != NULL)
  		printf("%s = %s
  ", env, ub_env_get(env));
  
  	printf("
  *** Display ***
  ");
  
  	if (ub_display_get_info(DISPLAY_TYPE_LCD, &disinfo)) {
  		printf("LCD info: failed
  ");
  	} else {
  		printf("LCD info:
  ");
  		printf("  pixel width:  %d
  ", disinfo.pixel_width);
  		printf("  pixel height: %d
  ", disinfo.pixel_height);
  		printf("  screen rows:  %d
  ", disinfo.screen_rows);
  		printf("  screen cols:  %d
  ", disinfo.screen_cols);
  	}
  	if (ub_display_get_info(DISPLAY_TYPE_VIDEO, &disinfo)) {
  		printf("video info: failed
  ");
  	} else {
  		printf("video info:
  ");
  		printf("  pixel width:  %d
  ", disinfo.pixel_width);
  		printf("  pixel height: %d
  ", disinfo.pixel_height);
  		printf("  screen rows:  %d
  ", disinfo.screen_rows);
  		printf("  screen cols:  %d
  ", disinfo.screen_cols);
  	}
  
  	printf("*** Press any key to continue ***
  ");
  	printf("got char 0x%x
  ", ub_getc());
  
  	/*
  	 * This only clears messages on screen, not on serial port. It is
  	 * equivalent to a no-op if no display is available.
  	 */
  	ub_display_clear();
  
  	/* reset */
  	printf("
  *** Resetting board ***
  ");
  	ub_reset();
  	printf("
  Hmm, reset returned...?!
  ");
  
  	return rv;
  }
  
  void test_dump_sig(struct api_signature *sig)
  {
  	printf("signature:
  ");
  	printf("  version\t= %d
  ", sig->version);
  	printf("  checksum\t= 0x%08x
  ", sig->checksum);
  	printf("  sc entry\t= 0x%08x
  ", (unsigned int)sig->syscall);
  }
  
  void test_dump_si(struct sys_info *si)
  {
  	int i;
  
  	printf("sys info:
  ");
  	printf("  clkbus\t= 0x%08x
  ", (unsigned int)si->clk_bus);
  	printf("  clkcpu\t= 0x%08x
  ", (unsigned int)si->clk_cpu);
  	printf("  bar\t\t= 0x%08x
  ", (unsigned int)si->bar);
  
  	printf("---
  ");
  	for (i = 0; i < si->mr_no; i++) {
  		if (si->mr[i].flags == 0)
  			break;
  
  		printf("  start\t= 0x%08lx
  ", si->mr[i].start);
  		printf("  size\t= 0x%08lx
  ", si->mr[i].size);
  
  		switch(si->mr[i].flags & 0x000F) {
  			case MR_ATTR_FLASH:
  				printf("  type FLASH
  ");
  				break;
  			case MR_ATTR_DRAM:
  				printf("  type DRAM
  ");
  				break;
  			case MR_ATTR_SRAM:
  				printf("  type SRAM
  ");
  				break;
  			default:
  				printf("  type UNKNOWN
  ");
  		}
  		printf("---
  ");
  	}
  }
  
  static char *test_stor_typ(int type)
  {
  	if (type & DT_STOR_IDE)
  		return "IDE";
  
  	if (type & DT_STOR_MMC)
  		return "MMC";
  
  	if (type & DT_STOR_SATA)
  		return "SATA";
  
  	if (type & DT_STOR_SCSI)
  		return "SCSI";
  
  	if (type & DT_STOR_USB)
  		return "USB";
  
  	return "Unknown";
  }
  
  void test_dump_buf(void *buf, int len)
  {
  	int i;
  	int line_counter = 0;
  	int sep_flag = 0;
  	int addr = 0;
  
  	printf("%07x:\t", addr);
  
  	for (i = 0; i < len; i++) {
  		if (line_counter++ > 15) {
  			line_counter = 0;
  			sep_flag = 0;
  			addr += 16;
  			i--;
  			printf("
  %07x:\t", addr);
  			continue;
  		}
  
  		if (sep_flag++ > 1) {
  			sep_flag = 1;
  			printf(" ");
  		}
  
  		printf("%02x", *((char *)buf++));
  	}
  
  	printf("
  ");
  }
  
  void test_dump_di(int handle)
  {
  	int i;
  	struct device_info *di = ub_dev_get(handle);
  
  	printf("device info (%d):
  ", handle);
  	printf("  cookie\t= 0x%08x
  ", (uint32_t)di->cookie);
  	printf("  type\t\t= 0x%08x
  ", di->type);
  
  	if (di->type == DEV_TYP_NET) {
  		printf("  hwaddr\t= ");
  		for (i = 0; i < 6; i++)
  			printf("%02x ", di->di_net.hwaddr[i]);
  
  		printf("
  ");
  
  	} else if (di->type & DEV_TYP_STOR) {
  		printf("  type\t\t= %s
  ", test_stor_typ(di->type));
  		printf("  blk size\t\t= %d
  ", (unsigned int)di->di_stor.block_size);
  		printf("  blk count\t\t= %d
  ", (unsigned int)di->di_stor.block_count);
  	}
  }