lsputil.cpp File Reference

#include "instlsp.h"

Include dependency graph for lsputil.cpp:

Go to the source code of this file.

Functions
BOOL	RemoveIdFromChain (WSAPROTOCOL_INFOW *pInfo, DWORD dwCatalogId)
BOOL	IsIdInChain (WSAPROTOCOL_INFOW *pInfo, DWORD dwId)
int	GetProviderCount (WSAPROTOCOL_INFOW *pProviders, int iProviderCount, int iProviderType)
int	GetLayeredEntriesByGuid (WSAPROTOCOL_INFOW *pMatchLayers, int *iLayeredCount, WSAPROTOCOL_INFOW *pEntries, int iEntryCount, GUID *MatchGuid)
BOOL	IsEqualProtocolEntries (WSAPROTOCOL_INFOW *pInfo1, WSAPROTOCOL_INFOW *pInfo2)
int	RetrieveLspGuid (__in_z char *LspPath, GUID *Guid)
BOOL	IsNonIfsProvider (WSAPROTOCOL_INFOW *pProvider, int iProviderCount, DWORD dwProviderId)
HMODULE	LoadUpdateProviderFunction ()
int	CountOrphanedChainEntries (WSAPROTOCOL_INFOW *pCatalog, int iCatalogCount)
WSAPROTOCOL_INFOW *	FindProviderById (DWORD CatalogId, WSAPROTOCOL_INFOW *Catalog, int CatalogCount)
WSAPROTOCOL_INFOW *	FindProviderByGuid (GUID *Guid, WSAPROTOCOL_INFOW *Catalog, int CatalogCount)
DWORD	GetCatalogIdForProviderGuid (GUID *Guid, WSAPROTOCOL_INFOW *Catalog, int CatalogCount)
DWORD	FindDummyIdFromProtocolChainId (DWORD CatalogId, WSAPROTOCOL_INFOW *Catalog, int CatalogCount)
void	InsertIdIntoProtocolChain (WSAPROTOCOL_INFOW *Entry, int Index, DWORD InsertId)
void	BuildSubsetLspChain (WSAPROTOCOL_INFOW *Entry, int Index, DWORD DummyId)

---

Function Documentation

void BuildSubsetLspChain	(	WSAPROTOCOL_INFOW *	Entry,
		int	Index,
		DWORD	DummyId
	)

Definition at line 570 of file lsputil.cpp.

{
    int     Idx, i;

    for(i=Index,Idx=1; i < Entry->ProtocolChain.ChainLen ;i++,Idx++)
    {
        Entry->ProtocolChain.ChainEntries[ Idx ] = Entry->ProtocolChain.ChainEntries[ i ];
    }

    Entry->ProtocolChain.ChainEntries[ 0 ] = DummyId;
    Entry->ProtocolChain.ChainLen = Entry->ProtocolChain.ChainLen - Index + 1;
}

int CountOrphanedChainEntries	(	WSAPROTOCOL_INFOW *	pCatalog,
		int	iCatalogCount
	)

Definition at line 380 of file lsputil.cpp.

{
    int     orphanCount = 0,
            i, j;

    for(i=0; i < iCatalogCount ;i++)
    {
        if ( pCatalog[ i ].ProtocolChain.ChainLen > 1 )
        {
            for(j=0; j < iCatalogCount ;j++)
            {
                if ( i == j )
                    continue;
                if ( pCatalog[ j ].dwCatalogEntryId == pCatalog[ i ].ProtocolChain.ChainEntries[ 0 ] )
                {
                    break;
                }
            }
            if ( j >= iCatalogCount )
                orphanCount++;
        }
    }

    return orphanCount;
}

DWORD FindDummyIdFromProtocolChainId	(	DWORD	CatalogId,
		WSAPROTOCOL_INFOW *	Catalog,
		int	CatalogCount
	)

Definition at line 501 of file lsputil.cpp.

