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

	}