/* Copyright (c) 2012 The Chromium OS Authors. All rights reserved.
 * Use of this source code is governed by a BSD-style license that can be
 * found in the LICENSE file.
 */

/* Console commands to trigger flash host commands */

#include "board.h"
#include "console.h"
#include "ec_commands.h"
#include "host_command.h"
#include "uart.h"
#include "util.h"

static int ro_image_size(int argc, char **argv)
{
	uart_printf("RO image size = 0x%x\n", CONFIG_SECTION_RO_SIZE);
	return EC_SUCCESS;
}
DECLARE_CONSOLE_COMMAND(rosize, ro_image_size,
			NULL,
			"Report size of RO image",
			NULL);

/* TODO(victoryang@): We should introduce a function to send fake host command
 *                    just like ec_command in ectool. See crosbug/p/11350 */
static int hc_flash_info(int argc, char **argv)
{
	uint8_t data[EC_HOST_PARAM_SIZE];
	enum ec_status res;
	struct ec_response_flash_info *r;
	struct host_cmd_handler_args args =
		{ .command = EC_CMD_FLASH_INFO,
		  .version = 0,
		  .params = NULL,
		  .params_size = 0,
		  .response = data,
		  .response_size = EC_HOST_PARAM_SIZE };

	res = host_command_process(&args);
	if (res != EC_RES_SUCCESS)
		return EC_ERROR_UNKNOWN;
	r = (struct ec_response_flash_info *)args.response;
	uart_printf("flash_size = %d\n", r->flash_size);
	uart_printf("write_block_size = %d\n", r->write_block_size);
	uart_printf("erase_block_size = %d\n", r->erase_block_size);
	uart_printf("protect_block_size = %d\n", r->protect_block_size);

	return EC_SUCCESS;
}
DECLARE_CONSOLE_COMMAND(hcflashinfo, hc_flash_info,
			NULL, NULL, NULL);

static int hc_flash_read(int argc, char **argv)
{
	uint8_t data[EC_HOST_PARAM_SIZE];
	enum ec_status res;
	struct ec_params_flash_read *p =
		(struct ec_params_flash_read *)data;
	struct host_cmd_handler_args args =
		{ .command = EC_CMD_FLASH_READ,
		  .version = 0,
		  .params = data,
		  .params_size = EC_HOST_PARAM_SIZE,
		  .response = data,
		  .response_size = EC_HOST_PARAM_SIZE };
	char *e;
	int i, size;

	if (argc != 3)
		return EC_ERROR_PARAM_COUNT;

	p->offset = strtoi(argv[1], &e, 0);
	if (*e)
		return EC_ERROR_PARAM1;
	size = strtoi(argv[2], &e, 0);
	p->size = size;
	if (*e)
		return EC_ERROR_PARAM2;

	res = host_command_process(&args);
	if (res != EC_RES_SUCCESS)
		return EC_ERROR_UNKNOWN;
	for (i = 0; i < size; ++i) {
		uart_printf("%02x", args.response[i]);
		if ((i & 31) == 31)
			uart_puts("\n");
	}
	return EC_SUCCESS;
}
DECLARE_CONSOLE_COMMAND(hcflashread, hc_flash_read,
			NULL, NULL, NULL);

static int hc_flash_write(int argc, char **argv)
{
	uint8_t data[EC_HOST_PARAM_SIZE];
	enum ec_status res;
	struct ec_params_flash_write *p =
		(struct ec_params_flash_write *)data;
	struct host_cmd_handler_args args =
		{ .command = EC_CMD_FLASH_WRITE,
		  .version = 0,
		  .params = data,
		  .params_size = EC_HOST_PARAM_SIZE,
		  .response = data,
		  .response_size = EC_HOST_PARAM_SIZE };
	char *e;
	int i, size;
	int seed, mult, add;

	if (argc != 6)
		return EC_ERROR_PARAM_COUNT;

	p->offset = strtoi(argv[1], &e, 0);
	if (*e)
		return EC_ERROR_PARAM1;
	size = strtoi(argv[2], &e, 0);
	p->size = size;
	if (*e)
		return EC_ERROR_PARAM2;
	seed = strtoi(argv[3], &e, 0);
	if (*e)
		return EC_ERROR_PARAM3;
	mult = strtoi(argv[4], &e, 0);
	if (*e)
		return EC_ERROR_PARAM4;
	add = strtoi(argv[5], &e, 0);
	if (*e)
		return EC_ERROR_PARAM5;

	for (i = 0; i < size; ++i) {
		p->data[i] = (uint8_t)(seed & 0xff);
		seed = seed * mult + add;
	}

	res = host_command_process(&args);
	if (res != EC_RES_SUCCESS)
		return EC_ERROR_UNKNOWN;
	return EC_SUCCESS;
}
DECLARE_CONSOLE_COMMAND(hcflashwrite, hc_flash_write,
			NULL, NULL, NULL);

static int hc_flash_erase(int argc, char **argv)
{
	uint8_t data[EC_HOST_PARAM_SIZE];
	enum ec_status res;
	struct ec_params_flash_erase *p =
		(struct ec_params_flash_erase *)data;
	struct host_cmd_handler_args args =
		{ .command = EC_CMD_FLASH_ERASE,
		  .version = 0,
		  .params = data,
		  .params_size = EC_HOST_PARAM_SIZE,
		  .response = data,
		  .response_size = EC_HOST_PARAM_SIZE };
	char *e;
	int size;

	if (argc != 3)
		return EC_ERROR_PARAM_COUNT;

	p->offset = strtoi(argv[1], &e, 0);
	if (*e)
		return EC_ERROR_PARAM1;
	size = strtoi(argv[2], &e, 0);
	p->size = size;
	if (*e)
		return EC_ERROR_PARAM2;

	res = host_command_process(&args);
	if (res != EC_RES_SUCCESS)
		return EC_ERROR_UNKNOWN;
	return EC_SUCCESS;
}
DECLARE_CONSOLE_COMMAND(hcflasherase, hc_flash_erase,
			NULL, NULL, NULL);