cache.c File Reference

#include "precomp.h"
#include <debug.h>

Include dependency graph for cache.c:

Go to the source code of this file.

Macros
#define	NDEBUG
#define	DnsCacheLock()   do { EnterCriticalSection(&DnsCache.Lock); } while (0)
#define	DnsCacheUnlock()   do { LeaveCriticalSection(&DnsCache.Lock); } while (0)

Functions
VOID	DnsIntCacheInitialize (VOID)
VOID	DnsIntCacheFree (VOID)
VOID	DnsIntCacheRemoveEntryItem (PRESOLVER_CACHE_ENTRY CacheEntry)
DNS_STATUS	DnsIntCacheFlush (_In_ ULONG ulFlags)
DNS_STATUS	DnsIntFlushCacheEntry (_In_ LPCWSTR pszName, _In_ WORD wType)
DNS_STATUS	DnsIntCacheGetEntryByName (LPCWSTR Name, WORD wType, DWORD dwFlags, PDNS_RECORDW *Record)
BOOL	DnsIntCacheRemoveEntryByName (LPCWSTR Name)
VOID	DnsIntCacheAddEntry (_In_ PDNS_RECORDW Record, _In_ BOOL bHostsFileEntry)
DNS_STATUS	DnsIntCacheGetEntries (_Out_ DNS_CACHE_ENTRY **ppCacheEntries)

Variables
static RESOLVER_CACHE	DnsCache
static BOOL	DnsCacheInitialized = FALSE

Macro Definition Documentation

DnsCacheLock

#define DnsCacheLock

(

)

do { EnterCriticalSection(&DnsCache.Lock); } while (0)

Definition at line 17 of file cache.c.

DnsCacheUnlock

#define DnsCacheUnlock

(

)

do { LeaveCriticalSection(&DnsCache.Lock); } while (0)

Definition at line 18 of file cache.c.

NDEBUG

#define NDEBUG

Definition at line 11 of file cache.c.

Function Documentation

DnsIntCacheAddEntry()

VOID DnsIntCacheAddEntry	(	_In_ PDNS_RECORDW	Record,
		_In_ BOOL	bHostsFileEntry
	)

Definition at line 233 of file cache.c.

{
    PRESOLVER_CACHE_ENTRY Entry;
 
    DPRINT("DnsIntCacheAddEntry(%p %u)\n",
           Record, bHostsFileEntry);
 
    DPRINT("Name: %S\n", Record->pName);
    DPRINT("TTL: %lu\n", Record->dwTtl);
 
    /* Lock the cache */
    DnsCacheLock();
 
    /* Match the Id with all the entries in the List */
    Entry = (PRESOLVER_CACHE_ENTRY)HeapAlloc(GetProcessHeap(), 0, sizeof(*Entry));
    if (!Entry)
        return;
 
    Entry->bHostsFileEntry = bHostsFileEntry;
    Entry->Record = DnsRecordSetCopyEx(Record, DnsCharSetUnicode, DnsCharSetUnicode);
 
    /* Insert it to our List */
    InsertTailList(&DnsCache.RecordList, &Entry->CacheLink);
 
    /* Release the cache */
    DnsCacheUnlock();
}

Referenced by AddIpv4HostEntries(), AddIpv6HostEntries(), and R_ResolverQuery().

DnsIntCacheFlush()

DNS_STATUS DnsIntCacheFlush

(

_In_ ULONG

ulFlags

)

Definition at line 69 of file cache.c.

{
    PLIST_ENTRY Entry, NextEntry;
    PRESOLVER_CACHE_ENTRY CacheEntry;
 
    DPRINT("DnsIntCacheFlush(%lu)\n", ulFlags);
 
    /* Lock the cache */
    DnsCacheLock();
 
    /* Loop every entry */
    Entry = DnsCache.RecordList.Flink;
    while (Entry != &DnsCache.RecordList)
    {
        NextEntry = Entry->Flink;
 
        /* Get this entry */
        CacheEntry = CONTAINING_RECORD(Entry, RESOLVER_CACHE_ENTRY, CacheLink);
 
        /* Remove it from list */
        if (((ulFlags & CACHE_FLUSH_HOSTS_FILE_ENTRIES) && (CacheEntry->bHostsFileEntry != FALSE)) ||
            ((ulFlags & CACHE_FLUSH_NON_HOSTS_FILE_ENTRIES) && (CacheEntry->bHostsFileEntry == FALSE)))
            DnsIntCacheRemoveEntryItem(CacheEntry);
 
        /* Move to the next entry */
        Entry = NextEntry;
    }
 
    /* Unlock the cache */
    DnsCacheUnlock();
 
    return ERROR_SUCCESS;
}

