dlls/ndis.sys/main.c

   1 /*
   2  * ndis.sys
   3  *
   4  * Copyright 2014 Austin English
   5  * Copyright 2016 André Hentschel
   6  *
   7  * This library is free software; you can redistribute it and/or
   8  * modify it under the terms of the GNU Lesser General Public
   9  * License as published by the Free Software Foundation; either
  10  * version 2.1 of the License, or (at your option) any later version.
  11  *
  12  * This library is distributed in the hope that it will be useful,
  13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  15  * Lesser General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU Lesser General Public
  18  * License along with this library; if not, write to the Free Software
  19  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  20  */
  21
  22 #include <stdarg.h>
  23
  24 #define NONAMELESSUNION
  25 #include "ntstatus.h"
  26 #define WIN32_NO_STATUS
  27 #include "windef.h"
  28 #include "winbase.h"
  29 #include "winternl.h"
  30 #include "winioctl.h"
  31 #include "winsock2.h"
  32 #include "ws2ipdef.h"
  33 #include "iphlpapi.h"
  34 #include "netioapi.h"
  35 #include "ddk/wdm.h"
  36 #include "ddk/ndis.h"
  37 #include "winreg.h"
  38 #include "wine/debug.h"
  39
  40 WINE_DEFAULT_DEBUG_CHANNEL(ndis);
  41
  42 static void query_global_stats(IRP *irp, const MIB_IF_ROW2 *netdev)
  43 {
  44     IO_STACK_LOCATION *irpsp = IoGetCurrentIrpStackLocation( irp );
  45     void *response = MmGetSystemAddressForMdlSafe( irp->MdlAddress, NormalPagePriority );
  46     DWORD len = irpsp->Parameters.DeviceIoControl.OutputBufferLength;
  47     DWORD oid;
  48
  49     if (irpsp->Parameters.DeviceIoControl.InputBufferLength != sizeof(oid))
  50     {
  51         irp->IoStatus.u.Status = STATUS_INVALID_PARAMETER;
  52         return;
  53     }
  54     oid = *(DWORD *)irp->AssociatedIrp.SystemBuffer;
  55
  56     switch (oid)
  57     {
  58     case OID_GEN_MEDIA_SUPPORTED:
  59     case OID_GEN_MEDIA_IN_USE:
  60     {
  61         if (len < sizeof(NDIS_MEDIUM))
  62         {
  63             irp->IoStatus.u.Status = STATUS_INVALID_PARAMETER;
  64             break;
  65         }
  66         *(NDIS_MEDIUM *)response = netdev->MediaType;
  67         irp->IoStatus.Information = sizeof(netdev->MediaType);
  68         irp->IoStatus.u.Status = STATUS_SUCCESS;
  69         break;
  70     }
  71     case OID_802_3_PERMANENT_ADDRESS:
  72     {
  73         irp->IoStatus.Information = netdev->PhysicalAddressLength;
  74         if (len < netdev->PhysicalAddressLength)
  75             irp->IoStatus.u.Status = STATUS_INVALID_PARAMETER;
  76         else
  77             memcpy( response, netdev->PermanentPhysicalAddress, sizeof(netdev->PermanentPhysicalAddress) );
  78         break;
  79     }
  80     case OID_802_3_CURRENT_ADDRESS:
  81     {
  82         irp->IoStatus.Information = netdev->PhysicalAddressLength;
  83         if (len < netdev->PhysicalAddressLength)
  84             irp->IoStatus.u.Status = STATUS_INVALID_PARAMETER;
  85         else
  86             memcpy( response, netdev->PhysicalAddress, sizeof(netdev->PhysicalAddress) );
  87         break;
  88
  89     }
  90     default:
  91         FIXME( "Unsupported OID %x\n", oid );
  92         irp->IoStatus.u.Status = STATUS_NOT_SUPPORTED;
  93         break;
  94     }
  95 }
  96
  97 static NTSTATUS WINAPI ndis_ioctl(DEVICE_OBJECT *device, IRP *irp)
  98 {
  99     IO_STACK_LOCATION *irpsp = IoGetCurrentIrpStackLocation( irp );
 100     MIB_IF_ROW2 *netdev = device->DeviceExtension;
 101
 102     TRACE( "ioctl %x insize %u outsize %u\n",
 103            irpsp->Parameters.DeviceIoControl.IoControlCode,
 104            irpsp->Parameters.DeviceIoControl.InputBufferLength,
 105            irpsp->Parameters.DeviceIoControl.OutputBufferLength );
 106
 107     switch (irpsp->Parameters.DeviceIoControl.IoControlCode)
 108     {
 109     case IOCTL_NDIS_QUERY_GLOBAL_STATS:
 110         query_global_stats(irp, netdev);
 111         break;
 112     default:
 113         FIXME( "ioctl %x not supported\n", irpsp->Parameters.DeviceIoControl.IoControlCode );
 114         irp->IoStatus.u.Status = STATUS_NOT_SUPPORTED;
 115         break;
 116     }
 117
 118     IoCompleteRequest( irp, IO_NO_INCREMENT );
 119     return STATUS_SUCCESS;
 120 }
 121
 122 static void add_key(const WCHAR *guidstrW, const MIB_IF_ROW2 *netdev)
 123 {
 124     HKEY card_key;
 125     WCHAR keynameW[100];
 126
 127     swprintf( keynameW, ARRAY_SIZE(keynameW), L"Software\\Microsoft\\Windows NT\\CurrentVersion\\NetworkCards\\%d", netdev->InterfaceIndex );
 128     if (RegCreateKeyExW( HKEY_LOCAL_MACHINE, keynameW, 0, NULL,
 129                  REG_OPTION_VOLATILE, KEY_ALL_ACCESS, NULL, &card_key, NULL ) == ERROR_SUCCESS)
 130     {
 131         RegSetValueExW( card_key, L"Description", 0, REG_SZ, (BYTE *)netdev->Description, (lstrlenW(netdev->Description) + 1) * sizeof(WCHAR) );
 132         RegSetValueExW( card_key, L"ServiceName", 0, REG_SZ, (BYTE *)guidstrW, (lstrlenW(guidstrW) + 1) * sizeof(WCHAR) );
 133         RegCloseKey( card_key );
 134     }
 135 }
 136
 137 static int add_device(DRIVER_OBJECT *driver, const WCHAR *guidstrW, MIB_IF_ROW2 *netdev)
 138 {
 139     WCHAR nameW[47], linkW[51];
 140     UNICODE_STRING name, link;
 141     DEVICE_OBJECT *device;
 142     NTSTATUS status;
 143
 144     swprintf( nameW, ARRAY_SIZE(nameW), L"\\Device\\%s", guidstrW );
 145     RtlInitUnicodeString( &name, nameW );
 146
 147     swprintf( linkW, ARRAY_SIZE(linkW), L"\\DosDevices\\%s", guidstrW );
 148     RtlInitUnicodeString( &link, linkW );
 149
 150     if (!(status = IoCreateDevice( driver, sizeof(*netdev), &name, 0, 0, FALSE, &device )))
 151         status = IoCreateSymbolicLink( &link, &name );
 152     if (status)
 153     {
 154         FIXME( "failed to create device error %x\n", status );
 155         return 0;
 156     }
 157
 158     memcpy( device->DeviceExtension, netdev, sizeof(*netdev) );
 159     return 1;
 160 }
 161
 162
 163 static void create_network_devices(DRIVER_OBJECT *driver)
 164 {
 165     MIB_IF_TABLE2 *table;
 166     ULONG i;
 167
 168     if (GetIfTable2( &table ) != NO_ERROR)
 169         return;
 170
 171     for (i = 0; i < table->NumEntries; i++)
 172     {
 173         GUID *guid = &table->Table[i].InterfaceGuid;
 174         WCHAR guidstrW[39];
 175
 176         swprintf( guidstrW, ARRAY_SIZE(guidstrW), L"{%08X-%04X-%04X-%02X%02X-%02X%02X%02X%02X%02X%02X}",
 177                   guid->Data1, guid->Data2, guid->Data3, guid->Data4[0], guid->Data4[1],
 178                   guid->Data4[2], guid->Data4[3], guid->Data4[4], guid->Data4[5],
 179                   guid->Data4[6], guid->Data4[7] );
 180
 181         if (add_device( driver, guidstrW, &table->Table[i] ))
 182             add_key( guidstrW, &table->Table[i] );
 183     }
 184
 185     FreeMibTable( table );
 186 }
 187
 188 NTSTATUS WINAPI DriverEntry(DRIVER_OBJECT *driver, UNICODE_STRING *path)
 189 {
 190     TRACE("(%p, %s)\n", driver, debugstr_w(path->Buffer));
 191
 192     driver->MajorFunction[IRP_MJ_DEVICE_CONTROL] = ndis_ioctl;
 193
 194     create_network_devices( driver );
 195
 196     return STATUS_SUCCESS;
 197 }
 198
 199 NDIS_STATUS WINAPI NdisAllocateMemoryWithTag(void **address, UINT length, ULONG tag)
 200 {
 201     FIXME("(%p, %u, %u): stub\n", address, length, tag);
 202     return NDIS_STATUS_FAILURE;
 203 }
 204
 205 void WINAPI NdisAllocateSpinLock(NDIS_SPIN_LOCK *lock)
 206 {
 207     FIXME("(%p): stub\n", lock);
 208 }
 209
 210 void WINAPI NdisRegisterProtocol(NDIS_STATUS *status, NDIS_HANDLE *handle,
 211                                  NDIS_PROTOCOL_CHARACTERISTICS *prot, UINT len)
 212 {
 213     FIXME("(%p, %p, %p, %u): stub\n", status, handle, prot, len);
 214     *status = NDIS_STATUS_FAILURE;
 215 }
 216
 217 CCHAR WINAPI NdisSystemProcessorCount(void)
 218 {
 219     SYSTEM_INFO si;
 220
 221     TRACE("()\n");
 222     GetSystemInfo(&si);
 223
 224     return si.dwNumberOfProcessors;
 225 }