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