sysadmin/ipxe
1
0
Fork 0
mirror of https://github.com/ipxe/ipxe synced 2025-12-17 01:52:08 +03:00
Code Issues Packages Projects Releases Wiki Activity

Added TFTP test code (currently just dumps file to console).

Browse Source
This commit is contained in:
Michael Brown
2006-08-09 00:09:29 +00:00
parent 3611cb17b7
commit d1a123b1f4
4 changed files with 67 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
src/tests/dhcptest.c
View File

@@ -57,6 +57,21 @@ static int test_dhcp_hello ( char *helloname ) {
return 0;
}
static int test_dhcp_tftp ( char *tftpname ) {
union {
struct sockaddr_in sin;
struct sockaddr_tcpip st;
} target;
memset ( &target, 0, sizeof ( target ) );
target.sin.sin_family = AF_INET;
target.sin.sin_port = htons ( 69 );
find_global_dhcp_ipv4_option ( DHCP_EB_SIADDR,
&target.sin.sin_addr );
return test_tftp ( &target.st, tftpname );
}
static int test_dhcp_boot ( struct net_device *netdev, char *filename ) {
if ( strncmp ( filename, "aoe:", 4 ) == 0 ) {
return test_dhcp_aoe_boot ( netdev, &filename[4] );
@@ -65,8 +80,7 @@ static int test_dhcp_boot ( struct net_device *netdev, char *filename ) {
} else if ( strncmp ( filename, "hello:", 6 ) == 0 ) {
return test_dhcp_hello ( &filename[6] );
} else {
printf ( "Don't know how to boot %s\n", filename );
return -EPROTONOSUPPORT;
return test_dhcp_tftp ( filename );
}
}

35
src/tests/tftptest.c Normal file
View File

@@ -0,0 +1,35 @@
#include <stdint.h>
#include <string.h>
#include <console.h>
#include <gpxe/udp.h>
#include <gpxe/tftp.h>
#include <gpxe/async.h>
static void test_tftp_callback ( struct tftp_session *tftp __unused,
unsigned int block __unused,
void *data, size_t len ) {
unsigned int i;
char c;
for ( i = 0 ; i < len ; i++ ) {
c = * ( ( char * ) data + i );
if ( c == '\r' ) {
/* Print nothing */
} else if ( ( c == '\n' ) || ( c >= 32 ) || ( c <= 126 ) ) {
putchar ( c );
} else {
putchar ( '.' );
}
}
}
int test_tftp ( struct sockaddr_tcpip *target, const char *filename ) {
struct tftp_session tftp;
memset ( &tftp, 0, sizeof ( tftp ) );
udp_connect ( &tftp.udp, target );
tftp.filename = filename;
tftp.callback = test_tftp_callback;
return async_wait ( tftp_get ( &tftp ) );
}