1.1 --- /dev/null Thu Jan 01 00:00:00 1970 +0000
1.2 +++ b/os/persistentdata/persistentstorage/sql/TEST/t_sqlcmdlineutil.cpp Fri Jun 15 03:10:57 2012 +0200
1.3 @@ -0,0 +1,313 @@
1.4 +// Copyright (c) 2010 Nokia Corporation and/or its subsidiary(-ies).
1.5 +// All rights reserved.
1.6 +// This component and the accompanying materials are made available
1.7 +// under the terms of "Eclipse Public License v1.0"
1.8 +// which accompanies this distribution, and is available
1.9 +// at the URL "http://www.eclipse.org/legal/epl-v10.html".
1.10 +//
1.11 +// Initial Contributors:
1.12 +// Nokia Corporation - initial contribution.
1.13 +//
1.14 +// Contributors:
1.15 +//
1.16 +// Description:
1.17 +//
1.18 +#include <e32test.h>
1.19 +#include "t_sqlcmdlineutil.h"
1.20 +
1.21 +static void GetCmdLine(RTest& aTest, const TDesC& aTestName, TDes& aCmdLine)
1.22 + {
1.23 + User::CommandLine(aCmdLine);
1.24 + aCmdLine.TrimAll();
1.25 + if(aCmdLine.Length() == 0)
1.26 + {
1.27 + aTest.Printf(_L("Usage: %S [ [/enc=<16/8>] /drv=<drive letter>:] [/page=<512/1024/2048/4096/8192/16384/32768>] ] [/cache=<number>] [/hlimit=<Kb>]\r\n"), &aTestName);
1.28 + return;
1.29 + }
1.30 + aCmdLine.Append(TChar('/'));
1.31 + }
1.32 +
1.33 +static void ExtractCmdLineParams(TDes& aCmdLine, RArray<TPtrC>& aPrmNames, RArray<TPtrC>& aPrmValues)
1.34 + {
1.35 + aPrmNames.Reset();
1.36 + aPrmValues.Reset();
1.37 +
1.38 + enum TState{EWaitPrmStart, EReadPrmName, EReadPrmValue};
1.39 + TState state = EWaitPrmStart;
1.40 + TInt startPos = -1;
1.41 + TPtr prmName(0, 0);
1.42 + TPtr prmValue(0, 0);
1.43 +
1.44 + aCmdLine.Append(TChar('/'));
1.45 +
1.46 + for(TInt i=0;i<aCmdLine.Length();++i)
1.47 + {
1.48 + switch(state)
1.49 + {
1.50 + case EWaitPrmStart:
1.51 + if(aCmdLine[i] == TChar('/'))
1.52 + {
1.53 + startPos = i + 1;
1.54 + prmName.Zero();
1.55 + state = EReadPrmName;
1.56 + }
1.57 + break;
1.58 + case EReadPrmName:
1.59 + if(aCmdLine[i] == TChar('='))
1.60 + {
1.61 + TPtr p = aCmdLine.MidTPtr(startPos, i - startPos);
1.62 + prmName.Set(p);
1.63 + prmName.TrimRight();
1.64 + startPos = i + 1;
1.65 + prmValue.Zero();
1.66 + state = EReadPrmValue;
1.67 + }
1.68 + break;
1.69 + case EReadPrmValue:
1.70 + if(aCmdLine[i] == TChar('/'))
1.71 + {
1.72 + TPtr p = aCmdLine.MidTPtr(startPos, i - startPos);
1.73 + prmValue.Set(p);
1.74 + prmValue.Trim();
1.75 + startPos = i + 1;
1.76 + aPrmNames.Append(prmName);
1.77 + aPrmValues.Append(prmValue);
1.78 + prmName.Zero();
1.79 + prmValue.Zero();
1.80 + state = EReadPrmName;
1.81 + }
1.82 + break;
1.83 + default:
1.84 + break;
1.85 + }
1.86 + }
1.87 + }
1.88 +
1.89 +static void ExtractParamNamesAndValues(const RArray<TPtrC>& aPrmNames, const RArray<TPtrC>& aPrmValues, TCmdLineParams& aCmdLineParams)
1.90 + {
1.91 + __ASSERT_ALWAYS(aPrmNames.Count() == aPrmValues.Count(), User::Invariant());
1.92 +
1.93 + aCmdLineParams.SetDefaults();
1.94 +
1.95 + for(TInt i=0;i<aPrmNames.Count();++i)
1.96 + {
1.97 + if(aPrmNames[i].CompareF(_L("enc")) == 0)
1.98 + {
1.99 + TLex lex(aPrmValues[i]);
1.100 + TInt enc = 0;
1.101 + TInt err = lex.Val(enc);
1.102 + if(err == KErrNone)
1.103 + {
1.104 + if(enc == 8)
1.105 + {
1.106 + aCmdLineParams.iDbEncoding = TCmdLineParams::EDbUtf8;
1.107 + }
1.108 + else if(enc == 16)
1.109 + {
1.110 + aCmdLineParams.iDbEncoding = TCmdLineParams::EDbUtf16;
1.111 + }
1.112 + }
1.113 + }
1.114 + else if(aPrmNames[i].CompareF(_L("drv")) == 0)
1.115 + {
1.116 + if(aPrmValues[i].Length() == 2 && aPrmValues[i][1] == TChar(':'))
1.117 + {
1.118 + TChar ch(aPrmValues[i][0]);
1.119 + ch.LowerCase();
1.120 + if(ch >= TChar('a') && ch <= TChar('z'))
1.121 + aCmdLineParams.iDriveName.Copy(aPrmValues[i]);
1.122 + }
1.123 + }
1.124 + else if(aPrmNames[i].CompareF(_L("page")) == 0)
1.125 + {
1.126 + TLex lex(aPrmValues[i]);
1.127 + TInt pageSize = 0;
1.128 + TInt err = lex.Val(pageSize);
1.129 + if(err == KErrNone && (pageSize == 512 || pageSize == 1024 || pageSize == 2048 ||
1.130 + pageSize == 4096 || pageSize == 8192 || pageSize == 16384 || pageSize == 32768))
1.131 + {
1.132 + aCmdLineParams.iPageSize = pageSize;
1.133 + }
1.134 + }
1.135 + else if(aPrmNames[i].CompareF(_L("cache")) == 0)
1.136 + {
1.137 + TLex lex(aPrmValues[i]);
1.138 + TInt cacheSize = 0;
1.139 + TInt err = lex.Val(cacheSize);
1.140 + if(err == KErrNone && (cacheSize > 0 && cacheSize < 1000000000))
1.141 + {
1.142 + aCmdLineParams.iCacheSize = cacheSize;
1.143 + }
1.144 + }
1.145 + else if(aPrmNames[i].CompareF(_L("hlimit")) == 0)
1.146 + {
1.147 + TLex lex(aPrmValues[i]);
1.148 + TInt softHeapLimit = 0;
1.149 + TInt err = lex.Val(softHeapLimit);
1.150 + if(err == KErrNone && (softHeapLimit >= 0 && softHeapLimit < 1000000000))
1.151 + {
1.152 + aCmdLineParams.iSoftHeapLimitKb = softHeapLimit;
1.153 + }
1.154 + }
1.155 + }
1.156 + }
1.157 +
1.158 +static void PrepareSqlConfigString(RTest& aTest, const TCmdLineParams& aCmdLineParams, TDes8& aConfigStr)
1.159 + {
1.160 + aConfigStr.Zero();
1.161 +
1.162 + if(aCmdLineParams.iDbEncoding == TCmdLineParams::EDbUtf8)
1.163 + {
1.164 + aTest.Printf(_L("--PRM--Database Encoding: UTF8\r\n"));
1.165 + aConfigStr.Append(_L8("encoding=\"UTF-8\";"));
1.166 + }
1.167 + else
1.168 + {
1.169 + aTest.Printf(_L("--PRM--Database Encoding: UTF16\r\n"));
1.170 + aConfigStr.Append(_L8("encoding=\"UTF-16\";"));
1.171 + }
1.172 +
1.173 + aTest.Printf(_L("--PRM--Database page size: %d\r\n"), aCmdLineParams.iPageSize);
1.174 + TBuf8<20> pageSizeBuf;
1.175 + pageSizeBuf.Format(_L8("page_size=%d;"), aCmdLineParams.iPageSize);
1.176 + aConfigStr.Append(pageSizeBuf);
1.177 +
1.178 + aTest.Printf(_L("--PRM--Database cache size: %d\r\n"), aCmdLineParams.iCacheSize);
1.179 + TBuf8<20> cacheSizeBuf;
1.180 + cacheSizeBuf.Format(_L8("cache_size=%d;"), aCmdLineParams.iCacheSize);
1.181 + aConfigStr.Append(cacheSizeBuf);
1.182 +
1.183 + aTest.Printf(_L("--PRM--Database drive: %S\r\n"), &aCmdLineParams.iDriveName);
1.184 +
1.185 + if(aCmdLineParams.iSoftHeapLimitKb > 0)
1.186 + {
1.187 + aTest.Printf(_L("--PRM--Soft heap limit: %d Kb\r\n"), aCmdLineParams.iSoftHeapLimitKb);
1.188 + }
1.189 + else
1.190 + {
1.191 + aTest.Printf(_L("--PRM--Soft heap limit: default\r\n"));
1.192 + }
1.193 + }
1.194 +
1.195 +#ifdef SQL_SOFT_HEAP_LIMIT_TEST
1.196 +
1.197 +static TInt KillProcess(const TDesC& aProcessName)
1.198 + {
1.199 + TFullName name;
1.200 + TBuf<64> pattern(aProcessName);
1.201 + TInt length = pattern.Length();
1.202 + pattern += _L("*");
1.203 + TFindProcess procFinder(pattern);
1.204 +
1.205 + while (procFinder.Next(name) == KErrNone)
1.206 + {
1.207 + if (name.Length() > length)
1.208 + {//If found name is a string containing aProcessName string.
1.209 + TChar c(name[length]);
1.210 + if (c.IsAlphaDigit() ||
1.211 + c == TChar('_') ||
1.212 + c == TChar('-'))
1.213 + {
1.214 + // If the found name is other valid application name
1.215 + // starting with aProcessName string.
1.216 + continue;
1.217 + }
1.218 + }
1.219 + RProcess proc;
1.220 + if (proc.Open(name) == KErrNone)
1.221 + {
1.222 + proc.Kill(0);
1.223 + }
1.224 + proc.Close();
1.225 + }
1.226 + return KErrNone;
1.227 + }
1.228 +
1.229 +_LIT(KSqlSrvName, "sqlsrv.exe");
1.230 +_LIT(KSqlSrvConfigFile, "c:\\test\\t_sqlserver.cfg");
1.231 +
1.232 +static void ReplaceConfigFile(const TDesC16& aConfig)
1.233 + {
1.234 + RFs fs;
1.235 + TInt err = fs.Connect();
1.236 + __ASSERT_ALWAYS(err == KErrNone, User::Invariant());
1.237 +
1.238 + (void)KillProcess(KSqlSrvName);
1.239 +
1.240 + (void)fs.MkDirAll(KSqlSrvConfigFile);
1.241 + (void)fs.Delete(KSqlSrvConfigFile);
1.242 +
1.243 + RFile file;
1.244 + err = file.Create(fs, KSqlSrvConfigFile, EFileRead | EFileWrite);
1.245 + __ASSERT_ALWAYS(err == KErrNone, User::Invariant());
1.246 +
1.247 + TPtrC8 p((const TUint8*)aConfig.Ptr(), aConfig.Length() * sizeof(TUint16));
1.248 + err = file.Write(p);
1.249 + file.Close();
1.250 + __ASSERT_ALWAYS(err == KErrNone, User::Invariant());
1.251 +
1.252 + fs.Close();
1.253 + }
1.254 +
1.255 +static void DeleteConfigFile()
1.256 + {
1.257 + RFs fs;
1.258 + TInt err = fs.Connect();
1.259 + __ASSERT_ALWAYS(err == KErrNone, User::Invariant());
1.260 +
1.261 + (void)KillProcess(KSqlSrvName);
1.262 +
1.263 + (void)fs.MkDirAll(KSqlSrvConfigFile);
1.264 + (void)fs.Delete(KSqlSrvConfigFile);
1.265 +
1.266 + fs.Close();
1.267 + }
1.268 +
1.269 +#endif //SQL_SOFT_HEAP_LIMIT_TEST
1.270 +
1.271 +void GetCmdLineParamsAndSqlConfigString(RTest& aTest, const TDesC& aTestName, TCmdLineParams& aCmdLineParams, TDes8& aConfigStr)
1.272 + {
1.273 + TBuf<200> cmdLine;
1.274 + GetCmdLine(aTest, aTestName, cmdLine);
1.275 + RArray<TPtrC> prmNames;
1.276 + RArray<TPtrC> prmValues;
1.277 + ExtractCmdLineParams(cmdLine, prmNames, prmValues);
1.278 + ExtractParamNamesAndValues(prmNames, prmValues, aCmdLineParams);
1.279 + prmValues.Close();
1.280 + prmNames.Close();
1.281 + PrepareSqlConfigString(aTest, aCmdLineParams, aConfigStr);
1.282 + }
1.283 +
1.284 +void PrepareDbName(const TDesC& aDeafultDbName, const TDriveName& aDriveName, TDes& aDbName)
1.285 + {
1.286 + TParse parse;
1.287 + parse.Set(aDriveName, &aDeafultDbName, 0);
1.288 + const TDesC& dbFilePath = parse.FullName();
1.289 + aDbName.Copy(dbFilePath);
1.290 + }
1.291 +
1.292 +void SetSoftHeapLimit(TInt aSoftHeapLimit)
1.293 + {
1.294 + if(aSoftHeapLimit > 0)
1.295 + {
1.296 +#ifdef SQL_SOFT_HEAP_LIMIT_TEST
1.297 + TBuf<50> configBuf;
1.298 + configBuf.Format(_L("soft_heap_limit_kb=%d"), aSoftHeapLimit);
1.299 + ReplaceConfigFile(configBuf);
1.300 +#else
1.301 + RDebug::Print(_L("The soft heap limit cannot be set if \"SQL_SOFT_HEAP_LIMIT_TEST\" macro is not defined!\r\n"));
1.302 +#endif
1.303 + }
1.304 + else if(aSoftHeapLimit < 0)
1.305 + {
1.306 + RDebug::Print(_L("Soft heap limit of %d Kb cannot be set!\r\n"), aSoftHeapLimit);
1.307 + }
1.308 + }
1.309 +
1.310 +void ResetSoftHeapLimit()
1.311 + {
1.312 +#ifdef SQL_SOFT_HEAP_LIMIT_TEST
1.313 + DeleteConfigFile();
1.314 +#endif
1.315 + }
1.316 +