csu3_am335x/EVSE/GPL/plc-utils.LNX.2.1.20/source/pib/modpib.c

/*====================================================================*
*
*   Copyright (c) 2013 Qualcomm Atheros, Inc.
*
*   All rights reserved.
*
*====================================================================*/

/*====================================================================*
 *
 *   modpib.c - 
 *
 *   Contributor(s):
 *      Charles Maier <cmaier@qca.qualcomm.com>
 *      Nathaniel Houghton <nhoughto@qca.qualcomm.com>
 *
 *--------------------------------------------------------------------*/

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

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <ctype.h>
#include <errno.h>
#include <sys/stat.h>
#include <netinet/in.h>
#include <arpa/inet.h>

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

#include "../tools/getoptv.h"
#include "../tools/number.h"
#include "../tools/symbol.h"
#include "../tools/flags.h"
#include "../tools/error.h"
#include "../tools/files.h"
#include "../tools/chars.h"
#include "../plc/plc.h"
#include "../key/HPAVKey.h"
#include "../key/keys.h"
#include "../pib/pib.h"

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

#ifndef MAKEFILE
#include "../tools/getoptv.c"
#include "../tools/putoptv.c"
#include "../tools/version.c"
#include "../tools/checksum32.c"
#include "../tools/fdchecksum32.c"
#include "../tools/uintspec.c"
#include "../tools/hexencode.c"
#include "../tools/hexdecode.c"
#include "../tools/hexstring.c"
#include "../tools/todigit.c"
#include "../tools/memincr.c"
#include "../tools/synonym.c"
#include "../tools/error.c"
#endif

#ifndef MAKEFILE
#include "../pib/pibfile.c"
#include "../pib/lightning_pib_file.c"
#include "../pib/lightning_pib_peek.c"
#include "../pib/panther_pib_file.c"
#include "../pib/panther_pib_peek.c"
#endif

#ifndef MAKEFILE
#include "../key/HPAVKeyNID.c"
#include "../key/SHA256Reset.c"
#include "../key/SHA256Write.c"
#include "../key/SHA256Block.c"
#include "../key/SHA256Fetch.c"
#include "../key/keys.c"
#endif

/*====================================================================*
 *   constants;
 *--------------------------------------------------------------------*/

#define MEMBERSHIP_OFFSET 0x00001E8C
#define MEMBERSHIP_STATUS 0x00000000

/*====================================================================*
 *
 *   signed pibimage1 (struct _file_ * file, simple_pib *sample_pib, signed level, flag_t flags);
 * 
 *   modify selected PIB header values and compute a new checksum;
 *   this function assumes that the file is open and positioned to
 *   the start of the parameter block;
 *
 *   Contributor(s):
 *	Nathaniel Houghton <nhoughto@qca.qualcomm.com>
 *	Charles Maier <cmaier@qca.qualcomm.com>
 *
 *--------------------------------------------------------------------*/

static signed pibimage1 (struct _file_ * file, simple_pib * sample_pib, signed level, flag_t flags) 

