fileobjs.h File Reference

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Functions
NTSTATUS	FileOpenAddress (PTDI_REQUEST Request, PTA_IP_ADDRESS AddrList, USHORT Protocol, BOOLEAN Shared, PVOID Options)
NTSTATUS	FileCloseAddress (PTDI_REQUEST Request)
NTSTATUS	FileOpenConnection (PTDI_REQUEST Request, PVOID ClientContext)
NTSTATUS	FileCloseConnection (PTDI_REQUEST Request)
NTSTATUS	FileOpenControlChannel (PTDI_REQUEST Request)
NTSTATUS	FileCloseControlChannel (PTDI_REQUEST Request)
VOID	LogActiveObjects (VOID)

Variables
LIST_ENTRY	AddressFileListHead
KSPIN_LOCK	AddressFileListLock
LIST_ENTRY	ConnectionEndpointListHead
KSPIN_LOCK	ConnectionEndpointListLock

Function Documentation

FileCloseAddress()

NTSTATUS FileCloseAddress

(

PTDI_REQUEST

Request

)

Definition at line 591 of file fileobjs.c.

{
  PADDRESS_FILE AddrFile = Request->Handle.AddressHandle;
  PCONNECTION_ENDPOINT Listener;
 
  if (!Request->Handle.AddressHandle) return STATUS_INVALID_PARAMETER;
 
  LockObject(AddrFile);
 
  if (InterlockedDecrement(&AddrFile->Sharers) != 0)
  {
      /* Still other guys have open handles to this, so keep it around */
      UnlockObject(AddrFile);
      return STATUS_SUCCESS;
  }
 
  /* We have to close this listener because we started it */
  Listener = AddrFile->Listener;
  UnlockObject(AddrFile);
  if( Listener )
  {
      TCPClose( Listener );
  }
 
  DereferenceObject(AddrFile);
 
  TI_DbgPrint(MAX_TRACE, ("Leaving.\n"));
 
  return STATUS_SUCCESS;
}

Referenced by DispEchoRequest(), EndRequestHandler(), and TiCloseFileObject().

FileCloseConnection()

NTSTATUS FileCloseConnection

(

PTDI_REQUEST

Request

)

Definition at line 667 of file fileobjs.c.

{
  PCONNECTION_ENDPOINT Connection;
 
  TI_DbgPrint(MID_TRACE, ("Called.\n"));
 
  Connection = Request->Handle.ConnectionContext;
 
  if (!Connection) return STATUS_INVALID_PARAMETER;
 
  TCPClose( Connection );
 
  Request->Handle.ConnectionContext = NULL;
 
  TI_DbgPrint(MAX_TRACE, ("Leaving.\n"));
 
  return STATUS_SUCCESS;
}

Referenced by TiCloseFileObject().

FileCloseControlChannel()

NTSTATUS FileCloseControlChannel

(

PTDI_REQUEST

Request

)

Definition at line 738 of file fileobjs.c.

{
  if (!Request->Handle.ControlChannel) return STATUS_INVALID_PARAMETER;
 
  DereferenceObject((PCONTROL_CHANNEL)Request->Handle.ControlChannel);
 
  Request->Handle.ControlChannel = NULL;
 
  return STATUS_SUCCESS;
}

Referenced by TiCloseFileObject().

FileOpenAddress()

NTSTATUS FileOpenAddress	(	PTDI_REQUEST	Request,
		PTA_IP_ADDRESS	AddrList,
		USHORT	Protocol,
		BOOLEAN	Shared,
		PVOID	Options
	)

Definition at line 399 of file fileobjs.c.

