#include <ws2_32.h>

Include dependency graph for dcatitem.c:

Functions
PTCATALOG_ENTRY WSAAPI	WsTcEntryAllocate (VOID)

VOID WSAAPI	WsTcEntryDelete (IN PTCATALOG_ENTRY CatalogEntry)

VOID WSAAPI	WsTcEntryDereference (IN PTCATALOG_ENTRY CatalogEntry)

DWORD WSAAPI	WsTcEntryInitializeFromRegistry (IN PTCATALOG_ENTRY CatalogEntry, IN HKEY ParentKey, IN DWORD UniqueId)

VOID WSAAPI	WsTcEntrySetProvider (IN PTCATALOG_ENTRY Entry, IN PTPROVIDER Provider)

Function Documentation

◆ WsTcEntryAllocate()

PTCATALOG_ENTRY WSAAPI WsTcEntryAllocate ( VOID )

Definition at line 17 of file dcatitem.c.

{
    PTCATALOG_ENTRY CatalogEntry;
 
    /* Allocate the catalog entry */
    CatalogEntry = HeapAlloc(WsSockHeap, HEAP_ZERO_MEMORY, sizeof(*CatalogEntry));
    if (CatalogEntry)
    {
        /* Set the default non-null members */
        CatalogEntry->RefCount = 1;
    }
 
    /* Return it */
    return CatalogEntry;
}

Referenced by WsTcRefreshFromRegistry().

◆ WsTcEntryDelete()

VOID WSAAPI WsTcEntryDelete ( IN PTCATALOG_ENTRY CatalogEntry )

Definition at line 35 of file dcatitem.c.

{
    /* Check if a provider is loaded */
    if (CatalogEntry->Provider)
    {
        /* Dereference it too */
        WsTpDereference(CatalogEntry->Provider);
        CatalogEntry->Provider = NULL;
    }
 
    /* Delete us */
    HeapFree(WsSockHeap, 0, CatalogEntry);
}

Referenced by WsTcEntryDereference().

◆ WsTcEntryDereference()

VOID WSAAPI WsTcEntryDereference ( IN PTCATALOG_ENTRY CatalogEntry )

Definition at line 51 of file dcatitem.c.

{
    /* Dereference and check if it's now 0 */
    if (!(InterlockedDecrement(&CatalogEntry->RefCount)))
    {
        /* We can delete the Provider now */
        WsTcEntryDelete(CatalogEntry);
    }
}

Referenced by WPUModifyIFSHandle(), WPUQueryBlockingCallback(), WSAAddressToStringA(), WSAAddressToStringW(), WSASocketW(), WSAStringToAddressA(), WSAStringToAddressW(), WsSockDelete(), WsTcDelete(), WsTcFindIfsProviderForSocket(), WsTcRefreshFromRegistry(), and WsTcUpdateProtocolList().

◆ WsTcEntryInitializeFromRegistry()

DWORD WSAAPI WsTcEntryInitializeFromRegistry	(	IN PTCATALOG_ENTRY	CatalogEntry,
		IN HKEY	ParentKey,
		IN DWORD	UniqueId
	)

Definition at line 63 of file dcatitem.c.

{
    CHAR CatalogEntryName[13];
    DWORD RegSize;
    DWORD RegType = REG_BINARY;
    HKEY EntryKey;
    DWORD Return;
    LPBYTE Buf;
    DWORD index;
 
    /* Convert to a 00000xxx string */
    sprintf(CatalogEntryName, "%0""12""lu", UniqueId);
 
    /* Open the Entry */
    Return = RegOpenKeyEx(ParentKey,
                          CatalogEntryName,
                          0,
                          KEY_READ,
                          &EntryKey);
 
    /* Get Size of Catalog Entry Structure */
    Return = RegQueryValueEx(EntryKey,
                              "PackedCatalogItem",
                              0,
                              NULL,
                              NULL,
                              &RegSize);
 
    if (!(Buf = HeapAlloc(WsSockHeap, HEAP_ZERO_MEMORY, RegSize)))
        return ERROR_NOT_ENOUGH_MEMORY;
 
    /* Read the Whole Catalog Entry Structure */
    Return = RegQueryValueEx(EntryKey,
                              "PackedCatalogItem",
                              0,
                              &RegType,
                              Buf,
                              &RegSize);
 
 
    memcpy(CatalogEntry->DllPath, Buf, sizeof(CatalogEntry->DllPath));
    index = sizeof(CatalogEntry->DllPath);
    if (index < RegSize)
    {
        memcpy(&CatalogEntry->ProtocolInfo, &Buf[index], sizeof(WSAPROTOCOL_INFOW));
        index += sizeof(WSAPROTOCOL_INFOW);
    }
    HeapFree(WsSockHeap, 0, Buf);
 
    /* Done */
    RegCloseKey(EntryKey);
    return Return;
}

◆ WsTcEntrySetProvider()

VOID WSAAPI WsTcEntrySetProvider	(	IN PTCATALOG_ENTRY	Entry,
		IN PTPROVIDER	Provider
	)

Definition at line 121 of file dcatitem.c.

{
    /* Reference the provider */
    InterlockedIncrement(&Provider->RefCount);
 
    /* Set it */
    Entry->Provider = Provider;
}