[sanboot] Extend the "keep-san" option to non-iSCSI SAN protocols
[people/peper/gpxe.git] / src / arch / i386 / interface / pcbios / iscsiboot.c
1 #include <stdint.h>
2 #include <string.h>
3 #include <stdlib.h>
4 #include <stdio.h>
5 #include <errno.h>
6 #include <gpxe/iscsi.h>
7 #include <gpxe/netdevice.h>
8 #include <gpxe/ibft.h>
9 #include <gpxe/sanboot.h>
10 #include <int13.h>
11
12 FILE_LICENCE ( GPL2_OR_LATER );
13
14 static int iscsiboot ( const char *root_path ) {
15         struct scsi_device *scsi;
16         struct int13_drive *drive;
17         int rc;
18
19         scsi = zalloc ( sizeof ( *scsi ) );
20         if ( ! scsi ) {
21                 rc = -ENOMEM;
22                 goto err_alloc_scsi;
23         }
24         drive = zalloc ( sizeof ( *drive ) );
25         if ( ! drive ) {
26                 rc = -ENOMEM;
27                 goto err_alloc_drive;
28         }
29
30         printf ( "iSCSI booting from %s\n", root_path );
31
32         if ( ( rc = iscsi_attach ( scsi, root_path ) ) != 0 ) {
33                 printf ( "Could not attach iSCSI device: %s\n",
34                          strerror ( rc ) );
35                 goto err_attach;
36         }
37         if ( ( rc = init_scsidev ( scsi ) ) != 0 ) {
38                 printf ( "Could not initialise iSCSI device: %s\n",
39                          strerror ( rc ) );
40                 goto err_init;
41         }
42
43         drive->blockdev = &scsi->blockdev;
44
45         /* FIXME: ugly, ugly hack */
46         struct net_device *netdev = last_opened_netdev();
47         struct iscsi_session *iscsi =
48                 container_of ( scsi->backend, struct iscsi_session, refcnt );
49         ibft_fill_data ( netdev, iscsi );
50
51         register_int13_drive ( drive );
52         printf ( "Registered as BIOS drive %#02x\n", drive->drive );
53         printf ( "Booting from BIOS drive %#02x\n", drive->drive );
54         rc = int13_boot ( drive->drive );
55         printf ( "Boot failed\n" );
56
57         /* Leave drive registered, if instructed to do so */
58         if ( keep_san() )
59                 return rc;
60
61         printf ( "Unregistering BIOS drive %#02x\n", drive->drive );
62         unregister_int13_drive ( drive );
63
64  err_init:
65         iscsi_detach ( scsi );
66  err_attach:
67         free ( drive );
68  err_alloc_drive:
69         free ( scsi );
70  err_alloc_scsi:
71         return rc;
72 }
73
74 struct sanboot_protocol iscsi_sanboot_protocol __sanboot_protocol = {
75         .prefix = "iscsi:",
76         .boot = iscsiboot,
77 };