nsprovid.c File Reference

#include <ws2_32.h>

Include dependency graph for nsprovid.c:

Go to the source code of this file.

Functions
PNSQUERY_PROVIDER WSAAPI	WsNqProvAllocate (VOID)
DWORD WSAAPI	WsNqProvInitialize (IN PNSQUERY_PROVIDER QueryProvider, IN PNS_PROVIDER Provider)
VOID WSAAPI	WsNqProvDelete (IN PNSQUERY_PROVIDER QueryProvider)
PNS_PROVIDER WSAAPI	WsNpAllocate (VOID)
DWORD WSAAPI	WsNpInitialize (IN PNS_PROVIDER Provider, IN LPWSTR DllName, IN LPGUID ProviderId)
DWORD WSAAPI	WsNpNSPCleanup (IN PNS_PROVIDER Provider)
VOID WSAAPI	WsNpDelete (IN PNS_PROVIDER Provider)
VOID WSAAPI	WsNpDereference (IN PNS_PROVIDER Provider)
DWORD WSAAPI	WsNqProvLookupServiceEnd (IN PNSQUERY_PROVIDER QueryProvider)
DWORD WSAAPI	WsNqProvLookupServiceNext (IN PNSQUERY_PROVIDER QueryProvider, IN DWORD ControlFlags, IN PDWORD BufferLength, LPWSAQUERYSETW Results)
DWORD WSAAPI	WsNqProvLookupServiceBegin (IN PNSQUERY_PROVIDER QueryProvider, IN LPWSAQUERYSETW Restrictions, IN LPWSASERVICECLASSINFOW ServiceClassInfo, IN DWORD ControlFlags)
DWORD WSAAPI	WsNpLookupServiceEnd (IN PNS_PROVIDER Provider, IN HANDLE LookupHandle)
DWORD WSAAPI	WsNpLookupServiceNext (IN PNS_PROVIDER Provider, IN HANDLE LookupHandle, IN DWORD ControlFlags, OUT PDWORD BufferLength, OUT LPWSAQUERYSETW Results)
DWORD WSAAPI	WsNpLookupServiceBegin (IN PNS_PROVIDER Provider, IN LPWSAQUERYSETW Restrictions, IN LPWSASERVICECLASSINFOW ServiceClassInfo, IN DWORD ControlFlags, OUT PHANDLE LookupHandle)

Function Documentation

WsNpAllocate()

PNS_PROVIDER WSAAPI WsNpAllocate

(

VOID

)

Definition at line 63 of file nsprovid.c.

{
    PNS_PROVIDER Provider;
 
    /* Allocate the object */
    Provider = HeapAlloc(WsSockHeap, HEAP_ZERO_MEMORY, sizeof(*Provider));
    if (Provider)
    {
        /* Set non-null data */
        Provider->RefCount = 1;
        Provider->Service.cbSize = sizeof(NSP_ROUTINE);
    }
 
    /* Return it */
    return Provider;
}

Referenced by WsNcLoadProvider().

WsNpDelete()

VOID WSAAPI WsNpDelete

(

IN PNS_PROVIDER

Provider

)

Definition at line 161 of file nsprovid.c.

{
    /* Make sure we have a loaded handle */
    if (Provider->DllHandle)
    {
        /* Clean us up */
        WsNpNSPCleanup(Provider);
 
        /* Unload the library */
        FreeLibrary(Provider->DllHandle);
        Provider->DllHandle = NULL;
    }
 
    /* Delete us */
    HeapFree(WsSockHeap, 0, Provider);
}

Referenced by WsNpDereference().

WsNpDereference()

VOID WSAAPI WsNpDereference

(

IN PNS_PROVIDER

Provider

)

Definition at line 180 of file nsprovid.c.

{
    /* Decrease the reference count and check if it's zero */
    if (!InterlockedDecrement(&Provider->RefCount))
    {
        /* Delete us*/
        WsNpDelete(Provider);
    }
}

Referenced by WsNcEntryDelete(), WsNcLoadProvider(), and WsNqProvDelete().

