enumprot.c File Reference

#include <ws2_32.h>
#include <debug.h>

Include dependency graph for enumprot.c:

Go to the source code of this file.

Macros
#define	NDEBUG

Functions
BOOL WSAAPI	CheckProtocolMatch (IN LPINT ProtocolSet, IN LPWSAPROTOCOL_INFOW ProtocolInfo)
VOID WSAAPI	ProtocolInfoFromContext (IN LPWSAPROTOCOL_INFOW ProtocolInfo, IN PPROTOCOL_ENUM_CONTEXT Context)
BOOL WSAAPI	ProtocolEnumerationProc (PVOID EnumContext, PTCATALOG_ENTRY Entry)
BOOL WSAAPI	ProviderEnumerationProc (PVOID EnumContext, PNSCATALOG_ENTRY Entry)
PTCATALOG WSAAPI	OpenInitializedCatalog (VOID)
INT WSPAPI	WSCEnumProtocols (IN LPINT lpiProtocols, OUT LPWSAPROTOCOL_INFOW lpProtocolBuffer, IN OUT LPDWORD lpdwBufferLength, OUT LPINT lpErrno)
INT WSAAPI	WSAEnumProtocolsA (IN LPINT lpiProtocols, OUT LPWSAPROTOCOL_INFOA lpProtocolBuffer, IN OUT LPDWORD lpdwBufferLength)
INT WSAAPI	WSAEnumProtocolsW (IN LPINT lpiProtocols, OUT LPWSAPROTOCOL_INFOW lpProtocolBuffer, IN OUT LPDWORD lpdwBufferLength)
INT WSPAPI	WPUGetProviderPath (IN LPGUID lpProviderId, OUT LPWSTR lpszProviderDllPath, IN OUT LPINT lpProviderDllPathLen, OUT LPINT lpErrno)
INT WSAAPI	WSAProviderConfigChange (IN OUT LPHANDLE lpNotificationHandle, IN LPWSAOVERLAPPED lpOverlapped, IN LPWSAOVERLAPPED_COMPLETION_ROUTINE lpCompletionRoutine)
INT WSPAPI	WSCGetProviderPath (IN LPGUID lpProviderId, OUT LPWSTR lpszProviderDllPath, IN OUT LPINT lpProviderDllPathLen, OUT LPINT lpErrno)

Macro Definition Documentation

NDEBUG

#define NDEBUG

Definition at line 14 of file enumprot.c.

Function Documentation

CheckProtocolMatch()

BOOL WSAAPI CheckProtocolMatch	(	IN LPINT	ProtocolSet,
		IN LPWSAPROTOCOL_INFOW	ProtocolInfo
	)

Definition at line 21 of file enumprot.c.

{
    BOOL Return = FALSE;
    DWORD i = 0;
    INT ProtocolId;
 
    /* Make sure we have a set */
    if (ProtocolSet)
    {
        /* Get the first ID */
        ProtocolId = ProtocolSet[i];
 
        /* Loop the list */
        while (ProtocolId != 0 && ProtocolInfo->iProtocol != 0)
        {
            /* Check if it's within ranges */
            if ((ProtocolId >= ProtocolInfo->iProtocol) &&
                (ProtocolId <= (ProtocolInfo->iProtocol +
                                ProtocolInfo->iProtocolMaxOffset)))
            {
                /* Found it */
                Return = TRUE;
                break;
            }
 
            /* Move on */
            i++;
            ProtocolId = ProtocolSet[i];
        }
    }
    else
    {
        /* Assume match */
        Return = TRUE;
    }
 
    /* Return result */
    return Return;
}

Referenced by ProtocolEnumerationProc().

OpenInitializedCatalog()

PTCATALOG WSAAPI OpenInitializedCatalog

(

VOID

)

Definition at line 137 of file enumprot.c.

{
    PTCATALOG Catalog;
    HKEY WsKey;
 
    /* Allocate the catalog */
    Catalog = WsTcAllocate();
    if (Catalog)
    {
        /* Open the WS Key */
        WsKey = WsOpenRegistryRoot();
 
        /* Initialize the catalog */
        WsTcInitializeFromRegistry(Catalog, WsKey, NULL);
 
        /* Close the key */
        RegCloseKey(WsKey);
    }
 
    /* Return it */
    return Catalog;
}