Referenced by DnsIntCacheFree(), and R_ResolverFlushCache().

DnsIntCacheFree()

VOID DnsIntCacheFree

(

VOID

)

Definition at line 36 of file cache.c.

{
    DPRINT("DnsIntCacheFree()\n");
 
    /* Check if we're initialized */
    if (!DnsCacheInitialized)
        return;
 
    if (!DnsCache.RecordList.Flink)
        return;
 
    DnsIntCacheFlush(CACHE_FLUSH_ALL);
 
    DeleteCriticalSection(&DnsCache.Lock);
    DnsCacheInitialized = FALSE;
}

Referenced by ServiceControlHandler(), and ServiceMain().

DnsIntCacheGetEntries()

DNS_STATUS DnsIntCacheGetEntries

(

_Out_ DNS_CACHE_ENTRY **

ppCacheEntries

)

Definition at line 264 of file cache.c.

{
    PRESOLVER_CACHE_ENTRY CacheEntry;
    PLIST_ENTRY NextEntry;
    PDNS_CACHE_ENTRY pLastEntry = NULL, pNewEntry;
 
    /* Lock the cache */
    DnsCacheLock();
 
    *ppCacheEntries = NULL;
 
    NextEntry = DnsCache.RecordList.Flink;
    while (NextEntry != &DnsCache.RecordList)
    {
        /* Get the Current Entry */
        CacheEntry = CONTAINING_RECORD(NextEntry, RESOLVER_CACHE_ENTRY, CacheLink);
 
        DPRINT("1 %S %lu\n", CacheEntry->Record->pName, CacheEntry->Record->wType);
        if (CacheEntry->Record->pNext)
        {
            DPRINT("2 %S %lu\n", CacheEntry->Record->pNext->pName, CacheEntry->Record->pNext->wType);
        }
 
        pNewEntry = midl_user_allocate(sizeof(DNS_CACHE_ENTRY));
        if (pNewEntry == NULL)
        {
            return ERROR_OUTOFMEMORY;
        }
 
        pNewEntry->pszName = midl_user_allocate((wcslen(CacheEntry->Record->pName) + 1) * sizeof(WCHAR));
        if (pNewEntry->pszName == NULL)
        {
            return ERROR_OUTOFMEMORY;
        }
 
        wcscpy(pNewEntry->pszName, CacheEntry->Record->pName);
        pNewEntry->wType1 = CacheEntry->Record->wType;
        pNewEntry->wType2 = 0;
        pNewEntry->wFlags = 0;
 
        if (pLastEntry == NULL)
            *ppCacheEntries = pNewEntry;
        else
            pLastEntry->pNext = pNewEntry;
        pLastEntry = pNewEntry;
 
        NextEntry = NextEntry->Flink;
    }
 
    /* Release the cache */
    DnsCacheUnlock();
 
    return ERROR_SUCCESS;
}

Referenced by CRrReadCache().

DnsIntCacheGetEntryByName()

DNS_STATUS DnsIntCacheGetEntryByName	(	LPCWSTR	Name,
		WORD	wType,
		DWORD	dwFlags,
		PDNS_RECORDW *	Record
	)

Definition at line 150 of file cache.c.

{
    DNS_STATUS Status = DNS_INFO_NO_RECORDS;
    PRESOLVER_CACHE_ENTRY CacheEntry;
    PLIST_ENTRY NextEntry;
 
    DPRINT("DnsIntCacheGetEntryByName(%S %hu 0x%lx %p)\n",
           Name, wType, dwFlags, Record);
 
    /* Assume failure */
    *Record = NULL;
 
    /* Lock the cache */
    DnsCacheLock();
 
    /* Match the Id with all the entries in the List */
    NextEntry = DnsCache.RecordList.Flink;
    while (NextEntry != &DnsCache.RecordList)
    {
        /* Get the Current Entry */
        CacheEntry = CONTAINING_RECORD(NextEntry, RESOLVER_CACHE_ENTRY, CacheLink);
 
        /* Check if this is the Catalog Entry ID we want */
        if (_wcsicmp(CacheEntry->Record->pName, Name) == 0)
        {
            /* Copy the entry and return it */
            *Record = DnsRecordSetCopyEx(CacheEntry->Record, DnsCharSetUnicode, DnsCharSetUnicode);
            Status = ERROR_SUCCESS;
            break;
        }
 
        NextEntry = NextEntry->Flink;
    }
 
    /* Release the cache */
    DnsCacheUnlock();
 
    return Status;
}

Referenced by R_ResolverQuery().

DnsIntCacheInitialize()

VOID DnsIntCacheInitialize

(

VOID

)

Definition at line 21 of file cache.c.

