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