[sanboot] Extend the "keep-san" option to non-iSCSI SAN protocols
[people/peper/gpxe.git] / src / arch / i386 / interface / pcbios / aoeboot.c
1 #include <stdint.h>
2 #include <string.h>
3 #include <stdio.h>
4 #include <byteswap.h>
5 #include <gpxe/aoe.h>
6 #include <gpxe/ata.h>
7 #include <gpxe/netdevice.h>
8 #include <gpxe/settings.h>
9 #include <gpxe/sanboot.h>
10 #include <gpxe/abft.h>
11 #include <int13.h>
12
13 FILE_LICENCE ( GPL2_OR_LATER );
14
15 static int aoeboot ( const char *root_path ) {
16         struct ata_device ata;
17         struct int13_drive drive;
18         int rc;
19
20         memset ( &ata, 0, sizeof ( ata ) );
21         memset ( &drive, 0, sizeof ( drive ) );
22
23         printf ( "AoE booting from %s\n", root_path );
24
25         /* FIXME: ugly, ugly hack */
26         struct net_device *netdev = last_opened_netdev();
27
28         if ( ( rc = aoe_attach ( &ata, netdev, root_path ) ) != 0 ) {
29                 printf ( "Could not attach AoE device: %s\n",
30                          strerror ( rc ) );
31                 goto error_attach;
32         }
33         if ( ( rc = init_atadev ( &ata ) ) != 0 ) {
34                 printf ( "Could not initialise AoE device: %s\n",
35                          strerror ( rc ) );
36                 goto error_init;
37         }
38
39         /* FIXME: ugly, ugly hack */
40         struct aoe_session *aoe =
41                 container_of ( ata.backend, struct aoe_session, refcnt );
42         abft_fill_data ( aoe );
43
44         drive.blockdev = &ata.blockdev;
45
46         register_int13_drive ( &drive );
47         printf ( "Registered as BIOS drive %#02x\n", drive.drive );
48         printf ( "Booting from BIOS drive %#02x\n", drive.drive );
49         rc = int13_boot ( drive.drive );
50         printf ( "Boot failed\n" );
51
52         /* Leave drive registered, if instructed to do so */
53         if ( keep_san() )
54                 return rc;
55
56         printf ( "Unregistering BIOS drive %#02x\n", drive.drive );
57         unregister_int13_drive ( &drive );
58
59  error_init:
60         aoe_detach ( &ata );
61  error_attach:
62         return rc;
63 }
64
65 struct sanboot_protocol aoe_sanboot_protocol __sanboot_protocol = {
66         .prefix = "aoe:",
67         .boot = aoeboot,
68 };