{
    int     i;

    for(i=0; i < CatalogCount ;i++)
    {
        if ( CatalogId == Catalog[ i ].dwCatalogEntryId )
        {
            if ( Catalog[ i ].ProtocolChain.ChainLen == LAYERED_PROTOCOL )
                return Catalog[ i ].dwCatalogEntryId;
            else
                return Catalog[ i ].ProtocolChain.ChainEntries[ 0 ];
        }
    }

    ASSERT( 0 );

    return 0;
}

WSAPROTOCOL_INFOW* FindProviderByGuid	(	GUID *	Guid,
		WSAPROTOCOL_INFOW *	Catalog,
		int	CatalogCount
	)

Definition at line 442 of file lsputil.cpp.

{
    int     i;

    for(i=0; i < CatalogCount ;i++)
    {
        if ( 0 == memcmp( &Catalog[ i ].ProviderId, Guid, sizeof( GUID ) ) )
        {
            return &Catalog[ i ];
        }
    }

    return NULL;
}

WSAPROTOCOL_INFOW* FindProviderById	(	DWORD	CatalogId,
		WSAPROTOCOL_INFOW *	Catalog,
		int	CatalogCount
	)

Definition at line 417 of file lsputil.cpp.

{
    int     i;

    for(i=0; i < CatalogCount ;i++)
    {
        if ( Catalog[ i ].dwCatalogEntryId == CatalogId )
            return &Catalog[ i ];
    }
    return NULL;
}

DWORD GetCatalogIdForProviderGuid	(	GUID *	Guid,
		WSAPROTOCOL_INFOW *	Catalog,
		int	CatalogCount
	)

Definition at line 469 of file lsputil.cpp.

{
    WSAPROTOCOL_INFOW *match = NULL;

    match = FindProviderByGuid( Guid, Catalog, CatalogCount );
    if ( NULL != match )
    {
        return match->dwCatalogEntryId;
    }

    return 0;
}

int GetLayeredEntriesByGuid	(	WSAPROTOCOL_INFOW *	pMatchLayers,
		int *	iLayeredCount,
		WSAPROTOCOL_INFOW *	pEntries,
		int	iEntryCount,
		GUID *	MatchGuid
	)

Definition at line 118 of file lsputil.cpp.

{
    int                count, 
                       err = SOCKET_ERROR,
                       i;

    // First count how many entries belong to this GUID
    count = 0;
    for(i=0; i < iEntryCount ;i++)
    {
        if ( 0 == memcmp( MatchGuid, &pEntries[i].ProviderId, sizeof( GUID ) ) )
            count++;
    }

    // Make sure the array passed in is large enough to hold the results
    if ( count > *iLayeredCount )
    {
        *iLayeredCount = count;
        goto cleanup;
    }

    // Go back and copy the matching providers into our array
    count = 0;
    for(i=0; i < iEntryCount ;i++)
    {
        if ( 0 == memcmp( MatchGuid, &pEntries[ i ].ProviderId, sizeof( GUID ) ) )
        {
            memcpy( &pMatchLayers[ count++ ], &pEntries[ i ], sizeof( WSAPROTOCOL_INFOW ) );
        }
    }

    *iLayeredCount = count;

    err = NO_ERROR;

cleanup:

    return err;
}

int GetProviderCount	(	WSAPROTOCOL_INFOW *	pProviders,
		int	iProviderCount,
		int	iProviderType
	)

Definition at line 83 of file lsputil.cpp.

{
    int Count, i;

    Count = 0;
    for(i=0; i < iProviderCount ;i++)
    {
        if ( ( LAYERED_CHAIN == iProviderType ) && ( pProviders[ i ].ProtocolChain.ChainLen > 1 ) )
            Count++;
        else if ( ( LAYERED_CHAIN != iProviderType) && ( pProviders[ i ].ProtocolChain.ChainLen == iProviderType ) )
            Count++;
    }
    return Count;
}

