Leaked source code of windows server 2003
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 

761 lines
20 KiB

//---------------------------------------------------------------------------
//
// Copyright (c) Microsoft Corporation 1993-1994
//
// File: atoms.c
//
// This files contains the atom list code.
//
// History:
// 01-31-94 ScottH Moved from cache.c
//
//---------------------------------------------------------------------------
///////////////////////////////////////////////////// INCLUDES
#include "brfprv.h" // common headers
///////////////////////////////////////////////////// TYPEDEFS
typedef struct tagA_ITEM
{
int atom; // index into hdsa
LPTSTR psz; // allocated
UINT ucRef;
} A_ITEM; // item for atom table
typedef struct tagATOMTABLE
{
CRITICAL_SECTION cs;
HDSA hdsa; // Actual list of A_ITEMs
HDPA hdpa; // List into hdsa (sorted). Values are indexes, not pointers
HDPA hdpaFree; // Free list. Values are indexes, not pointers.
} ATOMTABLE;
#define Atom_EnterCS(this) EnterCriticalSection(&(this)->cs)
#define Atom_LeaveCS(this) LeaveCriticalSection(&(this)->cs)
#define ATOM_GROW 32
#define Cache_Bogus(this) (!(this)->hdpa || !(this)->hdpaFree || !(this)->hdsa)
// Given an index into the DPA, get the pointer to the DSA
//
#define MyGetPtr(this, idpa) DSA_GetItemPtr((this)->hdsa, PtrToUlong(DPA_FastGetPtr((this)->hdpa, idpa)))
///////////////////////////////////////////////////// MODULE DATA
static ATOMTABLE s_atomtable;
#ifdef DEBUG
/*----------------------------------------------------------
Purpose: Validates the given atom is within the atomtable's range
Returns: --
Cond: --
*/
void PUBLIC Atom_ValidateFn(
int atom)
{
ATOMTABLE * this = &s_atomtable;
BOOL bError = FALSE;
Atom_EnterCS(this);
{
if (atom >= DSA_GetItemCount(this->hdsa) ||
atom < 0)
{
bError = TRUE;
}
}
Atom_LeaveCS(this);
if (bError)
{
// This is a problem!
//
DEBUG_MSG(TF_ERROR, TEXT("err BRIEFCASE: atom %d is out of range!"), atom);
DEBUG_BREAK(BF_ONVALIDATE);
}
}
/*----------------------------------------------------------
Purpose: Dump the table contents
Returns: --
Cond: For debugging purposes
*/
void PUBLIC Atom_DumpAll()
{
ATOMTABLE * this = &s_atomtable;
Atom_EnterCS(this);
{
if (IsFlagSet(g_uDumpFlags, DF_ATOMS))
{
A_ITEM * pitem;
int idpa;
int cItem;
ASSERT(this);
ASSERT(this->hdsa != NULL);
cItem = DPA_GetPtrCount(this->hdpa);
for (idpa = 0; idpa < cItem; idpa++)
{
pitem = MyGetPtr(this, idpa);
// The zero'th entry is reserved, so skip it
if (pitem->atom == 0)
continue;
TRACE_MSG(TF_ALWAYS, TEXT("ATOM: Atom %d [%u]: %s"),
pitem->atom, pitem->ucRef, pitem->psz);
}
}
}
Atom_LeaveCS(this);
}
#endif
/*----------------------------------------------------------
Purpose: Compare A_ITEMs
Returns: -1 if <, 0 if ==, 1 if >
Cond: --
*/
int CALLBACK _export Atom_CompareIndexes(
LPVOID lpv1,
LPVOID lpv2,
LPARAM lParam)
{
int i1 = PtrToUlong(lpv1);
int i2 = PtrToUlong(lpv2);
HDSA hdsa = (HDSA)lParam;
A_ITEM * pitem1 = DSA_GetItemPtr(hdsa, i1);
A_ITEM * pitem2 = DSA_GetItemPtr(hdsa, i2);
ASSERT(pitem1);
ASSERT(pitem2);
return lstrcmpi(pitem1->psz, pitem2->psz);
}
/*----------------------------------------------------------
Purpose: Compare A_ITEMs
Returns: -1 if <, 0 if ==, 1 if >
Cond: --
*/
int CALLBACK _export Atom_Compare(
LPVOID lpv1,
LPVOID lpv2,
LPARAM lParam)
{
// HACK: we know the first param is the address to a struct
// that contains the search criteria. The second is an index
// into the DSA.
//
int i2 = PtrToUlong(lpv2);
HDSA hdsa = (HDSA)lParam;
A_ITEM * pitem1 = (A_ITEM *)lpv1;
A_ITEM * pitem2 = DSA_GetItemPtr(hdsa, i2);
ASSERT(pitem1);
ASSERT(pitem2);
return lstrcmpi(pitem1->psz, pitem2->psz);
}
/*----------------------------------------------------------
Purpose: Initialize the atom table
Returns: TRUE on success
Cond: --
*/
BOOL PUBLIC Atom_Init()
{
BOOL bRet;
ATOMTABLE * this = &s_atomtable;
ASSERT(this);
ZeroInit(this, ATOMTABLE);
bRet = InitializeCriticalSectionAndSpinCount(&this->cs, 0);
if (bRet)
{
Atom_EnterCS(this);
{
if ((this->hdsa = DSA_Create(sizeof(A_ITEM), ATOM_GROW)) != NULL)
{
if ((this->hdpa = DPA_Create(ATOM_GROW)) == NULL)
{
DSA_Destroy(this->hdsa);
this->hdsa = NULL;
}
else
{
if ((this->hdpaFree = DPA_Create(ATOM_GROW)) == NULL)
{
DPA_Destroy(this->hdpa);
DSA_Destroy(this->hdsa);
this->hdpa = NULL;
this->hdsa = NULL;
}
else
{
// We've successfully initialized. Keep the zero'th
// atom reserved. This way null atoms will not accidentally
// munge data.
//
int atom = Atom_Add(TEXT("SHDD"));
ASSERT(atom == 0);
}
}
}
bRet = this->hdsa != NULL;
}
Atom_LeaveCS(this);
}
return bRet;
}
/*----------------------------------------------------------
Purpose: Destroy the atom table
Returns: --
Cond: --
*/
void PUBLIC Atom_Term()
{
ATOMTABLE * this = &s_atomtable;
Atom_EnterCS(this);
{
if (this->hdpa != NULL)
{
A_ITEM * pitem;
int idpa;
int cItem;
ASSERT(this->hdsa != NULL);
cItem = DPA_GetPtrCount(this->hdpa);
for (idpa = 0; idpa < cItem; idpa++)
{
pitem = MyGetPtr(this, idpa);
// The zero'th entry is reserved, so skip it
if (pitem->atom == 0)
continue;
Str_SetPtr(&pitem->psz, NULL);
}
DPA_Destroy(this->hdpa);
this->hdpa = NULL;
}
if (this->hdpaFree != NULL)
{
DPA_Destroy(this->hdpaFree);
this->hdpaFree = NULL;
}
if (this->hdsa != NULL)
{
DSA_Destroy(this->hdsa);
this->hdsa = NULL;
}
}
Atom_LeaveCS(this);
DeleteCriticalSection(&this->cs);
}
/*----------------------------------------------------------
Purpose: Add a string to the atom table. If the string already
exists, return its atom.
Returns: Atom
ATOM_ERR on failure
Cond: Reference count is incremented always
*/
int PUBLIC Atom_Add(
LPCTSTR psz)
{
ATOMTABLE * this = &s_atomtable;
A_ITEM * pitem = NULL;
A_ITEM item;
int atomRet = ATOM_ERR;
int idpa;
int cItem;
int cFree;
ASSERT(psz);
Atom_EnterCS(this);
{
int iItem;
DEBUG_CODE( iItem = -1; )
// Search for the string in the atom table first.
// If we find it, return the atom.
//
item.psz = (LPTSTR)(LPVOID)psz;
idpa = DPA_Search(this->hdpa, &item, 0, Atom_Compare, (LPARAM)this->hdsa, DPAS_SORTED);
if (idpa != -1)
{
// String is already in table
//
pitem = MyGetPtr(this, idpa);
pitem->ucRef++;
atomRet = pitem->atom;
ASSERT(IsSzEqual(psz, pitem->psz));
VALIDATE_ATOM(pitem->atom);
}
else
{
// Add the string to the table. Take any available entry
// from the free list first. Otherwise allocate more space
// in the table. Then add a ptr to the sorted ptr list.
//
cFree = DPA_GetPtrCount(this->hdpaFree);
if (cFree > 0)
{
// Use a free entry
//
cFree--;
iItem = PtrToUlong(DPA_DeletePtr(this->hdpaFree, cFree));
pitem = DSA_GetItemPtr(this->hdsa, iItem);
// atom field for pitem should already be set
VALIDATE_ATOM(pitem->atom);
}
else
{
// Allocate a new entry. item has bogus data in it.
// That's okay, we fill in good stuff below.
//
cItem = DSA_GetItemCount(this->hdsa);
if ((iItem = DSA_InsertItem(this->hdsa, cItem+1, &item)) != -1)
{
pitem = DSA_GetItemPtr(this->hdsa, iItem);
pitem->atom = iItem;
VALIDATE_ATOM(pitem->atom);
}
}
// Fill in the info
//
if (pitem)
{
pitem->ucRef = 1;
pitem->psz = 0;
if (!Str_SetPtr(&pitem->psz, psz))
goto Add_Fail;
// Add the new entry to the ptr list and sort
//
cItem = DPA_GetPtrCount(this->hdpa);
if (DPA_InsertPtr(this->hdpa, cItem+1, IntToPtr(iItem)) == -1)
goto Add_Fail;
DPA_Sort(this->hdpa, Atom_CompareIndexes, (LPARAM)this->hdsa);
atomRet = pitem->atom;
TRACE_MSG(TF_ATOM, TEXT("ATOM Adding %d [%u]: %s"), atomRet, pitem->ucRef, pitem->psz);
}
}
Add_Fail:
// Add the entry to the free list and fail. If even this fails,
// then we simply lose some slight efficiency, but this is not
// a memory leak.
//
#ifdef DEBUG
if (atomRet == ATOM_ERR)
TRACE_MSG(TF_ATOM, TEXT("ATOM **Failed adding %s"), psz);
#endif
if (atomRet == ATOM_ERR && pitem)
{
ASSERT(iItem != -1);
DPA_InsertPtr(this->hdpaFree, cFree+1, IntToPtr(iItem));
}
}
Atom_LeaveCS(this);
return atomRet;
}
/*----------------------------------------------------------
Purpose: Increment the reference count of this atom.
Returns: Last count
0 if the atom doesn't exist
Cond: --
*/
UINT PUBLIC Atom_AddRef(
int atom)
{
ATOMTABLE * this = &s_atomtable;
UINT cRef;
if (!Atom_IsValid(atom))
{
ASSERT(0);
return 0;
}
VALIDATE_ATOM(atom);
Atom_EnterCS(this);
{
A_ITEM * pitem = DSA_GetItemPtr(this->hdsa, atom);
if (pitem)
{
cRef = pitem->ucRef++;
}
else
{
cRef = 0;
}
}
Atom_LeaveCS(this);
return cRef;
}
/*----------------------------------------------------------
Purpose: Delete a string from the atom table.
If the reference count is not zero, we do nothing.
Returns: --
Cond: N.b. Decrements the reference count.
*/
void PUBLIC Atom_Delete(
int atom)
{
ATOMTABLE * this = &s_atomtable;
A_ITEM * pitem;
if (!Atom_IsValid(atom))
{
ASSERT(0);
return;
}
VALIDATE_ATOM(atom);
Atom_EnterCS(this);
{
pitem = DSA_GetItemPtr(this->hdsa, atom);
if (pitem)
{
int idpa;
int cFree;
ASSERT(pitem->atom == atom);
// Is the reference count already at zero?
if (0 == pitem->ucRef)
{
// Yes; somebody is calling Atom_Delete one-too-many times!
DEBUG_CODE( TRACE_MSG(TF_ATOM, TEXT("ATOM Deleting %d once-too-many!!"),
pitem->atom); )
ASSERT(0);
}
else if (0 == --pitem->ucRef)
{
// Yes
idpa = DPA_GetPtrIndex(this->hdpa, IntToPtr(atom)); // linear search
DEBUG_CODE( TRACE_MSG(TF_ATOM, TEXT("ATOM Deleting %d: %s"),
pitem->atom, pitem->psz ? pitem->psz : (LPCTSTR)TEXT("NULL")); )
ASSERT(atom == (int)DPA_GetPtr(this->hdpa, idpa));
if (DPA_ERR != idpa)
{
DPA_DeletePtr(this->hdpa, idpa);
ASSERT(pitem->psz);
Str_SetPtr(&pitem->psz, NULL);
DEBUG_CODE( pitem->psz = NULL; )
}
else
{
ASSERT(0); // Should never get here
}
// Add ptr to the free list. If this fails, we simply
// lose some efficiency in reusing this portion of the cache.
// This is not a memory leak.
//
cFree = DPA_GetPtrCount(this->hdpaFree);
DPA_InsertPtr(this->hdpaFree, cFree+1, IntToPtr(atom));
}
}
}
Atom_LeaveCS(this);
}
/*----------------------------------------------------------
Purpose: Replace the string corresponding with the atom with
another string. The atom will not change.
Returns: TRUE on success
Cond: --
*/
BOOL PUBLIC Atom_Replace(
int atom,
LPCTSTR pszNew)
{
ATOMTABLE * this = &s_atomtable;
BOOL bRet = FALSE;
A_ITEM * pitem;
ASSERT(pszNew);
if (!Atom_IsValid(atom))
{
return FALSE;
}
VALIDATE_ATOM(atom);
Atom_EnterCS(this);
{
pitem = DSA_GetItemPtr(this->hdsa, atom);
if (pitem)
{
ASSERT(atom == pitem->atom);
ASSERT(pitem->psz);
DEBUG_CODE( TRACE_MSG(TF_ATOM, TEXT("ATOM Change %d [%u]: %s -> %s"),
atom, pitem->ucRef, pitem->psz, pszNew); )
if (Str_SetPtr(&pitem->psz, pszNew))
{
DPA_Sort(this->hdpa, Atom_CompareIndexes, (LPARAM)this->hdsa);
bRet = TRUE;
}
#ifdef DEBUG
else
TRACE_MSG(TF_ATOM, TEXT("ATOM **Change failed"));
#endif
}
}
Atom_LeaveCS(this);
return bRet;
}
/*----------------------------------------------------------
Purpose: Translate all atoms with that contain the partial
string atomOld with the partial string atomNew.
Returns: TRUE on success
Cond: --
*/
BOOL PUBLIC Atom_Translate(
int atomOld,
int atomNew)
{
BOOL bRet = FALSE;
ATOMTABLE * this = &s_atomtable;
A_ITEM * pitem;
int idpa;
int cItem;
int atomSave = 0;
int cchOld;
LPCTSTR psz;
LPCTSTR pszOld;
LPCTSTR pszNew;
LPCTSTR pszRest;
TCHAR sz[MAXPATHLEN];
if ( !(Atom_IsValid(atomOld) && Atom_IsValid(atomNew)) )
{
return FALSE;
}
Atom_EnterCS(this);
{
pszOld = Atom_GetName(atomOld);
cchOld = lstrlen(pszOld);
pszNew = Atom_GetName(atomNew);
cItem = DPA_GetPtrCount(this->hdpa);
for (idpa = 0; idpa < cItem; idpa++)
{
pitem = MyGetPtr(this, idpa);
ASSERT(pitem);
if (pitem->atom == 0)
continue; // skip reserved atom
if (atomOld == pitem->atom)
{
atomSave = pitem->atom; // Save this one for last
continue;
}
psz = Atom_GetName(pitem->atom);
ASSERT(psz);
if (PathIsPrefix(psz, pszOld) && lstrlen(psz) >= cchOld)
{
// Translate this atom
//
pszRest = psz + cchOld; // whack up the path
PathCombine(sz, pszNew, pszRest);
DEBUG_CODE( TRACE_MSG(TF_ATOM, TEXT("ATOM Translate %d [%u]: %s -> %s"),
pitem->atom, pitem->ucRef, pitem->psz, (LPCTSTR)sz); )
if (!Str_SetPtr(&pitem->psz, sz))
goto Translate_Fail;
}
}
ASSERT(Atom_IsValid(atomSave)); // this means trouble
VALIDATE_ATOM(atomSave);
pitem = DSA_GetItemPtr(this->hdsa, atomSave);
if (pitem)
{
ASSERT(atomSave == pitem->atom);
ASSERT(pitem->psz);
DEBUG_CODE( TRACE_MSG(TF_ATOM, TEXT("ATOM Translate %d [%u]: %s -> %s"),
pitem->atom, pitem->ucRef, pitem->psz, pszNew); )
if (!Str_SetPtr(&pitem->psz, pszNew))
goto Translate_Fail;
}
bRet = TRUE;
Translate_Fail:
ASSERT(bRet);
// Sort here, even on a fail, so we correctly sort whatever
// got translated before the failure.
//
DPA_Sort(this->hdpa, Atom_CompareIndexes, (LPARAM)this->hdsa);
}
Atom_LeaveCS(this);
return bRet;
}
/*----------------------------------------------------------
Purpose: Search for a string in the atom table and return the atom
Returns: Atom
ATOM_ERR if the string is not in the table
Cond: Reference count is NOT incremented
*/
int PUBLIC Atom_Find(
LPCTSTR psz)
{
ATOMTABLE * this = &s_atomtable;
A_ITEM item;
A_ITEM * pitem;
int atomRet = ATOM_ERR;
int idpa;
ASSERT(psz);
Atom_EnterCS(this);
{
item.psz = (LPTSTR)(LPVOID)psz;
idpa = DPA_Search(this->hdpa, &item, 0, Atom_Compare, (LPARAM)this->hdsa,
DPAS_SORTED);
if (idpa != -1)
{
pitem = MyGetPtr(this, idpa);
atomRet = pitem->atom;
DEBUG_CODE( TRACE_MSG(TF_ATOM, TEXT("ATOM Find %s. Found %d [%u]: %s"),
psz, pitem->atom, pitem->ucRef, pitem->psz); )
ASSERT(IsSzEqual(psz, pitem->psz));
}
#ifdef DEBUG
else
TRACE_MSG(TF_ATOM, TEXT("ATOM **Not found %s"), psz);
#endif
}
Atom_LeaveCS(this);
return atomRet;
}
/*----------------------------------------------------------
Purpose: Get the string for this atom
Returns: Ptr to the string
NULL if the atom is bogus
Cond: The caller must serialize this.
*/
LPCTSTR PUBLIC Atom_GetName(
int atom)
{
ATOMTABLE * this = &s_atomtable;
LPCTSTR pszRet = NULL;
A_ITEM * pitem;
VALIDATE_ATOM(atom);
Atom_EnterCS(this);
{
pitem = DSA_GetItemPtr(this->hdsa, atom);
if (pitem)
{
pszRet = pitem->psz;
DEBUG_CODE( TRACE_MSG(TF_ATOM, TEXT("ATOM Getting name %d [%u]: %s"),
atom, pitem->ucRef, pszRet); )
ASSERT(atom == pitem->atom);
}
#ifdef DEBUG
else
TRACE_MSG(TF_ATOM, TEXT("ATOM **Cannot get %d"), atom);
#endif
}
Atom_LeaveCS(this);
return pszRet;
}
/*----------------------------------------------------------
Purpose: Return TRUE if atom2 is a partial path match of atom1.
Returns: boolean
Cond: Requires atom1 and atom2 to be valid.
*/
BOOL PUBLIC Atom_IsPartialMatch(
int atom1,
int atom2)
{
LPCTSTR psz1 = Atom_GetName(atom1);
LPCTSTR psz2 = Atom_GetName(atom2);
ASSERT(psz1);
ASSERT(psz2);
return PathIsPrefix(psz2, psz1);
}