context.c File Reference

#include "afd.h"

Include dependency graph for context.c:

Go to the source code of this file.

Functions
NTSTATUS NTAPI	AfdGetContext (PDEVICE_OBJECT DeviceObject, PIRP Irp, PIO_STACK_LOCATION IrpSp)
NTSTATUS NTAPI	AfdGetContextSize (PDEVICE_OBJECT DeviceObject, PIRP Irp, PIO_STACK_LOCATION IrpSp)
NTSTATUS NTAPI	AfdSetContext (PDEVICE_OBJECT DeviceObject, PIRP Irp, PIO_STACK_LOCATION IrpSp)

Function Documentation

AfdGetContext()

NTSTATUS NTAPI AfdGetContext	(	PDEVICE_OBJECT	DeviceObject,
		PIRP	Irp,
		PIO_STACK_LOCATION	IrpSp
	)

Definition at line 14 of file context.c.

                                          {
    NTSTATUS Status = STATUS_INVALID_PARAMETER;
    PFILE_OBJECT FileObject = IrpSp->FileObject;
    PAFD_FCB FCB = FileObject->FsContext;
    UINT ContextSize = IrpSp->Parameters.DeviceIoControl.OutputBufferLength;
 
    UNREFERENCED_PARAMETER(DeviceObject);
 
    if( !SocketAcquireStateLock( FCB ) ) return LostSocket( Irp );
 
    if (ContextSize == sizeof(FCB->SharedData))
    {
        RtlCopyMemory(Irp->UserBuffer,
                      &FCB->SharedData,
                      ContextSize);
        Status = STATUS_SUCCESS;
    }
    else if (FCB->Context)
    {
        ContextSize = min(ContextSize, FCB->ContextSize);
        RtlCopyMemory( Irp->UserBuffer,
                       FCB->Context,
                       ContextSize );
        Status = STATUS_SUCCESS;
    }
 
    AFD_DbgPrint(MID_TRACE,("Returning %x\n", Status));
 
    return UnlockAndMaybeComplete( FCB, Status, Irp, ContextSize );
}

Referenced by AfdDispatch().

AfdGetContextSize()

NTSTATUS NTAPI AfdGetContextSize	(	PDEVICE_OBJECT	DeviceObject,
		PIRP	Irp,
		PIO_STACK_LOCATION	IrpSp
	)

Definition at line 47 of file context.c.

{
    PFILE_OBJECT FileObject = IrpSp->FileObject;
    PAFD_FCB FCB = FileObject->FsContext;
 
    UNREFERENCED_PARAMETER(DeviceObject);
 
    if( !SocketAcquireStateLock( FCB ) ) return LostSocket( Irp );
 
    if (IrpSp->Parameters.DeviceIoControl.OutputBufferLength < sizeof(ULONG))
    {
        AFD_DbgPrint(MIN_TRACE,("Buffer too small\n"));
        return UnlockAndMaybeComplete(FCB, STATUS_BUFFER_TOO_SMALL, Irp, sizeof(ULONG));
    }
 
    RtlCopyMemory(Irp->UserBuffer,
                  &FCB->ContextSize,
                  sizeof(ULONG));
 
    return UnlockAndMaybeComplete(FCB, STATUS_SUCCESS, Irp, sizeof(ULONG));
}

Referenced by AfdDispatch().

AfdSetContext()

NTSTATUS NTAPI AfdSetContext	(	PDEVICE_OBJECT	DeviceObject,
		PIRP	Irp,
		PIO_STACK_LOCATION	IrpSp
	)

Definition at line 71 of file context.c.

                                          {
    PFILE_OBJECT FileObject = IrpSp->FileObject;
    PAFD_FCB FCB = FileObject->FsContext;
    PVOID Context = LockRequest(Irp, IrpSp, FALSE, NULL);
 
    UNREFERENCED_PARAMETER(DeviceObject);
 
    if( !SocketAcquireStateLock( FCB ) ) return LostSocket( Irp );
 
    if (!Context)
        return UnlockAndMaybeComplete(FCB, STATUS_NO_MEMORY, Irp, 0);
 
    if( FCB->Context ) {
        ExFreePoolWithTag(FCB->Context, TAG_AFD_SOCKET_CONTEXT);
        FCB->ContextSize = 0;
    }
 
    FCB->Context = ExAllocatePoolWithTag(PagedPool,
                                         IrpSp->Parameters.DeviceIoControl.InputBufferLength,
                                         TAG_AFD_SOCKET_CONTEXT);
 
    if( !FCB->Context ) return UnlockAndMaybeComplete( FCB, STATUS_NO_MEMORY, Irp, 0 );
 
    FCB->ContextSize = IrpSp->Parameters.DeviceIoControl.InputBufferLength;
 
    RtlCopyMemory( FCB->Context,
                   Context,
                   FCB->ContextSize );
 
    return UnlockAndMaybeComplete( FCB, STATUS_SUCCESS, Irp, 0 );
}

Referenced by AfdDispatch().