|
|
/*++
Copyright (c) 1998-2000 Microsoft Corporation
Module Name : IniFile.cpp
Abstract: Test harness for LKRhash
Author: George V. Reilly (GeorgeRe) 06-Jan-1998
Environment: Win32 - User Mode
Project: Internet Information Server RunTime Library
Revision History:
--*/
#include "precomp.hxx"
#include "WordHash.h"
#include "IniFile.h"
enum INI_TYPE { INI_WORD = 1, INI_DWORD, INI_DOUBLE, INI_STRING, INI_BOOL,};
typedef struct _ParseOptions { int m_nFieldOffset; const char* m_pszName; unsigned m_cchName; INI_TYPE m_type; DWORD_PTR m_default;} ParseOptions;
#define INI_ENTRY(_member, _name, _type, _default) \
{ \ FIELD_OFFSET(CIniFileSettings, _member), \ _name, \ sizeof(_name)-1, \ _type, \ (DWORD_PTR) _default, \ } \
static const ParseOptions g_po[] = { INI_ENTRY(m_tszDataFile, "DataFile", INI_STRING, _TEXT("??")), INI_ENTRY(m_nMaxKeys, "MaxKeys", INI_DWORD, MAXKEYS), INI_ENTRY(m_dblHighLoad, "MaxLoadFactor", INI_DOUBLE, LK_DFLT_MAXLOAD), INI_ENTRY(m_nInitSize, "InitSize", INI_DWORD, LK_DFLT_INITSIZE), INI_ENTRY(m_nSubTables, "NumSubTables", INI_DWORD,LK_DFLT_NUM_SUBTBLS), INI_ENTRY(m_nLookupFreq, "LookupFrequency",INI_DWORD, 5), INI_ENTRY(m_nMinThreads, "MinThreads", INI_DWORD, 1), INI_ENTRY(m_nMaxThreads, "MaxThreads", INI_DWORD, 4), INI_ENTRY(m_nRounds, "NumRounds", INI_DWORD, 1), INI_ENTRY(m_nSeed, "RandomSeed", INI_DWORD, 1234), INI_ENTRY(m_fCaseInsensitive,"CaseInsensitive",INI_BOOL, FALSE), INI_ENTRY(m_fMemCmp, "MemCmp", INI_BOOL, FALSE), INI_ENTRY(m_nLastChars, "NumLastChars", INI_DWORD, 0), INI_ENTRY(m_wTableSpin, "TableLockSpinCount",INI_WORD, LOCK_DEFAULT_SPINS), INI_ENTRY(m_wBucketSpin, "BucketLockSpinCount",INI_WORD,LOCK_DEFAULT_SPINS), INI_ENTRY(m_dblSpinAdjFctr,"SpinAdjustmentFactor",INI_DOUBLE, 1), INI_ENTRY(m_fTestIterators,"TestIterators", INI_BOOL, FALSE), INI_ENTRY(m_nInsertIfNotFound, "InsertIfNotFound",INI_DWORD, 0), INI_ENTRY(m_nFindKeyCopy, "FindKeyCopy", INI_DWORD, 0), INI_ENTRY(m_fNonPagedAllocs,"NonPagedAllocs", INI_BOOL, TRUE), INI_ENTRY(m_fRefTrace, "RefTrace", INI_BOOL, FALSE), {-1} // last entry
};
voidCIniFileSettings::Dump( LPCTSTR ptszProlog, LPCTSTR ptszEpilog) const{ TCHAR tsz[50];
_tprintf(_TEXT("%s\n"), ptszProlog); _tprintf(_TEXT("IniFile=\"%s\"\n"), m_tszIniFile); _tprintf(_TEXT("DataFile=\"%s\". %s keys.\n"), m_tszDataFile, CommaNumber(m_nMaxKeys, tsz)); _tprintf(_TEXT("Max load = %.1f, initsize = %d, %d subtables.\n"), m_dblHighLoad, m_nInitSize, m_nSubTables); _tprintf(_TEXT("Lookup freq = %d, %d-%d threads, %d round%s.\n"), m_nLookupFreq, m_nMinThreads, m_nMaxThreads, m_nRounds, (m_nRounds==1 ? "" : "s")); _tprintf(_TEXT("Seed=%d, CaseInsensitive=%d, MemCmp=%d, LastChars=%d\n"), m_nSeed, m_fCaseInsensitive, m_fMemCmp, m_nLastChars); _tprintf(_TEXT("Spin Count: Table = %hd, Bucket = %hd, AdjFactor=%.1f\n"), m_wTableSpin, m_wBucketSpin, m_dblSpinAdjFctr); _tprintf(_TEXT("TestIterators=%d, InsertIfNotFound=%d, FindKeyCopy=%d\n"), m_fTestIterators, m_nInsertIfNotFound, m_nFindKeyCopy); _tprintf(_TEXT("NonPagedAllocs=%d, RefTrace=%d\n"), m_fNonPagedAllocs, m_fRefTrace); _tprintf(_TEXT("%s\n"), ptszEpilog);}
#if defined(IRTLDEBUG)
# define DUMP_INIFILE(pifs, Pro, Epi) pifs->Dump(Pro, Epi)
#else
# define DUMP_INIFILE(pifs, Pro, Epi) ((void) 0)
#endif
DWORDReadFileIntoBuffer( LPCTSTR ptszFile, PBYTE pbBuffer, DWORD cbBuffer){#ifndef LKRHASH_KERNEL_MODE
HANDLE hFile = CreateFile( ptszFile, GENERIC_READ, 0, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
if (hFile == INVALID_HANDLE_VALUE) return 0;
DWORD cbRead = 0, cbFileSizeLow, cbFileSizeHigh;
cbFileSizeLow = GetFileSize(hFile, &cbFileSizeHigh);
bool fBadFile = (cbFileSizeHigh != 0 || cbFileSizeLow > cbBuffer || !ReadFile(hFile, pbBuffer, cbFileSizeLow, &cbRead, NULL));
CloseHandle(hFile);
return fBadFile ? 0 : cbRead;#else
return 0;#endif // !LKRHASH_KERNEL_MODE
}
// Do a case-insensitive match of first `cchStr' chars of ptszBuffer
// against ptszStr. Strings assumed to be alphanumeric
boolStrMatch( LPCSTR pszBuffer, LPCSTR pszStr, unsigned cchStr){ LPCSTR psz1 = pszBuffer; LPCSTR psz2 = pszStr; unsigned i, j; bool fMatch = true;
for (i = 0; i < cchStr; ++i) { IRTLASSERT(isalnum(*psz1) && isalnum(*psz2));
if (((*psz1++ ^ *psz2++) & 0xDF) != 0) { fMatch = false; break; } }
IRTLTRACE0("\tStrMatch: \""); for (j = 0; j < i + !fMatch; ++j) IRTLTRACE1("%hc", pszBuffer[j]); IRTLTRACE0(fMatch ? "\" == \"" : "\" != \""); for (j = 0; j < i + !fMatch; ++j) IRTLTRACE1("%hc", pszStr[j]); IRTLTRACE0("\"\n");
return fMatch;}
boolGetNum( char*& rpch, int& rn){ int fNegative = ('-' == *rpch); rn = 0;
if (fNegative) ++rpch; else if (!('0' <= *rpch && *rpch <= '9')) return false;
while ('0' <= *rpch && *rpch <= '9') rn = rn * 10 + (*rpch++ - '0');
if (fNegative) rn = -rn;
return true;}
boolGetDouble( char*& rpch, double& rdbl){ rdbl = 0;
int n; bool fValidInt = GetNum(rpch, n);
if (fValidInt) { rdbl = n;
// BUGBUG: ignore fractional part, if any
if ('.' == *rpch) { ++rpch; GetNum(rpch, n); } }
return fValidInt;}
boolGetString( char*& rpch, char* pszOutput, unsigned cchOutput){ // TODO: handle quoted strings and trailing blanks
bool fGotChars = false; while ('\0' != *rpch && '\r' != *rpch && '\n' != *rpch ) { fGotChars = true;
if (cchOutput-- > 0) *pszOutput++ = *rpch++; else ++rpch; }
if (cchOutput > 0) *pszOutput = '\0';
return fGotChars;}
// TODO: break the dependency upon g_po.
intCIniFileSettings::ParseIniFile( LPCSTR pszIniFile){ strncpy(m_tszIniFile, pszIniFile, _MAX_PATH);
int i, iMaxIndex = -1; int cMembers = 0;
for (i = 0; ; ++i) { if (g_po[i].m_nFieldOffset < 0) { iMaxIndex = i; break; }
PBYTE pbMember = g_po[i].m_nFieldOffset + ((BYTE*) this);
// Initialize the members of `this' with their default values
switch (g_po[i].m_type) { case INI_WORD: * (WORD*) pbMember = (WORD) g_po[i].m_default; break; case INI_DWORD: * (DWORD*) pbMember = (DWORD) g_po[i].m_default; break; case INI_DOUBLE: * (double*) pbMember = (float) g_po[i].m_default; break; case INI_STRING: strcpy((char*) pbMember, (const char*) g_po[i].m_default); break; case INI_BOOL: * (bool*) pbMember = (bool) (g_po[i].m_default != 0); break; default: IRTLASSERT(! "invalid INI_TYPE"); } }
DUMP_INIFILE(this, "Before", "");
BYTE abBuffer[2049]; DWORD cbRead = ReadFileIntoBuffer(m_tszIniFile, abBuffer, sizeof(abBuffer)-1);
if (cbRead == 0) { _tprintf(_TEXT("Can't open IniFile `%s'.\n"), m_tszIniFile) ; return 0; }
abBuffer[cbRead] = '\0';
bool fInSection = false, fSkipLine = false; bool fDone = false; const char szSectionName[] = "HashTest"; unsigned cchSectionName = strlen(szSectionName); char* pch = (char*) abBuffer; char* pszEOB = (char*) (abBuffer + cbRead);
// parse the in-memory buffer
while ('\0' != *pch) { while (' ' == *pch || '\r' == *pch || '\n' == *pch || '\t' == *pch) ++pch;
if ('\0' == *pch) break;
IRTLTRACE(_TEXT("Line starts with '%hc%hc%hc%hc'\n"), pch[0], pch[1], pch[2], pch[3]);
// Is this a section name?
if ('[' == *pch) { fInSection = false; ++pch; while (' ' == *pch || '\t' == *pch) ++pch; if (pch + cchSectionName < pszEOB && StrMatch(pch, szSectionName, cchSectionName)) { pch += cchSectionName; while (' ' == *pch || '\t' == *pch) ++pch; if (']' == *pch) { ++pch; fInSection = true; } } else fSkipLine = true;
continue; }
// skip comments and entire lines if we're not in the right section
if (fSkipLine || ';' == *pch || !fInSection) { // skip to end of line
while ('\0' != *pch && '\r' != *pch && '\n' != *pch) { IRTLTRACE1("%hc", *pch); ++pch; }
IRTLTRACE0("\n"); fSkipLine = false; continue; }
fSkipLine = true;
// try to match name=value
for (i = 0; i < iMaxIndex; ++i) { IRTLASSERT(isalnum(*pch)); if (pch + g_po[i].m_cchName >= pszEOB || !StrMatch(pch, g_po[i].m_pszName, g_po[i].m_cchName)) continue;
pch += g_po[i].m_cchName; while (' ' == *pch || '\t' == *pch) ++pch; if ('=' != *pch) { IRTLTRACE1("'=' not seen after <%hs>\n", g_po[i].m_pszName); break; } ++pch; while (' ' == *pch || '\t' == *pch) ++pch;
PBYTE pbMember = g_po[i].m_nFieldOffset + ((BYTE*) this); int n; char sz[_MAX_PATH]; double dbl;
IRTLTRACE1("<%hs>=", g_po[i].m_pszName);
switch (g_po[i].m_type) { case INI_WORD: if (GetNum(pch, n)) { IRTLTRACE1("%hu\n", (WORD) n); * (WORD*) pbMember = (WORD) n; } else IRTLTRACE("bad word\n"); break; case INI_DWORD: if (GetNum(pch, n)) { IRTLTRACE1("%u\n", (DWORD) n); * (DWORD*) pbMember = (DWORD) n; } else IRTLTRACE("bad dword\n"); break; case INI_DOUBLE: if (GetDouble(pch, dbl)) { IRTLTRACE1("%.1f\n", dbl); * (double*) pbMember = dbl; } else IRTLTRACE("bad double\n"); break; case INI_STRING: if (GetString(pch, sz, sizeof(sz)/sizeof(sz[0]))) { IRTLTRACE1("%hs\n", sz); strcpy((char*) pbMember, sz); } else IRTLTRACE("bad string\n"); break; case INI_BOOL: if (GetNum(pch, n)) { IRTLTRACE1("%d\n", n); * (bool*) pbMember = (n != 0); } else IRTLTRACE("bad bool\n"); break; default: IRTLASSERT(! "invalid INI_TYPE"); }
++cMembers; fSkipLine = false; break; } }
DUMP_INIFILE(this, "Parsed", "---");
return cMembers;}
voidCIniFileSettings::ReadIniFile( LPCTSTR ptszIniFile){ ParseIniFile(ptszIniFile);
m_nMaxKeys = min(max(1, m_nMaxKeys), MAXKEYS); m_dblHighLoad = max(1, m_dblHighLoad); m_nMinThreads = max(1, m_nMinThreads); m_nMaxThreads = min(MAX_THREADS, max(1, m_nMaxThreads));
// If we're not using a real lock, then we're not threadsafe
if (CWordHash::TableLock::LockType() == LOCK_FAKELOCK || CWordHash::BucketLock::LockType() == LOCK_FAKELOCK) m_nMinThreads = m_nMaxThreads = 1;
m_nRounds = max(1, m_nRounds); CWordHash::sm_fCaseInsensitive = m_fCaseInsensitive; CWordHash::sm_fMemCmp = m_fMemCmp; CWordHash::sm_nLastChars = m_nLastChars; CWordHash::sm_fNonPagedAllocs = m_fNonPagedAllocs; CWordHash::sm_fRefTrace = m_fRefTrace;
#ifdef LOCK_DEFAULT_SPIN_IMPLEMENTATION
# ifdef LKRHASH_GLOBAL_LOCK
CWordHash::GlobalLock::SetDefaultSpinAdjustmentFactor(m_dblSpinAdjFctr);# endif
CWordHash::TableLock::SetDefaultSpinAdjustmentFactor(m_dblSpinAdjFctr); CWordHash::BucketLock::SetDefaultSpinAdjustmentFactor(m_dblSpinAdjFctr);#endif // LOCK_DEFAULT_SPIN_IMPLEMENTATION
if (CWordHash::TableLock::Recursion() != LOCK_RECURSIVE || CWordHash::BucketLock::LockType() != LOCK_RECURSIVE) m_nInsertIfNotFound = 0;}
|