{
    DPRINT("DnsIntCacheInitialize()\n");
 
    /* Check if we're initialized */
    if (DnsCacheInitialized)
        return;
 
    /* Initialize the cache lock and namespace list */
    InitializeCriticalSection((LPCRITICAL_SECTION)&DnsCache.Lock);
    InitializeListHead(&DnsCache.RecordList);
    DnsCacheInitialized = TRUE;
}

Referenced by ServiceMain().

DnsIntCacheRemoveEntryByName()

BOOL DnsIntCacheRemoveEntryByName

(

LPCWSTR

Name

)

Definition at line 195 of file cache.c.

{
    BOOL Ret = FALSE;
    PRESOLVER_CACHE_ENTRY CacheEntry;
    PLIST_ENTRY NextEntry;
 
    DPRINT("DnsIntCacheRemoveEntryByName(%S)\n", Name);
 
    /* Lock the cache */
    DnsCacheLock();
 
    /* Match the Id with all the entries in the List */
    NextEntry = DnsCache.RecordList.Flink;
    while (NextEntry != &DnsCache.RecordList)
    {
        /* Get the Current Entry */
        CacheEntry = CONTAINING_RECORD(NextEntry, RESOLVER_CACHE_ENTRY, CacheLink);
 
        /* Check if this is the Catalog Entry ID we want */
        if (_wcsicmp(CacheEntry->Record->pName, Name) == 0)
        {
            /* Remove the entry */
            DnsIntCacheRemoveEntryItem(CacheEntry);
            Ret = TRUE;
            break;
        }
 
        NextEntry = NextEntry->Flink;
    }
 
    /* Release the cache */
    DnsCacheUnlock();
 
    /* Return */
    return Ret;
}

DnsIntCacheRemoveEntryItem()

VOID DnsIntCacheRemoveEntryItem

(

PRESOLVER_CACHE_ENTRY

CacheEntry

)

Definition at line 54 of file cache.c.

{
    DPRINT("DnsIntCacheRemoveEntryItem(%p)\n", CacheEntry);
 
    /* Remove the entry from the list */
    RemoveEntryList(&CacheEntry->CacheLink);
 
    /* Free record */
    DnsRecordListFree(CacheEntry->Record, DnsFreeRecordList);
 
    /* Delete us */
    HeapFree(GetProcessHeap(), 0, CacheEntry);
}

Referenced by DnsIntCacheFlush(), DnsIntCacheRemoveEntryByName(), and DnsIntFlushCacheEntry().

DnsIntFlushCacheEntry()

DNS_STATUS DnsIntFlushCacheEntry	(	_In_ LPCWSTR	pszName,
		_In_ WORD	wType
	)

Definition at line 106 of file cache.c.

{
    PLIST_ENTRY Entry, NextEntry;
    PRESOLVER_CACHE_ENTRY CacheEntry;
 
    DPRINT("DnsIntFlushCacheEntry(%S %x)\n", pszName, wType);
 
    /* Lock the cache */
    DnsCacheLock();
 
    /* Loop every entry */
    Entry = DnsCache.RecordList.Flink;
    while (Entry != &DnsCache.RecordList)
    {
        NextEntry = Entry->Flink;
 
        /* Get this entry */
        CacheEntry = CONTAINING_RECORD(Entry, RESOLVER_CACHE_ENTRY, CacheLink);
 
        /* Remove it from the list */
        if ((_wcsicmp(CacheEntry->Record->pName, pszName) == 0) &&
            (CacheEntry->bHostsFileEntry == FALSE))
        {
            if ((wType == DNS_TYPE_ANY) ||
                (CacheEntry->Record->wType == wType))
            {
                DnsIntCacheRemoveEntryItem(CacheEntry);
            }
        }
 
        /* Move to the next entry */
        Entry = NextEntry;
    }
 
    /* Unlock the cache */
    DnsCacheUnlock();
 
    return ERROR_SUCCESS;
}

Referenced by R_ResolverFlushCacheEntry().

Variable Documentation

DnsCache

RESOLVER_CACHE DnsCache

static

Definition at line 14 of file cache.c.

Referenced by DnsGetCacheDataTable(), DnsIntCacheAddEntry(), DnsIntCacheFlush(), DnsIntCacheFree(), DnsIntCacheGetEntries(), DnsIntCacheGetEntryByName(), DnsIntCacheInitialize(), DnsIntCacheRemoveEntryByName(), and DnsIntFlushCacheEntry().

DnsCacheInitialized

BOOL DnsCacheInitialized = FALSE

static

Definition at line 15 of file cache.c.

Referenced by DnsIntCacheFree(), and DnsIntCacheInitialize().