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