Chameleon

Chameleon Svn Source Tree

Root/trunk/i386/libsaio/disk.c

1/*
2 * Copyright (c) 1999-2003 Apple Computer, Inc. All rights reserved.
3 *
4 * @APPLE_LICENSE_HEADER_START@
5 *
6 * Portions Copyright (c) 1999-2003 Apple Computer, Inc. All Rights
7 * Reserved. This file contains Original Code and/or Modifications of
8 * Original Code as defined in and that are subject to the Apple Public
9 * Source License Version 2.0 (the "License"). You may not use this file
10 * except in compliance with the License. Please obtain a copy of the
11 * License at http://www.apple.com/publicsource and read it before using
12 * this file.
13 *
14 * The Original Code and all software distributed under the License are
15 * distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY KIND, EITHER
16 * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
17 * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
18 * FITNESS FOR A PARTICULAR PURPOSE OR NON- INFRINGEMENT. Please see the
19 * License for the specific language governing rights and limitations
20 * under the License.
21 *
22 * @APPLE_LICENSE_HEADER_END@
23 */
24/*
25 * Mach Operating System
26 * Copyright (c) 1990 Carnegie-Mellon University
27 * Copyright (c) 1989 Carnegie-Mellon University
28 * All rights reserved. The CMU software License Agreement specifies
29 * the terms and conditions for use and redistribution.
30 */
31
32/*
33 * INTEL CORPORATION PROPRIETARY INFORMATION
34 *
35 * This software is supplied under the terms of a license agreement or
36 * nondisclosure agreement with Intel Corporation and may not be copied
37 * nor disclosed except in accordance with the terms of that agreement.
38 *
39 * Copyright 1988, 1989 Intel Corporation
40 */
41
42/*
43 * Copyright 1993 NeXT Computer, Inc.
44 * All rights reserved.
45 */
46
47/*
48 * Copyright 2007 VMware Inc.
49 * "Preboot" ramdisk support added by David Elliott
50 * GPT support added by David Elliott. Based on IOGUIDPartitionScheme.cpp.
51 */
52
53//Azi: style the rest later...
54
55// Allow UFS_SUPPORT to be overridden with preprocessor option.
56#ifndef UFS_SUPPORT
57// zef: Disabled UFS support
58#define UFS_SUPPORT 0
59#endif
60
61#if UFS_SUPPORT
62#include "ufs.h"
63#endif
64#include <limits.h>
65#include <IOKit/storage/IOApplePartitionScheme.h>
66#include <IOKit/storage/IOGUIDPartitionScheme.h>
67#include "libsaio.h"
68#include "boot.h"
69#include "bootstruct.h"
70#include "memory.h"
71#include "fdisk.h"
72#include "hfs.h"
73#include "ntfs.h"
74#include "msdos.h"
75#include "ext2fs.h"
76#include "befs.h"
77#include "freebsd.h"
78#include "openbsd.h"
79#include "xml.h"
80#include "disk.h"
81// For EFI_GUID
82#include "efi.h"
83#include "efi_tables.h"
84
85typedef struct gpt_hdr gpt_hdr;
86typedef struct gpt_ent gpt_ent;
87
88#define PROBEFS_SIZE BPS * 4 /* buffer size for filesystem probe */
89#define CD_BPS 2048 /* CD-ROM block size */
90#define N_CACHE_SECS (BIOS_LEN / BPS) /* Must be a multiple of 4 for CD-ROMs */
91#define UFS_FRONT_PORCH 0
92#define kAPMSector 2 /* Sector number of Apple partition map */
93#define kAPMCDSector 8 /* Translated sector of Apple partition map on a CD */
94
95/*
96 * IORound and IOTrunc convenience functions, in the spirit
97 * of vm's round_page() and trunc_page().
98 */
99#define IORound(value,multiple) \
100 ((((value) + (multiple) - 1) / (multiple)) * (multiple))
101
102#define IOTrunc(value,multiple) \
103 (((value) / (multiple)) * (multiple));
104
105/*
106 * trackbuf points to the start of the track cache. Biosread()
107 * will store the sectors read from disk to this memory area.
108 *
109 * biosbuf points to a sector within the track cache, and is
110 * updated by Biosread().
111 */
112static char * const trackbuf = (char *) ptov(BIOS_ADDR);
113static char * biosbuf;
114
115/*
116 * Map a disk drive to bootable volumes contained within.
117 */
118struct DiskBVMap {
119 int biosdev; // BIOS device number (unique)
120 BVRef bvr; // chain of boot volumes on the disk
121 int bvrcnt; // number of boot volumes
122 struct DiskBVMap * next; // linkage to next mapping
123};
124
125static struct DiskBVMap * gDiskBVMap = NULL;
126static struct disk_blk0 * gBootSector = NULL;
127
128// Function pointers to be filled in if ramdisks are available:
129int (*p_ramdiskReadBytes)( int biosdev, unsigned int blkno,
130 unsigned int byteoff,
131 unsigned int byteCount, void * buffer ) = NULL;
132int (*p_get_ramdisk_info)(int biosdev, struct driveInfo *dip) = NULL;
133
134static bool getOSVersion(BVRef bvr, char *str);
135
136extern void spinActivityIndicator(int sectors);
137
138//==========================================================================
139
140static int getDriveInfo( int biosdev, struct driveInfo *dip )
141{
142static struct driveInfo cached_di;
143int cc;
144
145// Real BIOS devices are 8-bit, so anything above that is for internal use.
146// Don't cache ramdisk drive info since it doesn't require several BIOS
147// calls and is thus not worth it.
148if (biosdev >= 0x100)
149{
150if (p_get_ramdisk_info != NULL)
151{
152cc = (*p_get_ramdisk_info)(biosdev, dip);
153}
154else
155{
156cc = -1;
157}
158if (cc < 0)
159{
160dip->valid = 0;
161return -1;
162}
163else
164{
165return 0;
166}
167}
168
169if (!cached_di.valid || biosdev != cached_di.biosdev)
170{
171cc = get_drive_info(biosdev, &cached_di);
172
173if (cc < 0)
174{
175cached_di.valid = 0;
176DEBUG_DISK(("get_drive_info returned error\n"));
177return (-1); // BIOS call error
178}
179}
180
181bcopy(&cached_di, dip, sizeof(cached_di));
182
183return 0;
184}
185
186//==========================================================================
187// Maps (E)BIOS return codes to message strings.
188
189struct NamedValue {
190unsigned char value;
191const char * name;
192};
193
194//==========================================================================
195
196static const char * getNameForValue( const struct NamedValue * nameTable,
197 unsigned char value )
198{
199const struct NamedValue * np;
200
201for ( np = nameTable; np->value; np++)
202{
203if (np->value == value)
204{
205return np->name;
206}
207}
208
209return NULL;
210}
211
212#define ECC_CORRECTED_ERR 0x11
213
214static const struct NamedValue bios_errors[] =
215{
216{ 0x10, "Media error" },
217{ 0x11, "Corrected ECC error" },
218{ 0x20, "Controller or device error" },
219{ 0x40, "Seek failed" },
220{ 0x80, "Device timeout" },
221{ 0xAA, "Drive not ready" },
222{ 0x00, 0 }
223};
224
225
226//==============================================================================
227
228static const char * bios_error(int errnum)
229{
230static char errorstr[] = "Error 0x00";
231const char * errname;
232
233errname = getNameForValue(bios_errors, errnum);
234
235if (errname)
236{
237return errname;
238}
239
240sprintf(errorstr, "Error 0x%02x", errnum);
241return errorstr; // No string, print error code only
242}
243
244//==========================================================================
245// Use BIOS INT13 calls to read the sector specified. This function will
246// also perform read-ahead to cache a few subsequent sector to the sector
247// cache.
248//
249// Return:
250// 0 on success, or an error code from INT13/F2 or INT13/F42 BIOS call.
251
252static bool cache_valid = false;
253
254static int Biosread( int biosdev, unsigned long long secno )
255{
256static int xbiosdev, xcyl, xhead;
257static unsigned int xsec, xnsecs;
258struct driveInfo di;
259
260int rc = -1;
261int cyl, head, sec;
262int tries = 0;
263int bps, divisor;
264
265if (getDriveInfo(biosdev, &di) < 0)
266{
267return -1;
268}
269
270if (di.no_emulation)
271{
272bps = 2048; /* Always assume 2K block size since the BIOS may lie about the geometry */
273}
274else
275{
276bps = di.di.params.phys_nbps;
277
278if (bps == 0)
279{
280return -1;
281}
282}
283
284divisor = bps / BPS;
285
286DEBUG_DISK(("Biosread dev %x sec %d bps %d\n", biosdev, secno, bps));
287
288// To read the disk sectors, use EBIOS if we can. Otherwise,
289// revert to the standard BIOS calls.
290
291if ((biosdev >= kBIOSDevTypeHardDrive) && (di.uses_ebios & EBIOS_FIXED_DISK_ACCESS))
292{
293if (cache_valid && (biosdev == xbiosdev) && (secno >= xsec) && ((unsigned int)secno < (xsec + xnsecs)))
294{
295biosbuf = trackbuf + (BPS * (secno - xsec));
296return 0;
297}
298
299xnsecs = N_CACHE_SECS;
300xsec = (secno / divisor) * divisor;
301cache_valid = false;
302
303while ((rc = ebiosread(biosdev, secno / divisor, xnsecs / divisor)) && (++tries < 5))
304{
305if (rc == ECC_CORRECTED_ERR)
306{
307rc = 0; /* Ignore corrected ECC errors */
308break;
309}
310
311error(" EBIOS read error: %s\n", bios_error(rc), rc);
312error(" Block 0x%x Sectors %d\n", secno, xnsecs);
313sleep(1);
314}
315}
316
317else
318{
319/* spc = spt * heads */
320int spc = (di.di.params.phys_spt * di.di.params.phys_heads);
321cyl = secno / spc;
322head = (secno % spc) / di.di.params.phys_spt;
323sec = secno % di.di.params.phys_spt;
324
325if (cache_valid && (biosdev == xbiosdev) && (cyl == xcyl) &&
326(head == xhead) && ((unsigned int)sec >= xsec) && ((unsigned int)sec < (xsec + xnsecs)))
327
328{
329// this sector is in trackbuf cache
330biosbuf = trackbuf + (BPS * (sec - xsec));
331return 0;
332}
333
334// Cache up to a track worth of sectors, but do not cross a track boundary.
335
336xcyl = cyl;
337xhead = head;
338xsec = sec;
339xnsecs = ((unsigned int)(sec + N_CACHE_SECS) > di.di.params.phys_spt) ? (di.di.params.phys_spt - sec) : N_CACHE_SECS;
340
341cache_valid = false;
342
343while ((rc = biosread(biosdev, cyl, head, sec, xnsecs)) && (++tries < 5))
344{
345if (rc == ECC_CORRECTED_ERR)
346{
347rc = 0; /* Ignore corrected ECC errors */
348break;
349}
350error(" BIOS read error: %s\n", bios_error(rc), rc);
351error(" Block %d, Cyl %d Head %d Sector %d\n", secno, cyl, head, sec);
352sleep(1);
353}
354}
355
356// If the BIOS reported success, mark the sector cache as valid.
357
358if (rc == 0)
359{
360cache_valid = true;
361}
362
363biosbuf = trackbuf + (secno % divisor) * BPS;
364xbiosdev = biosdev;
365
366spinActivityIndicator(xnsecs);
367
368return rc;
369}
370
371
372//==============================================================================
373
374int testBiosread(int biosdev, unsigned long long secno)
375{
376return Biosread(biosdev, secno);
377}
378
379//==============================================================================
380
381static int readBytes(int biosdev, unsigned long long blkno, unsigned int byteoff, unsigned int byteCount, void * buffer)
382{
383// ramdisks require completely different code for reading.
384if(p_ramdiskReadBytes != NULL && biosdev >= 0x100)
385{
386return (*p_ramdiskReadBytes)(biosdev, blkno, byteoff, byteCount, buffer);
387}
388
389char * cbuf = (char *) buffer;
390int error;
391int copy_len;
392
393DEBUG_DISK(("%s: dev %x block %x [%d] -> 0x%x...", __FUNCTION__, biosdev, blkno, byteCount, (unsigned)cbuf));
394
395for (; byteCount; cbuf += copy_len, blkno++)
396{
397error = Biosread(biosdev, blkno);
398
399if (error)
400{
401DEBUG_DISK(("error\n"));
402
403return (-1);
404}
405
406copy_len = ((byteCount + byteoff) > BPS) ? (BPS - byteoff) : byteCount;
407bcopy( biosbuf + byteoff, cbuf, copy_len );
408byteCount -= copy_len;
409byteoff = 0;
410}
411
412DEBUG_DISK(("done\n"));
413
414return 0;
415}
416
417//==============================================================================
418
419static int isExtendedFDiskPartition( const struct fdisk_part * part )
420{
421static unsigned char extParts[] =
422{
4230x05, /* Extended */
4240x0f, /* Win95 extended */
4250x85, /* Linux extended */
426};
427
428unsigned int i;
429
430for (i = 0; i < sizeof(extParts)/sizeof(extParts[0]); i++)
431{
432if (extParts[i] == part->systid)
433{
434return 1;
435}
436}
437return 0;
438}
439
440//==============================================================================
441
442static int getNextFDiskPartition( int biosdev, int * partno,
443 const struct fdisk_part ** outPart )
444{
445static int sBiosdev = -1;
446static int sNextPartNo;
447static unsigned int sFirstBase;
448static unsigned int sExtBase;
449static unsigned int sExtDepth;
450static struct fdisk_part * sExtPart;
451struct fdisk_part * part;
452
453if ( sBiosdev != biosdev || *partno < 0 )
454{
455// Fetch MBR.
456if ( readBootSector( biosdev, DISK_BLK0, 0 ) )
457{
458return 0;
459}
460
461sBiosdev = biosdev;
462sNextPartNo = 0;
463sFirstBase = 0;
464sExtBase = 0;
465sExtDepth = 0;
466sExtPart = NULL;
467}
468
469while (1)
470{
471part = NULL;
472
473if ( sNextPartNo < FDISK_NPART )
474{
475part = (struct fdisk_part *) gBootSector->parts[sNextPartNo];
476}
477else if ( sExtPart )
478{
479unsigned int blkno = sExtPart->relsect + sFirstBase;
480
481// Save the block offset of the first extended partition.
482
483if (sExtDepth == 0)
484{
485sFirstBase = blkno;
486}
487sExtBase = blkno;
488
489// Load extended partition table.
490
491if ( readBootSector( biosdev, blkno, 0 ) == 0 )
492{
493sNextPartNo = 0;
494sExtDepth++;
495sExtPart = NULL;
496continue;
497}
498// Fall through to part == NULL
499}
500
501if ( part == NULL ) break; // Reached end of partition chain.
502
503// Advance to next partition number.
504
505sNextPartNo++;
506
507if ( isExtendedFDiskPartition(part) )
508{
509sExtPart = part;
510continue;
511}
512
513// Skip empty slots.
514
515if ( part->systid == 0x00 )
516{
517continue;
518}
519
520// Change relative offset to an absolute offset.
521part->relsect += sExtBase;
522
523*outPart = part;
524*partno = sExtDepth ? (int)(sExtDepth + FDISK_NPART) : sNextPartNo;
525
526break;
527}
528
529return (part != NULL);
530}
531
532//==============================================================================
533
534static BVRef newFDiskBVRef( int biosdev, int partno, unsigned int blkoff,
535 const struct fdisk_part * part,
536 FSInit initFunc, FSLoadFile loadFunc,
537 FSReadFile readFunc,
538 FSGetDirEntry getdirFunc,
539 FSGetFileBlock getBlockFunc,
540 FSGetUUID getUUIDFunc,
541 BVGetDescription getDescriptionFunc,
542 BVFree bvFreeFunc,
543 int probe, int type, unsigned int bvrFlags )
544{
545BVRef bvr = (BVRef) malloc( sizeof(*bvr) );
546if ( bvr )
547{
548bzero(bvr, sizeof(*bvr));
549
550bvr->biosdev = biosdev;
551bvr->part_no = partno;
552bvr->part_boff = blkoff;
553bvr->part_type = part->systid;
554bvr->fs_loadfile = loadFunc;
555bvr->fs_readfile = readFunc;
556bvr->fs_getdirentry = getdirFunc;
557bvr->fs_getfileblock= getBlockFunc;
558bvr->fs_getuuid = getUUIDFunc;
559bvr->description = getDescriptionFunc;
560bvr->type = type;
561bvr->bv_free = bvFreeFunc;
562
563if ((part->bootid & FDISK_ACTIVE) && (part->systid == FDISK_HFS))
564{
565bvr->flags |= kBVFlagPrimary;
566}
567
568// Probe the filesystem.
569
570if ( initFunc )
571{
572bvr->flags |= kBVFlagNativeBoot;
573
574if ( probe && initFunc( bvr ) != 0 )
575{
576// filesystem probe failed.
577
578DEBUG_DISK(("%s: failed probe on dev %x part %d\n", __FUNCTION__, biosdev, partno));
579
580(*bvr->bv_free)(bvr);
581bvr = NULL;
582}
583
584if ( readBootSector( biosdev, blkoff, (void *)0x7e00 ) == 0 )
585{
586bvr->flags |= kBVFlagBootable;
587}
588}
589else if ( readBootSector( biosdev, blkoff, (void *)0x7e00 ) == 0 )
590{
591bvr->flags |= kBVFlagForeignBoot;
592}
593else
594{
595(*bvr->bv_free)(bvr);
596bvr = NULL;
597}
598}
599
600if (bvr) bvr->flags |= bvrFlags;
601{
602return bvr;
603}
604}
605
606//==============================================================================
607
608BVRef newAPMBVRef( int biosdev, int partno, unsigned int blkoff,
609 const DPME * part,
610 FSInit initFunc, FSLoadFile loadFunc,
611 FSReadFile readFunc,
612 FSGetDirEntry getdirFunc,
613 FSGetFileBlock getBlockFunc,
614 FSGetUUID getUUIDFunc,
615 BVGetDescription getDescriptionFunc,
616 BVFree bvFreeFunc,
617 int probe, int type, unsigned int bvrFlags )
618{
619BVRef bvr = (BVRef) malloc( sizeof(*bvr) );
620if ( bvr )
621{
622bzero(bvr, sizeof(*bvr));
623
624bvr->biosdev = biosdev;
625bvr->part_no = partno;
626bvr->part_boff = blkoff;
627bvr->fs_loadfile = loadFunc;
628bvr->fs_readfile = readFunc;
629bvr->fs_getdirentry = getdirFunc;
630bvr->fs_getfileblock= getBlockFunc;
631bvr->fs_getuuid = getUUIDFunc;
632bvr->description = getDescriptionFunc;
633bvr->type = type;
634bvr->bv_free = bvFreeFunc;
635strlcpy(bvr->name, part->dpme_name, DPISTRLEN);
636strlcpy(bvr->type_name, part->dpme_type, DPISTRLEN);
637
638/*
639if ( part->bootid & FDISK_ACTIVE )
640{
641bvr->flags |= kBVFlagPrimary;
642}
643*/
644
645// Probe the filesystem.
646
647if ( initFunc )
648{
649bvr->flags |= kBVFlagNativeBoot | kBVFlagBootable | kBVFlagSystemVolume;
650
651if ( probe && initFunc( bvr ) != 0 )
652{
653// filesystem probe failed.
654
655DEBUG_DISK(("%s: failed probe on dev %x part %d\n", __FUNCTION__, biosdev, partno));
656
657(*bvr->bv_free)(bvr);
658bvr = NULL;
659}
660}
661/*
662else if ( readBootSector( biosdev, blkoff, (void *)0x7e00 ) == 0 )
663{
664bvr->flags |= kBVFlagForeignBoot;
665}
666*/
667else
668{
669(*bvr->bv_free)(bvr);
670bvr = NULL;
671}
672}
673if (bvr)
674{
675bvr->flags |= bvrFlags;
676}
677return bvr;
678}
679
680//==============================================================================
681
682// GUID's in LE form:
683// HFS+ partition - 48465300-0000-11AA-AA11-00306543ECAC
684EFI_GUID const GPT_HFS_GUID = { 0x48465300, 0x0000, 0x11AA, { 0xAA, 0x11, 0x00, 0x30, 0x65, 0x43, 0xEC, 0xAC } };
685
686// turbo - Apple Boot Partition - 426F6F74-0000-11AA-AA11-00306543ECAC
687EFI_GUID const GPT_BOOT_GUID = { 0x426F6F74, 0x0000, 0x11AA, { 0xAA, 0x11, 0x00, 0x30, 0x65, 0x43, 0xEC, 0xAC } };
688
689// turbo - or an EFI System Partition - C12A7328-F81F-11D2-BA4B-00A0C93EC93B
690EFI_GUID const GPT_EFISYS_GUID = { 0xC12A7328, 0xF81F, 0x11D2, { 0xBA, 0x4B, 0x00, 0xA0, 0xC9, 0x3E, 0xC9, 0x3B } };
691
692// zef - Basic Data Partition - EBD0A0A2-B9E5-4433-87C0-68B6B72699C7 for foreign OS support
693EFI_GUID const GPT_BASICDATA_GUID = { 0xEBD0A0A2, 0xB9E5, 0x4433, { 0x87, 0xC0, 0x68, 0xB6, 0xB7, 0x26, 0x99, 0xC7 } };
694
695// Microsoft Reserved Partition - E3C9E316-0B5C-4DB8-817DF92DF00215AE
696EFI_GUID const GPT_BASICDATA2_GUID = { 0xE3C9E316, 0x0B5C, 0x4DB8, { 0x81, 0x7D, 0xF9, 0x2D, 0xF0, 0x02, 0x15, 0xAE } };
697
698
699BVRef newGPTBVRef( int biosdev, int partno, unsigned int blkoff,
700 const gpt_ent * part,
701 FSInit initFunc, FSLoadFile loadFunc,
702 FSReadFile readFunc,
703 FSGetDirEntry getdirFunc,
704 FSGetFileBlock getBlockFunc,
705 FSGetUUID getUUIDFunc,
706 BVGetDescription getDescriptionFunc,
707 BVFree bvFreeFunc,
708 int probe, int type, unsigned int bvrFlags )
709{
710BVRef bvr = (BVRef) malloc( sizeof(*bvr) );
711if ( bvr ) {
712bzero(bvr, sizeof(*bvr));
713
714bvr->biosdev = biosdev;
715bvr->part_no = partno;
716bvr->part_boff = blkoff;
717bvr->fs_loadfile = loadFunc;
718bvr->fs_readfile = readFunc;
719bvr->fs_getdirentry = getdirFunc;
720bvr->fs_getfileblock= getBlockFunc;
721bvr->fs_getuuid = getUUIDFunc;
722bvr->description = getDescriptionFunc;
723bvr->type = type;
724bvr->bv_free = bvFreeFunc;
725// FIXME: UCS-2 -> UTF-8 the name
726strlcpy(bvr->name, "----", DPISTRLEN);
727if ( (efi_guid_compare(&GPT_BOOT_GUID, (EFI_GUID const*)part->ent_type) == 0) || (efi_guid_compare(&GPT_HFS_GUID, (EFI_GUID const*)part->ent_type) == 0) ) {
728strlcpy(bvr->type_name, "GPT HFS+", DPISTRLEN);
729} else {
730strlcpy(bvr->type_name, "GPT Unknown", DPISTRLEN);
731}
732
733/*
734if ( part->bootid & FDISK_ACTIVE ) {
735bvr->flags |= kBVFlagPrimary;
736}
737*/
738
739// Probe the filesystem.
740
741if ( initFunc ) {
742bvr->flags |= kBVFlagNativeBoot;
743
744if ( probe && initFunc( bvr ) != 0 ) {
745// filesystem probe failed.
746
747DEBUG_DISK(("%s: failed probe on dev %x part %d\n", __FUNCTION__, biosdev, partno));
748
749(*bvr->bv_free)(bvr);
750bvr = NULL;
751}
752if ( readBootSector( biosdev, blkoff, (void *)0x7e00 ) == 0 ) {
753bvr->flags |= kBVFlagBootable;
754}
755} else if ( readBootSector( biosdev, blkoff, (void *)0x7e00 ) == 0 ) {
756bvr->flags |= kBVFlagForeignBoot;
757} else {
758(*bvr->bv_free)(bvr);
759bvr = NULL;
760}
761}
762if (bvr) {
763bvr->flags |= bvrFlags;
764}
765return bvr;
766}
767
768//==============================================================================
769
770/* A note on partition numbers:
771 * IOKit makes the primary partitions numbers 1-4, and then
772 * extended partitions are numbered consecutively 5 and up.
773 * So, for example, if you have two primary partitions and
774 * one extended partition they will be numbered 1, 2, 5.
775 */
776
777static BVRef diskScanFDiskBootVolumes( int biosdev, int * countPtr )
778{
779 const struct fdisk_part * part;
780 struct DiskBVMap * map;
781 int partno = -1;
782 BVRef bvr;
783#if UFS_SUPPORT
784 BVRef booterUFS = NULL;
785#endif
786 int spc;
787 struct driveInfo di;
788 boot_drive_info_t *dp;
789
790/* Initialize disk info */
791
792if (getDriveInfo(biosdev, &di) != 0)
793{
794return NULL;
795}
796
797dp = &di.di;
798spc = (dp->params.phys_spt * dp->params.phys_heads);
799
800if (spc == 0)
801{
802/* This is probably a CD-ROM; punt on the geometry. */
803spc = 1;
804}
805
806 do {
807 // Create a new mapping.
808
809 map = (struct DiskBVMap *) malloc( sizeof(*map) );
810 if ( map )
811 {
812 map->biosdev = biosdev;
813 map->bvr = NULL;
814 map->bvrcnt = 0;
815 map->next = gDiskBVMap;
816 gDiskBVMap = map;
817
818 // Create a record for each partition found on the disk.
819
820 while ( getNextFDiskPartition( biosdev, &partno, &part ) )
821 {
822 DEBUG_DISK(("%s: part %d [%x]\n", __FUNCTION__,
823 partno, part->systid));
824 bvr = 0;
825
826 switch ( part->systid )
827 {
828#if UFS_SUPPORT
829 case FDISK_UFS:
830 bvr = newFDiskBVRef(
831 biosdev, partno,
832 part->relsect + UFS_FRONT_PORCH/BPS,
833 part,
834 UFSInitPartition,
835 UFSLoadFile,
836 UFSReadFile,
837 UFSGetDirEntry,
838 UFSGetFileBlock,
839 UFSGetUUID,
840 UFSGetDescription,
841 UFSFree,
842 0,
843 kBIOSDevTypeHardDrive, 0);
844 break;
845#endif
846
847 case FDISK_HFS:
848 bvr = newFDiskBVRef(
849 biosdev, partno,
850 part->relsect,
851 part,
852 HFSInitPartition,
853 HFSLoadFile,
854 HFSReadFile,
855 HFSGetDirEntry,
856 HFSGetFileBlock,
857 HFSGetUUID,
858 HFSGetDescription,
859 HFSFree,
860 0,
861 kBIOSDevTypeHardDrive, 0);
862 break;
863
864 // turbo - we want the booter type scanned also
865 case FDISK_BOOTER:
866 if (part->bootid & FDISK_ACTIVE)
867 gBIOSBootVolume = newFDiskBVRef(
868 biosdev, partno,
869 part->relsect,
870 part,
871 HFSInitPartition,
872 HFSLoadFile,
873 HFSReadFile,
874 HFSGetDirEntry,
875 HFSGetFileBlock,
876 HFSGetUUID,
877 HFSGetDescription,
878 HFSFree,
879 0,
880 kBIOSDevTypeHardDrive, 0);
881 break;
882
883#if UFS_SUPPORT
884 case FDISK_BOOTER:
885 booterUFS = newFDiskBVRef(
886 biosdev, partno,
887 ((part->relsect + spc - 1) / spc) * spc,
888 part,
889 UFSInitPartition,
890 UFSLoadFile,
891 UFSReadFile,
892 UFSGetDirEntry,
893 UFSGetFileBlock,
894 UFSGetUUID,
895 UFSGetDescription,
896 UFSFree,
897 0,
898 kBIOSDevTypeHardDrive, 0);
899 break;
900#endif
901
902 case FDISK_FAT32:
903 case FDISK_DOS12:
904 case FDISK_DOS16S:
905 case FDISK_DOS16B:
906 case FDISK_SMALLFAT32:
907 case FDISK_DOS16SLBA:
908 bvr = newFDiskBVRef(
909 biosdev, partno,
910 part->relsect,
911 part,
912 MSDOSInitPartition,
913 MSDOSLoadFile,
914 MSDOSReadFile,
915 MSDOSGetDirEntry,
916 MSDOSGetFileBlock,
917 MSDOSGetUUID,
918 MSDOSGetDescription,
919 MSDOSFree,
920 0,
921 kBIOSDevTypeHardDrive, 0);
922 break;
923
924 case FDISK_NTFS:
925 bvr = newFDiskBVRef(
926 biosdev, partno,
927 part->relsect,
928 part,
929 0, 0, 0, 0, 0,
930 NTFSGetUUID,
931 NTFSGetDescription,
932 (BVFree)free,
933 0, kBIOSDevTypeHardDrive, 0);
934 break;
935
936 case FDISK_LINUX:
937 bvr = newFDiskBVRef(
938 biosdev, partno,
939 part->relsect,
940 part,
941 0, 0, 0, 0, 0,
942 EX2GetUUID,
943 EX2GetDescription,
944 (BVFree)free,
945 0, kBIOSDevTypeHardDrive, 0);
946 break;
947
948 case FDISK_BEFS:
949 bvr = newFDiskBVRef(
950 biosdev, partno,
951 part->relsect,
952 part,
953 0, 0, 0, 0, 0, 0,
954 BeFSGetDescription,
955 (BVFree)free,
956 0, kBIOSDevTypeHardDrive, 0);
957 break;
958
959 case FDISK_FREEBSD:
960 bvr = newFDiskBVRef(
961 biosdev, partno,
962 part->relsect,
963 part,
964 0, 0, 0, 0, 0, 0,
965 FreeBSDGetDescription,
966 (BVFree)free,
967 0, kBIOSDevTypeHardDrive, 0);
968 break;
969
970 case FDISK_OPENBSD:
971 bvr = newFDiskBVRef(
972 biosdev, partno,
973 part->relsect,
974 part,
975 0, 0, 0, 0, 0, 0,
976 OpenBSDGetDescription,
977 (BVFree)free,
978 0, kBIOSDevTypeHardDrive, 0);
979 break;
980
981 default:
982 bvr = newFDiskBVRef(
983 biosdev, partno,
984 part->relsect,
985 part,
986 0, 0, 0, 0, 0, 0, 0,
987 (BVFree)free,
988 0,
989 kBIOSDevTypeHardDrive, 0);
990 break;
991 }
992
993 if ( bvr )
994 {
995 bvr->next = map->bvr;
996 map->bvr = bvr;
997 map->bvrcnt++;
998 }
999 }
1000
1001#if UFS_SUPPORT
1002 // Booting from a CD with an UFS filesystem embedded
1003 // in a booter partition.
1004
1005if ( booterUFS )
1006{
1007if ( map->bvrcnt == 0 )
1008{
1009map->bvr = booterUFS;
1010map->bvrcnt++;
1011}
1012else
1013{
1014free( booterUFS );
1015}
1016}
1017#endif
1018}
1019} while (0);
1020
1021/*
1022 * If no FDisk partition, then we will check for
1023 * an Apple partition map elsewhere.
1024 */
1025#if UNUSED
1026if (map->bvrcnt == 0)
1027{
1028static struct fdisk_part cdpart;
1029cdpart.systid = 0xCD;
1030
1031/* Let's try assuming we are on a hybrid HFS/ISO9660 CD. */
1032bvr = newFDiskBVRef(
1033biosdev, 0,
10340,
1035&cdpart,
1036HFSInitPartition,
1037HFSLoadFile,
1038HFSReadFile,
1039HFSGetDirEntry,
1040HFSGetFileBlock,
1041HFSGetUUID,
10420,
1043kBIOSDevTypeHardDrive);
1044bvr->next = map->bvr;
1045map->bvr = bvr;
1046map->bvrcnt++;
1047}
1048#endif
1049// Actually this should always be true given the above code
1050if(map == gDiskBVMap)
1051{
1052// Don't leave a null map in the chain
1053if(map->bvrcnt == 0 && map->bvr == NULL)
1054{
1055gDiskBVMap = map->next;
1056free(map);
1057map = NULL;
1058}
1059}
1060
1061if (countPtr) *countPtr = map ? map->bvrcnt : 0;
1062
1063return map ? map->bvr : NULL;
1064}
1065
1066//==============================================================================
1067
1068static BVRef diskScanAPMBootVolumes( int biosdev, int * countPtr )
1069{
1070struct DiskBVMap * map;
1071struct Block0 *block0_p;
1072unsigned int blksize;
1073unsigned int factor;
1074void *buffer = malloc(BPS);
1075
1076if (!buffer)
1077{
1078return NULL;
1079}
1080
1081/* Check for alternate block size */
1082if (readBytes( biosdev, 0, 0, BPS, buffer ) != 0)
1083{
1084return NULL;
1085}
1086block0_p = buffer;
1087if (OSSwapBigToHostInt16(block0_p->sbSig) == BLOCK0_SIGNATURE)
1088{
1089blksize = OSSwapBigToHostInt16(block0_p->sbBlkSize);
1090if (blksize != BPS)
1091{
1092free(buffer);
1093buffer = malloc(blksize);
1094if (!buffer)
1095{
1096return NULL;
1097}
1098}
1099factor = blksize / BPS;
1100}
1101else
1102{
1103blksize = BPS;
1104factor = 1;
1105}
1106
1107do
1108{
1109// Create a new mapping.
1110
1111map = (struct DiskBVMap *) malloc( sizeof(*map) );
1112if ( map )
1113{
1114int error;
1115DPME *dpme_p = (DPME *)buffer;
1116UInt32 i, npart = UINT_MAX;
1117BVRef bvr;
1118
1119map->biosdev = biosdev;
1120map->bvr = NULL;
1121map->bvrcnt = 0;
1122map->next = gDiskBVMap;
1123gDiskBVMap = map;
1124
1125for (i=0; i<npart; i++)
1126{
1127error = readBytes( biosdev, (kAPMSector + i) * factor, 0, blksize, buffer );
1128
1129if (error || OSSwapBigToHostInt16(dpme_p->dpme_signature) != DPME_SIGNATURE)
1130{
1131break;
1132}
1133
1134if (i==0)
1135{
1136npart = OSSwapBigToHostInt32(dpme_p->dpme_map_entries);
1137}
1138/*
1139printf("name = %s, %s%s %d -> %d [%d -> %d] {%d}\n",
1140dpme.dpme_name, dpme.dpme_type, (dpme.dpme_flags & DPME_FLAGS_BOOTABLE) ? "(bootable)" : "",
1141dpme.dpme_pblock_start, dpme.dpme_pblocks,
1142dpme.dpme_lblock_start, dpme.dpme_lblocks,
1143dpme.dpme_boot_block);
1144*/
1145
1146if (strcmp(dpme_p->dpme_type, "Apple_HFS") == 0)
1147{
1148bvr = newAPMBVRef(biosdev,
1149i,
1150OSSwapBigToHostInt32(dpme_p->dpme_pblock_start) * factor,
1151dpme_p,
1152HFSInitPartition,
1153HFSLoadFile,
1154HFSReadFile,
1155HFSGetDirEntry,
1156HFSGetFileBlock,
1157HFSGetUUID,
1158HFSGetDescription,
1159HFSFree,
11600,
1161kBIOSDevTypeHardDrive, 0);
1162bvr->next = map->bvr;
1163map->bvr = bvr;
1164map->bvrcnt++;
1165}
1166}
1167}
1168} while (0);
1169
1170free(buffer);
1171
1172if (countPtr) *countPtr = map ? map->bvrcnt : 0;
1173
1174return map ? map->bvr : NULL;
1175}
1176
1177//==============================================================================
1178
1179/*
1180 * Trying to figure out the filsystem type of a given partition.
1181 */
1182static int probeFileSystem(int biosdev, unsigned int blkoff)
1183{
1184// detected filesystem type;
1185int result = -1;
1186int fatbits;
1187
1188// Allocating buffer for 4 sectors.
1189const void * probeBuffer = malloc(PROBEFS_SIZE);
1190if (probeBuffer == NULL)
1191{
1192goto exit;
1193}
1194
1195// Reading first 4 sectors of current partition
1196int error = readBytes(biosdev, blkoff, 0, PROBEFS_SIZE, (void *)probeBuffer);
1197
1198if (error)
1199{
1200goto exit;
1201}
1202
1203if (HFSProbe(probeBuffer))
1204{
1205result = FDISK_HFS;
1206}
1207else if (EX2Probe(probeBuffer))
1208{
1209result = FDISK_LINUX;
1210}
1211else if (FreeBSDProbe(probeBuffer))
1212{
1213result = FDISK_FREEBSD;
1214}
1215
1216else if (OpenBSDProbe(probeBuffer))
1217{
1218result = FDISK_OPENBSD;
1219}
1220
1221else if (BeFSProbe(probeBuffer))
1222{
1223result = FDISK_BEFS;
1224}
1225
1226else if (NTFSProbe(probeBuffer))
1227{
1228result = FDISK_NTFS;
1229}
1230
1231else if ( (fatbits = MSDOSProbe(probeBuffer)) )
1232{
1233switch (fatbits)
1234{
1235case 32:
1236default:
1237result = FDISK_FAT32;
1238break;
1239case 16:
1240result = FDISK_DOS16B;
1241break;
1242case 12:
1243result = FDISK_DOS12;
1244break;
1245}
1246}
1247else
1248{
1249// Couldn't detect filesystem type
1250result = 0;
1251}
1252
1253exit:
1254if (probeBuffer != NULL) free((void *)probeBuffer);
1255{
1256return result;
1257}
1258}
1259
1260//==============================================================================
1261
1262static bool isPartitionUsed(gpt_ent * partition)
1263{
1264
1265// Ask whether the given partition is used.
1266
1267return efi_guid_is_null((EFI_GUID const*)partition->ent_type) ? false : true;
1268}
1269
1270//==============================================================================
1271
1272static BVRef diskScanGPTBootVolumes(int biosdev, int * countPtr)
1273{
1274struct DiskBVMap *map = NULL;
1275
1276void *buffer = malloc(BPS);
1277
1278int error;
1279if ( (error = readBytes( biosdev, /*secno*/0, 0, BPS, buffer )) != 0)
1280{
1281verbose("Failed to read boot sector from BIOS device %02xh. Error=%d\n", biosdev, error);
1282goto scanErr;
1283}
1284struct REAL_disk_blk0 *fdiskMap = buffer;
1285if ( OSSwapLittleToHostInt16(fdiskMap->signature) != DISK_SIGNATURE )
1286{
1287verbose("Failed to find boot signature on BIOS device %02xh\n", biosdev);
1288goto scanErr;
1289}
1290
1291int fdiskID = 0;
1292unsigned index;
1293for ( index = 0; index < FDISK_NPART; index++ )
1294{
1295if ( fdiskMap->parts[index].systid )
1296{
1297if ( fdiskMap->parts[index].systid == 0xEE )
1298{
1299// Fail if two 0xEE partitions are present which
1300// means the FDISK code will wind up parsing it.
1301if ( fdiskID )
1302{
1303goto scanErr;
1304}
1305
1306fdiskID = index + 1;
1307}
1308}
1309}
1310
1311if ( fdiskID == 0 )
1312{
1313goto scanErr;
1314}
1315
1316verbose("Attempting to read GPT\n");
1317
1318if(readBytes(biosdev, 1, 0, BPS, buffer) != 0)
1319{
1320goto scanErr;
1321}
1322
1323gpt_hdr *headerMap = buffer;
1324
1325// Determine whether the partition header signature is present.
1326
1327if ( memcmp(headerMap->hdr_sig, GPT_HDR_SIG, strlen(GPT_HDR_SIG)) )
1328{
1329goto scanErr;
1330}
1331
1332// Determine whether the partition header size is valid.
1333
1334UInt32 headerCheck = OSSwapLittleToHostInt32(headerMap->hdr_crc_self);
1335UInt32 headerSize = OSSwapLittleToHostInt32(headerMap->hdr_size);
1336
1337if ( headerSize < offsetof(gpt_hdr, padding) )
1338{
1339goto scanErr;
1340}
1341
1342if ( headerSize > BPS )
1343{
1344goto scanErr;
1345}
1346
1347// Determine whether the partition header checksum is valid.
1348
1349headerMap->hdr_crc_self = 0;
1350
1351if ( crc32(0, headerMap, headerSize) != headerCheck )
1352{
1353goto scanErr;
1354}
1355
1356// Determine whether the partition entry size is valid.
1357
1358UInt64 gptBlock = 0;
1359UInt32 gptCheck = 0;
1360UInt32 gptCount = 0;
1361UInt32 gptID = 0;
1362gpt_ent * gptMap = 0;
1363UInt32 gptSize = 0;
1364
1365gptBlock = OSSwapLittleToHostInt64(headerMap->hdr_lba_table);
1366gptCheck = OSSwapLittleToHostInt32(headerMap->hdr_crc_table);
1367gptCount = OSSwapLittleToHostInt32(headerMap->hdr_entries);
1368gptSize = OSSwapLittleToHostInt32(headerMap->hdr_entsz);
1369
1370if ( gptSize < sizeof(gpt_ent) )
1371{
1372goto scanErr;
1373}
1374
1375// Allocate a buffer large enough to hold one map, rounded to a media block.
1376free(buffer);
1377buffer = NULL;
1378
1379UInt32 bufferSize = IORound(gptCount * gptSize, BPS);
1380if (bufferSize == 0)
1381{
1382goto scanErr;
1383}
1384buffer = malloc(bufferSize);
1385if (!buffer)
1386{
1387 goto scanErr;
1388}
1389
1390if (readBytes(biosdev, gptBlock, 0, bufferSize, buffer) != 0)
1391{
1392goto scanErr;
1393}
1394verbose("Read GPT\n");
1395
1396// Allocate a new map for this BIOS device and insert it into the chain
1397map = malloc(sizeof(*map));
1398if (!map)
1399{
1400goto scanErr;
1401}
1402map->biosdev = biosdev;
1403map->bvr = NULL;
1404map->bvrcnt = 0;
1405map->next = gDiskBVMap;
1406gDiskBVMap = map;
1407
1408// fdisk like partition type id.
1409int fsType = 0;
1410
1411for(gptID = 1; gptID <= gptCount; ++gptID) {
1412BVRef bvr = NULL;
1413unsigned int bvrFlags = 0;
1414
1415// size on disk can be larger than sizeof(gpt_ent)
1416gptMap = (gpt_ent *) ( buffer + ( (gptID - 1) * gptSize) );
1417
1418// NOTE: EFI_GUID's are in LE and we know we're on an x86.
1419// The IOGUIDPartitionScheme.cpp code uses byte-based UUIDs, we don't.
1420
1421if (isPartitionUsed(gptMap)) {
1422char stringuuid[100];
1423efi_guid_unparse_upper((EFI_GUID*)gptMap->ent_type, stringuuid);
1424verbose("Reading GPT partition %d, type %s\n", gptID, stringuuid);
1425
1426// Getting fdisk like partition type.
1427fsType = probeFileSystem(biosdev, gptMap->ent_lba_start);
1428
1429if ( (efi_guid_compare(&GPT_BOOT_GUID, (EFI_GUID const*)gptMap->ent_type) == 0) || (efi_guid_compare(&GPT_HFS_GUID, (EFI_GUID const*)gptMap->ent_type) == 0) ) {
1430bvrFlags = (efi_guid_compare(&GPT_BOOT_GUID, (EFI_GUID const*)gptMap->ent_type) == 0) ? kBVFlagBooter : 0;
1431bvr = newGPTBVRef(biosdev,
1432gptID,
1433gptMap->ent_lba_start,
1434gptMap,
1435HFSInitPartition,
1436HFSLoadFile,
1437HFSReadFile,
1438HFSGetDirEntry,
1439HFSGetFileBlock,
1440HFSGetUUID,
1441HFSGetDescription,
1442HFSFree,
14430,
1444kBIOSDevTypeHardDrive, bvrFlags);
1445}
1446
1447// zef - foreign OS support
1448if ( (efi_guid_compare(&GPT_BASICDATA_GUID, (EFI_GUID const*)gptMap->ent_type) == 0) ||
1449(efi_guid_compare(&GPT_BASICDATA2_GUID, (EFI_GUID const*)gptMap->ent_type) == 0) ) {
1450switch (fsType)
1451{
1452case FDISK_NTFS:
1453bvr = newGPTBVRef(biosdev, gptID, gptMap->ent_lba_start, gptMap,
14540, 0, 0, 0, 0, 0, NTFSGetDescription,
1455(BVFree)free, 0, kBIOSDevTypeHardDrive, 0);
1456break;
1457
1458case FDISK_LINUX:
1459bvr = newGPTBVRef(biosdev, gptID, gptMap->ent_lba_start, gptMap,
14600, 0, 0, 0, 0, 0, EX2GetDescription,
1461(BVFree)free, 0, kBIOSDevTypeHardDrive, 0);
1462break;
1463
1464default:
1465bvr = newGPTBVRef(biosdev, gptID, gptMap->ent_lba_start, gptMap,
14660, 0, 0, 0, 0, 0, 0,
1467(BVFree)free, 0, kBIOSDevTypeHardDrive, 0);
1468break;
1469}
1470
1471}
1472
1473// turbo - save our booter partition
1474// zef - only on original boot device
1475if ( (efi_guid_compare(&GPT_EFISYS_GUID, (EFI_GUID const*)gptMap->ent_type) == 0) ) {
1476switch (fsType) {
1477case FDISK_HFS:
1478if (readBootSector( biosdev, gptMap->ent_lba_start, (void *)0x7e00 ) == 0) {
1479bvr = newGPTBVRef(biosdev, gptID, gptMap->ent_lba_start, gptMap,
1480HFSInitPartition,
1481HFSLoadFile,
1482HFSReadFile,
1483HFSGetDirEntry,
1484HFSGetFileBlock,
1485HFSGetUUID,
1486HFSGetDescription,
1487HFSFree,
14880, kBIOSDevTypeHardDrive, kBVFlagEFISystem);
1489}
1490break;
1491
1492case FDISK_FAT32:
1493if (testFAT32EFIBootSector( biosdev, gptMap->ent_lba_start, (void *)0x7e00 ) == 0) {
1494bvr = newGPTBVRef(biosdev, gptID, gptMap->ent_lba_start, gptMap,
1495MSDOSInitPartition,
1496MSDOSLoadFile,
1497MSDOSReadFile,
1498MSDOSGetDirEntry,
1499MSDOSGetFileBlock,
1500MSDOSGetUUID,
1501MSDOSGetDescription,
1502MSDOSFree,
15030, kBIOSDevTypeHardDrive, kBVFlagEFISystem);
1504}
1505break;
1506
1507default:
1508if (biosdev == gBIOSDev) {
1509gBIOSBootVolume = bvr;
1510}
1511break;
1512}
1513}
1514
1515if (bvr)
1516{
1517// Fixup bvr with the fake fdisk partition type.
1518if (fsType > 0) {
1519bvr->part_type = fsType;
1520}
1521
1522bvr->next = map->bvr;
1523map->bvr = bvr;
1524++map->bvrcnt;
1525}
1526
1527}
1528}
1529
1530scanErr:
1531if (buffer) {
1532free(buffer);
1533}
1534
1535if(map) {
1536if(countPtr) *countPtr = map->bvrcnt;
1537{
1538return map->bvr;
1539}
1540
1541} else {
1542if(countPtr) *countPtr = 0;
1543{
1544return NULL;
1545}
1546}
1547}
1548
1549//==============================================================================
1550
1551static bool getOSVersion(BVRef bvr, char *str)
1552{
1553bool valid = false;
1554config_file_t systemVersion;
1555char dirSpec[512];
1556
1557sprintf(dirSpec, "hd(%d,%d)/System/Library/CoreServices/SystemVersion.plist", BIOS_DEV_UNIT(bvr), bvr->part_no);
1558
1559if (!loadConfigFile(dirSpec, &systemVersion)) {
1560valid = true;
1561} else {
1562sprintf(dirSpec, "hd(%d,%d)/System/Library/CoreServices/ServerVersion.plist", BIOS_DEV_UNIT(bvr), bvr->part_no);
1563
1564if (!loadConfigFile(dirSpec, &systemVersion))
1565{
1566bvr->OSisServer = true;
1567valid = true;
1568}
1569}
1570
1571if (valid) {
1572const char *val;
1573int len;
1574
1575if (getValueForKey(kProductVersion, &val, &len, &systemVersion)) {
1576// getValueForKey uses const char for val
1577// so copy it and trim
1578*str = '\0';
1579strncat(str, val, MIN(len, 4));
1580} else {
1581valid = false;
1582}
1583}
1584
1585if(!valid) {
1586int fh = -1;
1587sprintf(dirSpec, "hd(%d,%d)/.PhysicalMediaInstall", BIOS_DEV_UNIT(bvr), bvr->part_no);
1588fh = open(dirSpec, 0);
1589
1590if (fh >= 0) {
1591valid = true;
1592bvr->OSisInstaller = true;
1593strcpy(bvr->OSVersion, "10.7"); // 10.7 +
1594close(fh);
1595} else {
1596close(fh);
1597}
1598}
1599return valid;
1600}
1601
1602//==============================================================================
1603
1604static void scanFSLevelBVRSettings(BVRef chain)
1605{
1606BVRef bvr;
1607char dirSpec[512], fileSpec[512];
1608char label[BVSTRLEN];
1609int ret;
1610long flags, time;
1611int fh, fileSize, error;
1612
1613for (bvr = chain; bvr; bvr = bvr->next) {
1614ret = -1;
1615error = 0;
1616
1617//
1618// Check for alternate volume label on boot helper partitions.
1619//
1620if (bvr->flags & kBVFlagBooter) {
1621sprintf(dirSpec, "hd(%d,%d)/System/Library/CoreServices/", BIOS_DEV_UNIT(bvr), bvr->part_no);
1622strcpy(fileSpec, ".disk_label.contentDetails");
1623ret = GetFileInfo(dirSpec, fileSpec, &flags, &time);
1624if (!ret) {
1625fh = open(strcat(dirSpec, fileSpec), 0);
1626fileSize = file_size(fh);
1627if (fileSize > 0 && fileSize < BVSTRLEN) {
1628if (read(fh, label, fileSize) != fileSize) {
1629error = -1;
1630}
1631} else {
1632error = -1;
1633}
1634
1635close(fh);
1636
1637if (!error) {
1638label[fileSize] = '\0';
1639strcpy(bvr->altlabel, label);
1640}
1641}
1642}
1643
1644// Check for SystemVersion.plist or ServerVersion.plist to determine if a volume hosts an installed system.
1645
1646if (bvr->flags & kBVFlagNativeBoot) {
1647if (getOSVersion(bvr,bvr->OSVersion) == true) {
1648bvr->flags |= kBVFlagSystemVolume;
1649}
1650}
1651}
1652}
1653
1654//==============================================================================
1655
1656void rescanBIOSDevice(int biosdev)
1657{
1658struct DiskBVMap *oldMap = diskResetBootVolumes(biosdev);
1659CacheReset();
1660diskFreeMap(oldMap);
1661oldMap = NULL;
1662scanBootVolumes(biosdev, 0);
1663}
1664
1665//==============================================================================
1666
1667struct DiskBVMap* diskResetBootVolumes(int biosdev)
1668{
1669struct DiskBVMap * map;
1670struct DiskBVMap *prevMap = NULL;
1671for ( map = gDiskBVMap; map; prevMap = map, map = map->next ) {
1672if ( biosdev == map->biosdev ) {
1673break;
1674}
1675}
1676
1677if(map != NULL) {
1678verbose("Resetting BIOS device %xh\n", biosdev);
1679// Reset the biosbuf cache
1680cache_valid = false;
1681if(map == gDiskBVMap) {
1682gDiskBVMap = map->next;
1683} else if(prevMap != NULL) {
1684prevMap->next = map->next;
1685} else {
1686stop("");
1687}
1688}
1689// Return the old map, either to be freed, or reinserted later
1690return map;
1691}
1692
1693//==============================================================================
1694
1695// Frees a DiskBVMap and all of its BootVolume's
1696void diskFreeMap(struct DiskBVMap *map)
1697{
1698if(map != NULL)
1699{
1700while(map->bvr != NULL)
1701{
1702BVRef bvr = map->bvr;
1703map->bvr = bvr->next;
1704(*bvr->bv_free)(bvr);
1705}
1706
1707free(map);
1708}
1709}
1710
1711//==============================================================================
1712
1713BVRef diskScanBootVolumes(int biosdev, int * countPtr)
1714{
1715struct DiskBVMap *map;
1716BVRef bvr;
1717int count = 0;
1718
1719// Find an existing mapping for this device.
1720
1721for (map = gDiskBVMap; map; map = map->next)
1722{
1723if (biosdev == map->biosdev)
1724{
1725count = map->bvrcnt;
1726break;
1727}
1728}
1729
1730if (map == NULL)
1731{
1732bvr = diskScanGPTBootVolumes(biosdev, &count);
1733if (bvr == NULL)
1734{
1735bvr = diskScanFDiskBootVolumes(biosdev, &count);
1736}
1737if (bvr == NULL)
1738{
1739bvr = diskScanAPMBootVolumes(biosdev, &count);
1740}
1741if (bvr)
1742{
1743scanFSLevelBVRSettings(bvr);
1744}
1745}
1746else
1747{
1748bvr = map->bvr;
1749}
1750if (countPtr)
1751{
1752*countPtr += count;
1753}
1754return bvr;
1755}
1756
1757//==============================================================================
1758
1759BVRef getBVChainForBIOSDev(int biosdev)
1760{
1761BVRef chain = NULL;
1762struct DiskBVMap * map = NULL;
1763
1764for (map = gDiskBVMap; map; map = map->next)
1765{
1766if (map->biosdev == biosdev)
1767{
1768chain = map->bvr;
1769break;
1770}
1771}
1772
1773return chain;
1774}
1775
1776//==============================================================================
1777
1778BVRef newFilteredBVChain(int minBIOSDev, int maxBIOSDev, unsigned int allowFlags, unsigned int denyFlags, int *count)
1779{
1780BVRef chain = NULL;
1781BVRef bvr = NULL;
1782BVRef newBVR = NULL;
1783BVRef prevBVR = NULL;
1784
1785struct DiskBVMap * map = NULL;
1786int bvCount = 0;
1787
1788const char *raw = 0;
1789char* val = 0;
1790int len;
1791
1792getValueForKey(kHidePartition, &raw, &len, &bootInfo->chameleonConfig);
1793if(raw)
1794{
1795val = XMLDecode(raw);
1796}
1797
1798/*
1799 * Traverse gDISKBVmap to get references for
1800 * individual bvr chains of each drive.
1801 */
1802for (map = gDiskBVMap; map; map = map->next)
1803{
1804for (bvr = map->bvr; bvr; bvr = bvr->next)
1805{
1806/*
1807 * Save the last bvr.
1808 */
1809if (newBVR)
1810{
1811prevBVR = newBVR;
1812}
1813
1814/*
1815 * Allocate and copy the matched bvr entry into a new one.
1816 */
1817newBVR = (BVRef) malloc(sizeof(*newBVR));
1818if (!newBVR)
1819{
1820continue;
1821}
1822bcopy(bvr, newBVR, sizeof(*newBVR));
1823
1824/*
1825 * Adjust the new bvr's fields.
1826 */
1827newBVR->next = NULL;
1828newBVR->filtered = true;
1829
1830if ( (!allowFlags || newBVR->flags & allowFlags)
1831&& (!denyFlags || !(newBVR->flags & denyFlags) )
1832&& (newBVR->biosdev >= minBIOSDev && newBVR->biosdev <= maxBIOSDev)
1833) {
1834newBVR->visible = true;
1835}
1836
1837/*
1838 * Looking for "Hide Partition" entries in 'hd(x,y)|uuid|"label" hd(m,n)|uuid|"label"' format,
1839 * to be able to hide foreign partitions from the boot menu.
1840 *
1841 */
1842if ( (newBVR->flags & kBVFlagForeignBoot) ) {
1843char *start, *next = val;
1844long len = 0;
1845do
1846{
1847start = strbreak(next, &next, &len);
1848if(len && matchVolumeToString(newBVR, start, len) )
1849{
1850newBVR->visible = false;
1851}
1852}
1853while ( next && *next );
1854}
1855
1856/*
1857 * Use the first bvr entry as the starting chain pointer.
1858 */
1859if (!chain) {
1860chain = newBVR;
1861}
1862
1863/*
1864 * Update the previous bvr's link pointer to use the new memory area.
1865 */
1866if (prevBVR) {
1867prevBVR->next = newBVR;
1868}
1869
1870if (newBVR->visible) {
1871bvCount++;
1872}
1873}
1874}
1875
1876#if DEBUG //Azi: warning - too big for boot-log.. far too big.. i mean HUGE!! :P
1877for (bvr = chain; bvr; bvr = bvr->next)
1878{
1879printf(" bvr: %d, dev: %d, part: %d, flags: %d, vis: %d\n", bvr, bvr->biosdev, bvr->part_no, bvr->flags, bvr->visible);
1880}
1881printf("count: %d\n", bvCount);
1882getchar();
1883#endif
1884
1885*count = bvCount;
1886
1887free(val);
1888return chain;
1889}
1890
1891//==============================================================================
1892
1893int freeFilteredBVChain(const BVRef chain)
1894{
1895int ret = 1;
1896BVRef bvr = chain;
1897BVRef nextBVR = NULL;
1898
1899while (bvr)
1900{
1901nextBVR = bvr->next;
1902
1903if (bvr->filtered)
1904{
1905free(bvr);
1906}
1907else
1908{
1909ret = 0;
1910break;
1911}
1912
1913bvr = nextBVR;
1914}
1915
1916return ret;
1917}
1918
1919//==============================================================================
1920
1921static const struct NamedValue fdiskTypes[] =
1922{
1923{ FDISK_NTFS,"Windows NTFS" },
1924{ FDISK_DOS12,"Windows FAT12" },
1925{ FDISK_DOS16B,"Windows FAT16" },
1926{ FDISK_DOS16S,"Windows FAT16" },
1927{ FDISK_DOS16SLBA,"Windows FAT16" },
1928{ FDISK_SMALLFAT32,"Windows FAT32" },
1929{ FDISK_FAT32,"Windows FAT32" },
1930{ FDISK_FREEBSD,"FreeBSD" },
1931{ FDISK_OPENBSD,"OpenBSD" },
1932{ FDISK_LINUX,"Linux" },
1933{ FDISK_UFS,"Apple UFS" },
1934{ FDISK_HFS,"Apple HFS" },
1935{ FDISK_BOOTER,"Apple Boot/UFS" },
1936{ FDISK_BEFS,"Haiku" },
1937{ 0xCD,"CD-ROM" },
1938{ 0x00,0 } /* must be last */
1939};
1940
1941//==============================================================================
1942
1943bool matchVolumeToString( BVRef bvr, const char* match, long matchLen)
1944{
1945char testStr[128];
1946
1947if ( !bvr || !match || !*match)
1948{
1949return 0;
1950}
1951
1952if ( bvr->biosdev < 0x80 || bvr->biosdev >= 0x100 )
1953{
1954 return 0;
1955}
1956
1957// Try to match hd(x,y) first.
1958sprintf(testStr, "hd(%d,%d)", BIOS_DEV_UNIT(bvr), bvr->part_no);
1959if ( matchLen ? !strncmp(match, testStr, matchLen) : !strcmp(match, testStr) )
1960{
1961return true;
1962}
1963
1964// Try to match volume UUID.
1965if ( bvr->fs_getuuid && bvr->fs_getuuid(bvr, testStr) == 0)
1966{
1967if ( matchLen ? !strncmp(match, testStr, matchLen) : !strcmp(match, testStr) )
1968{
1969return true;
1970}
1971}
1972
1973// Try to match volume label (always quoted).
1974if ( bvr->description )
1975{
1976bvr->description(bvr, testStr, sizeof(testStr)-1);
1977if ( matchLen ? !strncmp(match, testStr, matchLen) : !strcmp(match, testStr) )
1978{
1979return true;
1980}
1981}
1982
1983return false;
1984}
1985
1986//==============================================================================
1987
1988/* If Rename Partition has defined an alias, then extract it for description purpose.
1989 * The format for the rename string is the following:
1990 * hd(x,y)|uuid|"label" "alias";hd(m,n)|uuid|"label" "alias"; etc...
1991 */
1992
1993bool getVolumeLabelAlias(BVRef bvr, char* str, long strMaxLen)
1994{
1995char *aliasList, *entryStart, *entryNext;
1996
1997if ( !str || strMaxLen <= 0)
1998{
1999return false;
2000}
2001
2002aliasList = XMLDecode(getStringForKey(kRenamePartition, &bootInfo->chameleonConfig));
2003if ( !aliasList )
2004{
2005return false;
2006}
2007
2008for ( entryStart = entryNext = aliasList; entryNext && *entryNext; entryStart = entryNext )
2009{
2010char *volStart, *volEnd, *aliasStart;
2011long volLen, aliasLen;
2012
2013// Delimit current entry
2014entryNext = strchr(entryStart, ';');
2015if ( entryNext )
2016{
2017*entryNext = '\0';
2018entryNext++;
2019}
2020
2021volStart = strbreak(entryStart, &volEnd, &volLen);
2022if(!volLen)
2023{
2024continue;
2025}
2026
2027aliasStart = strbreak(volEnd, 0, &aliasLen);
2028if(!aliasLen)
2029{
2030continue;
2031}
2032
2033if ( matchVolumeToString(bvr, volStart, volLen) )
2034{
2035strncat(str, aliasStart, MIN(strMaxLen, aliasLen));
2036free(aliasList);
2037
2038return true;
2039}
2040}
2041
2042free(aliasList);
2043return false;
2044}
2045
2046//==============================================================================
2047
2048void getBootVolumeDescription( BVRef bvr, char * str, long strMaxLen, bool useDeviceDescription )
2049{
2050unsigned char type;
2051char *p = str;
2052
2053if(!bvr || !p || strMaxLen <= 0)
2054{
2055return;
2056}
2057
2058type = (unsigned char) bvr->part_type;
2059
2060if (useDeviceDescription)
2061{
2062int len = getDeviceDescription(bvr, str);
2063if(len >= strMaxLen)
2064{
2065return;
2066}
2067
2068strcpy(str + len, bvr->OSisInstaller ? " (Installer) " : " ");
2069len += bvr->OSisInstaller ? 13 : 1;
2070strMaxLen -= len;
2071p += len;
2072}
2073
2074/* See if a partition rename is preferred */
2075if (getVolumeLabelAlias(bvr, p, strMaxLen))
2076{
2077strncpy(bvr->label, p, strMaxLen);
2078return; // we're done here no need to seek for real name
2079}
2080
2081// Get the volume label using filesystem specific functions or use the alternate volume label if available.
2082
2083if (*bvr->altlabel != '\0')
2084{
2085strncpy(p, bvr->altlabel, strMaxLen);
2086}
2087else if (bvr->description)
2088{
2089bvr->description(bvr, p, strMaxLen);
2090}
2091
2092if (*p == '\0')
2093{
2094const char * name = getNameForValue( fdiskTypes, type );
2095
2096if (name == NULL)
2097{
2098name = bvr->type_name;
2099}
2100
2101if (name == NULL)
2102{
2103sprintf(p, "TYPE %02x", type);
2104}
2105else
2106{
2107strncpy(p, name, strMaxLen);
2108}
2109}
2110
2111// Set the devices label
2112sprintf(bvr->label, p);
2113}
2114
2115
2116//==============================================================================
2117
2118int readBootSector(int biosdev, unsigned int secno, void * buffer)
2119{
2120int error;
2121struct disk_blk0 * bootSector = (struct disk_blk0 *) buffer;
2122
2123if (bootSector == NULL)
2124{
2125if (gBootSector == NULL)
2126{
2127gBootSector = (struct disk_blk0 *) malloc(sizeof(*gBootSector));
2128
2129if (gBootSector == NULL)
2130{
2131return -1;
2132}
2133}
2134
2135bootSector = gBootSector;
2136}
2137
2138error = readBytes(biosdev, secno, 0, BPS, bootSector);
2139
2140if (error || bootSector->signature != DISK_SIGNATURE)
2141{
2142return -1;
2143}
2144return 0;
2145}
2146
2147//==============================================================================
2148
2149/*
2150 * Format of boot1f32 block.
2151 */
2152
2153#define BOOT1F32_MAGIC "BOOT "
2154#define BOOT1F32_MAGICLEN 11
2155
2156struct disk_boot1f32_blk
2157{
2158unsigned char init[3];
2159unsigned char fsheader[87];
2160unsigned char magic[BOOT1F32_MAGICLEN];
2161unsigned char bootcode[409];
2162unsigned short signature;
2163};
2164
2165//==============================================================================
2166
2167int testFAT32EFIBootSector( int biosdev, unsigned int secno, void * buffer )
2168{
2169struct disk_boot1f32_blk * bootSector = (struct disk_boot1f32_blk *) buffer;
2170int error;
2171
2172if ( bootSector == NULL )
2173{
2174if ( gBootSector == NULL )
2175{
2176gBootSector = (struct disk_blk0 *) malloc(sizeof(*gBootSector));
2177if ( gBootSector == NULL )
2178{
2179return -1;
2180}
2181}
2182bootSector = (struct disk_boot1f32_blk *) gBootSector;
2183}
2184
2185error = readBytes( biosdev, secno, 0, BPS, bootSector );
2186if ( error || bootSector->signature != DISK_SIGNATURE || strncmp((const char *)bootSector->magic, BOOT1F32_MAGIC, BOOT1F32_MAGICLEN) )
2187{
2188return -1;
2189}
2190return 0;
2191}
2192
2193
2194//==============================================================================
2195// Handle seek request from filesystem modules.
2196
2197void diskSeek(BVRef bvr, long long position)
2198{
2199bvr->fs_boff = position / BPS;
2200bvr->fs_byteoff = position % BPS;
2201}
2202
2203
2204//==============================================================================
2205// Handle read request from filesystem modules.
2206
2207int diskRead(BVRef bvr, long addr, long length)
2208{
2209return readBytes(bvr->biosdev, bvr->fs_boff + bvr->part_boff, bvr->fs_byteoff, length, (void *) addr);
2210}
2211
2212//==============================================================================
2213
2214int rawDiskRead( BVRef bvr, unsigned int secno, void *buffer, unsigned int len )
2215{
2216int secs;
2217unsigned char *cbuf = (unsigned char *)buffer;
2218unsigned int copy_len;
2219int rc;
2220
2221if ((len & (BPS-1)) != 0)
2222{
2223error("raw disk read not sector aligned");
2224return -1;
2225}
2226secno += bvr->part_boff;
2227
2228cache_valid = false;
2229
2230while (len > 0)
2231{
2232secs = len / BPS;
2233if (secs > N_CACHE_SECS)
2234{
2235secs = N_CACHE_SECS;
2236}
2237copy_len = secs * BPS;
2238
2239//printf("rdr: ebiosread(%d, %d, %d)\n", bvr->biosdev, secno, secs);
2240if ((rc = ebiosread(bvr->biosdev, secno, secs)) != 0)
2241{
2242/* Ignore corrected ECC errors */
2243if (rc != ECC_CORRECTED_ERR)
2244{
2245error(" EBIOS read error: %s\n", bios_error(rc), rc);
2246error(" Block %d Sectors %d\n", secno, secs);
2247return rc;
2248}
2249}
2250bcopy( trackbuf, cbuf, copy_len );
2251len -= copy_len;
2252cbuf += copy_len;
2253secno += secs;
2254spinActivityIndicator(secs);
2255}
2256
2257return 0;
2258}
2259
2260//==============================================================================
2261
2262int rawDiskWrite( BVRef bvr, unsigned int secno, void *buffer, unsigned int len )
2263{
2264 int secs;
2265 unsigned char *cbuf = (unsigned char *)buffer;
2266 unsigned int copy_len;
2267 int rc;
2268
2269if ((len & (BPS-1)) != 0)
2270{
2271error("raw disk write not sector aligned");
2272return -1;
2273}
2274secno += bvr->part_boff;
2275
2276cache_valid = false;
2277
2278while (len > 0)
2279{
2280secs = len / BPS;
2281if (secs > N_CACHE_SECS)
2282{
2283secs = N_CACHE_SECS;
2284}
2285copy_len = secs * BPS;
2286
2287bcopy( cbuf, trackbuf, copy_len );
2288//printf("rdr: ebioswrite(%d, %d, %d)\n", bvr->biosdev, secno, secs);
2289if ((rc = ebioswrite(bvr->biosdev, secno, secs)) != 0)
2290{
2291error(" EBIOS write error: %s\n", bios_error(rc), rc);
2292error(" Block %d Sectors %d\n", secno, secs);
2293return rc;
2294}
2295
2296len -= copy_len;
2297cbuf += copy_len;
2298secno += secs;
2299spinActivityIndicator(secs);
2300}
2301
2302return 0;
2303}
2304
2305//==============================================================================
2306
2307int diskIsCDROM(BVRef bvr)
2308{
2309struct driveInfo di;
2310
2311if (getDriveInfo(bvr->biosdev, &di) == 0 && di.no_emulation)
2312{
2313return 1;
2314}
2315return 0;
2316}
2317
2318//==============================================================================
2319
2320int biosDevIsCDROM(int biosdev)
2321{
2322struct driveInfo di;
2323
2324if (getDriveInfo(biosdev, &di) == 0 && di.no_emulation)
2325{
2326return 1;
2327}
2328return 0;
2329}
2330

Archive Download this file

Revision: 2381