libmgmt.c File Reference

#include "precomp.h"

Go to the source code of this file.

Functions
int	GetHmodTableIndex (PUNICODE_STRING pstrLibName)
VOID	AddHmodDependency (int iatom)
VOID	RemoveHmodDependency (int iatom)
HANDLE	xxxLoadHmodIndex (int iatom, BOOL bWx86KnownDll)
VOID	xxxDoSysExpunge (PTHREADINFO pti)
Variables
int	catomSysTableEntries
ATOM	aatomSysLoaded [CLIBS]
int	acatomSysUse [CLIBS]
int	acatomSysDepends [CLIBS]

---

Function Documentation

VOID AddHmodDependency (  int  iatom  )

Definition at line 136 of file libmgmt.c. References acatomSysDepends, catomSysTableEntries, and VOID(). Referenced by _SetWinEventHook(), and zzzSetWindowsHookEx(). { UserAssert(iatom >= 0); if (iatom < catomSysTableEntries) { acatomSysDepends[iatom]++; } }

int GetHmodTableIndex (  PUNICODE_STRING  pstrLibName  )

Definition at line 53 of file libmgmt.c. References aatomSysLoaded, acatomSysDepends, acatomSysUse, catomSysTableEntries, CLIBS, FALSE, ProbeAndReadUnicodeString, ProbeForReadUnicodeStringBuffer, UserAddAtom(), and UserDeleteAtom(). Referenced by _SetWinEventHook(), and zzzSetWindowsHookEx(). { int i; ATOM atom; UNICODE_STRING strLibName; /* * Probe string */ try { strLibName = ProbeAndReadUnicodeString(pstrLibName); ProbeForReadUnicodeStringBuffer(strLibName); atom = UserAddAtom(strLibName.Buffer, FALSE); } except (W32ExceptionHandler(FALSE, RIP_WARNING)) { return -1; } /* * If we can't add the atom we're hosed * so return an error. */ if (atom == 0) { return -1; } /* * Search for atom index */ for (i = 0; i < catomSysTableEntries && aatomSysLoaded[i] != atom; i++) ; if (i == catomSysTableEntries) { /* * Find empty entry for atom */ for (i = 0; i < catomSysTableEntries && aatomSysLoaded[i]; i++) ; /* * Check if no empty entry found */ if (i == catomSysTableEntries) { if (i == CLIBS) { UserDeleteAtom(atom); RIPERR0(ERROR_NOT_ENOUGH_MEMORY, RIP_WARNING, "Memory allocation failed in GetHmodTableIndex"); return -1; } /* * Increase table size */ catomSysTableEntries++; } /* * Set entry */ aatomSysLoaded[i] = atom; acatomSysUse[i] = 0; acatomSysDepends[i] = 0; } else { UserDeleteAtom(atom); } return i; }

VOID RemoveHmodDependency (  int  iatom  )

Definition at line 158 of file libmgmt.c. References aatomSysLoaded, acatomSysDepends, acatomSysUse, catomSysTableEntries, gcSysExpunge, gdwSysExpungeMask, and VOID(). Referenced by DestroyEventHook(), and FreeHook(). { UserAssert(iatom >= 0); if (iatom < catomSysTableEntries && --acatomSysDepends[iatom] == 0) { if (acatomSysUse[iatom]) { /* * Cause each thread to check for expunged dlls * the next time they awake. */ gcSysExpunge++; gdwSysExpungeMask |= (1 << iatom); } else { aatomSysLoaded[iatom] = 0; } } }

VOID xxxDoSysExpunge (  PTHREADINFO  pti  )

