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
   7 int test_dhcp ( struct net_device *netdev ) {
   8         struct dhcp_session dhcp;
   9         struct in_addr address = { htonl ( 0 ) };
  10         struct in_addr netmask = { htonl ( 0 ) };
  11         struct in_addr gateway = { INADDR_NONE };
  12         char filename[256];
  13         int rc;
  14
  15         /* Bring IP interface up with address 0.0.0.0 */
  16         if ( ( rc = add_ipv4_address ( netdev, address, netmask,
  17                                        gateway ) ) != 0 )
  18                 goto out_no_del_ipv4;
  19
  20         /* Issue DHCP request */
  21         memset ( &dhcp, 0, sizeof ( dhcp ) );
  22         dhcp.netdev = netdev;
  23         if ( ( rc = async_wait ( start_dhcp ( &dhcp ) ) ) != 0 )
  24                 goto out_no_options;
  25
  26         /* Register options received via DHCP */
  27         register_dhcp_options ( dhcp.options );
  28
  29         /* Retrieve IP address configuration */
  30         find_global_dhcp_ipv4_option ( DHCP_EB_YIADDR, &address );
  31         find_global_dhcp_ipv4_option ( DHCP_SUBNET_MASK, &netmask );
  32         find_global_dhcp_ipv4_option ( DHCP_ROUTERS, &gateway );
  33
  34         printf ( "IP %s", inet_ntoa ( address ) );
  35         printf ( " netmask %s", inet_ntoa ( netmask ) );
  36         printf ( " gateway %s\n", inet_ntoa ( gateway ) );
  37
  38         dhcp_snprintf ( filename, sizeof ( filename ),
  39                         find_global_dhcp_option ( DHCP_BOOTFILE_NAME ) );
  40         if ( ! filename[0] ) {
  41                 printf ( "No filename specified!\n" );
  42                 goto out;
  43         }
  44
  45         printf ( "Bootfile name %s\n", filename );
  46
  47         /* Remove old IP address configuration */
  48         del_ipv4_address ( netdev );
  49
  50         /* Set up new IP address configuration */
  51         if ( ( rc = add_ipv4_address ( netdev, address, netmask,
  52                                        gateway ) ) != 0 )
  53                 goto out_no_del_ipv4;
  54
  55         /* Proof of concept: check for "aoe:" prefix and if found, do
  56          * test AoE boot with AoE options.
  57          */
  58         if ( strncmp ( filename, "aoe:", 4 ) == 0 ) {
  59                 unsigned int drivenum;
  60
  61                 drivenum = find_global_dhcp_num_option ( DHCP_EB_BIOS_DRIVE );
  62                 test_aoeboot ( netdev, &filename[4], drivenum );
  63         } else {
  64                 printf ( "Don't know how to boot %s\n", filename );
  65         }
  66
  67  out:
  68         /* Unregister and free DHCP options */
  69         unregister_dhcp_options ( dhcp.options );
  70         free_dhcp_options ( dhcp.options );
  71  out_no_options:
  72         /* Take down IP interface */
  73         del_ipv4_address ( netdev );
  74  out_no_del_ipv4:
  75         return rc;
  76 }