WsNpInitialize()

DWORD WSAAPI WsNpInitialize	(	IN PNS_PROVIDER	Provider,
		IN LPWSTR	DllName,
		IN LPGUID	ProviderId
	)

Definition at line 82 of file nsprovid.c.

{
    INT ErrorCode = ERROR_SUCCESS;
    LPNSPSTARTUP NSPStartupProc;
    CHAR AnsiPath[MAX_PATH], ExpandedDllPath[MAX_PATH];
 
    /* Convert the path to ANSI */
    WideCharToMultiByte(CP_ACP,
                        0,
                        DllName,
                        -1,
                        AnsiPath,
                        MAX_PATH,
                        NULL,
                        NULL);
 
    /* Expand the DLL Path */
    ExpandEnvironmentStringsA(AnsiPath,
                              ExpandedDllPath,
                              MAX_PATH);
 
    /* Load the DLL */
    Provider->DllHandle = LoadLibraryA(ExpandedDllPath);
    if (!Provider->DllHandle)
    {
        /* Fail */
        ErrorCode = WSAEPROVIDERFAILEDINIT;
        goto Fail;
    }
 
    /* Get the pointer to NSPStartup */
    NSPStartupProc = (LPNSPSTARTUP)GetProcAddress(Provider->DllHandle,
                                                  "NSPStartup");
    if (!NSPStartupProc)
    {
        /* Fail */
        ErrorCode = WSAEPROVIDERFAILEDINIT;
        goto Fail;
    }
 
    /* Call it */
    (*NSPStartupProc)(ProviderId, &Provider->Service);
 
    /* Save the provider ID */
    Provider->ProviderId = *ProviderId;
    return ErrorCode;
 
Fail:
    /* Bail out */
    if (Provider->DllHandle) FreeLibrary(Provider->DllHandle);
    Provider->DllHandle = NULL;
    return ErrorCode;
}

Referenced by WsNcLoadProvider().

WsNpLookupServiceBegin()

DWORD WSAAPI WsNpLookupServiceBegin	(	IN PNS_PROVIDER	Provider,
		IN LPWSAQUERYSETW	Restrictions,
		IN LPWSASERVICECLASSINFOW	ServiceClassInfo,
		IN DWORD	ControlFlags,
		OUT PHANDLE	LookupHandle
	)

Definition at line 255 of file nsprovid.c.

{
    /* Call the NSP */
    return Provider->Service.NSPLookupServiceBegin(&Provider->ProviderId,
                                                   Restrictions,
                                                   ServiceClassInfo,
                                                   ControlFlags,
                                                   LookupHandle);
}

Referenced by WsNqProvLookupServiceBegin().

WsNpLookupServiceEnd()

DWORD WSAAPI WsNpLookupServiceEnd	(	IN PNS_PROVIDER	Provider,
		IN HANDLE	LookupHandle
	)

Definition at line 231 of file nsprovid.c.

{
    /* Call the NSP */
    return Provider->Service.NSPLookupServiceEnd(LookupHandle);
}

Referenced by WsNqProvLookupServiceEnd().

WsNpLookupServiceNext()

DWORD WSAAPI WsNpLookupServiceNext	(	IN PNS_PROVIDER	Provider,
		IN HANDLE	LookupHandle,
		IN DWORD	ControlFlags,
		OUT PDWORD	BufferLength,
		OUT LPWSAQUERYSETW	Results
	)

Definition at line 240 of file nsprovid.c.

{
    /* Call the NSP */
    return Provider->Service.NSPLookupServiceNext(LookupHandle,
                                                  ControlFlags,
                                                  BufferLength,
                                                  Results);
}

Referenced by WsNqProvLookupServiceNext().

WsNpNSPCleanup()

DWORD WSAAPI WsNpNSPCleanup

(

IN PNS_PROVIDER

Provider

)

Definition at line 140 of file nsprovid.c.

