src/tests/dhcptest.c

   1 #include <string.h>
   2 #include <stdlib.h>
   3 #include <vsprintf.h>
   4 #include <byteswap.h>
   5 #include <gpxe/ip.h>
   6 #include <gpxe/dhcp.h>
   7 #include <gpxe/iscsi.h>
   8 #include <gpxe/netdevice.h>
   9
  10 static int test_dhcp_aoe_boot ( struct net_device *netdev,
  11                                 char *aoename ) {
  12         unsigned int drivenum;
  13
  14         drivenum = find_global_dhcp_num_option ( DHCP_EB_BIOS_DRIVE );
  15         return test_aoeboot ( netdev, aoename, drivenum );
  16 }
  17
  18 enum {
  19         RP_LITERAL = 0,
  20         RP_SERVERNAME,
  21         RP_PROTOCOL,
  22         RP_PORT,
  23         RP_LUN,
  24         RP_TARGETNAME,
  25         NUM_RP_COMPONENTS
  26 };
  27
  28 static int iscsi_split_root_path ( char *root_path,
  29                                    char * components[NUM_RP_COMPONENTS] ) {
  30         int component = 0;
  31
  32         while ( 1 ) {
  33                 components[component++] = root_path;
  34                 if ( component == NUM_RP_COMPONENTS ) {
  35                         return 0;
  36                 }
  37                 for ( ; *root_path != ':' ; root_path++ ) {
  38                         if ( ! *root_path )
  39                                 return -EINVAL;
  40                 }
  41                 *(root_path++) = '\0';
  42         }
  43 }
  44
  45 static int test_dhcp_iscsi_boot ( struct net_device *netdev ) {
  46         char root_path[256];
  47         char *rp_components[NUM_RP_COMPONENTS];
  48         char initiator_iqn_buf[64];
  49         char *initiator_iqn = initiator_iqn_buf;
  50         char username[32];
  51         char password[32];
  52         union {
  53                 struct sockaddr_in sin;
  54                 struct sockaddr_tcpip st;
  55         } target;
  56         unsigned int drivenum;
  57         unsigned int lun;
  58         struct dhcp_option *option;
  59         int rc;
  60
  61         memset ( &target, 0, sizeof ( target ) );
  62         target.sin.sin_family = AF_INET;
  63
  64         dhcp_snprintf ( root_path, sizeof ( root_path ),
  65                         find_global_dhcp_option ( DHCP_ROOT_PATH ) );
  66
  67         printf ( "Root path \"%s\"\n", root_path );
  68
  69         if ( ( rc = iscsi_split_root_path ( root_path, rp_components ) ) != 0 )
  70                 goto bad_root_path;
  71
  72         if ( strcmp ( rp_components[RP_LITERAL], "iscsi" ) != 0 )
  73                 goto bad_root_path;
  74
  75         if ( inet_aton ( rp_components[RP_SERVERNAME],
  76                          &target.sin.sin_addr ) == 0 )
  77                 goto bad_root_path;
  78
  79         target.sin.sin_port = strtoul ( rp_components[RP_PORT], NULL, 0 );
  80         if ( ! target.st.st_port )
  81                 target.st.st_port = htons ( ISCSI_PORT );
  82
  83         lun = strtoul ( rp_components[RP_LUN], NULL, 0 );
  84
  85         dhcp_snprintf ( initiator_iqn_buf, sizeof ( initiator_iqn_buf ),
  86                         find_global_dhcp_option ( DHCP_ISCSI_INITIATOR_IQN ) );
  87         if ( ! initiator_iqn_buf[0] )
  88                 initiator_iqn = "iqn.1900-01.localdomain.localhost:initiator";
  89         dhcp_snprintf ( username, sizeof ( username ),
  90                         find_global_dhcp_option ( DHCP_EB_USERNAME ) );
  91         dhcp_snprintf ( password, sizeof ( password ),
  92                         find_global_dhcp_option ( DHCP_EB_PASSWORD ) );
  93
  94         drivenum = find_global_dhcp_num_option ( DHCP_EB_BIOS_DRIVE );
  95
  96         return test_iscsiboot ( initiator_iqn, &target.st,
  97                                 rp_components[RP_TARGETNAME], lun,
  98                                 username, password, netdev, drivenum );
  99
 100  bad_root_path:
 101         printf ( "Invalid iSCSI root path\n" );
 102         return -EINVAL;
 103 }
 104
 105 static int test_dhcp_hello ( char *helloname ) {
 106         char *message;
 107         union {
 108                 struct sockaddr_in sin;
 109                 struct sockaddr_tcpip st;
 110         } target;
 111
 112         memset ( &target, 0, sizeof ( target ) );
 113         target.sin.sin_family = AF_INET;
 114         target.sin.sin_port = htons ( 80 );
 115         message = strchr ( helloname, ':' );
 116         *message++ = '\0';
 117         if ( ! message ) {
 118                 printf ( "Invalid hello path\n" );
 119                 return -EINVAL;
 120         }
 121         inet_aton ( helloname, &target.sin.sin_addr );
 122
 123         test_hello ( &target.st, message );
 124         return 0;
 125 }
 126
 127 static int test_dhcp_http ( struct net_device *netdev, char *url ) {
 128         union {
 129                 struct sockaddr_in sin;
 130                 struct sockaddr_tcpip st;
 131         } target;
 132
 133         memset ( &target, 0, sizeof ( target ) );
 134         target.sin.sin_family = AF_INET;
 135         target.sin.sin_port = htons ( 80 );
 136
 137         char *addr = url + 7; // http://
 138         char *file = strchr(addr, '/');
 139         *file = '\0'; // for printf and inet_aton to work
 140         printf("connecting to %s\n", addr);
 141         inet_aton ( addr, &target.sin.sin_addr );
 142         *file = '/';
 143         test_http ( netdev, &target.st, file );
 144         return 0;
 145 }
 146
 147 static int test_dhcp_ftp ( struct net_device *netdev, char *ftpname ) {
 148         union {
 149                 struct sockaddr_in sin;
 150                 struct sockaddr_tcpip st;
 151         } target;
 152         char *filename;
 153
 154         filename = strchr ( ftpname, ':' );
 155         if ( ! filename ) {
 156                 printf ( "Invalid FTP path \"%s\"\n", ftpname );
 157                 return -EINVAL;
 158         }
 159         *filename++ = '\0';
 160
 161         memset ( &target, 0, sizeof ( target ) );
 162         target.sin.sin_family = AF_INET;
 163         target.sin.sin_port = htons ( 21 );
 164         inet_aton ( ftpname, &target.sin.sin_addr );
 165
 166         test_ftp ( &target, filename );
 167         return 0;
 168 }
 169
 170 static int test_dhcp_tftp ( struct net_device *netdev, char *tftpname ) {
 171         union {
 172                 struct sockaddr_in sin;
 173                 struct sockaddr_tcpip st;
 174         } target;
 175
 176         memset ( &target, 0, sizeof ( target ) );
 177         target.sin.sin_family = AF_INET;
 178         target.sin.sin_port = htons ( 69 );
 179         find_global_dhcp_ipv4_option ( DHCP_EB_SIADDR,
 180                                        &target.sin.sin_addr );
 181
 182         return test_tftp ( netdev, &target.st, tftpname );
 183 }
 184
 185 static int test_dhcp_boot ( struct net_device *netdev, char *filename ) {
 186         /*
 187         if ( strncmp ( filename, "aoe:", 4 ) == 0 ) {
 188                 return test_dhcp_aoe_boot ( netdev, &filename[4] );
 189         }
 190         */
 191         //      if ( strncmp ( filename, "iscsi:", 6 ) == 0 ) {
 192         if ( ! filename[0] ) {
 193                 return test_dhcp_iscsi_boot ( netdev );
 194         }
 195         /*
 196         if ( strncmp ( filename, "ftp:", 4 ) == 0 ) {
 197                 return test_dhcp_ftp ( netdev, &filename[4] );
 198         }
 199         */
 200         /*
 201         if ( strncmp ( filename, "hello:", 6 ) == 0 ) {
 202                 return test_dhcp_hello ( &filename[6] );
 203         }
 204         if ( strncmp ( filename, "http:", 5 ) == 0 ) {
 205                 return test_dhcp_http ( netdev, filename );
 206         }
 207         */
 208         return test_dhcp_tftp ( netdev, filename );
 209
 210         return -EPROTONOSUPPORT;
 211 }
 212
 213 int test_dhcp ( struct net_device *netdev ) {
 214         struct dhcp_session dhcp;
 215         struct in_addr address = { htonl ( 0 ) };
 216         struct in_addr netmask = { htonl ( 0 ) };
 217         struct in_addr gateway = { INADDR_NONE };
 218         char filename[256];
 219         int rc;
 220
 221         /* Bring IP interface up with address 0.0.0.0 */
 222         if ( ( rc = add_ipv4_address ( netdev, address, netmask,
 223                                        gateway ) ) != 0 )
 224                 goto out_no_del_ipv4;
 225
 226         /* Issue DHCP request */
 227         printf ( "DHCP (%s)...", netdev->name );
 228         memset ( &dhcp, 0, sizeof ( dhcp ) );
 229         dhcp.netdev = netdev;
 230         if ( ( rc = async_wait ( start_dhcp ( &dhcp ) ) ) != 0 ) {
 231                 printf ( "failed\n" );
 232                 goto out_no_options;
 233         }
 234         printf ( "done\n" );
 235
 236         /* Register options received via DHCP */
 237         register_dhcp_options ( dhcp.options );
 238
 239         /* Retrieve IP address configuration */
 240         find_global_dhcp_ipv4_option ( DHCP_EB_YIADDR, &address );
 241         find_global_dhcp_ipv4_option ( DHCP_SUBNET_MASK, &netmask );
 242         find_global_dhcp_ipv4_option ( DHCP_ROUTERS, &gateway );
 243
 244         printf ( "IP %s", inet_ntoa ( address ) );
 245         printf ( " netmask %s", inet_ntoa ( netmask ) );
 246         printf ( " gateway %s\n", inet_ntoa ( gateway ) );
 247
 248         dhcp_snprintf ( filename, sizeof ( filename ),
 249                         find_global_dhcp_option ( DHCP_BOOTFILE_NAME ) );
 250
 251         if ( filename[0] )
 252                 printf ( "Bootfile name \"%s\"\n", filename );
 253
 254         /* Remove old IP address configuration */
 255         del_ipv4_address ( netdev );
 256
 257         /* Set up new IP address configuration */
 258         if ( ( rc = add_ipv4_address ( netdev, address, netmask,
 259                                        gateway ) ) != 0 )
 260                 goto out_no_del_ipv4;
 261
 262         /* Test boot */
 263         if ( ( rc = test_dhcp_boot ( netdev, filename ) ) != 0 ) {
 264                 printf ( "Boot failed\n" );
 265                 goto out;
 266         }
 267
 268  out:
 269         /* Unregister and free DHCP options */
 270         unregister_dhcp_options ( dhcp.options );
 271         free_dhcp_options ( dhcp.options );
 272  out_no_options:
 273         /* Take down IP interface */
 274         del_ipv4_address ( netdev );
 275  out_no_del_ipv4:
 276         return rc;
 277 }