kerneltest/e32test/misc/t_ymodemz.cpp
changeset 0 a41df078684a
child 109 b3a1d9898418
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/kerneltest/e32test/misc/t_ymodemz.cpp	Mon Oct 19 15:55:17 2009 +0100
@@ -0,0 +1,291 @@
+// 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:
+// e32test\misc\t_ymodemz.cpp
+// 
+//
+
+#include <e32test.h>
+#include "ymodemu.h"
+#include <f32file.h>
+#include "unzip.h"
+
+RTest test(_L("YModemZ"));
+
+#define TEST(c)		((void)((c)||(test.Printf(_L("Failed at line %d\n"),__LINE__),test.Getch(),test(0),0)))
+#define CHECK(c)	((void)(((c)==0)||(test.Printf(_L("Error %d at line %d\n"),(c),__LINE__),test.Getch(),test(0),0)))
+
+const TInt KBufferSize=4096;
+
+_LIT(KLddName,"ECOMM");
+_LIT(KPddName,"EUART");
+
+RFs TheFs;
+RFile TheOutputFile;
+YModemU* TheYModem;
+
+void LoadCommDrivers()
+	{
+	test.Printf(_L("Load LDD\n"));
+	TInt r=User::LoadLogicalDevice(KLddName);
+	TEST(r==KErrNone || r==KErrAlreadyExists);
+
+	TInt i;
+	TInt n=0;
+	for (i=-1; i<10; ++i)
+		{
+		TBuf<16> pddName=KPddName();
+		if (i>=0)
+			pddName.Append('0'+i);
+		TInt r=User::LoadPhysicalDevice(pddName);
+		if (r==KErrNone || r==KErrAlreadyExists)
+			{
+			++n;
+			test.Printf(_L("%S found\n"),&pddName);
+			}
+		}
+	TEST(n!=0);
+	}
+
+GLDEF_C void AcceptUnzippedBlock(TZipInfo& aInfo, TUint8*& aOutPtr, TInt aError)
+	{
+	if (aError==KErrNone)
+		{
+		TInt avail=aOutPtr-aInfo.iOutBuf;
+		if (avail>=KZipWindowSize+0x1000)
+			{
+			TInt len=avail-KZipWindowSize;
+			TPtrC8 ptr(aInfo.iOutBuf,len);
+			TInt r=TheOutputFile.Write(ptr);
+			CHECK(r);
+			Mem::Copy(aInfo.iOutBuf,aInfo.iOutBuf+len,KZipWindowSize);
+			aOutPtr=aInfo.iOutBuf+KZipWindowSize;
+			}
+		}
+	}
+
+GLDEF_C TInt ReadInputData(TUint8* aDest, TInt& aLength)
+	{
+	TUint8* pD=aDest;
+//	test.Printf(_L("@%dms\n"),User::NTickCount());
+	TInt r=TheYModem->ReadPackets(pD,aLength);
+//	test.Printf(_L("ReadIP %d\n"),r);
+	aLength=pD-aDest;
+	return r;
+	}
+
+GLDEF_C TInt UnzipComplete(TZipInfo& a, TUint8* aOutPtr, TInt aError)
+	{
+	TInt r=aError;
+	if (r==KErrNone && aOutPtr>a.iOutBuf)
+		r=TheOutputFile.Write(TPtrC8(a.iOutBuf,aOutPtr-a.iOutBuf));
+	CHECK(r);
+	return r;
+	}
+
+_LIT(KLitThreadName,"Unzip");
+TInt Initialise(TZipInfo& a)
+	{
+	TInt r=InitInfo(a);
+	if (r!=KErrNone)
+		return r;
+	a.iFileBufSize=4*a.iInBufSize;
+	TAny* pFileBuf=MALLOC(a.iFileBufSize);
+	if (!pFileBuf)
+		return KErrNoMemory;
+	a.iFileBuf=(TUint8*)pFileBuf;
+	RThread t;
+	r=t.Create(KLitThreadName,UnzipThread,0x2000,NULL,&a);
+	if (r!=KErrNone)
+		{
+		FREE(pFileBuf);
+		a.iFileBuf=NULL;
+		return r;
+		}
+	t.SetPriority(EPriorityLess);
+	t.Logon(a.iThreadStatus);
+	t.Resume();
+	a.iThreadHandle=t.Handle();
+	return KErrNone;
+	}
+
+void ProcessHeader(TZipInfo& a)
+	{
+	test.Printf(_L("Flags=%d\n"),a.iFlags);
+	test.Printf(_L("Method=%d\n"),a.iMethod);
+	test.Printf(_L("Crc=%d\n"),a.iCrc);
+	test.Printf(_L("Compressed size=%d\n"),a.iCompressedSize);
+	test.Printf(_L("Uncompressed size=%d\n"),a.iUncompressedSize);
+	test.Printf(_L("File name %S\n"),&a.iName);
+	test.Printf(_L("Data offset %d\n\n"),a.iDataOffset);
+
+	TInt r=TheOutputFile.Replace(TheFs,a.iName,EFileWrite);
+	CHECK(r);
+	test.Printf(_L("Allocate memory for unzipped file\n"));
+	a.iOutBuf=(TUint8*)User::Alloc(262144);
+	TEST(a.iOutBuf!=NULL);
+	test.Printf(_L("Begin unzipping\n"));
+	a.iHeaderDone=2;
+	TRequestStatus* pS=&a.iProcessedHeader;
+	RThread t;
+	t.SetHandle(a.iThreadHandle);
+	t.RequestComplete(pS,0);
+	}
+
+void Cleanup(TZipInfo& a)
+	{
+	delete a.iFileBuf;
+	a.iFileBuf=NULL;
+	delete a.iOutBuf;
+	a.iOutBuf=NULL;
+	RThread& t=*(RThread*)&a.iThreadHandle;
+	t.Close();
+	}
+
+GLDEF_C TInt E32Main()
+	{
+//	RThread().SetSystem(ETrue);
+	RThread().SetPriority(EPriorityAbsoluteForeground);
+	test.SetLogged(EFalse);
+	test.Title();
+
+	TBuf<256> cmd;
+	User::CommandLine(cmd);
+	TInt port=0;
+	if (cmd.Length()!=0)
+		{
+		TUint8 c=(TUint8)cmd[0];
+		if (c>='0' && c<='9')
+			{
+			port=c-'0';
+			}
+		}
+
+	TInt r=KErrNone;
+	LoadCommDrivers();
+
+	test.Printf(_L("Connect to file server\n"));
+	r=TheFs.Connect();
+	CHECK(r);
+	r=TheFs.ShareAuto();
+	CHECK(r);
+
+	test.Printf(_L("Create YModem object\n"));
+	YModemU* pY=NULL;
+	TRAP(r,pY=YModemU::NewL(port,ETrue));
+	TEST(r==KErrNone && pY!=NULL);
+	TheYModem=pY;
+
+	test.Printf(_L("Create buffer\n"));
+	TUint8* buffer=(TUint8*)User::Alloc(KBufferSize);
+	TEST(buffer!=NULL);
+
+	test.Printf(_L("Receive...\n"));
+
+	TBool mode=1;
+	FOREVER
+		{
+		TInt total_size=0;
+		TInt size=-1;
+		TBuf<256> name;
+		r=pY->StartDownload(mode, size, name);
+//		test.Printf(_L("@%dms"),User::NTickCount());
+		if (r!=KErrNone)
+			break;
+		test.Printf(_L("r=%d, size=%d, name %S\n"),r,size,&name);
+		if (r==KErrNone && name.Right(4).CompareF(_L(".zip"))==0 && size!=0)
+			{
+			test.Printf(_L("Initialising unzip...\n"));
+			TZipInfo z;
+			z.iRemain=size;
+			r=Initialise(z);
+			CHECK(r);
+			test.Printf(_L("Read header\n"));
+			TUint32 c=0;
+			RThread t;
+			t.SetHandle(z.iThreadHandle);
+			while (z.iRemain && z.iThreadStatus==KRequestPending)
+				{
+				TRequestStatus dummy;
+				TRequestStatus* pS=&dummy;
+//				test.Printf(_L("remain=%d\n"),z.iRemain);
+				r=ReadBlockToBuffer(z);
+				CHECK(r);
+				t.RequestComplete(pS,0);		// same process
+				while(z.iHeaderDone==0 && z.iThreadStatus==KRequestPending)
+					DELAY(20000);
+				if (z.iHeaderDone==1 && z.iThreadStatus==KRequestPending)
+					{
+					// after reading first block, process the header
+					ProcessHeader(z);
+					c=User::NTickCount();
+					}
+				}
+			test.Printf(_L("\nWait for thread to exit\n"));
+			User::WaitForRequest(z.iThreadStatus);
+			TInt exitType=t.ExitType();
+			TInt exitReason=t.ExitReason();
+			if (z.iRemain || exitType!=EExitKill || exitReason!=KErrNone)
+				{
+				TBuf<32> exitCat=t.ExitCategory();
+				test.Printf(_L("Exit code %d,%d,%S\n"),exitType,exitReason,&exitCat); test.Getch(); test(0);
+				}
+			TUint8* pD=buffer;
+			r=pY->ReadPackets(pD,KBufferSize);	// should get EOF response
+			TEST(r==KErrEof);
+			Cleanup(z);
+			TheOutputFile.Close();
+			}
+		else if (r==KErrNone)
+			{
+			test.Printf(_L("Opening file for write\n"));
+			RFile file;
+			r=file.Replace(TheFs,name,EFileWrite);
+			if (r!=KErrNone)
+				{
+				test.Printf(_L("RFile::Replace returns %d\n"),r); test.Getch();	test(0);
+				}
+			while (r==KErrNone)
+				{
+				TUint8* pD=buffer;
+				r=pY->ReadPackets(pD,KBufferSize);
+				if (r==KErrNone || r==KErrEof)
+					{
+					TInt blen=pD-buffer;
+					if (size>0)				// size was transmitted
+						{
+						if (blen>size-total_size)
+							blen=size-total_size;
+						}
+					total_size+=blen;
+					TPtrC8 fptr(buffer,blen);
+					TInt s=file.Write(fptr);
+					if (s!=KErrNone)
+						{
+						test.Printf(_L("RFile::Write returns %d\n"),s); test.Getch(); test(0);
+						}
+					}
+				}
+			file.Close();
+			test.Printf(_L("rx size=%d\n"),total_size);
+			}
+		}
+	delete buffer;
+	delete pY;
+	TheFs.Close();
+	test.Printf(_L("r=%d\n"),r);
+	test.Getch();
+
+	return KErrNone;
+	}
+