// Copyright (c) 2003-2009 Nokia Corporation and/or its subsidiary(-ies).
// All rights reserved.
// This component and the accompanying materials are made available
// under the terms of "Eclipse Public License v1.0"
// which accompanies this distribution, and is available
// at the URL "http://www.eclipse.org/legal/epl-v10.html".
//
// Initial Contributors:
// Nokia Corporation - initial contribution.
//
// Contributors:
//
// Description:
// Add a simple coff header to a binary file
// optionally strip off the first 256 bytes
// This is primarily for generating an image
// file that can be downloaded via the ether
// to cogent using the SmartFirmWare boot rom
// 
//

#include <sys/stat.h>
#include <fcntl.h>
#include <io.h>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>

// little error functionet
void error(const char* p)
{
	fprintf(stderr, "%s\n", p);
	exit(1);
}

// little usage functionet
void usage(const char* p)
{
	fprintf(stderr, "%s\n", p);
	exit(EXIT_FAILURE);
}


int main(int argc, char* argv[])
{
	// check for usage
	if (argc < 3 || argc > 4)
		{
		usage("Usage: bin2coff infile outfile nostrip\nOption nostrip: don't strip off the first 256 bytes");
		}

	// make sure we are not operating on only one file
	if (!strcmp(argv[1], argv[2]))
		{
		usage("Usage: bin2coff must specify different input and output files");
		}

	// try to open the input file
	FILE *pInFile;
	if((pInFile  = fopen(argv[1], "rb" )) == NULL)
   		{
   		error("Cannot open input file");
   		}

	// open output file
	FILE *pOutFile;
	if((pOutFile = fopen(argv[2], "wb") ) == NULL)
		{
   		error("Cannot open output file");
		}

	// strip the 256 long header?
	// ok, so it's simple, so what
	bool stripheader = (argc == 3);

	// how big is the input file
	struct stat filelength;
	stat(argv[1], &filelength);
	unsigned long fsize = filelength.st_size;
	if (stripheader)
		{
		fsize -= 256;
		}

	// write out the coff header to the new file
	// somewhere to put a simple coff header
	unsigned char coffhead[0x58] = {0};  // zero all the elements

	// fill in the constant bits
	// this is supposed to be simple, remember
	coffhead[1] = 0x0a;
	coffhead[2] = 0x01;
	coffhead[0x10] = 0x1c;
	coffhead[0x12] = 0x0f;
	coffhead[0x13] = 0xa1;
	coffhead[0x14] = 0x0b;
	coffhead[0x15] = 0x01;
	coffhead[0x26] = 0x40; // entry point 0x00400000
	coffhead[0x2a] = 0x40;
	coffhead[0x30] = 0x2e;
	coffhead[0x31] = 0x74;
	coffhead[0x32] = 0x65;
	coffhead[0x33] = 0x78;
	coffhead[0x34] = 0x74;
	coffhead[0x3a] = 0x40;
	coffhead[0x3e] = 0x40;
	coffhead[0x44] = 0x58;
	coffhead[0x54] = 0x20;

	// now fill in the text segment size
	*(int *) (&coffhead[0x18]) = fsize;
	*(int *) (&coffhead[0x40]) = fsize;

	// write out the new file
	// first the header
	int numwritten=0;
	numwritten = fwrite(coffhead, sizeof(char), 0x58, pOutFile);
	if (numwritten < 0x58)
		{
   		error("The number of items written is less than 0x58");
		}

	// now the rest of the file
	if (stripheader)
		{
		fpos_t filePositionIndicator = 0x100;
		if (fsetpos(pInFile, &filePositionIndicator) !=0)
			{
			error("The file is not accessible ");
			}
		}
	
	static const int buffsize = 1024;
	unsigned char fbuff[buffsize];
	while (!feof(pInFile))
	        {
			int nread = fread(fbuff,sizeof(unsigned char), sizeof(fbuff), pInFile);
			int nwritten = fwrite(fbuff, sizeof(unsigned char), nread, pOutFile);
			if (nwritten < nread)
		        {
				error("The number of items written is less than number of items read");
				}
			}
   
	fclose(pOutFile);
	fclose(pInFile);

	return 0;
}