DNS code now records a whole struct sockaddr for the name server.
[people/dverkamp/gpxe.git] / 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 sockaddr_tcpip nameserver;
39
40 /* Avoid dragging in syslog.o */
41 struct in_addr syslogserver;
42
43 /**
44  * Configure network device via DHCP
45  *
46  * @v netdev            Network device
47  * @ret rc              Return status code
48  */
49 int dhcp ( struct net_device *netdev ) {
50         static struct dhcp_option_block *dhcp_options = NULL;
51         struct dhcp_session dhcp;
52         struct in_addr address = { 0 };
53         struct in_addr netmask = { 0 };
54         struct in_addr gateway = { INADDR_NONE };
55         struct sockaddr_in *sin_nameserver;
56         struct async async;
57         int rc;
58
59         /* Check we can open the interface first */
60         if ( ( rc = ifopen ( netdev ) ) != 0 )
61                 return rc;
62
63         /* Free up any previously-acquired options */
64         if ( dhcp_options ) {
65                 unregister_dhcp_options ( dhcp_options );
66                 free_dhcp_options ( dhcp_options );
67                 dhcp_options = NULL;
68         }
69
70         /* Clear any existing routing table entry */
71         del_ipv4_address ( netdev );
72
73         /* Issue DHCP request */
74         printf ( "DHCP (%s %s)...", netdev->name, netdev_hwaddr ( netdev ) );
75         memset ( &dhcp, 0, sizeof ( dhcp ) );
76         dhcp.netdev = netdev;
77         if ( ( rc = async_block ( &async,
78                                   start_dhcp ( &dhcp, &async ) ) ) != 0 ) {
79                 printf ( "failed (%s)\n", strerror ( rc ) );
80                 return rc;
81         }
82         printf ( "done\n" );
83
84         /* Store and register options */
85         dhcp_options = dhcp.options;
86         register_dhcp_options ( dhcp_options );
87
88         /* Retrieve IP address configuration */
89         find_dhcp_ipv4_option ( dhcp_options, DHCP_EB_YIADDR, &address );
90         find_dhcp_ipv4_option ( dhcp_options, DHCP_SUBNET_MASK, &netmask );
91         find_dhcp_ipv4_option ( dhcp_options, DHCP_ROUTERS, &gateway );
92
93         /* Set up new IP address configuration */
94         if ( ( rc = add_ipv4_address ( netdev, address, netmask,
95                                        gateway ) ) != 0 ) {
96                 printf ( "Could not configure %s with DHCP results: %s\n",
97                          netdev->name, strerror ( rc ) );
98                 return rc;
99         }
100
101         /* Retrieve other DHCP options that we care about */
102         sin_nameserver = ( struct sockaddr_in * ) &nameserver;
103         sin_nameserver->sin_family = AF_INET;
104         find_dhcp_ipv4_option ( dhcp_options, DHCP_DNS_SERVERS,
105                                 &sin_nameserver->sin_addr );
106         find_dhcp_ipv4_option ( dhcp_options, DHCP_LOG_SERVERS,
107                                 &syslogserver );
108
109         return 0;
110 }