/*	$OpenBSD: build.c,v 1.9 2017/08/27 08:15:48 otto Exp $	*/

/*
 * Copyright (c) 2004 Theo de Raadt <deraadt@openbsd.org>
 *
 * Permission to use, copy, modify, and distribute this software for any
 * purpose with or without fee is hereby granted, provided that the above
 * copyright notice and this permission notice appear in all copies.
 *
 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 */
#include <sys/types.h>
#include <dev/pci/if_bnxreg.h>
#include <fcntl.h>
#include <stdlib.h>
#include <err.h>
#include <unistd.h>
#include <string.h>
#include <stdio.h>

#include "bnxfw.h"

#define nitems(_a) (sizeof((_a)) / sizeof((_a)[0]))

int	bnx_rv2p_proc1len;
int	bnx_rv2p_proc2len;

struct chunks {
	void *start;
	int *len;
};

#define FILENAME_B06 "bnx-b06"
struct chunks chunks_b06[] = {
	{ bnx_COM_b06FwText, &bnx_COM_b06FwTextLen },
	{ bnx_COM_b06FwData, &bnx_COM_b06FwDataLen },
	{ bnx_COM_b06FwRodata, &bnx_COM_b06FwRodataLen },
	{ bnx_COM_b06FwBss, &bnx_COM_b06FwBssLen },
	{ bnx_COM_b06FwSbss, &bnx_COM_b06FwSbssLen },

	{ bnx_RXP_b06FwText, &bnx_RXP_b06FwTextLen },
	{ bnx_RXP_b06FwData, &bnx_RXP_b06FwDataLen },
	{ bnx_RXP_b06FwRodata, &bnx_RXP_b06FwRodataLen },
	{ bnx_RXP_b06FwBss, &bnx_RXP_b06FwBssLen },
	{ bnx_RXP_b06FwSbss, &bnx_RXP_b06FwSbssLen },

	{ bnx_TPAT_b06FwText, &bnx_TPAT_b06FwTextLen },
	{ bnx_TPAT_b06FwData, &bnx_TPAT_b06FwDataLen },
	{ bnx_TPAT_b06FwRodata, &bnx_TPAT_b06FwRodataLen },
	{ bnx_TPAT_b06FwBss, &bnx_TPAT_b06FwBssLen },
	{ bnx_TPAT_b06FwSbss, &bnx_TPAT_b06FwSbssLen },

	{ bnx_TXP_b06FwText, &bnx_TXP_b06FwTextLen },
	{ bnx_TXP_b06FwData, &bnx_TXP_b06FwDataLen },
	{ bnx_TXP_b06FwRodata, &bnx_TXP_b06FwRodataLen },
	{ bnx_TXP_b06FwBss, &bnx_TXP_b06FwBssLen },
	{ bnx_TXP_b06FwSbss, &bnx_TXP_b06FwSbssLen }
};

#define FILENAME_B09 "bnx-b09"
struct chunks chunks_b09[] = {
	{ bnx_COM_b09FwText, &bnx_COM_b09FwTextLen },
	{ bnx_COM_b09FwData, &bnx_COM_b09FwDataLen },
	{ bnx_COM_b09FwRodata, &bnx_COM_b09FwRodataLen },
	{ bnx_COM_b09FwBss, &bnx_COM_b09FwBssLen },
	{ bnx_COM_b09FwSbss, &bnx_COM_b09FwSbssLen },

	{ bnx_RXP_b09FwText, &bnx_RXP_b09FwTextLen },
	{ bnx_RXP_b09FwData, &bnx_RXP_b09FwDataLen },
	{ bnx_RXP_b09FwRodata, &bnx_RXP_b09FwRodataLen },
	{ bnx_RXP_b09FwBss, &bnx_RXP_b09FwBssLen },
	{ bnx_RXP_b09FwSbss, &bnx_RXP_b09FwSbssLen },

	{ bnx_TPAT_b09FwText, &bnx_TPAT_b09FwTextLen },
	{ bnx_TPAT_b09FwData, &bnx_TPAT_b09FwDataLen },
	{ bnx_TPAT_b09FwRodata, &bnx_TPAT_b09FwRodataLen },
	{ bnx_TPAT_b09FwBss, &bnx_TPAT_b09FwBssLen },
	{ bnx_TPAT_b09FwSbss, &bnx_TPAT_b09FwSbssLen },

