diff options
Diffstat (limited to 'roms/ipxe/src/usr/lotest.c')
| -rw-r--r-- | roms/ipxe/src/usr/lotest.c | 217 |
1 files changed, 217 insertions, 0 deletions
diff --git a/roms/ipxe/src/usr/lotest.c b/roms/ipxe/src/usr/lotest.c new file mode 100644 index 000000000..7f9f2fd07 --- /dev/null +++ b/roms/ipxe/src/usr/lotest.c @@ -0,0 +1,217 @@ +/* + * Copyright (C) 2010 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 <string.h> +#include <stdio.h> +#include <errno.h> +#include <byteswap.h> +#include <ipxe/iobuf.h> +#include <ipxe/netdevice.h> +#include <ipxe/if_ether.h> +#include <ipxe/keys.h> +#include <ipxe/console.h> +#include <usr/ifmgmt.h> +#include <usr/lotest.h> + +/** @file + * + * Loopback testing + * + */ + +#define LINK_WAIT_MS 15000 + +/** + * Process received packet + * + * @v iobuf I/O buffer + * @v netdev Network device + * @v ll_dest Link-layer destination address + * @v ll_source Link-layer source address + * @ret rc Return status code + */ +static int lotest_rx ( struct io_buffer *iobuf, + struct net_device *netdev __unused, + const void *ll_dest __unused, + const void *ll_source __unused ) { + free_iob ( iobuf ); + return -ENOTSUP; +} + +/** + * Transcribe network-layer address + * + * @v net_addr Network-layer address + * @ret string Human-readable transcription of address + */ +static const char * lotest_ntoa ( const void *net_addr __unused ) { + return "<INVALID>"; +} + +/** + * Loopback test network-layer protocol + * + * Using a dedicated network-layer protocol avoids problems caused by + * cards supporting features such as IPv4 checksum offload trying to + * interpret the (randomly generated) network-layer content. + */ +static struct net_protocol lotest_protocol __net_protocol = { + .name = "LOTEST", + .rx = lotest_rx, + .ntoa = lotest_ntoa, + .net_proto = htons ( 0x6950 ), /* Not a genuine protocol number */ + .net_addr_len = 0, +}; + +/** + * Perform loopback test between two network devices + * + * @v sender Sending network device + * @v receiver Received network device + * @v mtu Packet size (excluding link-layer headers) + * @ret rc Return status code + */ +int loopback_test ( struct net_device *sender, struct net_device *receiver, + size_t mtu ) { + uint8_t buf[mtu]; + struct io_buffer *iobuf; + const void *ll_dest; + const void *ll_source; + uint16_t net_proto; + unsigned int i; + unsigned int successes; + int rc; + + /* Open network devices */ + if ( ( rc = ifopen ( sender ) ) != 0 ) + return rc; + if ( ( rc = ifopen ( receiver ) ) != 0 ) + return rc; + + /* Wait for link-up */ + if ( ( rc = iflinkwait ( sender, LINK_WAIT_MS ) ) != 0 ) + return rc; + if ( ( rc = iflinkwait ( receiver, LINK_WAIT_MS ) ) != 0 ) + return rc; + + /* Print initial statistics */ + printf ( "Performing loopback test from %s to %s with %zd byte MTU\n", + sender->name, receiver->name, mtu ); + ifstat ( sender ); + ifstat ( receiver ); + + /* Freeze receive queue processing on the receiver, so that we + * can extract all received packets. + */ + netdev_rx_freeze ( receiver ); + + /* Perform loopback test */ + for ( successes = 0 ; ; successes++ ) { + + /* Print running total */ + printf ( "\r%d", successes ); + + /* Generate random packet */ + for ( i = 0 ; i < sizeof ( buf ) ; i++ ) + buf[i] = random(); + iobuf = alloc_iob ( MAX_LL_HEADER_LEN + sizeof ( buf ) ); + if ( ! iobuf ) { + printf ( "\nFailed to allocate I/O buffer" ); + rc = -ENOMEM; + goto done; + } + iob_reserve ( iobuf, MAX_LL_HEADER_LEN ); + memcpy ( iob_put ( iobuf, sizeof ( buf ) ), + buf, sizeof ( buf ) ); + + /* Transmit packet */ + if ( ( rc = net_tx ( iob_disown ( iobuf ), sender, + &lotest_protocol, receiver->ll_addr, + sender->ll_addr ) ) != 0 ) { + printf ( "\nFailed to transmit packet: %s", + strerror ( rc ) ); + goto done; + } + + /* Poll until packet arrives */ + do { + /* Check for cancellation */ + if ( iskey() && ( getchar() == CTRL_C ) ) { + rc = -ECANCELED; + goto done; + } + /* Poll network devices */ + net_poll(); + } while ( ( iobuf = netdev_rx_dequeue ( receiver ) ) == NULL ); + + /* Check received packet */ + if ( ( rc = receiver->ll_protocol->pull ( receiver, iobuf, + &ll_dest, &ll_source, + &net_proto ) ) != 0 ){ + printf ( "\nFailed to strip link-layer header: %s", + strerror ( rc ) ); + goto done; + } + if ( net_proto == lotest_protocol.net_proto ) { + if ( iob_len ( iobuf ) != sizeof ( buf ) ) { + printf ( "\nLength mismatch: sent %zd, " + "received %zd", + sizeof ( buf ), iob_len ( iobuf ) ); + DBG ( "\nSent:\n" ); + DBG_HDA ( 0, buf, sizeof ( buf ) ); + DBG ( "Received:\n" ); + DBG_HDA ( 0, iobuf->data, iob_len ( iobuf ) ); + rc = -EINVAL; + goto done; + } + if ( memcmp ( iobuf->data, buf, sizeof ( buf ) ) != 0){ + printf ( "\nContent mismatch" ); + DBG ( "\nSent:\n" ); + DBG_HDA ( 0, buf, sizeof ( buf ) ); + DBG ( "Received:\n" ); + DBG_HDA ( 0, iobuf->data, iob_len ( iobuf ) ); + rc = -EINVAL; + goto done; + } + } else { + printf ( "\nReceived spurious packet type %04x\n", + ntohs ( net_proto ) ); + /* Continue; this allows for the fact that + * there may have been packets outstanding on + * the wire when we started the test. + */ + } + + free_iob ( iob_disown ( iobuf ) ); + } + + done: + printf ( "\n"); + free_iob ( iobuf ); + netdev_rx_unfreeze ( receiver ); + + /* Dump final statistics */ + ifstat ( sender ); + ifstat ( receiver ); + + return 0; +} |