diff -r 000000000000 -r 2c201484c85f cryptoservices/filebasedcertificateandkeystores/test/keytool/keytool.cpp --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/cryptoservices/filebasedcertificateandkeystores/test/keytool/keytool.cpp Wed Jul 08 11:25:26 2009 +0100 @@ -0,0 +1,387 @@ +// Copyright (c) 2004-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 "Symbian Foundation License v1.0" +// which accompanies this distribution, and is available +// at the URL "http://www.symbianfoundation.org/legal/sfl-v10.html". +// +// Initial Contributors: +// Nokia Corporation - initial contribution. +// +// Contributors: +// +// Description: +// + +#include +#include + +#include "keytool_utils.h" +#include "keytool_defs.h" +#include "keytool_controller.h" +#include "keytool_view_imp.h" +#include "keytool_commands.h" +#include "keytoolfileview.h" + + +// Boiler plate +_LIT(KShortName, "Symbian OS KeyTool"); +_LIT(KName, "Symbian OS KeyStore Manipulation Tool"); +_LIT(KCopyright, "Copyright (c) 2004 - 2007 Symbian Software Ltd. All rights reserved."); + +_LIT(KNewLine, "\n"); + +_LIT(KDone, "Press any key to continue... \n"); + +// Keytool command line parameters +_LIT(KList, "-list"); +_LIT(KListShort, "-l"); + +_LIT(KListStores, "-liststores"); +_LIT(KListStoresShort, "-ls"); + +_LIT(KImport, "-import"); +_LIT(KImportShort, "-i"); + + +_LIT(KRemove, "-remove"); +_LIT(KRemoveShort, "-r"); + +_LIT(KHelp, "-help"); +_LIT(KHelpShort, "-h"); + +_LIT(KSetUser, "-setuser"); +_LIT(KSetUserShort, "-s"); + +_LIT(KSetAllUsers, "-setallusers"); +_LIT(KSetAllUsersShort, "-a"); + +_LIT(KSetManager, "-setmanager"); +_LIT(KSetManagerShort, "-m"); + +_LIT(KRemoveUser, "-removeuser"); +_LIT(KRemoveUserShort, "-ru"); + +// Command parameters +_LIT(KLabel, "-label"); +_LIT(KKeyUsage, "-usage"); +_LIT(KAccess, "-access"); + +_LIT(KStore, "-store"); + +//_LIT(KExpiry, "-expiry"); // Functionality not present at the moment... +_LIT(KDetails, "-details"); +_LIT(KDetailsShort, "-d"); + +_LIT(KPageWise, "-page"); +_LIT(KPageWiseShort, "-p"); + + +const TInt KMaxArgs = 10; + +/** + * Displays tool name and copyright informations. + */ +LOCAL_D void BoilerPlateL(CConsoleBase* console) + { + console->Printf(KNewLine); + console->Printf(KName); + console->Printf(KNewLine); + console->Printf(KCopyright); + console->Printf(KNewLine); + console->Printf(KNewLine); + } + + +LOCAL_D TBool VerifyCommand(const TDesC& aCommand, TInt& aCmdNum, TInt& aCmdCount) + { + if ((aCmdNum != -1) && (aCommand[0] == '-')) + { + aCmdNum = KeyToolDefController::KUsageCommand; + aCmdCount = KMaxArgs; + return 1; + } + if (aCommand.CompareF(KList) == 0 || aCommand.Compare(KListShort) == 0) + { + aCmdNum = KeyToolDefController::KListCommand; + } + else if (aCommand.CompareF(KListStores) == 0 || aCommand.Compare(KListStoresShort) == 0) + { + aCmdNum = KeyToolDefController::KListStoresCommand; + } + else if (aCommand.CompareF(KImport) == 0 || aCommand.Compare(KImportShort) == 0) + { + aCmdNum = KeyToolDefController::KImportCommand; + } + else if (aCommand.CompareF(KRemove) == 0 || aCommand.Compare(KRemoveShort) == 0) + { + aCmdNum = KeyToolDefController::KRemoveCommand; + } + else if (aCommand.CompareF(KSetUser) == 0 || aCommand.Compare(KSetUserShort) == 0) + { + aCmdNum = KeyToolDefController::KSetUserCommand; + } + else if (aCommand.CompareF(KSetManager) == 0 || aCommand.Compare(KSetManagerShort) == 0) + { + aCmdNum = KeyToolDefController::KSetManagerCommand; + } + else if (aCommand.CompareF(KRemoveUser) == 0 || aCommand.Compare(KRemoveUserShort) == 0) + { + aCmdNum = KeyToolDefController::KRemoveUserCommand; + } + else if (aCommand.CompareF(KSetAllUsers) == 0 || aCommand.Compare(KSetAllUsersShort) == 0) + { + aCmdNum = KeyToolDefController::KSetAllUsersCommand; + } + else + { + return 0; + } + + return 1; + } + + +/** + * Parses the command line and given control to the handler to deal with the request. + */ +LOCAL_D void DoMainL() + { + TBool interactiveMode = ETrue; + + RFs fs; + User::LeaveIfError(fs.Connect()); + CleanupClosePushL(fs); + + CConsoleBase* console = Console::NewL(KShortName, TSize(KConsFullScreen, KConsFullScreen)); + CleanupStack::PushL(console); + CCommandLineArguments* cmdArgs = CCommandLineArguments::NewLC(); + TInt cmdArgsCount = cmdArgs->Count(); + + RFile file; + // command: keytool inputfile outputfile + if (cmdArgsCount == 3) + { + interactiveMode = EFalse; + TInt error = file.Open(fs, cmdArgs->Arg(1), EFileRead|EFileShareAny); + file.Close(); + TInt error1 = file.Replace(fs, cmdArgs->Arg(2), EFileWrite|EFileShareExclusive); + CleanupClosePushL(file); + // If the input file doesn't exist or not able to create outputfile + // switch to Interactive mode + if (error != KErrNone || error1 != KErrNone) + { + CleanupStack::PopAndDestroy(&file); + interactiveMode = ETrue; + } + } + + CKeytoolConsoleView* view = CKeytoolConsoleView::NewLC(*console); + CKeyToolController* controller = CKeyToolController::NewLC(*view); + CArrayFixFlat* args = new (ELeave) CArrayFixFlat (10); + CleanupStack::PushL(args); + CKeytoolFileView* view1 = NULL; + + TInt cmdCount = 0; + if (interactiveMode) + { + KeyToolUtils::SetConsole(console); + BoilerPlateL(console); + for (TInt i = 0; i < cmdArgsCount; i++) + { + args->AppendL(cmdArgs->Arg(i)); + } + // In Interactive mode by default we can execute 1 command only. + cmdCount = 1; + } + else + { + CleanupStack::PopAndDestroy(3, view); // args, controller, view + KeyToolUtils::SetFile(&file); + view1 = CKeytoolFileView::NewLC(cmdArgs->Arg(1)); + cmdCount = view1->SplitFileInputToArrayL(); + } + + for (TInt j = 0; j < cmdCount; j++) + { + if (!interactiveMode) + { + controller = CKeyToolController::NewLC(*view1); + args = view1->ReadArrayArgumentsLC(j); + } + + CKeyToolParameters* params = CKeyToolParameters::NewLC(); + + TInt command = -1; + TInt i = -1; + + TInt argsCount = args->Count(); + while (i < (argsCount-1)) + { + i++; + + if ((args->At(i).CompareF(KDetails)==0)|| (args->At(i).Compare(KDetailsShort)==0)) + { + params->iIsDetailed = ETrue; + continue; + } + + if (args->At(i).CompareF(KPageWise)==0 || (args->At(i).Compare(KPageWiseShort)==0)) + { + params->iPageWise = ETrue; + continue; + } + + TDesC& commd = args->At(i); + if (VerifyCommand(commd, command, i)) + { + if (commd.CompareF(KSetAllUsers) == 0 || commd.Compare(KSetAllUsersShort) == 0 || + commd.CompareF(KRemoveUser) == 0 || commd.Compare(KRemoveUserShort) == 0 || + commd.CompareF(KSetManager) == 0 || commd.Compare(KSetManagerShort) == 0 ) + { + i++; + if (args->At(i)[0] == '-' || args->At(i).Left(2) == _L("0x")) + { + i = argsCount; + command = KeyToolDefController::KUsageCommand; + } + else + { + i--; + } + } + if (commd.CompareF(KSetAllUsers) == 0 || commd.Compare(KSetAllUsersShort) == 0) + { + params->iPolicy = CKeyToolParameters::ESetAllUsersPolicy; + params->iIsDetailed = ETrue; + } + if (commd.CompareF(KRemoveUser) == 0 || commd.Compare(KRemoveUserShort) == 0) + { + params->iPolicy = CKeyToolParameters::ERemoveUserPolicy; + params->iIsDetailed = ETrue; + } + if (commd.CompareF(KSetManager) == 0 || commd.Compare(KSetManagerShort) == 0) + { + params->iPolicy = CKeyToolParameters::ESetManagerPolicy; + params->iIsDetailed = ETrue; + } + if (commd.CompareF(KSetUser) == 0 || commd.Compare(KSetUserShort) == 0) + { + params->iPolicy = CKeyToolParameters::ESetUserPolicy; + i++; + if (i >= argsCount || args->At(i)[0] == '-' || args->At(i).Left(2) != _L("0x")) + { + i = argsCount; + command = KeyToolDefController::KUsageCommand; + } + else + { + TUid app; + TLex lex(args->At(i).Mid(2)); + TUint uid =0; + TInt err = lex.Val(uid, EHex); + if (err == KErrNone) + { + app = TUid::Uid(uid); + } + params->iUIDs.Append(app); + params->iIsDetailed = ETrue; + } + } + continue; + } + + TDesC& cmd = args->At(i); + if (cmd.CompareF(KHelp) == 0 || cmd.Compare(KHelpShort) == 0 || + cmd.CompareF(KLabel) == 0 || cmd.CompareF(KKeyUsage) == 0 || + cmd.CompareF(KStore) == 0 || cmd.CompareF(KAccess) == 0) + { + i++; + if (i >= argsCount || args->At(i)[0] == '-') + { + i = argsCount; + command = KeyToolDefController::KUsageCommand; + } + else if (cmd.CompareF(KHelp) == 0 || cmd.Compare(KHelpShort) == 0) + { + params->iDefault = args->At(i).AllocL(); + i = argsCount; + } + else if (cmd.CompareF(KLabel) == 0) + { + params->iLabel = args->At(i).AllocL(); + } + else if (cmd.CompareF(KKeyUsage) == 0) + { + params->iUsage = KeyToolUtils::ParseKeyUsage(args->At(i)); + } + else if (cmd.CompareF(KStore) == 0) + { + TLex parser(args->At(i)); + TInt err = parser.Val(params->iKeystoreIndex); + params->iIsDetailed = ETrue; + } + else if (cmd.CompareF(KAccess) == 0) + { + params->iAccess = KeyToolUtils::ParseKeyAccess(args->At(i)); + } + else + { + // no action required + } + continue; + } + + if (i!=0) + { + if (args->At(i)[0] == '-') + { + i = argsCount; + command = KeyToolDefController::KUsageCommand; + continue; + } + delete params->iDefault; + params->iDefault = NULL; + params->iDefault = args->At(i).AllocL(); + } + } + + + if (command != -1) + { + TRAP_IGNORE(controller->HandleCommandL(command, params)); + } + else + { + controller->HandleCommandL(KeyToolDefController::KUsageCommand, params); + } + + CleanupStack::PopAndDestroy(3, controller); // params, args, controller + } + if (interactiveMode) + { + CleanupStack::PopAndDestroy(view); + // We are done! + console->Printf(KNewLine); + console->Printf(KDone); + console->Getch(); + } + else + { + CleanupStack::PopAndDestroy(2, &file); //view1 and file + } + CleanupStack::PopAndDestroy(3, &fs); //cmdArgs, console, fs + } + + +GLDEF_C TInt E32Main() // main function called by E32 + { + __UHEAP_MARK; + CTrapCleanup* cleanup=CTrapCleanup::New(); + + TRAP_IGNORE(DoMainL()); + + delete cleanup; + __UHEAP_MARKEND; + return 0; + }