src/tests/dhcptest.c

   1 #include <string.h>
   2 #include <vsprintf.h>
   3 #include <byteswap.h>
   4 #include <gpxe/ip.h>
   5 #include <gpxe/dhcp.h>
   6 #include <gpxe/iscsi.h>
   7 #include <gpxe/netdevice.h>
   8
   9 static int test_dhcp_aoe_boot ( struct net_device *netdev,
  10                                 char *aoename ) {
  11         unsigned int drivenum;
  12
  13         drivenum = find_global_dhcp_num_option ( DHCP_EB_BIOS_DRIVE );
  14         return test_aoeboot ( netdev, aoename, drivenum );
  15 }
  16
  17 static int test_dhcp_iscsi_boot ( char *iscsiname ) {
  18         char *initiator_iqn = "iqn.1900-01.localdomain.localhost:initiator";
  19         char *target_iqn;
  20         union {
  21                 struct sockaddr_in sin;
  22                 struct sockaddr_tcpip st;
  23         } target;
  24
  25         memset ( &target, 0, sizeof ( target ) );
  26         target.sin.sin_family = AF_INET;
  27         target.sin.sin_port = htons ( ISCSI_PORT );
  28         target_iqn = strchr ( iscsiname, ':' );
  29         *target_iqn++ = '\0';
  30         if ( ! target_iqn ) {
  31                 printf ( "Invalid iSCSI DHCP path\n" );
  32                 return -EINVAL;
  33         }
  34         inet_aton ( iscsiname, &target.sin.sin_addr );
  35
  36         return test_iscsiboot ( initiator_iqn, &target.st, target_iqn );
  37 }
  38
  39 static int test_dhcp_hello ( char *helloname ) {
  40         char *message;
  41         union {
  42                 struct sockaddr_in sin;
  43                 struct sockaddr_tcpip st;
  44         } target;
  45
  46         memset ( &target, 0, sizeof ( target ) );
  47         target.sin.sin_family = AF_INET;
  48         target.sin.sin_port = htons ( 80 );
  49         message = strchr ( helloname, ':' );
  50         *message++ = '\0';
  51         if ( ! message ) {
  52                 printf ( "Invalid hello path\n" );
  53                 return -EINVAL;
  54         }
  55         inet_aton ( helloname, &target.sin.sin_addr );
  56
  57         test_hello ( &target.st, message );
  58         return 0;
  59 }
  60
  61 static int test_dhcp_http ( struct net_device *netdev, char *url ) {
  62         union {
  63                 struct sockaddr_in sin;
  64                 struct sockaddr_tcpip st;
  65         } target;
  66
  67         memset ( &target, 0, sizeof ( target ) );
  68         target.sin.sin_family = AF_INET;
  69         target.sin.sin_port = htons ( 80 );
  70
  71         char *addr = url + 7; // http://
  72         char *file = strchr(addr, '/');
  73         *file = '\0'; // for printf and inet_aton to work
  74         printf("connecting to %s\n", addr);
  75         inet_aton ( addr, &target.sin.sin_addr );
  76         *file = '/';
  77         test_http ( netdev, &target.st, file );
  78         return 0;
  79 }
  80
  81 static int test_dhcp_tftp ( struct net_device *netdev, char *tftpname ) {
  82         union {
  83                 struct sockaddr_in sin;
  84                 struct sockaddr_tcpip st;
  85         } target;
  86
  87         memset ( &target, 0, sizeof ( target ) );
  88         target.sin.sin_family = AF_INET;
  89         target.sin.sin_port = htons ( 69 );
  90         find_global_dhcp_ipv4_option ( DHCP_EB_SIADDR,
  91                                        &target.sin.sin_addr );
  92
  93         return test_tftp ( netdev, &target.st, tftpname );
  94 }
  95
  96 static int test_dhcp_boot ( struct net_device *netdev, char *filename ) {
  97         if ( strncmp ( filename, "aoe:", 4 ) == 0 ) {
  98                 return test_dhcp_aoe_boot ( netdev, &filename[4] );
  99         } else if ( strncmp ( filename, "iscsi:", 6 ) == 0 ) {
 100                 return test_dhcp_iscsi_boot ( &filename[6] );
 101         } else if ( strncmp ( filename, "hello:", 6 ) == 0 ) {
 102                 return test_dhcp_hello ( &filename[6] );
 103         } else if ( strncmp ( filename, "http:", 5 ) == 0 ) {
 104                 return test_dhcp_http ( netdev, filename );
 105         } else {
 106                 return test_dhcp_tftp ( netdev, filename );
 107         }
 108 }
 109
 110 int test_dhcp ( struct net_device *netdev ) {
 111         struct dhcp_session dhcp;
 112         struct in_addr address = { htonl ( 0 ) };
 113         struct in_addr netmask = { htonl ( 0 ) };
 114         struct in_addr gateway = { INADDR_NONE };
 115         char filename[256];
 116         int rc;
 117
 118         /* Bring IP interface up with address 0.0.0.0 */
 119         if ( ( rc = add_ipv4_address ( netdev, address, netmask,
 120                                        gateway ) ) != 0 )
 121                 goto out_no_del_ipv4;
 122
 123         /* Issue DHCP request */
 124         printf ( "DHCP (%s)...", netdev_name ( netdev ) );
 125         memset ( &dhcp, 0, sizeof ( dhcp ) );
 126         dhcp.netdev = netdev;
 127         if ( ( rc = async_wait ( start_dhcp ( &dhcp ) ) ) != 0 ) {
 128                 printf ( "failed\n" );
 129                 goto out_no_options;
 130         }
 131         printf ( "done\n" );
 132
 133         /* Register options received via DHCP */
 134         register_dhcp_options ( dhcp.options );
 135
 136         /* Retrieve IP address configuration */
 137         find_global_dhcp_ipv4_option ( DHCP_EB_YIADDR, &address );
 138         find_global_dhcp_ipv4_option ( DHCP_SUBNET_MASK, &netmask );
 139         find_global_dhcp_ipv4_option ( DHCP_ROUTERS, &gateway );
 140
 141         printf ( "IP %s", inet_ntoa ( address ) );
 142         printf ( " netmask %s", inet_ntoa ( netmask ) );
 143         printf ( " gateway %s\n", inet_ntoa ( gateway ) );
 144
 145         dhcp_snprintf ( filename, sizeof ( filename ),
 146                         find_global_dhcp_option ( DHCP_BOOTFILE_NAME ) );
 147         if ( ! filename[0] ) {
 148                 printf ( "No filename specified!\n" );
 149                 goto out;
 150         }
 151
 152         printf ( "Bootfile name %s\n", filename );
 153
 154         /* Remove old IP address configuration */
 155         del_ipv4_address ( netdev );
 156
 157         /* Set up new IP address configuration */
 158         if ( ( rc = add_ipv4_address ( netdev, address, netmask,
 159                                        gateway ) ) != 0 )
 160                 goto out_no_del_ipv4;
 161
 162         /* Test boot */
 163         if ( ( rc = test_dhcp_boot ( netdev, filename ) ) != 0 ) {
 164                 printf ( "Boot failed\n" );
 165                 goto out;
 166         }
 167
 168  out:
 169         /* Unregister and free DHCP options */
 170         unregister_dhcp_options ( dhcp.options );
 171         free_dhcp_options ( dhcp.options );
 172  out_no_options:
 173         /* Take down IP interface */
 174         del_ipv4_address ( netdev );
 175  out_no_del_ipv4:
 176         return rc;
 177 }