	{ bnx_TXP_b09FwText, &bnx_TXP_b09FwTextLen },
	{ bnx_TXP_b09FwData, &bnx_TXP_b09FwDataLen },
	{ bnx_TXP_b09FwRodata, &bnx_TXP_b09FwRodataLen },
	{ bnx_TXP_b09FwBss, &bnx_TXP_b09FwBssLen },
	{ bnx_TXP_b09FwSbss, &bnx_TXP_b09FwSbssLen }
};

#define FILENAME_RV2P "bnx-rv2p"
struct chunks chunks_rv2p[] = {
	{ bnx_rv2p_proc1, &bnx_rv2p_proc1len },
	{ bnx_rv2p_proc2, &bnx_rv2p_proc2len }
};

#define FILENAME_XI_RV2P "bnx-xi-rv2p"
struct chunks chunks_xi_rv2p[] = {
	{ bnx_xi_rv2p_proc1, &bnx_rv2p_proc1len },
	{ bnx_xi_rv2p_proc2, &bnx_rv2p_proc2len }
};

#define FILENAME_XI90_RV2P "bnx-xi90-rv2p"
struct chunks chunks_xi90_rv2p[] = {
	{ bnx_xi90_rv2p_proc1, &bnx_rv2p_proc1len },
	{ bnx_xi90_rv2p_proc2, &bnx_rv2p_proc2len }
};

void
hswapn(u_int32_t *p, int wcount)
{
	for (; wcount; wcount -=4) {
		*p = htonl(*p);
		p++;
	}
}

void
write_firmware(char *filename, void *header, size_t hlen,
    struct chunks *chunks, u_int nchunks)
{
	int fd, i, total;
	ssize_t rlen;

	printf("creating %s", filename);
	fd = open(filename, O_WRONLY|O_CREAT|O_TRUNC, 0644);
	if (fd == -1)
		err(1, "%s", filename);

	rlen = write(fd, header, hlen);
	if (rlen == -1)
		err(1, "%s", filename);
	if (rlen != hlen)
		errx(1, "%s: short write", filename);
	total = rlen;
	printf(" [%d", total);
	fflush(stdout);

	for (i = 0; i < nchunks; i++) {
		hswapn(chunks[i].start, *chunks[i].len);
		rlen = write(fd, chunks[i].start, *chunks[i].len);
		if (rlen == -1) {
			printf("\n");
			err(1, "%s", filename);
		}
		if (rlen != *chunks[i].len) {
			printf("\n");
			errx(1, "%s: short write", filename);
		}
		printf("+%zd", rlen);
		fflush(stdout);
		total += rlen;
	}

	printf("] total %d\n", total);

	close(fd);
}

