kerneltest/e32test/misc/ymodemtx.cpp
changeset 90 947f0dc9f7a8
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/kerneltest/e32test/misc/ymodemtx.cpp	Fri Apr 16 16:24:37 2010 +0300
@@ -0,0 +1,472 @@
+// Copyright (c) 2009-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\ymodemtx.cpp
+// 
+//
+
+#include <e32std.h>
+#include <e32svr.h>
+#include <f32file.h>
+#include <d32comm.h>
+
+RFs TheFs;
+RFile TheFile;
+TInt FileSize;
+RBusDevComm TheComm;
+RTimer TheTimer;
+TFileName FileName;
+TFileName FileName8b;
+TPtr8 FileName8(0,0,0);
+
+const TUint8 STX=0x02;
+const TUint8 EOT=0x04;
+const TUint8 ACK=0x06;
+const TUint8 BIGC=0x43;
+const TUint8 BIGG=0x47;
+
+#define PACKET_SIZE		1024
+
+#define MIN(a,b)		((a)<(b)?(a):(b))
+#define OFFSET(p,off)	((void*)((char*)p+off))
+
+#define RESET_COMM()	TheComm.ResetBuffers()
+
+#define assert(x)			((void)( (x) || (__Panic(__LINE__, #x),0) ))
+#define assert_KErrNone(r)	((void)( ((r)==KErrNone) || (__Panic(__LINE__, (r)),0) ))
+
+
+void __Panic(TInt aLine, TInt aError)
+	{
+	RDebug::Printf("Line %d Expected KErrNone got %d", aLine, aError);
+	User::Panic(_L("YMODEMTX"), aLine);
+	}
+
+void __Panic(TInt aLine, const char* aMessage)
+	{
+	RDebug::Printf("Line %d Assertion \"%s\" failed", aLine, aMessage);
+	User::Panic(_L("YMODEMTX"), aLine);
+	}
+
+/*
+YModem packet structure:
+Byte 0 = STX
+Byte 1 = sequence number (first user data packet is 1)
+Byte 2 = complement of sequence number
+Bytes 3-1026 = data (1K per packet)
+Bytes 1027, 1028 = 16-bit CRC (big-endian)
+
+A herald packet is sent first, with sequence number 0
+The data field consists of the null-terminated file name
+followed by the null-terminated file size in ASCII decimal
+digits.
+*/
+struct SPacket
+	{
+	TUint8 iPTI;
+	TUint8 iSeq;
+	TUint8 iSeqBar;
+	TUint8 iData[PACKET_SIZE];
+	TUint8 iCRC1;
+	TUint8 iCRC0;
+	};
+
+static const TUint16 crcTab[256] =
+    {
+	0x0000,0x1021,0x2042,0x3063,0x4084,0x50a5,0x60c6,0x70e7,0x8108,0x9129,0xa14a,
+	0xb16b,0xc18c,0xd1ad,0xe1ce,0xf1ef,0x1231,0x0210,0x3273,0x2252,0x52b5,0x4294,
+	0x72f7,0x62d6,0x9339,0x8318,0xb37b,0xa35a,0xd3bd,0xc39c,0xf3ff,0xe3de,0x2462,
+	0x3443,0x0420,0x1401,0x64e6,0x74c7,0x44a4,0x5485,0xa56a,0xb54b,0x8528,0x9509,
+	0xe5ee,0xf5cf,0xc5ac,0xd58d,0x3653,0x2672,0x1611,0x0630,0x76d7,0x66f6,0x5695,
+	0x46b4,0xb75b,0xa77a,0x9719,0x8738,0xf7df,0xe7fe,0xd79d,0xc7bc,0x48c4,0x58e5,
+	0x6886,0x78a7,0x0840,0x1861,0x2802,0x3823,0xc9cc,0xd9ed,0xe98e,0xf9af,0x8948,
+	0x9969,0xa90a,0xb92b,0x5af5,0x4ad4,0x7ab7,0x6a96,0x1a71,0x0a50,0x3a33,0x2a12,
+	0xdbfd,0xcbdc,0xfbbf,0xeb9e,0x9b79,0x8b58,0xbb3b,0xab1a,0x6ca6,0x7c87,0x4ce4,
+	0x5cc5,0x2c22,0x3c03,0x0c60,0x1c41,0xedae,0xfd8f,0xcdec,0xddcd,0xad2a,0xbd0b,
+	0x8d68,0x9d49,0x7e97,0x6eb6,0x5ed5,0x4ef4,0x3e13,0x2e32,0x1e51,0x0e70,0xff9f,
+	0xefbe,0xdfdd,0xcffc,0xbf1b,0xaf3a,0x9f59,0x8f78,0x9188,0x81a9,0xb1ca,0xa1eb,
+	0xd10c,0xc12d,0xf14e,0xe16f,0x1080,0x00a1,0x30c2,0x20e3,0x5004,0x4025,0x7046,
+	0x6067,0x83b9,0x9398,0xa3fb,0xb3da,0xc33d,0xd31c,0xe37f,0xf35e,0x02b1,0x1290,
+	0x22f3,0x32d2,0x4235,0x5214,0x6277,0x7256,0xb5ea,0xa5cb,0x95a8,0x8589,0xf56e,
+	0xe54f,0xd52c,0xc50d,0x34e2,0x24c3,0x14a0,0x0481,0x7466,0x6447,0x5424,0x4405,
+	0xa7db,0xb7fa,0x8799,0x97b8,0xe75f,0xf77e,0xc71d,0xd73c,0x26d3,0x36f2,0x0691,
+	0x16b0,0x6657,0x7676,0x4615,0x5634,0xd94c,0xc96d,0xf90e,0xe92f,0x99c8,0x89e9,
+	0xb98a,0xa9ab,0x5844,0x4865,0x7806,0x6827,0x18c0,0x08e1,0x3882,0x28a3,0xcb7d,
+	0xdb5c,0xeb3f,0xfb1e,0x8bf9,0x9bd8,0xabbb,0xbb9a,0x4a75,0x5a54,0x6a37,0x7a16,
+	0x0af1,0x1ad0,0x2ab3,0x3a92,0xfd2e,0xed0f,0xdd6c,0xcd4d,0xbdaa,0xad8b,0x9de8,
+	0x8dc9,0x7c26,0x6c07,0x5c64,0x4c45,0x3ca2,0x2c83,0x1ce0,0x0cc1,0xef1f,0xff3e,
+	0xcf5d,0xdf7c,0xaf9b,0xbfba,0x8fd9,0x9ff8,0x6e17,0x7e36,0x4e55,0x5e74,0x2e93,
+	0x3eb2,0x0ed1,0x1ef0
+    };
+
+void UpdateCrc(const void* aPtr, TInt aLength, TUint16& aCrc)
+//
+// Perform a CCITT CRC checksum.
+//
+	{
+
+	register const TUint8* pB = (const TUint8*)aPtr;
+	register TUint16 crc=aCrc;
+    while (aLength--)
+		crc=TUint16((crc<<8)^crcTab[(crc>>8)^*pB++]);
+	aCrc=crc;
+	}
+
+void ClearCommError()
+	{
+	}
+
+TInt CommRead1(TInt aTimeout)
+	{
+	TBuf8<1> c;
+	TRequestStatus s0, s1;
+	TheComm.ReadOneOrMore(s1, c);
+	TheTimer.After(s0, aTimeout*1000);
+	User::WaitForRequest(s0, s1);
+	if (s1 != KRequestPending)
+		{
+		TheTimer.Cancel();
+		User::WaitForRequest(s0);
+		return (s1==KErrNone) ? c[0] : s1.Int();
+		}
+	TheComm.ReadCancel();
+	User::WaitForRequest(s1);
+	return KErrTimedOut;
+	}
+
+TInt CommRead(TDes8& aBuf, TInt aTimeout)
+	{
+	aBuf.Zero();
+	TRequestStatus s0, s1;
+	TInt timeout = aTimeout * 1000;
+
+	TUint8* b = (TUint8*)aBuf.Ptr();
+	while (aBuf.Length() < aBuf.MaxLength())
+		{
+		TPtr8 p(b+aBuf.Length(), 0, aBuf.MaxLength()-aBuf.Length());
+		TheComm.ReadOneOrMore(s1, p);
+		TheTimer.After(s0, timeout);
+		User::WaitForRequest(s0, s1);
+		if (s1 == KErrNone)
+			{
+			TheTimer.Cancel();
+			User::WaitForRequest(s0);
+			aBuf.SetLength(aBuf.Length() + p.Length());
+			timeout = 500000;
+			continue;
+			}
+		if (s1 != KRequestPending)
+			{
+			TheTimer.Cancel();
+			User::WaitForRequest(s0);
+			break;
+			}
+		if (s0 != KRequestPending)
+			{
+			TheComm.ReadCancel();
+			User::WaitForRequest(s1);
+			break;
+			}
+		}
+	if (aBuf.Length()==0)
+		return KErrTimedOut;
+	return aBuf.Length();
+	}
+
+void CommWrite(const TDesC8& aBuf)
+	{
+	TRequestStatus s;
+	TheComm.Write(s, aBuf);
+	User::WaitForRequest(s);
+	}
+
+void CommWriteC(TUint aChar)
+	{
+	TBuf8<1> b;
+	b.SetLength(1);
+	b[0] = (TUint8)aChar;
+	CommWrite(b);
+	}
+
+void CommWriteS(const char* aString)
+	{
+	CommWrite(TPtrC8((const TUint8*)aString));
+	}
+
+TInt PreparePacket(SPacket& pkt, TInt aSeq)
+	{
+	TInt r = KErrNone;
+	TUint16 crc = 0;
+
+	pkt.iPTI = STX;
+	pkt.iSeq = (TUint8)(aSeq>=0 ? aSeq : 0);
+	pkt.iSeqBar = (TUint8)~pkt.iSeq;
+	if (aSeq>0)
+		{
+		TInt l;
+		TInt fpos = (aSeq-1)*PACKET_SIZE;	// file position of packet
+		if ( fpos >= FileSize )
+			return KErrEof;
+		l = MIN(PACKET_SIZE, FileSize-fpos);
+		TPtr8 d(pkt.iData, 0, l);
+		assert_KErrNone(TheFile.Read(d));
+		if (l<PACKET_SIZE)
+			memset(pkt.iData+l, 0, PACKET_SIZE-l);
+		}
+	else
+		{
+		memset(pkt.iData, 0, PACKET_SIZE);
+		if (aSeq==0)
+			{
+			TInt pos = FileName8.LocateReverse('\\');
+			TPtrC8 fn8(FileName8.Mid(pos+1));
+			memcpy(pkt.iData, fn8.Ptr(), fn8.Length());
+			TPtr8 d2(pkt.iData+fn8.Length()+1, 0, PACKET_SIZE-fn8.Length()-1);
+			d2.Num(FileSize);
+			}
+		}
+	UpdateCrc(pkt.iData, PACKET_SIZE, crc);
+	pkt.iCRC1 = (TUint8)(crc>>8);
+	pkt.iCRC0 = (TUint8)crc;
+	return r;
+	}
+
+TInt SendPacket(TInt& aSeq, TBool aStream)
+	{
+	TInt c;
+	SPacket pkt;
+	TInt retries = 10;
+	TInt tmout = (aSeq>=0) ? 2000 : 500;
+	TInt r = PreparePacket(pkt, aSeq);
+	if (r!=KErrNone)
+		return r;
+	for(;;)
+		{
+		RESET_COMM();
+		CommWrite(TPtrC8( (const TUint8*)&pkt, sizeof(pkt) ));
+		if (aStream)
+			break;
+		c = CommRead1(tmout);
+		if (c==KErrTimedOut && aSeq<0)
+			return KErrNone;
+		if (c>=0)
+			{
+			if (c==ACK)
+				break;
+			}
+		if (--retries==0)
+			return KErrTimedOut;
+		}
+	if (aSeq==0)
+		{
+		c = CommRead1(100);
+		if (c==KErrTimedOut)
+			{
+			++aSeq;
+			return KErrNone;
+			}
+		if (aStream && c!=BIGG)
+			return KErrTimedOut;
+		else if (!aStream && c!=BIGC)
+			return KErrTimedOut;
+		}
+	++aSeq;
+	return r;
+	}
+
+TInt SendImageFile()
+	{
+	TInt r = 0;
+	TInt b1;
+	TBool stream;
+	TInt seq = 0;
+
+	RESET_COMM();
+	r = CommRead1(20000);
+	if (r<0)
+		return r;
+	if (r!=BIGG && r!=BIGC)
+		return KErrTimedOut;
+	stream = (r==BIGG);
+	seq = 0;
+	r = KErrNone;
+	while (r==KErrNone)
+		{
+		r = SendPacket(seq, stream);
+		}
+	if (r!=KErrEof)
+		return r;
+	r=KErrNone;
+	RESET_COMM();
+	CommWriteC(EOT);
+	b1 = CommRead1(500);
+	if (b1==KErrTimedOut)
+		return KErrNone;
+	if (b1!=ACK)
+		return KErrNone;
+	b1 = CommRead1(1000);
+	if (b1==KErrTimedOut)
+		return KErrNone;
+	if (stream && b1!=BIGG)
+		return KErrNone;
+	else if (!stream && b1!=BIGC)
+		return KErrNone;
+	seq = -1;
+	r = SendPacket(seq, stream);
+	return r;
+	}
+
+
+#define	ALLOWBPS(x)	case x : bps = EBps##x; break
+
+void ParseCommandLine()
+	{
+	TInt r = 0;
+	TInt cmdLineLen = User::CommandLineLength();
+	HBufC* cmdLine = HBufC::New(cmdLineLen);
+	assert(cmdLine!=0);
+	TPtr cmdLineW(cmdLine->Des());
+	User::CommandLine(cmdLineW);
+	TLex16 cmdLineLex(*cmdLine);
+	TInt baud = 115200;
+	TInt port = 0;
+	while (!cmdLineLex.Eos())
+		{
+		TPtrC token(cmdLineLex.NextToken());
+		cmdLineLex.SkipSpace();
+		if (token[0]=='-')
+			{
+			switch (token[1])
+				{
+				case 'p':
+					{
+					r = cmdLineLex.Val(port);
+					assert_KErrNone(r);
+					break;
+					}
+				case 'b':
+					{
+					r = cmdLineLex.Val(baud);
+					assert_KErrNone(r);
+					break;
+					}
+				default:
+					assert(0);
+					break;
+				}
+			continue;
+			}
+		FileName = token;
+		assert(cmdLineLex.Eos());
+		}
+	assert(FileName.Length()>0);
+
+	TBps bps = EBpsSpecial;
+	switch (baud)
+		{
+		ALLOWBPS(50);
+		ALLOWBPS(75);
+		ALLOWBPS(110);
+		ALLOWBPS(134);
+		ALLOWBPS(150);
+		ALLOWBPS(300);
+		ALLOWBPS(600);
+		ALLOWBPS(1200);
+		ALLOWBPS(1800);
+		ALLOWBPS(2000);
+		ALLOWBPS(2400);
+		ALLOWBPS(3600);
+		ALLOWBPS(4800);
+		ALLOWBPS(7200);
+		ALLOWBPS(9600);
+		ALLOWBPS(19200);
+		ALLOWBPS(38400);
+		ALLOWBPS(57600);
+		ALLOWBPS(115200);
+		ALLOWBPS(230400);
+		ALLOWBPS(460800);
+		ALLOWBPS(576000);
+		ALLOWBPS(1152000);
+		ALLOWBPS(4000000);
+		ALLOWBPS(921600);
+		}
+	assert(bps != EBpsSpecial);
+
+	assert_KErrNone(TheFs.Connect());
+	assert_KErrNone(TheFile.Open(TheFs, FileName, EFileShareReadersOnly|EFileRead));
+	assert_KErrNone(TheFile.Size(FileSize));
+	FileName8b = FileName;
+	new (&FileName8) TPtr8(FileName8b.Collapse());
+	assert(TheComm.Open(port)==KErrNone);
+	TCommConfig cfgBuf;
+	TCommConfigV01& cfg = cfgBuf();
+	cfg.iRate = bps;
+	cfg.iDataBits = EData8;
+	cfg.iStopBits = EStop1;
+	cfg.iParity = EParityNone;
+	cfg.iHandshake = 0;
+	cfg.iParityError = 0;
+	cfg.iFifo = EFifoEnable;
+	cfg.iSpecialRate = 0;
+	cfg.iTerminatorCount = 0;
+	cfg.iSIREnable = ESIRDisable;
+	cfg.iSIRSettings = KConfigSIRShutDown;
+	assert_KErrNone(TheComm.SetConfig(cfgBuf));
+	CommWrite(_L8("Sending "));
+	CommWrite(FileName8);
+	CommWrite(_L8("\r\nWaiting for YModem or YModem-G receive...\r\n"));
+	}
+
+_LIT(KLddName,"ECOMM");
+_LIT(KPddName,"EUART");
+
+void LoadCommDrivers()
+	{
+	TInt r=User::LoadLogicalDevice(KLddName);
+	if (r!=KErrAlreadyExists)
+		assert_KErrNone(r);
+
+	TInt i;
+	TInt n=0;
+	for (i=-1; i<10; ++i)
+		{
+		TBuf<16> pddName=KPddName();
+		if (i>=0)
+			pddName.Append('0'+i);
+		r=User::LoadPhysicalDevice(pddName);
+		if (r==KErrNone || r==KErrAlreadyExists)
+			++n;
+		}
+	assert(n!=0);
+	}
+
+TInt E32Main()
+	{
+	TInt r;
+
+	LoadCommDrivers();
+	ParseCommandLine();
+
+	r = TheTimer.CreateLocal();
+	assert_KErrNone(r);
+
+	r = SendImageFile();
+	assert_KErrNone(r);
+
+	TheTimer.Close();
+	TheComm.Close();
+	TheFile.Close();
+	TheFs.Close();
+
+	return KErrNone;
+	}