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