// 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 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:

 // e32test\dll\t_xxver2w.cpp

 // Overview:

 // Check of DLL & EXE module info and DDL function ordinals

 // API Information:

 // RLibrary

 // Details:

 // - Use RLibrary::GetInfoFromHeader() and RLibrary::GetInfo() to get DLL 

 // information. Verify results are as expected.

 // - Use RLibrary::GetInfoFromHeader() and RLibrary::GetInfo() to get EXE 

 // information. Verify results are as expected.

 // - Load a DLL, lookup and verify the function at the specified ordinal 

 // within the DLL.

 // Platforms/Drives/Compatibility:

 // All.

 // Assumptions/Requirement/Pre-requisites:

 // Failures and causes:

 // Base Port information:

 // 

 //

 #include <e32uid.h>

 #include <e32test.h>

 #include <f32file.h>

 #include <e32wins.h>

 #include <emulator.h>

 #include <d_ldrtst.h>

 RTest test(_L("T_XXVER2W"));

 RLdrTest LdrTest;

 TFileName CopyBinaryFromZ(const TFileName& aFileName)

 	{

 	TParse parse;

 	test(parse.Set(aFileName,0,0)==KErrNone);

 	TFileName source = _L("z:");

 	source.Append(parse.NameAndExt());

 	TBuf<MAX_PATH> sName;

 	TInt r = MapEmulatedFileName(sName, parse.NameAndExt());

 	test(r==KErrNone);

 	TFileName destination = _S16("C:\\t_zzver2w.tmp");

 	TBuf<MAX_PATH> dName;

 	r = MapEmulatedFileName(dName, destination);

 	test(r==KErrNone);

 	r=Emulator::CopyFile((LPCTSTR)sName.PtrZ(),(LPCTSTR)dName.PtrZ(),FALSE);

 	test(r);

 	return destination;

 	}

 void TestDllInfo()

 	{

 	TFileName fn;

 	if(PlatSec::ConfigSetting(PlatSec::EPlatSecEnforceSysBin))

 		fn = _S16("Z:\\sys\\bin\\t_ver2.dll");

 	else

 		fn = _S16("Z:\\system\\bin\\t_ver2.dll");

 	test.Printf(_L("Getting info for %S\n"), &fn);

 	TBool formHeader=EFalse;

 	for(;;)

 		{

 		RLibrary::TInfo info;

 		TPckg<RLibrary::TInfo> infoBuf(info);

 		TInt r;

 		if(formHeader)

 			{

 			TUint8* buf;

 			RFs fs;

 			test(fs.Connect()==KErrNone);

 			RFile file;

 			test((r=file.Open(fs,CopyBinaryFromZ(fn),0))==KErrNone);

 			TInt size;

 			test((r=file.Size(size))==KErrNone);

 			buf=new TUint8[size];

 			test(buf!=0);

 			TPtr8 header(buf,size);

 			test((r=file.Read(header))==KErrNone);

 			file.Close();

 			fs.Close();

 			r = RLibrary::GetInfoFromHeader(header, infoBuf);

 			test.Printf(_L("GetInfoFromHeader returns %d\n"), r);

 			delete buf;

 			}

 		else

 			{

 			r = RLibrary::GetInfo(fn, infoBuf);

 			test.Printf(_L("GetInfo returns %d\n"), r);

 			}

 		test(r==KErrNone);

 		const TUint32* uid = (const TUint32*)&info.iUids;

 		test.Printf(_L("VER  %08x\n"), info.iModuleVersion);

 		test.Printf(_L("UID1 %08x\n"), uid[0]);

 		test.Printf(_L("UID2 %08x\n"), uid[1]);

 		test.Printf(_L("UID3 %08x\n"), uid[2]);

 		test.Printf(_L("SID  %08x\n"), info.iSecurityInfo.iSecureId);

 		test.Printf(_L("VID  %08x\n"), info.iSecurityInfo.iVendorId);

 		test.Printf(_L("CAP0 %08x\n"), ((SSecurityInfo&)info.iSecurityInfo).iCaps[0]);

 		test.Printf(_L("CAP1 %08x\n"), ((SSecurityInfo&)info.iSecurityInfo).iCaps[1]);

 		TUint32 v = 0x00030000u;

 		test(info.iModuleVersion == v);

 		test(uid[0] == (TUint32)KDynamicLibraryUidValue);

 		test(uid[2] == (TUint32)0x40abcdef);

 		TUint32 xsid = ((v>>16)<<4)|(v&0x0f)|0x89abcd00u;

 		test(info.iSecurityInfo.iSecureId == xsid);

 		test(info.iSecurityInfo.iVendorId == 0x01234500+(xsid&0xff));

 		test(((SSecurityInfo&)info.iSecurityInfo).iCaps[0]==0x0002aaab);

 		test(((SSecurityInfo&)info.iSecurityInfo).iCaps[1]==0);

 		if(formHeader)

 			break;

 		formHeader = ETrue;

 		}

 	}

 void TestExeInfo()

 	{

 	TFileName fn;

 	if(PlatSec::ConfigSetting(PlatSec::EPlatSecEnforceSysBin))

 		fn = _S16("Z:\\sys\\bin\\t_xxver2w.exe");

 	else

 		fn = _S16("Z:\\system\\bin\\t_xxver2w.exe");

 	test.Printf(_L("Getting info for %S\n"), &fn);

 	TBool formHeader=EFalse;

 	for(;;)

 		{

 		RLibrary::TInfo info;

 		TPckg<RLibrary::TInfo> infoBuf(info);

 		TInt r;

 		if(formHeader)

 			{

 			TUint8* buf;

 			RFs fs;

 			test(fs.Connect()==KErrNone);

 			RFile file;

 			test((r=file.Open(fs,CopyBinaryFromZ(fn),0))==KErrNone);

 			TInt size;

 			test((r=file.Size(size))==KErrNone);

 			buf=new TUint8[size];

 			test(buf!=0);

 			TPtr8 header(buf,size);

 			test((r=file.Read(header))==KErrNone);

 			file.Close();

 			fs.Close();

 			r = RLibrary::GetInfoFromHeader(header, infoBuf);

 			test.Printf(_L("GetInfoFromHeader returns %d\n"), r);

 			delete buf;

 			}

 		else

 			{

 			r = RLibrary::GetInfo(fn, infoBuf);

 			test.Printf(_L("GetInfo returns %d\n"), r);

 			}

 		test(r==KErrNone);

 		const TUint32* uid = (const TUint32*)&info.iUids;

 		test.Printf(_L("VER  %08x\n"), info.iModuleVersion);

 		test.Printf(_L("UID1 %08x\n"), uid[0]);

 		test.Printf(_L("UID2 %08x\n"), uid[1]);

 		test.Printf(_L("UID3 %08x\n"), uid[2]);

 		test.Printf(_L("SID  %08x\n"), info.iSecurityInfo.iSecureId);

 		test.Printf(_L("VID  %08x\n"), info.iSecurityInfo.iVendorId);

 		test.Printf(_L("CAP0 %08x\n"), ((SSecurityInfo&)info.iSecurityInfo).iCaps[0]);

 		test.Printf(_L("CAP1 %08x\n"), ((SSecurityInfo&)info.iSecurityInfo).iCaps[1]);

 		test(info.iModuleVersion == 0x00010000);

 		test(uid[0] == (TUint32)KExecutableImageUidValue);

 		TUint32 xuid3 = 0x40abcd77u;

 		test(uid[2] == xuid3);

 		test(info.iSecurityInfo.iSecureId == xuid3);

 		test(info.iSecurityInfo.iVendorId == 0x01234577);

 		test(((SSecurityInfo&)info.iSecurityInfo).iCaps[0]==0x0002aaab);

 		test(((SSecurityInfo&)info.iSecurityInfo).iCaps[1]==0);

 		if(formHeader)

 			break;

 		formHeader = ETrue;

 		}

 	}

 struct SExportInfo

 	{

 	TInt	iTotal;

 	TInt	iHoles;

 	TInt	iHole[1];

 	};

 const TInt DllExportInfo[] = {59,6,2,3,4,23,24,39};

 void CheckExports(RLibrary aLib)

 	{

 	const TFileName& fn = aLib.FileName();

 	test.Printf(_L("Filename %S\n"), &fn);

 	const SExportInfo* e = (const SExportInfo*)DllExportInfo;

 	TAny* libcs = LdrTest.LibraryCodeSeg(aLib.Handle());

 	test.Printf(_L("Code seg @%08x\n"), libcs);

 	test(libcs != NULL);

 	TInt n = e->iTotal;

 	TInt nh = e->iHoles;

 	TInt ord;

 	for (ord=1; ord<=n+1; ++ord)

 		{

 		TLibraryFunction f = aLib.Lookup(ord);

 		test.Printf(_L("Ord %3d->%08x\n"), ord, f);

 		if (ord>n)

 			{

 			test(!f);

 			continue;

 			}

 		TInt i;

 		for (i=0; i<nh && e->iHole[i]!=ord; ++i) {}

 		if (i<nh)

 			test(!f);	// hole

 		else

 			test(f!=NULL);

 		TAny* cs = LdrTest.CodeSegFromAddr((TLinAddr)f);

 		test(f ? (cs==libcs) : !cs);

 		}

 	}

 void TestMissingOrdinals()

 	{

 	RLibrary l;

 	TFileName fn;

 	if(PlatSec::ConfigSetting(PlatSec::EPlatSecEnforceSysBin))

 		fn = _S("Z:\\sys\\bin\\t_ver2.dll");

 	else

 		fn = _S("Z:\\system\\bin\\t_ver2.dll");

 	TInt r = l.Load(fn);

 	test.Printf(_L("Load %S returns %d\n"), &fn, r);

 	CheckExports(l);

 	l.Close();

 	}

 TInt E32Main()

 	{

 	test.Title();

 	test.Start(_L("Connect to test driver"));

 	TInt r = User::LoadLogicalDevice(_L("d_ldrtst"));

 	test(r==KErrNone || r==KErrAlreadyExists);

 	r = LdrTest.Open();

 	test(r==KErrNone);

 	test.Next(_L("Test DLL Info"));

 	TestDllInfo();

 	test.Next(_L("Test EXE Info"));

 	TestExeInfo();

 	test.Next(_L("Test Missing Ordinals"));

 	TestMissingOrdinals();

 	LdrTest.Close();

 	test.End();

 	return KErrNone;

 	}