sysadmin/ipxe
1
0
Fork 0
mirror of https://github.com/ipxe/ipxe synced 2025-12-22 13:00:39 +03:00
Code Issues Packages Projects Releases Wiki Activity

[pxe] Maintain a queue for received PXE UDP packets

Browse Source 
Some devices return multiple packets in a single poll.  Handle such
devices gracefully by enqueueing received PXE UDP packets (along with
a pseudo-header to hold the IPv4 addresses and port numbers) and
dequeueing them on subsequent calls to PXENV_UDP_READ.

Signed-off-by: Michael Brown <mcb30@ipxe.org>
This commit is contained in:
Michael Brown
2015-02-06 14:13:26 +00:00
parent c9dbe1d39c
commit 95bc563f0c
1 changed files with 85 additions and 38 deletions
Download Patch File Download Diff File Expand all files Collapse all files

123
src/arch/i386/interface/pxe/pxe_udp.c
View File

@@ -34,14 +34,26 @@
FILE_LICENCE ( GPL2_OR_LATER ); FILE_LICENCE ( GPL2_OR_LATER );
/** A PXE UDP pseudo-header */
struct pxe_udp_pseudo_header {
/** Source IP address */
IP4_t src_ip;
/** Source port */
UDP_PORT_t s_port;
/** Destination IP address */
IP4_t dest_ip;
/** Destination port */
UDP_PORT_t d_port;
} __attribute__ (( packed ));
/** A PXE UDP connection */ /** A PXE UDP connection */
struct pxe_udp_connection { struct pxe_udp_connection {
/** Data transfer interface to UDP stack */ /** Data transfer interface to UDP stack */
struct interface xfer; struct interface xfer;
/** Local address */ /** Local address */
struct sockaddr_in local; struct sockaddr_in local;
/** Current PXENV_UDP_READ parameter block */ /** List of received packets */
struct s_PXENV_UDP_READ *pxenv_udp_read; struct list_head list;
}; };
/** /**
@@ -58,45 +70,38 @@ struct pxe_udp_connection {
static int pxe_udp_deliver ( struct pxe_udp_connection *pxe_udp, static int pxe_udp_deliver ( struct pxe_udp_connection *pxe_udp,
struct io_buffer *iobuf, struct io_buffer *iobuf,
struct xfer_metadata *meta ) { struct xfer_metadata *meta ) {
struct s_PXENV_UDP_READ *pxenv_udp_read = pxe_udp->pxenv_udp_read; struct pxe_udp_pseudo_header *pshdr;
struct sockaddr_in *sin_src; struct sockaddr_in *sin_src;
struct sockaddr_in *sin_dest; struct sockaddr_in *sin_dest;
userptr_t buffer; int rc;
size_t len;
int rc = 0;
if ( ! pxenv_udp_read ) { /* Extract metadata */
DBG ( "PXE discarded UDP packet\n" );
rc = -ENOBUFS;
goto done;
}
/* Copy packet to buffer and record length */
buffer = real_to_user ( pxenv_udp_read->buffer.segment,
pxenv_udp_read->buffer.offset );
len = iob_len ( iobuf );
if ( len > pxenv_udp_read->buffer_size )
len = pxenv_udp_read->buffer_size;
copy_to_user ( buffer, 0, iobuf->data, len );
pxenv_udp_read->buffer_size = len;
/* Fill in source/dest information */
assert ( meta ); assert ( meta );
sin_src = ( struct sockaddr_in * ) meta->src; sin_src = ( struct sockaddr_in * ) meta->src;
assert ( sin_src ); assert ( sin_src );
assert ( sin_src->sin_family == AF_INET ); assert ( sin_src->sin_family == AF_INET );
pxenv_udp_read->src_ip = sin_src->sin_addr.s_addr;
pxenv_udp_read->s_port = sin_src->sin_port;
sin_dest = ( struct sockaddr_in * ) meta->dest; sin_dest = ( struct sockaddr_in * ) meta->dest;
assert ( sin_dest ); assert ( sin_dest );
assert ( sin_dest->sin_family == AF_INET ); assert ( sin_dest->sin_family == AF_INET );
pxenv_udp_read->dest_ip = sin_dest->sin_addr.s_addr;
pxenv_udp_read->d_port = sin_dest->sin_port;
/* Mark as received */ /* Construct pseudo-header */
pxe_udp->pxenv_udp_read = NULL; if ( ( rc = iob_ensure_headroom ( iobuf, sizeof ( *pshdr ) ) ) != 0 ) {
DBG ( "PXE could not prepend pseudo-header\n" );
rc = -ENOMEM;
goto drop;
}
pshdr = iob_push ( iobuf, sizeof ( *pshdr ) );
pshdr->src_ip = sin_src->sin_addr.s_addr;
pshdr->s_port = sin_src->sin_port;
pshdr->dest_ip = sin_dest->sin_addr.s_addr;
pshdr->d_port = sin_dest->sin_port;
done: /* Add to queue */
list_add_tail ( &iobuf->list, &pxe_udp->list );
return 0;
drop:
free_iob ( iobuf ); free_iob ( iobuf );
return rc; return rc;
} }
@@ -116,6 +121,7 @@ static struct pxe_udp_connection pxe_udp = {
.local = { .local = {
.sin_family = AF_INET, .sin_family = AF_INET,
}, },
.list = LIST_HEAD_INIT ( pxe_udp.list ),
}; };
/** /**
@@ -205,11 +211,20 @@ static PXENV_EXIT_t pxenv_udp_open ( struct s_PXENV_UDP_OPEN *pxenv_udp_open ) {
*/ */
static PXENV_EXIT_t static PXENV_EXIT_t
pxenv_udp_close ( struct s_PXENV_UDP_CLOSE *pxenv_udp_close ) { pxenv_udp_close ( struct s_PXENV_UDP_CLOSE *pxenv_udp_close ) {
struct io_buffer *iobuf;
struct io_buffer *tmp;
DBG ( "PXENV_UDP_CLOSE\n" ); DBG ( "PXENV_UDP_CLOSE\n" );
/* Close UDP connection */ /* Close UDP connection */
intf_restart ( &pxe_udp.xfer, 0 ); intf_restart ( &pxe_udp.xfer, 0 );
/* Discard any received packets */
list_for_each_entry_safe ( iobuf, tmp, &pxe_udp.list, list ) {
list_del ( &iobuf->list );
free_iob ( iobuf );
}
pxenv_udp_close->Status = PXENV_STATUS_SUCCESS; pxenv_udp_close->Status = PXENV_STATUS_SUCCESS;
return PXENV_EXIT_SUCCESS; return PXENV_EXIT_SUCCESS;
} }
@@ -365,20 +380,32 @@ pxenv_udp_write ( struct s_PXENV_UDP_WRITE *pxenv_udp_write ) {
static PXENV_EXIT_t pxenv_udp_read ( struct s_PXENV_UDP_READ *pxenv_udp_read ) { static PXENV_EXIT_t pxenv_udp_read ( struct s_PXENV_UDP_READ *pxenv_udp_read ) {
struct in_addr dest_ip_wanted = { .s_addr = pxenv_udp_read->dest_ip }; struct in_addr dest_ip_wanted = { .s_addr = pxenv_udp_read->dest_ip };
struct in_addr dest_ip; struct in_addr dest_ip;
struct io_buffer *iobuf;
struct pxe_udp_pseudo_header *pshdr;
uint16_t d_port_wanted = pxenv_udp_read->d_port; uint16_t d_port_wanted = pxenv_udp_read->d_port;
uint16_t d_port; uint16_t d_port;
userptr_t buffer;
size_t len;
/* Try receiving a packet */ /* Try receiving a packet, if the queue is empty */
pxe_udp.pxenv_udp_read = pxenv_udp_read; if ( list_empty ( &pxe_udp.list ) )
step(); step();
if ( pxe_udp.pxenv_udp_read ) {
/* Remove first packet from the queue */
iobuf = list_first_entry ( &pxe_udp.list, struct io_buffer, list );
if ( ! iobuf ) {
/* No packet received */ /* No packet received */
DBG2 ( "PXENV_UDP_READ\n" ); DBG2 ( "PXENV_UDP_READ\n" );
pxe_udp.pxenv_udp_read = NULL;
goto no_packet; goto no_packet;
} }
dest_ip.s_addr = pxenv_udp_read->dest_ip; list_del ( &iobuf->list );
d_port = pxenv_udp_read->d_port;
/* Strip pseudo-header */
assert ( iob_len ( iobuf ) >= sizeof ( *pshdr ) );
pshdr = iobuf->data;
iob_pull ( iobuf, sizeof ( *pshdr ) );
dest_ip.s_addr = pshdr->dest_ip;
d_port = pshdr->d_port;
DBG ( "PXENV_UDP_READ" ); DBG ( "PXENV_UDP_READ" );
/* Filter on destination address and/or port */ /* Filter on destination address and/or port */
@@ -386,14 +413,29 @@ static PXENV_EXIT_t pxenv_udp_read ( struct s_PXENV_UDP_READ *pxenv_udp_read ) {
( dest_ip_wanted.s_addr != dest_ip.s_addr ) ) { ( dest_ip_wanted.s_addr != dest_ip.s_addr ) ) {
DBG ( " wrong IP %s", inet_ntoa ( dest_ip ) ); DBG ( " wrong IP %s", inet_ntoa ( dest_ip ) );
DBG ( " (wanted %s)\n", inet_ntoa ( dest_ip_wanted ) ); DBG ( " (wanted %s)\n", inet_ntoa ( dest_ip_wanted ) );
goto no_packet; goto drop;
} }
if ( d_port_wanted && ( d_port_wanted != d_port ) ) { if ( d_port_wanted && ( d_port_wanted != d_port ) ) {
DBG ( " wrong port %d", htons ( d_port ) ); DBG ( " wrong port %d", htons ( d_port ) );
DBG ( " (wanted %d)\n", htons ( d_port_wanted ) ); DBG ( " (wanted %d)\n", htons ( d_port_wanted ) );
goto no_packet; goto drop;
} }
/* Copy packet to buffer and record length */
buffer = real_to_user ( pxenv_udp_read->buffer.segment,
pxenv_udp_read->buffer.offset );
len = iob_len ( iobuf );
if ( len > pxenv_udp_read->buffer_size )
len = pxenv_udp_read->buffer_size;
copy_to_user ( buffer, 0, iobuf->data, len );
pxenv_udp_read->buffer_size = len;
/* Fill in source/dest information */
pxenv_udp_read->src_ip = pshdr->src_ip;
pxenv_udp_read->s_port = pshdr->s_port;
pxenv_udp_read->dest_ip = pshdr->dest_ip;
pxenv_udp_read->d_port = pshdr->d_port;
DBG ( " %04x:%04x+%x %s:", pxenv_udp_read->buffer.segment, DBG ( " %04x:%04x+%x %s:", pxenv_udp_read->buffer.segment,
pxenv_udp_read->buffer.offset, pxenv_udp_read->buffer_size, pxenv_udp_read->buffer.offset, pxenv_udp_read->buffer_size,
inet_ntoa ( *( ( struct in_addr * ) &pxenv_udp_read->src_ip ) )); inet_ntoa ( *( ( struct in_addr * ) &pxenv_udp_read->src_ip ) ));
@@ -401,9 +443,14 @@ static PXENV_EXIT_t pxenv_udp_read ( struct s_PXENV_UDP_READ *pxenv_udp_read ) {
inet_ntoa ( *( ( struct in_addr * ) &pxenv_udp_read->dest_ip ) ), inet_ntoa ( *( ( struct in_addr * ) &pxenv_udp_read->dest_ip ) ),
ntohs ( pxenv_udp_read->d_port ) ); ntohs ( pxenv_udp_read->d_port ) );
/* Free I/O buffer */
free_iob ( iobuf );
pxenv_udp_read->Status = PXENV_STATUS_SUCCESS; pxenv_udp_read->Status = PXENV_STATUS_SUCCESS;
return PXENV_EXIT_SUCCESS; return PXENV_EXIT_SUCCESS;
drop:
free_iob ( iobuf );
no_packet: no_packet:
pxenv_udp_read->Status = PXENV_STATUS_FAILURE; pxenv_udp_read->Status = PXENV_STATUS_FAILURE;
return PXENV_EXIT_FAILURE; return PXENV_EXIT_FAILURE;