Print out "DHCP..." message just in case people think the demo is sitting
[people/dverkamp/gpxe.git] / 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         printf ( "DHCP..." );
22         memset ( &dhcp, 0, sizeof ( dhcp ) );
23         dhcp.netdev = netdev;
24         if ( ( rc = async_wait ( start_dhcp ( &dhcp ) ) ) != 0 ) {
25                 printf ( "failed\n" );
26                 goto out_no_options;
27         }
28         printf ( "done\n" );
29
30         /* Register options received via DHCP */
31         register_dhcp_options ( dhcp.options );
32
33         /* Retrieve IP address configuration */
34         find_global_dhcp_ipv4_option ( DHCP_EB_YIADDR, &address );
35         find_global_dhcp_ipv4_option ( DHCP_SUBNET_MASK, &netmask );
36         find_global_dhcp_ipv4_option ( DHCP_ROUTERS, &gateway );
37
38         printf ( "IP %s", inet_ntoa ( address ) );
39         printf ( " netmask %s", inet_ntoa ( netmask ) );
40         printf ( " gateway %s\n", inet_ntoa ( gateway ) );
41
42         dhcp_snprintf ( filename, sizeof ( filename ),
43                         find_global_dhcp_option ( DHCP_BOOTFILE_NAME ) );
44         if ( ! filename[0] ) {
45                 printf ( "No filename specified!\n" );
46                 goto out;
47         }
48         
49         printf ( "Bootfile name %s\n", filename );
50
51         /* Remove old IP address configuration */
52         del_ipv4_address ( netdev );
53
54         /* Set up new IP address configuration */
55         if ( ( rc = add_ipv4_address ( netdev, address, netmask,
56                                        gateway ) ) != 0 )
57                 goto out_no_del_ipv4;
58
59         /* Proof of concept: check for "aoe:" prefix and if found, do
60          * test AoE boot with AoE options.
61          */
62         if ( strncmp ( filename, "aoe:", 4 ) == 0 ) {
63                 unsigned int drivenum;
64                 
65                 drivenum = find_global_dhcp_num_option ( DHCP_EB_BIOS_DRIVE );
66                 test_aoeboot ( netdev, &filename[4], drivenum );
67         } else {
68                 printf ( "Don't know how to boot %s\n", filename );
69         }
70         
71  out:
72         /* Unregister and free DHCP options */
73         unregister_dhcp_options ( dhcp.options );
74         free_dhcp_options ( dhcp.options );
75  out_no_options:
76         /* Take down IP interface */
77         del_ipv4_address ( netdev );
78  out_no_del_ipv4:
79         return rc;
80 }