diff options
| author | Guo Mang <mang.guo@intel.com> | 2017-08-02 09:54:47 +0800 |
|---|---|---|
| committer | Guo Mang <mang.guo@intel.com> | 2017-09-05 19:45:08 +0800 |
| commit | 6c128c65b5ec0e5b8b5a0ccb165f3afd29e485f8 (patch) | |
| tree | 444372d92a0ae8991fe4d15eb3937df43690dfda /AppPkg/Applications/Sockets/TftpServer | |
| parent | b207c6434d7a5a4502975d322312e07017e8a8cb (diff) | |
| download | edk2-platforms-6c128c65b5ec0e5b8b5a0ccb165f3afd29e485f8.tar.xz | |
Remove core packages since we can get them from edk2 repository
Contributed-under: TianoCore Contribution Agreement 1.0
Signed-off-by: Guo Mang <mang.guo@intel.com>
Diffstat (limited to 'AppPkg/Applications/Sockets/TftpServer')
| -rw-r--r-- | AppPkg/Applications/Sockets/TftpServer/TftpServer.c | 2367 | ||||
| -rw-r--r-- | AppPkg/Applications/Sockets/TftpServer/TftpServer.h | 431 | ||||
| -rw-r--r-- | AppPkg/Applications/Sockets/TftpServer/TftpServer.inf | 70 |
3 files changed, 0 insertions, 2868 deletions
diff --git a/AppPkg/Applications/Sockets/TftpServer/TftpServer.c b/AppPkg/Applications/Sockets/TftpServer/TftpServer.c deleted file mode 100644 index 30d9dae41d..0000000000 --- a/AppPkg/Applications/Sockets/TftpServer/TftpServer.c +++ /dev/null @@ -1,2367 +0,0 @@ -/** @file
- This is a simple TFTP server application
-
- Copyright (c) 2011, 2012, Intel Corporation
- All rights reserved. This program and the accompanying materials
- are licensed and made available under the terms and conditions of the BSD License
- which accompanies this distribution. The full text of the license may be found at
- http://opensource.org/licenses/bsd-license.php
-
- THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,
- WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
-
-**/
-
-#include <TftpServer.h>
-
-TSDT_TFTP_SERVER mTftpServer; ///< TFTP server's control structure
-volatile BOOLEAN mbTftpServerExit; ///< Set TRUE to cause TFTP server to exit
-
-
-/**
- Read file data into a buffer
-
- @param [in] pContext Connection context structure address
-
- @retval TRUE if a read error occurred
-
-**/
-BOOLEAN
-BufferFill (
- IN TSDT_CONNECTION_CONTEXT * pContext
- )
-{
- BOOLEAN bReadError;
- size_t BytesRead;
- UINT64 LengthInBytes;
-
- DBG_ENTER ( );
-
- //
- // Use break instead of goto
- //
- bReadError = FALSE;
- for ( ; ; ) {
- //
- // Determine if there is any work to do
- //
- LengthInBytes = DIM ( pContext->FileData ) >> 1;
- if (( pContext->ValidBytes > LengthInBytes )
- || ( 0 == pContext->BytesRemaining )) {
- break;
- }
-
- //
- // Determine the number of bytes to read
- //
- if ( LengthInBytes > pContext->BytesRemaining ) {
- LengthInBytes = pContext->BytesRemaining;
- }
-
- //
- // Read in the next portion of the file
- //
- BytesRead = fread ( pContext->pFill,
- 1,
- (size_t)LengthInBytes,
- pContext->File );
- if ( -1 == BytesRead ) {
- bReadError = TRUE;
- break;
- }
-
- //
- // Account for the file data read
- //
- pContext->BytesRemaining -= BytesRead;
- pContext->ValidBytes += BytesRead;
- DEBUG (( DEBUG_FILE_BUFFER,
- "0x%08x: Buffer filled with %Ld bytes, %Ld bytes ramaining\r\n",
- pContext->pFill,
- BytesRead,
- pContext->BytesRemaining ));
-
- //
- // Set the next buffer location
- //
- pContext->pFill += BytesRead;
- if ( pContext->pEnd <= pContext->pFill ) {
- pContext->pFill = &pContext->FileData[ 0 ];
- }
-
- //
- // Verify that the end of the buffer is reached
- //
- ASSERT ( 0 == ( DIM ( pContext->FileData ) & 1 ));
- break;
- }
-
- //
- // Return the read status
- //
- DBG_EXIT ( );
- return bReadError;
-}
-
-
-/**
- Add a connection context to the list of connection contexts.
-
- @param [in] pTftpServer Address of the ::TSDT_TFTP_SERVER structure
- @param [in] SocketFd Socket file descriptor
-
- @retval Context structure address, NULL if allocation fails
-
-**/
-TSDT_CONNECTION_CONTEXT *
-ContextAdd (
- IN TSDT_TFTP_SERVER * pTftpServer,
- IN int SocketFd
- )
-{
- TSDT_CONNECTION_CONTEXT * pContext;
- TFTP_PACKET * pEnd;
- TFTP_PACKET * pPacket;
-
- DBG_ENTER ( );
-
- //
- // Allocate a new context
- //
- pContext = (TSDT_CONNECTION_CONTEXT *)AllocateZeroPool ( sizeof ( *pContext ));
- if ( NULL != pContext ) {
- //
- // Initialize the context
- //
- pContext->SocketFd = SocketFd;
- CopyMem ( &pContext->RemoteAddress,
- &pTftpServer->RemoteAddress,
- sizeof ( pContext->RemoteAddress ));
- pContext->BlockSize = 512;
-
- //
- // Buffer management
- //
- pContext->pFill = &pContext->FileData[ 0 ];
- pContext->pEnd = &pContext->FileData[ sizeof ( pContext->FileData )];
- pContext->pBuffer = pContext->pFill;
-
- //
- // Window management
- //
- pContext->MaxTimeout = MultU64x32 ( PcdGet32 ( Tftp_MaxTimeoutInSec ),
- 2 * 1000 * 1000 * 1000 );
- pContext->Rtt2x = pContext->MaxTimeout;
- pContext->WindowSize = MAX_PACKETS;
- WindowTimeout ( pContext );
-
- //
- // Place the packets on the free list
- //
- pPacket = &pContext->Tx[ 0 ];
- pEnd = &pPacket[ DIM ( pContext->Tx )];
- while ( pEnd > pPacket ) {
- PacketFree ( pContext, pPacket );
- pPacket += 1;
- }
-
- //
- // Display the new context
- //
- if ( AF_INET == pTftpServer->RemoteAddress.v4.sin_family ) {
- DEBUG (( DEBUG_PORT_WORK,
- "0x%08x: Context for %d.%d.%d.%d:%d\r\n",
- pContext,
- (UINT8)pTftpServer->RemoteAddress.v4.sin_addr.s_addr,
- (UINT8)( pTftpServer->RemoteAddress.v4.sin_addr.s_addr >> 8 ),
- (UINT8)( pTftpServer->RemoteAddress.v4.sin_addr.s_addr >> 16 ),
- (UINT8)( pTftpServer->RemoteAddress.v4.sin_addr.s_addr >> 24 ),
- htons ( pTftpServer->RemoteAddress.v4.sin_port )));
- }
- else {
- DEBUG (( DEBUG_PORT_WORK,
- "0x%08x: Context for [%02x%02x:%02x%02x:%02x%02x:%02x%02x:%02x%02x:%02x%02x:%02x%02x:%02x%02x]:%d\r\n",
- pContext,
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 0 ],
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 1 ],
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 2 ],
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 3 ],
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 4 ],
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 5 ],
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 6 ],
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 7 ],
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 8 ],
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 9 ],
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 10 ],
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 11 ],
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 12 ],
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 13 ],
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 14 ],
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 15 ],
- htons ( pTftpServer->RemoteAddress.v6.sin6_port )));
- }
-
- //
- // Add the context to the context list
- //
- pContext->pNext = pTftpServer->pContextList;
- pTftpServer->pContextList = pContext;
- }
-
- //
- // Return the connection context
- //
- DBG_EXIT_STATUS ( pContext );
- return pContext;
-}
-
-
-/**
- Locate a remote connection context.
-
- @param [in] pTftpServer Address of the ::TSDT_TFTP_SERVER structure
- @param [in] pIpAddress The start of the remote IP address in network order
- @param [in] Port The remote port number
-
- @retval Context structure address, NULL if not found
-
-**/
-TSDT_CONNECTION_CONTEXT *
-ContextFind (
- IN TSDT_TFTP_SERVER * pTftpServer
- )
-{
- TSDT_CONNECTION_CONTEXT * pContext;
-
- DBG_ENTER ( );
-
- //
- // Walk the list of connection contexts
- //
- pContext = pTftpServer->pContextList;
- while ( NULL != pContext ) {
- //
- // Attempt to locate the remote network connection
- //
- if ( 0 == memcmp ( &pTftpServer->RemoteAddress,
- &pContext->RemoteAddress,
- pTftpServer->RemoteAddress.v6.sin6_len )) {
- //
- // The connection was found
- //
- DEBUG (( DEBUG_TFTP_REQUEST,
- "0x%08x: pContext found\r\n",
- pContext ));
- break;
- }
-
- //
- // Set the next context
- //
- pContext = pContext->pNext;
- }
-
- //
- // Return the connection context structure address
- //
- DBG_EXIT_HEX ( pContext );
- return pContext;
-}
-
-
-/**
- Remove a context from the list.
-
- @param [in] pTftpServer Address of the ::TSDT_TFTP_SERVER structure
- @param [in] pContext Address of a ::TSDT_CONNECTION_CONTEXT structure
-
-**/
-VOID
-ContextRemove (
- IN TSDT_TFTP_SERVER * pTftpServer,
- IN TSDT_CONNECTION_CONTEXT * pContext
- )
-{
- TSDT_CONNECTION_CONTEXT * pNextContext;
- TSDT_CONNECTION_CONTEXT * pPreviousContext;
-
- DBG_ENTER ( );
-
- //
- // Attempt to locate the context in the list
- //
- pPreviousContext = NULL;
- pNextContext = pTftpServer->pContextList;
- while ( NULL != pNextContext ) {
- //
- // Determine if the context was found
- //
- if ( pNextContext == pContext ) {
- //
- // Remove the context from the list
- //
- if ( NULL == pPreviousContext ) {
- pTftpServer->pContextList = pContext->pNext;
- }
- else {
- pPreviousContext->pNext = pContext->pNext;
- }
- break;
- }
-
- //
- // Set the next context
- //
- pPreviousContext = pNextContext;
- pNextContext = pNextContext->pNext;
- }
-
- //
- // Determine if the context was found
- //
- if ( NULL != pContext ) {
- //
- // Return the resources
- //
- gBS->FreePool ( pContext );
- }
-
- DBG_EXIT ( );
-}
-
-
-/**
- Queue data packets for transmission
-
- @param [in] pContext Connection context structure address
-
- @retval TRUE if a read error occurred
-
-**/
-BOOLEAN
-PacketFill (
- IN TSDT_CONNECTION_CONTEXT * pContext
- )
-{
- BOOLEAN bReadError;
- UINT64 LengthInBytes;
- UINT8 * pBuffer;
- TFTP_PACKET * pPacket;
-
- DBG_ENTER ( );
-
- //
- // Use break instead of goto
- //
- bReadError = FALSE;
- for ( ; ; ) {
- //
- // Fill the buffer if necessary
- //
- bReadError = BufferFill ( pContext );
- if ( bReadError ) {
- //
- // File access mode not supported
- //
- DEBUG (( DEBUG_ERROR | DEBUG_TFTP_REQUEST,
- "ERROR - File read failure!\r\n" ));
-
- //
- // Tell the client of the error
- //
- SendError ( pContext,
- TFTP_ERROR_SEE_MSG,
- (UINT8 *)"Read failure" );
- break;
- }
-
- //
- // Determine if any packets can be filled
- //
- if ( pContext->bEofSent
- || ( NULL == pContext->pFreeList )) {
- //
- // All of the packets are filled
- //
- break;
- }
-
- //
- // Set the TFTP opcode and block number
- //
- pPacket = PacketGet ( pContext );
- pBuffer = &pPacket->TxBuffer[ 0 ];
- *pBuffer++ = 0;
- *pBuffer++ = TFTP_OP_DATA;
- *pBuffer++ = (UINT8)( pContext->BlockNumber >> 8 );
- *pBuffer++ = (UINT8)pContext->BlockNumber;
-
- //
- // Determine how much data needs to be sent
- //
- LengthInBytes = pContext->BlockSize;
- if (( pContext->BytesToSend < TFTP_MAX_BLOCK_SIZE )
- && ( LengthInBytes > pContext->BytesToSend )) {
- LengthInBytes = pContext->BytesToSend;
- pContext->bEofSent = TRUE;
- }
- DEBUG (( DEBUG_TX_PACKET,
- "0x%08x: Packet, Block %d filled with %d bytes\r\n",
- pPacket,
- pContext->BlockNumber,
- (UINT32)LengthInBytes ));
-
- //
- // Copy the file data into the packet
- //
- pPacket->TxBytes = (ssize_t)( 2 + 2 + LengthInBytes );
- if ( 0 < LengthInBytes ) {
- CopyMem ( pBuffer,
- pContext->pBuffer,
- (UINTN)LengthInBytes );
- DEBUG (( DEBUG_FILE_BUFFER,
- "0x%08x: Buffer consumed %d bytes of file data\r\n",
- pContext->pBuffer,
- LengthInBytes ));
-
- //
- // Account for the file data consumed
- //
- pContext->ValidBytes -= LengthInBytes;
- pContext->BytesToSend -= LengthInBytes;
- pContext->pBuffer += LengthInBytes;
- if ( pContext->pEnd <= pContext->pBuffer ) {
- pContext->pBuffer = &pContext->FileData[ 0 ];
- }
- }
-
- //
- // Queue the packet for transmission
- //
- PacketQueue ( pContext, pPacket );
- }
-
- //
- // Return the read status
- //
- DBG_EXIT ( );
- return bReadError;
-}
-
-
-/**
- Free the packet
-
- @param [in] pContext Address of a ::TSDT_CONNECTION_CONTEXT structure
- @param [in] pPacket Address of a ::TFTP_PACKET structure
-
-**/
-VOID
-PacketFree(
- IN TSDT_CONNECTION_CONTEXT * pContext,
- IN TFTP_PACKET * pPacket
- )
-{
- DBG_ENTER ( );
-
- //
- // Don't free the error packet
- //
- if ( pPacket != &pContext->ErrorPacket ) {
- //
- // Place the packet on the free list
- //
- pPacket->pNext = pContext->pFreeList;
- pContext->pFreeList = pPacket;
- DEBUG (( DEBUG_TX_PACKET,
- "0x%08x: Packet queued to free list\r\n",
- pPacket ));
- }
-
- DBG_EXIT ( );
-}
-
-
-/**
- Get a packet from the free list for transmission
-
- @param [in] pContext Address of a ::TSDT_CONNECTION_CONTEXT structure
-
- @retval Address of a ::TFTP_PACKET structure
-
-**/
-TFTP_PACKET *
-PacketGet (
- IN TSDT_CONNECTION_CONTEXT * pContext
- )
-{
- TFTP_PACKET * pPacket;
-
- DBG_ENTER ( );
-
- //
- // Get the next packet from the free list
- //
- pPacket = pContext->pFreeList;
- if ( NULL != pPacket ) {
- pContext->pFreeList = pPacket->pNext;
- pPacket->RetryCount = 0;
- DEBUG (( DEBUG_TX_PACKET,
- "0x%08x: Packet removed from free list\r\n",
- pPacket ));
- }
-
- //
- // Return the packet
- //
- DBG_EXIT_HEX ( pPacket );
- return pPacket;
-}
-
-
-/**
- Queue the packet for transmission
-
- @param [in] pContext Address of a ::TSDT_CONNECTION_CONTEXT structure
- @param [in] pPacket Address of a ::TFTP_PACKET structure
-
- @retval TRUE if a transmission error has occurred
-
-**/
-BOOLEAN
-PacketQueue (
- IN TSDT_CONNECTION_CONTEXT * pContext,
- IN TFTP_PACKET * pPacket
- )
-{
- BOOLEAN bTransmitError;
- TFTP_PACKET * pTail;
- EFI_STATUS Status;
-
- DBG_ENTER ( );
-
- //
- // Account for this data block
- //
- pPacket->BlockNumber = pContext->BlockNumber;
- pContext->BlockNumber += 1;
-
- //
- // Queue the packet for transmission
- //
- pTail = pContext->pTxTail;
- if ( NULL == pTail ) {
- pContext->pTxHead = pPacket;
- }
- else {
- pTail->pNext = pPacket;
- }
- pContext->pTxTail = pPacket;
- pPacket->pNext = NULL;
- DEBUG (( DEBUG_TX_PACKET,
- "0x%08x: Packet queued to TX list\r\n",
- pPacket ));
-
- //
- // Start the transmission if necessary
- //
- bTransmitError = FALSE;
- if ( pContext->PacketsInWindow < pContext->WindowSize ) {
- Status = PacketTx ( pContext, pPacket );
- bTransmitError = (BOOLEAN)( EFI_ERROR ( Status ));
- }
-
- //
- // Return the transmit status
- //
- DBG_EXIT_TF ( bTransmitError );
- return bTransmitError;
-}
-
-
-/**
- Remove a packet from the transmit queue
-
- @param [in] pContext Address of a ::TSDT_CONNECTION_CONTEXT structure
-
-**/
-TFTP_PACKET *
-PacketRemove(
- IN TSDT_CONNECTION_CONTEXT * pContext
- )
-{
- TFTP_PACKET * pNext;
- TFTP_PACKET * pPacket;
-
- DBG_ENTER ( );
-
- //
- // Remove a packet from the transmit queue
- //
- //
- pPacket = pContext->pTxHead;
- if ( NULL != pPacket ) {
- pNext = pPacket->pNext;
- pContext->pTxHead = pNext;
- if ( NULL == pNext ) {
- pContext->pTxTail = NULL;
- }
- DEBUG (( DEBUG_TX_PACKET,
- "0x%08x: Packet removed from TX list\r\n",
- pPacket ));
-
- //
- // Remove this packet from the window
- //
- pContext->PacketsInWindow -= 1;
- }
-
- //
- // Return the packet
- //
- DBG_EXIT_HEX ( pPacket );
- return pPacket;
-}
-
-
-/**
- Transmit the packet
-
- @param [in] pContext Address of a ::TSDT_CONNECTION_CONTEXT structure
- @param [in] pPacket Address of a ::TFTP_PACKET structure
-
- @retval EFI_SUCCESS Message processed successfully
-
-**/
-EFI_STATUS
-PacketTx (
- IN TSDT_CONNECTION_CONTEXT * pContext,
- IN TFTP_PACKET * pPacket
- )
-{
- ssize_t LengthInBytes;
- EFI_STATUS Status;
-
- DBG_ENTER ( );
-
- //
- // Assume success
- //
- Status = EFI_SUCCESS;
-
- //
- // Determine if this packet should be transmitted
- //
- if ( PcdGet32 ( Tftp_MaxRetry ) >= pPacket->RetryCount ) {
- pPacket->RetryCount += 1;
-
- //
- // Display the operation
- //
- DEBUG (( DEBUG_TX_PACKET,
- "0x%08x: Packet transmiting\r\n",
- pPacket ));
- DEBUG (( DEBUG_TX,
- "0x%08x: pContext sending 0x%08x bytes\r\n",
- pContext,
- pPacket->TxBytes ));
-
- //
- // Keep track of when the packet was transmitted
- //
- if ( PcdGetBool ( Tftp_HighSpeed )) {
- pPacket->TxTime = GetPerformanceCounter ( );
- }
-
- //
- // Send the TFTP packet
- //
- pContext->PacketsInWindow += 1;
- LengthInBytes = sendto ( pContext->SocketFd,
- &pPacket->TxBuffer[ 0 ],
- pPacket->TxBytes,
- 0,
- (struct sockaddr *)&pContext->RemoteAddress,
- pContext->RemoteAddress.sin6_len );
- if ( -1 == LengthInBytes ) {
- DEBUG (( DEBUG_ERROR | DEBUG_TX,
- "ERROR - Transmit failure, errno: 0x%08x\r\n",
- errno ));
- pContext->PacketsInWindow -= 1;
- Status = EFI_DEVICE_ERROR;
- }
- }
- else {
- //
- // Too many retries
- //
- Status = EFI_NO_RESPONSE;
- DEBUG (( DEBUG_WARN | DEBUG_WINDOW,
- "WARNING - No response from TFTP client\r\n" ));
- }
-
- //
- // Return the operation status
- //
- DBG_EXIT_STATUS ( Status );
- return Status;
-}
-
-
-/**
- Process the work for the sockets.
-
- @param [in] pTftpServer Address of the ::TSDT_TFTP_SERVER structure
- @param [in] pIndex Address of an index into the pollfd array
-
-**/
-VOID
-PortWork (
- IN TSDT_TFTP_SERVER * pTftpServer,
- IN int * pIndex
- )
-{
- int Index;
- TSDT_CONNECTION_CONTEXT * pContext;
- struct pollfd * pTftpPort;
- socklen_t RemoteAddressLength;
- int revents;
-
- DBG_ENTER ( );
-
- //
- // Locate the port
- //
- Index = *pIndex;
- if ( -1 != Index ) {
- pTftpPort = &pTftpServer->TftpPort[ *pIndex ];
-
- //
- // Handle input events
- //
- revents = pTftpPort->revents;
- pTftpPort->revents = 0;
- if ( 0 != ( revents & POLLRDNORM )) {
- //
- // Receive the message from the remote system
- //
- RemoteAddressLength = sizeof ( pTftpServer->RemoteAddress );
- pTftpServer->RxBytes = recvfrom ( pTftpPort->fd,
- &pTftpServer->RxBuffer[ 0 ],
- sizeof ( pTftpServer->RxBuffer ),
- 0,
- (struct sockaddr *) &pTftpServer->RemoteAddress,
- &RemoteAddressLength );
- if ( -1 != pTftpServer->RxBytes ) {
- if ( PcdGetBool ( Tftp_HighSpeed )) {
- pTftpServer->RxTime = GetPerformanceCounter ( );
- }
- if ( AF_INET == pTftpServer->RemoteAddress.v4.sin_family ) {
- DEBUG (( DEBUG_TFTP_PORT,
- "Received %d bytes from %d.%d.%d.%d:%d\r\n",
- pTftpServer->RxBytes,
- pTftpServer->RemoteAddress.v4.sin_addr.s_addr & 0xff,
- ( pTftpServer->RemoteAddress.v4.sin_addr.s_addr >> 8 ) & 0xff,
- ( pTftpServer->RemoteAddress.v4.sin_addr.s_addr >> 16 ) & 0xff,
- ( pTftpServer->RemoteAddress.v4.sin_addr.s_addr >> 24 ) & 0xff,
- htons ( pTftpServer->RemoteAddress.v4.sin_port )));
- }
- else {
- DEBUG (( DEBUG_TFTP_PORT,
- "Received %d bytes from [%02x%02x:%02x%02x:%02x%02x:%02x%02x:%02x%02x:%02x%02x:%02x%02x:%02x%02x]:%d\r\n",
- pTftpServer->RxBytes,
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 0 ],
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 1 ],
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 2 ],
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 3 ],
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 4 ],
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 5 ],
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 6 ],
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 7 ],
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 8 ],
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 9 ],
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 10 ],
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 11 ],
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 12 ],
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 13 ],
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 14 ],
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 15 ],
- htons ( pTftpServer->RemoteAddress.v6.sin6_port )));
- }
-
- //
- // Lookup connection context using the remote system address and port
- // to determine if an existing connection to this remote
- // system exists
- //
- pContext = ContextFind ( pTftpServer );
-
- //
- // Process the received message
- //
- TftpProcessRequest ( pTftpServer, pContext, pTftpPort->fd );
- }
- else {
- //
- // Receive error on the TFTP server port
- // Close the server socket
- //
- DEBUG (( DEBUG_ERROR,
- "ERROR - Failed receive on TFTP server port, errno: 0x%08x\r\n",
- errno ));
- revents |= POLLHUP;
- }
- }
-
- //
- // Handle the close event
- //
- if ( 0 != ( revents & POLLHUP )) {
- //
- // Close the port
- //
- close ( pTftpPort->fd );
- pTftpPort->fd = -1;
- *pIndex = -1;
- pTftpServer->Entries -= 1;
- ASSERT ( 0 <= pTftpServer->Entries );
- }
- }
-
- DBG_EXIT ( );
-}
-
-
-/**
- Build and send an error packet
-
- @param [in] pContext Address of a ::TSDT_CONNECTION_CONTEXT structure
- @param [in] Error Error number for the packet
- @param [in] pError Zero terminated error string address
-
- @retval EFI_SUCCESS Message processed successfully
-
-**/
-EFI_STATUS
-SendError (
- IN TSDT_CONNECTION_CONTEXT * pContext,
- IN UINT16 Error,
- IN UINT8 * pError
- )
-{
- UINT8 Character;
- UINT8 * pBuffer;
- TFTP_PACKET * pPacket;
- EFI_STATUS Status;
-
- DBG_ENTER ( );
-
- //
- // Build the error packet
- //
- pPacket = &pContext->ErrorPacket;
- pBuffer = &pPacket->TxBuffer[ 0 ];
- pBuffer[ 0 ] = 0;
- pBuffer[ 1 ] = TFTP_OP_ERROR;
- pBuffer[ 2 ] = (UINT8)( Error >> 8 );
- pBuffer[ 3 ] = (UINT8)Error;
-
- //
- // Copy the zero terminated string into the buffer
- //
- pBuffer += 4;
- do {
- Character = *pError++;
- *pBuffer++ = Character;
- } while ( 0 != Character );
-
- //
- // Send the error message
- //
- pPacket->TxBytes = pBuffer - &pPacket->TxBuffer[ 0 ];
- Status = PacketTx ( pContext, pPacket );
-
- //
- // Return the operation status
- //
- DBG_EXIT_STATUS ( Status );
- return Status;
-}
-
-
-/**
- Scan the list of sockets and process any pending work
-
- @param [in] pTftpServer Address of the ::TSDT_TFTP_SERVER structure
-
-**/
-VOID
-SocketPoll (
- IN TSDT_TFTP_SERVER * pTftpServer
- )
-{
- int FDCount;
-
- DEBUG (( DEBUG_SOCKET_POLL, "Entering SocketPoll\r\n" ));
-
- //
- // Determine if any ports are active
- //
- if ( 0 != pTftpServer->Entries ) {
- FDCount = poll ( &pTftpServer->TftpPort[ 0 ],
- pTftpServer->Entries,
- CLIENT_POLL_DELAY );
- if ( 0 < FDCount ) {
- //
- // Process this port
- //
- PortWork ( pTftpServer, &pTftpServer->Udpv4Index );
- PortWork ( pTftpServer, &pTftpServer->Udpv6Index );
- }
- }
-
- DEBUG (( DEBUG_SOCKET_POLL, "Exiting SocketPoll\r\n" ));
-}
-
-
-/**
- Process the ACK
-
- @param [in] pTftpServer Address of the ::TSDT_TFTP_SERVER structure
- @param [in] pContext Connection context structure address
-
- @retval TRUE if the context should be closed
-
-**/
-BOOLEAN
-TftpAck (
- IN TSDT_TFTP_SERVER * pTftpServer,
- IN TSDT_CONNECTION_CONTEXT * pContext
- )
-{
- INTN AckNumber;
- BOOLEAN bCloseContext;
- UINT16 BlockNumber;
- UINT8 * pBuffer;
- TFTP_PACKET * pPacket;
- EFI_STATUS Status;
-
- DBG_ENTER ( );
-
- //
- // Use break instead of goto
- //
- bCloseContext = FALSE;
- for ( ; ; ) {
- //
- // Validate the parameters
- //
- if ( NULL == pContext ) {
- if ( AF_INET == pTftpServer->RemoteAddress.v4.sin_family ) {
- DEBUG (( DEBUG_ERROR,
- "ERROR - File not open for %d.%d.%d.%d:%d\r\n",
- (UINT8)pTftpServer->RemoteAddress.v4.sin_addr.s_addr,
- (UINT8)( pTftpServer->RemoteAddress.v4.sin_addr.s_addr >> 8 ),
- (UINT8)( pTftpServer->RemoteAddress.v4.sin_addr.s_addr >> 16 ),
- (UINT8)( pTftpServer->RemoteAddress.v4.sin_addr.s_addr >> 24 ),
- htons ( pTftpServer->RemoteAddress.v4.sin_port )));
- }
- else {
- DEBUG (( DEBUG_ERROR,
- "ERROR - File not open for [%02x%02x:%02x%02x:%02x%02x:%02x%02x:%02x%02x:%02x%02x:%02x%02x:%02x%02x]:%d\r\n",
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 0 ],
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 1 ],
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 2 ],
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 3 ],
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 4 ],
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 5 ],
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 6 ],
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 7 ],
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 8 ],
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 9 ],
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 10 ],
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 11 ],
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 12 ],
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 13 ],
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 14 ],
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 15 ],
- htons ( pTftpServer->RemoteAddress.v6.sin6_port )));
- }
- break;
- }
-
- //
- // Verify that the ACK was expected
- //
- pPacket = pContext->pTxHead;
- if ( NULL == pPacket ) {
- //
- // ACK not expected!
- //
- DEBUG (( DEBUG_ERROR,
- "ERROR - Expecting data not ACKs for pContext 0x%08x\r\n",
- pContext ));
- break;
- }
-
- //
- // Get the ACKed block number
- //
- pBuffer = &pTftpServer->RxBuffer[ 0 ];
- BlockNumber = HTONS ( *(UINT16 *)&pBuffer[ 2 ]);
-
- //
- // Determine if this is the correct ACK
- //
- DEBUG (( DEBUG_TFTP_ACK,
- "ACK for block 0x%04x received\r\n",
- BlockNumber ));
- AckNumber = BlockNumber - pPacket->BlockNumber;
- if (( 0 > AckNumber ) || ( AckNumber >= (INTN)pContext->PacketsInWindow )){
- DEBUG (( DEBUG_WARN | DEBUG_TFTP_ACK,
- "WARNING - Expecting ACK 0x%0x4 not received ACK 0x%08x\r\n",
- pPacket->BlockNumber,
- BlockNumber ));
- break;
- }
-
- //
- // Release the ACKed packets
- //
- do {
- //
- // Remove the packet from the transmit list and window
- //
- pPacket = PacketRemove ( pContext );
-
- //
- // Get the block number of this packet
- //
- AckNumber = pPacket->BlockNumber;
-
- //
- // Increase the size of the transmit window
- //
- if ( PcdGetBool ( Tftp_HighSpeed )
- && ( AckNumber == BlockNumber )) {
- WindowAck ( pTftpServer, pContext, pPacket );
- }
-
- //
- // Free this packet
- //
- PacketFree ( pContext, pPacket );
- } while (( NULL != pContext->pTxHead ) && ( AckNumber != BlockNumber ));
-
- //
- // Fill the window with packets
- //
- pPacket = pContext->pTxHead;
- while (( NULL != pPacket )
- && ( pContext->PacketsInWindow < pContext->WindowSize )
- && ( !bCloseContext )) {
- Status = PacketTx ( pContext, pPacket );
- bCloseContext = (BOOLEAN)( EFI_ERROR ( Status ));
- pPacket = pPacket->pNext;
- }
-
- //
- // Get more packets ready for transmission
- //
- PacketFill ( pContext );
-
- //
- // Close the context when the last packet is ACKed
- //
- if ( 0 == pContext->PacketsInWindow ) {
- bCloseContext = TRUE;
-
- //
- // Display the bandwidth
- //
- if ( PcdGetBool ( Tftp_Bandwidth )) {
- UINT64 Bandwidth;
- UINT64 DeltaTime;
- UINT64 NanoSeconds;
- UINT32 Value;
-
- //
- // Compute the download time
- //
- DeltaTime = GetPerformanceCounter ( );
- if ( pTftpServer->Time2 > pTftpServer->Time1 ) {
- DeltaTime = DeltaTime - pContext->TimeStart;
- }
- else {
- DeltaTime = pContext->TimeStart - DeltaTime;
- }
- NanoSeconds = GetTimeInNanoSecond ( DeltaTime );
- Bandwidth = pContext->LengthInBytes;
- DEBUG (( DEBUG_WINDOW,
- "File Length %Ld, Transfer Time: %d.%03d Sec\r\n",
- Bandwidth,
- DivU64x32 ( NanoSeconds, 1000 * 1000 * 1000 ),
- ((UINT32)DivU64x32 ( NanoSeconds, 1000 * 1000 )) % 1000 ));
-
- //
- // Display the round trip time
- //
- Bandwidth = MultU64x32 ( Bandwidth, 8 * 1000 * 1000 );
- Bandwidth /= NanoSeconds;
- if ( 1000 > Bandwidth ) {
- Value = (UINT32)Bandwidth;
- Print ( L"Bandwidth: %d Kbits/Sec\r\n",
- Value );
- }
- else if (( 1000 * 1000 ) > Bandwidth ) {
- Value = (UINT32)Bandwidth;
- Print ( L"Bandwidth: %d.%03d Mbits/Sec\r\n",
- Value / 1000,
- Value % 1000 );
- }
- else {
- Value = (UINT32)DivU64x32 ( Bandwidth, 1000 );
- Print ( L"Bandwidth: %d.%03d Gbits/Sec\r\n",
- Value / 1000,
- Value % 1000 );
- }
- }
- }
- break;
- }
-
- //
- // Return the operation status
- //
- DBG_EXIT ( );
- return bCloseContext;
-}
-
-
-/**
- Get the next TFTP option
-
- @param [in] pOption Address of a zero terminated option string
- @param [in] pEnd End of buffer address
- @param [in] ppNextOption Address to receive the address of the next
- zero terminated option string
-
- @retval EFI_SUCCESS Message processed successfully
-
-**/
-EFI_STATUS
-TftpOptionGet (
- IN UINT8 * pOption,
- IN UINT8 * pEnd,
- IN UINT8 ** ppNextOption
- )
-{
- UINT8 * pNextOption;
- EFI_STATUS Status;
-
- //
- // Locate the end of the option
- //
- pNextOption = pOption;
- while (( pEnd > pNextOption ) && ( 0 != *pNextOption )) {
- pNextOption += 1;
- }
- if ( pEnd <= pNextOption ) {
- //
- // Error - end of buffer reached
- //
- DEBUG (( DEBUG_ERROR | DEBUG_TFTP_REQUEST,
- "ERROR - Option without zero termination received!\r\n" ));
- Status = EFI_INVALID_PARAMETER;
- }
- else {
- //
- // Zero terminated option found
- //
- pNextOption += 1;
-
- //
- // Display the zero terminated ASCII option string
- //
- DEBUG (( DEBUG_TFTP_REQUEST,
- "Option: %a\r\n",
- pOption ));
- Status = EFI_SUCCESS;
- }
-
- //
- // Return the next option address
- //
- *ppNextOption = pNextOption;
-
- //
- // Return the operation status
- //
- return Status;
-}
-
-
-/**
- Place an option value into the option acknowledgement
-
- @param [in] pOack Option acknowledgement address
- @param [in] Value Value to translate into ASCII decimal
-
- @return Option acknowledgement address
-
-**/
-UINT8 *
-TftpOptionSet (
- IN UINT8 * pOack,
- IN UINT64 Value
- )
-{
- UINT64 NextValue;
-
- //
- // Determine the next value
- //
- NextValue = Value / 10;
-
- //
- // Supress leading zeros
- //
- if ( 0 != NextValue ) {
- pOack = TftpOptionSet ( pOack, NextValue );
- }
-
- //
- // Output this digit
- //
- *pOack++ = (UINT8)( Value - ( NextValue * 10 ) + '0' );
-
- //
- // Return the next option acknowledgement location
- //
- return pOack;
-}
-
-
-/**
- Process the TFTP request
-
- @param [in] pContext Address of a ::TSDT_CONNECTION_CONTEXT structure
- @param [in] pOption Address of the first zero terminated option string
- @param [in] pEnd End of buffer address
-
-**/
-VOID
-TftpOptions (
- IN TSDT_CONNECTION_CONTEXT * pContext,
- IN UINT8 * pOption,
- IN UINT8 * pEnd
- )
-{
- UINT8 * pNextOption;
- UINT8 * pOack;
- TFTP_PACKET * pPacket;
- UINT8 * pTemp;
- UINT8 * pValue;
- EFI_STATUS Status;
- INT32 Value;
-
- //
- // Get a packet
- //
- pPacket = PacketGet ( pContext );
-
- //
- // Start the OACK packet
- // Let the OACK handle the parsing errors
- // See http://tools.ietf.org/html/rfc2347
- //
- pOack = &pPacket->TxBuffer[ 0 ];
- *pOack++ = 0;
- *pOack++ = TFTP_OP_OACK;
- pPacket->TxBytes = 2;
- pPacket->BlockNumber = 0;
-
- //
- // Walk the list of options
- //
- do {
- //
- // Get the next option, skip junk at end of message
- //
- Status = TftpOptionGet ( pOption, pEnd, &pNextOption );
- if ( !EFI_ERROR ( Status )) {
- //
- // Process the option
- //
-
- //
- // blksize - See http://tools.ietf.org/html/rfc2348
- //
- pValue = pNextOption;
- if ( 0 == strcasecmp ((char *)pOption, "blksize" )) {
- //
- // Get the value
- //
- Status = TftpOptionGet ( pValue, pEnd, &pNextOption );
- if ( !EFI_ERROR ( Status )) {
- //
- // Validate the block size, skip non-numeric block sizes
- //
- Status = TftpOptionValue ( pValue, &Value );
- if ( !EFI_ERROR ( Status )) {
- //
- // Propose a smaller block size if necessary
- //
- if ( Value > TFTP_MAX_BLOCK_SIZE ) {
- Value = TFTP_MAX_BLOCK_SIZE;
- }
-
- //
- // Set the new block size
- //
- pContext->BlockSize = Value;
- DEBUG (( DEBUG_TFTP_REQUEST,
- "Using block size of %d bytes\r\n",
- pContext->BlockSize ));
-
- //
- // Update the OACK
- //
- pTemp = pOack;
- *pOack++ = 'b';
- *pOack++ = 'l';
- *pOack++ = 'k';
- *pOack++ = 's';
- *pOack++ = 'i';
- *pOack++ = 'z';
- *pOack++ = 'e';
- *pOack++ = 0;
- pOack = TftpOptionSet ( pOack, pContext->BlockSize );
- *pOack++ = 0;
- pPacket->TxBytes += pOack - pTemp;
- }
- }
- }
-
- //
- // timeout - See http://tools.ietf.org/html/rfc2349
- //
- else if ( 0 == strcasecmp ((char *)pOption, "timeout" )) {
- //
- // Get the value
- //
- Status = TftpOptionGet ( pValue, pEnd, &pNextOption );
- if ( !EFI_ERROR ( Status )) {
- Status = TftpOptionValue ( pValue, &Value );
- if ( !EFI_ERROR ( Status )) {
- //
- // Set the timeout value
- //
- pContext->MaxTimeout = Value;
- DEBUG (( DEBUG_TFTP_REQUEST,
- "Using timeout of %d seconds\r\n",
- pContext->MaxTimeout ));
-
- //
- // Update the OACK
- //
- pTemp = pOack;
- *pOack++ = 't';
- *pOack++ = 'i';
- *pOack++ = 'm';
- *pOack++ = 'e';
- *pOack++ = 'o';
- *pOack++ = 'u';
- *pOack++ = 't';
- *pOack++ = 0;
- pOack = TftpOptionSet ( pOack, pContext->MaxTimeout );
- *pOack++ = 0;
- pPacket->TxBytes += pOack - pTemp;
- }
- }
- }
-
- //
- // tsize - See http://tools.ietf.org/html/rfc2349
- //
- else if ( 0 == strcasecmp ((char *)pOption, "tsize" )) {
- //
- // Get the value
- //
- Status = TftpOptionGet ( pValue, pEnd, &pNextOption );
- if ( !EFI_ERROR ( Status )) {
- Status = TftpOptionValue ( pValue, &Value );
- if ( !EFI_ERROR ( Status )) {
- //
- // Return the file size
- //
- DEBUG (( DEBUG_TFTP_REQUEST,
- "Returning file size of %Ld bytes\r\n",
- pContext->LengthInBytes ));
-
- //
- // Update the OACK
- //
- pTemp = pOack;
- *pOack++ = 't';
- *pOack++ = 's';
- *pOack++ = 'i';
- *pOack++ = 'z';
- *pOack++ = 'e';
- *pOack++ = 0;
- pOack = TftpOptionSet ( pOack, pContext->LengthInBytes );
- *pOack++ = 0;
- pPacket->TxBytes += pOack - pTemp;
- }
- }
- }
- else {
- //
- // Unknown option - Ignore it
- //
- DEBUG (( DEBUG_WARN | DEBUG_TFTP_REQUEST,
- "WARNING - Skipping unknown option: %a\r\n",
- pOption ));
- }
- }
-
- //
- // Set the next option
- //
- pOption = pNextOption;
- } while ( pEnd > pOption );
-
- //
- // Transmit the OACK if necessary
- //
- if ( 2 < pPacket->TxBytes ) {
- PacketQueue ( pContext, pPacket );
- }
- else {
- PacketFree ( pContext, pPacket );
- }
-}
-
-
-/**
- Process the TFTP request
-
- @param [in] pOption Address of the first zero terminated option string
- @param [in] pValue Address to receive the value
-
- @retval EFI_SUCCESS Option translated into a value
-
-**/
-EFI_STATUS
-TftpOptionValue (
- IN UINT8 * pOption,
- IN INT32 * pValue
- )
-{
- UINT8 Digit;
- EFI_STATUS Status;
- INT32 Value;
-
- //
- // Assume success
- //
- Status = EFI_SUCCESS;
-
- //
- // Walk the characters in the option
- //
- Value = 0;
- while ( 0 != *pOption ) {
- //
- // Convert the next digit to binary
- //
- Digit = *pOption++;
- if (( '0' <= Digit ) && ( '9' >= Digit )) {
- Value *= 10;
- Value += Digit - '0';
- }
- else {
- DEBUG (( DEBUG_ERROR | DEBUG_TFTP_REQUEST,
- "ERROR - Invalid character '0x%02x' in the value\r\n",
- Digit ));
- Status = EFI_INVALID_PARAMETER;
- break;
- }
- }
-
- //
- // Return the value
- //
- *pValue = Value;
-
- //
- // Return the conversion status
- //
- return Status;
-}
-
-
-/**
- Process the TFTP request
-
- @param [in] pTftpServer Address of the ::TSDT_TFTP_SERVER structure
- @param [in] pContext Address of a ::TSDT_CONNECTION_CONTEXT structure
- @param [in] SocketFd Socket file descriptor
-
-**/
-VOID
-TftpProcessRequest (
- IN TSDT_TFTP_SERVER * pTftpServer,
- IN TSDT_CONNECTION_CONTEXT * pContext,
- IN int SocketFd
- )
-{
- BOOLEAN bCloseContext;
- UINT16 Opcode;
-
- DBG_ENTER ( );
-
- //
- // Get the opcode
- //
- Opcode = HTONS ( *(UINT16 *)&pTftpServer->RxBuffer[ 0 ]);
- DEBUG (( DEBUG_TFTP_REQUEST,
- "TFTP Opcode: 0x%08x\r\n",
- Opcode ));
-
- //
- // Validate the parameters
- //
- bCloseContext = FALSE;
- switch ( Opcode ) {
- default:
- DEBUG (( DEBUG_TFTP_REQUEST,
- "ERROR - Unknown TFTP opcode: %d\r\n",
- Opcode ));
- break;
-
- case TFTP_OP_ACK:
- bCloseContext = TftpAck ( pTftpServer, pContext );
- break;
-
- case TFTP_OP_READ_REQUEST:
- bCloseContext = TftpRead ( pTftpServer, pContext, SocketFd );
- break;
-
-
-
-
- case TFTP_OP_DATA:
- if ( NULL == pContext ) {
- if ( AF_INET == pTftpServer->RemoteAddress.v4.sin_family ) {
- DEBUG (( DEBUG_ERROR,
- "ERROR - File not open for %d.%d.%d.%d:%d\r\n",
- (UINT8)pTftpServer->RemoteAddress.v4.sin_addr.s_addr,
- (UINT8)( pTftpServer->RemoteAddress.v4.sin_addr.s_addr >> 8 ),
- (UINT8)( pTftpServer->RemoteAddress.v4.sin_addr.s_addr >> 16 ),
- (UINT8)( pTftpServer->RemoteAddress.v4.sin_addr.s_addr >> 24 ),
- htons ( pTftpServer->RemoteAddress.v4.sin_port )));
- }
- else {
- DEBUG (( DEBUG_ERROR,
- "ERROR - File not open for [%02x%02x:%02x%02x:%02x%02x:%02x%02x:%02x%02x:%02x%02x:%02x%02x:%02x%02x]:%d\r\n",
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 0 ],
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 1 ],
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 2 ],
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 3 ],
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 4 ],
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 5 ],
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 6 ],
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 7 ],
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 8 ],
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 9 ],
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 10 ],
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 11 ],
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 12 ],
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 13 ],
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 14 ],
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 15 ],
- htons ( pTftpServer->RemoteAddress.v6.sin6_port )));
- }
- break;
- }
- if ( 0 != pContext->PacketsInWindow ) {
- DEBUG (( DEBUG_ERROR,
- "ERROR - Expecting ACKs not data for pContext 0x%08x\r\n",
- pContext ));
- break;
- }
- if ( pTftpServer->RxBytes > (ssize_t)( pContext->BlockSize + 2 + 2 )) {
- DEBUG (( DEBUG_ERROR,
- "ERROR - Receive data length of %d > %d bytes (maximum block size) for pContext 0x%08x\r\n",
- pTftpServer->RxBytes - 2 - 2,
- pContext->BlockSize,
- pContext ));
- break;
- }
- break;
-
- case TFTP_OP_ERROR:
- if ( NULL == pContext ) {
- if ( AF_INET == pTftpServer->RemoteAddress.v4.sin_family ) {
- DEBUG (( DEBUG_ERROR,
- "ERROR - File not open for %d.%d.%d.%d:%d\r\n",
- (UINT8)pTftpServer->RemoteAddress.v4.sin_addr.s_addr,
- (UINT8)( pTftpServer->RemoteAddress.v4.sin_addr.s_addr >> 8 ),
- (UINT8)( pTftpServer->RemoteAddress.v4.sin_addr.s_addr >> 16 ),
- (UINT8)( pTftpServer->RemoteAddress.v4.sin_addr.s_addr >> 24 ),
- htons ( pTftpServer->RemoteAddress.v4.sin_port )));
- }
- else {
- DEBUG (( DEBUG_ERROR,
- "ERROR - File not open for [%02x%02x:%02x%02x:%02x%02x:%02x%02x:%02x%02x:%02x%02x:%02x%02x:%02x%02x]:%d\r\n",
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 0 ],
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 1 ],
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 2 ],
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 3 ],
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 4 ],
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 5 ],
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 6 ],
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 7 ],
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 8 ],
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 9 ],
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 10 ],
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 11 ],
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 12 ],
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 13 ],
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 14 ],
- pTftpServer->RemoteAddress.v6.sin6_addr.__u6_addr.__u6_addr8[ 15 ],
- htons ( pTftpServer->RemoteAddress.v6.sin6_port )));
- }
- }
- break;
- }
-
- //
- // Determine if the context should be closed
- //
- if ( bCloseContext ) {
- ContextRemove ( pTftpServer, pContext );
- }
-
- DBG_EXIT ( );
-}
-
-
-/**
- Process the read request
-
- @param [in] pTftpServer Address of the ::TSDT_TFTP_SERVER structure
- @param [in] pContext Address of a ::TSDT_CONNECTION_CONTEXT structure
- @param [in] SocketFd Socket file descriptor
-
- @retval TRUE if the context should be closed
-
-**/
-BOOLEAN
-TftpRead (
- IN TSDT_TFTP_SERVER * pTftpServer,
- IN TSDT_CONNECTION_CONTEXT * pContext,
- IN int SocketFd
- )
-{
- BOOLEAN bCloseContext;
- struct stat FileStatus;
- UINT8 * pBuffer;
- UINT8 * pEnd;
- UINT8 * pFileName;
- UINT8 * pMode;
- UINT8 * pOption;
- CHAR8 * pReadMode;
- UINT64 TimeStart;
-
- DBG_ENTER ( );
-
- //
- // Log the receive time
- //
- TimeStart = 0;
- if ( PcdGetBool ( Tftp_Bandwidth )) {
- TimeStart = GetPerformanceCounter ( );
- }
-
- //
- // Close the context if necessary
- //
- bCloseContext = FALSE;
- if ( NULL != pContext ) {
- ContextRemove ( pTftpServer, pContext );
- }
-
- //
- // Use break instead of goto
- //
- for ( ; ; ) {
- //
- // Create the connection context
- //
- pContext = ContextAdd ( pTftpServer, SocketFd );
- if ( NULL == pContext ) {
- break;
- }
-
- //
- // Set the start time
- //
- if ( PcdGetBool ( Tftp_Bandwidth )) {
- pContext->TimeStart = TimeStart;
- }
-
- //
- // Locate the mode
- //
- pBuffer = &pTftpServer->RxBuffer[ 0 ];
- pEnd = &pBuffer[ pTftpServer->RxBytes ];
- pFileName = &pBuffer[ 2 ];
- pMode = pFileName;
- while (( pEnd > pMode ) && ( 0 != *pMode )) {
- pMode += 1;
- }
- if ( pEnd <= pMode ) {
- //
- // Mode not found
- //
- DEBUG (( DEBUG_ERROR | DEBUG_RX,
- "ERROR - File mode not found\r\n" ));
- //
- // Tell the client of the error
- //
- SendError ( pContext,
- TFTP_ERROR_SEE_MSG,
- (UINT8 *)"File open mode not found" );
- break;
- }
- pMode += 1;
- DEBUG (( DEBUG_TFTP_REQUEST,
- "TFTP - FileName: %a\r\n",
- pFileName ));
-
- //
- // Locate the options
- //
- pOption = pMode;
- while (( pEnd > pOption ) && ( 0 != *pOption )) {
- pOption += 1;
- }
- if ( pEnd <= pOption ) {
- //
- // End of mode not found
- //
- DEBUG (( DEBUG_ERROR | DEBUG_RX,
- "ERROR - File mode not valid\r\n" ));
- //
- // Tell the client of the error
- //
- SendError ( pContext,
- TFTP_ERROR_SEE_MSG,
- (UINT8 *)"File open mode not valid" );
- break;
- }
- pOption += 1;
- DEBUG (( DEBUG_TFTP_REQUEST,
- "TFTP - Mode: %a\r\n",
- pMode ));
-
- //
- // Verify the mode is supported
- //
- pReadMode = "r";
- if ( 0 == strcasecmp ((char *)pMode, "octet" )) {
- //
- // Read the file as binary input
- //
- pReadMode = "rb";
- }
-
- //
- // Determine the file length
- //
- pContext->File = fopen ((const char *)pFileName, pReadMode );
- if (( NULL == pContext->File )
- || ( -1 == stat ((const char *)pFileName, &FileStatus ))) {
- //
- // File not found
- //
- DEBUG (( DEBUG_ERROR | DEBUG_TFTP_REQUEST,
- ( NULL == pContext->File )
- ? "ERROR - File not found!\r\n"
- : "ERROR - Unable to determine file %a size!\r\n",
- pFileName ));
-
- //
- // Tell the client of the error
- //
- SendError ( pContext,
- TFTP_ERROR_NOT_FOUND,
- (UINT8 *)"File not found" );
- break;
- }
- pContext->LengthInBytes = FileStatus.st_size;
- pContext->BytesRemaining = pContext->LengthInBytes;
- pContext->BytesToSend = pContext->LengthInBytes;
-
- //
- // Display the file size
- //
- DEBUG_CODE_BEGIN ( );
- UINT32 Value;
-
- if ( 1024 > pContext->LengthInBytes ) {
- Value = (UINT32)pContext->LengthInBytes;
- DEBUG (( DEBUG_FILE_BUFFER,
- "%a size: %d Bytes\r\n",
- pFileName,
- Value ));
- }
- else if (( 1024 * 1024 ) > pContext->LengthInBytes ) {
- Value = (UINT32)pContext->LengthInBytes;
- DEBUG (( DEBUG_FILE_BUFFER,
- "%a size: %d.%03d KiBytes (%Ld Bytes)\r\n",
- pFileName,
- Value / 1024,
- (( Value % 1024 ) * 1000 ) / 1024,
- pContext->LengthInBytes ));
- }
- else if (( 1024 * 1024 * 1024 ) > pContext->LengthInBytes ) {
- Value = (UINT32)DivU64x32 ( pContext->LengthInBytes, 1024 );
- DEBUG (( DEBUG_FILE_BUFFER,
- "%a size: %d.%03d MiBytes (%Ld Bytes)\r\n",
- pFileName,
- Value / 1024,
- (( Value % 1024 ) * 1000 ) / 1024,
- pContext->LengthInBytes ));
- }
- else {
- Value = (UINT32)DivU64x32 ( pContext->LengthInBytes, 1024 * 1024 );
- DEBUG (( DEBUG_FILE_BUFFER,
- "%a size: %d.%03d GiBytes (%Ld Bytes)\r\n",
- pFileName,
- Value / 1024,
- (( Value % 1024 ) * 1000 ) / 1024,
- pContext->LengthInBytes ));
- }
- DEBUG_CODE_END ( );
-
- //
- // Process the options
- //
- if ( pEnd > pOption ) {
- TftpOptions ( pContext, pOption, pEnd );
- }
- else {
- //
- // Skip the open ACK
- //
- pContext->BlockNumber = 1;
- }
-
- //
- // Send the first packet (OACK or data block)
- //
- bCloseContext = PacketFill ( pContext );
- break;
- }
-
- //
- // Return the close status
- //
- DBG_EXIT ( );
- return bCloseContext;
-}
-
-
-/**
- Create the port for the TFTP server
-
- This routine polls the network layer to create the TFTP port for the
- TFTP server. More than one attempt may be necessary since it may take
- some time to get the IP address and initialize the upper layers of
- the network stack.
-
- @param [in] pTftpServer Address of the ::TSDT_TFTP_SERVER structure
- @param [in] AddressFamily The address family to use for the conection.
- @param [in] pIndex Address of the index into the port array
-
-**/
-VOID
-TftpServerSocket (
- IN TSDT_TFTP_SERVER * pTftpServer,
- IN sa_family_t AddressFamily,
- IN int * pIndex
- )
-{
- int SocketStatus;
- struct pollfd * pTftpPort;
- UINT16 TftpPort;
- union {
- struct sockaddr_in v4;
- struct sockaddr_in6 v6;
- } TftpServerAddress;
-
- DEBUG (( DEBUG_SERVER_TIMER, "Entering TftpServerListen\r\n" ));
-
- //
- // Determine if the socket is already initialized
- //
- if ( -1 == *pIndex ) {
- //
- // Attempt to create the socket for the TFTP server
- //
- pTftpPort = &pTftpServer->TftpPort[ pTftpServer->Entries ];
- pTftpPort->fd = socket ( AddressFamily,
- SOCK_DGRAM,
- IPPROTO_UDP );
- if ( -1 != pTftpPort->fd ) {
- //
- // Initialize the poll structure
- //
- pTftpPort->events = POLLRDNORM | POLLHUP;
- pTftpPort->revents = 0;
-
- //
- // Set the socket address
- //
- TftpPort = 69;
- ZeroMem ( &TftpServerAddress, sizeof ( TftpServerAddress ));
- TftpServerAddress.v4.sin_port = htons ( TftpPort );
- if ( AF_INET == AddressFamily ) {
- TftpServerAddress.v4.sin_len = sizeof ( TftpServerAddress.v4 );
- TftpServerAddress.v4.sin_family = AF_INET;
- }
- else {
- TftpServerAddress.v6.sin6_len = sizeof ( TftpServerAddress.v6 );
- TftpServerAddress.v6.sin6_family = AF_INET6;
- }
-
- //
- // Bind the socket to the TFTP port
- //
- SocketStatus = bind ( pTftpPort->fd,
- (struct sockaddr *) &TftpServerAddress,
- TftpServerAddress.v6.sin6_len );
- if ( -1 != SocketStatus ) {
- DEBUG (( DEBUG_TFTP_PORT,
- "0x%08x: Socket bound to port %d\r\n",
- pTftpPort->fd,
- TftpPort ));
-
- //
- // Account for this connection
- //
- *pIndex = pTftpServer->Entries;
- pTftpServer->Entries += 1;
- ASSERT ( DIM ( pTftpServer->TftpPort ) >= pTftpServer->Entries );
- }
-
- //
- // Release the socket if necessary
- //
- if ( -1 == SocketStatus ) {
- close ( pTftpPort->fd );
- pTftpPort->fd = -1;
- }
- }
- }
-
- DEBUG (( DEBUG_SERVER_TIMER, "Exiting TftpServerListen\r\n" ));
-}
-
-
-/**
- Update the window due to the ACK
-
- @param [in] pTftpServer Address of the ::TSDT_TFTP_SERVER structure
- @param [in] pContext Address of a ::TSDT_CONNECTION_CONTEXT structure
- @param [in] pPacket Address of a ::TFTP_PACKET structure
-
-**/
-VOID
-WindowAck (
- IN TSDT_TFTP_SERVER * pTftpServer,
- IN TSDT_CONNECTION_CONTEXT * pContext,
- IN TFTP_PACKET * pPacket
- )
-{
- if ( PcdGetBool ( Tftp_HighSpeed )) {
- UINT64 DeltaTime;
- UINT64 NanoSeconds;
-
- DBG_ENTER ( );
-
- //
- // Compute the round trip time
- //
- if ( pTftpServer->Time2 > pTftpServer->Time1 ) {
- DeltaTime = pTftpServer->RxTime - pPacket->TxTime;
- }
- else {
- DeltaTime = pPacket->TxTime - pTftpServer->RxTime;
- }
-
- //
- // Adjust the round trip time
- //
- NanoSeconds = GetTimeInNanoSecond ( DeltaTime );
- DeltaTime = RShiftU64 ( pContext->Rtt2x, ACK_SHIFT );
- pContext->Rtt2x += NanoSeconds + NanoSeconds - DeltaTime;
- if ( pContext->Rtt2x > pContext->MaxTimeout ) {
- pContext->Rtt2x = pContext->MaxTimeout;
- }
-
- //
- // Account for the ACK
- //
- if ( pContext->WindowSize < MAX_PACKETS ) {
- pContext->AckCount -= 1;
- if ( 0 == pContext->AckCount ) {
- //
- // Increase the window
- //
- pContext->WindowSize += 1;
-
- //
- // Set the ACK count
- //
- if ( pContext->WindowSize < pContext->Threshold ) {
- pContext->AckCount = pContext->WindowSize * PcdGet32 ( Tftp_AckMultiplier );
- }
- else {
- pContext->AckCount = PcdGet32 ( Tftp_AckLogBase ) << pContext->WindowSize;
- }
-
- //
- // Display the round trip time
- //
- DEBUG_CODE_BEGIN ( );
- UINT32 Value;
-
- DeltaTime = RShiftU64 ( pContext->Rtt2x, 1 );
- if ( 1000 > DeltaTime ) {
- DEBUG (( DEBUG_WINDOW,
- "WindowSize: %d, Threshold: %d, AckCount: %4d, RTT: %Ld nSec\r\n",
- pContext->WindowSize,
- pContext->Threshold,
- pContext->AckCount,
- DeltaTime ));
- }
- else if (( 1000 * 1000 ) > DeltaTime ) {
- Value = (UINT32)DeltaTime;
- DEBUG (( DEBUG_WINDOW,
- "WindowSize: %d, Threshold: %d, AckCount: %4d, RTT: %d.%03d uSec\r\n",
- pContext->WindowSize,
- pContext->Threshold,
- pContext->AckCount,
- Value / 1000,
- Value % 1000 ));
- }
- else if (( 1000 * 1000 * 1000 ) > DeltaTime ) {
- Value = (UINT32)DivU64x32 ( DeltaTime, 1000 );
- DEBUG (( DEBUG_WINDOW,
- "WindowSize: %d, Threshold: %d, AckCount: %4d, RTT: %d.%03d mSec\r\n",
- pContext->WindowSize,
- pContext->Threshold,
- pContext->AckCount,
- Value / 1000,
- Value % 1000 ));
- }
- else {
- Value = (UINT32)DivU64x32 ( DeltaTime, 1000 * 1000 );
- DEBUG (( DEBUG_WINDOW,
- "WindowSize: %d, Threshold: %d, AckCount: %4d, RTT: %d.%03d Sec\r\n",
- pContext->WindowSize,
- pContext->Threshold,
- pContext->AckCount,
- Value / 1000,
- Value % 1000 ));
- }
- DEBUG_CODE_END ( );
- }
- }
-
- DBG_EXIT ( );
- }
-}
-
-
-/**
- A timeout has occurred, close the window
-
- @param [in] pContext Address of a ::TSDT_CONNECTION_CONTEXT structure
-
-**/
-VOID
-WindowTimeout (
- IN TSDT_CONNECTION_CONTEXT * pContext
- )
-{
- if ( PcdGetBool ( Tftp_HighSpeed )) {
- TFTP_PACKET * pPacket;
-
- DBG_ENTER ( );
-
- //
- // Set the threshold at half the previous window size
- //
- pContext->Threshold = ( pContext->WindowSize + 1 ) >> 1;
-
- //
- // Close the transmit window
- //
- pContext->WindowSize = 1;
- pContext->PacketsInWindow = 0;
-
- //
- // Double the round trip time
- //
- pContext->Rtt2x = LShiftU64 ( pContext->Rtt2x, 1 );
- if ( pContext->Rtt2x > pContext->MaxTimeout ) {
- pContext->Rtt2x = pContext->MaxTimeout;
- }
-
- //
- // Set the ACK count
- //
- if ( pContext->WindowSize < pContext->Threshold ) {
- pContext->AckCount = pContext->WindowSize * PcdGet32 ( Tftp_AckMultiplier );
- }
- else {
- pContext->AckCount = PcdGet32 ( Tftp_AckLogBase ) << pContext->WindowSize;
- }
-
- //
- // Display the round trip time
- //
- DEBUG_CODE_BEGIN ( );
- UINT64 DeltaTime;
- UINT32 Value;
-
- DeltaTime = RShiftU64 ( pContext->Rtt2x, 1 );
- if ( 1000 > DeltaTime ) {
- DEBUG (( DEBUG_WINDOW,
- "WindowSize: %d, Threshold: %d, AckCount: %4d, RTT: %Ld nSec\r\n",
- pContext->WindowSize,
- pContext->Threshold,
- pContext->AckCount,
- DeltaTime ));
- }
- else if (( 1000 * 1000 ) > DeltaTime ) {
- Value = (UINT32)DeltaTime;
- DEBUG (( DEBUG_WINDOW,
- "WindowSize: %d, Threshold: %d, AckCount: %4d, RTT: %d.%03d uSec\r\n",
- pContext->WindowSize,
- pContext->Threshold,
- pContext->AckCount,
- Value / 1000,
- Value % 1000 ));
- }
- else if (( 1000 * 1000 * 1000 ) > DeltaTime ) {
- Value = (UINT32)DivU64x32 ( DeltaTime, 1000 );
- DEBUG (( DEBUG_WINDOW,
- "WindowSize: %d, Threshold: %d, AckCount: %4d, RTT: %d.%03d mSec\r\n",
- pContext->WindowSize,
- pContext->Threshold,
- pContext->AckCount,
- Value / 1000,
- Value % 1000 ));
- }
- else {
- Value = (UINT32)DivU64x32 ( DeltaTime, 1000 * 1000 );
- DEBUG (( DEBUG_WINDOW,
- "WindowSize: %d, Threshold: %d, AckCount: %4d, RTT: %d.%03d Sec\r\n",
- pContext->WindowSize,
- pContext->Threshold,
- pContext->AckCount,
- Value / 1000,
- Value % 1000 ));
- }
- DEBUG_CODE_END ( );
-
- //
- // Retransmit the first packet in the window
- //
- pPacket = pContext->pTxHead;
- if ( NULL != pPacket ) {
- PacketTx ( pContext, pPacket );
- }
-
- DBG_EXIT ( );
- }
-}
-
-
-/**
- Entry point for the TFTP server application.
-
- @param [in] Argc The number of arguments
- @param [in] Argv The argument value array
-
- @retval 0 The application exited normally.
- @retval Other An error occurred.
-**/
-int
-main (
- IN int Argc,
- IN char **Argv
- )
-{
- UINTN Index;
- TSDT_TFTP_SERVER * pTftpServer;
- EFI_STATUS Status;
- UINT64 TriggerTime;
-
- //
- // Get the performance counter characteristics
- //
- pTftpServer = &mTftpServer;
- if ( PcdGetBool ( Tftp_HighSpeed )
- || PcdGetBool ( Tftp_Bandwidth )) {
- pTftpServer->ClockFrequency = GetPerformanceCounterProperties ( &pTftpServer->Time1,
- &pTftpServer->Time2 );
- }
-
- //
- // Create a timer event to start TFTP port
- //
- Status = gBS->CreateEvent ( EVT_TIMER,
- TPL_TFTP_SERVER,
- NULL,
- NULL,
- &pTftpServer->TimerEvent );
- if ( !EFI_ERROR ( Status )) {
- //
- // Compute the poll interval
- //
- TriggerTime = TFTP_PORT_POLL_DELAY * ( 1000 * 10 );
- Status = gBS->SetTimer ( pTftpServer->TimerEvent,
- TimerPeriodic,
- TriggerTime );
- if ( !EFI_ERROR ( Status )) {
- DEBUG (( DEBUG_TFTP_PORT, "TFTP port timer started\r\n" ));
-
- //
- // Run the TFTP server forever
- //
- pTftpServer->Udpv4Index = -1;
- pTftpServer->Udpv6Index = -1;
- do {
- //
- // Poll the network layer to create the TFTP port
- // for the tftp server. More than one attempt may
- // be necessary since it may take some time to get
- // the IP address and initialize the upper layers
- // of the network stack.
- //
- if ( DIM ( pTftpServer->TftpPort ) != pTftpServer->Entries ) {
- do {
- //
- // Wait a while before polling for a connection
- //
- if ( EFI_SUCCESS != gBS->CheckEvent ( pTftpServer->TimerEvent )) {
- if ( 0 == pTftpServer->Entries ) {
- break;
- }
- gBS->WaitForEvent ( 1, &pTftpServer->TimerEvent, &Index );
- }
-
- //
- // Poll for a network connection
- //
- TftpServerSocket ( pTftpServer,
- AF_INET,
- &pTftpServer->Udpv4Index );
- TftpServerSocket ( pTftpServer,
- AF_INET6,
- &pTftpServer->Udpv6Index );
- } while ( 0 == pTftpServer->Entries );
- }
-
- //
- // Poll the socket for activity
- //
- do {
- SocketPoll ( pTftpServer );
-
- //
- // Normal TFTP lets the client request the retransmit by
- // sending another ACK for the previous packet
- //
- if ( PcdGetBool ( Tftp_HighSpeed )) {
- UINT64 CurrentTime;
- UINT64 ElapsedTime;
- TSDT_CONNECTION_CONTEXT * pContext;
- TFTP_PACKET * pPacket;
-
- //
- // High speed TFTP uses an agressive retransmit to
- // get the TFTP client moving again when the ACK or
- // previous data packet was lost.
- //
- // Get the current time
- //
- CurrentTime = GetPerformanceCounter ( );
-
- //
- // Walk the list of contexts
- //
- pContext = pTftpServer->pContextList;
- while ( NULL != pContext )
- {
- //
- // Check for a transmit timeout
- //
- pPacket = pContext->pTxHead;
- if ( NULL != pPacket ) {
- //
- // Compute the elapsed time
- //
- if ( pTftpServer->Time2 > pTftpServer->Time1 ) {
- ElapsedTime = CurrentTime - pPacket->TxTime;
- }
- else {
- ElapsedTime = pPacket->TxTime - CurrentTime;
- }
- ElapsedTime = GetTimeInNanoSecond ( ElapsedTime );
-
- //
- // Determine if a retransmission is necessary
- //
- if ( ElapsedTime >= pContext->Rtt2x ) {
- DEBUG (( DEBUG_WINDOW,
- "0x%08x: Context TX timeout for packet 0x%08x, Window: %d\r\n",
- pContext,
- pPacket,
- pContext->WindowSize ));
- WindowTimeout ( pContext );
- }
- }
-
- //
- // Set the next context
- //
- pContext = pContext->pNext;
- }
- }
- } while ( DIM ( pTftpServer->TftpPort ) == pTftpServer->Entries );
- } while ( !mbTftpServerExit );
-
- //
- // Done with the timer event
- //
- gBS->SetTimer ( pTftpServer->TimerEvent,
- TimerCancel,
- 0 );
- }
- gBS->CloseEvent ( pTftpServer->TimerEvent );
- }
-
- //
- // Return the final status
- //
- DBG_EXIT_STATUS ( Status );
- return Status;
-}
diff --git a/AppPkg/Applications/Sockets/TftpServer/TftpServer.h b/AppPkg/Applications/Sockets/TftpServer/TftpServer.h deleted file mode 100644 index 2166893570..0000000000 --- a/AppPkg/Applications/Sockets/TftpServer/TftpServer.h +++ /dev/null @@ -1,431 +0,0 @@ -/** @file
- Definitions for the TFTP server.
-
- Copyright (c) 2011, 2012, Intel Corporation
- All rights reserved. This program and the accompanying materials
- are licensed and made available under the terms and conditions of the BSD License
- which accompanies this distribution. The full text of the license may be found at
- http://opensource.org/licenses/bsd-license.php
-
- THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,
- WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
-
-**/
-
-#ifndef _TFTP_SERVER_H_
-#define _TFTP_SERVER_H_
-
-#include <errno.h>
-#include <stdio.h>
-#include <stdlib.h>
-#include <string.h>
-#include <Uefi.h>
-
-#include <Guid/EventGroup.h>
-
-#include <Library/BaseMemoryLib.h>
-#include <Library/DebugLib.h>
-#include <Library/MemoryAllocationLib.h>
-#include <Library/PcdLib.h>
-#include <Library/TimerLib.h>
-#include <Library/UefiApplicationEntryPoint.h>
-#include <Library/UefiBootServicesTableLib.h>
-#include <Library/UefiLib.h>
-#include <Protocol/BlockIo.h>
-
-#include <netinet/in.h>
-#include <netinet6/in6.h>
-
-#include <sys/EfiSysCall.h>
-#include <sys/poll.h>
-#include <sys/socket.h>
-#include <sys/stat.h>
-
-//------------------------------------------------------------------------------
-// Macros
-//------------------------------------------------------------------------------
-
-#if defined(_MSC_VER) /* Handle Microsoft VC++ compiler specifics. */
-#define DBG_ENTER() DEBUG (( DEBUG_ENTER_EXIT, "Entering " __FUNCTION__ "\n" )) ///< Display routine entry
-#define DBG_EXIT() DEBUG (( DEBUG_ENTER_EXIT, "Exiting " __FUNCTION__ "\n" )) ///< Display routine exit
-#define DBG_EXIT_DEC(Status) DEBUG (( DEBUG_ENTER_EXIT, "Exiting " __FUNCTION__ ", Status: %d\n", Status )) ///< Display routine exit with decimal value
-#define DBG_EXIT_HEX(Status) DEBUG (( DEBUG_ENTER_EXIT, "Exiting " __FUNCTION__ ", Status: 0x%08x\n", Status )) ///< Display routine exit with hex value
-#define DBG_EXIT_STATUS(Status) DEBUG (( DEBUG_ENTER_EXIT, "Exiting " __FUNCTION__ ", Status: %r\n", Status )) ///< Display routine exit with status value
-#define DBG_EXIT_TF(Status) DEBUG (( DEBUG_ENTER_EXIT, "Exiting " __FUNCTION__ ", returning %s\n", (FALSE == Status) ? L"FALSE" : L"TRUE" )) ///< Display routine with TRUE/FALSE value
-#else // _MSC_VER
-#define DBG_ENTER()
-#define DBG_EXIT()
-#define DBG_EXIT_DEC(Status)
-#define DBG_EXIT_HEX(Status)
-#define DBG_EXIT_STATUS(Status)
-#define DBG_EXIT_TF(Status)
-#endif // _MSC_VER
-
-#define DIM(x) ( sizeof ( x ) / sizeof ( x[0] )) ///< Compute the number of entries in an array
-
-//------------------------------------------------------------------------------
-// Constants
-//------------------------------------------------------------------------------
-
-#define ACK_SHIFT 4 ///< Number of samples in ACK average
-
-#define DEBUG_WINDOW 0x00000001 ///< Display the window messages
-#define DEBUG_TX_PACKET 0x00000002 ///< Display the transmit packet messages
-#define DEBUG_FILE_BUFFER 0x00000004 ///< Display the file buffer messages
-#define DEBUG_SERVER_TIMER 0x00000008 ///< Display the socket poll messages
-#define DEBUG_TFTP_REQUEST 0x00000010 ///< Display the TFTP request messages
-#define DEBUG_PORT_WORK 0x00000020 ///< Display the port work messages
-#define DEBUG_SOCKET_POLL 0x00000040 ///< Display the socket poll messages
-#define DEBUG_TFTP_PORT 0x00000080 ///< Display the TFTP port messages
-#define DEBUG_TX 0x00000100 ///< Display transmit messages
-#define DEBUG_RX 0x00000200 ///< Display receive messages
-#define DEBUG_TFTP_ACK 0x00000400 ///< Display the TFTP ACK messages
-#define DEBUG_ENTER_EXIT 0x00000800 ///< Display entry and exit messages
-
-#define MAX_PACKETS 8 ///< Maximum number of packets in the window
-
-#define TFTP_PORT_POLL_DELAY ( 2 * 1000 ) ///< Delay in milliseconds for attempts to open the TFTP port
-#define CLIENT_POLL_DELAY 50 ///< Delay in milliseconds between client polls
-
-#define TPL_TFTP_SERVER TPL_CALLBACK ///< TPL for routine synchronization
-
-/**
- Verify new TPL value
-
- This macro which is enabled when debug is enabled verifies that
- the new TPL value is >= the current TPL value.
-**/
-#ifdef VERIFY_TPL
-#undef VERIFY_TPL
-#endif // VERIFY_TPL
-
-#if !defined(MDEPKG_NDEBUG)
-
-#define VERIFY_TPL(tpl) \
-{ \
- EFI_TPL PreviousTpl; \
- \
- PreviousTpl = gBS->RaiseTPL ( TPL_HIGH_LEVEL ); \
- gBS->RestoreTPL ( PreviousTpl ); \
- if ( PreviousTpl > tpl ) { \
- DEBUG (( DEBUG_ERROR, "Current TPL: %d, New TPL: %d\r\n", PreviousTpl, tpl )); \
- ASSERT ( PreviousTpl <= tpl ); \
- } \
-}
-
-#else // MDEPKG_NDEBUG
-
-#define VERIFY_TPL(tpl)
-
-#endif // MDEPKG_NDEBUG
-
-#define TFTP_SERVER_SIGNATURE SIGNATURE_32('T','F','T','P') ///< TSDT_TFTP_SERVER memory signature
-
-//
-// See: http://www.rfc-editor.org/rfc/pdfrfc/rfc1350.txt.pdf
-//
-// TFTP Operations
-//
-
-#define TFTP_OP_READ_REQUEST 1 ///< Read request, zero terminated file name, zero terminated mode
-#define TFTP_OP_WRITE_REQUEST 2 ///< Write request, zero terminated file name, zero terminated mode
-#define TFTP_OP_DATA 3 ///< Data block, end-of-file indicated by short block
-#define TFTP_OP_ACK 4 ///< ACK block number
-#define TFTP_OP_ERROR 5 ///< Error number and explaination
-#define TFTP_OP_OACK 6 ///< ACK the options
-
-#define TFTP_MAX_BLOCK_SIZE 4096 ///< Maximum block size
-
-#define TFTP_ERROR_SEE_MSG 0 ///< See the error message
-#define TFTP_ERROR_NOT_FOUND 1 ///< File not found
-#define TFTP_ERROR_ACCESS_VIOLATION 2 ///< Access violation
-#define TFTP_ERROR_DISK_FULL 3 ///< Disk full
-#define TFTP_ERROR_ILLEGAL_OP 4 ///< Illegal operation
-#define TFTP_ERROR_UNKNOWN_XFER_ID 5 ///< Unknown transfer ID
-#define TFTP_ERROR_FILE_EXISTS 6 ///< File already exists
-#define TFTP_ERROR_NO_SUCH_USER 7 ///< No such user
-
-//------------------------------------------------------------------------------
-// Data Types
-//------------------------------------------------------------------------------
-
-/**
- Packet structure
-**/
-typedef struct _TFTP_PACKET TFTP_PACKET;
-typedef struct _TFTP_PACKET {
- TFTP_PACKET * pNext; ///< Next packet in list
- UINT64 TxTime; ///< Time the transmit was performed
- ssize_t TxBytes; ///< Bytes in the TX buffer
- UINT32 RetryCount; ///< Number of transmissions
- UINT16 BlockNumber; ///< Block number of this packet
- UINT8 TxBuffer[ 2 + 2 + TFTP_MAX_BLOCK_SIZE ]; ///< Transmit buffer
-} GCC_TFTP_PACKET;
-
-/**
- Port control structure
-**/
-typedef struct _TSDT_CONNECTION_CONTEXT TSDT_CONNECTION_CONTEXT;
-typedef struct _TSDT_CONNECTION_CONTEXT {
- //
- // Remote connection management
- //
- TSDT_CONNECTION_CONTEXT * pNext; ///< Next context in the connection list
- struct sockaddr_in6 RemoteAddress; ///< Remote address
- int SocketFd; ///< Socket file descriptor
-
- //
- // File management parameters
- //
- FILE * File; ///< NULL while file is closed
- UINT64 LengthInBytes; ///< Size of the file
- UINT64 BytesRemaining; ///< Number of bytes remaining to be sent
- UINT64 BytesToSend; ///< Number of bytes to send
- UINT64 ValidBytes; ///< Number of valid bytes in the buffer
- BOOLEAN bEofSent; ///< End of file sent
- UINT8 * pFill; ///< Next portion of the buffer to fill
- UINT8 * pBuffer; ///< Pointer into the file data
- UINT8 * pEnd; ///< End of the file data
- UINT8 FileData[ 2 * MAX_PACKETS * TFTP_MAX_BLOCK_SIZE ]; ///< File data to send
- UINT64 TimeStart; ///< Start of file transfer
-
- //
- // TFTP management parameters
- //
- UINT16 BlockNumber; ///< Next block to be transmitted
- UINT32 BlockSize; ///< Negotiated block size
-
- //
- // Window management
- //
- UINT32 AckCount; ///< Number of ACKs to receive before increasing the window
- UINT32 PacketsInWindow; ///< Number of packets in the window
- UINT32 Threshold; ///< Size of window when ACK count becomes logrithmic
- UINT32 WindowSize; ///< Size of the transmit window
- UINT64 MaxTimeout; ///< Maximum number of seconds to wait before retransmission
- UINT64 Rtt2x; ///< Twice the average round trip time in nanoseconds
-
- //
- // Buffer management
- //
- TFTP_PACKET * pFreeList; ///< List of free packets
- TFTP_PACKET * pTxHead; ///< First packet in the list of packets for transmission
- TFTP_PACKET * pTxTail; ///< Last packet in the list of packets for transmission
- TFTP_PACKET ErrorPacket; ///< Error packet
- TFTP_PACKET Tx[ MAX_PACKETS ];///< Transmit packets
-}GCC_TSDT_CONNECTION_CONTEXT;
-
-/**
- TFTP server control structure
-**/
-typedef struct {
- UINTN Signature; ///< Structure identification
-
- //
- // Image attributes
- //
- EFI_HANDLE ImageHandle; ///< Image handle
-
- //
- // Performance management
- //
- UINT64 ClockFrequency; ///< Frequency of the clock
- UINT64 Time1; ///< Clock value after rollover
- UINT64 Time2; ///< Clock value before rollover
- UINT64 RxTime; ///< Time when the packet was recevied
-
- //
- // TFTP port management
- //
- EFI_EVENT TimerEvent; ///< Timer to open TFTP port
- int Udpv4Index; ///< Entry for UDPv4
- int Udpv6Index; ///< Entry for UDPv6
- int Entries; ///< Number of TFTP ports
- struct pollfd TftpPort [ 2 ]; ///< Poll descriptor for the TFTP ports (UDP4, UDP6)
-
- //
- // Request management
- //
- union {
- struct sockaddr_in v4; ///< UDP4 address
- struct sockaddr_in6 v6; ///< UDP6 address
- } RemoteAddress; ///< Remote address
- ssize_t RxBytes; ///< Receive data length in bytes
- UINT8 RxBuffer[ 2 + 2 + TFTP_MAX_BLOCK_SIZE ]; ///< Receive buffer
-
- //
- // Client port management
- //
- TSDT_CONNECTION_CONTEXT * pContextList; ///< List of connection context structures
-} TSDT_TFTP_SERVER;
-
-//#define SERVER_FROM_SERVICE(a) CR(a, TSDT_TFTP_SERVER, ServiceBinding, TFTP_SERVER_SIGNATURE) ///< Locate DT_LAYER from service binding
-
-extern TSDT_TFTP_SERVER mTftpServer;
-
-//------------------------------------------------------------------------------
-// Support routines
-//------------------------------------------------------------------------------
-
-/**
- Queue data packets for transmission
-
- @param [in] pContext Connection context structure address
-
- @retval TRUE if a read error occurred
-
-**/
-BOOLEAN
-PacketFill (
- IN TSDT_CONNECTION_CONTEXT * pContext
- );
-
-/**
- Free the packet
-
- @param [in] pContext Address of a ::TSDT_CONNECTION_CONTEXT structure
- @param [in] pPacket Address of a ::TFTP_PACKET structure
-
-**/
-VOID
-PacketFree(
- IN TSDT_CONNECTION_CONTEXT * pContext,
- IN TFTP_PACKET * pPacket
- );
-
-/**
- Get a packet for transmission
-
- @param [in] pContext Address of a ::TSDT_CONNECTION_CONTEXT structure
-
- @retval Address of a ::TFTP_PACKET structure
-
-**/
-TFTP_PACKET *
-PacketGet (
- IN TSDT_CONNECTION_CONTEXT * pContext
- );
-
-/**
- Queue the packet for transmission
-
- @param [in] pContext Address of a ::TSDT_CONNECTION_CONTEXT structure
- @param [in] pPacket Address of a ::TFTP_PACKET structure
-
- @retval TRUE if a transmission error has occurred
-
-**/
-BOOLEAN
-PacketQueue (
- IN TSDT_CONNECTION_CONTEXT * pContext,
- IN TFTP_PACKET * pPacket
- );
-
-/**
- Transmit the packet
-
- @param [in] pContext Address of a ::TSDT_CONNECTION_CONTEXT structure
- @param [in] pPacket Address of a ::TFTP_PACKET structure
-
- @retval EFI_SUCCESS Message processed successfully
-
-**/
-EFI_STATUS
-PacketTx (
- IN TSDT_CONNECTION_CONTEXT * pContext,
- IN TFTP_PACKET * pPacket
- );
-
-/**
- Build and send an error packet
-
- @param [in] pContext The context structure address.
- @param [in] Error Error number for the packet
- @param [in] pError Zero terminated error string address
-
- @retval EFI_SUCCESS Message processed successfully
-
-**/
-EFI_STATUS
-SendError (
- IN TSDT_CONNECTION_CONTEXT * pContext,
- IN UINT16 Error,
- IN UINT8 * pError
- );
-
-/**
- Process the TFTP request
-
- @param [in] pOption Address of the first zero terminated option string
- @param [in] pValue Address to receive the value
-
- @retval EFI_SUCCESS Option translated into a value
-
-**/
-EFI_STATUS
-TftpOptionValue (
- IN UINT8 * pOption,
- IN INT32 * pValue
- );
-
-/**
- Process the TFTP request
-
- @param [in] pTftpServer The TFTP server control structure address.
- @param [in] pContext Connection context structure address
- @param [in] SocketFd Socket file descriptor
-
-**/
-VOID
-TftpProcessRequest (
- IN TSDT_TFTP_SERVER * pTftpServer,
- IN TSDT_CONNECTION_CONTEXT * pContext,
- IN int SocketFd
- );
-
-/**
- Process the read request
-
- @param [in] pTftpServer Address of the ::TSDT_TFTP_SERVER structure
- @param [in] pContext Connection context structure address
- @param [in] SocketFd Socket file descriptor
-
- @retval TRUE if the context should be closed
-
-**/
-BOOLEAN
-TftpRead (
- IN TSDT_TFTP_SERVER * pTftpServer,
- IN TSDT_CONNECTION_CONTEXT * pContext,
- IN int SocketFd
- );
-
-/**
- Update the window due to the ACK
-
- @param [in] pTftpServer Address of the ::TSDT_TFTP_SERVER structure
- @param [in] pContext Address of a ::TSDT_CONNECTION_CONTEXT structure
- @param [in] pPacket Address of a ::TFTP_PACKET structure
-
-**/
-VOID
-WindowAck (
- IN TSDT_TFTP_SERVER * pTftpServer,
- IN TSDT_CONNECTION_CONTEXT * pContext,
- IN TFTP_PACKET * pPacket
- );
-
-/**
- A timeout has occurred, close the window
-
- @param [in] pContext Address of a ::TSDT_CONNECTION_CONTEXT structure
-
-**/
-VOID
-WindowTimeout (
- IN TSDT_CONNECTION_CONTEXT * pContext
- );
-
-//------------------------------------------------------------------------------
-
-#endif // _TFTP_SERVER_H_
diff --git a/AppPkg/Applications/Sockets/TftpServer/TftpServer.inf b/AppPkg/Applications/Sockets/TftpServer/TftpServer.inf deleted file mode 100644 index 79513c2215..0000000000 --- a/AppPkg/Applications/Sockets/TftpServer/TftpServer.inf +++ /dev/null @@ -1,70 +0,0 @@ -## @file
-# TFTP Server Application
-#
-# Copyright (c) 2011-2012, Intel Corporation
-# All rights reserved. This program and the accompanying materials
-# are licensed and made available under the terms and conditions of the BSD License
-# which accompanies this distribution. The full text of the license may be found at
-# http://opensource.org/licenses/bsd-license.php
-#
-# THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,
-# WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
-#
-##
-
-
-[Defines]
- INF_VERSION = 0x00010005
- BASE_NAME = TftpServer
- FILE_GUID = 94621F9E-B5CA-4cfd-82BE-0C542EB0D9BE
- MODULE_TYPE = UEFI_APPLICATION
- VERSION_STRING = 1.0
- ENTRY_POINT = ShellCEntryLib
-
-#
-# The following information is for reference only and not required by the build tools.
-#
-# VALID_ARCHITECTURES = IA32 X64 IPF EBC
-#
-
-[Sources]
- TftpServer.h
- TftpServer.c
-
-
-[Pcd]
- gAppPkgTokenSpaceGuid.Tftp_AckLogBase
- gAppPkgTokenSpaceGuid.Tftp_AckMultiplier
- gAppPkgTokenSpaceGuid.Tftp_Bandwidth
- gAppPkgTokenSpaceGuid.Tftp_HighSpeed
- gAppPkgTokenSpaceGuid.Tftp_MaxRetry
- gAppPkgTokenSpaceGuid.Tftp_MaxTimeoutInSec
-
-
-[Packages]
- AppPkg/AppPkg.dec
- MdePkg/MdePkg.dec
- ShellPkg/ShellPkg.dec
- StdLib/StdLib.dec
-
-
-[LibraryClasses]
- BaseMemoryLib
- BsdSocketLib
- DebugLib
- DevMedia
- EfiSocketLib
- LibC
- ShellLib
- ShellCEntryLib
- TimerLib
- UefiBootServicesTableLib
- UefiLib
- UefiRuntimeServicesTableLib
-# UseSocketDxe
-
-[BuildOptions]
- INTEL:*_*_*_CC_FLAGS = /Qdiag-disable:181,186
- MSFT:*_*_*_CC_FLAGS = /Od
- GCC:*_*_*_CC_FLAGS = -O0 -Wno-unused-variable
-
|