Definition at line 253 of file libmgmt.c. References aatomSysLoaded, acatomSysDepends, acatomSysUse, tagPROCESSINFO::ahmodLibLoaded, catomSysTableEntries, CLEARHMODLOADED, ClientFreeLibrary(), tagPROCESSINFO::cSysExpunge, gcSysExpunge, gdwSysExpungeMask, tagTHREADINFO::ppi, TESTHMODLOADED, UserDeleteAtom(), and VOID(). Referenced by xxxInternalGetMessage(), and xxxSleepThread(). { int i; /* * Clear this first before we potentially leave the critical section. */ pti->ppi->cSysExpunge = gcSysExpunge; /* * Scan for libraries that have been freed */ for (i = 0; i < catomSysTableEntries; i++) { if ((acatomSysDepends[i] == 0) && (aatomSysLoaded[i] != 0) && TESTHMODLOADED(pti, i)) { HANDLE hmodFree = pti->ppi->ahmodLibLoaded[i]; /* * Clear this hmod for this process before we leave the * critical section. */ CLEARHMODLOADED(pti, i); /* * Decrement the count of processes that have loaded this * .dll. If there are no more, then destroy the reference * to this .dll. */ if (--acatomSysUse[i] == 0) { UserDeleteAtom(aatomSysLoaded[i]); aatomSysLoaded[i] = 0; gdwSysExpungeMask &= ~(1 << i); } /* * Call back the client to free the library... */ ClientFreeLibrary(hmodFree); } } }

HANDLE xxxLoadHmodIndex (  int  iatom, BOOL  bWx86KnownDll )

Definition at line 192 of file libmgmt.c. References aatomSysLoaded, acatomSysUse, tagPROCESSINFO::ahmodLibLoaded, catomSysTableEntries, ClientFreeLibrary(), ClientLoadLibrary(), gptiRit, MAX_PATH, NULL, tagTHREADINFO::ppi, PtiCurrent, RtlInitUnicodeString(), SETHMODLOADED, TESTHMODLOADED, and UserGetAtomName(). Referenced by xxxCallHook2(), and xxxGetEventProc(). { WCHAR pszLibName[MAX_PATH]; HANDLE hmod; UNICODE_STRING strLibrary; PTHREADINFO ptiCurrent = PtiCurrent(); UserAssert((!gptiRit || gptiRit->ppi != PtiCurrent()->ppi) && "Shouldn't load global hooks on system process - gptiRit->ppi is the system process"); if (iatom >= catomSysTableEntries) { RIPERR0(ERROR_INVALID_PARAMETER, RIP_WARNING, "Index out of range"); return NULL; } UserGetAtomName(aatomSysLoaded[iatom], pszLibName, sizeof(pszLibName)/sizeof(WCHAR)); /* * Call back the client to load the library. */ RtlInitUnicodeString(&strLibrary, pszLibName); hmod = ClientLoadLibrary(&strLibrary, bWx86KnownDll); if (hmod != NULL) { /* * Check to make sure another thread hasn't loaded this library * while we were outside the critical section. */ if (!TESTHMODLOADED(ptiCurrent, iatom)) { /* * Go ahead and bump the reference count. */ acatomSysUse[iatom]++; SETHMODLOADED(ptiCurrent, iatom, hmod); } else { /* * Another thread loaded it while we were outside the * critical section. Unload it so the system's * reference count is correct. */ ClientFreeLibrary(ptiCurrent->ppi->ahmodLibLoaded[iatom]); } } return hmod; }

---

Variable Documentation

ATOM aatomSysLoaded[CLIBS]

Definition at line 29 of file libmgmt.c. Referenced by GetHmodTableIndex(), RemoveHmodDependency(), xxxDoSysExpunge(), and xxxLoadHmodIndex().

int acatomSysDepends[CLIBS]

Definition at line 39 of file libmgmt.c. Referenced by AddHmodDependency(), GetHmodTableIndex(), RemoveHmodDependency(), and xxxDoSysExpunge().

int acatomSysUse[CLIBS]

Definition at line 34 of file libmgmt.c. Referenced by GetHmodTableIndex(), RemoveHmodDependency(), xxxDoSysExpunge(), and xxxLoadHmodIndex().

int catomSysTableEntries

Definition at line 23 of file libmgmt.c. Referenced by AddHmodDependency(), GetHmodTableIndex(), RemoveHmodDependency(), xxxDoSysExpunge(), and xxxLoadHmodIndex().

---