/*
 * Copyright (c) 2014 Brian Swetland
 *
 * Permission is hereby granted, free of charge, to any person obtaining
 * a copy of this software and associated documentation files
 * (the "Software"), to deal in the Software without restriction,
 * including without limitation the rights to use, copy, modify, merge,
 * publish, distribute, sublicense, and/or sell copies of the Software,
 * and to permit persons to whom the Software is furnished to do so,
 * subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be
 * included in all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
 * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
 * IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY
 * CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
 * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
 * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
 */

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <errno.h>
#include <fcntl.h>
#include <sys/types.h>

#include "liblkboot.h"

void usage(void) {
	fprintf(stderr,
"usage: lkboot <hostname> <command> ...\n"
"\n"
"       lkboot <hostname> flash <partition> <filename>\n"
"       lkboot <hostname> erase <partition>\n"
"       lkboot <hostname> remove <partition>\n"
"       lkboot <hostname> fpga <bitfile>\n"
"       lkboot <hostname> boot <binary>\n"
"       lkboot <hostname> getsysparam <name>\n"
"       lkboot <hostname> reboot\n"
"       lkboot <hostname> :<commandname> [ <arg>* ]\n"
"\n"
"NOTE: If <hostname> is 'jtag', lkboot will attempt to use\n"
"       a tool 'zynq-dcc' to communicate with the device.\n"
"       Make sure it is in your path.\n"
"\n"
	);
	exit(1);
}

void printsysparam(void *data, int len) {
	unsigned char *x = data;
	int i;
	for (i = 0; i < len; i++) {
		if ((x[i] < ' ') || (x[1] > 127)) goto printhex;
	}
	write(STDERR_FILENO, "\"", 1);
	write(STDERR_FILENO, data, len);
	write(STDERR_FILENO, "\"\n", 2);
	return;
printhex:
	fprintf(stderr, "[");
	for (i = 0; i < len; i++) fprintf(stderr, " %02x", x[i]);
	fprintf(stderr, " ]\n");
}

int main(int argc, char **argv) {
	const char *host = argv[1];
	const char *cmd = argv[2];
	const char *args = argv[3];
	const char *fn = NULL;
	int fd = -1;

	if (argc == 3) {
		if (!strcmp(cmd, "reboot")) {
			return lkboot_txn(host, cmd, fd, "");
		} else if (cmd[0] == ':') {
			return lkboot_txn(host, cmd + 1, fd, "");
		} else {
			usage();
		}
	}

	if (argc < 4) usage();

	if (!strcmp(cmd, "flash")) {
		if (argc < 5) usage();
		fn = argv[4];
	} else if (!strcmp(cmd, "fpga")) {
		fn = args;
		args = "";
	} else if (!strcmp(cmd, "boot")) {
		fn = args;
		args = "";
	} else if (!strcmp(cmd, "erase")) {
	} else if (!strcmp(cmd, "remove")) {
	} else if (!strcmp(cmd, "getsysparam")) {
		if (lkboot_txn(host, cmd, -1, args) == 0) {
			void *rbuf = NULL;
			printsysparam(rbuf, lkboot_get_reply(&rbuf));
			return 0;
		} else {
			return -1;
		}
	} else if (cmd[0] == ':') {
		return lkboot_txn(host, cmd + 1, -1, args);
	} else {
		usage();
	}
	if (fn) {
		if ((fd = open(fn, O_RDONLY)) < 0) {
			fprintf(stderr, "error; cannot open '%s'\n", fn);
			return -1;
		}
	}
	return lkboot_txn(host, cmd, fd, args);
}

// vim: noexpandtab