{ 
	struct simple_pib simple_pib; 
	memset (& simple_pib, 0, sizeof (simple_pib)); 
	if (read (file->file, & simple_pib, sizeof (simple_pib)) != sizeof (simple_pib)) 
	{ 
		error (1, errno, FILE_CANTREAD, file->name); 
	} 
	if (_anyset (flags, PIB_MAC)) 
	{ 
		memcpy (simple_pib.MAC, sample_pib->MAC, sizeof (simple_pib.MAC)); 
	} 
	if (_anyset (flags, PIB_MACINC)) 
	{ 
		memincr (simple_pib.MAC, sizeof (simple_pib.MAC)); 
	} 
	if (_anyset (flags, PIB_DAK)) 
	{ 
		memcpy (simple_pib.DAK, sample_pib->DAK, sizeof (simple_pib.DAK)); 
	} 
	if (_anyset (flags, PIB_NMK)) 
	{ 
		memcpy (simple_pib.NMK, sample_pib->NMK, sizeof (simple_pib.NMK)); 
	} 
	if (_anyset (flags, PIB_NETWORK)) 
	{ 
		memcpy (simple_pib.NET, sample_pib->NET, sizeof (simple_pib.NET)); 
	} 
	if (_anyset (flags, PIB_MFGSTRING)) 
	{ 
		memcpy (simple_pib.MFG, sample_pib->MFG, sizeof (simple_pib.MFG)); 
	} 
	if (_anyset (flags, PIB_USER)) 
	{ 
		memcpy (simple_pib.USR, sample_pib->USR, sizeof (simple_pib.USR)); 
	} 
	if (_anyset (flags, PIB_CCO_MODE)) 
	{ 
		simple_pib.CCoSelection = sample_pib->CCoSelection; 
	} 
	if (_anyset (flags, PIB_NMK | PIB_NID)) 
	{ 

#if 0

/*
 *	clear the AVLNMembership bit whenever the NMK or NID change; this is an important
 *	step because it prevents false network association;
 */

		if (BE16TOH (simple_pib.PIBVERSION) > 0x0200) 
		{ 
			uint32_t membership = MEMBERSHIP_STATUS; 
			if (lseek (file->file, MEMBERSHIP_OFFSET, SEEK_SET) != MEMBERSHIP_OFFSET) 
			{ 
				error (1, errno, FILE_CANTHOME, file->name); 
			} 
			if (write (file->file, & membership, sizeof (membership)) != sizeof (membership)) 
			{ 
				error (1, errno, FILE_CANTSAVE, file->name); 
			} 
		} 

#endif

		if (_allclr (flags, PIB_NID)) 
		{ 
			level = simple_pib.PreferredNID [HPAVKEY_NID_LEN - 1] >> 4; 
		} 
		HPAVKeyNID (simple_pib.PreferredNID, simple_pib.NMK, level & 1); 
	} 
	if (lseek (file->file, (off_t) (0) - sizeof (simple_pib), SEEK_CUR) == - 1) 
	{ 
		error (1, errno, FILE_CANTHOME, file->name); 
	} 
	if (write (file->file, & simple_pib, sizeof (simple_pib)) != sizeof (simple_pib)) 
	{ 
		error (1, errno, FILE_CANTSAVE, file->name); 
	} 
	if (lseek (file->file, (off_t) (0) - sizeof (simple_pib), SEEK_CUR) == - 1) 
	{ 
		error (1, errno, FILE_CANTHOME, file->name); 
	} 
	simple_pib.CHECKSUM = fdchecksum32 (file->file, LE16TOH (simple_pib.PIBLENGTH), simple_pib.CHECKSUM); 
	if (lseek (file->file, (off_t) (0) - LE16TOH (simple_pib.PIBLENGTH), SEEK_CUR) == - 1) 
	{ 
		error (1, errno, FILE_CANTHOME, file->name); 
	} 
	if (write (file->file, & simple_pib, sizeof (simple_pib)) != sizeof (simple_pib)) 
	{ 
		error (1, errno, FILE_CANTSAVE, file->name); 
	} 
	if (_anyset (flags, PIB_VERBOSE)) 
	{ 
		lightning_pib_peek (& simple_pib); 
	} 
	return (0); 
} 

/*====================================================================*
 *
 *   signed pibimage2 (struct _file_ * file, simple_pib *sample_pib, signed level, flag_t flags);
 * 
 *   modify selected PIB header values but do not compute a checksum;
 *
 *   Contributor(s):
 *	Charles Maier <cmaier@qca.qualcomm.com>
 *
 *--------------------------------------------------------------------*/

static signed pibimage2 (struct _file_ * file, simple_pib * sample_pib, signed level, flag_t flags) 