Referenced by WSCEnumProtocols().

ProtocolEnumerationProc()

BOOL WSAAPI ProtocolEnumerationProc	(	PVOID	EnumContext,
		PTCATALOG_ENTRY	Entry
	)

Definition at line 84 of file enumprot.c.

{
    PPROTOCOL_ENUM_CONTEXT Context = (PPROTOCOL_ENUM_CONTEXT)EnumContext;
    LPWSAPROTOCOL_INFOW ProtocolInfo = &Entry->ProtocolInfo;
 
    /* Check if this protocol matches */
    if (CheckProtocolMatch(Context->Protocols, ProtocolInfo))
    {
        /* Copy the information */
        ProtocolInfoFromContext(ProtocolInfo, Context);
        Context->BufferUsed += sizeof(*ProtocolInfo);
    }
 
    /* Continue enumeration */
    return TRUE;
}

Referenced by WSCEnumProtocols().

ProtocolInfoFromContext()

VOID WSAAPI ProtocolInfoFromContext	(	IN LPWSAPROTOCOL_INFOW	ProtocolInfo,
		IN PPROTOCOL_ENUM_CONTEXT	Context
	)

Definition at line 64 of file enumprot.c.

{
    /* Check if we'll have space */
    if ((Context->BufferUsed + sizeof(*ProtocolInfo)) <=
        (Context->BufferLength))
    {
        /* Copy the data */
        RtlMoveMemory((PVOID)((ULONG_PTR)Context->ProtocolBuffer +
                              Context->BufferUsed),
                      ProtocolInfo,
                      sizeof(*ProtocolInfo));
 
        /* Increase the count */
        Context->Count++;
    }
}

Referenced by ProtocolEnumerationProc().

ProviderEnumerationProc()

BOOL WSAAPI ProviderEnumerationProc	(	PVOID	EnumContext,
		PNSCATALOG_ENTRY	Entry
	)

Definition at line 104 of file enumprot.c.

{
    INT PathLen;
    PPROVIDER_ENUM_CONTEXT Context = (PPROVIDER_ENUM_CONTEXT)EnumContext;
 
    /* Check if this provider matches */
    if (IsEqualGUID(&Entry->ProviderId, &Context->ProviderId))
    {
        /* Get the information about the provider */
        PathLen = wcslen(Entry->DllPath) + 1;
        Context->FoundPathLen = PathLen;
        Context->Found = 1;
 
        /* If we have enough room, copy path */
        if (PathLen <= Context->ProviderDllPathLen)
        {
            wcscpy(Context->ProviderDllPath, Entry->DllPath);
        }
 
        /* Stop enumeration */
        return FALSE;
    }
    else
    {
        /* Continue enumeration */
        return TRUE;
    }
 
}

Referenced by WSCGetProviderPath().

WPUGetProviderPath()

INT WSPAPI WPUGetProviderPath	(	IN LPGUID	lpProviderId,
		OUT LPWSTR	lpszProviderDllPath,
		IN OUT LPINT	lpProviderDllPathLen,
		OUT LPINT	lpErrno
	)

Definition at line 297 of file enumprot.c.

{
    return WSCGetProviderPath(lpProviderId, lpszProviderDllPath, lpProviderDllPathLen, lpErrno);
}

Referenced by WsTpInitialize().

WSAEnumProtocolsA()

INT WSAAPI WSAEnumProtocolsA	(	IN LPINT	lpiProtocols,
		OUT LPWSAPROTOCOL_INFOA	lpProtocolBuffer,
		IN OUT LPDWORD	lpdwBufferLength
	)

Definition at line 229 of file enumprot.c.