{
  PADDRESS_FILE AddrFile;
  UINT AllocatedPort;
 
  TI_DbgPrint(MID_TRACE, ("Called (Proto %d).\n", Protocol));
 
  /* If it's shared and has a port specified, look for a match */
  if ((Shared != FALSE) && (Address->Address[0].Address[0].sin_port != 0))
  {
    AddrFile = AddrFindShared(NULL, Address->Address[0].Address[0].sin_port, Protocol);
    if (AddrFile != NULL)
    {
        Request->Handle.AddressHandle = AddrFile;
        return STATUS_SUCCESS;
    }
  }
 
  AddrFile = ExAllocatePoolWithTag(NonPagedPool, sizeof(ADDRESS_FILE),
                                   ADDR_FILE_TAG);
  if (!AddrFile) {
    TI_DbgPrint(MIN_TRACE, ("Insufficient resources.\n"));
    return STATUS_INSUFFICIENT_RESOURCES;
  }
 
  RtlZeroMemory(AddrFile, sizeof(ADDRESS_FILE));
 
  AddrFile->RefCount = 1;
  AddrFile->Free = AddrFileFree;
  AddrFile->Sharers = 1;
 
  /* Set our default options */
  AddrFile->TTL = 128;
  AddrFile->DF = 0;
  AddrFile->BCast = 1;
  AddrFile->HeaderIncl = 1;
  AddrFile->ProcessId = PsGetCurrentProcessId();
 
  _SEH2_TRY {
      PTEB Teb;
 
      Teb = PsGetCurrentThreadTeb();
      if (Teb != NULL)
         AddrFile->SubProcessTag = Teb->SubProcessTag;
  } _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER) {
      AddrFile->SubProcessTag = 0;
  } _SEH2_END;
 
  KeQuerySystemTime(&AddrFile->CreationTime);
 
  /* Make sure address is a local unicast address or 0 */
  /* FIXME: IPv4 only */
  AddrFile->Family = Address->Address[0].AddressType;
  AddrFile->Address.Address.IPv4Address = Address->Address[0].Address[0].in_addr;
  AddrFile->Address.Type = IP_ADDRESS_V4;
 
  if (!AddrIsUnspecified(&AddrFile->Address) &&
      !AddrLocateInterface(&AddrFile->Address)) {
      TI_DbgPrint(MIN_TRACE, ("Non-local address given (0x%X).\n", A2S(&AddrFile->Address)));
      ExFreePoolWithTag(AddrFile, ADDR_FILE_TAG);
      return STATUS_INVALID_ADDRESS;
  }
 
  TI_DbgPrint(MID_TRACE, ("Opening address %s for communication (P=%d U=%d).\n",
    A2S(&AddrFile->Address), Protocol, IPPROTO_UDP));
 
  /* Protocol specific handling */
  switch (Protocol) {
  case IPPROTO_TCP:
      if (Address->Address[0].Address[0].sin_port)
      {
          /* The client specified an explicit port so we force a bind to this */
          AllocatedPort = TCPAllocatePort(Address->Address[0].Address[0].sin_port);
 
          /* Check for bind success */
          if (AllocatedPort == (UINT)-1)
          {
              ExFreePoolWithTag(AddrFile, ADDR_FILE_TAG);
              return STATUS_ADDRESS_ALREADY_EXISTS;
          }
          AddrFile->Port = AllocatedPort;
 
          /* Sanity check */
          ASSERT(Address->Address[0].Address[0].sin_port == AddrFile->Port);
      }
      else if (!AddrIsUnspecified(&AddrFile->Address))
      {
          /* The client is trying to bind to a local address so allocate a port now too */
          AllocatedPort = TCPAllocatePort(0);
 
          /* Check for bind success */
          if (AllocatedPort == (UINT)-1)
          {
              ExFreePoolWithTag(AddrFile, ADDR_FILE_TAG);
              return STATUS_ADDRESS_ALREADY_EXISTS;
          }
          AddrFile->Port = AllocatedPort;
      }
      else
      {
          /* The client wants an unspecified port with an unspecified address so we wait to see what the TCP library gives us */
          AddrFile->Port = 0;
      }
 
      AddEntity(CO_TL_ENTITY, AddrFile, CO_TL_TCP);
 
      AddrFile->Send = NULL; /* TCPSendData */
      break;
 
  case IPPROTO_UDP:
      TI_DbgPrint(MID_TRACE,("Allocating udp port\n"));
      AllocatedPort = UDPAllocatePort(Address->Address[0].Address[0].sin_port);
 
      if ((Address->Address[0].Address[0].sin_port &&
           AllocatedPort != Address->Address[0].Address[0].sin_port) ||
           AllocatedPort == (UINT)-1)
      {
          ExFreePoolWithTag(AddrFile, ADDR_FILE_TAG);
          return STATUS_ADDRESS_ALREADY_EXISTS;
      }
      AddrFile->Port = AllocatedPort;
 
      TI_DbgPrint(MID_TRACE,("Setting port %d (wanted %d)\n",
                             AddrFile->Port,
                             Address->Address[0].Address[0].sin_port));
 
      AddEntity(CL_TL_ENTITY, AddrFile, CL_TL_UDP);
 
      AddrFile->Send = UDPSendDatagram;
      break;
 
  case IPPROTO_ICMP:
    AddrFile->Port = 0;
    AddrFile->Send = ICMPSendDatagram;
 
    /* FIXME: Verify this */
    AddEntity(ER_ENTITY, AddrFile, ER_ICMP);
    break;
 
  default:
    /* Use raw IP for all other protocols */
    AddrFile->Port = 0;
    AddrFile->Send = RawIPSendDatagram;
 
    /* FIXME: Verify this */
    AddEntity(CL_TL_ENTITY, AddrFile, 0);
    break;
  }
 
  TI_DbgPrint(MID_TRACE, ("IP protocol number for address file object is %d.\n",
    Protocol));
 
  TI_DbgPrint(MID_TRACE, ("Port number for address file object is %d.\n",
    WN2H(AddrFile->Port)));
 
  /* Set protocol */
  AddrFile->Protocol = Protocol;
 
  /* Initialize receive and transmit queues */
  InitializeListHead(&AddrFile->ReceiveQueue);
  InitializeListHead(&AddrFile->TransmitQueue);
 
  /* Initialize spin lock that protects the address file object */
  ExInitializeResourceLite(&AddrFile->Resource);
 
  /* Return address file object */
  Request->Handle.AddressHandle = AddrFile;
 
  /* Add address file to global list */
  ExInterlockedInsertTailList(
    &AddressFileListHead,
    &AddrFile->ListEntry,
    &AddressFileListLock);
 
  TI_DbgPrint(MAX_TRACE, ("Leaving.\n"));
 
  return STATUS_SUCCESS;
}

