Sweden-Number/dlls/ndis.sys/main.c

227 lines
7.0 KiB
C
Raw Normal View History

2014-10-29 18:30:39 +01:00
/*
* ndis.sys
*
* Copyright 2014 Austin English
* Copyright 2016 André Hentschel
2014-10-29 18:30:39 +01:00
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation; either
* version 2.1 of the License, or (at your option) any later version.
*
* This library is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this library; if not, write to the Free Software
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
*/
#include <stdarg.h>
#define NONAMELESSUNION
#include "ntstatus.h"
#define WIN32_NO_STATUS
2014-10-29 18:30:39 +01:00
#include "windef.h"
#include "winbase.h"
#include "winternl.h"
#include "winioctl.h"
#include "winsock2.h"
#include "ws2ipdef.h"
#include "iphlpapi.h"
#include "netioapi.h"
#include "ntddndis.h"
#include "ddk/wdm.h"
#include "ddk/ndis.h"
#include "winreg.h"
#include "wine/debug.h"
WINE_DEFAULT_DEBUG_CHANNEL(ndis);
2014-10-29 18:30:39 +01:00
static void query_global_stats(IRP *irp, const MIB_IF_ROW2 *netdev)
{
IO_STACK_LOCATION *irpsp = IoGetCurrentIrpStackLocation( irp );
void *response = MmGetSystemAddressForMdlSafe( irp->MdlAddress, NormalPagePriority );
DWORD len = irpsp->Parameters.DeviceIoControl.OutputBufferLength;
DWORD oid;
if (irpsp->Parameters.DeviceIoControl.InputBufferLength != sizeof(oid))
{
irp->IoStatus.u.Status = STATUS_INVALID_PARAMETER;
return;
}
oid = *(DWORD *)irp->AssociatedIrp.SystemBuffer;
switch (oid)
{
case OID_GEN_MEDIA_SUPPORTED:
case OID_GEN_MEDIA_IN_USE:
{
if (len < sizeof(NDIS_MEDIUM))
{
irp->IoStatus.u.Status = STATUS_INVALID_PARAMETER;
break;
}
*(NDIS_MEDIUM *)response = netdev->MediaType;
irp->IoStatus.Information = sizeof(netdev->MediaType);
irp->IoStatus.u.Status = STATUS_SUCCESS;
break;
}
case OID_802_3_PERMANENT_ADDRESS:
{
irp->IoStatus.Information = netdev->PhysicalAddressLength;
if (len < netdev->PhysicalAddressLength)
irp->IoStatus.u.Status = STATUS_INVALID_PARAMETER;
else
memcpy( response, netdev->PermanentPhysicalAddress, sizeof(netdev->PermanentPhysicalAddress) );
break;
}
case OID_802_3_CURRENT_ADDRESS:
{
irp->IoStatus.Information = netdev->PhysicalAddressLength;
if (len < netdev->PhysicalAddressLength)
irp->IoStatus.u.Status = STATUS_INVALID_PARAMETER;
else
memcpy( response, netdev->PhysicalAddress, sizeof(netdev->PhysicalAddress) );
break;
}
default:
FIXME( "Unsupported OID %x\n", oid );
irp->IoStatus.u.Status = STATUS_NOT_SUPPORTED;
break;
}
}
static NTSTATUS WINAPI ndis_ioctl(DEVICE_OBJECT *device, IRP *irp)
{
IO_STACK_LOCATION *irpsp = IoGetCurrentIrpStackLocation( irp );
MIB_IF_ROW2 *netdev = device->DeviceExtension;
TRACE( "ioctl %x insize %u outsize %u\n",
irpsp->Parameters.DeviceIoControl.IoControlCode,
irpsp->Parameters.DeviceIoControl.InputBufferLength,
irpsp->Parameters.DeviceIoControl.OutputBufferLength );
switch (irpsp->Parameters.DeviceIoControl.IoControlCode)
{
case IOCTL_NDIS_QUERY_GLOBAL_STATS:
query_global_stats(irp, netdev);
break;
default:
FIXME( "ioctl %x not supported\n", irpsp->Parameters.DeviceIoControl.IoControlCode );
irp->IoStatus.u.Status = STATUS_NOT_SUPPORTED;
break;
}
IoCompleteRequest( irp, IO_NO_INCREMENT );
return STATUS_SUCCESS;
}
static void add_key(const WCHAR *guidstrW, const MIB_IF_ROW2 *netdev)
{
HKEY card_key;
WCHAR keynameW[100];
swprintf( keynameW, ARRAY_SIZE(keynameW), L"Software\\Microsoft\\Windows NT\\CurrentVersion\\NetworkCards\\%d", netdev->InterfaceIndex );
if (RegCreateKeyExW( HKEY_LOCAL_MACHINE, keynameW, 0, NULL,
REG_OPTION_VOLATILE, KEY_ALL_ACCESS, NULL, &card_key, NULL ) == ERROR_SUCCESS)
{
RegSetValueExW( card_key, L"Description", 0, REG_SZ, (BYTE *)netdev->Description, (lstrlenW(netdev->Description) + 1) * sizeof(WCHAR) );
RegSetValueExW( card_key, L"ServiceName", 0, REG_SZ, (BYTE *)guidstrW, (lstrlenW(guidstrW) + 1) * sizeof(WCHAR) );
RegCloseKey( card_key );
}
}
static int add_device(DRIVER_OBJECT *driver, const WCHAR *guidstrW, MIB_IF_ROW2 *netdev)
{
WCHAR nameW[47], linkW[51];
UNICODE_STRING name, link;
DEVICE_OBJECT *device;
NTSTATUS status;
swprintf( nameW, ARRAY_SIZE(nameW), L"\\Device\\%s", guidstrW );
RtlInitUnicodeString( &name, nameW );
swprintf( linkW, ARRAY_SIZE(linkW), L"\\DosDevices\\%s", guidstrW );
RtlInitUnicodeString( &link, linkW );
if (!(status = IoCreateDevice( driver, sizeof(*netdev), &name, 0, 0, FALSE, &device )))
status = IoCreateSymbolicLink( &link, &name );
if (status)
{
FIXME( "failed to create device error %x\n", status );
return 0;
}
memcpy( device->DeviceExtension, netdev, sizeof(*netdev) );
return 1;
}
static void create_network_devices(DRIVER_OBJECT *driver)
{
MIB_IF_TABLE2 *table;
ULONG i;
if (GetIfTable2( &table ) != NO_ERROR)
return;
for (i = 0; i < table->NumEntries; i++)
{
GUID *guid = &table->Table[i].InterfaceGuid;
WCHAR guidstrW[39];
swprintf( guidstrW, ARRAY_SIZE(guidstrW), L"{%08X-%04X-%04X-%02X%02X-%02X%02X%02X%02X%02X%02X}",
guid->Data1, guid->Data2, guid->Data3, guid->Data4[0], guid->Data4[1],
guid->Data4[2], guid->Data4[3], guid->Data4[4], guid->Data4[5],
guid->Data4[6], guid->Data4[7] );
if (add_device( driver, guidstrW, &table->Table[i] ))
add_key( guidstrW, &table->Table[i] );
}
FreeMibTable( table );
}
NTSTATUS WINAPI DriverEntry(DRIVER_OBJECT *driver, UNICODE_STRING *path)
2014-10-29 18:30:39 +01:00
{
TRACE("(%p, %s)\n", driver, debugstr_w(path->Buffer));
2014-10-29 18:30:39 +01:00
driver->MajorFunction[IRP_MJ_DEVICE_CONTROL] = ndis_ioctl;
create_network_devices( driver );
return STATUS_SUCCESS;
2014-10-29 18:30:39 +01:00
}
NDIS_STATUS WINAPI NdisAllocateMemoryWithTag(void **address, UINT length, ULONG tag)
{
FIXME("(%p, %u, %u): stub\n", address, length, tag);
return NDIS_STATUS_FAILURE;
}
void WINAPI NdisAllocateSpinLock(NDIS_SPIN_LOCK *lock)
{
FIXME("(%p): stub\n", lock);
}
void WINAPI NdisRegisterProtocol(NDIS_STATUS *status, NDIS_HANDLE *handle,
NDIS_PROTOCOL_CHARACTERISTICS *prot, UINT len)
{
FIXME("(%p, %p, %p, %u): stub\n", status, handle, prot, len);
*status = NDIS_STATUS_FAILURE;
}
CCHAR WINAPI NdisSystemProcessorCount(void)
{
SYSTEM_INFO si;
TRACE("()\n");
GetSystemInfo(&si);
return si.dwNumberOfProcessors;
}