Print net device name when making request.
[people/xl0/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 #include <gpxe/iscsi.h>
7 #include <gpxe/netdevice.h>
8
9 static int test_dhcp_aoe_boot ( struct net_device *netdev,
10                                 char *aoename ) {
11         unsigned int drivenum;
12         
13         drivenum = find_global_dhcp_num_option ( DHCP_EB_BIOS_DRIVE );
14         return test_aoeboot ( netdev, aoename, drivenum );
15 }
16
17 static int test_dhcp_iscsi_boot ( char *iscsiname ) {
18         char *initiator_iqn = "iqn.1900-01.localdomain.localhost:initiator";
19         char *target_iqn;
20         union {
21                 struct sockaddr_in sin;
22                 struct sockaddr_tcpip st;
23         } target;
24
25         memset ( &target, 0, sizeof ( target ) );
26         target.sin.sin_family = AF_INET;
27         target.sin.sin_port = htons ( ISCSI_PORT );
28         target_iqn = strchr ( iscsiname, ':' );
29         *target_iqn++ = '\0';
30         if ( ! target_iqn ) {
31                 printf ( "Invalid iSCSI DHCP path\n" );
32                 return -EINVAL;
33         }
34         inet_aton ( iscsiname, &target.sin.sin_addr );
35
36         return test_iscsiboot ( initiator_iqn, &target.st, target_iqn );
37 }
38
39 static int test_dhcp_hello ( char *helloname ) {
40         char *message;
41         union {
42                 struct sockaddr_in sin;
43                 struct sockaddr_tcpip st;
44         } target;
45
46         memset ( &target, 0, sizeof ( target ) );
47         target.sin.sin_family = AF_INET;
48         target.sin.sin_port = htons ( 80 );
49         message = strchr ( helloname, ':' );
50         *message++ = '\0';
51         if ( ! message ) {
52                 printf ( "Invalid hello path\n" );
53                 return -EINVAL;
54         }
55         inet_aton ( helloname, &target.sin.sin_addr );  
56
57         test_hello ( &target.st, message );
58         return 0;
59 }
60
61 static int test_dhcp_http ( struct net_device *netdev, char *url ) {
62         union {
63                 struct sockaddr_in sin;
64                 struct sockaddr_tcpip st;
65         } target;
66
67         memset ( &target, 0, sizeof ( target ) );
68         target.sin.sin_family = AF_INET;
69         target.sin.sin_port = htons ( 80 );
70
71         char *addr = url + 7; // http://
72         char *file = strchr(addr, '/');
73         *file = '\0'; // for printf and inet_aton to work
74         printf("connecting to %s\n", addr);
75         inet_aton ( addr, &target.sin.sin_addr );
76         *file = '/';
77         test_http ( netdev, &target.st, file );
78         return 0;
79 }
80
81 static int test_dhcp_tftp ( struct net_device *netdev, char *tftpname ) {
82         union {
83                 struct sockaddr_in sin;
84                 struct sockaddr_tcpip st;
85         } target;
86
87         memset ( &target, 0, sizeof ( target ) );
88         target.sin.sin_family = AF_INET;
89         target.sin.sin_port = htons ( 69 );
90         find_global_dhcp_ipv4_option ( DHCP_EB_SIADDR,
91                                        &target.sin.sin_addr );
92
93         return test_tftp ( netdev, &target.st, tftpname );
94 }
95
96 static int test_dhcp_boot ( struct net_device *netdev, char *filename ) {
97         if ( strncmp ( filename, "aoe:", 4 ) == 0 ) {
98                 return test_dhcp_aoe_boot ( netdev, &filename[4] );
99         } else if ( strncmp ( filename, "iscsi:", 6 ) == 0 ) {
100                 return test_dhcp_iscsi_boot ( &filename[6] );
101         } else if ( strncmp ( filename, "hello:", 6 ) == 0 ) {
102                 return test_dhcp_hello ( &filename[6] );
103         } else if ( strncmp ( filename, "http:", 5 ) == 0 ) {
104                 return test_dhcp_http ( netdev, filename );
105         } else {
106                 return test_dhcp_tftp ( netdev, filename );
107         }
108 }
109
110 int test_dhcp ( struct net_device *netdev ) {
111         struct dhcp_session dhcp;
112         struct in_addr address = { htonl ( 0 ) };
113         struct in_addr netmask = { htonl ( 0 ) };
114         struct in_addr gateway = { INADDR_NONE };
115         char filename[256];
116         int rc;
117
118         /* Bring IP interface up with address 0.0.0.0 */
119         if ( ( rc = add_ipv4_address ( netdev, address, netmask,
120                                        gateway ) ) != 0 )
121                 goto out_no_del_ipv4;
122
123         /* Issue DHCP request */
124         printf ( "DHCP (%s)...", netdev_name ( netdev ) );
125         memset ( &dhcp, 0, sizeof ( dhcp ) );
126         dhcp.netdev = netdev;
127         if ( ( rc = async_wait ( start_dhcp ( &dhcp ) ) ) != 0 ) {
128                 printf ( "failed\n" );
129                 goto out_no_options;
130         }
131         printf ( "done\n" );
132
133         /* Register options received via DHCP */
134         register_dhcp_options ( dhcp.options );
135
136         /* Retrieve IP address configuration */
137         find_global_dhcp_ipv4_option ( DHCP_EB_YIADDR, &address );
138         find_global_dhcp_ipv4_option ( DHCP_SUBNET_MASK, &netmask );
139         find_global_dhcp_ipv4_option ( DHCP_ROUTERS, &gateway );
140
141         printf ( "IP %s", inet_ntoa ( address ) );
142         printf ( " netmask %s", inet_ntoa ( netmask ) );
143         printf ( " gateway %s\n", inet_ntoa ( gateway ) );
144
145         dhcp_snprintf ( filename, sizeof ( filename ),
146                         find_global_dhcp_option ( DHCP_BOOTFILE_NAME ) );
147         if ( ! filename[0] ) {
148                 printf ( "No filename specified!\n" );
149                 goto out;
150         }
151         
152         printf ( "Bootfile name %s\n", filename );
153
154         /* Remove old IP address configuration */
155         del_ipv4_address ( netdev );
156
157         /* Set up new IP address configuration */
158         if ( ( rc = add_ipv4_address ( netdev, address, netmask,
159                                        gateway ) ) != 0 )
160                 goto out_no_del_ipv4;
161
162         /* Test boot */
163         if ( ( rc = test_dhcp_boot ( netdev, filename ) ) != 0 ) {
164                 printf ( "Boot failed\n" );
165                 goto out;
166         }
167         
168  out:
169         /* Unregister and free DHCP options */
170         unregister_dhcp_options ( dhcp.options );
171         free_dhcp_options ( dhcp.options );
172  out_no_options:
173         /* Take down IP interface */
174         del_ipv4_address ( netdev );
175  out_no_del_ipv4:
176         return rc;
177 }