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