{
    INT ErrorCode = ERROR_SUCCESS;
    LPNSPCLEANUP lpNSPCleanup = NULL;
 
    /* Make sure we have a loaded handle */
    if (Provider->DllHandle)
    {
        /* Get the pointer and clear it */
        lpNSPCleanup = InterlockedExchangePointer((PVOID*)&Provider->Service.NSPCleanup,
                                                  NULL);
        /* If it's not NULL, call it */
        if (lpNSPCleanup) ErrorCode = lpNSPCleanup(&Provider->ProviderId);
    }
 
    /* Return */
    return ErrorCode;
}

Referenced by CleanupNamespaceProviders(), and WsNpDelete().

WsNqProvAllocate()

PNSQUERY_PROVIDER WSAAPI WsNqProvAllocate

(

VOID

)

Definition at line 17 of file nsprovid.c.

{
    PNSQUERY_PROVIDER Provider;
 
    /* Allocate the object */
    Provider = HeapAlloc(WsSockHeap, HEAP_ZERO_MEMORY, sizeof(*Provider));
 
    /* Return it */
    return Provider;
}

Referenced by WsNqAddProvider().

WsNqProvDelete()

VOID WSAAPI WsNqProvDelete

(

IN PNSQUERY_PROVIDER

QueryProvider

)

Definition at line 45 of file nsprovid.c.

{
    /* Check if we have a provider */
    if (QueryProvider->Provider)
    {
        /* Dereference it */
        WsNpDereference(QueryProvider->Provider);
 
        /* Clear it */
        QueryProvider->Provider = NULL;
    }
 
    /* Delete us */
    HeapFree(WsSockHeap, 0, QueryProvider);
}

Referenced by WsNqDelete(), WsNqLookupServiceBegin(), and WsNqLookupServiceNext().

WsNqProvInitialize()

DWORD WSAAPI WsNqProvInitialize	(	IN PNSQUERY_PROVIDER	QueryProvider,
		IN PNS_PROVIDER	Provider
	)

Definition at line 30 of file nsprovid.c.

{
    /* Reference the provider */
    InterlockedIncrement(&Provider->RefCount);
 
    /* Set it as our own */
    QueryProvider->Provider = Provider;
 
    /* Return success */
    return ERROR_SUCCESS;
}

Referenced by WsNqAddProvider().

WsNqProvLookupServiceBegin()

DWORD WSAAPI WsNqProvLookupServiceBegin	(	IN PNSQUERY_PROVIDER	QueryProvider,
		IN LPWSAQUERYSETW	Restrictions,
		IN LPWSASERVICECLASSINFOW	ServiceClassInfo,
		IN DWORD	ControlFlags
	)

Definition at line 216 of file nsprovid.c.

{
    /* Simply call the provider */
    return WsNpLookupServiceBegin(QueryProvider->Provider,
                                  Restrictions,
                                  ServiceClassInfo,
                                  ControlFlags,
                                  &QueryProvider->LookupHandle);
}

Referenced by WsNqLookupServiceBegin().

WsNqProvLookupServiceEnd()

DWORD WSAAPI WsNqProvLookupServiceEnd

(

IN PNSQUERY_PROVIDER

QueryProvider

)

Definition at line 192 of file nsprovid.c.

{
    /* Simply call the provider */
    return WsNpLookupServiceEnd(QueryProvider->Provider,
                                QueryProvider->LookupHandle);
}

Referenced by WsNqLookupServiceEnd(), and WsNqLookupServiceNext().

WsNqProvLookupServiceNext()

DWORD WSAAPI WsNqProvLookupServiceNext	(	IN PNSQUERY_PROVIDER	QueryProvider,
		IN DWORD	ControlFlags,
		IN PDWORD	BufferLength,
		LPWSAQUERYSETW	Results
	)

Definition at line 201 of file nsprovid.c.

{
    /* Simply call the provider */
    return WsNpLookupServiceNext(QueryProvider->Provider,
                                 QueryProvider->LookupHandle,
                                 ControlFlags,
                                 BufferLength,
                                 Results);
}