int
main(int argc, char *argv[])
{
	struct	bnx_firmware_header *bf;
	struct	bnx_rv2p_header	*rh;

	bf = (struct bnx_firmware_header *)malloc(sizeof *bf);
	bzero(bf, sizeof *bf);

	/* initialize the file header */
	bf->bnx_COM_FwReleaseMajor = htonl(bnx_COM_b06FwReleaseMajor);
	bf->bnx_COM_FwReleaseMinor = htonl(bnx_COM_b06FwReleaseMinor);
	bf->bnx_COM_FwReleaseFix = htonl(bnx_COM_b06FwReleaseFix);
	bf->bnx_COM_FwStartAddr = htonl(bnx_COM_b06FwStartAddr);
	bf->bnx_COM_FwTextAddr = htonl(bnx_COM_b06FwTextAddr);
	bf->bnx_COM_FwTextLen = htonl(bnx_COM_b06FwTextLen);
	bf->bnx_COM_FwDataAddr = htonl(bnx_COM_b06FwDataAddr);
	bf->bnx_COM_FwDataLen = htonl(bnx_COM_b06FwDataLen);
	bf->bnx_COM_FwRodataAddr = htonl(bnx_COM_b06FwRodataAddr);
	bf->bnx_COM_FwRodataLen = htonl(bnx_COM_b06FwRodataLen);
	bf->bnx_COM_FwBssAddr = htonl(bnx_COM_b06FwBssAddr);
	bf->bnx_COM_FwBssLen = htonl(bnx_COM_b06FwBssLen);
	bf->bnx_COM_FwSbssAddr = htonl(bnx_COM_b06FwSbssAddr);
	bf->bnx_COM_FwSbssLen = htonl(bnx_COM_b06FwSbssLen);

	bf->bnx_RXP_FwReleaseMajor = htonl(bnx_RXP_b06FwReleaseMajor);
	bf->bnx_RXP_FwReleaseMinor = htonl(bnx_RXP_b06FwReleaseMinor);
	bf->bnx_RXP_FwReleaseFix = htonl(bnx_RXP_b06FwReleaseFix);
	bf->bnx_RXP_FwStartAddr = htonl(bnx_RXP_b06FwStartAddr);
	bf->bnx_RXP_FwTextAddr = htonl(bnx_RXP_b06FwTextAddr);
	bf->bnx_RXP_FwTextLen = htonl(bnx_RXP_b06FwTextLen);
	bf->bnx_RXP_FwDataAddr = htonl(bnx_RXP_b06FwDataAddr);
	bf->bnx_RXP_FwDataLen = htonl(bnx_RXP_b06FwDataLen);
	bf->bnx_RXP_FwRodataAddr = htonl(bnx_RXP_b06FwRodataAddr);
	bf->bnx_RXP_FwRodataLen = htonl(bnx_RXP_b06FwRodataLen);
	bf->bnx_RXP_FwBssAddr = htonl(bnx_RXP_b06FwBssAddr);
	bf->bnx_RXP_FwBssLen = htonl(bnx_RXP_b06FwBssLen);
	bf->bnx_RXP_FwSbssAddr = htonl(bnx_RXP_b06FwSbssAddr);
	bf->bnx_RXP_FwSbssLen = htonl(bnx_RXP_b06FwSbssLen);

	bf->bnx_TPAT_FwReleaseMajor = htonl(bnx_TPAT_b06FwReleaseMajor);
	bf->bnx_TPAT_FwReleaseMinor = htonl(bnx_TPAT_b06FwReleaseMinor);
	bf->bnx_TPAT_FwReleaseFix = htonl(bnx_TPAT_b06FwReleaseFix);
	bf->bnx_TPAT_FwStartAddr = htonl(bnx_TPAT_b06FwStartAddr);
	bf->bnx_TPAT_FwTextAddr = htonl(bnx_TPAT_b06FwTextAddr);
	bf->bnx_TPAT_FwTextLen = htonl(bnx_TPAT_b06FwTextLen);
	bf->bnx_TPAT_FwDataAddr = htonl(bnx_TPAT_b06FwDataAddr);
	bf->bnx_TPAT_FwDataLen = htonl(bnx_TPAT_b06FwDataLen);
	bf->bnx_TPAT_FwRodataAddr = htonl(bnx_TPAT_b06FwRodataAddr);
	bf->bnx_TPAT_FwRodataLen = htonl(bnx_TPAT_b06FwRodataLen);
	bf->bnx_TPAT_FwBssAddr = htonl(bnx_TPAT_b06FwBssAddr);
	bf->bnx_TPAT_FwBssLen = htonl(bnx_TPAT_b06FwBssLen);
	bf->bnx_TPAT_FwSbssAddr = htonl(bnx_TPAT_b06FwSbssAddr);
	bf->bnx_TPAT_FwSbssLen = htonl(bnx_TPAT_b06FwSbssLen);

	bf->bnx_TXP_FwReleaseMajor = htonl(bnx_TXP_b06FwReleaseMajor);
	bf->bnx_TXP_FwReleaseMinor = htonl(bnx_TXP_b06FwReleaseMinor);
	bf->bnx_TXP_FwReleaseFix = htonl(bnx_TXP_b06FwReleaseFix);
	bf->bnx_TXP_FwStartAddr = htonl(bnx_TXP_b06FwStartAddr);
	bf->bnx_TXP_FwTextAddr = htonl(bnx_TXP_b06FwTextAddr);
	bf->bnx_TXP_FwTextLen = htonl(bnx_TXP_b06FwTextLen);
	bf->bnx_TXP_FwDataAddr = htonl(bnx_TXP_b06FwDataAddr);
	bf->bnx_TXP_FwDataLen = htonl(bnx_TXP_b06FwDataLen);
	bf->bnx_TXP_FwRodataAddr = htonl(bnx_TXP_b06FwRodataAddr);
	bf->bnx_TXP_FwRodataLen = htonl(bnx_TXP_b06FwRodataLen);
	bf->bnx_TXP_FwBssAddr = htonl(bnx_TXP_b06FwBssAddr);
	bf->bnx_TXP_FwBssLen = htonl(bnx_TXP_b06FwBssLen);
	bf->bnx_TXP_FwSbssAddr = htonl(bnx_TXP_b06FwSbssAddr);
	bf->bnx_TXP_FwSbssLen = htonl(bnx_TXP_b06FwSbssLen);

	write_firmware(FILENAME_B06, bf, sizeof(*bf), chunks_b06,
	    nitems(chunks_b06));

	bzero(bf, sizeof *bf);

	bf->bnx_COM_FwReleaseMajor = htonl(bnx_COM_b09FwReleaseMajor);
	bf->bnx_COM_FwReleaseMinor = htonl(bnx_COM_b09FwReleaseMinor);
	bf->bnx_COM_FwReleaseFix = htonl(bnx_COM_b09FwReleaseFix);
	bf->bnx_COM_FwStartAddr = htonl(bnx_COM_b09FwStartAddr);
	bf->bnx_COM_FwTextAddr = htonl(bnx_COM_b09FwTextAddr);
	bf->bnx_COM_FwTextLen = htonl(bnx_COM_b09FwTextLen);
	bf->bnx_COM_FwDataAddr = htonl(bnx_COM_b09FwDataAddr);
	bf->bnx_COM_FwDataLen = htonl(bnx_COM_b09FwDataLen);
	bf->bnx_COM_FwRodataAddr = htonl(bnx_COM_b09FwRodataAddr);
	bf->bnx_COM_FwRodataLen = htonl(bnx_COM_b09FwRodataLen);
	bf->bnx_COM_FwBssAddr = htonl(bnx_COM_b09FwBssAddr);
	bf->bnx_COM_FwBssLen = htonl(bnx_COM_b09FwBssLen);
	bf->bnx_COM_FwSbssAddr = htonl(bnx_COM_b09FwSbssAddr);
	bf->bnx_COM_FwSbssLen = htonl(bnx_COM_b09FwSbssLen);

	bf->bnx_RXP_FwReleaseMajor = htonl(bnx_RXP_b09FwReleaseMajor);
	bf->bnx_RXP_FwReleaseMinor = htonl(bnx_RXP_b09FwReleaseMinor);
	bf->bnx_RXP_FwReleaseFix = htonl(bnx_RXP_b09FwReleaseFix);
	bf->bnx_RXP_FwStartAddr = htonl(bnx_RXP_b09FwStartAddr);
	bf->bnx_RXP_FwTextAddr = htonl(bnx_RXP_b09FwTextAddr);
	bf->bnx_RXP_FwTextLen = htonl(bnx_RXP_b09FwTextLen);
	bf->bnx_RXP_FwDataAddr = htonl(bnx_RXP_b09FwDataAddr);
	bf->bnx_RXP_FwDataLen = htonl(bnx_RXP_b09FwDataLen);
	bf->bnx_RXP_FwRodataAddr = htonl(bnx_RXP_b09FwRodataAddr);
	bf->bnx_RXP_FwRodataLen = htonl(bnx_RXP_b09FwRodataLen);
	bf->bnx_RXP_FwBssAddr = htonl(bnx_RXP_b09FwBssAddr);
	bf->bnx_RXP_FwBssLen = htonl(bnx_RXP_b09FwBssLen);
	bf->bnx_RXP_FwSbssAddr = htonl(bnx_RXP_b09FwSbssAddr);
	bf->bnx_RXP_FwSbssLen = htonl(bnx_RXP_b09FwSbssLen);

	bf->bnx_TPAT_FwReleaseMajor = htonl(bnx_TPAT_b09FwReleaseMajor);
	bf->bnx_TPAT_FwReleaseMinor = htonl(bnx_TPAT_b09FwReleaseMinor);
	bf->bnx_TPAT_FwReleaseFix = htonl(bnx_TPAT_b09FwReleaseFix);
	bf->bnx_TPAT_FwStartAddr = htonl(bnx_TPAT_b09FwStartAddr);
	bf->bnx_TPAT_FwTextAddr = htonl(bnx_TPAT_b09FwTextAddr);
	bf->bnx_TPAT_FwTextLen = htonl(bnx_TPAT_b09FwTextLen);
	bf->bnx_TPAT_FwDataAddr = htonl(bnx_TPAT_b09FwDataAddr);
	bf->bnx_TPAT_FwDataLen = htonl(bnx_TPAT_b09FwDataLen);
	bf->bnx_TPAT_FwRodataAddr = htonl(bnx_TPAT_b09FwRodataAddr);
	bf->bnx_TPAT_FwRodataLen = htonl(bnx_TPAT_b09FwRodataLen);
	bf->bnx_TPAT_FwBssAddr = htonl(bnx_TPAT_b09FwBssAddr);
	bf->bnx_TPAT_FwBssLen = htonl(bnx_TPAT_b09FwBssLen);
	bf->bnx_TPAT_FwSbssAddr = htonl(bnx_TPAT_b09FwSbssAddr);
	bf->bnx_TPAT_FwSbssLen = htonl(bnx_TPAT_b09FwSbssLen);

	bf->bnx_TXP_FwReleaseMajor = htonl(bnx_TXP_b09FwReleaseMajor);
	bf->bnx_TXP_FwReleaseMinor = htonl(bnx_TXP_b09FwReleaseMinor);
	bf->bnx_TXP_FwReleaseFix = htonl(bnx_TXP_b09FwReleaseFix);
	bf->bnx_TXP_FwStartAddr = htonl(bnx_TXP_b09FwStartAddr);
	bf->bnx_TXP_FwTextAddr = htonl(bnx_TXP_b09FwTextAddr);
	bf->bnx_TXP_FwTextLen = htonl(bnx_TXP_b09FwTextLen);
	bf->bnx_TXP_FwDataAddr = htonl(bnx_TXP_b09FwDataAddr);
	bf->bnx_TXP_FwDataLen = htonl(bnx_TXP_b09FwDataLen);
	bf->bnx_TXP_FwRodataAddr = htonl(bnx_TXP_b09FwRodataAddr);
	bf->bnx_TXP_FwRodataLen = htonl(bnx_TXP_b09FwRodataLen);
	bf->bnx_TXP_FwBssAddr = htonl(bnx_TXP_b09FwBssAddr);
	bf->bnx_TXP_FwBssLen = htonl(bnx_TXP_b09FwBssLen);
	bf->bnx_TXP_FwSbssAddr = htonl(bnx_TXP_b09FwSbssAddr);
	bf->bnx_TXP_FwSbssLen = htonl(bnx_TXP_b09FwSbssLen);

	write_firmware(FILENAME_B09, bf, sizeof(*bf), chunks_b09,
	    nitems(chunks_b09));

	free(bf);

	rh = (struct bnx_rv2p_header *)malloc(sizeof *rh);

	bzero(rh, sizeof *rh);
	bnx_rv2p_proc1len = sizeof bnx_rv2p_proc1;
	bnx_rv2p_proc2len = sizeof bnx_rv2p_proc2;
	rh->bnx_rv2p_proc1len = htonl(bnx_rv2p_proc1len);
	rh->bnx_rv2p_proc2len = htonl(bnx_rv2p_proc2len);

	write_firmware(FILENAME_RV2P, rh, sizeof(*rh), chunks_rv2p,
	    nitems(chunks_rv2p));

	bzero(rh, sizeof *rh);
	bnx_rv2p_proc1len = sizeof bnx_xi_rv2p_proc1;
	bnx_rv2p_proc2len = sizeof bnx_xi_rv2p_proc2;
	rh->bnx_rv2p_proc1len = htonl(bnx_rv2p_proc1len);
	rh->bnx_rv2p_proc2len = htonl(bnx_rv2p_proc2len);

	write_firmware(FILENAME_XI_RV2P, rh, sizeof(*rh), chunks_xi_rv2p,
	    nitems(chunks_xi_rv2p));

	bzero(rh, sizeof *rh);
	bnx_rv2p_proc1len = sizeof bnx_xi90_rv2p_proc1;
	bnx_rv2p_proc2len = sizeof bnx_xi90_rv2p_proc2;
	rh->bnx_rv2p_proc1len = htonl(bnx_rv2p_proc1len);
	rh->bnx_rv2p_proc2len = htonl(bnx_rv2p_proc2len);

	write_firmware(FILENAME_XI90_RV2P, rh, sizeof(*rh), chunks_xi90_rv2p,
	    nitems(chunks_xi90_rv2p));

	free(rh);

	return 0;
}