Add route() function to display routing table.
[people/dverkamp/gpxe.git] / src / tests / dhcptest.c
1 #include <string.h>
2 #include <stdlib.h>
3 #include <vsprintf.h>
4 #include <byteswap.h>
5 #include <gpxe/ip.h>
6 #include <gpxe/dhcp.h>
7 #include <gpxe/iscsi.h>
8 #include <gpxe/netdevice.h>
9
10 static int test_dhcp_aoe_boot ( struct net_device *netdev,
11                                 char *aoename ) {
12         unsigned int drivenum;
13         
14         drivenum = find_global_dhcp_num_option ( DHCP_EB_BIOS_DRIVE );
15         return test_aoeboot ( netdev, aoename, drivenum );
16 }
17
18 enum {
19         RP_LITERAL = 0,
20         RP_SERVERNAME,
21         RP_PROTOCOL,
22         RP_PORT,
23         RP_LUN,
24         RP_TARGETNAME,
25         NUM_RP_COMPONENTS
26 };
27
28 static int iscsi_split_root_path ( char *root_path,
29                                    char * components[NUM_RP_COMPONENTS] ) {
30         int component = 0;
31         
32         while ( 1 ) {
33                 components[component++] = root_path;
34                 if ( component == NUM_RP_COMPONENTS ) {
35                         return 0;
36                 }
37                 for ( ; *root_path != ':' ; root_path++ ) {
38                         if ( ! *root_path )
39                                 return -EINVAL;
40                 }
41                 *(root_path++) = '\0';
42         }
43 }
44
45 static int test_dhcp_iscsi_boot ( struct net_device *netdev ) {
46         char root_path[256];
47         char *rp_components[NUM_RP_COMPONENTS];
48         char initiator_iqn_buf[64];
49         char *initiator_iqn = initiator_iqn_buf;
50         char username[32];
51         char password[32];
52         union {
53                 struct sockaddr_in sin;
54                 struct sockaddr_tcpip st;
55         } target;
56         unsigned int drivenum;
57         unsigned int lun;
58         struct dhcp_option *option;
59         int rc;
60
61         memset ( &target, 0, sizeof ( target ) );
62         target.sin.sin_family = AF_INET;
63
64         dhcp_snprintf ( root_path, sizeof ( root_path ),
65                         find_global_dhcp_option ( DHCP_ROOT_PATH ) );
66
67         printf ( "Root path \"%s\"\n", root_path );
68
69         if ( ( rc = iscsi_split_root_path ( root_path, rp_components ) ) != 0 )
70                 goto bad_root_path;
71
72         if ( strcmp ( rp_components[RP_LITERAL], "iscsi" ) != 0 )
73                 goto bad_root_path;
74
75         if ( inet_aton ( rp_components[RP_SERVERNAME],
76                          &target.sin.sin_addr ) == 0 )
77                 goto bad_root_path;
78
79         target.sin.sin_port = strtoul ( rp_components[RP_PORT], NULL, 0 );
80         if ( ! target.st.st_port )
81                 target.st.st_port = htons ( ISCSI_PORT );
82
83         lun = strtoul ( rp_components[RP_LUN], NULL, 0 );
84         
85         dhcp_snprintf ( initiator_iqn_buf, sizeof ( initiator_iqn_buf ),
86                         find_global_dhcp_option ( DHCP_ISCSI_INITIATOR_IQN ) );
87         if ( ! initiator_iqn_buf[0] )
88                 initiator_iqn = "iqn.1900-01.localdomain.localhost:initiator";
89         dhcp_snprintf ( username, sizeof ( username ),
90                         find_global_dhcp_option ( DHCP_EB_USERNAME ) );
91         dhcp_snprintf ( password, sizeof ( password ),
92                         find_global_dhcp_option ( DHCP_EB_PASSWORD ) );
93
94         drivenum = find_global_dhcp_num_option ( DHCP_EB_BIOS_DRIVE );
95
96         return test_iscsiboot ( initiator_iqn, &target.st,
97                                 rp_components[RP_TARGETNAME], lun,
98                                 username, password, netdev, drivenum );
99
100  bad_root_path:
101         printf ( "Invalid iSCSI root path\n" );
102         return -EINVAL;
103 }
104
105 static int test_dhcp_hello ( char *helloname ) {
106         char *message;
107         union {
108                 struct sockaddr_in sin;
109                 struct sockaddr_tcpip st;
110         } target;
111
112         memset ( &target, 0, sizeof ( target ) );
113         target.sin.sin_family = AF_INET;
114         target.sin.sin_port = htons ( 80 );
115         message = strchr ( helloname, ':' );
116         *message++ = '\0';
117         if ( ! message ) {
118                 printf ( "Invalid hello path\n" );
119                 return -EINVAL;
120         }
121         inet_aton ( helloname, &target.sin.sin_addr );  
122
123         test_hello ( &target.st, message );
124         return 0;
125 }
126
127 static int test_dhcp_http ( struct net_device *netdev, char *url ) {
128         union {
129                 struct sockaddr_in sin;
130                 struct sockaddr_tcpip st;
131         } target;
132
133         memset ( &target, 0, sizeof ( target ) );
134         target.sin.sin_family = AF_INET;
135         target.sin.sin_port = htons ( 80 );
136
137         char *addr = url + 7; // http://
138         char *file = strchr(addr, '/');
139         *file = '\0'; // for printf and inet_aton to work
140         printf("connecting to %s\n", addr);
141         inet_aton ( addr, &target.sin.sin_addr );
142         *file = '/';
143         test_http ( netdev, &target.st, file );
144         return 0;
145 }
146
147 static int test_dhcp_ftp ( struct net_device *netdev, char *ftpname ) {
148         union {
149                 struct sockaddr_in sin;
150                 struct sockaddr_tcpip st;
151         } target;
152         char *filename;
153
154         filename = strchr ( ftpname, ':' );
155         if ( ! filename ) {
156                 printf ( "Invalid FTP path \"%s\"\n", ftpname );
157                 return -EINVAL;
158         }
159         *filename++ = '\0';
160
161         memset ( &target, 0, sizeof ( target ) );
162         target.sin.sin_family = AF_INET;
163         target.sin.sin_port = htons ( 21 );
164         inet_aton ( ftpname, &target.sin.sin_addr );
165
166         test_ftp ( &target, filename );
167         return 0;       
168 }
169
170 static int test_dhcp_tftp ( struct net_device *netdev, char *tftpname ) {
171         union {
172                 struct sockaddr_in sin;
173                 struct sockaddr_tcpip st;
174         } target;
175
176         memset ( &target, 0, sizeof ( target ) );
177         target.sin.sin_family = AF_INET;
178         target.sin.sin_port = htons ( 69 );
179         find_global_dhcp_ipv4_option ( DHCP_EB_SIADDR,
180                                        &target.sin.sin_addr );
181
182         return test_tftp ( netdev, &target.st, tftpname );
183 }
184
185 static int test_dhcp_boot ( struct net_device *netdev, char *filename ) {
186         /*
187         if ( strncmp ( filename, "aoe:", 4 ) == 0 ) {
188                 return test_dhcp_aoe_boot ( netdev, &filename[4] );
189         } 
190         */
191         //      if ( strncmp ( filename, "iscsi:", 6 ) == 0 ) {
192         if ( ! filename[0] ) {
193                 return test_dhcp_iscsi_boot ( netdev );
194         }
195         /*
196         if ( strncmp ( filename, "ftp:", 4 ) == 0 ) {
197                 return test_dhcp_ftp ( netdev, &filename[4] );
198         }
199         */
200         /*
201         if ( strncmp ( filename, "hello:", 6 ) == 0 ) {
202                 return test_dhcp_hello ( &filename[6] );
203         }
204         if ( strncmp ( filename, "http:", 5 ) == 0 ) {
205                 return test_dhcp_http ( netdev, filename );
206         }
207         */
208         return test_dhcp_tftp ( netdev, filename );
209
210         return -EPROTONOSUPPORT;
211 }
212
213 int test_dhcp ( struct net_device *netdev ) {
214         struct dhcp_session dhcp;
215         struct in_addr address = { htonl ( 0 ) };
216         struct in_addr netmask = { htonl ( 0 ) };
217         struct in_addr gateway = { INADDR_NONE };
218         char filename[256];
219         int rc;
220
221         /* Issue DHCP request */
222         printf ( "DHCP (%s)...", netdev->name );
223         memset ( &dhcp, 0, sizeof ( dhcp ) );
224         dhcp.netdev = netdev;
225         if ( ( rc = async_wait ( start_dhcp ( &dhcp ) ) ) != 0 ) {
226                 printf ( "failed\n" );
227                 goto out_no_options;
228         }
229         printf ( "done\n" );
230
231         /* Register options received via DHCP */
232         register_dhcp_options ( dhcp.options );
233
234         /* Retrieve IP address configuration */
235         find_global_dhcp_ipv4_option ( DHCP_EB_YIADDR, &address );
236         find_global_dhcp_ipv4_option ( DHCP_SUBNET_MASK, &netmask );
237         find_global_dhcp_ipv4_option ( DHCP_ROUTERS, &gateway );
238
239         dhcp_snprintf ( filename, sizeof ( filename ),
240                         find_global_dhcp_option ( DHCP_BOOTFILE_NAME ) );
241         
242         if ( filename[0] )
243                 printf ( "Bootfile name \"%s\"\n", filename );
244
245         /* Set up new IP address configuration */
246         if ( ( rc = add_ipv4_address ( netdev, address, netmask,
247                                        gateway ) ) != 0 )
248                 goto out_no_del_ipv4;
249
250         route();
251
252         /* Test boot */
253         if ( ( rc = test_dhcp_boot ( netdev, filename ) ) != 0 ) {
254                 printf ( "Boot failed\n" );
255                 goto out;
256         }
257         
258  out:
259         /* Unregister and free DHCP options */
260         unregister_dhcp_options ( dhcp.options );
261         free_dhcp_options ( dhcp.options );
262  out_no_options:
263         /* Take down IP interface */
264         del_ipv4_address ( netdev );
265  out_no_del_ipv4:
266         return rc;
267 }