{ 
	struct simple_pib simple_pib; 
	memset (& simple_pib, 0, sizeof (simple_pib)); 
	if (read (file->file, & simple_pib, sizeof (simple_pib)) != sizeof (simple_pib)) 
	{ 
		error (1, errno, FILE_CANTREAD, file->name); 
	} 
	if (_anyset (flags, PIB_MAC)) 
	{ 
		memcpy (simple_pib.MAC, sample_pib->MAC, sizeof (simple_pib.MAC)); 
	} 
	if (_anyset (flags, PIB_MACINC)) 
	{ 
		memincr (simple_pib.MAC, sizeof (simple_pib.MAC)); 
	} 
	if (_anyset (flags, PIB_DAK)) 
	{ 
		memcpy (simple_pib.DAK, sample_pib->DAK, sizeof (simple_pib.DAK)); 
	} 
	if (_anyset (flags, PIB_NMK)) 
	{ 
		memcpy (simple_pib.NMK, sample_pib->NMK, sizeof (simple_pib.NMK)); 
	} 
	if (_anyset (flags, PIB_NETWORK)) 
	{ 
		memcpy (simple_pib.NET, sample_pib->NET, sizeof (simple_pib.NET)); 
	} 
	if (_anyset (flags, PIB_MFGSTRING)) 
	{ 
		memcpy (simple_pib.MFG, sample_pib->MFG, sizeof (simple_pib.MFG)); 
	} 
	if (_anyset (flags, PIB_USER)) 
	{ 
		memcpy (simple_pib.USR, sample_pib->USR, sizeof (simple_pib.USR)); 
	} 
	if (_anyset (flags, PIB_CCO_MODE)) 
	{ 
		simple_pib.CCoSelection = sample_pib->CCoSelection; 
	} 
	if (_anyset (flags, PIB_NMK | PIB_NID)) 
	{ 

#if 0

/*
 *	clear the AVLNMembership bit whenever the NMK or NID change; this is an important
 *	step because it prevents false network association;
 */

		if (BE16TOH (simple_pib.PIBVERSION) > 0x0200) 
		{ 
			uint32_t membership = MEMBERSHIP_STATUS; 
			if (lseek (file->file, MEMBERSHIP_OFFSET, SEEK_SET) != MEMBERSHIP_OFFSET) 
			{ 
				error (1, errno, FILE_CANTHOME, file->name); 
			} 
			if (write (file->file, & membership, sizeof (membership)) != sizeof (membership)) 
			{ 
				error (1, errno, FILE_CANTSAVE, file->name); 
			} 
		} 

#endif

		if (_allclr (flags, PIB_NID)) 
		{ 
			level = simple_pib.PreferredNID [HPAVKEY_NID_LEN - 1] >> 4; 
		} 
		HPAVKeyNID (simple_pib.PreferredNID, simple_pib.NMK, level & 1); 
	} 
	if (lseek (file->file, (off_t) (0) - sizeof (simple_pib), SEEK_CUR) == - 1) 
	{ 
		error (1, errno, FILE_CANTHOME, file->name); 
	} 
	if (write (file->file, & simple_pib, sizeof (simple_pib)) != sizeof (simple_pib)) 
	{ 
		error (1, errno, FILE_CANTSAVE, file->name); 
	} 
	if (lseek (file->file, (off_t) (0) - sizeof (simple_pib), SEEK_CUR) == - 1) 
	{ 
		error (1, errno, FILE_CANTHOME, file->name); 
	} 
	if (_anyset (flags, PIB_VERBOSE)) 
	{ 
		panther_pib_peek (& simple_pib); 
	} 
	return (0); 
} 

/*====================================================================*
 *
 *   signed pibchain2 (struct _file_ * file, simple_pib * sample_pib, signed level, flag_t flags);
 *
 *   Contributor(s):
 *	Charles Maier <cmaier@qca.qualcomm.com>
 *
 *--------------------------------------------------------------------*/

static signed pibchain2 (struct _file_ * file, simple_pib * sample_pib, signed level, flag_t flags) 

