#

# Copyright (c) 2009 Nokia Corporation and/or its subsidiary(-ies).

# All rights reserved.

# This component and the accompanying materials are made available

# under the terms of the License "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: 

#

package spitool;

use strict;

use Exporter;

use vars qw($VERSION @ISA @EXPORT @EXPORT_OK);

use romutl;

$VERSION     = 1.00;

@ISA         = qw(Exporter);

@EXPORT      = ();

@EXPORT_OK   = qw(createSpi);

sub binarize { #converts decimal number to 4 byte litte endian format

	my $value = shift;

	my $remainder;

	my $returnValue;

	for(my $i=0;$i<4;$i++) {

		$remainder=$value % 256;

		$returnValue.=chr($remainder);

		$value = ($value-$remainder)/256;

	}

	return $returnValue;

}

sub convertUidFromText { #converts UID from hexadeciaml text value to decimal value, passes decimal value unchanged, returns -1 if invalid UID

	my $value = shift;

	if($value =~ /^0x([\da-fA-F]{1,8})$/i) {

		return hex $1;

	} elsif ($value =~ /^\d*$/) {

		return $value;

	} else {

		return -1;

	}

}

sub bin2hex { #converts 4 byte little endian value to 0x... hex text value

	my $value=shift;

	my $byte;

	my $quotient;

	my $remainder;

	my $hexValue="";

	for(my $i=0;$i<4;$i++) {

		$byte=ord(substr($value,$i,1));

		$remainder=$byte%16;

		$quotient=($byte-$remainder)/16;

		if($remainder>9) {

			$remainder= chr($remainder+55);

		}

		if($quotient>9) {

			$quotient= chr($quotient+55);

		}		

		$hexValue=$quotient . $remainder . $hexValue;

	}

	return "0x" . $hexValue;

}

sub uidcrc { #returns decimal UID checksum value for the three inputs

	my $output = `uidcrc $_[0] $_[1] $_[2]`;

	if($output =~ /([^ ]*)$/i) {

		$output =$1;

		chomp $output;

		return hex($output);

	}	

}

sub printZeroes { #prints as many hexadecimal zeroes to OUTPUTFILE as specified by input

	my $numberOfZeroes=shift;

	for(my $i=0;$i<$numberOfZeroes;$i++) {

		print OUTPUTFILE chr(0);

	}

}

sub bytes2dec { #calculates decimal value from inputted 4 byte little endian value 

	my $bytes=shift;

	my @byteArray;

	for(my $i=0;$i<length $bytes;$i++) {

		$byteArray[$i]=ord(substr($bytes,$i,1));

	}

	my $decValue;

	for(my $i=0;$i<scalar @byteArray;$i++) {

		$decValue+=($byteArray[$i]*(2**($i*8)));

	}

	return $decValue;

}

sub print_usage

	{

#........1.........2.........3.........4.........5.........6.........7.....

	print <<USAGE_EOF;

Usage:

  spitool.pl [options] files directories   

Create an SPI file by concatenating the files and contents of directories,

based on the options passed. 

The available options are:

-tSPIFileName       -- SPIFileName is the name the produced SPI file will 

                       have (i.e. ecom-0-0.spi). If not specified, the name 

                       will be ecom-0-0.spi by default

-dTargetDir         -- TargetDir is the directory where the SPI file should

                       be created, ending with a \

-iExisting          -- Existing is address and name of existing SPI file to

                       concatenate specified files to 

-uid<x>=<y>         -- <x> has value 1, 2 or 3, <y> is an UID value in

                       decimal or 0x... hexadecimal

-existinguid<x>=<y> -- <x> has value 1, 2 or 3, <y> is an UID value in 

                       decimal or 0x... hexadecimal

-existinguidcrc=<x> -- <x> is an UID value in decimal or 0x.. hexadecimal

-hide<ResourceFileNames> -- <ResourceFileNames> is the list of the resource files

			    that are to be hidden in the SPI file separated by

			    space or comma.

If an existing SPI file is specified with the -i option then this file is

used as a base and other data files are added to the end of this file,

otherwise a new SPI file is created. In either case the produced SPI file

is placed in the directory specified by the -d option and given the name 

specified with the -t option.

Files which are to be concatenated into the SPI file should be specified

on the command line by either specifying the file's name (and location), or

by including a directory name, in which case all files from that directory

will be included.

The -uid options can be used to filter files for inclusion in the SPI file.

This option can be included multiple times, so a list of UID1 values can be

built up, and the same for UID2 and UID3 values. Each file on the command

line is compared with this list, and if any of its UID values match a

relevant value in the UID lists then that file will be included in the SPI

file. If the file does not match any values it will be excluded.

The -existinguid options allow the UID values that an existing SPI file

should have to be specified. This will allow the possibility of checking

that the correct type of files are being concatenated together.

The -hide option can be used to mark a resource file as hidden in the SPI file.

To mark a resource file as a hidden entry in the SPI file, the resource data 

length will be written as 0.

USAGE_EOF

	}

