reactos/drivers/network/afd/afd/info.c

   1 /* $Id$
   2  * COPYRIGHT:        See COPYING in the top level directory
   3  * PROJECT:          ReactOS kernel
   4  * FILE:             drivers/net/afd/afd/info.c
   5  * PURPOSE:          Ancillary functions driver
   6  * PROGRAMMER:       Art Yerkes (ayerkes@speakeasy.net)
   7  * UPDATE HISTORY:
   8  * 20040708 Created
   9  */
  10 #include "afd.h"
  11
  12 NTSTATUS NTAPI
  13 AfdGetInfo( PDEVICE_OBJECT DeviceObject, PIRP Irp,
  14             PIO_STACK_LOCATION IrpSp ) {
  15     NTSTATUS Status = STATUS_SUCCESS;
  16     PAFD_INFO InfoReq = LockRequest(Irp, IrpSp);
  17     PFILE_OBJECT FileObject = IrpSp->FileObject;
  18     PAFD_FCB FCB = FileObject->FsContext;
  19     PLIST_ENTRY CurrentEntry;
  20
  21     AFD_DbgPrint(MID_TRACE,("Called %x %x\n", InfoReq,
  22                             InfoReq ? InfoReq->InformationClass : 0));
  23
  24     if( !SocketAcquireStateLock( FCB ) ) return LostSocket( Irp );
  25
  26     if (!InfoReq)
  27         return UnlockAndMaybeComplete(FCB, STATUS_NO_MEMORY, Irp, 0);
  28
  29     _SEH2_TRY {
  30         switch( InfoReq->InformationClass ) {
  31         case AFD_INFO_RECEIVE_WINDOW_SIZE:
  32             InfoReq->Information.Ulong = FCB->Recv.Size;
  33             break;
  34
  35         case AFD_INFO_SEND_WINDOW_SIZE:
  36             InfoReq->Information.Ulong = FCB->Send.Size;
  37             AFD_DbgPrint(MID_TRACE,("Send window size %d\n", FCB->Send.Size));
  38             break;
  39
  40         case AFD_INFO_GROUP_ID_TYPE:
  41             InfoReq->Information.LargeInteger.u.HighPart = FCB->GroupType;
  42             InfoReq->Information.LargeInteger.u.LowPart = FCB->GroupID;
  43             AFD_DbgPrint(MID_TRACE, ("Group ID: %d Group Type: %d\n", FCB->GroupID, FCB->GroupType));
  44             break;
  45
  46         case AFD_INFO_BLOCKING_MODE:
  47             InfoReq->Information.Boolean = FCB->NonBlocking;
  48             break;
  49
  50     case AFD_INFO_INLINING_MODE:
  51         InfoReq->Information.Boolean = FCB->OobInline;
  52         break;
  53
  54     case AFD_INFO_RECEIVE_CONTENT_SIZE:
  55         /* Only touch InfoReq if a socket has been set up.
  56            Behaviour was verified under WinXP SP2. */
  57         if(FCB->AddressFile.Object || FCB->Connection.Object)
  58             InfoReq->Information.Ulong = FCB->Recv.Content - FCB->Recv.BytesUsed;
  59
  60         break;
  61
  62         case AFD_INFO_SENDS_IN_PROGRESS:
  63             InfoReq->Information.Ulong = 0;
  64
  65             /* Count the queued sends */
  66             CurrentEntry = FCB->PendingIrpList[FUNCTION_SEND].Flink;
  67             while (CurrentEntry != &FCB->PendingIrpList[FUNCTION_SEND])
  68             {
  69                  InfoReq->Information.Ulong++;
  70                  CurrentEntry = CurrentEntry->Flink;
  71             }
  72
  73         /* This needs to count too because when this is dispatched
  74          * the user-mode IRP has already been completed and therefore
  75          * will NOT be in our pending IRP list. We count this as one send
  76          * outstanding although it could be multiple since we batch sends
  77          * when waiting for the in flight request to return, so this number
  78          * may not be accurate but it really doesn't matter that much since
  79          * it's more or less a zero/non-zero comparison to determine whether
  80          * we can shutdown the socket
  81          */
  82         if (FCB->SendIrp.InFlightRequest)
  83             InfoReq->Information.Ulong++;
  84         break;
  85
  86         default:
  87             AFD_DbgPrint(MIN_TRACE,("Unknown info id %x\n",
  88                                     InfoReq->InformationClass));
  89             Status = STATUS_INVALID_PARAMETER;
  90             break;
  91         }
  92     } _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER) {
  93         AFD_DbgPrint(MIN_TRACE,("Exception executing GetInfo\n"));
  94         Status = STATUS_INVALID_PARAMETER;
  95     } _SEH2_END;
  96
  97     AFD_DbgPrint(MID_TRACE,("Returning %x\n", Status));
  98
  99     return UnlockAndMaybeComplete( FCB, Status, Irp, 0 );
 100 }
 101
 102 NTSTATUS NTAPI
 103 AfdSetInfo( PDEVICE_OBJECT DeviceObject, PIRP Irp,
 104             PIO_STACK_LOCATION IrpSp ) {
 105     NTSTATUS Status = STATUS_SUCCESS;
 106     PAFD_INFO InfoReq = LockRequest(Irp, IrpSp);
 107     PFILE_OBJECT FileObject = IrpSp->FileObject;
 108     PAFD_FCB FCB = FileObject->FsContext;
 109     PCHAR NewBuffer;
 110
 111     if (!SocketAcquireStateLock(FCB)) return LostSocket(Irp);
 112
 113     if (!InfoReq)
 114         return UnlockAndMaybeComplete(FCB, STATUS_NO_MEMORY, Irp, 0);
 115
 116     _SEH2_TRY {
 117       switch (InfoReq->InformationClass) {
 118         case AFD_INFO_BLOCKING_MODE:
 119           AFD_DbgPrint(MID_TRACE,("Blocking mode set to %d\n", InfoReq->Information.Boolean));
 120           FCB->NonBlocking = InfoReq->Information.Boolean;
 121           break;
 122         case AFD_INFO_INLINING_MODE:
 123           FCB->OobInline = InfoReq->Information.Boolean;
 124           break;
 125         case AFD_INFO_RECEIVE_WINDOW_SIZE:
 126           NewBuffer = ExAllocatePool(PagedPool, InfoReq->Information.Ulong);
 127           if (NewBuffer)
 128           {
 129               if (FCB->Recv.Content > InfoReq->Information.Ulong)
 130                   FCB->Recv.Content = InfoReq->Information.Ulong;
 131
 132               if (FCB->Recv.Window)
 133               {
 134                   RtlCopyMemory(NewBuffer,
 135                                 FCB->Recv.Window,
 136                                 FCB->Recv.Content);
 137
 138                   ExFreePool(FCB->Recv.Window);
 139               }
 140
 141               FCB->Recv.Size = InfoReq->Information.Ulong;
 142               FCB->Recv.Window = NewBuffer;
 143
 144               Status = STATUS_SUCCESS;
 145           }
 146           else
 147           {
 148               Status = STATUS_NO_MEMORY;
 149           }
 150           break;
 151         case AFD_INFO_SEND_WINDOW_SIZE:
 152           NewBuffer = ExAllocatePool(PagedPool, InfoReq->Information.Ulong);
 153           if (NewBuffer)
 154           {
 155               if (FCB->Send.BytesUsed > InfoReq->Information.Ulong)
 156                   FCB->Send.BytesUsed = InfoReq->Information.Ulong;
 157
 158               if (FCB->Send.Window)
 159               {
 160                   RtlCopyMemory(NewBuffer,
 161                                 FCB->Send.Window,
 162                                 FCB->Send.BytesUsed);
 163
 164                   ExFreePool(FCB->Send.Window);
 165               }
 166
 167               FCB->Send.Size = InfoReq->Information.Ulong;
 168               FCB->Send.Window = NewBuffer;
 169
 170               Status = STATUS_SUCCESS;
 171           }
 172           else
 173           {
 174               Status = STATUS_NO_MEMORY;
 175           }
 176           break;
 177         default:
 178           AFD_DbgPrint(MIN_TRACE,("Unknown request %d\n", InfoReq->InformationClass));
 179           break;
 180       }
 181     } _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER) {
 182       AFD_DbgPrint(MIN_TRACE,("Exception executing SetInfo\n"));
 183       Status = STATUS_INVALID_PARAMETER;
 184     } _SEH2_END;
 185
 186     AFD_DbgPrint(MID_TRACE,("Returning %x\n", Status));
 187
 188     return UnlockAndMaybeComplete(FCB, Status, Irp, 0);
 189 }
 190
 191 NTSTATUS NTAPI
 192 AfdGetSockName( PDEVICE_OBJECT DeviceObject, PIRP Irp,
 193                       PIO_STACK_LOCATION IrpSp ) {
 194     NTSTATUS Status = STATUS_SUCCESS;
 195     PFILE_OBJECT FileObject = IrpSp->FileObject;
 196     PAFD_FCB FCB = FileObject->FsContext;
 197     PMDL Mdl = NULL;
 198
 199     if( !SocketAcquireStateLock( FCB ) ) return LostSocket( Irp );
 200
 201     if( FCB->AddressFile.Object == NULL && FCB->Connection.Object == NULL ) {
 202          return UnlockAndMaybeComplete( FCB, STATUS_INVALID_PARAMETER, Irp, 0 );
 203     }
 204
 205     Mdl = IoAllocateMdl
 206         ( Irp->UserBuffer,
 207           IrpSp->Parameters.DeviceIoControl.OutputBufferLength,
 208           FALSE,
 209           FALSE,
 210           NULL );
 211
 212     if( Mdl != NULL ) {
 213         _SEH2_TRY {
 214             MmProbeAndLockPages( Mdl, Irp->RequestorMode, IoModifyAccess );
 215         } _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER) {
 216             AFD_DbgPrint(MIN_TRACE, ("MmProbeAndLockPages() failed.\n"));
 217             Status = _SEH2_GetExceptionCode();
 218         } _SEH2_END;
 219
 220         if( NT_SUCCESS(Status) ) {
 221                 Status = TdiQueryInformation
 222                     ( FCB->Connection.Object ? FCB->Connection.Object : FCB->AddressFile.Object,
 223                       TDI_QUERY_ADDRESS_INFO,
 224                       Mdl );
 225         }
 226     } else
 227         Status = STATUS_INSUFFICIENT_RESOURCES;
 228
 229     return UnlockAndMaybeComplete( FCB, Status, Irp, 0 );
 230 }
 231
 232 NTSTATUS NTAPI
 233 AfdGetPeerName( PDEVICE_OBJECT DeviceObject, PIRP Irp,
 234                       PIO_STACK_LOCATION IrpSp ) {
 235     NTSTATUS Status;
 236     PFILE_OBJECT FileObject = IrpSp->FileObject;
 237     PAFD_FCB FCB = FileObject->FsContext;
 238
 239
 240     if( !SocketAcquireStateLock( FCB ) ) return LostSocket( Irp );
 241
 242     if (FCB->RemoteAddress == NULL) {
 243         AFD_DbgPrint(MIN_TRACE,("Invalid parameter\n"));
 244         return UnlockAndMaybeComplete( FCB, STATUS_INVALID_PARAMETER, Irp, 0 );
 245     }
 246
 247     if (IrpSp->Parameters.DeviceIoControl.OutputBufferLength >= TaLengthOfTransportAddress(FCB->RemoteAddress))
 248     {
 249         RtlCopyMemory(Irp->UserBuffer, FCB->RemoteAddress, TaLengthOfTransportAddress(FCB->RemoteAddress));
 250         Status = STATUS_SUCCESS;
 251     }
 252     else
 253     {
 254         AFD_DbgPrint(MIN_TRACE,("Buffer too small\n"));
 255         Status = STATUS_BUFFER_TOO_SMALL;
 256     }
 257
 258     return UnlockAndMaybeComplete( FCB, Status, Irp, 0 );
 259 }