void InsertIdIntoProtocolChain	(	WSAPROTOCOL_INFOW *	Entry,
		int	Index,
		DWORD	InsertId
	)

Definition at line 535 of file lsputil.cpp.

{
    int     i;

    for(i=Entry->ProtocolChain.ChainLen; i > Index ;i--)
    {
        Entry->ProtocolChain.ChainEntries[ i ] = Entry->ProtocolChain.ChainEntries[ i - 1 ];
    }

    Entry->ProtocolChain.ChainEntries[ Index ] = InsertId;
    Entry->ProtocolChain.ChainLen++;
}

BOOL IsEqualProtocolEntries	(	WSAPROTOCOL_INFOW *	pInfo1,
		WSAPROTOCOL_INFOW *	pInfo2
	)

Definition at line 177 of file lsputil.cpp.

{
    if ( (memcmp(&pInfo1->ProviderId, &pInfo2->ProviderId, sizeof(GUID)) == 0) &&
         (pInfo1->dwServiceFlags1 == pInfo2->dwServiceFlags1) &&
         (pInfo1->dwServiceFlags2 == pInfo2->dwServiceFlags2) &&
         (pInfo1->dwServiceFlags3 == pInfo2->dwServiceFlags3) &&
         (pInfo1->dwServiceFlags4 == pInfo2->dwServiceFlags4) &&
         (pInfo1->ProtocolChain.ChainLen == pInfo2->ProtocolChain.ChainLen) &&
         (pInfo1->iVersion == pInfo2->iVersion) &&
         (pInfo1->iAddressFamily == pInfo2->iAddressFamily) &&
         (pInfo1->iMaxSockAddr == pInfo2->iMaxSockAddr) &&
         (pInfo1->iMinSockAddr == pInfo2->iMinSockAddr) &&
         (pInfo1->iSocketType == pInfo2->iSocketType) &&
         (pInfo1->iProtocol == pInfo2->iProtocol) &&
         (pInfo1->iProtocolMaxOffset == pInfo2->iProtocolMaxOffset) &&
         (pInfo1->iNetworkByteOrder == pInfo2->iNetworkByteOrder) &&
         (pInfo1->iSecurityScheme == pInfo2->iSecurityScheme) &&
         (pInfo1->dwMessageSize == pInfo2->dwMessageSize)
       )
    {
        return TRUE;
    }
    else
    {
        return FALSE;
    }
}

BOOL IsIdInChain	(	WSAPROTOCOL_INFOW *	pInfo,
		DWORD	dwId
	)

Definition at line 59 of file lsputil.cpp.

{
    int     i;

    for(i=0; i < pInfo->ProtocolChain.ChainLen ;i++)
    {
        if ( pInfo->ProtocolChain.ChainEntries[ i ] == dwId )
            return TRUE;
    }
    return FALSE;
}

BOOL IsNonIfsProvider	(	WSAPROTOCOL_INFOW *	pProvider,
		int	iProviderCount,
		DWORD	dwProviderId
	)

Definition at line 270 of file lsputil.cpp.

{
    int     i;

    for(i=0; i < iProviderCount ;i++)
    {
        if ( pProvider[ i ].dwCatalogEntryId == dwProviderId )
        {
            return !( pProvider[ i ].dwServiceFlags1 & XP1_IFS_HANDLES );
        }
    }
    
    return FALSE;
}

HMODULE LoadUpdateProviderFunction

(

)

Definition at line 302 of file lsputil.cpp.