{
    INT error, i, count;
    LPWSAPROTOCOL_INFOW protocolInfoW;
    DWORD size;
    DPRINT("WSAEnumProtocolsA: %p %p %p\n", lpiProtocols, lpProtocolBuffer, lpdwBufferLength);
    if (!lpdwBufferLength)
    {
        SetLastError(WSAENOBUFS);
        return SOCKET_ERROR;
    }
    count = WSCEnumProtocols(lpiProtocols, NULL, &size, &error);
    if (!lpProtocolBuffer || *lpdwBufferLength < (size/sizeof(WSAPROTOCOL_INFOW))*sizeof(WSAPROTOCOL_INFOA))
    {
        *lpdwBufferLength = (size/sizeof(WSAPROTOCOL_INFOW))*sizeof(WSAPROTOCOL_INFOA);
        SetLastError(WSAENOBUFS);
        return SOCKET_ERROR;
    }
    protocolInfoW = HeapAlloc(WsSockHeap, 0, size);
    count = WSCEnumProtocols(lpiProtocols, protocolInfoW, &size, &error);
    if (SOCKET_ERROR == count)
    {
        HeapFree(WsSockHeap, 0, protocolInfoW);
        SetLastError(error);
        return SOCKET_ERROR;
    }
    *lpdwBufferLength = 0;
    for (i = 0; i < count; i++)
    {
        /* Copy the data */
        RtlMoveMemory(&lpProtocolBuffer[i],
                      &protocolInfoW[i],
                      sizeof(lpProtocolBuffer[0])-sizeof(lpProtocolBuffer[0].szProtocol));
        wcstombs(lpProtocolBuffer[i].szProtocol, protocolInfoW[i].szProtocol, sizeof(lpProtocolBuffer[0].szProtocol));
        *lpdwBufferLength += sizeof(WSAPROTOCOL_INFOA);
    }
    HeapFree(WsSockHeap, 0, protocolInfoW);
    return i;
}

Referenced by test_WSAEnumProtocolsA().

WSAEnumProtocolsW()

INT WSAAPI WSAEnumProtocolsW	(	IN LPINT	lpiProtocols,
		OUT LPWSAPROTOCOL_INFOW	lpProtocolBuffer,
		IN OUT LPDWORD	lpdwBufferLength
	)

Definition at line 276 of file enumprot.c.

{
    INT error, count;
    DPRINT("WSAEnumProtocolsW: %p %p %p\n", lpiProtocols, lpProtocolBuffer, lpdwBufferLength);
    count = WSCEnumProtocols(lpiProtocols, lpProtocolBuffer, lpdwBufferLength, &error);
    if (SOCKET_ERROR == count)
    {
        SetLastError(error);
        return SOCKET_ERROR;
    }
    return count;
}

Referenced by test_WSAEnumProtocolsW().

WSAProviderConfigChange()

INT WSAAPI WSAProviderConfigChange	(	IN OUT LPHANDLE	lpNotificationHandle,
		IN LPWSAOVERLAPPED	lpOverlapped,
		IN LPWSAOVERLAPPED_COMPLETION_ROUTINE	lpCompletionRoutine
	)

Definition at line 310 of file enumprot.c.

{
    DPRINT("WSAProviderConfigChange: %p\n", lpNotificationHandle);
    UNIMPLEMENTED;
    SetLastError(WSAEINVAL);
    return SOCKET_ERROR;
}

WSCEnumProtocols()

INT WSPAPI WSCEnumProtocols	(	IN LPINT	lpiProtocols,
		OUT LPWSAPROTOCOL_INFOW	lpProtocolBuffer,
		IN OUT LPDWORD	lpdwBufferLength,
		OUT LPINT	lpErrno
	)

Definition at line 165 of file enumprot.c.