sub createSpi 

	{

	my @resourceFiles=();

	my @hideresourceFiles=();

	my $spiFileName;

	my $targetDirectory;

	my $existingSpiFileName;

	my @uid;

	my @uidLengths = (0, 0, 0, 0);

	my @uid1;

	my @uid2;

	my @uid3;

	my @existingUid = (-1,-1,-1,-1);

	my $uidNumber;

	my $defaultSpiFileName = "ecom-0-0.spi";

	my $defaultTargetDirectory = &get_epocroot."epoc32\/tools\/";

	my @defaultUid = (-1,-1,-1,-1);

##########################################################################################

# Reading arguments phase

##########################################################################################

	foreach my $arg (@_) {

		if ($arg =~ /^-t(.*)/i) { # set target SPI file

			$spiFileName = $1;

			next;

			}

		if ($arg =~ /^-d(.*)/i) { # set target ouput directory

			my $tempDirectory=$1;

			if((-d $tempDirectory) ) {

				$targetDirectory = $tempDirectory;

				next;

				}

				else

				{

				 print "Output directory \'",$tempDirectory,"\' does not exist.\n";

				 exit(1);

				 }				

			}

		if ($arg =~ /^-i(.*)/i) { # existing SPI file to use as a base

			my $tempFileName = $1;

			if((-e $tempFileName) && (!(-d $tempFileName))) {

				$existingSpiFileName = $tempFileName;

				next;

				}

			}

		if ($arg =~ /^-uid([1-3])\=(.*)/i) {

			$uid[$1-1][$uidLengths[$1-1]++] = $2;

			next;

			}

		if($arg=~/^-existinguidcrc\=(.*)/i) {

			$existingUid[3]=$1;

			next;

			}

		if ($arg =~ /^-existinguid([1-3])\=(.*)/i) {

			$existingUid[$1-1]=$2;

			next;

			}

		if ($arg =~ /^-hide(.*)/i) { # Collect the files to be hidden

			my $line = $1;

			$line =~ s/,/ /g;

			my @files = split(' ' , $line);

			foreach my $file (@files)

			{

				push @hideresourceFiles, $file;

			}

			next;

			}

		if (-d $arg) {

			if(($arg =~ m-^.:-) && ($arg =~ m-[\\\/]$-)) {

				unless(opendir(DIRECTORY, $arg)) { print "Exiting: $arg"; exit; }

				while (my $file=readdir(DIRECTORY)) {

					my $newfile = $arg.$file;

					if(!(-d $newfile)) {

						push @resourceFiles, $newfile;

					}

				}

				close(DIRECTORY);

				next;

				}

			}

		if ((-e $arg) && (!(-d $arg))) {

			push @resourceFiles, $arg;

			next;

			}

		if ($arg eq "-h") {

			print_usage;

			exit(1);

		}	

		print "Unknown command: $arg\n";

	}

#####################################################################################

# UID phase

#####################################################################################

	if(!(defined $spiFileName)) { #use default file name if none passed on command line

		$spiFileName = $defaultSpiFileName;

	}

	if(!(defined $targetDirectory)) { #use default target dir if none passed on command line

		$targetDirectory = $defaultTargetDirectory;

	}

	for(my $i=0;$i<3;$i++) { #if default UIDs specified then added to UID match lists

		if($defaultUid[$i]>=0) {

			$uid[$i][$uidLengths[$i]++] = $defaultUid[$i];

		}

	}

	for(my $i=0;$i<3;$i++) { #makes sure UIDs are valid UIDs

		my @tempUidArray;

		my $iterator=0;

		while(defined $uid[$i][$iterator]) {

			my $convertedUid=convertUidFromText($uid[$i][$iterator]);

			if ($convertedUid != -1) {

				push @tempUidArray, binarize($convertedUid);

			} else {

				print "Invalid UID: $uid[$i][$iterator]\n";

			}

			$iterator++;

		}

		for(my $j=0;$i<scalar @tempUidArray;$j++) {

			$uid[$i][$j]=$tempUidArray[$j];

		}

		for(my $j=scalar@tempUidArray;defined $uid[$i][$j];$j++) {

			undef $uid[$i][$j];

		}

	}

#####################################################################################

# Phase to split up resource names

#####################################################################################

	my @resourceFilePaths;

	my @resourceFileNames;

	my @resourceExtensions;

	my @filestobehidden;

# To mark the resource files as hidden in the SPI file by writing the data length as zero

	foreach my $file (@hideresourceFiles)

	{

		my $matchfound =0;

		my $i=0;

		for(;$i<scalar @resourceFiles && !$matchfound;$i++)

		{

			if (lc($file) eq lc($resourceFiles[$i]))

			{

				$filestobehidden[$i] = 1;

				$matchfound =1;

			}

		}

		if (!$matchfound)

		{

# Those files that are to be hidden in the SPI file but not existing in the SPI

			if (! -e $file)

			{

				print "Warning: Hiding non-existent file $file\n";

			}

			push @resourceFiles,$file;

			$filestobehidden[$i] = 1;

		}

	}

	for(my $i=0;$i<scalar @resourceFiles;$i++) {

		if($resourceFiles[$i]=~m|\/|) {

			if($resourceFiles[$i]=~m|(.*)\/([^\/]*)$|) {

				$resourceFilePaths[$i]=$1;

				$resourceFileNames[$i]=$2;

			}

			if($resourceFileNames[$i]=~m|(.*)\.([^\.]*)|) {

				$resourceFileNames[$i]= $1;

				$resourceExtensions[$i]=$2;

			}

		} else {

			$resourceFilePaths[$i]="";

			if($resourceFiles[$i]=~m|(.*)\.([^\.]*)|) {

				$resourceFileNames[$i]= $1;

				$resourceExtensions[$i]=$2;

			}

		}

	}

	my %uid2values; #hash to hold UID2 values for each type of SPI file

	$uid2values{"ecom"} = 270556204;

##########################################################

# Existing file stage

##########################################################

	my @spiUid = (270556203, 0, 0); #holds spi values (including CRC value)

	foreach my $key (keys(%uid2values)) { #searches through SPI types to match UID2 value

		if($spiFileName =~/^$key/) {

			$spiUid[1]=$uid2values{$key};

		}

	}

	$spiUid[3] = uidcrc($spiUid[0], $spiUid[1], $spiUid[2]);

	my $total=0; #used to keep track of position in SPI file

	my $buffer;

	my $spifile=File::Spec->catpath( "", $targetDirectory, $spiFileName );

 	open OUTPUTFILE, ">$spifile" or die $!;

	binmode (OUTPUTFILE);

	if($existingSpiFileName) {

		open EXISTINGFILE, "$existingSpiFileName" or die $!;

		binmode (EXISTINGFILE);

		my @fileNameLengths;

		my @fileLengths;

		my @fileNames;

		my @fileContents;

		read(EXISTINGFILE,$buffer,4);

		read(EXISTINGFILE,$buffer,4);

		if(bytes2dec($buffer)!=$spiUid[1]) {

			print "Incompatible SPI files.\n";

		}

		read(EXISTINGFILE,$buffer,24);

		$total=32;

		my $existingSpiFileSize = (stat(EXISTINGFILE))[7];

		while($total<$existingSpiFileSize) { #loop to store information from files which are not being replaced

			read(EXISTINGFILE,$buffer,4);

			push @fileNameLengths, bytes2dec($buffer);

			read(EXISTINGFILE,$buffer,4);

			push @fileLengths, bytes2dec($buffer);

			read(EXISTINGFILE,$buffer,$fileNameLengths[$#fileNameLengths]);

			push @fileNames, $buffer;

			$total=$total+8+$fileNameLengths[$#fileNameLengths]+$fileLengths[$#fileLengths];

			my $padding = (4-(($fileNameLengths[$#fileNameLengths]+$fileLengths[$#fileLengths])%4))%4;

			read(EXISTINGFILE,$buffer,$fileLengths[$#fileLengths]+$padding);

			push @fileContents, $buffer;

			$total += (4-($total%4))%4;			

		}

		close EXISTINGFILE;	

		#next part prints to OUTPUTFILE the header and files which are not being replaced

		print OUTPUTFILE binarize($spiUid[0]) . binarize($spiUid[1]) . binarize($spiUid[2]) . binarize($spiUid[3]);

		printZeroes(16);

		$total=32;

		for(my $i=0; $i<scalar @fileNames; $i++) {

			my $flag=1;

			for(my $j=0; $j<scalar @resourceFileNames && $flag==1; $j++) {

				if($fileNames[$i] eq $resourceFileNames[$j]) {

					$flag=0;

				}

			}

			if($flag) {

				print OUTPUTFILE binarize($fileNameLengths[$i]) . binarize($fileLengths[$i]) . $fileNames[$i] . $fileContents[$i];

				$total=$total+8+length($fileNames[$i])+length($fileContents[$i]);

			}

		}

	} else { #prints header for target SPI file if there is no existing SPI file

		print OUTPUTFILE binarize($spiUid[0]) . binarize($spiUid[1]) . binarize($spiUid[2]) . binarize($spiUid[3]);

		printZeroes(16);

		$total=32;

	}

####################################################################

# Appending new data files to the SPI file

####################################################################

	my $resourceFileSize;

	my $resourceFileSizeInBinary;

	my $resourceFileNameSize;

	my $resourceFileNameSizeInBinary;

	for(my $i=0; $i<scalar @resourceExtensions;$i++) {

# To mark the resource files as hidden in the SPI file by writing the data length as zero

	   if ($filestobehidden[$i] == 1)

	   {

		$resourceFileNameSize = length($resourceFileNames[$i]);

		$resourceFileNameSizeInBinary = binarize($resourceFileNameSize);

		$resourceFileSize = 0;

		$resourceFileSizeInBinary = binarize($resourceFileSize);

		print OUTPUTFILE $resourceFileNameSizeInBinary . $resourceFileSizeInBinary . $resourceFileNames[$i];

		$total+=$resourceFileNameSize;

		my $padding = (4-(($resourceFileSize + $resourceFileNameSize)%4))%4;

		printZeroes($padding);

		$total+=$padding;

	   }

	   else

	   {

		open RESOURCEFILE, "<$resourceFiles[$i]";

		binmode(RESOURCEFILE);

		my @fileUid; #stores UIDs from particular data file

		my $fileUid1;

		my $fileUid2;

		my $fileUid3;

		read(RESOURCEFILE,$fileUid[0],4);

		read(RESOURCEFILE,$fileUid[1],4);

		read(RESOURCEFILE,$fileUid[2],4);

		my $uidFlag=0; #changes to 1 if a UID value in data file matches a specified UID value

		my $uidExists=1; #changes to 1 if there are specified UIDs to match to

		for(my $j=0;$j<3 && (!$uidFlag);$j++) {

			my $k=0;

			while(defined $uid[$j][$k] && (!$uidFlag)) {

				$uidExists=0;

				if($uid[$j][$k] eq bin2hex($fileUid[$j])) {

					$uidFlag=1;

				}

				$k++;

			}

		}	

		if(($uidFlag) || ($uidExists)) { #if suitable UIDs writes data file to SPI file

			$resourceFileSize = (stat(RESOURCEFILE))[7];

			$resourceFileNameSize = length($resourceFileNames[$i]);

			$resourceFileNameSizeInBinary = binarize($resourceFileNameSize);

			$resourceFileSizeInBinary = binarize($resourceFileSize);

			print OUTPUTFILE $resourceFileNameSizeInBinary . $resourceFileSizeInBinary . $resourceFileNames[$i];

			print OUTPUTFILE "$fileUid[0]$fileUid[1]$fileUid[2]";

			while(read(RESOURCEFILE,$buffer,1)) {

				print OUTPUTFILE $buffer;

			}

			$total+=$resourceFileSize;

			$total+=8;

			$total+=$resourceFileNameSize;

			my $padding = (4-(($resourceFileSize + $resourceFileNameSize)%4))%4;

			printZeroes($padding);

			$total+=$padding;

		}

	   }

	}

	print "Created $spiFileName\n";

}

1;