mirror of
https://github.com/reactos/reactos.git
synced 2024-06-19 19:28:22 +00:00
d99f998d2d
- It was a race condition caused by another thread clearing our IRP while we were still using it svn path=/trunk/; revision=40183
387 lines
10 KiB
C
387 lines
10 KiB
C
/* $Id$
|
|
*
|
|
* COPYRIGHT: See COPYING in the top level directory
|
|
* PROJECT: ReactOS kernel
|
|
* FILE: drivers/net/afd/include/afd.h
|
|
* PURPOSE: Ancillary functions driver -- constants and structures
|
|
* PROGRAMMER: Art Yerkes (ayerkes@speakeasy.net)
|
|
* UPDATE HISTORY:
|
|
* 20040630 Created
|
|
*/
|
|
|
|
#ifndef _AFD_H
|
|
#define _AFD_H
|
|
|
|
#include <ntddk.h>
|
|
#include <ntifs.h>
|
|
#include <ndk/obtypes.h>
|
|
#include <ndk/obfuncs.h>
|
|
#include <ndk/mmtypes.h>
|
|
#include <ndk/mmfuncs.h>
|
|
#include <tdi.h>
|
|
#include <tdikrnl.h>
|
|
#include <tdiinfo.h>
|
|
#include <string.h>
|
|
#define _WINBASE_
|
|
#define _WINDOWS_H
|
|
#define _INC_WINDOWS
|
|
#include <windef.h>
|
|
#include <winsock2.h>
|
|
#include <afd/shared.h>
|
|
|
|
#ifndef MIN
|
|
#define MIN(x,y) (((x)<(y))?(x):(y))
|
|
#endif
|
|
|
|
#define IOCTL_TCP_QUERY_INFORMATION_EX \
|
|
CTL_CODE(FILE_DEVICE_NETWORK, 0, METHOD_NEITHER, FILE_ANY_ACCESS)
|
|
|
|
#define TL_INSTANCE 0
|
|
#define IP_MIB_STATS_ID 1
|
|
#define IP_MIB_ADDRTABLE_ENTRY_ID 0x102
|
|
|
|
typedef struct IPSNMP_INFO {
|
|
ULONG Forwarding;
|
|
ULONG DefaultTTL;
|
|
ULONG InReceives;
|
|
ULONG InHdrErrors;
|
|
ULONG InAddrErrors;
|
|
ULONG ForwDatagrams;
|
|
ULONG InUnknownProtos;
|
|
ULONG InDiscards;
|
|
ULONG InDelivers;
|
|
ULONG OutRequests;
|
|
ULONG RoutingDiscards;
|
|
ULONG OutDiscards;
|
|
ULONG OutNoRoutes;
|
|
ULONG ReasmTimeout;
|
|
ULONG ReasmReqds;
|
|
ULONG ReasmOks;
|
|
ULONG ReasmFails;
|
|
ULONG FragOks;
|
|
ULONG FragFails;
|
|
ULONG FragCreates;
|
|
ULONG NumIf;
|
|
ULONG NumAddr;
|
|
ULONG NumRoutes;
|
|
} IPSNMP_INFO, *PIPSNMP_INFO;
|
|
|
|
typedef struct IPADDR_ENTRY {
|
|
ULONG Addr;
|
|
ULONG Index;
|
|
ULONG Mask;
|
|
ULONG BcastAddr;
|
|
ULONG ReasmSize;
|
|
USHORT Context;
|
|
USHORT Pad;
|
|
} IPADDR_ENTRY, *PIPADDR_ENTRY;
|
|
|
|
#define DN2H(dw) \
|
|
((((dw) & 0xFF000000L) >> 24) | \
|
|
(((dw) & 0x00FF0000L) >> 8) | \
|
|
(((dw) & 0x0000FF00L) << 8) | \
|
|
(((dw) & 0x000000FFL) << 24))
|
|
|
|
#define SOCKET_STATE_INVALID_TRANSITION ((DWORD)-1)
|
|
#define SOCKET_STATE_CREATED 0
|
|
#define SOCKET_STATE_BOUND 1
|
|
#define SOCKET_STATE_CONNECTING 2
|
|
#define SOCKET_STATE_CONNECTED 3
|
|
#define SOCKET_STATE_LISTENING 4
|
|
#define SOCKET_STATE_MASK 0x0000ffff
|
|
#define SOCKET_STATE_EOF_READ 0x20000000
|
|
#define SOCKET_STATE_LOCKED 0x40000000
|
|
#define SOCKET_STATE_NEW 0x80000000
|
|
#define SOCKET_STATE_CLOSED 0x00000100
|
|
|
|
#define FUNCTION_CONNECT 0
|
|
#define FUNCTION_RECV 1
|
|
#define FUNCTION_SEND 2
|
|
#define FUNCTION_PREACCEPT 3
|
|
#define FUNCTION_ACCEPT 4
|
|
#define FUNCTION_CLOSE 5
|
|
#define MAX_FUNCTIONS 6
|
|
|
|
#define IN_FLIGHT_REQUESTS 4
|
|
|
|
#define EXTRA_LOCK_BUFFERS 2 /* Number of extra buffers needed
|
|
* for ancillary data on packet
|
|
* requests. */
|
|
|
|
#define DEFAULT_SEND_WINDOW_SIZE 16384
|
|
#define DEFAULT_RECEIVE_WINDOW_SIZE 16384
|
|
|
|
#define SGID_CONNECTIONLESS 1 /* XXX Find this flag */
|
|
|
|
/* XXX This is a hack we should clean up later
|
|
* We do this in order to get some storage for the locked handle table
|
|
* Maybe I'll use some tail item in the irp instead */
|
|
#define AFD_HANDLES(x) ((PAFD_HANDLE)(x)->Exclusive)
|
|
#define SET_AFD_HANDLES(x,y) (((x)->Exclusive) = (ULONG_PTR)(y))
|
|
|
|
typedef struct _AFD_MAPBUF {
|
|
PVOID BufferAddress;
|
|
PMDL Mdl;
|
|
} AFD_MAPBUF, *PAFD_MAPBUF;
|
|
|
|
typedef struct _AFD_DEVICE_EXTENSION {
|
|
PDEVICE_OBJECT DeviceObject;
|
|
LIST_ENTRY Polls;
|
|
KSPIN_LOCK Lock;
|
|
} AFD_DEVICE_EXTENSION, *PAFD_DEVICE_EXTENSION;
|
|
|
|
typedef struct _AFD_ACTIVE_POLL {
|
|
LIST_ENTRY ListEntry;
|
|
PIRP Irp;
|
|
PAFD_DEVICE_EXTENSION DeviceExt;
|
|
KDPC TimeoutDpc;
|
|
KTIMER Timer;
|
|
PKEVENT EventObject;
|
|
BOOLEAN Exclusive;
|
|
} AFD_ACTIVE_POLL, *PAFD_ACTIVE_POLL;
|
|
|
|
typedef struct _IRP_LIST {
|
|
LIST_ENTRY ListEntry;
|
|
PIRP Irp;
|
|
} IRP_LIST, *PIRP_LIST;
|
|
|
|
typedef struct _AFD_TDI_OBJECT {
|
|
PFILE_OBJECT Object;
|
|
HANDLE Handle;
|
|
} AFD_TDI_OBJECT, *PAFD_TDI_OBJECT;
|
|
|
|
typedef struct _AFD_TDI_OBJECT_QELT {
|
|
LIST_ENTRY ListEntry;
|
|
UINT Seq;
|
|
PTDI_CONNECTION_INFORMATION ConnInfo;
|
|
AFD_TDI_OBJECT Object;
|
|
} AFD_TDI_OBJECT_QELT, *PAFD_TDI_OBJECT_QELT;
|
|
|
|
typedef struct _AFD_IN_FLIGHT_REQUEST {
|
|
PIRP InFlightRequest;
|
|
IO_STATUS_BLOCK Iosb;
|
|
PTDI_CONNECTION_INFORMATION ConnectionCallInfo;
|
|
PTDI_CONNECTION_INFORMATION ConnectionReturnInfo;
|
|
} AFD_IN_FLIGHT_REQUEST, *PAFD_IN_FLIGHT_REQUEST;
|
|
|
|
typedef struct _AFD_DATA_WINDOW {
|
|
PCHAR Window;
|
|
UINT BytesUsed, Size, Content;
|
|
} AFD_DATA_WINDOW, *PAFD_DATA_WINDOW;
|
|
|
|
typedef struct _AFD_STORED_DATAGRAM {
|
|
LIST_ENTRY ListEntry;
|
|
UINT Len;
|
|
PTRANSPORT_ADDRESS Address;
|
|
CHAR Buffer[1];
|
|
} AFD_STORED_DATAGRAM, *PAFD_STORED_DATAGRAM;
|
|
|
|
typedef struct _AFD_FCB {
|
|
BOOLEAN Locked, Critical, Overread;
|
|
UINT State, Flags;
|
|
KIRQL OldIrql;
|
|
UINT LockCount;
|
|
PVOID CurrentThread;
|
|
KSPIN_LOCK SpinLock;
|
|
PFILE_OBJECT FileObject;
|
|
PAFD_DEVICE_EXTENSION DeviceExt;
|
|
BOOLEAN DelayedAccept, NeedsNewListen;
|
|
UINT ConnSeq;
|
|
PTRANSPORT_ADDRESS LocalAddress, RemoteAddress;
|
|
PTDI_CONNECTION_INFORMATION AddressFrom;
|
|
AFD_TDI_OBJECT AddressFile, Connection;
|
|
AFD_IN_FLIGHT_REQUEST ConnectIrp, ListenIrp, ReceiveIrp, SendIrp;
|
|
AFD_DATA_WINDOW Send, Recv;
|
|
FAST_MUTEX Mutex;
|
|
KEVENT StateLockedEvent;
|
|
PKEVENT EventSelect;
|
|
DWORD EventSelectTriggers;
|
|
DWORD EventsFired;
|
|
UNICODE_STRING TdiDeviceName;
|
|
PVOID Context;
|
|
DWORD PollState;
|
|
UINT ContextSize;
|
|
LIST_ENTRY PendingIrpList[MAX_FUNCTIONS];
|
|
LIST_ENTRY DatagramList;
|
|
LIST_ENTRY PendingConnections;
|
|
} AFD_FCB, *PAFD_FCB;
|
|
|
|
/* bind.c */
|
|
|
|
NTSTATUS WarmSocketForBind( PAFD_FCB FCB );
|
|
NTSTATUS NTAPI
|
|
AfdBindSocket(PDEVICE_OBJECT DeviceObject, PIRP Irp,
|
|
PIO_STACK_LOCATION IrpSp);
|
|
|
|
/* connect.c */
|
|
|
|
NTSTATUS MakeSocketIntoConnection( PAFD_FCB FCB );
|
|
NTSTATUS WarmSocketForConnection( PAFD_FCB FCB );
|
|
NTSTATUS NTAPI
|
|
AfdStreamSocketConnect(PDEVICE_OBJECT DeviceObject, PIRP Irp,
|
|
PIO_STACK_LOCATION IrpSp);
|
|
|
|
/* context.c */
|
|
|
|
NTSTATUS NTAPI
|
|
AfdGetContext( PDEVICE_OBJECT DeviceObject, PIRP Irp,
|
|
PIO_STACK_LOCATION IrpSp );
|
|
NTSTATUS NTAPI
|
|
AfdSetContext( PDEVICE_OBJECT DeviceObject, PIRP Irp,
|
|
PIO_STACK_LOCATION IrpSp );
|
|
|
|
/* info.c */
|
|
|
|
NTSTATUS NTAPI
|
|
AfdGetInfo( PDEVICE_OBJECT DeviceObject, PIRP Irp,
|
|
PIO_STACK_LOCATION IrpSp );
|
|
|
|
NTSTATUS NTAPI
|
|
AfdGetSockName( PDEVICE_OBJECT DeviceObject, PIRP Irp,
|
|
PIO_STACK_LOCATION IrpSp );
|
|
|
|
NTSTATUS NTAPI
|
|
AfdGetPeerName( PDEVICE_OBJECT DeviceObject, PIRP Irp,
|
|
PIO_STACK_LOCATION IrpSp );
|
|
|
|
/* listen.c */
|
|
NTSTATUS AfdWaitForListen( PDEVICE_OBJECT DeviceObject, PIRP Irp,
|
|
PIO_STACK_LOCATION IrpSp );
|
|
|
|
NTSTATUS AfdListenSocket(PDEVICE_OBJECT DeviceObject, PIRP Irp,
|
|
PIO_STACK_LOCATION IrpSp);
|
|
|
|
NTSTATUS AfdAccept( PDEVICE_OBJECT DeviceObject, PIRP Irp,
|
|
PIO_STACK_LOCATION IrpSp );
|
|
|
|
/* lock.c */
|
|
|
|
PAFD_WSABUF LockBuffers( PAFD_WSABUF Buf, UINT Count,
|
|
PVOID AddressBuf, PINT AddressLen,
|
|
BOOLEAN Write, BOOLEAN LockAddress );
|
|
VOID UnlockBuffers( PAFD_WSABUF Buf, UINT Count, BOOL Address );
|
|
UINT SocketAcquireStateLock( PAFD_FCB FCB );
|
|
NTSTATUS NTAPI UnlockAndMaybeComplete
|
|
( PAFD_FCB FCB, NTSTATUS Status, PIRP Irp,
|
|
UINT Information,
|
|
PIO_COMPLETION_ROUTINE Completion );
|
|
VOID SocketStateUnlock( PAFD_FCB FCB );
|
|
NTSTATUS LostSocket( PIRP Irp );
|
|
PAFD_HANDLE LockHandles( PAFD_HANDLE HandleArray, UINT HandleCount );
|
|
VOID UnlockHandles( PAFD_HANDLE HandleArray, UINT HandleCount );
|
|
PVOID LockRequest( PIRP Irp, PIO_STACK_LOCATION IrpSp );
|
|
VOID UnlockRequest( PIRP Irp, PIO_STACK_LOCATION IrpSp );
|
|
|
|
/* main.c */
|
|
|
|
VOID OskitDumpBuffer( PCHAR Buffer, UINT Len );
|
|
NTSTATUS LeaveIrpUntilLater( PAFD_FCB FCB, PIRP Irp, UINT Function );
|
|
VOID DestroySocket( PAFD_FCB FCB );
|
|
|
|
/* read.c */
|
|
|
|
NTSTATUS NTAPI ReceiveComplete
|
|
( PDEVICE_OBJECT DeviceObject,
|
|
PIRP Irp,
|
|
PVOID Context );
|
|
|
|
NTSTATUS NTAPI PacketSocketRecvComplete
|
|
( PDEVICE_OBJECT DeviceObject,
|
|
PIRP Irp,
|
|
PVOID Context );
|
|
|
|
NTSTATUS NTAPI
|
|
AfdConnectedSocketReadData(PDEVICE_OBJECT DeviceObject, PIRP Irp,
|
|
PIO_STACK_LOCATION IrpSp, BOOLEAN Short);
|
|
NTSTATUS NTAPI
|
|
AfdPacketSocketReadData(PDEVICE_OBJECT DeviceObject, PIRP Irp,
|
|
PIO_STACK_LOCATION IrpSp );
|
|
|
|
/* select.c */
|
|
|
|
NTSTATUS NTAPI
|
|
AfdSelect( PDEVICE_OBJECT DeviceObject, PIRP Irp,
|
|
PIO_STACK_LOCATION IrpSp );
|
|
NTSTATUS NTAPI
|
|
AfdEventSelect( PDEVICE_OBJECT DeviceObject, PIRP Irp,
|
|
PIO_STACK_LOCATION IrpSp );
|
|
NTSTATUS NTAPI
|
|
AfdEnumEvents( PDEVICE_OBJECT DeviceObject, PIRP Irp,
|
|
PIO_STACK_LOCATION IrpSp );
|
|
VOID PollReeval( PAFD_DEVICE_EXTENSION DeviceObject, PFILE_OBJECT FileObject );
|
|
VOID KillSelectsForFCB( PAFD_DEVICE_EXTENSION DeviceExt,
|
|
PFILE_OBJECT FileObject, BOOLEAN ExclusiveOnly );
|
|
|
|
/* tdi.c */
|
|
|
|
NTSTATUS TdiOpenAddressFile(
|
|
PUNICODE_STRING DeviceName,
|
|
PTRANSPORT_ADDRESS Name,
|
|
PHANDLE AddressHandle,
|
|
PFILE_OBJECT *AddressObject);
|
|
|
|
NTSTATUS TdiAssociateAddressFile(
|
|
HANDLE AddressHandle,
|
|
PFILE_OBJECT ConnectionObject);
|
|
|
|
NTSTATUS TdiListen
|
|
( PIRP *Irp,
|
|
PFILE_OBJECT ConnectionObject,
|
|
PTDI_CONNECTION_INFORMATION *RequestConnectionInfo,
|
|
PTDI_CONNECTION_INFORMATION *ReturnConnectionInfo,
|
|
PIO_STATUS_BLOCK Iosb,
|
|
PIO_COMPLETION_ROUTINE CompletionRoutine,
|
|
PVOID CompletionContext);
|
|
|
|
NTSTATUS TdiReceive
|
|
( PIRP *Irp,
|
|
PFILE_OBJECT ConnectionObject,
|
|
USHORT Flags,
|
|
PCHAR Buffer,
|
|
UINT BufferLength,
|
|
PIO_STATUS_BLOCK Iosb,
|
|
PIO_COMPLETION_ROUTINE CompletionRoutine,
|
|
PVOID CompletionContext);
|
|
|
|
NTSTATUS TdiSend
|
|
( PIRP *Irp,
|
|
PFILE_OBJECT ConnectionObject,
|
|
USHORT Flags,
|
|
PCHAR Buffer,
|
|
UINT BufferLength,
|
|
PIO_STATUS_BLOCK Iosb,
|
|
PIO_COMPLETION_ROUTINE CompletionRoutine,
|
|
PVOID CompletionContext);
|
|
|
|
NTSTATUS TdiReceiveDatagram(
|
|
PIRP *Irp,
|
|
PFILE_OBJECT TransportObject,
|
|
USHORT Flags,
|
|
PCHAR Buffer,
|
|
UINT BufferLength,
|
|
PTDI_CONNECTION_INFORMATION From,
|
|
PIO_STATUS_BLOCK Iosb,
|
|
PIO_COMPLETION_ROUTINE CompletionRoutine,
|
|
PVOID CompletionContext);
|
|
|
|
NTSTATUS TdiSendDatagram(
|
|
PIRP *Irp,
|
|
PFILE_OBJECT TransportObject,
|
|
PCHAR Buffer,
|
|
UINT BufferLength,
|
|
PTDI_CONNECTION_INFORMATION To,
|
|
PIO_STATUS_BLOCK Iosb,
|
|
PIO_COMPLETION_ROUTINE CompletionRoutine,
|
|
PVOID CompletionContext);
|
|
|
|
/* write.c */
|
|
|
|
NTSTATUS NTAPI
|
|
AfdConnectedSocketWriteData(PDEVICE_OBJECT DeviceObject, PIRP Irp,
|
|
PIO_STACK_LOCATION IrpSp, BOOLEAN Short);
|
|
NTSTATUS NTAPI
|
|
AfdPacketSocketWriteData(PDEVICE_OBJECT DeviceObject, PIRP Irp,
|
|
PIO_STACK_LOCATION IrpSp);
|
|
|
|
#endif/*_AFD_H*/
|