// ymodem.cpp+
−
// +
−
// Copyright (c) 2008 - 2010 Accenture. All rights reserved.+
−
// This component and the accompanying materials are made available+
−
// under the terms of the "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:+
−
// Accenture - Initial contribution+
−
//+
−
+
−
#include <fshell/ioutils.h>+
−
#include "ymodem.h"+
−
+
−
using namespace IoUtils;+
−
+
−
CCommandBase* CCmdYmodem::NewLC()+
−
	{+
−
	CCmdYmodem* self = new(ELeave) CCmdYmodem();+
−
	CleanupStack::PushL(self);+
−
	self->BaseConstructL();+
−
	self->ConstructL();+
−
	return self;+
−
	}+
−
+
−
CCmdYmodem::~CCmdYmodem()+
−
	{+
−
	iFileNames.Close();+
−
	}+
−
+
−
CCmdYmodem::CCmdYmodem()+
−
	{+
−
	}+
−
+
−
const TDesC& CCmdYmodem::Name() const+
−
	{+
−
	_LIT(KName, "ymodem");	+
−
	return KName;+
−
	}+
−
+
−
class TFailedFile+
−
	{+
−
public:+
−
	TFailedFile();+
−
public:+
−
	TInt iError;+
−
	TFileName2 iFileName;+
−
	};+
−
+
−
TFailedFile::TFailedFile()+
−
	: iError(KErrNone)+
−
	{+
−
	}+
