[iSCSI] Support Windows Server 2008 direct iSCSI installation
[people/dverkamp/gpxe.git] / src / usr / autoboot.c
1 /*
2  * Copyright (C) 2006 Michael Brown <mbrown@fensystems.co.uk>.
3  *
4  * This program is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU General Public License as
6  * published by the Free Software Foundation; either version 2 of the
7  * License, or any later version.
8  *
9  * This program is distributed in the hope that it will be useful, but
10  * WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
12  * General Public License for more details.
13  *
14  * You should have received a copy of the GNU General Public License
15  * along with this program; if not, write to the Free Software
16  * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
17  */
18
19 #include <string.h>
20 #include <stdio.h>
21 #include <errno.h>
22 #include <gpxe/netdevice.h>
23 #include <gpxe/dhcp.h>
24 #include <gpxe/settings.h>
25 #include <gpxe/image.h>
26 #include <gpxe/embedded.h>
27 #include <usr/ifmgmt.h>
28 #include <usr/route.h>
29 #include <usr/dhcpmgmt.h>
30 #include <usr/imgmgmt.h>
31 #include <usr/iscsiboot.h>
32 #include <usr/aoeboot.h>
33 #include <usr/autoboot.h>
34
35 /** @file
36  *
37  * Automatic booting
38  *
39  */
40
41 /** Time to wait for link-up */
42 #define LINK_WAIT_MS 15000
43
44 /** Shutdown flags for exit */
45 int shutdown_exit_flags = 0;
46
47 /**
48  * Identify the boot network device
49  *
50  * @ret netdev          Boot network device
51  */
52 static struct net_device * find_boot_netdev ( void ) {
53         return NULL;
54 }
55
56 /**
57  * Boot embedded image
58  *
59  * @ret rc              Return status code
60  */
61 static int boot_embedded_image ( void ) {
62         struct image *image;
63         int rc;
64
65         image = embedded_image();
66         if ( !image )
67                 return ENOENT;
68
69         if ( ( rc = imgload ( image ) ) != 0 ) {
70                 printf ( "Could not load embedded image: %s\n",
71                          strerror ( rc ) );
72         } else if ( ( rc = imgexec ( image ) ) != 0 ) {
73                 printf ( "Could not boot embedded image: %s\n",
74                          strerror ( rc ) );
75         }
76         image_put ( image );
77         return rc;
78 }
79
80 /**
81  * Boot using filename
82  *
83  * @v filename          Boot filename
84  * @ret rc              Return status code
85  */
86 static int boot_filename ( const char *filename ) {
87         struct image *image;
88         int rc;
89
90         image = alloc_image();
91         if ( ! image ) {
92                 printf ( "Out of memory\n" );
93                 return -ENOMEM;
94         }
95         if ( ( rc = imgfetch ( image, filename,
96                                register_and_autoload_image ) ) != 0 ) {
97                 printf ( "Could not load %s: %s\n",
98                          filename, strerror ( rc ) );
99                 goto done;
100         }
101         if ( ( rc = imgexec ( image ) ) != 0 ) {
102                 printf ( "Could not boot %s: %s\n",
103                          filename, strerror ( rc ) );
104                 goto done;
105         }
106
107  done:
108         image_put ( image );
109         return rc;
110 }
111
112 /**
113  * Boot using root path
114  *
115  * @v root_path         Root path
116  * @ret rc              Return status code
117  */
118 int boot_root_path ( const char *root_path ) {
119
120         /* Quick hack */
121         if ( strncmp ( root_path, "iscsi:", 6 ) == 0 ) {
122                 return iscsiboot ( root_path );
123         } else if ( strncmp ( root_path, "aoe:", 4 ) == 0 ) {
124                 return aoeboot ( root_path );
125         }
126
127         return -ENOTSUP;
128 }
129
130 /**
131  * Boot from a network device
132  *
133  * @v netdev            Network device
134  * @ret rc              Return status code
135  */
136 static int netboot ( struct net_device *netdev ) {
137         char buf[256];
138         int rc;
139
140         /* Open device and display device status */
141         if ( ( rc = ifopen ( netdev ) ) != 0 )
142                 return rc;
143         ifstat ( netdev );
144
145         /* Wait for link-up */
146         printf ( "Waiting for link-up on %s...", netdev->name );
147         if ( ( rc = iflinkwait ( netdev, LINK_WAIT_MS ) ) != 0 ) {
148                 printf ( " no link detected\n" );
149                 return rc;
150         }
151         printf ( " ok\n" );
152
153         /* Configure device via DHCP */
154         if ( ( rc = dhcp ( netdev ) ) != 0 )
155                 return rc;
156         route();
157
158         /* Try to boot an embedded image if we have one */
159         rc = boot_embedded_image ();
160         if ( rc != ENOENT )
161                 return rc;
162
163         /* Try to download and boot whatever we are given as a filename */
164         fetch_string_setting ( NULL, &filename_setting, buf, sizeof ( buf ) );
165         if ( buf[0] ) {
166                 printf ( "Booting from filename \"%s\"\n", buf );
167                 return boot_filename ( buf );
168         }
169         
170         /* No filename; try the root path */
171         fetch_string_setting ( NULL, &root_path_setting, buf, sizeof ( buf ) );
172         if ( buf[0] ) {
173                 printf ( "Booting from root path \"%s\"\n", buf );
174                 return boot_root_path ( buf );
175         }
176
177         printf ( "No filename or root path specified\n" );
178         return -ENOENT;
179 }
180
181 /**
182  * Close all open net devices
183  *
184  * Called before a fresh boot attempt in order to free up memory.  We
185  * don't just close the device immediately after the boot fails,
186  * because there may still be TCP connections in the process of
187  * closing.
188  */
189 static void close_all_netdevs ( void ) {
190         struct net_device *netdev;
191
192         for_each_netdev ( netdev ) {
193                 ifclose ( netdev );
194         }
195 }
196
197 /**
198  * Boot the system
199  */
200 void autoboot ( void ) {
201         struct net_device *boot_netdev;
202         struct net_device *netdev;
203
204         /* If we have an identifable boot device, try that first */
205         close_all_netdevs();
206         if ( ( boot_netdev = find_boot_netdev() ) )
207                 netboot ( boot_netdev );
208
209         /* If that fails, try booting from any of the other devices */
210         for_each_netdev ( netdev ) {
211                 if ( netdev == boot_netdev )
212                         continue;
213                 close_all_netdevs();
214                 netboot ( netdev );
215         }
216
217         printf ( "No more network devices\n" );
218 }