435 lines
9.9 KiB
C
435 lines
9.9 KiB
C
|
/*++
|
||
|
|
|
||
|
|
Copyright (c) 1995 Microsoft Corporation
|
||
|
|
|
||
|
|
Module Name:
|
||
|
|
|
||
|
|
|
||
|
|
Abstract:
|
||
|
|
|
||
|
|
Revision History:
|
||
|
|
|
||
|
|
Amritansh Raghav
|
||
|
|
|
||
|
|
--*/
|
||
|
|
#include "inc.h"
|
||
|
|
#pragma hdrstop
|
||
|
|
|
||
|
|
int
|
||
|
|
TCPQueryInformationEx(
|
||
|
|
DWORD Family,
|
||
|
|
void *InBuf,
|
||
|
|
ulong *InBufLen,
|
||
|
|
void *OutBuf,
|
||
|
|
ulong *OutBufLen
|
||
|
|
);
|
||
|
|
|
||
|
|
int
|
||
|
|
TCPSetInformationEx(
|
||
|
|
void *InBuf,
|
||
|
|
ULONG *InBufLen,
|
||
|
|
void *OutBuf,
|
||
|
|
ULONG *OutBufLen
|
||
|
|
);
|
||
|
|
|
||
|
|
TDIEntityID*
|
||
|
|
GetTdiEntityCount(
|
||
|
|
PULONG pulNumEntities
|
||
|
|
);
|
||
|
|
|
||
|
|
|
||
|
|
VOID
|
||
|
|
InitAdapterMappingTable(VOID)
|
||
|
|
{
|
||
|
|
DWORD i;
|
||
|
|
|
||
|
|
for(i = 0; i < MAP_HASH_SIZE; i++)
|
||
|
|
{
|
||
|
|
InitializeListHead(&g_pAdapterMappingTable[i]);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
VOID
|
||
|
|
UnInitAdapterMappingTable(VOID)
|
||
|
|
{
|
||
|
|
DWORD i;
|
||
|
|
|
||
|
|
for(i = 0; i < MAP_HASH_SIZE; i++)
|
||
|
|
{
|
||
|
|
PLIST_ENTRY pleHead = &g_pAdapterMappingTable[i];
|
||
|
|
|
||
|
|
while(!IsListEmpty(pleHead))
|
||
|
|
{
|
||
|
|
PLIST_ENTRY pleNode;
|
||
|
|
LPAIHASH lpAIBlock;
|
||
|
|
|
||
|
|
pleNode = RemoveHeadList(pleHead);
|
||
|
|
lpAIBlock = CONTAINING_RECORD(pleNode,AIHASH,leList);
|
||
|
|
|
||
|
|
HeapFree(g_hPrivateHeap,0,lpAIBlock);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
DWORD
|
||
|
|
StoreAdapterToATInstanceMap(
|
||
|
|
DWORD dwAdapterIndex,
|
||
|
|
DWORD dwATInst
|
||
|
|
)
|
||
|
|
{
|
||
|
|
LPAIHASH lpAIBlock;
|
||
|
|
|
||
|
|
if((lpAIBlock = LookUpAdapterMap(dwAdapterIndex)) isnot NULL)
|
||
|
|
{
|
||
|
|
lpAIBlock->dwATInstance = dwATInst;
|
||
|
|
return NO_ERROR;
|
||
|
|
}
|
||
|
|
|
||
|
|
//
|
||
|
|
// Wasnt found
|
||
|
|
//
|
||
|
|
|
||
|
|
if((lpAIBlock = HeapAlloc(g_hPrivateHeap,0,sizeof(AIHASH))) is NULL)
|
||
|
|
{
|
||
|
|
return ERROR_NOT_ENOUGH_MEMORY;
|
||
|
|
}
|
||
|
|
|
||
|
|
//
|
||
|
|
// New one so make everything else unmapped to avoid nasty surprises
|
||
|
|
//
|
||
|
|
|
||
|
|
lpAIBlock->dwAdapterIndex = dwAdapterIndex;
|
||
|
|
lpAIBlock->dwATInstance = dwATInst;
|
||
|
|
lpAIBlock->dwIFInstance = INVALID_IF_INSTANCE;
|
||
|
|
|
||
|
|
InsertAdapterMap(lpAIBlock);
|
||
|
|
|
||
|
|
return NO_ERROR;
|
||
|
|
}
|
||
|
|
|
||
|
|
DWORD
|
||
|
|
StoreAdapterToIFInstanceMap(
|
||
|
|
DWORD dwAdapterIndex,
|
||
|
|
DWORD dwIFInst
|
||
|
|
)
|
||
|
|
{
|
||
|
|
LPAIHASH lpAIBlock;
|
||
|
|
|
||
|
|
if((lpAIBlock = LookUpAdapterMap(dwAdapterIndex)) isnot NULL)
|
||
|
|
{
|
||
|
|
lpAIBlock->dwIFInstance = dwIFInst;
|
||
|
|
return NO_ERROR;
|
||
|
|
}
|
||
|
|
|
||
|
|
//
|
||
|
|
// Wasnt found
|
||
|
|
//
|
||
|
|
|
||
|
|
if((lpAIBlock = HeapAlloc(g_hPrivateHeap,0,sizeof(AIHASH))) is NULL)
|
||
|
|
{
|
||
|
|
return ERROR_NOT_ENOUGH_MEMORY;
|
||
|
|
}
|
||
|
|
|
||
|
|
//
|
||
|
|
// New one so make everything else unmapped to avoid nasty surprises
|
||
|
|
//
|
||
|
|
|
||
|
|
lpAIBlock->dwAdapterIndex = dwAdapterIndex;
|
||
|
|
lpAIBlock->dwATInstance = INVALID_AT_INSTANCE;
|
||
|
|
lpAIBlock->dwIFInstance = dwIFInst;
|
||
|
|
|
||
|
|
InsertAdapterMap(lpAIBlock);
|
||
|
|
|
||
|
|
return NO_ERROR;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
DWORD
|
||
|
|
GetIFInstanceFromAdapter(
|
||
|
|
DWORD dwAdapterIndex
|
||
|
|
)
|
||
|
|
{
|
||
|
|
LPAIHASH lpAIBlock;
|
||
|
|
|
||
|
|
if((lpAIBlock = LookUpAdapterMap(dwAdapterIndex)) isnot NULL)
|
||
|
|
{
|
||
|
|
return lpAIBlock->dwIFInstance;
|
||
|
|
}
|
||
|
|
|
||
|
|
return INVALID_IF_INSTANCE;
|
||
|
|
}
|
||
|
|
|
||
|
|
DWORD
|
||
|
|
GetATInstanceFromAdapter(
|
||
|
|
DWORD dwAdapterIndex
|
||
|
|
)
|
||
|
|
{
|
||
|
|
LPAIHASH lpAIBlock;
|
||
|
|
|
||
|
|
if((lpAIBlock = LookUpAdapterMap(dwAdapterIndex)) isnot NULL)
|
||
|
|
{
|
||
|
|
return lpAIBlock->dwATInstance;
|
||
|
|
}
|
||
|
|
|
||
|
|
return INVALID_IF_INSTANCE;
|
||
|
|
}
|
||
|
|
|
||
|
|
LPAIHASH
|
||
|
|
LookUpAdapterMap(
|
||
|
|
DWORD dwAdapterIndex
|
||
|
|
)
|
||
|
|
{
|
||
|
|
DWORD dwHashIndex;
|
||
|
|
LPAIHASH lpAIBlock;
|
||
|
|
PLIST_ENTRY lpleCurrent;
|
||
|
|
|
||
|
|
dwHashIndex = dwAdapterIndex % MAP_HASH_SIZE;
|
||
|
|
|
||
|
|
//
|
||
|
|
// The list is not ordered, travel the whole hash bucket
|
||
|
|
//
|
||
|
|
|
||
|
|
for(lpleCurrent = g_pAdapterMappingTable[dwHashIndex].Flink;
|
||
|
|
lpleCurrent isnot &g_pAdapterMappingTable[dwHashIndex];
|
||
|
|
lpleCurrent = lpleCurrent->Flink)
|
||
|
|
{
|
||
|
|
lpAIBlock = CONTAINING_RECORD(lpleCurrent,AIHASH,leList);
|
||
|
|
if(lpAIBlock->dwAdapterIndex is dwAdapterIndex)
|
||
|
|
{
|
||
|
|
return lpAIBlock;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return NULL;
|
||
|
|
}
|
||
|
|
|
||
|
|
VOID
|
||
|
|
InsertAdapterMap(
|
||
|
|
LPAIHASH lpaiBlock
|
||
|
|
)
|
||
|
|
{
|
||
|
|
DWORD dwHashIndex;
|
||
|
|
|
||
|
|
dwHashIndex = lpaiBlock->dwAdapterIndex % MAP_HASH_SIZE;
|
||
|
|
InsertHeadList(&g_pAdapterMappingTable[dwHashIndex],&lpaiBlock->leList);
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
DWORD
|
||
|
|
UpdateAdapterToIFInstanceMapping(
|
||
|
|
VOID
|
||
|
|
)
|
||
|
|
{
|
||
|
|
DWORD dwResult;
|
||
|
|
DWORD dwOutBufLen;
|
||
|
|
DWORD dwInBufLen;
|
||
|
|
TCP_REQUEST_QUERY_INFORMATION_EX trqiInBuf;
|
||
|
|
TDIObjectID *ID;
|
||
|
|
BYTE *Context;
|
||
|
|
MIB_IFROW maxIfEntry;
|
||
|
|
DWORD i;
|
||
|
|
ULONG ulNumEntities;
|
||
|
|
TDIEntityID *entityList;
|
||
|
|
TDIEntityID *pEntity = NULL;
|
||
|
|
|
||
|
|
entityList = GetTdiEntityCount(&ulNumEntities);
|
||
|
|
|
||
|
|
if(entityList == NULL)
|
||
|
|
{
|
||
|
|
Trace0(ERR,
|
||
|
|
"UpdateAdapterToIFInstanceMapping: Couldnt get num entities\n");
|
||
|
|
|
||
|
|
return ERROR_GEN_FAILURE;
|
||
|
|
}
|
||
|
|
|
||
|
|
dwInBufLen = sizeof(TCP_REQUEST_QUERY_INFORMATION_EX);
|
||
|
|
ID = &(trqiInBuf.ID);
|
||
|
|
Context = (BYTE *) &(trqiInBuf.Context[0]);
|
||
|
|
ID->toi_entity.tei_entity = IF_ENTITY;
|
||
|
|
ID->toi_class = INFO_CLASS_PROTOCOL;
|
||
|
|
ID->toi_type = INFO_TYPE_PROVIDER;
|
||
|
|
ID->toi_id = IF_MIB_STATS_ID;
|
||
|
|
|
||
|
|
//
|
||
|
|
// First go through and invalidate all the mappings
|
||
|
|
//
|
||
|
|
|
||
|
|
for(i = 0; i < MAP_HASH_SIZE; i ++)
|
||
|
|
{
|
||
|
|
PLIST_ENTRY pCurrentNode;
|
||
|
|
LPAIHASH lpAIBlock;
|
||
|
|
|
||
|
|
for(pCurrentNode = g_pAdapterMappingTable[i].Flink;
|
||
|
|
pCurrentNode isnot &(g_pAdapterMappingTable[i]);
|
||
|
|
pCurrentNode = pCurrentNode->Flink)
|
||
|
|
{
|
||
|
|
lpAIBlock = CONTAINING_RECORD(pCurrentNode,AIHASH,leList);
|
||
|
|
|
||
|
|
lpAIBlock->dwIFInstance = INVALID_IF_INSTANCE;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
//
|
||
|
|
// Read the interface entry items
|
||
|
|
//
|
||
|
|
|
||
|
|
for ( i = 0, pEntity = entityList; i < ulNumEntities ; ++i, ++pEntity)
|
||
|
|
{
|
||
|
|
if (pEntity->tei_entity == IF_ENTITY)
|
||
|
|
{
|
||
|
|
dwOutBufLen = sizeof(MIB_IFROW) - FIELD_OFFSET(MIB_IFROW, dwIndex);
|
||
|
|
|
||
|
|
ID->toi_entity.tei_instance = pEntity->tei_instance;
|
||
|
|
|
||
|
|
ZeroMemory(Context,CONTEXT_SIZE);
|
||
|
|
|
||
|
|
dwResult = TCPQueryInformationEx(AF_INET,
|
||
|
|
&trqiInBuf,
|
||
|
|
&dwInBufLen,
|
||
|
|
(LPVOID)&(maxIfEntry.dwIndex),
|
||
|
|
&dwOutBufLen );
|
||
|
|
if (dwResult isnot NO_ERROR)
|
||
|
|
{
|
||
|
|
Trace1(ERR,
|
||
|
|
"Error %x querying information from stack. Continuing",
|
||
|
|
dwResult);
|
||
|
|
|
||
|
|
continue;
|
||
|
|
}
|
||
|
|
|
||
|
|
StoreAdapterToIFInstanceMap(maxIfEntry.dwIndex,
|
||
|
|
pEntity->tei_instance);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
//
|
||
|
|
// Now delete the blocks that have invalid mappings
|
||
|
|
//
|
||
|
|
|
||
|
|
for(i = 0; i < MAP_HASH_SIZE; i ++)
|
||
|
|
{
|
||
|
|
PLIST_ENTRY pCurrentNode;
|
||
|
|
LPAIHASH lpAIBlock;
|
||
|
|
|
||
|
|
pCurrentNode = g_pAdapterMappingTable[i].Flink;
|
||
|
|
|
||
|
|
while(pCurrentNode isnot &(g_pAdapterMappingTable[i]))
|
||
|
|
{
|
||
|
|
PLIST_ENTRY pTempNode;
|
||
|
|
|
||
|
|
pTempNode = pCurrentNode;
|
||
|
|
|
||
|
|
pCurrentNode = pCurrentNode->Flink;
|
||
|
|
|
||
|
|
lpAIBlock = CONTAINING_RECORD(pTempNode,
|
||
|
|
AIHASH,
|
||
|
|
leList);
|
||
|
|
|
||
|
|
if(lpAIBlock->dwIFInstance is INVALID_IF_INSTANCE)
|
||
|
|
{
|
||
|
|
//
|
||
|
|
// Delete it
|
||
|
|
//
|
||
|
|
|
||
|
|
RemoveEntryList(&(lpAIBlock->leList));
|
||
|
|
|
||
|
|
HeapFree(g_hPrivateHeap,
|
||
|
|
0,
|
||
|
|
lpAIBlock);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
g_dwLastIfUpdateTime = GetCurrentTime();
|
||
|
|
|
||
|
|
HeapFree(g_hPrivateHeap,0, entityList);
|
||
|
|
|
||
|
|
return NO_ERROR;
|
||
|
|
}
|
||
|
|
|
||
|
|
DWORD
|
||
|
|
UpdateAdapterToATInstanceMapping(
|
||
|
|
VOID
|
||
|
|
)
|
||
|
|
{
|
||
|
|
DWORD dwResult;
|
||
|
|
DWORD dwInBufLen;
|
||
|
|
DWORD i;
|
||
|
|
TCP_REQUEST_QUERY_INFORMATION_EX trqiInBuf;
|
||
|
|
TDIObjectID *ID;
|
||
|
|
UCHAR *Context;
|
||
|
|
DWORD dwSize;
|
||
|
|
AddrXlatInfo AXI;
|
||
|
|
|
||
|
|
//
|
||
|
|
// When we update the mapping, we also update the ArpEntities
|
||
|
|
// Infact we use the same function to do both
|
||
|
|
//
|
||
|
|
|
||
|
|
if(g_pdwArpEntTable)
|
||
|
|
{
|
||
|
|
HeapFree(g_hPrivateHeap,
|
||
|
|
HEAP_NO_SERIALIZE,
|
||
|
|
g_pdwArpEntTable);
|
||
|
|
|
||
|
|
g_pdwArpEntTable = NULL;
|
||
|
|
}
|
||
|
|
|
||
|
|
dwResult = AllocateAndGetArpEntTableFromStack(&g_pdwArpEntTable,
|
||
|
|
&g_dwNumArpEntEntries,
|
||
|
|
g_hPrivateHeap,
|
||
|
|
0,
|
||
|
|
0);
|
||
|
|
|
||
|
|
if(dwResult isnot NO_ERROR)
|
||
|
|
{
|
||
|
|
Trace1(ERR,
|
||
|
|
"UpdateAdapterToATInstanceMapping: Couldnt get ArpEntTable. Error %d",
|
||
|
|
dwResult);
|
||
|
|
|
||
|
|
g_pdwArpEntTable = NULL;
|
||
|
|
|
||
|
|
return ERROR_CAN_NOT_COMPLETE;
|
||
|
|
}
|
||
|
|
|
||
|
|
dwInBufLen = sizeof( TCP_REQUEST_QUERY_INFORMATION_EX );
|
||
|
|
|
||
|
|
Context = (BYTE *) &(trqiInBuf.Context[0]);
|
||
|
|
ID = &(trqiInBuf.ID);
|
||
|
|
|
||
|
|
for (i = 0; i < g_dwNumArpEntEntries; i++ )
|
||
|
|
{
|
||
|
|
|
||
|
|
ID->toi_entity.tei_entity = AT_ENTITY;
|
||
|
|
ID->toi_type = INFO_TYPE_PROVIDER;
|
||
|
|
ID->toi_class = INFO_CLASS_PROTOCOL;
|
||
|
|
ID->toi_id = AT_MIB_ADDRXLAT_INFO_ID;
|
||
|
|
ID->toi_entity.tei_instance = g_pdwArpEntTable[i];
|
||
|
|
|
||
|
|
dwSize = sizeof(AXI);
|
||
|
|
ZeroMemory(Context, CONTEXT_SIZE);
|
||
|
|
|
||
|
|
dwResult = TCPQueryInformationEx(AF_INET,
|
||
|
|
&trqiInBuf,
|
||
|
|
&dwInBufLen,
|
||
|
|
&AXI,
|
||
|
|
&dwSize );
|
||
|
|
|
||
|
|
if(dwResult isnot NO_ERROR)
|
||
|
|
{
|
||
|
|
Trace1(ERR,
|
||
|
|
"UpdateAdapterToATInstanceMapping: Error %x querying stack",
|
||
|
|
dwResult);
|
||
|
|
|
||
|
|
return dwResult;
|
||
|
|
}
|
||
|
|
|
||
|
|
StoreAdapterToATInstanceMap(AXI.axi_index,g_pdwArpEntTable[i]);
|
||
|
|
}
|
||
|
|
|
||
|
|
g_dwLastArpUpdateTime = GetCurrentTime();
|
||
|
|
|
||
|
|
return NO_ERROR;
|
||
|
|
}
|