Referenced by DispEchoRequest(), and TiCreateFileObject().

FileOpenConnection()

NTSTATUS FileOpenConnection	(	PTDI_REQUEST	Request,
		PVOID	ClientContext
	)

Definition at line 632 of file fileobjs.c.

{
  NTSTATUS Status;
  PCONNECTION_ENDPOINT Connection;
 
  TI_DbgPrint(MID_TRACE, ("Called.\n"));
 
  Connection = TCPAllocateConnectionEndpoint( ClientContext );
 
  if( !Connection ) return STATUS_NO_MEMORY;
 
  Status = TCPSocket( Connection, AF_INET, SOCK_STREAM, IPPROTO_TCP );
 
  if( !NT_SUCCESS(Status) ) {
      DereferenceObject( Connection );
      return Status;
  }
 
  /* Return connection endpoint file object */
  Request->Handle.ConnectionContext = Connection;
 
  TI_DbgPrint(MAX_TRACE, ("Leaving.\n"));
 
  return STATUS_SUCCESS;
}

Referenced by TiCreateFileObject().

FileOpenControlChannel()

NTSTATUS FileOpenControlChannel

(

PTDI_REQUEST

Request

)

Definition at line 694 of file fileobjs.c.

{
  PCONTROL_CHANNEL ControlChannel;
  TI_DbgPrint(MID_TRACE, ("Called.\n"));
 
  ControlChannel = ExAllocatePoolWithTag(NonPagedPool, sizeof(*ControlChannel),
                                         CONTROL_CHANNEL_TAG);
 
  if (!ControlChannel) {
    TI_DbgPrint(MIN_TRACE, ("Insufficient resources.\n"));
    return STATUS_INSUFFICIENT_RESOURCES;
  }
 
  RtlZeroMemory(ControlChannel, sizeof(CONTROL_CHANNEL));
 
  /* Make sure address is a local unicast address or 0 */
 
  /* Locate address entry. If specified address is 0, a random address is chosen */
 
  /* Initialize receive and transmit queues */
  InitializeListHead(&ControlChannel->ListEntry);
 
  /* Initialize spin lock that protects the address file object */
  KeInitializeSpinLock(&ControlChannel->Lock);
 
  ControlChannel->RefCount = 1;
  ControlChannel->Free = ControlChannelFree;
 
  /* Return address file object */
  Request->Handle.ControlChannel = ControlChannel;
 
  TI_DbgPrint(MAX_TRACE, ("Leaving.\n"));
 
  return STATUS_SUCCESS;
}

Referenced by TiCreateFileObject().

LogActiveObjects()

VOID LogActiveObjects

(

VOID

)

Definition at line 109 of file fileobjs.c.

