nvmsplit.c
Author: cmaier@cmassoc.net

    /*====================================================================*
     *
     *   Copyright (c) 2013 Qualcomm Atheros, Inc.
     *
     *   All rights reserved.
     *
     *   Redistribution and use in source and binary forms, with or
     *   without modification, are permitted (subject to the limitations
     *   in the disclaimer below) provided that the following conditions
     *   are met:
     *
     *   * Redistributions of source code must retain the above copyright
     *     notice, this list of conditions and the following disclaimer.
     *
     *   * Redistributions in binary form must reproduce the above
     *     copyright notice, this list of conditions and the following
     *     disclaimer in the documentation and/or other materials
     *     provided with the distribution.
     *
     *   * Neither the name of Qualcomm Atheros nor the names of
     *     its contributors may be used to endorse or promote products
     *     derived from this software without specific prior written
     *     permission.
     *
     *   NO EXPRESS OR IMPLIED LICENSES TO ANY PARTY'S PATENT RIGHTS ARE
     *   GRANTED BY THIS LICENSE. THIS SOFTWARE IS PROVIDED BY THE
     *   COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY EXPRESS OR
     *   IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
     *   WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
     *   PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER
     *   OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
     *   SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
     *   NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
     *   LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
     *   HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
     *   CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE
     *   OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
     *   SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
     *
     *--------------------------------------------------------------------*/

    /*====================================================================*"
     *
     *   nvmsplit.c -
     *
     *
     *   Contributor(s):
     *      Charles Maier <cmaier@qca.qualcomm.com>
     *
     *--------------------------------------------------------------------*/

    /*====================================================================*
     *   system header files;
     *--------------------------------------------------------------------*/

    #include <stdio.h>
    #include <unistd.h>
    #include <string.h>
    #include <errno.h>

    /*====================================================================*
     *   custom header files;
     *--------------------------------------------------------------------*/

    #include "../tools/getoptv.h"
    #include "../tools/memory.h"
    #include "../tools/flags.h"
    #include "../tools/files.h"
    #include "../tools/error.h"
    #include "../nvm/nvm.h"

    /*====================================================================*
     *   custom source files;
     *--------------------------------------------------------------------*/

    #ifndef MAKEFILE
    #include "../tools/getoptv.c"
    #include "../tools/putoptv.c"
    #include "../tools/version.c"
    #include "../tools/checksum32.c"
    #include "../tools/error.c"
    #endif

    /*====================================================================*
     *
     *   void function1 (char const * filename, flag_t flags);
     *
     *   open an Atheros Qualcomm PLC firmware image file and write each
     *   component as an independent image file;
     *
     *
     *   Contributor(s):
     *	Charles Maier <cmaier@qca.qualcomm.com>
     *
     *--------------------------------------------------------------------*/

    static void function1 (char const * filename, flag_t flags)

    {
    	struct nvm_header1 nvm_header;
    	char savename [FILENAME_MAX];
    	byte buffer [BUFSIZ];
    	signed image = 0;
    	signed offset = 0;
    	signed ifd;
    	signed ofd;
    	char * sp;
    	char * cp;
    	memset (&nvm_header, 0, sizeof (nvm_header));
    	for (sp = cp = strcpy (savename, filename); *cp; cp++)
    	{
    		if (*cp == FILE_C_EXTENDER)
    		{
    			sp = cp;
    		}
    	}
    	if (sp == savename)
    	{
    		sp = cp;
    	}
    	if ((ifd = open (filename, O_BINARY|O_RDONLY)) == -1)
    	{
    		error (1, errno, FILE_CANTOPEN, filename);
    	}
    	do
    	{
    		signed extent;
    		signed length;
    		if (lseek (ifd, 0, SEEK_CUR) != offset)
    		{
    			error (1, errno, NVM_HDR_LINK, filename, image);
    		}
    		if (read (ifd, &nvm_header, sizeof (nvm_header)) != sizeof (nvm_header))
    		{
    			error (1, errno, NVM_HDR_CANTREAD, filename, image);
    		}
    		if (LE32TOH (nvm_header.HEADERVERSION) != 0x60000000)
    		{
    			error (1, 0, NVM_HDR_VERSION, filename, image);
    		}
    		if (checksum32 (&nvm_header, sizeof (nvm_header), 0))
    		{
    			error (1, 0, NVM_HDR_CHECKSUM, filename, image);
    		}

    #if defined (WIN32)

    		sprintf (sp, "-%02d.nvm", image);

    #else

    		snprintf (sp, savename + sizeof (savename) - sp, "-%02d.nvm", image);

    #endif

    		if ((ofd = open (savename, O_BINARY|O_CREAT|O_RDWR|O_TRUNC, FILE_FILEMODE)) == -1)
    		{
    			error (1, errno, FILE_CANTOPEN, savename);
    		}
    		if (_anyset (flags, NVM_VERBOSE))
    		{
    			error (0, 0, "%s", savename);
    		}
    		offset = LE32TOH (nvm_header.NEXTHEADER);
    		nvm_header.NEXTHEADER = 0;
    		nvm_header.HEADERCHECKSUM = 0;
    		nvm_header.HEADERCHECKSUM = checksum32 (&nvm_header, sizeof (nvm_header), 0);
    		if (write (ofd, &nvm_header, sizeof (nvm_header)) != sizeof (nvm_header))
    		{
    			error (1, errno, FILE_CANTSAVE, savename);
    		}
    		extent = LE32TOH (nvm_header.IMAGELENGTH);
    		length = sizeof (buffer);
    		while (extent)
    		{
    			if (length > extent)
    			{
    				length = extent;
    			}
    			if (read (ifd, buffer, length) < length)
    			{
    				error (1, errno, NVM_IMG_CANTREAD, filename, image);
    			}
    			if (write (ofd, buffer, length) < length)
    			{
    				error (1, errno, NVM_IMG_CANTSAVE, savename, image);
    			}
    			extent -= length;
    		}
    		close (ofd);
    		image++;
    	}
    	while (offset);
    	close (ifd);
    	return;
    }

    /*====================================================================*
     *
     *   void function2 (char const * filename, flag_t flags);
     *
     *   open an Atheros Qualcomm PLC firmware image file and write each
     *   component as an independent image file;
     *
     *
     *   Contributor(s):
     *	Charles Maier <cmaier@qca.qualcomm.com>
     *
     *--------------------------------------------------------------------*/

    static void function2 (char const * filename, flag_t flags)

    {
    	struct nvm_header2 nvm_header;
    	char savename [FILENAME_MAX];
    	byte buffer [BUFSIZ];
    	signed image = 0;
    	signed offset = 0;
    	signed ifd;
    	signed ofd;
    	char * sp;
    	char * cp;
    	memset (&nvm_header, 0, sizeof (nvm_header));
    	for (sp = cp = strcpy (savename, filename); *cp; cp++)
    	{
    		if (*cp == FILE_C_EXTENDER)
    		{
    			sp = cp;
    		}
    	}
    	if (sp == savename)
    	{
    		sp = cp;
    	}
    	if ((ifd = open (filename, O_BINARY|O_RDONLY)) == -1)
    	{
    		error (1, errno, FILE_CANTOPEN, filename);
    	}
    	do
    	{
    		signed extent;
    		signed length;
    		if (lseek (ifd, 0, SEEK_CUR) != offset)
    		{
    			error (1, errno, NVM_HDR_LINK, filename, image);
    		}
    		if (read (ifd, &nvm_header, sizeof (nvm_header)) != sizeof (nvm_header))
    		{
    			error (1, errno, NVM_HDR_CANTREAD, filename, image);
    		}
    		if (LE16TOH (nvm_header.MinorVersion) != 1)
    		{
    			error (1, 0, NVM_HDR_VERSION, filename, image);
    		}
    		if (LE16TOH (nvm_header.MajorVersion) != 1)
    		{
    			error (1, 0, NVM_HDR_VERSION, filename, image);
    		}
    		if (checksum32 (&nvm_header, sizeof (nvm_header), 0))
    		{
    			error (1, 0, NVM_HDR_CHECKSUM, filename, image);
    		}

    #if defined (WIN32)

    		sprintf (sp, "-%02d.nvm", image);

    #else

    		snprintf (sp, savename + sizeof (savename) - sp, "-%02d.nvm", image);

    #endif

    		if ((ofd = open (savename, O_BINARY|O_WRONLY|O_CREAT|O_TRUNC, FILE_FILEMODE)) == -1)
    		{
    			error (1, errno, FILE_CANTOPEN, savename);
    		}
    		if (_anyset (flags, NVM_VERBOSE))
    		{
    			error (0, 0, "%s --> %s", filename, savename);
    		}
    		offset = LE32TOH (nvm_header.NextHeader);
    		nvm_header.NextHeader = ~0;
    		nvm_header.PrevHeader = ~0;
    		nvm_header.HeaderChecksum = 0;
    		nvm_header.HeaderChecksum = checksum32 (&nvm_header, sizeof (nvm_header), 0);
    		if (write (ofd, &nvm_header, sizeof (nvm_header)) != sizeof (nvm_header))
    		{
    			error (1, errno, NVM_HDR_CANTSAVE, savename, image);
    		}
    		extent = LE32TOH (nvm_header.ImageLength);
    		length = sizeof (buffer);
    		while (extent)
    		{
    			if (length > extent)
    			{
    				length = extent;
    			}
    			if (read (ifd, buffer, length) < length)
    			{
    				error (1, errno, NVM_IMG_CANTREAD, filename, image);
    			}
    			if (write (ofd, buffer, length) < length)
    			{
    				error (1, errno, NVM_IMG_CANTSAVE, savename, image);
    			}
    			extent -= length;
    		}
    		close (ofd);
    		image++;
    	}
    	while (~offset);
    	close (ifd);
    	return;
    }

    /*====================================================================*
     *
     *   void function (struct _file_ * file, flag_t flags);
     *
     *   open an Atheros Qualcomm PLC firmware image file and write each
     *   component as an independent, single-image firmware file;
     *
     *
     *   Contributor(s):
     *	Charles Maier <cmaier@qca.qualcomm.com>
     *
     *--------------------------------------------------------------------*/

    static void function (char const * filename, flag_t flags)

    {
    	uint32_t version;
    	signed ifd;
    	if ((ifd = open (filename, O_BINARY|O_RDONLY)) == -1)
    	{
    		error (1, errno, FILE_CANTOPEN, filename);
    	}
    	if (read (ifd, &version, sizeof (version)) != sizeof (version))
    	{
    		error (1, errno, FILE_CANTREAD, filename);
    	}
    	close (ifd);
    	if (LE32TOH (version) == 0x60000000)
    	{
    		function1 (filename, flags);
    	}
    	else
    	{
    		function2 (filename, flags);
    	}
    	return;
    }

    /*====================================================================*
     *
     *   int main (int argc, char const * argv []);
     *
     *
     *   Contributor(s):
     *	Charles Maier <cmaier@qca.qualcomm.com>
     *
     *--------------------------------------------------------------------*/

    int main (int argc, char const * argv [])

    {
    	static char const * optv [] =
    	{
    		"qv",
    		"file [file] [...]",
    		"Qualcomm Atheros PLC Firmware Image File Slicer",
    		"q\tsuppress messages",
    		"v\tverbose messages",
    		(char const *) (0)
    	};
    	flag_t flags = (flag_t)(0);
    	signed c;
    	optind = 1;
    	while ((c = getoptv (argc, argv, optv)) != -1)
    	{
    		switch (c)
    		{
    		case 'q':
    			_setbits (flags, NVM_SILENCE);
    			break;
    		case 'v':
    			_setbits (flags, NVM_VERBOSE);
    			break;
    		default:
    			break;
    		}
    	}
    	argc -= optind;
    	argv += optind;
    	while ((argc) && (* argv))
    	{
    		function (* argv, flags);
    		argc--;
    		argv++;
    	}
    	return (0);
    }

[PREV] [HOME] [NEXT]