reactos/boot/freeldr/freeldr/arch/i386/i386disk.c

   1 /*
   2  *  FreeLoader
   3  *  Copyright (C) 1998-2003  Brian Palmer  <brianp@sginet.com>
   4  *
   5  *  This program is free software; you can redistribute it and/or modify
   6  *  it under the terms of the GNU General Public License as published by
   7  *  the Free Software Foundation; either version 2 of the License, or
   8  *  (at your option) any later version.
   9  *
  10  *  This program is distributed in the hope that it will be useful,
  11  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13  *  GNU General Public License for more details.
  14  *
  15  *  You should have received a copy of the GNU General Public License
  16  *  along with this program; if not, write to the Free Software
  17  *  Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  18  */
  19
  20 #include <freeldr.h>
  21
  22 #define NDEBUG
  23 #include <debug.h>
  24
  25 /////////////////////////////////////////////////////////////////////////////////////////////
  26 // FUNCTIONS
  27 /////////////////////////////////////////////////////////////////////////////////////////////
  28
  29 #ifdef __i386__
  30
  31 BOOLEAN DiskResetController(ULONG DriveNumber)
  32 {
  33         REGS    RegsIn;
  34         REGS    RegsOut;
  35
  36         DbgPrint((DPRINT_DISK, "DiskResetController(0x%x) DISK OPERATION FAILED -- RESETTING CONTROLLER\n", DriveNumber));
  37
  38         // BIOS Int 13h, function 0 - Reset disk system
  39         // AH = 00h
  40         // DL = drive (if bit 7 is set both hard disks and floppy disks reset)
  41         // Return:
  42         // AH = status
  43         // CF clear if successful
  44         // CF set on error
  45         RegsIn.b.ah = 0x00;
  46         RegsIn.b.dl = DriveNumber;
  47
  48         // Reset the disk controller
  49         Int386(0x13, &RegsIn, &RegsOut);
  50
  51         return INT386_SUCCESS(RegsOut);
  52 }
  53
  54 BOOLEAN DiskInt13ExtensionsSupported(ULONG DriveNumber)
  55 {
  56         REGS    RegsIn;
  57         REGS    RegsOut;
  58
  59         DbgPrint((DPRINT_DISK, "DiskInt13ExtensionsSupported()\n"));
  60
  61         // IBM/MS INT 13 Extensions - INSTALLATION CHECK
  62         // AH = 41h
  63         // BX = 55AAh
  64         // DL = drive (80h-FFh)
  65         // Return:
  66         // CF set on error (extensions not supported)
  67         // AH = 01h (invalid function)
  68         // CF clear if successful
  69         // BX = AA55h if installed
  70         // AH = major version of extensions
  71         // 01h = 1.x
  72         // 20h = 2.0 / EDD-1.0
  73         // 21h = 2.1 / EDD-1.1
  74         // 30h = EDD-3.0
  75         // AL = internal use
  76         // CX = API subset support bitmap
  77         // DH = extension version (v2.0+ ??? -- not present in 1.x)
  78         //
  79         // Bitfields for IBM/MS INT 13 Extensions API support bitmap
  80         // Bit 0, extended disk access functions (AH=42h-44h,47h,48h) supported
  81         // Bit 1, removable drive controller functions (AH=45h,46h,48h,49h,INT 15/AH=52h) supported
  82         // Bit 2, enhanced disk drive (EDD) functions (AH=48h,AH=4Eh) supported
  83         //        extended drive parameter table is valid
  84         // Bits 3-15 reserved
  85         RegsIn.b.ah = 0x41;
  86         RegsIn.w.bx = 0x55AA;
  87         RegsIn.b.dl = DriveNumber;
  88
  89         // Reset the disk controller
  90         Int386(0x13, &RegsIn, &RegsOut);
  91
  92         if (!INT386_SUCCESS(RegsOut))
  93         {
  94                 // CF set on error (extensions not supported)
  95                 return FALSE;
  96         }
  97
  98         if (RegsOut.w.bx != 0xAA55)
  99         {
 100                 // BX = AA55h if installed
 101                 return FALSE;
 102         }
 103
 104         // Note:
 105         // The original check is too strict because some BIOSes report that
 106         // extended disk access functions are not suported when booting
 107         // from a CD (e.g. Phoenix BIOS v6.00PG). Argh!
 108 #if 0
 109         if (!(RegsOut.w.cx & 0x0001))
 110         {
 111                 // CX = API subset support bitmap
 112                 // Bit 0, extended disk access functions (AH=42h-44h,47h,48h) supported
 113                 return FALSE;
 114         }
 115 #endif
 116
 117         // Use this relaxed check instead
 118         if (RegsOut.w.cx == 0x0000)
 119         {
 120                 // CX = API subset support bitmap
 121                 return FALSE;
 122         }
 123
 124         return TRUE;
 125 }
 126
 127 VOID DiskStopFloppyMotor(VOID)
 128 {
 129         WRITE_PORT_UCHAR((PUCHAR)0x3F2, 0);
 130 }
 131
 132 BOOLEAN DiskGetExtendedDriveParameters(ULONG DriveNumber, PVOID Buffer, USHORT BufferSize)
 133 {
 134         REGS    RegsIn;
 135         REGS    RegsOut;
 136         PUSHORT Ptr = (PUSHORT)(BIOSCALLBUFFER);
 137
 138         DbgPrint((DPRINT_DISK, "DiskGetExtendedDriveParameters()\n"));
 139
 140         // Initialize transfer buffer
 141         *Ptr = BufferSize;
 142
 143         // BIOS Int 13h, function 48h - Get drive parameters
 144         // AH = 48h
 145         // DL = drive (bit 7 set for hard disk)
 146         // DS:SI = result buffer
 147         // Return:
 148         // CF set on error
 149         // AH = status (07h)
 150         // CF clear if successful
 151         // AH = 00h
 152         // DS:SI -> result buffer
 153         RegsIn.b.ah = 0x48;
 154         RegsIn.b.dl = DriveNumber;
 155         RegsIn.x.ds = BIOSCALLBUFSEGMENT;       // DS:SI -> result buffer
 156         RegsIn.w.si = BIOSCALLBUFOFFSET;
 157
 158         // Get drive parameters
 159         Int386(0x13, &RegsIn, &RegsOut);
 160
 161         if (!INT386_SUCCESS(RegsOut))
 162         {
 163                 return FALSE;
 164         }
 165
 166         memcpy(Buffer, Ptr, BufferSize);
 167
 168     DbgPrint((DPRINT_DISK, "size of buffer:                          %x\n", Ptr[0]));
 169     DbgPrint((DPRINT_DISK, "information flags:                       %x\n", Ptr[1]));
 170     DbgPrint((DPRINT_DISK, "number of physical cylinders on drive:   %u\n", *(PULONG)&Ptr[2]));
 171     DbgPrint((DPRINT_DISK, "number of physical heads on drive:       %u\n", *(PULONG)&Ptr[4]));
 172     DbgPrint((DPRINT_DISK, "number of physical sectors per track:    %u\n", *(PULONG)&Ptr[6]));
 173     DbgPrint((DPRINT_DISK, "total number of sectors on drive:        %I64u\n", *(unsigned long long*)&Ptr[8]));
 174     DbgPrint((DPRINT_DISK, "bytes per sector:                        %u\n", Ptr[12]));
 175     if (Ptr[0] >= 0x1e)
 176     {
 177         DbgPrint((DPRINT_DISK, "EED configuration parameters:            %x:%x\n", Ptr[13], Ptr[14]));
 178         if (Ptr[13] != 0xffff && Ptr[14] != 0xffff)
 179         {
 180            PUCHAR SpecPtr = (PUCHAR)((Ptr[13] << 4) + Ptr[14]);
 181            DbgPrint((DPRINT_DISK, "SpecPtr:                                 %x\n", SpecPtr));
 182            DbgPrint((DPRINT_DISK, "physical I/O port base address:          %x\n", *(PUSHORT)&SpecPtr[0]));
 183            DbgPrint((DPRINT_DISK, "disk-drive control port address:         %x\n", *(PUSHORT)&SpecPtr[2]));
 184            DbgPrint((DPRINT_DISK, "drive flags:                             %x\n", SpecPtr[4]));
 185            DbgPrint((DPRINT_DISK, "proprietary information:                 %x\n", SpecPtr[5]));
 186            DbgPrint((DPRINT_DISK, "IRQ for drive:                           %u\n", SpecPtr[6]));
 187            DbgPrint((DPRINT_DISK, "sector count for multi-sector transfers: %u\n", SpecPtr[7]));
 188            DbgPrint((DPRINT_DISK, "DMA control:                             %x\n", SpecPtr[8]));
 189            DbgPrint((DPRINT_DISK, "programmed I/O control:                  %x\n", SpecPtr[9]));
 190            DbgPrint((DPRINT_DISK, "drive options:                           %x\n", *(PUSHORT)&SpecPtr[10]));
 191         }
 192     }
 193     if (Ptr[0] >= 0x42)
 194     {
 195         DbgPrint((DPRINT_DISK, "signature:                             %x\n", Ptr[15]));
 196     }
 197
 198         return TRUE;
 199 }
 200
 201 BOOLEAN i386DiskGetBootVolume(PULONG DriveNumber, PULONGLONG StartSector, PULONGLONG SectorCount, int *FsType)
 202 {
 203         PARTITION_TABLE_ENTRY   PartitionTableEntry;
 204         UCHAR                   VolumeType;
 205         ULONG                   ActivePartition;
 206
 207         DbgPrint((DPRINT_FILESYSTEM, "FsOpenVolume() DriveNumber: 0x%x PartitionNumber: 0x%x\n", i386BootDrive, i386BootPartition));
 208
 209         // Check and see if it is a floppy drive
 210         // If so then just assume FAT12 file system type
 211         if (DiskIsDriveRemovable(i386BootDrive))
 212         {
 213                 DbgPrint((DPRINT_FILESYSTEM, "Drive is a floppy diskette drive. Assuming FAT12 file system.\n"));
 214
 215                 *DriveNumber = i386BootDrive;
 216                 *StartSector = 0;
 217                 *SectorCount = 2 * 80 * 18; /* FIXME hardcoded for 1.44 Mb */
 218                 *FsType = FS_FAT;
 219                 return TRUE;
 220         }
 221
 222         // Check for ISO9660 file system type
 223         if (i386BootDrive >= 0x80 && FsRecIsIso9660(i386BootDrive))
 224         {
 225                 DbgPrint((DPRINT_FILESYSTEM, "Drive is a cdrom drive. Assuming ISO-9660 file system.\n"));
 226
 227                 *DriveNumber = i386BootDrive;
 228                 *StartSector = 0;
 229                 *SectorCount = 0;
 230                 *FsType = FS_ISO9660;
 231                 return TRUE;
 232         }
 233
 234         // Get the requested partition entry
 235         if (i386BootPartition == 0)
 236         {
 237                 // Partition requested was zero which means the boot partition
 238                 if (! DiskGetActivePartitionEntry(i386BootDrive, &PartitionTableEntry, &ActivePartition))
 239                 {
 240                         /* Try partition-less disk */
 241                         *StartSector = 0;
 242                         *SectorCount = 0;
 243                 }
 244                 /* Check for valid partition */
 245                 else if (PartitionTableEntry.SystemIndicator == PARTITION_ENTRY_UNUSED)
 246                 {
 247                         return FALSE;
 248                 }
 249                 else
 250                 {
 251                         *StartSector = PartitionTableEntry.SectorCountBeforePartition;
 252                         *SectorCount = PartitionTableEntry.PartitionSectorCount;
 253                 }
 254         }
 255         else if (0xff == i386BootPartition)
 256         {
 257                 /* Partition-less disk */
 258                 *StartSector = 0;
 259                 *SectorCount = 0;
 260         }
 261         else
 262         {
 263                 // Get requested partition
 264                 if (! MachDiskGetPartitionEntry(i386BootDrive, i386BootPartition, &PartitionTableEntry))
 265                 {
 266                         return FALSE;
 267                 }
 268                 /* Check for valid partition */
 269                 else if (PartitionTableEntry.SystemIndicator == PARTITION_ENTRY_UNUSED)
 270                 {
 271                         return FALSE;
 272                 }
 273                 else
 274                 {
 275                         *StartSector = PartitionTableEntry.SectorCountBeforePartition;
 276                         *SectorCount = PartitionTableEntry.PartitionSectorCount;
 277                 }
 278         }
 279
 280         // Try to recognize the file system
 281         if (!FsRecognizeVolume(i386BootDrive, *StartSector, &VolumeType))
 282         {
 283                 return FALSE;
 284         }
 285
 286         *DriveNumber = i386BootDrive;
 287
 288         switch (VolumeType)
 289         {
 290         case PARTITION_FAT_12:
 291         case PARTITION_FAT_16:
 292         case PARTITION_HUGE:
 293         case PARTITION_XINT13:
 294         case PARTITION_FAT32:
 295         case PARTITION_FAT32_XINT13:
 296                 *FsType = FS_FAT;
 297                 return TRUE;
 298         case PARTITION_EXT2:
 299                 *FsType = FS_EXT2;
 300                 return TRUE;
 301         case PARTITION_NTFS:
 302                 *FsType = FS_NTFS;
 303                 return TRUE;
 304         default:
 305                 *FsType = 0;
 306                 return FALSE;
 307         }
 308
 309         return TRUE;
 310 }
 311
 312 VOID
 313 i386DiskGetBootDevice(PULONG BootDevice)
 314 {
 315   ((char *)BootDevice)[0] = (char)i386BootDrive;
 316   ((char *)BootDevice)[1] = (char)i386BootPartition;
 317 }
 318
 319 BOOLEAN
 320 i386DiskBootingFromFloppy(VOID)
 321 {
 322   return i386BootDrive < 0x80;
 323 }
 324
 325 #define IsRecognizedPartition(P)  \
 326     ((P) == PARTITION_FAT_12       || \
 327      (P) == PARTITION_FAT_16       || \
 328      (P) == PARTITION_HUGE         || \
 329      (P) == PARTITION_IFS          || \
 330      (P) == PARTITION_EXT2         || \
 331      (P) == PARTITION_FAT32        || \
 332      (P) == PARTITION_FAT32_XINT13 || \
 333      (P) == PARTITION_XINT13)
 334
 335 #define IsContainerPartition(P) \
 336     ((P) == PARTITION_EXTENDED         || \
 337      (P) == PARTITION_XINT13_EXTENDED)
 338
 339 BOOLEAN i386DiskGetSystemVolume(char *SystemPath,
 340                              char *RemainingPath,
 341                              PULONG Device,
 342                              PULONG DriveNumber,
 343                              PULONGLONG StartSector,
 344                              PULONGLONG SectorCount,
 345                              int *FsType)
 346 {
 347         ULONG PartitionNumber;
 348         PARTITION_TABLE_ENTRY PartitionTableEntry;
 349         UCHAR VolumeType;
 350         CHAR BootPath[256];
 351         unsigned i, RosPartition;
 352
 353         /*
 354          * Verify system path
 355          */
 356         if (!DissectArcPath(SystemPath, BootPath, DriveNumber, &PartitionNumber))
 357         {
 358                 return FALSE;
 359         }
 360         if (NULL != RemainingPath)
 361         {
 362                 strcpy(RemainingPath, BootPath);
 363         }
 364
 365         /* 0xff -> no partition table present, use whole device */
 366         if (0xff == PartitionNumber)
 367         {
 368                 PartitionTableEntry.SectorCountBeforePartition = 0;
 369                 i = 0xff;
 370         }
 371         else
 372         {
 373                 /* recalculate the boot partition for freeldr */
 374                 i = 0;
 375                 RosPartition = 0;
 376                 while (1)
 377                 {
 378                         if (!MachDiskGetPartitionEntry(*DriveNumber, ++i, &PartitionTableEntry))
 379                         {
 380                                 return FALSE;
 381                         }
 382                         if (!IsContainerPartition(PartitionTableEntry.SystemIndicator) &&
 383                             PartitionTableEntry.SystemIndicator != PARTITION_ENTRY_UNUSED)
 384                         {
 385                                 if (++RosPartition == PartitionNumber)
 386                                 {
 387                                         break;
 388                                 }
 389                         }
 390                 }
 391         }
 392
 393         /* Check for ISO9660 file system type */
 394         if (*DriveNumber >= 0x80 && FsRecIsIso9660(*DriveNumber))
 395         {
 396                 DbgPrint((DPRINT_FILESYSTEM, "Drive is a cdrom drive. Assuming ISO-9660 file system.\n"));
 397
 398                 if (NULL != Device)
 399                 {
 400                         ((char *)Device)[0] = (char)(*DriveNumber);
 401                         ((char *)Device)[1] = (char)i;
 402                 }
 403                 *StartSector = 0;
 404                 *SectorCount = 0;
 405                 *FsType = FS_ISO9660;
 406                 return TRUE;
 407         }
 408
 409         if (!FsRecognizeVolume(*DriveNumber, PartitionTableEntry.SectorCountBeforePartition, &VolumeType))
 410         {
 411                 return FALSE;
 412         }
 413
 414         if (NULL != Device)
 415         {
 416                 ((char *)Device)[0] = (char)(*DriveNumber);
 417                 ((char *)Device)[1] = (char)i;
 418         }
 419         *StartSector = PartitionTableEntry.SectorCountBeforePartition;
 420         *SectorCount = PartitionTableEntry.PartitionSectorCount;
 421
 422         switch (VolumeType)
 423         {
 424         case PARTITION_FAT_12:
 425         case PARTITION_FAT_16:
 426         case PARTITION_HUGE:
 427         case PARTITION_XINT13:
 428         case PARTITION_FAT32:
 429         case PARTITION_FAT32_XINT13:
 430                 *FsType = FS_FAT;
 431                 return TRUE;
 432         case PARTITION_EXT2:
 433                 *FsType = FS_EXT2;
 434                 return TRUE;
 435         case PARTITION_NTFS:
 436                 *FsType = FS_NTFS;
 437                 return TRUE;
 438         default:
 439                 *FsType = 0;
 440                 return FALSE;
 441         }
 442
 443         return FALSE;
 444 }
 445
 446 BOOLEAN
 447 i386DiskGetBootPath(char *BootPath, unsigned Size)
 448 {
 449         static char Path[] = "multi(0)disk(0)";
 450         char Device[4];
 451
 452         _itoa(i386BootDrive, Device, 10);
 453         if (Size <= sizeof(Path) + 6 + strlen(Device))
 454         {
 455                 return FALSE;
 456         }
 457         strcpy(BootPath, Path);
 458         strcat(BootPath, MachDiskBootingFromFloppy() ? "fdisk" : "cdrom");
 459         strcat(strcat(strcat(BootPath, "("), Device), ")");
 460
 461         return TRUE;
 462 }
 463
 464 BOOLEAN
 465 i386DiskNormalizeSystemPath(char *SystemPath, unsigned Size)
 466 {
 467         CHAR BootPath[256];
 468         ULONG PartitionNumber;
 469         ULONG DriveNumber;
 470         PARTITION_TABLE_ENTRY PartEntry;
 471         char *p;
 472
 473         if (!DissectArcPath(SystemPath, BootPath, &DriveNumber, &PartitionNumber))
 474         {
 475                 return FALSE;
 476         }
 477
 478         if (0 != PartitionNumber)
 479         {
 480                 return TRUE;
 481         }
 482
 483         if (! DiskGetActivePartitionEntry(DriveNumber,
 484                                           &PartEntry,
 485                                           &PartitionNumber) ||
 486             PartitionNumber < 1 || 9 < PartitionNumber)
 487         {
 488                 return FALSE;
 489         }
 490
 491         p = SystemPath;
 492         while ('\0' != *p && 0 != _strnicmp(p, "partition(", 10)) {
 493                 p++;
 494         }
 495         p = strchr(p, ')');
 496         if (NULL == p || '0' != *(p - 1)) {
 497                 return FALSE;
 498         }
 499         *(p - 1) = '0' + PartitionNumber;
 500
 501         return TRUE;
 502 }
 503
 504 #endif /* defined __i386__ */
 505
 506 /* EOF */