/*
 * 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);
}