/*
* Copyright (C) 2006 Michael Brown <mbrown@fensystems.co.uk>.
*
* This program is free software; you can redistribute it and/or
* modify it under the terms of the GNU General Public License as
* published by the Free Software Foundation; either version 2 of the
* License, or any later version.
*
* This program is distributed in the hope that it will be useful, but
* WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program; if not, write to the Free Software
* Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
*/
FILE_LICENCE ( GPL2_OR_LATER );
#include <stdint.h>
#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <strings.h>
#include <byteswap.h>
#include <errno.h>
#include <assert.h>
#include <gpxe/refcnt.h>
#include <gpxe/xfer.h>
#include <gpxe/open.h>
#include <gpxe/uri.h>
#include <gpxe/tcpip.h>
#include <gpxe/retry.h>
#include <gpxe/features.h>
#include <gpxe/bitmap.h>
#include <gpxe/settings.h>
#include <gpxe/dhcp.h>
#include <gpxe/uri.h>
#include <gpxe/tftp.h>
/** @file
*
* TFTP protocol
*
*/
FEATURE ( FEATURE_PROTOCOL, "TFTP", DHCP_EB_FEATURE_TFTP, 1 );
/* TFTP-specific error codes */
#define ETFTP_INVALID_BLKSIZE EUNIQ_01
#define ETFTP_INVALID_TSIZE EUNIQ_02
#define ETFTP_MC_NO_PORT EUNIQ_03
#define ETFTP_MC_NO_MC EUNIQ_04
#define ETFTP_MC_INVALID_MC EUNIQ_05
#define ETFTP_MC_INVALID_IP EUNIQ_06
#define ETFTP_MC_INVALID_PORT EUNIQ_07
/**
* A TFTP request
*
* This data structure holds the state for an ongoing TFTP transfer.
*/
struct tftp_request {
/** Reference count */
struct refcnt refcnt;
/** Data transfer interface */
struct xfer_interface xfer;
/** URI being fetched */
struct uri *uri;
/** Transport layer interface */
struct xfer_interface socket;
/** Multicast transport layer interface */
struct xfer_interface mc_socket;
/** Data block size
*
* This is the "blksize" option negotiated with the TFTP
* server. (If the TFTP server does not support TFTP options,
* this will default to 512).
*/
unsigned int blksize;
/** File size
*
* This is the value returned in the "tsize" option from the
* TFTP server. If the TFTP server does not support the
* "tsize" option, this value will be zero.
*/
unsigned long tsize;
/** Server port
*
* This is the port to which RRQ packets are sent.
*/
unsigned int port;
/** Peer address
*
* The peer address is determined by the first response
* received to the TFTP RRQ.
*/
struct sockaddr_tcpip peer;
/** Request flags */
unsigned int flags;
/** MTFTP timeout count */
unsigned int mtftp_timeouts;
/** Block bitmap */
struct bitmap bitmap;
/** Maximum known length
*
* We don't always know the file length in advance. In
* particular, if the TFTP server doesn't support the tsize
* option, or we are using MTFTP, then we don't know the file
* length until we see the end-of-file block (which, in the
* case of MTFTP, may not be the last block we see).
*
* This value is updated whenever we obtain information about
* the file length.
*/
size_t filesize;
/** Retransmission timer */
struct retry_timer timer;
};
/** TFTP request flags */
enum {
/** Send ACK packets */
TFTP_FL_SEND_ACK = 0x0001,
/** Request blksize and tsize options */
TFTP_FL_RRQ_SIZES = 0x0002,
/** Request multicast option */
TFTP_FL_RRQ_MULTICAST = 0x0004,
/** Perform MTFTP recovery on timeout */
TFTP_FL_MTFTP_RECOVERY = 0x0008,
/** Only get filesize and then abort the transfer */
TFTP_FL_SIZEONLY = 0x0010,
};
/** Maximum number of MTFTP open requests before falling back to TFTP */
#define MTFTP_MAX_TIMEOUTS 3
/**
* Free TFTP request
*
* @v refcnt Reference counter
*/
static void tftp_free ( struct refcnt *refcnt ) {
struct tftp_request *tftp =
container_of ( refcnt, struct tftp_request, refcnt );
uri_put ( tftp->uri );
bitmap_free ( &tftp->bitmap );
free ( tftp );
}
/**
* Mark TFTP request as complete
*
* @v tftp TFTP connection
* @v rc Return status code
*/
static void tftp_done ( struct tftp_request *tftp, int rc ) {
DBGC ( tftp, "TFTP %p finished with status %d (%s)\n",
tftp, rc, strerror ( rc ) );
/* Stop the retry timer */
stop_timer ( &tftp->timer );
/* Close all data transfer interfaces */
xfer_nullify ( &tftp->socket );
xfer_close ( &tftp->socket, rc );
xfer_nullify ( &tftp->mc_socket );
xfer_close ( &tftp->mc_socket, rc );
xfer_nullify ( &tftp->xfer );
xfer_close ( &tftp->xfer, rc );
}
/**
* Reopen TFTP socket
*
* @v tftp TFTP connection
* @ret rc Return status code
*/
static int tftp_reopen ( struct tftp_request *tftp ) {
struct sockaddr_tcpip server;
int rc;
/* Close socket */
xfer_close ( &tftp->socket, 0 );
/* Disable ACK sending. */
tftp->flags &= ~TFTP_FL_SEND_ACK;
/* Reset peer address */
memset ( &tftp->peer, 0, sizeof ( tftp->peer ) );
/* Open socket */
memset ( &server, 0, sizeof ( server ) );
server.st_port = htons ( tftp->port );
if ( ( rc = xfer_open_named_socket ( &tftp->socket, SOCK_DGRAM,
( struct sockaddr * ) &server,
tftp->uri->host, NULL ) ) != 0 ) {
DBGC ( tftp, "TFTP %p could not open socket: %s\n",
tftp, strerror ( rc ) );
return rc;
}
return 0;
}
/**
* Reopen TFTP multicast socket
*
* @v tftp TFTP connection
* @v local Local socket address
* @ret rc Return status code
*/
static int tftp_reopen_mc ( struct tftp_request *tftp,
struct sockaddr *local ) {
int rc;
/* Close multicast socket */
xfer_close ( &tftp->mc_socket, 0 );
/* Open multicast socket. We never send via this socket, so
* use the local address as the peer address (since the peer
* address cannot be NULL).
*/
if ( ( rc = xfer_open_socket ( &tftp->mc_socket, SOCK_DGRAM,
local, local ) ) != 0 ) {
DBGC ( tftp, "TFTP %p could not open multicast "
"socket: %s\n", tftp, strerror ( rc ) );
return rc;
}
return 0;
}
/**
* Presize TFTP receive buffers and block bitmap
*
* @v tftp TFTP connection
* @v filesize Known minimum file size
* @ret rc Return status code
*/
static int tftp_presize ( struct tftp_request *tftp, size_t filesize ) {
unsigned int num_blocks;
int rc;
/* Do nothing if we are already large enough */
if ( filesize <= tftp->filesize )
return 0;
/* Record filesize */
tftp->filesize = filesize;
/* Notify recipient of file size */
xfer_seek ( &tftp->xfer, filesize, SEEK_SET );
xfer_seek ( &tftp->xfer, 0, SEEK_SET );
/* Calculate expected number of blocks. Note that files whose
* length is an exact multiple of the blocksize will have a
* trailing zero-length block, which must be included.
*/
num_blocks = ( ( filesize / tftp->blksize ) + 1 );
if ( ( rc = bitmap_resize ( &tftp->bitmap, num_blocks ) ) != 0 ) {
DBGC ( tftp, "TFTP %p could not resize bitmap to %d blocks: "
"%s\n", tftp, num_blocks, strerror ( rc ) );
return rc;
}
return 0;
}
/**
* TFTP requested blocksize
*
* This is treated as a global configuration parameter.
*/
static unsigned int tftp_request_blksize = TFTP_MAX_BLKSIZE;
/**
* Set TFTP request blocksize
*
* @v blksize Requested block size
*/
void tftp_set_request_blksize ( unsigned int blksize ) {
if ( blksize < TFTP_DEFAULT_BLKSIZE )
blksize = TFTP_DEFAULT_BLKSIZE;
tftp_request_blksize = blksize;
}
/**
* MTFTP multicast receive address
*
* This is treated as a global configuration parameter.
*/
static struct sockaddr_in tftp_mtftp_socket = {
.sin_family = AF_INET,
.sin_addr.s_addr = htonl ( 0xefff0101 ),
.sin_port = htons ( 3001 ),
};
/**
* Set MTFTP multicast address
*
* @v address Multicast IPv4 address
*/
void tftp_set_mtftp_address ( struct in_addr address ) {
tftp_mtftp_socket.sin_addr = address;
}
/**
* Set MTFTP multicast port
*
* @v port Multicast port
*/
void tftp_set_mtftp_port ( unsigned int port ) {
tftp_mtftp_socket.sin_port = htons ( port );
}
/**
* Transmit RRQ
*
* @v tftp TFTP connection
* @ret rc Return status code
*/
static int tftp_send_rrq ( struct tftp_request *tftp ) {
struct tftp_rrq *rrq;
const char *path;
size_t len;
struct io_buffer *iobuf;
/* Strip initial '/' if present. If we were opened via the
* URI interface, then there will be an initial '/', since a
* full tftp:// URI provides no way to specify a non-absolute
* path. However, many TFTP servers (particularly Windows
* TFTP servers) complain about having an initial '/', and it
* violates user expectations to have a '/' silently added to
* the DHCP-specified filename.
*/
path = tftp->uri->path;
if ( *path == '/' )
path++;
DBGC ( tftp, "TFTP %p requesting \"%s\"\n", tftp, path );
/* Allocate buffer */
len = ( sizeof ( *rrq ) + strlen ( path ) + 1 /* NUL */
+ 5 + 1 /* "octet" + NUL */
+ 7 + 1 + 5 + 1 /* "blksize" + NUL + ddddd + NUL */
+ 5 + 1 + 1 + 1 /* "tsize" + NUL + "0" + NUL */
+ 9 + 1 + 1 /* "multicast" + NUL + NUL */ );
iobuf = xfer_alloc_iob ( &tftp->socket, len );
if ( ! iobuf )
return -ENOMEM;
/* Build request */
rrq = iob_put ( iobuf, sizeof ( *rrq ) );
rrq->opcode = htons ( TFTP_RRQ );
iob_put ( iobuf, snprintf ( iobuf->tail, iob_tailroom ( iobuf ),
"%s%coctet", path, 0 ) + 1 );
if ( tftp->flags & TFTP_FL_RRQ_SIZES ) {
iob_put ( iobuf, snprintf ( iobuf->tail,
iob_tailroom ( iobuf ),
"blksize%c%d%ctsize%c0", 0,
tftp_request_blksize, 0, 0 ) + 1 );
}
if ( tftp->flags & TFTP_FL_RRQ_MULTICAST ) {
iob_put ( iobuf, snprintf ( iobuf->tail,
iob_tailroom ( iobuf ),
"multicast%c", 0 ) + 1 );
}
/* RRQ always goes to the address specified in the initial
* xfer_open() call
*/
return xfer_deliver_iob ( &tftp->socket, iobuf );
}
/**
* Transmit ACK
*
* @v tftp TFTP connection
* @ret rc Return status code
*/
static int tftp_send_ack ( struct tftp_request *tftp ) {
struct tftp_ack *ack;
struct io_buffer *iobuf;
struct xfer_metadata meta = {
.dest = ( struct sockaddr * ) &tftp->peer,
};
unsigned int block;
/* Determine next required block number */
block = bitmap_first_gap ( &tftp->bitmap );
DBGC2 ( tftp, "TFTP %p sending ACK for block %d\n", tftp, block );
/* Allocate buffer */
iobuf = xfer_alloc_iob ( &tftp->socket, sizeof ( *ack ) );
if ( ! iobuf )
return -ENOMEM;
/* Build ACK */
ack = iob_put ( iobuf, sizeof ( *ack ) );
ack->opcode = htons ( TFTP_ACK );
ack->block = htons ( block );
/* ACK always goes to the peer recorded from the RRQ response */
return xfer_deliver_iob_meta ( &tftp->socket, iobuf, &meta );
}
/**
* Transmit ERROR (Abort)
*
* @v tftp TFTP connection
* @v errcode TFTP error code
* @v errmsg Error message string
* @ret rc Return status code
*/
static int tftp_send_error ( struct tftp_request *tftp, int errcode,
const char *errmsg ) {
struct tftp_error *err;
struct io_buffer *iobuf;
struct xfer_metadata meta = {
.dest = ( struct sockaddr * ) &tftp->peer,
};
size_t msglen;
DBGC2 ( tftp, "TFTP %p sending ERROR %d: %s\n", tftp, errcode,
errmsg );
/* Allocate buffer */
msglen = sizeof ( *err ) + strlen ( errmsg ) + 1 /* NUL */;
iobuf = xfer_alloc_iob ( &tftp->socket, msglen );
if ( ! iobuf )
return -ENOMEM;
/* Build ERROR */
err = iob_put ( iobuf, msglen );
err->opcode = htons ( TFTP_ERROR );
err->errcode = htons ( errcode );
strcpy ( err->errmsg, errmsg );
/* ERR always goes to the peer recorded from the RRQ response */
return xfer_deliver_iob_meta ( &tftp->socket, iobuf, &meta );
}
/**
* Transmit next relevant packet
*
* @v tftp TFTP connection
* @ret rc Return status code
*/
static int tftp_send_packet ( struct tftp_request *tftp ) {
/* Update retransmission timer. While name resolution takes place the
* window is zero. Avoid unnecessary delay after name resolution
* completes by retrying immediately.
*/
stop_timer ( &tftp->timer );
if ( xfer_window ( &tftp->socket ) ) {
start_timer ( &tftp->timer );
} else {
start_timer_nodelay ( &tftp->timer );
}
/* Send RRQ or ACK as appropriate */
if ( ! tftp->peer.st_family ) {
return tftp_send_rrq ( tftp );
} else {
if ( tftp->flags & TFTP_FL_SEND_ACK ) {
return tftp_send_ack ( tftp );
} else {
return 0;
}
}
}
/**
* Handle TFTP retransmission timer expiry
*
* @v timer Retry timer
* @v fail Failure indicator
*/
static void tftp_timer_expired ( struct retry_timer *timer, int fail ) {
struct tftp_request *tftp =
container_of ( timer, struct tftp_request, timer );
int rc;
/* If we are doing MTFTP, attempt the various recovery strategies */
if ( tftp->flags & TFTP_FL_MTFTP_RECOVERY ) {
if ( tftp->peer.st_family ) {
/* If we have received any response from the server,
* try resending the RRQ to restart the download.
*/
DBGC ( tftp, "TFTP %p attempting reopen\n", tftp );
if ( ( rc = tftp_reopen ( tftp ) ) != 0 )
goto err;
} else {
/* Fall back to plain TFTP after several attempts */
tftp->mtftp_timeouts++;
DBGC ( tftp, "TFTP %p timeout %d waiting for MTFTP "
"open\n", tftp, tftp->mtftp_timeouts );
if ( tftp->mtftp_timeouts > MTFTP_MAX_TIMEOUTS ) {
DBGC ( tftp, "TFTP %p falling back to plain "
"TFTP\n", tftp );
tftp->flags = TFTP_FL_RRQ_SIZES;
/* Close multicast socket */
xfer_close ( &tftp->mc_socket, 0 );
/* Reset retry timer */
start_timer_nodelay ( &tftp->timer );
/* The blocksize may change: discard
* the block bitmap
*/
bitmap_free ( &tftp->bitmap );
memset ( &tftp->bitmap, 0,
sizeof ( tftp->bitmap ) );
/* Reopen on standard TFTP port */
tftp->port = TFTP_PORT;
if ( ( rc = tftp_reopen ( tftp ) ) != 0 )
goto err;
}
}
} else {
/* Not doing MTFTP (or have fallen back to plain
* TFTP); fail as per normal.
*/
if ( fail ) {
rc = -ETIMEDOUT;
goto err;
}
}
tftp_send_packet ( tftp );
return;
err:
tftp_done ( tftp, rc );
}
/**
* Process TFTP "blksize" option
*
* @v tftp TFTP connection
* @v value Option value
* @ret rc Return status code
*/
static int tftp_process_blksize ( struct tftp_request *tftp,
const char *value ) {
char *end;
tftp->blksize = strtoul ( value, &end, 10 );
if ( *end ) {
DBGC ( tftp, "TFTP %p got invalid blksize \"%s\"\n",
tftp, value );
return -( EINVAL | ETFTP_INVALID_BLKSIZE );
}
DBGC ( tftp, "TFTP %p blksize=%d\n", tftp, tftp->blksize );
return 0;
}
/**
* Process TFTP "tsize" option
*
* @v tftp TFTP connection
* @v value Option value
* @ret rc Return status code
*/
static int tftp_process_tsize ( struct tftp_request *tftp,
const char *value ) {
char *end;
tftp->tsize = strtoul ( value, &end, 10 );
if ( *end ) {
DBGC ( tftp, "TFTP %p got invalid tsize \"%s\"\n",
tftp, value );
return -( EINVAL | ETFTP_INVALID_TSIZE );
}
DBGC ( tftp, "TFTP %p tsize=%ld\n", tftp, tftp->tsize );
return 0;
}
/**
* Process TFTP "multicast" option
*
* @v tftp TFTP connection
* @v value Option value
* @ret rc Return status code
*/
static int tftp_process_multicast ( struct tftp_request *tftp,
const char *value ) {
union {
struct sockaddr sa;
struct sockaddr_in sin;
} socket;
char buf[ strlen ( value ) + 1 ];
char *addr;
char *port;
char *port_end;
char *mc;
char *mc_end;
int rc;
/* Split value into "addr,port,mc" fields */
memcpy ( buf, value, sizeof ( buf ) );
addr = buf;
port = strchr ( addr, ',' );
if ( ! port ) {
DBGC ( tftp, "TFTP %p multicast missing port,mc\n", tftp );
return -( EINVAL | ETFTP_MC_NO_PORT );
}
*(port++) = '\0';
mc = strchr ( port, ',' );
if ( ! mc ) {
DBGC ( tftp, "TFTP %p multicast missing mc\n", tftp );
return -( EINVAL | ETFTP_MC_NO_MC );
}
*(mc++) = '\0';
/* Parse parameters */
if ( strtoul ( mc, &mc_end, 0 ) == 0 )
tftp->flags &= ~TFTP_FL_SEND_ACK;
if ( *mc_end ) {
DBGC ( tftp, "TFTP %p multicast invalid mc %s\n", tftp, mc );
return -( EINVAL | ETFTP_MC_INVALID_MC );
}
DBGC ( tftp, "TFTP %p is%s the master client\n",
tftp, ( ( tftp->flags & TFTP_FL_SEND_ACK ) ? "" : " not" ) );
if ( *addr && *port ) {
socket.sin.sin_family = AF_INET;
if ( inet_aton ( addr, &socket.sin.sin_addr ) == 0 ) {
DBGC ( tftp, "TFTP %p multicast invalid IP address "
"%s\n", tftp, addr );
return -( EINVAL | ETFTP_MC_INVALID_IP );
}
DBGC ( tftp, "TFTP %p multicast IP address %s\n",
tftp, inet_ntoa ( socket.sin.sin_addr ) );
socket.sin.sin_port = htons ( strtoul ( port, &port_end, 0 ) );
if ( *port_end ) {
DBGC ( tftp, "TFTP %p multicast invalid port %s\n",
tftp, port );
return -( EINVAL | ETFTP_MC_INVALID_PORT );
}
DBGC ( tftp, "TFTP %p multicast port %d\n",
tftp, ntohs ( socket.sin.sin_port ) );
if ( ( rc = tftp_reopen_mc ( tftp, &socket.sa ) ) != 0 )
return rc;
}
return 0;
}
/** A TFTP option */
struct tftp_option {
/** Option name */
const char *name;
/** Option processor
*
* @v tftp TFTP connection
* @v value Option value
* @ret rc Return status code
*/
int ( * process ) ( struct tftp_request *tftp, const char *value );
};
/** Recognised TFTP options */
static struct tftp_option tftp_options[] = {
{ "blksize", tftp_process_blksize },
{ "tsize", tftp_process_tsize },
{ "multicast", tftp_process_multicast },
{ NULL, NULL }
};
/**
* Process TFTP option
*
* @v tftp TFTP connection
* @v name Option name
* @v value Option value
* @ret rc Return status code
*/
static int tftp_process_option ( struct tftp_request *tftp,
const char *name, const char *value ) {
struct tftp_option *option;
for ( option = tftp_options ; option->name ; option++ ) {
if ( strcasecmp ( name, option->name ) == 0 )
return option->process ( tftp, value );
}
DBGC ( tftp, "TFTP %p received unknown option \"%s\" = \"%s\"\n",
tftp, name, value );
/* Unknown options should be silently ignored */
return 0;
}
/**
* Receive OACK
*
* @v tftp TFTP connection
* @v buf Temporary data buffer
* @v len Length of temporary data buffer
* @ret rc Return status code
*/
static int tftp_rx_oack ( struct tftp_request *tftp, void *buf, size_t len ) {
struct tftp_oack *oack = buf;
char *end = buf + len;
char *name;
char *value;
char *next;
int rc = 0;
/* Sanity check */
if ( len < sizeof ( *oack ) ) {
DBGC ( tftp, "TFTP %p received underlength OACK packet "
"length %zd\n", tftp, len );
rc = -EINVAL;
goto done;
}
/* Process each option in turn */
for ( name = oack->data ; name < end ; name = next ) {
/* Parse option name and value
*
* We treat parsing errors as non-fatal, because there
* exists at least one TFTP server (IBM Tivoli PXE
* Server 5.1.0.3) that has been observed to send
* malformed OACKs containing trailing garbage bytes.
*/
value = ( name + strnlen ( name, ( end - name ) ) + 1 );
if ( value > end ) {
DBGC ( tftp, "TFTP %p received OACK with malformed "
"option name:\n", tftp );
DBGC_HD ( tftp, oack, len );
break;
}
if ( value == end ) {
DBGC ( tftp, "TFTP %p received OACK missing value "
"for option \"%s\"\n", tftp, name );
DBGC_HD ( tftp, oack, len );
break;
}
next = ( value + strnlen ( value, ( end - value ) ) + 1 );
if ( next > end ) {
DBGC ( tftp, "TFTP %p received OACK with malformed "
"value for option \"%s\":\n", tftp, name );
DBGC_HD ( tftp, oack, len );
break;
}
/* Process option */
if ( ( rc = tftp_process_option ( tftp, name, value ) ) != 0 )
goto done;
}
/* Process tsize information, if available */
if ( tftp->tsize ) {
if ( ( rc = tftp_presize ( tftp, tftp->tsize ) ) != 0 )
goto done;
}
/* Abort request if only trying to determine file size */
if ( tftp->flags & TFTP_FL_SIZEONLY ) {
rc = 0;
tftp_send_error ( tftp, 0, "TFTP Aborted" );
tftp_done ( tftp, rc );
return rc;
}
/* Request next data block */
tftp_send_packet ( tftp );
done:
if ( rc )
tftp_done ( tftp, rc );
return rc;
}
/**
* Receive DATA
*
* @v tftp TFTP connection
* @v iobuf I/O buffer
* @ret rc Return status code
*
* Takes ownership of I/O buffer.
*/
static int tftp_rx_data ( struct tftp_request *tftp,
struct io_buffer *iobuf ) {
struct tftp_data *data = iobuf->data;
struct xfer_metadata meta;
unsigned int block;
off_t offset;
size_t data_len;
int rc;
if ( tftp->flags & TFTP_FL_SIZEONLY ) {
/* If we get here then server doesn't support SIZE option */
rc = -ENOTSUP;
tftp_send_error ( tftp, 0, "TFTP Aborted" );
goto done;
}
/* Sanity check */
if ( iob_len ( iobuf ) < sizeof ( *data ) ) {
DBGC ( tftp, "TFTP %p received underlength DATA packet "
"length %zd\n", tftp, iob_len ( iobuf ) );
rc = -EINVAL;
goto done;
}
/* Calculate block number */
block = ( ( bitmap_first_gap ( &tftp->bitmap ) + 1 ) & ~0xffff );
if ( data->block == 0 && block == 0 ) {
DBGC ( tftp, "TFTP %p received data block 0\n", tftp );
rc = -EINVAL;
goto done;
}
block += ( ntohs ( data->block ) - 1 );
/* Extract data */
offset = ( block * tftp->blksize );
iob_pull ( iobuf, sizeof ( *data ) );
data_len = iob_len ( iobuf );
if ( data_len > tftp->blksize ) {
DBGC ( tftp, "TFTP %p received overlength DATA packet "
"length %zd\n", tftp, data_len );
rc = -EINVAL;
goto done;
}
/* Deliver data */
memset ( &meta, 0, sizeof ( meta ) );
meta.whence = SEEK_SET;
meta.offset = offset;
if ( ( rc = xfer_deliver_iob_meta ( &tftp->xfer, iob_disown ( iobuf ),
&meta ) ) != 0 ) {
DBGC ( tftp, "TFTP %p could not deliver data: %s\n",
tftp, strerror ( rc ) );
goto done;
}
/* Ensure block bitmap is ready */
if ( ( rc = tftp_presize ( tftp, ( offset + data_len ) ) ) != 0 )
goto done;
/* Mark block as received */
bitmap_set ( &tftp->bitmap, block );
/* Acknowledge block */
tftp_send_packet ( tftp );
/* If all blocks have been received, finish. */
if ( bitmap_full ( &tftp->bitmap ) )
tftp_done ( tftp, 0 );
done:
free_iob ( iobuf );
if ( rc )
tftp_done ( tftp, rc );
return rc;
}
/** Translation between TFTP errors and internal error numbers */
static const int tftp_errors[] = {
[TFTP_ERR_FILE_NOT_FOUND] = ENOENT,
[TFTP_ERR_ACCESS_DENIED] = EACCES,
[TFTP_ERR_ILLEGAL_OP] = ENOTSUP,
};
/**
* Receive ERROR
*
* @v tftp TFTP connection
* @v buf Temporary data buffer
* @v len Length of temporary data buffer
* @ret rc Return status code
*/
static int tftp_rx_error ( struct tftp_request *tftp, void *buf, size_t len ) {
struct tftp_error *error = buf;
unsigned int err;
int rc = 0;
/* Sanity check */
if ( len < sizeof ( *error ) ) {
DBGC ( tftp, "TFTP %p received underlength ERROR packet "
"length %zd\n", tftp, len );
return -EINVAL;
}
DBGC ( tftp, "TFTP %p received ERROR packet with code %d, message "
"\"%s\"\n", tftp, ntohs ( error->errcode ), error->errmsg );
/* Determine final operation result */
err = ntohs ( error->errcode );
if ( err < ( sizeof ( tftp_errors ) / sizeof ( tftp_errors[0] ) ) )
rc = -tftp_errors[err];
if ( ! rc )
rc = -ENOTSUP;
/* Close TFTP request */
tftp_done ( tftp, rc );
return 0;
}
/**
* Receive new data
*
* @v tftp TFTP connection
* @v iobuf I/O buffer
* @v meta Transfer metadata
* @ret rc Return status code
*/
static int tftp_rx ( struct tftp_request *tftp,
struct io_buffer *iobuf,
struct xfer_metadata *meta ) {
struct sockaddr_tcpip *st_src;
struct tftp_common *common = iobuf->data;
size_t len = iob_len ( iobuf );
int rc = -EINVAL;
/* Sanity checks */
if ( len < sizeof ( *common ) ) {
DBGC ( tftp, "TFTP %p received underlength packet length "
"%zd\n", tftp, len );
goto done;
}
if ( ! meta->src ) {
DBGC ( tftp, "TFTP %p received packet without source port\n",
tftp );
goto done;
}
/* Filter by TID. Set TID on first response received */
st_src = ( struct sockaddr_tcpip * ) meta->src;
if ( ! tftp->peer.st_family ) {
memcpy ( &tftp->peer, st_src, sizeof ( tftp->peer ) );
DBGC ( tftp, "TFTP %p using remote port %d\n", tftp,
ntohs ( tftp->peer.st_port ) );
} else if ( memcmp ( &tftp->peer, st_src,
sizeof ( tftp->peer ) ) != 0 ) {
DBGC ( tftp, "TFTP %p received packet from wrong source (got "
"%d, wanted %d)\n", tftp, ntohs ( st_src->st_port ),
ntohs ( tftp->peer.st_port ) );
goto done;
}
switch ( common->opcode ) {
case htons ( TFTP_OACK ):
rc = tftp_rx_oack ( tftp, iobuf->data, len );
break;
case htons ( TFTP_DATA ):
rc = tftp_rx_data ( tftp, iob_disown ( iobuf ) );
break;
case htons ( TFTP_ERROR ):
rc = tftp_rx_error ( tftp, iobuf->data, len );
break;
default:
DBGC ( tftp, "TFTP %p received strange packet type %d\n",
tftp, ntohs ( common->opcode ) );
break;
};
done:
free_iob ( iobuf );
return rc;
}
/**
* Receive new data via socket
*
* @v socket Transport layer interface
* @v iobuf I/O buffer
* @v meta Transfer metadata
* @ret rc Return status code
*/
static int tftp_socket_deliver_iob ( struct xfer_interface *socket,
struct io_buffer *iobuf,
struct xfer_metadata *meta ) {
struct tftp_request *tftp =
container_of ( socket, struct tftp_request, socket );
/* Enable sending ACKs when we receive a unicast packet. This
* covers three cases:
*
* 1. Standard TFTP; we should always send ACKs, and will
* always receive a unicast packet before we need to send the
* first ACK.
*
* 2. RFC2090 multicast TFTP; the only unicast packets we will
* receive are the OACKs; enable sending ACKs here (before
* processing the OACK) and disable it when processing the
* multicast option if we are not the master client.
*
* 3. MTFTP; receiving a unicast datagram indicates that we
* are the "master client" and should send ACKs.
*/
tftp->flags |= TFTP_FL_SEND_ACK;
return tftp_rx ( tftp, iobuf, meta );
}
/** TFTP socket operations */
static struct xfer_interface_operations tftp_socket_operations = {
.close = ignore_xfer_close,
.vredirect = xfer_vreopen,
.window = unlimited_xfer_window,
.alloc_iob = default_xfer_alloc_iob,
.deliver_iob = tftp_socket_deliver_iob,
.deliver_raw = xfer_deliver_as_iob,
};
/**
* Receive new data via multicast socket
*
* @v mc_socket Multicast transport layer interface
* @v iobuf I/O buffer
* @v meta Transfer metadata
* @ret rc Return status code
*/
static int tftp_mc_socket_deliver_iob ( struct xfer_interface *mc_socket,
struct io_buffer *iobuf,
struct xfer_metadata *meta ) {
struct tftp_request *tftp =
container_of ( mc_socket, struct tftp_request, mc_socket );
return tftp_rx ( tftp, iobuf, meta );
}
/** TFTP multicast socket operations */
static struct xfer_interface_operations tftp_mc_socket_operations = {
.close = ignore_xfer_close,
.vredirect = xfer_vreopen,
.window = unlimited_xfer_window,
.alloc_iob = default_xfer_alloc_iob,
.deliver_iob = tftp_mc_socket_deliver_iob,
.deliver_raw = xfer_deliver_as_iob,
};
/**
* Close TFTP data transfer interface
*
* @v xfer Data transfer interface
* @v rc Reason for close
*/
static void tftp_xfer_close ( struct xfer_interface *xfer, int rc ) {
struct tftp_request *tftp =
container_of ( xfer, struct tftp_request, xfer );
DBGC ( tftp, "TFTP %p interface closed: %s\n",
tftp, strerror ( rc ) );
tftp_done ( tftp, rc );
}
/**
* Check flow control window
*
* @v xfer Data transfer interface
* @ret len Length of window
*/
static size_t tftp_xfer_window ( struct xfer_interface *xfer ) {
struct tftp_request *tftp =
container_of ( xfer, struct tftp_request, xfer );
/* We abuse this data-xfer method to convey the blocksize to
* the caller. This really should be done using some kind of
* stat() method, but we don't yet have the facility to do
* that.
*/
return tftp->blksize;
}
/** TFTP data transfer interface operations */
static struct xfer_interface_operations tftp_xfer_operations = {
.close = tftp_xfer_close,
.vredirect = ignore_xfer_vredirect,
.window = tftp_xfer_window,
.alloc_iob = default_xfer_alloc_iob,
.deliver_iob = xfer_deliver_as_raw,
.deliver_raw = ignore_xfer_deliver_raw,
};
/**
* Initiate TFTP/TFTM/MTFTP download
*
* @v xfer Data transfer interface
* @v uri Uniform Resource Identifier
* @ret rc Return status code
*/
static int tftp_core_open ( struct xfer_interface *xfer, struct uri *uri,
unsigned int default_port,
struct sockaddr *multicast,
unsigned int flags ) {
struct tftp_request *tftp;
int rc;
/* Sanity checks */
if ( ! uri->host )
return -EINVAL;
if ( ! uri->path )
return -EINVAL;
/* Allocate and populate TFTP structure */
tftp = zalloc ( sizeof ( *tftp ) );
if ( ! tftp )
return -ENOMEM;
tftp->refcnt.free = tftp_free;
xfer_init ( &tftp->xfer, &tftp_xfer_operations, &tftp->refcnt );
tftp->uri = uri_get ( uri );
xfer_init ( &tftp->socket, &tftp_socket_operations, &tftp->refcnt );
xfer_init ( &tftp->mc_socket, &tftp_mc_socket_operations,
&tftp->refcnt );
tftp->blksize = TFTP_DEFAULT_BLKSIZE;
tftp->flags = flags;
tftp->timer.expired = tftp_timer_expired;
/* Open socket */
tftp->port = uri_port ( tftp->uri, default_port );
if ( ( rc = tftp_reopen ( tftp ) ) != 0 )
goto err;
/* Open multicast socket */
if ( multicast ) {
if ( ( rc = tftp_reopen_mc ( tftp, multicast ) ) != 0 )
goto err;
}
/* Start timer to initiate RRQ */
start_timer_nodelay ( &tftp->timer );
/* Attach to parent interface, mortalise self, and return */
xfer_plug_plug ( &tftp->xfer, xfer );
ref_put ( &tftp->refcnt );
return 0;
err:
DBGC ( tftp, "TFTP %p could not create request: %s\n",
tftp, strerror ( rc ) );
tftp_done ( tftp, rc );
ref_put ( &tftp->refcnt );
return rc;
}
/**
* Initiate TFTP download
*
* @v xfer Data transfer interface
* @v uri Uniform Resource Identifier
* @ret rc Return status code
*/
static int tftp_open ( struct xfer_interface *xfer, struct uri *uri ) {
return tftp_core_open ( xfer, uri, TFTP_PORT, NULL,
TFTP_FL_RRQ_SIZES );
}
/** TFTP URI opener */
struct uri_opener tftp_uri_opener __uri_opener = {
.scheme = "tftp",
.open = tftp_open,
};
/**
* Initiate TFTP-size request
*
* @v xfer Data transfer interface
* @v uri Uniform Resource Identifier
* @ret rc Return status code
*/
static int tftpsize_open ( struct xfer_interface *xfer, struct uri *uri ) {
return tftp_core_open ( xfer, uri, TFTP_PORT, NULL,
( TFTP_FL_RRQ_SIZES |
TFTP_FL_SIZEONLY ) );
}
/** TFTP URI opener */
struct uri_opener tftpsize_uri_opener __uri_opener = {
.scheme = "tftpsize",
.open = tftpsize_open,
};
/**
* Initiate TFTM download
*
* @v xfer Data transfer interface
* @v uri Uniform Resource Identifier
* @ret rc Return status code
*/
static int tftm_open ( struct xfer_interface *xfer, struct uri *uri ) {
return tftp_core_open ( xfer, uri, TFTP_PORT, NULL,
( TFTP_FL_RRQ_SIZES |
TFTP_FL_RRQ_MULTICAST ) );
}
/** TFTM URI opener */
struct uri_opener tftm_uri_opener __uri_opener = {
.scheme = "tftm",
.open = tftm_open,
};
/**
* Initiate MTFTP download
*
* @v xfer Data transfer interface
* @v uri Uniform Resource Identifier
* @ret rc Return status code
*/
static int mtftp_open ( struct xfer_interface *xfer, struct uri *uri ) {
return tftp_core_open ( xfer, uri, MTFTP_PORT,
( struct sockaddr * ) &tftp_mtftp_socket,
TFTP_FL_MTFTP_RECOVERY );
}
/** MTFTP URI opener */
struct uri_opener mtftp_uri_opener __uri_opener = {
.scheme = "mtftp",
.open = mtftp_open,
};
/******************************************************************************
*
* Settings
*
******************************************************************************
*/
/** TFTP server setting */
struct setting next_server_setting __setting = {
.name = "next-server",
.description = "TFTP server",
.tag = DHCP_EB_SIADDR,
.type = &setting_type_ipv4,
};
/**
* Apply TFTP configuration settings
*
* @ret rc Return status code
*/
static int tftp_apply_settings ( void ) {
static struct in_addr tftp_server = { 0 };
struct in_addr last_tftp_server;
char uri_string[32];
struct uri *uri;
/* Retrieve TFTP server setting */
last_tftp_server = tftp_server;
fetch_ipv4_setting ( NULL, &next_server_setting, &tftp_server );
/* If TFTP server setting has changed, set the current working
* URI to match. Do it only when the TFTP server has changed
* to try to minimise surprises to the user, who probably
* won't expect the CWURI to change just because they updated
* an unrelated setting and triggered all the settings
* applicators.
*/
if ( tftp_server.s_addr != last_tftp_server.s_addr ) {
snprintf ( uri_string, sizeof ( uri_string ),
"tftp://%s/", inet_ntoa ( tftp_server ) );
uri = parse_uri ( uri_string );
if ( ! uri )
return -ENOMEM;
churi ( uri );
uri_put ( uri );
}
return 0;
}
/** TFTP settings applicator */
struct settings_applicator tftp_settings_applicator __settings_applicator = {
.apply = tftp_apply_settings,
};