{ 
	unsigned module = 0; 
	struct panther_nvm_header header; 
	uint32_t prev = ~ 0; 
	uint32_t next = 0; 
	if (lseek (file->file, 0, SEEK_SET)) 
	{ 
		error (1, errno, NVM_IMG_CHECKSUM, file->name, module); 
	} 
	do 
	{ 
		if (read (file->file, & header, sizeof (header)) != sizeof (header)) 
		{ 
			error (1, errno, NVM_HDR_CANTREAD, file->name, module); 
		} 
		if (LE16TOH (header.MajorVersion) != 1) 
		{ 
			error (1, errno, NVM_HDR_VERSION, file->name, module); 
		} 
		if (LE16TOH (header.MinorVersion) != 1) 
		{ 
			error (1, errno, NVM_HDR_VERSION, file->name, module); 
		} 
		if (checksum32 (& header, sizeof (header), 0)) 
		{ 
			error (1, errno, NVM_HDR_CHECKSUM, file->name, module); 
		} 
		if (LE32TOH (header.PrevHeader) != prev) 
		{ 
			error (1, errno, NVM_HDR_LINK, file->name, module); 
		} 
		if (LE32TOH (header.ImageType) == NVM_IMAGE_PIB) 
		{ 
			pibimage2 (file, sample_pib, level, flags); 
			header.ImageChecksum = fdchecksum32 (file->file, LE32TOH (header.ImageLength), 0); 
			if (lseek (file->file, (off_t) (0) - LE32TOH (header.ImageLength), SEEK_CUR) == - 1) 
			{ 
				error (1, errno, FILE_CANTHOME, file->name); 
			} 
			header.HeaderChecksum = checksum32 (& header, sizeof (header), header.HeaderChecksum); 
			if (lseek (file->file, (off_t) (0) - sizeof (header), SEEK_CUR) == - 1) 
			{ 
				error (1, errno, FILE_CANTHOME, file->name); 
			} 
			if (write (file->file, & header, sizeof (header)) != sizeof (header)) 
			{ 
				error (1, errno, FILE_CANTSAVE, file->name); 
			} 
			if (lseek (file->file, (off_t) (0) - sizeof (header), SEEK_CUR) == - 1) 
			{ 
				error (1, errno, FILE_CANTHOME, file->name); 
			} 
			break; 
		} 
		if (fdchecksum32 (file->file, LE32TOH (header.ImageLength), header.ImageChecksum)) 
		{ 
			error (1, errno, NVM_IMG_CHECKSUM, file->name, module); 
		} 
		prev = next; 
		next = LE32TOH (header.NextHeader); 
		module++; 
	} 
	while (~ header.NextHeader); 
	return (0); 
} 

/*====================================================================*
 *
 *   signed function (char const * filename, struct simple_pib * simple_pib, unsigned level, flag_t flags);
 *
 *   Contributor(s):
 *	Charles Maier <cmaier@qca.qualcomm.com>
 *
 *--------------------------------------------------------------------*/

static signed function (char const * filename, struct simple_pib * sample_pib, unsigned level, flag_t flags) 

{ 
	uint32_t version; 
	signed status; 
	struct _file_ file; 
	file.name = filename; 
	if ((file.file = open (file.name, O_BINARY | O_RDWR, FILE_FILEMODE)) == - 1) 
	{ 
		if (_allclr (flags, PIB_SILENCE)) 
		{ 
			error (0, errno, FILE_CANTOPEN, file.name); 
		} 
		return (- 1); 
	} 
	if (read (file.file, & version, sizeof (version)) != sizeof (version)) 
	{ 
		if (_allclr (flags, PIB_SILENCE)) 
		{ 
			error (0, errno, FILE_CANTREAD, file.name); 
		} 
		return (- 1); 
	} 
	if (lseek (file.file, 0, SEEK_SET)) 
	{ 
		error (1, errno, FILE_CANTHOME, file.name); 
	} 
	if (LE32TOH (version) == 0x00010001) 
	{ 
		status = pibchain2 (& file, sample_pib, level, flags); 
	} 
	else 
	{ 
		status = pibimage1 (& file, sample_pib, level, flags); 
	} 
	close (file.file); 
	return (status); 
} 

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

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