−
+
−
void CCmdYmodem::DoRunL()+
−
	{+
−
	PrepareConsoleToTransferL();+
−
+
−
	if (iMode == EReceive)+
−
		{+
−
		TPtrC receiveDir;+
−
		if (iFileNames.Count() > 1)+
−
			{+
−
			PrintError(KErrArgument, _L("Too many arguments - only the receive directory may be specified in receive mode."));+
−
			User::Leave(KErrArgument);+
−
			}+
−
		else if (iFileNames.Count() == 1)+
−
			{+
−
			TFileName2 fileName(iFileNames[0]);+
−
			receiveDir.Set(iFileNames[0]);+
−
+
−
			if ((fileName.Exists(FsL()) && !fileName.IsDirL(FsL())) || (!fileName.Exists(FsL()) && !iOverwrite))+
−
				{+
−
				LeaveIfErr(KErrArgument, _L("\"%S\" is not a directory"), &fileName);+
−
				}+
−
			}+
−
		else+
−
			{+
−
			receiveDir.Set(Env().Pwd());+
−
			}+
−
+
−
		Progress(_L("YMODEM receive to \"%S\"\r\n"), &receiveDir);+
−
		RArray<TFileName2> receivedFiles;+
−
		CleanupClosePushL(receivedFiles);+
−
		TFailedFile failedFile;+
−
		TBool finished(EFalse);+
−
		while (!finished)+
−
			{+
−
			TSyncResult syncResult = SendSyncL();+
−
			if (syncResult == EEot)+
−
				{+
−
				finished = ETrue;+
−
				}+
−
			else+
−
				{+
−
				iPacketNumber = 0;+
−
				TBool isFinalBlock(EFalse);+
−
				TPtrC block(ReceiveBlockL(ETrue, isFinalBlock));+
−
				TPtrC fileName(block.Ptr()); // filename is the first null terminated string in the block.+
−
				if (fileName.Length() > 0)+
−
					{+
−
					TFileName2 fileName2(fileName);+
−
					fileName2.MakeAbsoluteL(receiveDir);+
−
					TLex lex(block.Mid(fileName.Length() + 1));+
−
					TInt size;+
−
					User::LeaveIfError(lex.Val(size));+
−
					TInt err = KErrNone;+
−
					if (!iOverwrite && fileName2.Exists(FsL()))+
−
						{+
−
						err = KErrAlreadyExists;+
−
						}+
−
					RFile file;+
−
					if (err == KErrNone)+
−
						{+
−
						if (size == 0)+
−
							{+
−
							if (iOverwrite)+
−
								{+
−
								err = file.Replace(FsL(), fileName2, EFileWrite);+
−
								}+
−
							else+
−
								{+
−
								err = file.Create(FsL(), fileName2, EFileWrite);+
−
								}+
−
							}+
−
						else+
−
							{+
−
							Progress(_L("Starting receive of \"%S\" (%d bytes)\r\n"), &fileName, size);+
−
							SendSyncL();+
−
							TRAP(err, ReceiveToFileL(fileName2));+
−
							if (err == KErrNone)+
−
								{+
−
								err = file.Open(FsL(), fileName2, EFileWrite);+
−
								if (err == KErrNone)+
−
									{+
−
									err = file.SetSize(size);+
−
									}+
−
								}+
−
							else+
−
								{+
−
								Fs().Delete(fileName2);+
−
								}+
−
							}+
−
						}+
−
					else+
−
						{+
−
						TRAP_IGNORE(ReceiveToNullL());+
−
						}+
−
					file.Close();+
−
					if (err)+
−
						{+
−
						failedFile.iError = err;+
−
						failedFile.iFileName = fileName2;+
−
						finished = ETrue;+
−
						}+
−
					else+
−
						{+
−
						receivedFiles.Append(fileName2); // Ignore error.+
−
						}+
−
					}+
−
				else+
−
					{+
−
					finished = ETrue;+
−
					}+
−
				}+
−
			}+
−
+
−
		CleanupClonsoleAfterTransferL();+
−
+
−
		const TInt numFilesReceived = receivedFiles.Count();+
−
		if (numFilesReceived == 0)+
−
			{+
−
			if (failedFile.iError == KErrNone)+
−
				{+
−
				Printf(_L("No files to receive.\r\n"));+
−
				}+
−
			}+
−
		else if (numFilesReceived == 1)+
−
			{+
−
			const TFileName2& fileName = receivedFiles[0];+
−
			Printf(_L("Successfully received \"%S\".\r\n"), &fileName);+
−
			}+
−
		else+
−
			{+
−
			Printf(_L("Successfully received:\r\n"));+
−
			for (TInt i = 0; i < numFilesReceived; ++i)+
−
				{+
−
				const TFileName2& fileName = receivedFiles[i];+
−
				Printf(_L("\t\"%S\"\r\n"), &fileName);+
−
				}+
−
			}+
−
+
−
		if (failedFile.iError)+
−
			{+
−
			PrintError(failedFile.iError, _L("Failed to receive \"%S\""), &failedFile.iFileName);+
−
			}+
−
		CleanupStack::PopAndDestroy(&receivedFiles);+
−
		}+
−
	else if (iMode == ESend)+
−
		{+
−
		const TInt numFiles = iFileNames.Count();+
−
		if (numFiles == 0)+
−
			{+
−
			PrintError(KErrArgument, _L("No files to send specified"));+
−
			User::Leave(KErrArgument);+
−
			}+
−
		for (TInt i = 0; i < numFiles; ++i)+
−
			{+
−
			const TFileName2& fileName = iFileNames[i];+
−
			if (fileName.Exists(FsL()))+
−
				{+
−
				if (fileName.IsDirL(FsL()))+
−
					{+
−
					LeaveIfErr(KErrArgument, _L("\"%S\" is not a file"), &fileName);+
−
					}+
−
				}+
−
			else+
−
				{+
−
				LeaveIfErr(KErrArgument, _L("\"%S\" does not exist"), &fileName);+
−
				}+
−
			}+
−
		WaitForSyncL();+
−
		HBufC* buf = HBufC::NewLC(iBlockSize);+
−
		TPtr bufPtr(buf->Des());+
−
		for (TInt i = 0; i < numFiles; ++i)+
−
			{+
−
			const TFileName2& fileName = iFileNames[i];+
−
			LeaveIfFileNotFound(fileName);+
−
			TEntry entry;+
−
			User::LeaveIfError(FsL().Entry(fileName, entry));+
−
			bufPtr.SetLength(bufPtr.MaxLength());+
−
			bufPtr.FillZ();+
−
			bufPtr.Copy(fileName.NameAndExt());+
−
			bufPtr.SetLength(buf->Length() + 1); // To leave a terminating null after the file name.+
−
			bufPtr.AppendNum(entry.iSize);+
−
			if (bufPtr.Length() < iBlockSize)+
−
				{+
−
				// There's not enough data to fill this block, so pad with NULL.+
−
				bufPtr.AppendFill(0, iBlockSize - bufPtr.Length());+
−
				}+
−
			iPacketNumber = 0;+
−
			SendBlockL(*buf);+
−
			WaitForSyncL();+
−
			SendFileL(fileName);+
−
			}+
−
		iPacketNumber = 0;+
−
		bufPtr.SetLength(bufPtr.MaxLength());+
−
		bufPtr.FillZ();+
−
		SendBlockL(*buf);+
−
		CleanupStack::PopAndDestroy(buf);+
−
		CleanupClonsoleAfterTransferL();+
−
+
−
		if (numFiles == 1)+
−
			{+
−
			Printf(_L("Successfully sent \"%S\".\r\n"), &iFileNames[0]);+
−
			}+
−
		else+
−
			{+
−
			Printf(_L("Successfully sent:\r\n"));+
−
			for (TInt i = 0; i < numFiles; ++i)+
−
				{+
−
				const TFileName2& fileName = iFileNames[i];+
−
				Printf(_L("\t\"%S\"\r\n"), &fileName);+
−
				}+
−
			}+
−
		}+
−
	User::LeaveIfError(Stdin().CancelCaptureAllKeys());+
−
	}+
−
+
−
void CCmdYmodem::ArgumentsL(RCommandArgumentList& aArguments)+
−
	{+
−
	_LIT(KArg1, "mode");+
−
	aArguments.AppendEnumL((TInt&)iMode, KArg1);+
−
	_LIT(KArg2, "file_name");+
−
	aArguments.AppendFileNameL(iFileNames, KArg2);+
−
	}+
−
+
−
void CCmdYmodem::OptionsL(RCommandOptionList& aOptions)+
−
	{+
−
	CCmdXmodem::OptionsL(aOptions);+
−
	}+
−
+
−