Add INT 13,41 (extensions installation check). LILO's MBR now uses
[people/xl0/gpxe-arm.git] / src / arch / i386 / interface / pcbios / int13.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 <stdint.h>
20 #include <limits.h>
21 #include <byteswap.h>
22 #include <errno.h>
23 #include <assert.h>
24 #include <gpxe/list.h>
25 #include <gpxe/blockdev.h>
26 #include <realmode.h>
27 #include <bios.h>
28 #include <biosint.h>
29 #include <int13.h>
30
31 /** @file
32  *
33  * INT 13 emulation
34  *
35  * This module provides a mechanism for exporting block devices via
36  * the BIOS INT 13 disk interrupt interface.  
37  *
38  */
39
40 /** Vector for chaining to other INT 13 handlers */
41 static struct segoff __text16 ( int13_vector );
42 #define int13_vector __use_text16 ( int13_vector )
43
44 /** Assembly wrapper */
45 extern void int13_wrapper ( void );
46
47 /** Vector for storing original INT 18 handler
48  *
49  * We do not chain to this vector, so there is no need to place it in
50  * .text16.
51  */
52 static struct segoff int18_vector;
53
54 /** Vector for storing original INT 19 handler
55  *
56  * We do not chain to this vector, so there is no need to place it in
57  * .text16.
58  */
59 static struct segoff int19_vector;
60
61 /** Restart point for INT 18 or 19 */
62 extern void int13_exec_fail ( void );
63
64 /** List of registered emulated drives */
65 static LIST_HEAD ( drives );
66
67 /**
68  * Convert CHS address to linear address
69  *
70  * @v drive             Emulated drive
71  * @v ch                Low bits of cylinder number
72  * @v cl (bits 7:6)     High bits of cylinder number
73  * @v cl (bits 5:0)     Sector number
74  * @v dh                Head number
75  * @ret lba             LBA address
76  * 
77  */
78 static unsigned long chs_to_lba ( struct int13_drive *drive,
79                                   struct i386_all_regs *ix86 ) {
80         unsigned int cylinder;
81         unsigned int head;
82         unsigned int sector;
83         unsigned long lba;
84
85         cylinder = ( ( ( ix86->regs.cl & 0xc0 ) << 8 ) | ix86->regs.ch );
86         head = ix86->regs.dh;
87         sector = ( ix86->regs.cl & 0x3f );
88
89         assert ( cylinder < drive->cylinders );
90         assert ( head < drive->heads );
91         assert ( ( sector >= 1 ) && ( sector <= drive->sectors_per_track ) );
92
93         lba = ( ( ( ( cylinder * drive->heads ) + head )
94                   * drive->sectors_per_track ) + sector - 1 );
95
96         DBG ( "C/H/S address %x/%x/%x -> LBA %x\n",
97               cylinder, head, sector, lba );
98
99         return lba;
100 }
101
102 /**
103  * Read from drive to real-mode data buffer
104  *
105  * @v drive             Emulated drive
106  * @v lba               LBA starting sector number
107  * @v data              Data buffer
108  * @v count             Number of sectors to read
109  * @ret status          Status code
110  */
111 static int int13_read ( struct int13_drive *drive, uint64_t lba,
112                         struct segoff data, unsigned long count ) {
113         struct block_device *blockdev = drive->blockdev;
114         size_t blksize = blockdev->blksize;
115         uint8_t buffer[blksize];
116         int rc;
117
118         DBG ( "Read %lx sectors from %llx to %04x:%04x\n", count,
119               ( unsigned long long ) lba, data.segment, data.offset );
120         while ( count-- ) {
121                 if ( ( rc = blockdev->read ( blockdev, lba, buffer ) ) != 0 )
122                         return INT13_STATUS_READ_ERROR;
123                 copy_to_real ( data.segment, data.offset, buffer, blksize );
124                 data.offset += blksize;
125                 lba++;
126         }
127         return 0;
128 }
129
130 /**
131  * Write from real-mode data buffer to drive
132  *
133  * @v drive             Emulated drive
134  * @v lba               LBA starting sector number
135  * @v data              Data buffer
136  * @v count             Number of sectors to read
137  * @ret status          Status code
138  */
139 static int int13_write ( struct int13_drive *drive, uint64_t lba,
140                          struct segoff data, unsigned long count ) {
141         struct block_device *blockdev = drive->blockdev;
142         size_t blksize = blockdev->blksize;
143         uint8_t buffer[blksize];
144         int rc;
145
146         DBG ( "Write %lx sectors from %04x:%04x to %llx\n", count,
147               data.segment, data.offset, ( unsigned long long ) lba );
148         while ( count-- ) {
149                 copy_from_real ( buffer, data.segment, data.offset, blksize );
150                 if ( ( rc = blockdev->write ( blockdev, lba, buffer ) ) != 0 )
151                         return INT13_STATUS_WRITE_ERROR;
152                 data.offset += blksize;
153                 lba++;
154         }
155         return 0;
156 }
157
158 /**
159  * INT 13, 00 - Reset disk system
160  *
161  * @v drive             Emulated drive
162  * @ret status          Status code
163  */
164 static int int13_reset ( struct int13_drive *drive __unused,
165                          struct i386_all_regs *ix86 __unused ) {
166         DBG ( "Reset drive\n" );
167         return 0;
168 }
169
170 /**
171  * INT 13, 01 - Get status of last operation
172  *
173  * @v drive             Emulated drive
174  * @ret status          Status code
175  */
176 static int int13_get_last_status ( struct int13_drive *drive,
177                                    struct i386_all_regs *ix86 __unused ) {
178         DBG ( "Get status of last operation\n" );
179         return drive->last_status;
180 }
181
182 /**
183  * INT 13, 02 - Read sectors
184  *
185  * @v drive             Emulated drive
186  * @v al                Number of sectors to read (must be nonzero)
187  * @v ch                Low bits of cylinder number
188  * @v cl (bits 7:6)     High bits of cylinder number
189  * @v cl (bits 5:0)     Sector number
190  * @v dh                Head number
191  * @v es:bx             Data buffer
192  * @ret status          Status code
193  * @ret al              Number of sectors read
194  */
195 static int int13_read_sectors ( struct int13_drive *drive,
196                                 struct i386_all_regs *ix86 ) {
197         unsigned long lba = chs_to_lba ( drive, ix86 );
198         unsigned int count = ix86->regs.al;
199         struct segoff data = {
200                 .segment = ix86->segs.es,
201                 .offset = ix86->regs.bx,
202         };
203
204         if ( drive->blockdev->blksize != INT13_BLKSIZE ) {
205                 DBG ( "Invalid blocksize (%d) for non-extended read\n",
206                       drive->blockdev->blksize );
207                 return INT13_STATUS_INVALID;
208         }
209
210         return int13_read ( drive, lba, data, count );
211 }
212
213 /**
214  * INT 13, 03 - Write sectors
215  *
216  * @v drive             Emulated drive
217  * @v al                Number of sectors to write (must be nonzero)
218  * @v ch                Low bits of cylinder number
219  * @v cl (bits 7:6)     High bits of cylinder number
220  * @v cl (bits 5:0)     Sector number
221  * @v dh                Head number
222  * @v es:bx             Data buffer
223  * @ret status          Status code
224  * @ret al              Number of sectors written
225  */
226 static int int13_write_sectors ( struct int13_drive *drive,
227                                  struct i386_all_regs *ix86 ) {
228         unsigned long lba = chs_to_lba ( drive, ix86 );
229         unsigned int count = ix86->regs.al;
230         struct segoff data = {
231                 .segment = ix86->segs.es,
232                 .offset = ix86->regs.bx,
233         };
234
235         if ( drive->blockdev->blksize != INT13_BLKSIZE ) {
236                 DBG ( "Invalid blocksize (%d) for non-extended write\n",
237                       drive->blockdev->blksize );
238                 return INT13_STATUS_INVALID;
239         }
240
241         return int13_write ( drive, lba, data, count );
242 }
243
244 /**
245  * INT 13, 08 - Get drive parameters
246  *
247  * @v drive             Emulated drive
248  * @ret status          Status code
249  * @ret ch              Low bits of maximum cylinder number
250  * @ret cl (bits 7:6)   High bits of maximum cylinder number
251  * @ret cl (bits 5:0)   Maximum sector number
252  * @ret dh              Maximum head number
253  * @ret dl              Number of drives
254  */
255 static int int13_get_parameters ( struct int13_drive *drive,
256                                   struct i386_all_regs *ix86 ) {
257         unsigned int max_cylinder = drive->cylinders - 1;
258         unsigned int max_head = drive->heads - 1;
259         unsigned int max_sector = drive->sectors_per_track; /* sic */
260
261         DBG ( "Get drive parameters\n" );
262
263         ix86->regs.ch = ( max_cylinder & 0xff );
264         ix86->regs.cl = ( ( ( max_cylinder >> 8 ) << 6 ) | max_sector );
265         ix86->regs.dh = max_head;
266         get_real ( ix86->regs.dl, BDA_SEG, BDA_NUM_DRIVES );
267         return 0;
268 }
269
270 /**
271  * INT 13, 41 - Extensions installation check
272  *
273  * @v drive             Emulated drive
274  * @v bx                0x55aa
275  * @ret bx              0xaa55
276  * @ret cx              Extensions API support bitmap
277  * @ret status          Status code
278  */
279 static int int13_extension_check ( struct int13_drive *drive __unused,
280                                    struct i386_all_regs *ix86 ) {
281         if ( ix86->regs.bx == 0x55aa ) {
282                 DBG ( "INT 13 extensions installation check\n" );
283                 ix86->regs.bx = 0xaa55;
284                 ix86->regs.cx = INT13_EXTENSION_LINEAR;
285                 return 0;
286         } else {
287                 return INT13_STATUS_INVALID;
288         }
289 }
290
291 /**
292  * INT 13, 42 - Extended read
293  *
294  * @v drive             Emulated drive
295  * @v ds:si             Disk address packet
296  * @ret status          Status code
297  */
298 static int int13_extended_read ( struct int13_drive *drive,
299                                  struct i386_all_regs *ix86 ) {
300         struct int13_disk_address addr;
301
302         copy_from_real ( &addr, ix86->segs.ds, ix86->regs.si,
303                          sizeof ( addr ) );
304         return int13_read ( drive, addr.lba, addr.buffer, addr.count );
305 }
306
307 /**
308  * INT 13, 43 - Extended write
309  *
310  * @v drive             Emulated drive
311  * @v ds:si             Disk address packet
312  * @ret status          Status code
313  */
314 static int int13_extended_write ( struct int13_drive *drive,
315                                   struct i386_all_regs *ix86 ) {
316         struct int13_disk_address addr;
317
318         copy_from_real ( &addr, ix86->segs.ds, ix86->regs.si,
319                          sizeof ( addr ) );
320         return int13_write ( drive, addr.lba, addr.buffer, addr.count );
321 }
322
323 /**
324  * INT 13, 48 - Get extended parameters
325  *
326  * @v drive             Emulated drive
327  * @v ds:si             Drive parameter table
328  * @ret status          Status code
329  */
330 static int int13_get_extended_parameters ( struct int13_drive *drive,
331                                            struct i386_all_regs *ix86 ) {
332         struct int13_disk_parameters params = {
333                 .bufsize = sizeof ( params ),
334                 .flags = INT13_FL_DMA_TRANSPARENT,
335                 .cylinders = drive->cylinders,
336                 .heads = drive->heads,
337                 .sectors_per_track = drive->sectors_per_track,
338                 .sectors = drive->blockdev->blocks,
339                 .sector_size = drive->blockdev->blksize,
340         };
341         
342         DBG ( "Get extended drive parameters to %04x:%04x\n",
343               ix86->segs.ds, ix86->regs.si );
344
345         copy_to_real ( ix86->segs.ds, ix86->regs.si, &params,
346                        sizeof ( params ) );
347         return 0;
348 }
349
350 /**
351  * INT 13 handler
352  *
353  */
354 static void int13 ( struct i386_all_regs *ix86 ) {
355         struct int13_drive *drive;
356         int status;
357
358         list_for_each_entry ( drive, &drives, list ) {
359                 if ( drive->drive != ix86->regs.dl )
360                         continue;
361                 
362                 DBG ( "INT 13, %02x on drive %02x\n", ix86->regs.ah,
363                       ix86->regs.dl );
364         
365                 switch ( ix86->regs.ah ) {
366                 case INT13_RESET:
367                         status = int13_reset ( drive, ix86 );
368                         break;
369                 case INT13_GET_LAST_STATUS:
370                         status = int13_get_last_status ( drive, ix86 );
371                         break;
372                 case INT13_READ_SECTORS:
373                         status = int13_read_sectors ( drive, ix86 );
374                         break;
375                 case INT13_WRITE_SECTORS:
376                         status = int13_write_sectors ( drive, ix86 );
377                         break;
378                 case INT13_GET_PARAMETERS:
379                         status = int13_get_parameters ( drive, ix86 );
380                         break;
381                 case INT13_EXTENSION_CHECK:
382                         status = int13_extension_check ( drive, ix86 );
383                         break;
384                 case INT13_EXTENDED_READ:
385                         status = int13_extended_read ( drive, ix86 );
386                         break;
387                 case INT13_EXTENDED_WRITE:
388                         status = int13_extended_write ( drive, ix86 );
389                         break;
390                 case INT13_GET_EXTENDED_PARAMETERS:
391                         status = int13_get_extended_parameters ( drive, ix86 );
392                         break;
393                 default:
394                         DBG ( "Unrecognised INT 13\n" );
395                         status = INT13_STATUS_INVALID;
396                         break;
397                 }
398
399                 /* Store status for INT 13,01 */
400                 drive->last_status = status;
401                 /* All functions return status via %ah and CF */
402                 ix86->regs.ah = status;
403                 if ( status ) {
404                         ix86->flags |= CF;
405                         DBG ( "INT13 failed with status %x\n", status );
406                 }
407                 /* Set OF to indicate to wrapper not to chain this call */
408                 ix86->flags |= OF;
409         }
410 }
411
412 /**
413  * Hook INT 13 handler
414  *
415  */
416 static void hook_int13 ( void ) {
417         /* Assembly wrapper to call int13().  int13() sets OF if we
418          * should not chain to the previous handler.  (The wrapper
419          * clears CF and OF before calling int13()).
420          */
421         __asm__  __volatile__ ( ".section \".text16\", \"ax\", @progbits\n\t"
422                                 ".code16\n\t"
423                                 "\nint13_wrapper:\n\t"
424                                 "orb $0, %%al\n\t" /* clear CF and OF */
425                                 "pushl %0\n\t" /* call int13() */
426                                 "data32 call prot_call\n\t"
427                                 "jo 1f\n\t" /* chain if OF not set */
428                                 "pushfw\n\t"
429                                 "lcall *%%cs:int13_vector\n\t"
430                                 "\n1:\n\t"
431                                 "call 2f\n\t" /* return with flags intact */
432                                 "lret $2\n\t"
433                                 "\n2:\n\t"
434                                 "ret $4\n\t"
435                                 ".previous\n\t"
436                                 ".code32\n\t" : :
437                                 "i" ( int13 ) );
438
439         hook_bios_interrupt ( 0x13, ( unsigned int ) int13_wrapper,
440                               &int13_vector );
441 }
442
443 /**
444  * Unhook INT 13 handler
445  */
446 static void unhook_int13 ( void ) {
447         unhook_bios_interrupt ( 0x13, ( unsigned int ) int13_wrapper,
448                                 &int13_vector );
449 }
450
451 /**
452  * Register INT 13 emulated drive
453  *
454  * @v drive             Emulated drive
455  *
456  * Registers the drive with the INT 13 emulation subsystem, and hooks
457  * the INT 13 interrupt vector (if not already hooked).
458  *
459  * The underlying block device must be valid.  A drive number and
460  * geometry will be assigned if left blank.
461  */
462 void register_int13_drive ( struct int13_drive *drive ) {
463         uint8_t num_drives;
464         unsigned long blocks;
465         unsigned long blocks_per_cyl;
466
467         /* Give drive a default geometry if none specified */
468         if ( ! drive->heads )
469                 drive->heads = 255;
470         if ( ! drive->sectors_per_track )
471                 drive->sectors_per_track = 63;
472         if ( ! drive->cylinders ) {
473                 /* Avoid attempting a 64-bit divide on a 32-bit system */
474                 blocks = ( ( drive->blockdev->blocks <= ULONG_MAX ) ?
475                            drive->blockdev->blocks : ULONG_MAX );
476                 blocks_per_cyl = ( drive->heads * drive->sectors_per_track );
477                 assert ( blocks_per_cyl != 0 );
478                 drive->cylinders = ( blocks / blocks_per_cyl );
479         }
480
481         /* Assign drive number if none specified, update BIOS drive count */
482         get_real ( num_drives, BDA_SEG, BDA_NUM_DRIVES );
483         if ( ! drive->drive )
484                 drive->drive = ( num_drives | 0x80 );
485         if ( num_drives <= ( drive->drive & 0x7f ) )
486                 num_drives = ( ( drive->drive & 0x7f ) + 1 );
487         put_real ( num_drives, BDA_SEG, BDA_NUM_DRIVES );
488
489         DBG ( "Registered INT13 drive %02x with C/H/S geometry %d/%d/%d\n",
490               drive->drive, drive->cylinders, drive->heads,
491               drive->sectors_per_track );
492
493         /* Hook INT 13 vector if not already hooked */
494         if ( list_empty ( &drives ) )
495                 hook_int13();
496
497         /* Add to list of emulated drives */
498         list_add ( &drive->list, &drives );
499 }
500
501 /**
502  * Unregister INT 13 emulated drive
503  *
504  * @v drive             Emulated drive
505  *
506  * Unregisters the drive from the INT 13 emulation subsystem.  If this
507  * is the last emulated drive, the INT 13 vector is unhooked (if
508  * possible).
509  */
510 void unregister_int13_drive ( struct int13_drive *drive ) {
511         /* Remove from list of emulated drives */
512         list_del ( &drive->list );
513
514         DBG ( "Unregistered INT13 drive %02x\n", drive->drive );
515
516         /* Unhook INT 13 vector if no more drives */
517         if ( list_empty ( &drives ) )
518                 unhook_int13();
519 }
520
521 /**
522  * Attempt to boot from an INT 13 drive
523  *
524  * @v drive             Drive number
525  * @ret rc              Return status code
526  *
527  * This boots from the specified INT 13 drive by loading the Master
528  * Boot Record to 0000:7c00 and jumping to it.  INT 18 is hooked to
529  * capture an attempt by the MBR to boot the next device.  (This is
530  * the closest thing to a return path from an MBR).
531  *
532  * Note that this function can never return success, by definition.
533  */
534 int int13_boot ( unsigned int drive ) {
535         int status, signature;
536         int d0, d1;
537
538         DBG ( "Booting from INT 13 drive %02x\n", drive );
539
540         /* Use INT 13 to read the boot sector */
541         REAL_EXEC ( rm_int13_boot,
542                     "pushw $0\n\t"
543                     "popw %%es\n\t"
544                     "int $0x13\n\t"
545                     "jc 1f\n\t"
546                     "xorl %%eax, %%eax\n\t"
547                     "\n1:\n\t"
548                     "movzwl %%es:0x7dfe, %%ebx\n\t",
549                     4,
550                     OUT_CONSTRAINTS ( "=a" ( status ), "=b" ( signature ),
551                                       "=c" ( d0 ), "=d" ( drive ) ),
552                     IN_CONSTRAINTS ( "0" ( 0x0201 ), "1" ( 0x7c00 ),
553                                      "2" ( 0x0001 ), "3" ( drive ) ),
554                     CLOBBER ( "ebp" ) );
555         if ( status )
556                 return -EIO;
557
558         /* Check signature is correct */
559         if ( signature != be16_to_cpu ( 0x55aa ) ) {
560                 DBG ( "Invalid disk signature %#04x (should be 0x55aa)\n",
561                       cpu_to_be16 ( signature ) );
562                 return -ENOEXEC;
563         }
564
565         /* Hook INTs 18 and 19 to capture failure paths */
566         hook_bios_interrupt ( 0x18, ( unsigned int ) int13_exec_fail,
567                               &int18_vector );
568         hook_bios_interrupt ( 0x19, ( unsigned int ) int13_exec_fail,
569                               &int19_vector );
570
571         /* Boot the loaded sector */
572         REAL_EXEC ( rm_int13_exec,
573                     "movw %%ss, %%ax\n\t" /* Preserve stack pointer */
574                     "movw %%ax, %%cs:int13_exec_saved_ss\n\t"
575                     "movw %%sp, %%cs:int13_exec_saved_sp\n\t"
576                     "ljmp $0, $0x7c00\n\t"
577                     "\nint13_exec_saved_ss: .word 0\n\t"
578                     "\nint13_exec_saved_sp: .word 0\n\t"
579                     "\nint13_exec_fail:\n\t"
580                     "movw %%cs:int13_exec_saved_ss, %%ax\n\t"
581                     "movw %%ax, %%ss\n\t"
582                     "movw %%cs:int13_exec_saved_sp, %%sp\n\t"
583                     "\n99:\n\t",
584                     1,
585                     OUT_CONSTRAINTS ( "=d" ( d1 ) ),
586                     IN_CONSTRAINTS ( "0" ( drive ) ),
587                     CLOBBER ( "eax", "ebx", "ecx", "esi", "edi", "ebp" ) );
588
589         DBG ( "Booted disk returned via INT 18 or 19\n" );
590
591         /* Unhook INTs 18 and 19 */
592         unhook_bios_interrupt ( 0x18, ( unsigned int ) int13_exec_fail,
593                                 &int18_vector );
594         unhook_bios_interrupt ( 0x19, ( unsigned int ) int13_exec_fail,
595                                 &int19_vector );
596         
597         return -ECANCELED;
598 }