{
    INT Status;
    PTCATALOG Catalog;
    PROTOCOL_ENUM_CONTEXT Context;
    DPRINT("WSCEnumProtocols: %p\n", lpiProtocols);
 
    /* Create a catalog object from the current one */
    Catalog = OpenInitializedCatalog();
    if (!Catalog)
    {
        /* Fail if we couldn't */
        *lpErrno = WSAENOBUFS;
        return SOCKET_ERROR;
    }
 
    /* Setup the context */
    Context.Protocols = lpiProtocols;
    Context.ProtocolBuffer = lpProtocolBuffer;
    Context.BufferLength = lpProtocolBuffer ? *lpdwBufferLength : 0;
    Context.BufferUsed = 0;
    Context.Count = 0;
    Context.ErrorCode = ERROR_SUCCESS;
 
    /* Enumerate the catalog */
    WsTcEnumerateCatalogItems(Catalog, ProtocolEnumerationProc, &Context);
 
    /* Get status */
    Status = Context.Count;
 
    /* Check the error code */
    if (Context.ErrorCode == ERROR_SUCCESS)
    {
        /* Check if enough space was available */
        if (Context.BufferLength < Context.BufferUsed)
        {
            /* Fail and tell them how much we need */
            *lpdwBufferLength = Context.BufferUsed;
            *lpErrno = WSAENOBUFS;
            Status = SOCKET_ERROR;
        }
    }
    else
    {
        /* Failure, normalize error */
        Status = SOCKET_ERROR;
        *lpErrno = Context.ErrorCode;
    }
 
    /* Delete the catalog object */
    WsTcDelete(Catalog);
 
    /* Return */
    return Status;
}

Referenced by WSAEnumProtocolsA(), and WSAEnumProtocolsW().

WSCGetProviderPath()

INT WSPAPI WSCGetProviderPath	(	IN LPGUID	lpProviderId,
		OUT LPWSTR	lpszProviderDllPath,
		IN OUT LPINT	lpProviderDllPathLen,
		OUT LPINT	lpErrno
	)

Definition at line 325 of file enumprot.c.

{
    PWSTHREAD Thread;
    PWSPROCESS Process;
    PNSCATALOG Catalog;
    INT ErrorCode, PathLen;
    PROVIDER_ENUM_CONTEXT Context;
 
    DPRINT("WSCGetProviderPath: %p %p %p %p\n", lpProviderId, lpszProviderDllPath, lpProviderDllPathLen, lpErrno);
 
    /* Enter prolog */
    if ((ErrorCode = WsApiProlog(&Process, &Thread)) != ERROR_SUCCESS)
    {
        /* FIXME: if WSANOTINITIALISED, we should init
         * and perform the search!
         */
 
        /* Leave now */
        *lpErrno = ErrorCode;
        return SOCKET_ERROR;
    }
 
    /* Get the catalog */
    Catalog = WsProcGetNsCatalog(Process);
 
    _SEH2_TRY
    {
        /* Setup the context */
        Context.ProviderId = *lpProviderId;
        Context.ProviderDllPath = lpszProviderDllPath;
        Context.ProviderDllPathLen = *lpProviderDllPathLen;
        Context.FoundPathLen = 0;
        Context.Found = 0;
        Context.ErrorCode = ERROR_SUCCESS;
 
        ErrorCode = ERROR_SUCCESS;
 
        /* Enumerate the catalog */
        WsNcEnumerateCatalogItems(Catalog, ProviderEnumerationProc, &Context);
 
        /* Check the error code */
        if (Context.ErrorCode == ERROR_SUCCESS)
        {
            /* Check if provider was found */
            if (Context.Found)
            {
                PathLen = Context.FoundPathLen;
 
                /* Check whether buffer is too small
                 * If it isn't, return length without null char
                 * (see ProviderEnumerationProc)
                 */
                if (Context.FoundPathLen <= *lpProviderDllPathLen)
                {
                    PathLen = Context.FoundPathLen - 1;
                }
                else
                {
                    ErrorCode = WSAEFAULT;
                }
 
                /* Set returned/required length */
                *lpProviderDllPathLen = PathLen;
            }
            else
            {
                ErrorCode = WSAEINVAL;
            }
        }
        else
        {
            ErrorCode = Context.ErrorCode;
        }
    }
    _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER)
    {
        ErrorCode = WSAEFAULT;
    }
    _SEH2_END;
 
    /* Do we have to return failure? */
    if (ErrorCode != ERROR_SUCCESS)
    {
        *lpErrno = ErrorCode;
        return SOCKET_ERROR;
    }
 
    return 0;
}

Referenced by WPUGetProviderPath().