commands/hash/hash.cpp
changeset 0 7f656887cf89
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/commands/hash/hash.cpp	Wed Jun 23 15:52:26 2010 +0100
@@ -0,0 +1,178 @@
+// hash.cpp
+// 
+// Copyright (c) 2009 - 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 <fshell/ltkutils.h>
+#include <fshell/common.mmh>
+#include <hash.h>
+
+const TInt KBlockSize = 512;
+
+using namespace IoUtils;
+
+class CCmdHash : public CCommandBase
+	{
+public:
+	static CCommandBase* NewLC();
+	~CCmdHash();
+private:
+	CCmdHash();
+	CMessageDigest::THashId HashIdL();
+private: // From CCommandBase.
+	virtual const TDesC& Name() const;
+	virtual void DoRunL();
+	virtual void OptionsL(RCommandOptionList& aOptions);
+private:
+	TBool iVerbose;
+	TFileName2 iInputFileName;
+	TFileName2 iOutputFileName;
+	enum TAlgorithm
+		{
+		ESha1,
+		ESha224,
+		ESha256,
+		ESha384,
+		ESha512,
+		EMd2,
+		EMd4,
+		EMd5
+		} iAlgorithm;
+	CMessageDigest* iMessageDigest;
+	};
+
+
+CCommandBase* CCmdHash::NewLC()
+	{
+	CCmdHash* self = new(ELeave) CCmdHash();
+	CleanupStack::PushL(self);
+	self->BaseConstructL();
+	return self;
+	}
+
+CCmdHash::~CCmdHash()
+	{
+	delete iMessageDigest;
+	}
+
+CCmdHash::CCmdHash()
+	{
+	}
+
+CMessageDigest::THashId CCmdHash::HashIdL()
+	{
+	switch (iAlgorithm)
+		{
+		case ESha1:
+			return CMessageDigest::ESHA1;
+		case EMd2:
+			return CMessageDigest::EMD2;
+#ifdef FSHELL_MD4_SUPPORT
+		case EMd4:
+			return CMessageDigest::EMD4;
+#endif // FSHELL_MD4_SUPPORT
+		case EMd5:
+			return CMessageDigest::EMD5;
+#ifdef FSHELL_SHA2_SUPPORT
+		case ESha224:
+			return CMessageDigest::ESHA224;
+		case ESha256:
+			return CMessageDigest::ESHA256;
+		case ESha384:
+			return CMessageDigest::ESHA384;
+		case ESha512:
+			return CMessageDigest::ESHA512;
+#endif // FSHELL_SHA2_SUPPORT
+		default:
+			LeaveIfErr(KErrNotSupported, _L("Requested hashing algorithm is not available in this Symbian OS platform"));
+			return CMessageDigest::ESHA1;
+		}
+	}
+
+const TDesC& CCmdHash::Name() const
+	{
+	_LIT(KName, "hash");	
+	return KName;
+	}
+
+void CCmdHash::DoRunL()
+	{
+	iMessageDigest = CMessageDigestFactory::NewDigestL(HashIdL());
+
+	RIoReadHandle readHandle;
+	readHandle.Create(IoSession());
+	CleanupClosePushL(readHandle);
+
+	RIoFile ioFile;
+	if (iOptions.IsPresent(&iInputFileName))
+		{
+		LeaveIfErr(ioFile.Create(IoSession(), iInputFileName, RIoFile::ERead), _L("Couldn't open handle to file '%S'"), &iInputFileName);
+		CleanupClosePushL(ioFile);
+		LeaveIfErr(ioFile.Attach(readHandle, RIoEndPoint::EForeground), _L("Couldn't attach read handle to file object"));
+		}
+	else
+		{
+		LeaveIfErr(readHandle.Duplicate(Stdin()), _L("Couldn't duplicate STDIN"));
+		}
+
+	LeaveIfErr(readHandle.SetReadMode(RIoReadHandle::EOneOrMore), _L("Couldn't set read handle into read one or more mode"));
+
+	TBuf<KBlockSize> buf;
+	TInt err = KErrNone;
+	do
+		{
+		err = readHandle.Read(buf);
+		if (err == KErrNone)
+			{
+			iMessageDigest->Update(buf.Collapse());
+			}
+		}
+		while (err == KErrNone);
+
+	if (err != KErrEof)
+		{
+		LeaveIfErr(err, _L("Unable to read input"));
+		}
+
+	TPtrC8 hash(iMessageDigest->Final());
+
+	if (iOptions.IsPresent(&iOutputFileName))
+		{
+		RFile outputFile;
+		LeaveIfErr(outputFile.Create(FsL(), iOutputFileName, EFileWrite), _L("Couldn't create '%S'"), &iOutputFileName);
+		CleanupClosePushL(outputFile);
+		outputFile.Write(hash);
+		CleanupStack::PopAndDestroy(&outputFile);
+		}
+	else
+		{
+		LtkUtils::HexDumpToOutput(hash, Stdout());
+		}
+
+	if (iOptions.IsPresent(&iInputFileName))
+		{
+		CleanupStack::PopAndDestroy(&ioFile);
+		}
+
+	CleanupStack::PopAndDestroy(&readHandle);
+	}
+
+void CCmdHash::OptionsL(RCommandOptionList& aOptions)
+	{
+	aOptions.AppendBoolL(iVerbose, _L("verbose"));
+	aOptions.AppendFileNameL(iInputFileName, _L("input"));
+	aOptions.AppendFileNameL(iOutputFileName, _L("output"));
+	aOptions.AppendEnumL((TInt&)iAlgorithm, _L("algorithm"));
+	}
+
+
+EXE_BOILER_PLATE(CCmdHash)
+