{ 
	extern struct _term_ const daks []; 
	extern struct _term_ const nmks []; 
	static char const * optv [] = 
	{ 
		"C:D:L:M:N:S:T:U:v", 
		"file [file] [...]", 
		"modify selected PIB parameters and update checksum", 
		"C n\tCCo Selection is n", 
		"D x\tDAK as xx:xx:xx:xx:xx:xx:xx:xx:xx:xx:xx:xx:xx:xx:xx:xx", 
		"L n\tsecurity level is n", 
		"M x\tMAC as xx:xx:xx:xx:xx:xx", 
		"N x\tNMK as xx:xx:xx:xx:xx:xx:xx:xx:xx:xx:xx:xx:xx:xx:xx:xx", 
		"S s\tMFG string is s", 
		"T s\tNET string is s", 
		"U s\tUSR string is s", 
		"v\tverbose messages", 
		(char const *) (0)
	}; 
	struct simple_pib sample_pib; 
	signed level = 0; 
	signed state = 0; 
	flag_t flags = (flag_t) (0); 
	signed c; 
	optind = 1; 
	memset (& sample_pib, 0, sizeof (sample_pib)); 
	while (~ (c = getoptv (argc, argv, optv))) 
	{ 
		char const * sp; 
		switch ((char) (c)) 
		{ 
		case 'C': 
			_setbits (flags, PIB_CCO_MODE); 
			sample_pib.CCoSelection = (uint8_t) (uintspec (optarg, 0, 4)); 
			break; 
		case 'D': 
			if (!hexencode (sample_pib.DAK, sizeof (sample_pib.DAK), synonym (optarg, daks, SIZEOF (daks))))
			{
				error (1, errno, PLC_BAD_DAK, optarg);
			}
			_setbits (flags, PIB_DAK); 
			break; 
		case 'L': 
			level = (uint8_t) (uintspec (optarg, 0, 1)); 
			_setbits (flags, PIB_NID); 
			break; 
		case 'M': 
			if (! strcmp (optarg, "auto")) 
			{ 
				_setbits (flags, PIB_MACINC); 
				break; 
			} 
			if (! strcmp (optarg, "next")) 
			{ 
				_setbits (flags, PIB_MACINC); 
				break; 
			} 
			if (! strcmp (optarg, "plus")) 
			{ 
				_setbits (flags, PIB_MACINC); 
				break; 
			} 
			if (! hexencode (sample_pib.MAC, sizeof (sample_pib.MAC), optarg)) 
			{ 
				error (1, errno, PLC_BAD_MAC, optarg); 
			} 
			_setbits (flags, PIB_MAC); 
			break; 
		case 'N': 
			if (!hexencode (sample_pib.NMK, sizeof (sample_pib.NMK), synonym (optarg, nmks, SIZEOF (nmks))))
			{
				error (1, errno, PLC_BAD_NMK, optarg);
			}
			_setbits (flags, PIB_NMK); 
			break; 
		case 'S': 
			for (sp = optarg; isprint ((unsigned char)* sp); ++ sp);
			if (* sp) 
			{ 
				error (1, EINVAL, "NMK contains illegal characters"); 
			} 
			if ((sp - optarg) > (signed) (sizeof (sample_pib.MFG) - 1)) 
			{ 
				error (1, 0, "Manufacturing string is at most %u chars", (unsigned) (sizeof (sample_pib.MFG) - 1)); 
			} 
			memcpy (sample_pib.MFG, optarg, strlen (optarg)); 
			_setbits (flags, PIB_MFGSTRING); 
			break; 
		case 'T': 
			for (sp = optarg; isprint ((unsigned char)* sp); ++ sp); 
			if (* sp) 
			{ 
				error (1, EINVAL, "NMK contains illegal characters"); 
			} 
			if ((sp - optarg) > (signed) (sizeof (sample_pib.NET) - 1)) 
			{ 
				error (1, 0, "NET string is at most %u chars", (unsigned) (sizeof (sample_pib.NET) - 1)); 
			} 
			memcpy (sample_pib.NET, optarg, strlen (optarg)); 
			_setbits (flags, PIB_NETWORK); 
			break; 
		case 'U': 
			for (sp = optarg; isprint ((unsigned char)* sp); ++ sp); 
			if (* sp) 
			{ 
				error (1, EINVAL, "NMK contains illegal characters"); 
			} 
			if ((sp - optarg) > (signed) (sizeof (sample_pib.USR) - 1)) 
			{ 
				error (1, 0, "USR string is at most %u chars", (unsigned) (sizeof (sample_pib.USR) - 1)); 
			} 
			memcpy (sample_pib.USR, optarg, strlen (optarg)); 
			_setbits (flags, PIB_USER); 
			break; 
		case 'q': 
			_setbits (flags, PIB_SILENCE); 
			break; 
		case 'v': 
			_setbits (flags, PIB_VERBOSE); 
			break; 
		default: 
			break; 
		} 
	} 
	argc -= optind; 
	argv += optind; 
	while ((argc) && (* argv)) 
	{ 
		if (function (* argv, & sample_pib, level, flags)) 
		{ 
			state = 1; 
		} 
		argc--; 
		argv++; 
	} 
	exit (state); 
}