[PATCH 4/5] mountmgr.sys: Implement IOCTL_NDIS_QUERY_GLOBAL_STATS on network cards.

[Zebediah Figura]

On 8/25/20 9:32 AM, Isabella Bosia wrote:
> Signed-off-by: Isabella Bosia <ibosia at codeweavers.com>
> ---
>  dlls/mountmgr.sys/ndis.c | 105 +++++++++++++++++++++++++++++++++++++++
>  1 file changed, 105 insertions(+)
> 

This patch could probably be split up: one patch to create the device,
and maybe even split up the rest by individual ioctls.

> diff --git a/dlls/mountmgr.sys/ndis.c b/dlls/mountmgr.sys/ndis.c
> index 0a3f87d410..379d70e6d0 100644
> --- a/dlls/mountmgr.sys/ndis.c
> +++ b/dlls/mountmgr.sys/ndis.c
> @@ -40,6 +40,86 @@
>  
>  WINE_DEFAULT_DEBUG_CHANNEL(ndis);
>  
> +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 char *guidstr, const MIB_IF_ROW2 *netdev)
>  {
>      HKEY card_key;
> @@ -61,6 +141,29 @@ static void add_key(const char *guidstr, const MIB_IF_ROW2 *netdev)
>  
>  static int add_device(DRIVER_OBJECT *driver, const char *guidstr, MIB_IF_ROW2 *netdev)
>  {
> +    char data[300];
> +    ANSI_STRING nameA, linkA;
> +    UNICODE_STRING nameW, linkW;
> +    DEVICE_OBJECT *device;
> +    NTSTATUS status;
> +
> +    snprintf( data, sizeof(data), "\\Device\\%s", guidstr );
> +    RtlInitAnsiString( &nameA, data );
> +    RtlAnsiStringToUnicodeString( &nameW, &nameA, TRUE );
> +
> +    snprintf( data, sizeof(data), "\\DosDevices\\%s", guidstr );
> +    RtlInitAnsiString( &linkA, data );
> +    RtlAnsiStringToUnicodeString( &linkW, &linkA, TRUE );
> +
> +    if (!(status = IoCreateDevice( driver, sizeof(*netdev), &nameW, 0, 0, FALSE, &device )))
> +        status = IoCreateSymbolicLink( &linkW, &nameW );
> +    if (status)
> +    {
> +        FIXME( "failed to create device error %x\n", status );
> +        return 0;
> +    }
> +
> +    memcpy( device->DeviceExtension, netdev, sizeof(*netdev) );
>      return 1;
>  }
>  
> @@ -92,6 +195,8 @@ NTSTATUS WINAPI ndis_driver_entry(DRIVER_OBJECT *driver, UNICODE_STRING *path)
>  {
>      TRACE("(%p, %s)\n", driver, debugstr_w(path->Buffer));
>  
> +    driver->MajorFunction[IRP_MJ_DEVICE_CONTROL] = ndis_ioctl;
> +
>      create_network_devices( driver );
>  
>      return STATUS_SUCCESS;
> 

-------------- next part --------------
A non-text attachment was scrubbed...
Name: signature.asc
Type: application/pgp-signature
Size: 488 bytes
Desc: OpenPGP digital signature
URL: <http://www.winehq.org/pipermail/wine-devel/attachments/20200826/aebff565/attachment.sig>