src/usr/dhcpmgmt.c

   1 /*
   2  * Copyright (C) 2007 Michael Brown <mbrown@fensystems.co.uk>.
   3  *
   4  * This program is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU General Public License as
   6  * published by the Free Software Foundation; either version 2 of the
   7  * License, or any later version.
   8  *
   9  * This program is distributed in the hope that it will be useful, but
  10  * WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU General Public License
  15  * along with this program; if not, write to the Free Software
  16  * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  17  */
  18
  19 #include <string.h>
  20 #include <stdio.h>
  21 #include <byteswap.h>
  22 #include <gpxe/in.h>
  23 #include <gpxe/ip.h>
  24 #include <gpxe/dhcp.h>
  25 #include <gpxe/async.h>
  26 #include <gpxe/netdevice.h>
  27 #include <gpxe/dns.h>
  28 #include <usr/ifmgmt.h>
  29 #include <usr/dhcpmgmt.h>
  30
  31 /** @file
  32  *
  33  * DHCP management
  34  *
  35  */
  36
  37 /* Avoid dragging in dns.o */
  38 struct in_addr nameserver;
  39
  40 /**
  41  * Configure network device via DHCP
  42  *
  43  * @v netdev            Network device
  44  * @ret rc              Return status code
  45  */
  46 int dhcp ( struct net_device *netdev ) {
  47         static struct dhcp_option_block *dhcp_options = NULL;
  48         struct dhcp_session dhcp;
  49         struct in_addr address = { 0 };
  50         struct in_addr netmask = { 0 };
  51         struct in_addr gateway = { INADDR_NONE };
  52         struct async async;
  53         int rc;
  54
  55         /* Check we can open the interface first */
  56         if ( ( rc = ifopen ( netdev ) ) != 0 )
  57                 return rc;
  58
  59         /* Free up any previously-acquired options */
  60         if ( dhcp_options ) {
  61                 unregister_dhcp_options ( dhcp_options );
  62                 free_dhcp_options ( dhcp_options );
  63                 dhcp_options = NULL;
  64         }
  65
  66         /* Clear any existing routing table entry */
  67         del_ipv4_address ( netdev );
  68
  69         /* Issue DHCP request */
  70         printf ( "DHCP (%s %s)...", netdev->name, netdev_hwaddr ( netdev ) );
  71         memset ( &dhcp, 0, sizeof ( dhcp ) );
  72         dhcp.netdev = netdev;
  73         if ( ( rc = async_block ( &async,
  74                                   start_dhcp ( &dhcp, &async ) ) ) != 0 ) {
  75                 printf ( "failed (%s)\n", strerror ( rc ) );
  76                 return rc;
  77         }
  78         printf ( "done\n" );
  79
  80         /* Store and register options */
  81         dhcp_options = dhcp.options;
  82         register_dhcp_options ( dhcp_options );
  83
  84         /* Retrieve IP address configuration */
  85         find_dhcp_ipv4_option ( dhcp_options, DHCP_EB_YIADDR, &address );
  86         find_dhcp_ipv4_option ( dhcp_options, DHCP_SUBNET_MASK, &netmask );
  87         find_dhcp_ipv4_option ( dhcp_options, DHCP_ROUTERS, &gateway );
  88
  89         /* Set up new IP address configuration */
  90         if ( ( rc = add_ipv4_address ( netdev, address, netmask,
  91                                        gateway ) ) != 0 ) {
  92                 printf ( "Could not configure %s with DHCP results: %s\n",
  93                          netdev->name, strerror ( rc ) );
  94                 return rc;
  95         }
  96
  97         /* Retrieve other DHCP options that we care about */
  98         find_dhcp_ipv4_option ( dhcp_options, DHCP_DNS_SERVERS,
  99                                 &nameserver );
 100
 101         return 0;
 102 }