{
    HMODULE hModule = NULL;
    HRESULT hr;
    char    WinsockLibraryPath[ MAX_PATH+1 ],
            szExpandPath[ MAX_PATH+1 ];

    //
    // See if we're on a platform that supports WSCUpdateProvider. If so then
    // uninstalling an LSP is easy; otherwise, it is very painful if you're
    // removing an LSP that has other LSPs on top if it.
    //
    if ( GetSystemDirectoryA( WinsockLibraryPath, MAX_PATH+1 ) == 0 )
    {
        hr = StringCchCopyA( szExpandPath, MAX_PATH+1, "%SYSTEMROOT%\\system32" );
        if ( FAILED( hr ) )
        {
            fprintf( stderr, "LoadUpdateProviderFunctions: StringCchCopyA failed: 0x%x\n", hr );
            goto cleanup;
        }

        if ( ExpandEnvironmentStringsA( WinsockLibraryPath, szExpandPath, MAX_PATH+1 ) == 0 )
        {
            fprintf(stderr, "LoadUpdateProviderFunctions: Unable to expand environment string: %d\n", 
                    GetLastError()
                   );
            goto cleanup;
        }
    }

    hr = StringCchCatA( WinsockLibraryPath, MAX_PATH+1, WINSOCK_DLL );
    if ( FAILED( hr ) )
    {
        fprintf( stderr, "LoadUpdateProviderFunctions: StringCchCatA failed: 0x%x\n", hr );
        goto cleanup;
    }

    hModule = LoadLibraryA( WinsockLibraryPath );
    if (hModule == NULL)
    {
        fprintf(stderr, "LoadUpdateProviderFunctions: Unable to load %s: %d\n", 
                WinsockLibraryPath, GetLastError()
                );
        goto cleanup;
    }
#ifdef _WIN64
    fnWscUpdateProvider   = (LPWSCUPDATEPROVIDER)GetProcAddress(hModule, "WSCUpdateProvider");

    fnWscUpdateProvider32 = (LPWSCUPDATEPROVIDER)GetProcAddress(hModule, "WSCUpdateProvider32");
#else
    fnWscUpdateProvider   = (LPWSCUPDATEPROVIDER)GetProcAddress(hModule, "WSCUpdateProvider");
#endif

    return hModule;

cleanup:

    if ( NULL != hModule )
    {
        FreeLibrary( hModule );
        hModule = NULL;
    }

    return NULL;
}

BOOL RemoveIdFromChain	(	WSAPROTOCOL_INFOW *	pInfo,
		DWORD	dwCatalogId
	)

Definition at line 27 of file lsputil.cpp.

{
    int     i, 
            j;

    for(i=0; i < pInfo->ProtocolChain.ChainLen ;i++)
    {
        if ( pInfo->ProtocolChain.ChainEntries[ i ] == dwCatalogId )
        {
            for(j=i; j < pInfo->ProtocolChain.ChainLen-1 ; j++)
            {
                pInfo->ProtocolChain.ChainEntries[ j ] = 
                        pInfo->ProtocolChain.ChainEntries[ j+1 ];
            }
            pInfo->ProtocolChain.ChainLen--;
            return TRUE;
        }
    }
    return FALSE;
}

int RetrieveLspGuid	(	__in_z char *	LspPath,
		GUID *	Guid
	)

Definition at line 218 of file lsputil.cpp.

{
    HMODULE         hMod = NULL;
    LPFN_GETLSPGUID fnGetLspGuid = NULL;
    int             retval = SOCKET_ERROR;

    // Load teh library
    hMod = LoadLibraryA( LspPath );
    if ( NULL == hMod )
    {
        fprintf( stderr, "RetrieveLspGuid: LoadLibraryA failed: %d\n", GetLastError() );
        goto cleanup;
    }

    // Get a pointer to the LSPs GetLspGuid function
    fnGetLspGuid = (LPFN_GETLSPGUID) GetProcAddress( hMod, "GetLspGuid" );
    if ( NULL == fnGetLspGuid )
    {
        fprintf( stderr, "RetrieveLspGuid: GetProcAddress failed: %d\n", GetLastError() );
        goto cleanup;
    }

    // Retrieve the LSPs GUID
    fnGetLspGuid( Guid );

    retval = NO_ERROR;

cleanup:

    if ( NULL != hMod )
        FreeLibrary( hMod );

    return retval;
}