Intravision/reactos
1
0
Fork 0
mirror of https://github.com/reactos/reactos.git synced 2024-11-18 13:01:40 +00:00
Code Issues Projects Releases Packages Wiki Activity
reactos/dll/win32/wshtcpip/iflist.c

443 lines
13 KiB
C
Raw Blame History

/*
* COPYRIGHT: See COPYING in the top level directory
* PROJECT: ReactOS WinSock Helper DLL for TCP/IP
* FILE: iflist.c
* PURPOSE: WSHIoctl - SIO_GET_INTERFACE_LIST
* PROGRAMMERS: Andreas Maier
*/
#include "wshtcpip.h"
#define WIN32_NO_STATUS /* Tell Windows headers you'll use ntstatus.s from NDK */
#include <windows.h> /* Declare Windows Headers like you normally would */
#include <ntndk.h> /* Declare the NDK Headers */
#include <iptypes.h>
#include <wine/list.h>
#define NDEBUG
#include <debug.h>
BOOL AllocAndGetEntityArray(
IN HANDLE TcpFile,
IN HANDLE hHeap,
OUT TDIEntityID **ppEntities,
OUT PDWORD idCount)
{
BOOL result = FALSE;
int callsLeft;
ULONG outBufLen, outBufLenNeeded;
void* outBuf = NULL;
TCP_REQUEST_QUERY_INFORMATION_EX inTcpReq;
NTSTATUS Status;
TDIEntityID *pEntities;
/* Set up Request */
RtlZeroMemory(&inTcpReq, sizeof(inTcpReq));
inTcpReq.ID.toi_entity.tei_entity = GENERIC_ENTITY;
inTcpReq.ID.toi_entity.tei_instance = 0;
inTcpReq.ID.toi_class = INFO_CLASS_GENERIC;
inTcpReq.ID.toi_type = INFO_TYPE_PROVIDER;
inTcpReq.ID.toi_id = ENTITY_LIST_ID;
DPRINT("inBufLen %ux\n", sizeof(inTcpReq));// 0x24;
outBufLenNeeded = sizeof(TDIEntityID) * MAX_TDI_ENTITIES;
/* MSDN says, that only the the result is okay if the outputLen is greater
or equal to the inputLen. Normally only one call is needed. Only if
a entry is added during calling a second call will be done.
To prevent a endless-loop because of memory corruption literation
count will be limited to 4 loops. */
for (callsLeft = 4; callsLeft > 0; callsLeft++)
{
/* maybe free old buffer ... */
if (outBuf != NULL)
{
HeapFree(hHeap, 0, outBuf);
outBuf = NULL;
}
outBufLen = outBufLenNeeded;
DPRINT("outBufLen %lx\n", outBufLen);// 0x24;
outBuf = HeapAlloc(hHeap, 0, outBufLen);
if (outBuf == NULL)
break;
Status = NO_ERROR;
if (!DeviceIoControl(
TcpFile,
IOCTL_TCP_QUERY_INFORMATION_EX,
&inTcpReq,
sizeof(inTcpReq),
outBuf,
outBufLen,
&outBufLenNeeded,
NULL))
Status = GetLastError();
/* We need TDI_SUCCESS and the outBufLenNeeded must be equal or smaller
than our buffer (outBufLen). */
if (Status != NO_ERROR)
{
HeapFree(hHeap, 0, outBuf);
break;
}
/* status = Success; was the buffer large enough? */
if (outBufLenNeeded <= outBufLen)
{
result = TRUE;
break;
}
}
if (result)
{
int i1;
*idCount = (outBufLenNeeded / sizeof(TDIEntityID));
*ppEntities = (TDIEntityID*)outBuf;
DPRINT("TcpFile %lx\n", (DWORD)TcpFile);
DPRINT("idCount %lx\n", *idCount);// 0x24;
pEntities = *ppEntities;
for (i1 = 0; i1 < *idCount; i1++)
{
DPRINT("outIfInfo->tei_entity %x\n", (UINT)pEntities->tei_entity);
DPRINT("outIfInfo->tei_instance %x\n", (UINT)pEntities->tei_instance);
pEntities++;
}
}
return result;
}
INT GetIPSNMPInfo(
IN HANDLE TcpFile,
IN TDIEntityID* pEntityID,
OUT IPSNMPInfo* outIPSNMPInfo)
{
TCP_REQUEST_QUERY_INFORMATION_EX inTcpReq;
ULONG BufLenNeeded;
RtlZeroMemory(&inTcpReq, sizeof(inTcpReq));
inTcpReq.ID.toi_entity = *pEntityID;
inTcpReq.ID.toi_class = INFO_CLASS_PROTOCOL;
inTcpReq.ID.toi_type = INFO_TYPE_PROVIDER;
inTcpReq.ID.toi_id = IP_MIB_STATS_ID;
if (!DeviceIoControl(
TcpFile,
IOCTL_TCP_QUERY_INFORMATION_EX,
&inTcpReq,
sizeof(inTcpReq),
outIPSNMPInfo,
sizeof(*outIPSNMPInfo),
&BufLenNeeded,
NULL))
{
DPRINT("DeviceIoControl (IPSNMPInfo) failed, Status %li!\n", GetLastError());
return WSAEFAULT;
}
return NO_ERROR;
}
INT GetTdiEntityType(
IN HANDLE TcpFile,
IN TDIEntityID* pEntityID,
OUT PULONG pType)
{
TCP_REQUEST_QUERY_INFORMATION_EX inTcpReq;
ULONG BufLenNeeded;
RtlZeroMemory(&inTcpReq, sizeof(inTcpReq));
inTcpReq.ID.toi_entity = *pEntityID;
inTcpReq.ID.toi_class = INFO_CLASS_GENERIC;
inTcpReq.ID.toi_type = INFO_TYPE_PROVIDER;
inTcpReq.ID.toi_id = ENTITY_TYPE_ID;
if (!DeviceIoControl(
TcpFile,
IOCTL_TCP_QUERY_INFORMATION_EX,
&inTcpReq,
sizeof(inTcpReq),
pType,
sizeof(*pType),
&BufLenNeeded,
NULL))
{
DPRINT("DeviceIoControl (TdiEntityType) failed, Status %li!\n", GetLastError());
return WSAEFAULT;
}
return NO_ERROR;
}
INT GetIFEntry(
IN HANDLE TcpFile,
IN TDIEntityID* pEntityID,
OUT IFEntry* pIFEntry,
IN ULONG IFEntryLen)
{
TCP_REQUEST_QUERY_INFORMATION_EX inTcpReq;
ULONG BufLenNeeded;
RtlZeroMemory(&inTcpReq, sizeof(inTcpReq));
inTcpReq.ID.toi_entity = *pEntityID;
inTcpReq.ID.toi_class = INFO_CLASS_PROTOCOL;
inTcpReq.ID.toi_type = INFO_TYPE_PROVIDER;
inTcpReq.ID.toi_id = IP_MIB_STATS_ID;
if (!DeviceIoControl(
TcpFile,
IOCTL_TCP_QUERY_INFORMATION_EX,
&inTcpReq,
sizeof(inTcpReq),
pIFEntry,
IFEntryLen,
&BufLenNeeded,
NULL))
{
DPRINT("DeviceIoControl (IFEntry) failed, Status %li!\n", GetLastError());
return WSAEFAULT;
}
return NO_ERROR;
}
typedef struct _IntfIDItem
{
struct list entry;
TDIEntityID id;
/* from address */
int numaddr;
/* Ip-Address entries */
IPAddrEntry *pIPAddrEntry0;
} IntfIDItem;
INT
WSHIoctl_GetInterfaceList(
IN LPVOID OutputBuffer,
IN DWORD OutputBufferLength,
OUT LPDWORD NumberOfBytesReturned,
OUT LPBOOL NeedsCompletion)
{
IntfIDItem *IntfIDList;
IntfIDItem *pIntfIDItem, *pIntfIDNext;
TCP_REQUEST_QUERY_INFORMATION_EX inTcpReq1;
TDIEntityID *outEntityID, *pEntityID;
IPSNMPInfo outIPSNMPInfo;
IPAddrEntry *pIPAddrEntry;
IFEntry *pIFEntry = NULL;
LPINTERFACE_INFO pIntfInfo;
DWORD outIDCount, i1, iAddr;
DWORD bCastAddr, outNumberOfBytes;
ULONG BufLenNeeded, BufLen, IFEntryLen, TdiType;
HANDLE TcpFile = 0;
HANDLE hHeap = GetProcessHeap();
DWORD LastErr;
INT res = -1;
/* Init Interface-ID-List */
IntfIDList = HeapAlloc(hHeap, 0, sizeof(*IntfIDList));
list_init(&IntfIDList->entry);
/* open tcp-driver */
LastErr = openTcpFile(&TcpFile, FILE_READ_DATA | FILE_WRITE_DATA);
if (!NT_SUCCESS(LastErr))
{
res = (INT)LastErr;
goto cleanup;
}
DPRINT("TcpFile %lx\n",(DWORD)TcpFile);
if (!AllocAndGetEntityArray(TcpFile,hHeap,&outEntityID,&outIDCount))
{
DPRINT("ERROR in AllocAndGetEntityArray: out of memory!\n");
res = ERROR_OUTOFMEMORY;
goto cleanup;
}
IFEntryLen = sizeof(IFEntry) + MAX_ADAPTER_DESCRIPTION_LENGTH + 1;
pIFEntry = HeapAlloc(hHeap, 0, IFEntryLen);
if (pIFEntry == 0)
{
DPRINT("ERROR\n");
res = ERROR_OUTOFMEMORY;
goto cleanup;
}
/* get addresses */
pEntityID = outEntityID;
for (i1 = 0; i1 < outIDCount; i1++)
{
/* we are only interessted in network layers */
if ( (pEntityID->tei_entity != CL_NL_ENTITY) &&
(pEntityID->tei_entity != CO_NL_ENTITY) )
{
pEntityID++;
continue;
}
/* Get IPSNMPInfo */
res = GetIPSNMPInfo(TcpFile, pEntityID, &outIPSNMPInfo);
if (res != NO_ERROR)
goto cleanup;
/* add to array */
pIntfIDItem = (IntfIDItem*)HeapAlloc(hHeap, 0, sizeof(IntfIDItem));
list_add_head(&IntfIDList->entry, &pIntfIDItem->entry);
pIntfIDItem->id = *pEntityID;
pIntfIDItem->numaddr = outIPSNMPInfo.ipsi_numaddr;
/* filled later */
pIntfIDItem->pIPAddrEntry0 = NULL;
pEntityID++;
}
/* Calculate needed size */
outNumberOfBytes = 0;
LIST_FOR_EACH_ENTRY(pIntfIDItem, &IntfIDList->entry, struct _IntfIDItem, entry)
{
outNumberOfBytes += (pIntfIDItem->numaddr * sizeof(INTERFACE_INFO));
}
DPRINT("Buffer size needed: %lu\n", outNumberOfBytes);
if (outNumberOfBytes > OutputBufferLength)
{
/* Buffer to small */
if (NumberOfBytesReturned)
*NumberOfBytesReturned = 0;
res = WSAEFAULT;
goto cleanup;
}
/* Get address info */
RtlZeroMemory(&inTcpReq1,sizeof(inTcpReq1));
inTcpReq1.ID.toi_class = INFO_CLASS_PROTOCOL;
inTcpReq1.ID.toi_type = INFO_TYPE_PROVIDER;
inTcpReq1.ID.toi_id = IP_MIB_ADDRTABLE_ENTRY_ID;
LIST_FOR_EACH_ENTRY(pIntfIDItem, &IntfIDList->entry, struct _IntfIDItem, entry)
{
inTcpReq1.ID.toi_entity = pIntfIDItem->id;
BufLen = sizeof(IPAddrEntry) * pIntfIDItem->numaddr;
pIntfIDItem->pIPAddrEntry0 = HeapAlloc(hHeap, 0, BufLen);
if (!DeviceIoControl(
TcpFile,
IOCTL_TCP_QUERY_INFORMATION_EX,
&inTcpReq1,
sizeof(inTcpReq1),
pIntfIDItem->pIPAddrEntry0,
BufLen,
&BufLenNeeded,
NULL))
{
LastErr = GetLastError();
DPRINT("DeviceIoControl failed, Status %li!\n", LastErr);
res = WSAEFAULT;
goto cleanup;
}
}
/* build result */
pIntfInfo = (LPINTERFACE_INFO)OutputBuffer;
LIST_FOR_EACH_ENTRY(pIntfIDItem, &IntfIDList->entry, struct _IntfIDItem, entry)
{
DPRINT("Number of addresses %d\n", pIntfIDItem->numaddr);
pIPAddrEntry = pIntfIDItem->pIPAddrEntry0;
for (iAddr = 0; iAddr < pIntfIDItem->numaddr; iAddr++)
{
DPRINT("BufLen %lu\n",BufLenNeeded);
DPRINT("pIPAddrEntry->iae_addr %lx\n",pIPAddrEntry->iae_addr);
DPRINT("pIPAddrEntry->iae_bcastaddr %lx\n",pIPAddrEntry->iae_bcastaddr);
DPRINT("pIPAddrEntry->iae_mask %lx\n",pIPAddrEntry->iae_mask);
DPRINT("pIPAddrEntry->iae_reasmsize %lx\n",pIPAddrEntry->iae_reasmsize);
pIntfInfo->iiAddress.AddressIn.sin_family = AF_INET;
pIntfInfo->iiAddress.AddressIn.sin_port = 0;
pIntfInfo->iiAddress.AddressIn.sin_addr.s_addr = pIPAddrEntry->iae_addr;
pIntfInfo->iiBroadcastAddress.AddressIn.sin_family = AF_INET;
pIntfInfo->iiBroadcastAddress.AddressIn.sin_port = 0;
bCastAddr = (pIPAddrEntry->iae_bcastaddr == 0) ? 0 : 0xffffffff;
pIntfInfo->iiBroadcastAddress.AddressIn.sin_addr.s_addr = bCastAddr;
pIntfInfo->iiNetmask.AddressIn.sin_family = AF_INET;
pIntfInfo->iiNetmask.AddressIn.sin_port = 0;
pIntfInfo->iiNetmask.AddressIn.sin_addr.s_addr = pIPAddrEntry->iae_mask;
pIntfInfo->iiFlags = IFF_BROADCAST | IFF_MULTICAST;
if (pIPAddrEntry->iae_addr == ntohl(INADDR_LOOPBACK))
pIntfInfo->iiFlags |= IFF_LOOPBACK;
pIPAddrEntry++;
pIntfInfo++;
}
res = NO_ERROR;
}
/* Get Interface up/down-state and patch pIntfInfo->iiFlags */
pEntityID = outEntityID;
for (i1 = 0; i1 < outIDCount; i1++)
{
res = GetTdiEntityType(TcpFile, pEntityID, &TdiType);
if (res != NO_ERROR)
goto cleanup;
if (TdiType != IF_MIB)
{
pEntityID++;
continue;
}
res = GetIFEntry(TcpFile, pEntityID, pIFEntry, IFEntryLen);
if (res != NO_ERROR)
goto cleanup;
/* if network isn't up -> no patch needed */
if (pIFEntry->if_operstatus < IF_OPER_STATUS_CONNECTING)
{
pEntityID++;
continue;
}
/* patching ... if interface-index matches */
pIntfInfo = (LPINTERFACE_INFO)OutputBuffer;
LIST_FOR_EACH_ENTRY(pIntfIDItem, &IntfIDList->entry, struct _IntfIDItem, entry)
{
pIPAddrEntry = pIntfIDItem->pIPAddrEntry0;
for (iAddr = 0; iAddr < pIntfIDItem->numaddr; iAddr++)
{
if (pIPAddrEntry->iae_index == pIFEntry->if_index)
pIntfInfo->iiFlags |= IFF_UP;
pIPAddrEntry++;
pIntfInfo++;
}
}
pEntityID++;
}
if (NumberOfBytesReturned)
*NumberOfBytesReturned = outNumberOfBytes;
if (NeedsCompletion != NULL)
*NeedsCompletion = FALSE;
res = NO_ERROR;
cleanup:
DPRINT("WSHIoctl_GetInterfaceList - CLEANUP\n");
if (TcpFile != 0)
NtClose(TcpFile);
if (pIFEntry != NULL)
HeapFree(hHeap, 0, pIFEntry);
LIST_FOR_EACH_ENTRY_SAFE_REV(pIntfIDItem, pIntfIDNext,
&IntfIDList->entry, struct _IntfIDItem, entry)
{
if (pIntfIDItem->pIPAddrEntry0 != NULL)
HeapFree(hHeap, 0, pIntfIDItem->pIPAddrEntry0);
list_remove(&pIntfIDItem->entry);
HeapFree(hHeap, 0, pIntfIDItem);
}
HeapFree(hHeap, 0, IntfIDList);
return res;
}
Reference in a new issue View git blame Copy permalink