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