{
#ifdef LOG_OBJECTS
    PLIST_ENTRY CurrentEntry;
    KIRQL OldIrql;
    PADDRESS_FILE AddrFile;
    PCONNECTION_ENDPOINT Conn;
 
    DbgPrint("----------- TCP/IP Active Object Dump -------------\n");
 
    TcpipAcquireSpinLock(&AddressFileListLock, &OldIrql);
 
    CurrentEntry = AddressFileListHead.Flink;
    while (CurrentEntry != &AddressFileListHead)
    {
        AddrFile = CONTAINING_RECORD(CurrentEntry, ADDRESS_FILE, ListEntry);
 
        DbgPrint("Address File (%s, %d, %d) @ 0x%p | Ref count: %d | Sharers: %d\n",
                 A2S(&AddrFile->Address), WN2H(AddrFile->Port), AddrFile->Protocol,
                 AddrFile, AddrFile->RefCount, AddrFile->Sharers);
        DbgPrint("\tListener: ");
        if (AddrFile->Listener == NULL)
            DbgPrint("<None>\n");
        else
            DbgPrint("0x%p\n", AddrFile->Listener);
        DbgPrint("\tAssociated endpoints: ");
        if (AddrFile->Connection == NULL)
            DbgPrint("<None>\n");
        else
        {
            Conn = AddrFile->Connection;
            while (Conn)
            {
                DbgPrint("0x%p ", Conn);
                Conn = Conn->Next;
            }
            DbgPrint("\n");
        }
 
        CurrentEntry = CurrentEntry->Flink;
    }
 
    TcpipReleaseSpinLock(&AddressFileListLock, OldIrql);
 
    TcpipAcquireSpinLock(&ConnectionEndpointListLock, &OldIrql);
 
    CurrentEntry = ConnectionEndpointListHead.Flink;
    while (CurrentEntry != &ConnectionEndpointListHead)
    {
        Conn = CONTAINING_RECORD(CurrentEntry, CONNECTION_ENDPOINT, ListEntry);
 
        DbgPrint("Connection @ 0x%p | Ref count: %d\n", Conn, Conn->RefCount);
        DbgPrint("\tPCB: ");
        if (Conn->SocketContext == NULL)
            DbgPrint("<None>\n");
        else
        {
            DbgPrint("0x%p\n", Conn->SocketContext);
            LibTCPDumpPcb(Conn->SocketContext);
        }
        DbgPrint("\tPacket queue status: %s\n", IsListEmpty(&Conn->PacketQueue) ? "Empty" : "Not Empty");
        DbgPrint("\tRequest lists: Connect: %s | Recv: %s | Send: %s | Shutdown: %s | Listen: %s\n",
                 IsListEmpty(&Conn->ConnectRequest) ? "Empty" : "Not Empty",
                 IsListEmpty(&Conn->ReceiveRequest) ? "Empty" : "Not Empty",
                 IsListEmpty(&Conn->SendRequest) ? "Empty" : "Not Empty",
                 IsListEmpty(&Conn->ShutdownRequest) ? "Empty" : "Not Empty",
                 IsListEmpty(&Conn->ListenRequest) ? "Empty" : "Not Empty");
        DbgPrint("\tSend shutdown: %s\n", Conn->SendShutdown ? "Yes" : "No");
        DbgPrint("\tReceive shutdown: %s\n", Conn->ReceiveShutdown ? "Yes" : "No");
        if (Conn->ReceiveShutdown) DbgPrint("\tReceive shutdown status: 0x%x\n", Conn->ReceiveShutdownStatus);
        DbgPrint("\tClosing: %s\n", Conn->Closing ? "Yes" : "No");
 
        CurrentEntry = CurrentEntry->Flink;
    }
 
    TcpipReleaseSpinLock(&ConnectionEndpointListLock, OldIrql);
 
    DbgPrint("---------------------------------------------------\n");
#endif
}

Referenced by IPTimeoutDpcFn().

Variable Documentation

AddressFileListHead

LIST_ENTRY AddressFileListHead

extern

Definition at line 21 of file fileobjs.c.

Referenced by DriverEntry(), and TiUnload().

AddressFileListLock

KSPIN_LOCK AddressFileListLock

extern

Definition at line 22 of file fileobjs.c.

Referenced by DriverEntry(), and TiUnload().

ConnectionEndpointListHead

LIST_ENTRY ConnectionEndpointListHead

extern

Definition at line 25 of file fileobjs.c.

Referenced by DriverEntry(), LogActiveObjects(), and TCPAllocateConnectionEndpoint().

ConnectionEndpointListLock

KSPIN_LOCK ConnectionEndpointListLock

extern

Definition at line 26 of file fileobjs.c.

Referenced by ConnectionFree(), DriverEntry(), LogActiveObjects(), and TCPAllocateConnectionEndpoint().