// Copyright (c) 1998-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:

 // e32utils\setcap\setcap.cpp

 // 

 //

 #include "setcap.h"

 #include <e32std.h>

 #include <e32std_private.h>

 GLDEF_D TBool CapabilitySet;

 GLDEF_D SCapabilitySet Capability;

 GLDEF_D TBool SecureIdSet;

 GLDEF_D TSecureId SecureId;

 GLDEF_D TBool VendorIdSet;

 GLDEF_D TVendorId VendorId;

 #ifdef __TOOLS__

 #define		Mem		HMem

 #include "h_utl.h"

 #endif //__TOOLS__

 #ifndef __EPOC32__

 GLDEF_C TInt SetCap(HANDLE hFile)

 	{

 	DWORD ret;

 	TText8 checkedUidBuf[sizeof(TCheckedUid)];

 	ReadFile(hFile,checkedUidBuf,sizeof(TCheckedUid),&ret,NULL);

 	if (ret!=sizeof(TCheckedUid))

 		goto close;

 	//Look at PE file for UID section

 	{

 	const TInt KPeHeaderAddrAddr=0x3c;

 	const TInt KPeHeaderAddrSize=0x01;

 	const TInt KNumberOfSectionsOffset=0x06;

 	const TInt KNumberOfSectionsSize=0x02;

 	const TInt KSectionTableOffset=0xf8;

 	const TInt KSectionHeaderSize=0x28;

 	const TInt KSectionNameLength=0x08;

 	const TInt KPtrToRawDataOffset=0x14;

 	const TInt KPtrToRawDataSize=0x04;

 	const TText8 peText[4]={'P','E',0,0};

 	const TText8 uidText[8]={'.','S','Y','M','B','I','A','N'};

 	//Read address of start of PE header

 	if (SetFilePointer(hFile,KPeHeaderAddrAddr,0,FILE_BEGIN)==0xFFFFFFFF)

 		goto close;

 	TInt peAddr=0;

 	ReadFile(hFile,&peAddr,KPeHeaderAddrSize,&ret,NULL);

 	if (ret!=KPeHeaderAddrSize)

 		goto close;

 	//Check it really is the start of PE header

 	if (SetFilePointer(hFile,peAddr,0,FILE_BEGIN)==0xFFFFFFFF)

 		goto close;

 	TText8 text[4];

 	ReadFile(hFile,text,4,&ret,NULL);

 	if (*(TInt32*)text!=*(TInt32*)peText)

 		goto close;

 	//Read number of sections

 	if (SetFilePointer(hFile,peAddr+KNumberOfSectionsOffset,0,FILE_BEGIN)==0xFFFFFFFF)

 		goto close;

 	TInt sections=0;

 	ReadFile(hFile,&sections,KNumberOfSectionsSize,&ret,NULL);

 	if (ret!=KNumberOfSectionsSize)

 		goto close;

 	//Go through section headers looking for UID section

 	if (SetFilePointer(hFile,peAddr+KSectionTableOffset,0,FILE_BEGIN)==0xFFFFFFFF)

 		goto close;

 	TInt i=0;

 	for(;i<sections;i++)

 		{

 		TText8 name[KSectionNameLength];

 		ReadFile(hFile,name,KSectionNameLength,&ret,NULL);

 		if (ret!=KSectionNameLength)

 			goto close;

 		if (*(TInt64*)name==*(TInt64*)uidText)

 			break;

 		if (SetFilePointer(hFile,KSectionHeaderSize-KSectionNameLength,0,FILE_CURRENT)==0xFFFFFFFF)

 			goto close;

 		}

 	if (i==sections)

 		goto close;

 	//Read RVA/Offset

 	if (SetFilePointer(hFile,KPtrToRawDataOffset-KSectionNameLength,0,FILE_CURRENT)==0xFFFFFFFF)

 		goto close;

 	TInt uidOffset;

 	ReadFile(hFile,&uidOffset,KPtrToRawDataSize,&ret,NULL);

 	if (ret!=KPtrToRawDataSize)

 		goto close;

 	//SYMBIAN Header

 	if (SetFilePointer(hFile,uidOffset,0,FILE_BEGIN)==0xFFFFFFFF)

 		goto close;

 	TEmulatorImageHeader header;

 	ReadFile(hFile,&header,sizeof(header),&ret,NULL);

 	if (ret!=sizeof(header))

 		goto close;

 	// set new capabilities

 	if (CapabilitySet)

 		{

 		header.iS.iCaps = Capability;

 		}

 	// set new secure id and vendor id if specified

 	if (SecureIdSet)

 		{

 		header.iS.iSecureId = SecureId.iId;

 		}

 	if (VendorIdSet)

 		{

 		header.iS.iVendorId = VendorId.iId;

 		}

 	// save header values

 	Capability = header.iS.iCaps;

 	SecureId.iId = header.iS.iSecureId;

 	VendorId.iId = header.iS.iVendorId;

 	if (SetFilePointer(hFile,uidOffset,0,FILE_BEGIN)==0xFFFFFFFF)

 		goto close;

 	BOOL b = WriteFile(hFile,&header,sizeof(header),&ret,NULL);

 	if (b==FALSE)

 		goto close;

 	CloseHandle(hFile);

 	return KErrNone;

 	}

 close:

 	CloseHandle(hFile);

 	return KErrGeneral;

 	}

 #endif //__EPOC32__

 #ifndef __WINS__

 GLDEF_C TInt SetCap(E32ImageHeader* h)

 	{

 	TUint hdrfmt = h->HeaderFormat();

 	if (hdrfmt < KImageHdrFmt_V)

 		return KErrNotSupported;

 	E32ImageHeaderV* v = (E32ImageHeaderV*)h;

 	if (CapabilitySet)

 		v->iS.iCaps = Capability;

 	if (SecureIdSet)

 		v->iS.iSecureId = SecureId.iId;

 	if (VendorIdSet)

 		v->iS.iVendorId = VendorId.iId;	

 	// save header values

 	Capability = v->iS.iCaps;

 	SecureId.iId = v->iS.iSecureId;

 	VendorId.iId = v->iS.iVendorId;

 	// regenerate CRC

 	v->iHeaderCrc = KImageCrcInitialiser;

 	TUint32 crc = 0;

 	Mem::Crc32(crc, v, v->TotalSize());

 	v->iHeaderCrc = crc;

 	return KErrNone;

 	}

 #endif //__WINS__