f200c16fda3a8047021b588508bcbe18dc960071
[people/pcmattman/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/settings.h>
8 #include <gpxe/dhcp.h>
9 #include <gpxe/netdevice.h>
10 #include <gpxe/ibft.h>
11 #include <gpxe/init.h>
12 #include <gpxe/sanboot.h>
13 #include <int13.h>
14 #include <usr/autoboot.h>
15
16 FILE_LICENCE ( GPL2_OR_LATER );
17
18 struct setting keep_san_setting __setting = {
19         .name = "keep-san",
20         .description = "Preserve SAN connection",
21         .tag = DHCP_EB_KEEP_SAN,
22         .type = &setting_type_int8,
23 };
24
25 static int iscsiboot ( const char *root_path ) {
26         struct scsi_device *scsi;
27         struct int13_drive *drive;
28         int keep_san;
29         int rc;
30
31         scsi = zalloc ( sizeof ( *scsi ) );
32         if ( ! scsi ) {
33                 rc = -ENOMEM;
34                 goto err_alloc_scsi;
35         }
36         drive = zalloc ( sizeof ( *drive ) );
37         if ( ! drive ) {
38                 rc = -ENOMEM;
39                 goto err_alloc_drive;
40         }
41
42         printf ( "iSCSI booting from %s\n", root_path );
43
44         if ( ( rc = iscsi_attach ( scsi, root_path ) ) != 0 ) {
45                 printf ( "Could not attach iSCSI device: %s\n",
46                          strerror ( rc ) );
47                 goto err_attach;
48         }
49         if ( ( rc = init_scsidev ( scsi ) ) != 0 ) {
50                 printf ( "Could not initialise iSCSI device: %s\n",
51                          strerror ( rc ) );
52                 goto err_init;
53         }
54
55         drive->blockdev = &scsi->blockdev;
56
57         /* FIXME: ugly, ugly hack */
58         struct net_device *netdev = last_opened_netdev();
59         struct iscsi_session *iscsi =
60                 container_of ( scsi->backend, struct iscsi_session, refcnt );
61         ibft_fill_data ( netdev, iscsi );
62
63         register_int13_drive ( drive );
64         printf ( "Registered as BIOS drive %#02x\n", drive->drive );
65         printf ( "Booting from BIOS drive %#02x\n", drive->drive );
66         rc = int13_boot ( drive->drive );
67         printf ( "Boot failed\n" );
68
69         /* Leave drive registered, if instructed to do so */
70         keep_san = fetch_intz_setting ( NULL, &keep_san_setting );
71         if ( keep_san ) {
72                 printf ( "Preserving connection to SAN disk\n" );
73                 shutdown_exit_flags |= SHUTDOWN_KEEP_DEVICES;
74                 return rc;
75         }
76
77         printf ( "Unregistering BIOS drive %#02x\n", drive->drive );
78         unregister_int13_drive ( drive );
79
80  err_init:
81         iscsi_detach ( scsi );
82  err_attach:
83         free ( drive );
84  err_alloc_drive:
85         free ( scsi );
86  err_alloc_scsi:
87         return rc;
88 }
89
90 struct sanboot_protocol iscsi_sanboot_protocol __sanboot_protocol = {
91         .prefix = "iscsi:",
92         .boot = iscsiboot,
93 };