1.1 --- /dev/null Thu Jan 01 00:00:00 1970 +0000
1.2 +++ b/os/security/cryptomgmtlibs/securitytestfw/test/testhandler2/t_policy.cpp Fri Jun 15 03:10:57 2012 +0200
1.3 @@ -0,0 +1,541 @@
1.4 +/*
1.5 +* Copyright (c) 2004-2009 Nokia Corporation and/or its subsidiary(-ies).
1.6 +* All rights reserved.
1.7 +* This component and the accompanying materials are made available
1.8 +* under the terms of the License "Eclipse Public License v1.0"
1.9 +* which accompanies this distribution, and is available
1.10 +* at the URL "http://www.eclipse.org/legal/epl-v10.html".
1.11 +*
1.12 +* Initial Contributors:
1.13 +* Nokia Corporation - initial contribution.
1.14 +*
1.15 +* Contributors:
1.16 +*
1.17 +* Description:
1.18 +*
1.19 +*/
1.20 +
1.21 +
1.22 +#include "t_policy.h"
1.23 +#include "t_input.h"
1.24 +#include "t_output.h"
1.25 +#include "t_testhandler.h"
1.26 +#include "utf.h"
1.27 +
1.28 +#include <s32file.h>
1.29 +
1.30 +// run failure tests first
1.31 +
1.32 +_LIT8(KTestExeStart, "<testexe>");
1.33 +_LIT8(KExcludedCapsStart, "<excludedcapabilities>");
1.34 +_LIT8(KPolicyStart, "<policy>");
1.35 +_LIT8(KPreActionsStart, "<preactions>");
1.36 +_LIT8(KPassAcionStart, "<passactions>");
1.37 +_LIT8(KFailAcionStart, "<failactions>");
1.38 +_LIT8(KPostActionsStart, "<postactions>");
1.39 +_LIT8(KSecureIdStart, "<secureid>");
1.40 +_LIT8(KSecureIdEnd, "</secureid>");
1.41 +_LIT8(KVendorIdStart, "<vendorid>");
1.42 +_LIT8(KVendorIdEnd, "</vendorid>");
1.43 +
1.44 +_LIT(KFormat,"Action Name : %S \n");
1.45 +_LIT(KSetCapExe, "setcap");
1.46 +_LIT(KSetCapExe2, "setcap : ");
1.47 +//NOTE :If the below literal is uncommented , remove the c:\hardcoded reference
1.48 +// and replace with RFs::GetSystemDrive().
1.49 +//_LIT(KTestExeTmpPath, "c:\\sys\\bin\\policytest_exe.exe");
1.50 +_LIT(KPassScriptPath, "\\policytest_script_pass.txt");
1.51 +_LIT(KFailScriptPath, "\\policytest_script_fail.txt");
1.52 +_LIT(KLogTmpPath, "\\policytest_log.txt");
1.53 +_LIT(KTestPath, "policytest_");
1.54 +_LIT(KFailTestRunning, "Fail Test Script Running");
1.55 +_LIT(KPassTestRunning, "Pass Test Script Running");
1.56 +
1.57 +
1.58 +const TUint KCapabilityAll = 0xffffffff;
1.59 +
1.60 +CPolicyTest* CPolicyTest::NewL(CConsoleBase& aConsole,
1.61 + Output& aOut,
1.62 + const TTestActionSpec& aTestActionSpec)
1.63 + {
1.64 + CPolicyTest* self = CPolicyTest::NewLC(aConsole, aOut, aTestActionSpec);
1.65 + CleanupStack::Pop(self);
1.66 + return self;
1.67 + }
1.68 +
1.69 +CPolicyTest* CPolicyTest::NewLC(CConsoleBase& aConsole,
1.70 + Output& aOut,
1.71 + const TTestActionSpec& aTestActionSpec)
1.72 + {
1.73 + CPolicyTest* self = new(ELeave) CPolicyTest(aConsole, aOut);
1.74 + CleanupStack::PushL(self);
1.75 + self->ConstructL(aTestActionSpec);
1.76 + return self;
1.77 + }
1.78 +
1.79 +CPolicyTest::CPolicyTest(CConsoleBase& aConsole,
1.80 + Output& aOut)
1.81 + : CTestAction(aConsole, aOut)
1.82 + {
1.83 + }
1.84 +
1.85 +void CPolicyTest::ConstructL(const TTestActionSpec& aTestActionSpec)
1.86 + {
1.87 + CTestAction::ConstructL(aTestActionSpec);
1.88 + iExpectedResult = KErrNone;
1.89 +
1.90 + User::LeaveIfError(iFs.Connect());
1.91 +
1.92 + RProcess thisProcess;
1.93 + User::LeaveIfError(thisProcess.Open(thisProcess.Id()));
1.94 + iProcessSecureId = thisProcess.SecureId();
1.95 + iProcessVendorId = thisProcess.VendorId();
1.96 + thisProcess.Close();
1.97 +
1.98 + SetTestExeL(Input::ParseElement(aTestActionSpec.iActionBody, KTestExeStart));
1.99 + Input::ParseCapabilitySetL(Input::ParseElement(aTestActionSpec.iActionBody, KExcludedCapsStart), iExcludedCaps);
1.100 + SetPolicyL(Input::ParseElement(aTestActionSpec.iActionBody, KPolicyStart));
1.101 + iPreActions = Input::ParseElement(aTestActionSpec.iActionBody, KPreActionsStart).AllocL();
1.102 + SetTestActionL(Input::ParseElement(aTestActionSpec.iActionBody, KPassAcionStart),
1.103 + Input::ParseElement(aTestActionSpec.iActionBody, KFailAcionStart));
1.104 + iPostActions = Input::ParseElement(aTestActionSpec.iActionBody, KPostActionsStart).AllocL();
1.105 + }
1.106 +
1.107 +CPolicyTest::~CPolicyTest()
1.108 + {
1.109 + iFs.Close();
1.110 + delete iTestExe;
1.111 + iCapabilities.Close();
1.112 + delete iPreActions;
1.113 + delete iPassAction;
1.114 + delete iFailAction;
1.115 + delete iPostActions;
1.116 + iProcess.Close();
1.117 + }
1.118 +
1.119 +void CPolicyTest::BadUsageL(const TDesC& aMessage)
1.120 + {
1.121 + iOut.writeString(_L("Error in script action testpolicy"));
1.122 + iOut.writeNewLine();
1.123 + iOut.writeString(aMessage);
1.124 + iOut.writeNewLine();
1.125 + User::Leave(KErrArgument);
1.126 + }
1.127 +
1.128 +void CPolicyTest::SetTestExeL(const TDesC8& aPath)
1.129 + {
1.130 + if (aPath == KNullDesC8)
1.131 + {
1.132 + BadUsageL(_L("testexe not specified"));
1.133 + }
1.134 +
1.135 + iTestExe = HBufC::NewMaxL(aPath.Length());
1.136 + TPtr ptr = iTestExe->Des();
1.137 + ptr.Copy(aPath);
1.138 + }
1.139 +
1.140 +void CPolicyTest::SetPolicyL(const TDesC8& aSpec)
1.141 + {
1.142 + iSecureId = Input::ParseIntElement(aSpec, KSecureIdStart, KSecureIdEnd);
1.143 + iVendorId = Input::ParseIntElement(aSpec, KVendorIdStart, KVendorIdEnd);
1.144 +
1.145 + TCapabilitySet capSet;
1.146 + Input::ParseCapabilitySetL(aSpec, capSet);
1.147 +
1.148 + // Extract capabilities into array
1.149 + for (TInt i = 0 ; i < ECapability_Limit ; ++i)
1.150 + {
1.151 + TCapability c = static_cast<TCapability>(i);
1.152 + if (capSet.HasCapability(c))
1.153 + {
1.154 + User::LeaveIfError(iCapabilities.Append(c));
1.155 + }
1.156 + }
1.157 + }
1.158 +
1.159 +void CPolicyTest::SetTestActionL(const TDesC8& aPassAction, const TDesC8& aFailAction)
1.160 + {
1.161 + if (aPassAction == KNullDesC8)
1.162 + {
1.163 + BadUsageL(_L("passactions not specified"));
1.164 + }
1.165 +
1.166 + iPassAction = aPassAction.AllocL();
1.167 +
1.168 + if (aFailAction == KNullDesC8)
1.169 + {
1.170 + BadUsageL(_L("failactions not specified"));
1.171 + }
1.172 +
1.173 + iFailAction = aFailAction.AllocL();
1.174 +}
1.175 +
1.176 +void CPolicyTest::PerformAction(TRequestStatus& aStatus)
1.177 + {
1.178 + if (aStatus < 0)
1.179 + {
1.180 + iState = EFinished;
1.181 + }
1.182 +
1.183 + switch (iState)
1.184 + {
1.185 + case EInit:
1.186 + {
1.187 + TDriveUnit sysDrive (RFs::GetSystemDrive());
1.188 + TDriveName sysDriveName (sysDrive.Name());
1.189 +
1.190 + TBuf<128> scriptFile (sysDriveName);
1.191 + scriptFile.Append(KPassScriptPath);
1.192 + WriteScriptFileL(scriptFile, *iPassAction);
1.193 +
1.194 + scriptFile.Copy(sysDriveName);
1.195 + scriptFile.Append(KFailScriptPath);
1.196 + WriteScriptFileL(scriptFile, *iFailAction);
1.197 + }
1.198 + // fall through
1.199 +
1.200 + case ESetupTest:
1.201 + GetNextTest();
1.202 + if (iTestState == ETestFinished)
1.203 + {
1.204 + iState = EFinished;
1.205 + TRequestStatus* status = &aStatus;
1.206 + User::RequestComplete(status, KErrNone);
1.207 + }
1.208 + else
1.209 + {
1.210 + SetupTestL(aStatus);
1.211 + iState = ERunTest;
1.212 + }
1.213 + break;
1.214 +
1.215 + case ERunTest:
1.216 + CheckProcessTermintationL();
1.217 + RunTestL(aStatus);
1.218 + iState = EProcessResults;
1.219 + break;
1.220 +
1.221 + case EProcessResults:
1.222 + CheckProcessTermintationL();
1.223 + ProcessResultsL(aStatus);
1.224 + iState = ESetupTest;
1.225 + break;
1.226 +
1.227 + case EFinished:
1.228 + iActionState = EPostrequisite;
1.229 + TRequestStatus* status = &aStatus;
1.230 + User::RequestComplete(status, aStatus.Int());
1.231 + break;
1.232 + }
1.233 + }
1.234 +
1.235 +void CPolicyTest::StartProcessL(const TDesC& aExe, const TDesC& aCommandLine, TRequestStatus& aStatus)
1.236 + {
1.237 + iOut.writeString(_L("Starting child process: "));
1.238 + iOut.writeString(aExe);
1.239 + iOut.writeString(_L(" "));
1.240 + iOut.writeString(aCommandLine);
1.241 + iOut.writeNewLine();
1.242 +
1.243 + User::LeaveIfError(iProcess.Create(aExe, aCommandLine));
1.244 + iProcess.Logon(aStatus);
1.245 + iProcess.Resume();
1.246 + }
1.247 +
1.248 +void CPolicyTest::CheckProcessTermintationL()
1.249 + {
1.250 + if (iProcess.ExitType() == EExitPanic)
1.251 + {
1.252 + iOut.writeString(_L("Child process panicked: "));
1.253 + iOut.writeString(iProcess.ExitCategory());
1.254 + iOut.writeString(_L(" "));
1.255 + iOut.writeNum(iProcess.ExitReason());
1.256 + iOut.writeNewLine();
1.257 + User::Leave(KErrGeneral);
1.258 + }
1.259 + ASSERT(iProcess.ExitType() == EExitKill);
1.260 + iProcess.Close();
1.261 + }
1.262 +
1.263 +void CPolicyTest::GetNextTest()
1.264 + {
1.265 + // Step through capabilities to be tested
1.266 + if (iTestState == ETestFailCap)
1.267 + {
1.268 + ++iCapIndex;
1.269 + if (iCapIndex < iCapabilities.Count())
1.270 + return;
1.271 + }
1.272 +
1.273 + // Step through possible tests until we hit a vaild test
1.274 + do
1.275 + {
1.276 + iTestState = static_cast<TTestState>(iTestState + 1);
1.277 + }
1.278 + while (!((iTestState == ETestFailSID && iSecureId) ||
1.279 + (iTestState == ETestFailVID && iVendorId) ||
1.280 + (iTestState == ETestFailCap && iCapabilities.Count()) ||
1.281 + (iTestState == ETestPass) ||
1.282 + (iTestState == ETestFinished)));
1.283 + }
1.284 +
1.285 +void CPolicyTest::SetupTestL(TRequestStatus& aStatus)
1.286 + {
1.287 + switch (iTestState)
1.288 + {
1.289 + case ETestFailSID:
1.290 + {
1.291 + TInt wrongSecureId = iSecureId + 1;
1.292 + iOut.write(_L("Failure test: Wrong SID (%08x):\n\n"), wrongSecureId);
1.293 + SetTestSecurityInfoL(wrongSecureId, iVendorId, KCapabilityAll, aStatus);
1.294 + }
1.295 + break;
1.296 +
1.297 + case ETestFailVID:
1.298 + {
1.299 + TInt wrongVendorId = iVendorId + 1;
1.300 + iOut.write(_L("Failure test: Wrong VID (%08x):\n\n"), wrongVendorId);
1.301 + SetTestSecurityInfoL(iSecureId, wrongVendorId, KCapabilityAll, aStatus);
1.302 + }
1.303 + break;
1.304 +
1.305 + case ETestFailCap:
1.306 + {
1.307 + TCapability missingCap = iCapabilities[iCapIndex];
1.308 +
1.309 + iOut.writeString(_L("Failure test: Missing capability ("));
1.310 + iOut.writeCapabilityL(missingCap);
1.311 + iOut.writeString(_L("):\n\n"));
1.312 +
1.313 + TUint capSet = ~ (1 << missingCap);
1.314 + SetTestSecurityInfoL(iSecureId, iVendorId, capSet, aStatus);
1.315 + }
1.316 + break;
1.317 +
1.318 + case ETestPass:
1.319 + {
1.320 + iOut.write(_L("Pass test:\n\n"));
1.321 +
1.322 + TUint capSet = 0;
1.323 + for (TInt i = 0 ; i < iCapabilities.Count() ; ++i)
1.324 + {
1.325 + capSet |= 1 << iCapabilities[i];
1.326 + }
1.327 +
1.328 + SetTestSecurityInfoL(iSecureId, iVendorId, capSet, aStatus);
1.329 + }
1.330 + break;
1.331 +
1.332 + default:
1.333 + User::Invariant();
1.334 + }
1.335 + }
1.336 +
1.337 +void CPolicyTest::SetTestSecurityInfoL(TInt aSecureId, TInt aVendorId, TUint aCapSet, TRequestStatus& aStatus)
1.338 + {
1.339 + // Remove excluded capabilities
1.340 + for (TInt i = 0 ; i < ECapability_Limit ; ++i)
1.341 + {
1.342 + if (iExcludedCaps.HasCapability(static_cast<TCapability>(i)))
1.343 + {
1.344 + aCapSet &= ~ (1 << i);
1.345 + }
1.346 + }
1.347 +
1.348 + TBuf<128> commandLine;
1.349 + commandLine.AppendFormat(_L("%S %08x "), iTestExe, aCapSet);
1.350 + if (aSecureId)
1.351 + {
1.352 + commandLine.AppendFormat(_L("-SID %08x "), aSecureId);
1.353 + }
1.354 + if (aVendorId)
1.355 + {
1.356 + commandLine.AppendFormat(_L("-VID %08x "), aVendorId);
1.357 + }
1.358 + // commandLine.Append(KTestExeTmpPath);
1.359 + iTestExeTmpNewPath = KTestPath;
1.360 + iTestExeTmpNewPath.Append(*iTestExe);
1.361 + commandLine.Append(iTestExeTmpNewPath);
1.362 +
1.363 + TBuf<128> isetcapTmpNewPath1; //stores the value of commandline
1.364 + TBuf<128> isetcapTmpNewPath; //stores the value of KsetCapexe2
1.365 + isetcapTmpNewPath = KSetCapExe2;
1.366 + isetcapTmpNewPath1= commandLine;
1.367 + isetcapTmpNewPath.Append(isetcapTmpNewPath1);
1.368 + RDebug::RawPrint(isetcapTmpNewPath);
1.369 + StartProcessL(KSetCapExe, commandLine, aStatus);
1.370 +
1.371 + }
1.372 +
1.373 +void CPolicyTest::WriteScriptFileL(const TDesC& aPath, const TDesC8& aAction)
1.374 + {
1.375 + iFs.Delete(aPath); // ignore errors
1.376 +
1.377 + RFile file;
1.378 + User::LeaveIfError(file.Create(iFs, aPath, EFileShareExclusive | EFileWrite));
1.379 + CleanupClosePushL(file);
1.380 +
1.381 + User::LeaveIfError(file.Write(*iPreActions));
1.382 + User::LeaveIfError(file.Write(aAction));
1.383 + User::LeaveIfError(file.Write(*iPostActions));
1.384 +
1.385 + CleanupStack::PopAndDestroy(&file);
1.386 + }
1.387 +
1.388 +void CPolicyTest::RunTestL(TRequestStatus& aStatus)
1.389 + {
1.390 +
1.391 + HBufC* hptr16;
1.392 + hptr16 = CnvUtfConverter::ConvertToUnicodeFromUtf8L(*iNameInfo);
1.393 + RDebug::Print(KFormat,hptr16);
1.394 + delete hptr16;
1.395 +
1.396 + TDriveUnit sysDrive (RFs::GetSystemDrive());
1.397 + TDriveName sysDriveName (sysDrive.Name());
1.398 +
1.399 + TBuf<128> passScriptFile (sysDriveName);
1.400 + passScriptFile.Append(KPassScriptPath);
1.401 +
1.402 + TBuf<128> failScriptFile (sysDriveName);
1.403 + failScriptFile.Append(KFailScriptPath);
1.404 +
1.405 + TPtrC script = (iTestState == ETestPass) ? passScriptFile : failScriptFile;
1.406 + (iTestState == ETestPass) ? RDebug::RawPrint(KPassTestRunning) : RDebug::RawPrint(KFailTestRunning);
1.407 +
1.408 +
1.409 + TBuf<128> logTmpFile (sysDriveName);
1.410 + logTmpFile.Append(KLogTmpPath);
1.411 + iFs.Delete(logTmpFile); // ignore errors
1.412 +
1.413 + TBuf<128> commandLine;
1.414 + commandLine.AppendFormat(_L("%S %S"), &script, &logTmpFile);
1.415 +
1.416 + StartProcessL(iTestExeTmpNewPath, commandLine, aStatus);
1.417 + }
1.418 +
1.419 +void CPolicyTest::ProcessResultsL(TRequestStatus& aStatus)
1.420 + {
1.421 + _LIT8(KSummaryLine, " tests failed out of ");
1.422 + _LIT8(KNewLine, "\r\n");
1.423 +
1.424 + TInt failCount = KErrNotFound, runCount;
1.425 +
1.426 + // Read entire log file into memory to process
1.427 + RFile file;
1.428 + TDriveUnit sysDrive (RFs::GetSystemDrive());
1.429 + TBuf<128> logTmpFile (sysDrive.Name());
1.430 + logTmpFile.Append(KLogTmpPath);
1.431 + User::LeaveIfError(file.Open(iFs, logTmpFile, EFileShareReadersOnly | EFileRead));
1.432 + CleanupClosePushL(file);
1.433 +
1.434 + TInt size;
1.435 + User::LeaveIfError(file.Size(size));
1.436 + HBufC8* buffer = HBufC8::NewLC(size);
1.437 + TPtr8 ptr = buffer->Des();
1.438 +
1.439 + User::LeaveIfError(file.Read(ptr));
1.440 +
1.441 + iOut.writeString(_L("Child test output:\n"));
1.442 +
1.443 + TInt pos = 0;
1.444 + while (pos < size)
1.445 + {
1.446 + TInt nextNewline = buffer->Mid(pos).Find(KNewLine);
1.447 +
1.448 + // Split buffer into lines
1.449 + TPtrC8 line;
1.450 + if (nextNewline == KErrNotFound)
1.451 + {
1.452 + line.Set(buffer->Mid(pos));
1.453 + }
1.454 + else
1.455 + {
1.456 + line.Set(buffer->Mid(pos, nextNewline + KNewLine().Length()));
1.457 + }
1.458 + pos += line.Length();
1.459 +
1.460 + // Search for summary line
1.461 + TInt pos2 = line.Find(KSummaryLine);
1.462 + if (pos2 != KErrNotFound)
1.463 + {
1.464 + // Parse the summary line to work out if the test passed
1.465 + TLex8 lex1(line.Left(pos2));
1.466 + TInt err1 = lex1.Val(failCount);
1.467 + TLex8 lex2(line.Mid(pos2 + KSummaryLine().Length()));
1.468 + TInt err2 = lex2.Val(runCount);
1.469 +
1.470 + if (err1 != KErrNone || err2 != KErrNone)
1.471 + {
1.472 + iOut.writeString(_L("Failed to parse summary line\n"));
1.473 + User::LeaveIfError(err1);
1.474 + User::LeaveIfError(err2);
1.475 + }
1.476 + }
1.477 + else
1.478 + {
1.479 + // Don't print the summary line as this will confuse whatever parsed
1.480 + // the main log
1.481 + iOut.writeString(_L("> "));
1.482 + iOut.writeString(line);
1.483 + }
1.484 + }
1.485 +
1.486 + if (failCount == KErrNotFound)
1.487 + {
1.488 + iOut.writeString(_L("Couldn't find summary line in test output\n"));
1.489 + User::Leave(KErrNotFound);
1.490 + }
1.491 + iFailCount += failCount;
1.492 +
1.493 + // Print results in different format
1.494 + iOut.write(_L("Tests run: %d\n"), runCount);
1.495 + iOut.write(_L("Tests failed: %d\n"), failCount);
1.496 + iOut.writeNewLine();
1.497 +
1.498 + CleanupStack::PopAndDestroy(2, &file);
1.499 +
1.500 + TRequestStatus* status = &aStatus;
1.501 + User::RequestComplete(status, KErrNone);
1.502 + }
1.503 +
1.504 +void CPolicyTest::PerformCancel()
1.505 + {
1.506 + // not implemented - need to pass original status object to LogonCancel
1.507 + User::Invariant();
1.508 + /*
1.509 + switch (iState)
1.510 + {
1.511 + case ESetCapsPass:
1.512 + case ERunTest:
1.513 + iProcess.LogonCancel();
1.514 + iProcess.Kill(KErrCancel);
1.515 + iProcess.Close();
1.516 + break;
1.517 + }
1.518 + */
1.519 + }
1.520 +
1.521 +void CPolicyTest::Reset()
1.522 + {
1.523 + iProcess.Close();
1.524 + iState = ESetupTest;
1.525 + iTestState = ETestNone;
1.526 + iCapIndex = -1;
1.527 + iFailCount = 0;
1.528 + }
1.529 +
1.530 +void CPolicyTest::DoReportAction()
1.531 + {
1.532 + iOut.writeString(_L("Running policy tests...\n\n"));
1.533 + }
1.534 +
1.535 +void CPolicyTest::DoCheckResult(TInt aError)
1.536 + {
1.537 + if (aError == KErrNone && iFailCount > 0)
1.538 + {
1.539 + iOut.write(_L("%d tests failed\n"), iFailCount);
1.540 + aError = KErrGeneral;
1.541 + }
1.542 +
1.543 + iResult = (aError == iExpectedResult);
1.544 + }