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 } }; // 0xAF00 "Apple HFS/HFS+"
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 } }; // 0xAB00 "Apple boot"
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 } }; // 0xEF00 "EFI System"
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 } }; // 0x0100 "Microsoft basic data"
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 } }; // 0x0C01 "Microsoft reserved"
697
698// Apple OSX
699//EFI_GUID const GPT_UFS_GUID= { 0x55465300, 0x0000, 0x11AA, { 0xAA, 0x11, 0x00, 0x30, 0x65, 0x43, 0xEC, 0xAC } }; // 0xA800 "Apple UFS"
700//EFI_GUID const GPT_RAID_GUID= { 0x52414944, 0x0000, 0x11AA, { 0xAA, 0x11, 0x00, 0x30, 0x65, 0x43, 0xEC, 0xAC } }; // 0xAF01 "Apple RAID"
701//EFI_GUID const GPT_RAID_OFFLINE_GUID= { 0x52414944, 0x5f4f, 0x11AA, { 0xAA, 0x11, 0x00, 0x30, 0x65, 0x43, 0xEC, 0xAC } }; // 0xAF02 "Apple RAID offline"
702//EFI_GUID const GPT_LABEL_GUID= { 0x4C616265, 0x6C00, 0x11AA, { 0xAA, 0x11, 0x00, 0x30, 0x65, 0x43, 0xEC, 0xAC } }; // 0xAF03 "Apple label"
703//EFI_GUID const GPT_APPLETV_GUID= { 0x5265636F, 0x7665, 0x11AA, { 0xAA, 0x11, 0x00, 0x30, 0x65, 0x43, 0xEC, 0xAC } }; // 0xAF04 "Apple TV recovery"
704//EFI_GUID const GPT_CORESTORAGE_GUID= { 0x53746F72, 0x6167, 0x11AA, { 0xAA, 0x11, 0x00, 0x30, 0x65, 0x43, 0xEC, 0xAC } }; // 0xAF05 "Apple Core storage"
705// same as Apple ZFS
706//EFI_GUID const GPT_ZFS_GUID= { 0x6A898CC3, 0x1DD2, 0x11B2, { 0x99, 0xA6, 0x08, 0x00, 0x20, 0x73, 0x66, 0x31 } }; // 0xBF01 "Solaris /usr & Apple ZFS
707
708BVRef newGPTBVRef( int biosdev, int partno, unsigned int blkoff,
709 const gpt_ent * part,
710 FSInit initFunc, FSLoadFile loadFunc,
711 FSReadFile readFunc,
712 FSGetDirEntry getdirFunc,
713 FSGetFileBlock getBlockFunc,
714 FSGetUUID getUUIDFunc,
715 BVGetDescription getDescriptionFunc,
716 BVFree bvFreeFunc,
717 int probe, int type, unsigned int bvrFlags )
718{
719BVRef bvr = (BVRef) malloc( sizeof(*bvr) );
720if ( bvr )
721{
722bzero(bvr, sizeof(*bvr));
723
724bvr->biosdev = biosdev;
725bvr->part_no = partno;
726bvr->part_boff = blkoff;
727bvr->fs_loadfile = loadFunc;
728bvr->fs_readfile = readFunc;
729bvr->fs_getdirentry = getdirFunc;
730bvr->fs_getfileblock= getBlockFunc;
731bvr->fs_getuuid = getUUIDFunc;
732bvr->description = getDescriptionFunc;
733bvr->type = type;
734bvr->bv_free = bvFreeFunc;
735// FIXME: UCS-2 -> UTF-8 the name
736strlcpy(bvr->name, "----", DPISTRLEN);
737if ( (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) )
738{
739strlcpy(bvr->type_name, "GPT HFS+", DPISTRLEN);
740}
741else
742{
743strlcpy(bvr->type_name, "GPT Unknown", DPISTRLEN);
744}
745
746/*
747if ( part->bootid & FDISK_ACTIVE )
748{
749bvr->flags |= kBVFlagPrimary;
750}
751*/
752
753// Probe the filesystem.
754
755if ( initFunc )
756{
757bvr->flags |= kBVFlagNativeBoot;
758
759if ( probe && initFunc( bvr ) != 0 )
760{
761// filesystem probe failed.
762
763DEBUG_DISK(("%s: failed probe on dev %x part %d\n", __FUNCTION__, biosdev, partno));
764
765(*bvr->bv_free)(bvr);
766bvr = NULL;
767}
768if ( readBootSector( biosdev, blkoff, (void *)0x7e00 ) == 0 )
769{
770bvr->flags |= kBVFlagBootable;
771}
772}
773else if ( readBootSector( biosdev, blkoff, (void *)0x7e00 ) == 0 )
774{
775bvr->flags |= kBVFlagForeignBoot;
776}
777else
778{
779(*bvr->bv_free)(bvr);
780bvr = NULL;
781}
782}
783if (bvr)
784{
785bvr->flags |= bvrFlags;
786}
787return bvr;
788}
789
790//==============================================================================
791
792/* A note on partition numbers:
793 * IOKit makes the primary partitions numbers 1-4, and then
794 * extended partitions are numbered consecutively 5 and up.
795 * So, for example, if you have two primary partitions and
796 * one extended partition they will be numbered 1, 2, 5.
797 */
798
799static BVRef diskScanFDiskBootVolumes( int biosdev, int * countPtr )
800{
801 const struct fdisk_part * part;
802 struct DiskBVMap * map;
803 int partno = -1;
804 BVRef bvr;
805#if UFS_SUPPORT
806 BVRef booterUFS = NULL;
807#endif
808 int spc;
809 struct driveInfo di;
810 boot_drive_info_t *dp;
811
812/* Initialize disk info */
813
814if (getDriveInfo(biosdev, &di) != 0)
815{
816return NULL;
817}
818
819dp = &di.di;
820spc = (dp->params.phys_spt * dp->params.phys_heads);
821
822if (spc == 0)
823{
824/* This is probably a CD-ROM; punt on the geometry. */
825spc = 1;
826}
827
828do {
829// Create a new mapping.
830
831map = (struct DiskBVMap *) malloc( sizeof(*map) );
832if ( map )
833 {
834map->biosdev = biosdev;
835map->bvr = NULL;
836map->bvrcnt = 0;
837map->next = gDiskBVMap;
838gDiskBVMap = map;
839
840// Create a record for each partition found on the disk.
841
842while ( getNextFDiskPartition( biosdev, &partno, &part ) )
843{
844DEBUG_DISK(("%s: part %d [%x]\n", __FUNCTION__,
845partno, part->systid));
846bvr = 0;
847
848switch ( part->systid )
849{
850#if UFS_SUPPORT
851case FDISK_UFS:
852bvr = newFDiskBVRef(
853biosdev, partno,
854part->relsect + UFS_FRONT_PORCH/BPS,
855part,
856UFSInitPartition,
857UFSLoadFile,
858UFSReadFile,
859UFSGetDirEntry,
860UFSGetFileBlock,
861UFSGetUUID,
862UFSGetDescription,
863UFSFree,
8640,
865kBIOSDevTypeHardDrive, 0);
866break;
867#endif
868
869 case FDISK_HFS:
870 bvr = newFDiskBVRef(
871 biosdev, partno,
872 part->relsect,
873 part,
874 HFSInitPartition,
875 HFSLoadFile,
876 HFSReadFile,
877 HFSGetDirEntry,
878 HFSGetFileBlock,
879 HFSGetUUID,
880 HFSGetDescription,
881 HFSFree,
882 0,
883 kBIOSDevTypeHardDrive, 0);
884 break;
885
886 // turbo - we want the booter type scanned also
887 case FDISK_BOOTER:
888 if (part->bootid & FDISK_ACTIVE)
889 gBIOSBootVolume = newFDiskBVRef(
890 biosdev, partno,
891 part->relsect,
892 part,
893 HFSInitPartition,
894 HFSLoadFile,
895 HFSReadFile,
896 HFSGetDirEntry,
897 HFSGetFileBlock,
898 HFSGetUUID,
899 HFSGetDescription,
900 HFSFree,
901 0,
902 kBIOSDevTypeHardDrive, 0);
903 break;
904
905#if UFS_SUPPORT
906case FDISK_BOOTER:
907booterUFS = newFDiskBVRef(
908 biosdev, partno,
909 ((part->relsect + spc - 1) / spc) * spc,
910 part,
911 UFSInitPartition,
912 UFSLoadFile,
913 UFSReadFile,
914 UFSGetDirEntry,
915 UFSGetFileBlock,
916 UFSGetUUID,
917 UFSGetDescription,
918 UFSFree,
919 0,
920 kBIOSDevTypeHardDrive, 0);
921break;
922#endif
923
924 case FDISK_FAT32:
925 case FDISK_DOS12:
926 case FDISK_DOS16S:
927 case FDISK_DOS16B:
928 case FDISK_SMALLFAT32:
929 case FDISK_DOS16SLBA:
930 bvr = newFDiskBVRef(
931 biosdev, partno,
932 part->relsect,
933 part,
934 MSDOSInitPartition,
935 MSDOSLoadFile,
936 MSDOSReadFile,
937 MSDOSGetDirEntry,
938 MSDOSGetFileBlock,
939 MSDOSGetUUID,
940 MSDOSGetDescription,
941 MSDOSFree,
942 0,
943 kBIOSDevTypeHardDrive, 0);
944 break;
945
946 case FDISK_NTFS:
947 bvr = newFDiskBVRef(
948 biosdev, partno,
949 part->relsect,
950 part,
951 0, 0, 0, 0, 0,
952 NTFSGetUUID,
953 NTFSGetDescription,
954 (BVFree)free,
955 0, kBIOSDevTypeHardDrive, 0);
956 break;
957
958 case FDISK_LINUX:
959 bvr = newFDiskBVRef(
960 biosdev, partno,
961 part->relsect,
962 part,
963 0, 0, 0, 0, 0,
964 EX2GetUUID,
965 EX2GetDescription,
966 (BVFree)free,
967 0, kBIOSDevTypeHardDrive, 0);
968 break;
969
970 case FDISK_BEFS:
971 bvr = newFDiskBVRef(
972 biosdev, partno,
973 part->relsect,
974 part,
975 0, 0, 0, 0, 0, 0,
976 BeFSGetDescription,
977 (BVFree)free,
978 0, kBIOSDevTypeHardDrive, 0);
979 break;
980
981 case FDISK_FREEBSD:
982 bvr = newFDiskBVRef(
983 biosdev, partno,
984 part->relsect,
985 part,
986 0, 0, 0, 0, 0, 0,
987 FreeBSDGetDescription,
988 (BVFree)free,
989 0, kBIOSDevTypeHardDrive, 0);
990 break;
991
992 case FDISK_OPENBSD:
993 bvr = newFDiskBVRef(
994 biosdev, partno,
995 part->relsect,
996 part,
997 0, 0, 0, 0, 0, 0,
998 OpenBSDGetDescription,
999 (BVFree)free,
1000 0, kBIOSDevTypeHardDrive, 0);
1001 break;
1002
1003 default:
1004 bvr = newFDiskBVRef(
1005 biosdev, partno,
1006 part->relsect,
1007 part,
1008 0, 0, 0, 0, 0, 0, 0,
1009 (BVFree)free,
1010 0,
1011 kBIOSDevTypeHardDrive, 0);
1012 break;
1013 }
1014
1015 if ( bvr )
1016 {
1017 bvr->next = map->bvr;
1018 map->bvr = bvr;
1019 map->bvrcnt++;
1020 }
1021 }
1022
1023#if UFS_SUPPORT
1024 // Booting from a CD with an UFS filesystem embedded
1025 // in a booter partition.
1026
1027if ( booterUFS )
1028{
1029if ( map->bvrcnt == 0 )
1030{
1031map->bvr = booterUFS;
1032map->bvrcnt++;
1033}
1034else
1035{
1036free( booterUFS );
1037}
1038}
1039#endif
1040}
1041} while (0);
1042
1043/*
1044 * If no FDisk partition, then we will check for
1045 * an Apple partition map elsewhere.
1046 */
1047#if UNUSED
1048if (map->bvrcnt == 0)
1049{
1050static struct fdisk_part cdpart;
1051cdpart.systid = 0xCD;
1052
1053/* Let's try assuming we are on a hybrid HFS/ISO9660 CD. */
1054bvr = newFDiskBVRef(
1055biosdev, 0,
10560,
1057&cdpart,
1058HFSInitPartition,
1059HFSLoadFile,
1060HFSReadFile,
1061HFSGetDirEntry,
1062HFSGetFileBlock,
1063HFSGetUUID,
1064HFSGetDescription,
1065HFSFree,
10660,
1067kBIOSDevTypeHardDrive, 0);
1068bvr->next = map->bvr;
1069map->bvr = bvr;
1070map->bvrcnt++;
1071}
1072#endif
1073// Actually this should always be true given the above code
1074if(map == gDiskBVMap)
1075{
1076// Don't leave a null map in the chain
1077if(map->bvrcnt == 0 && map->bvr == NULL)
1078{
1079gDiskBVMap = map->next;
1080free(map);
1081map = NULL;
1082}
1083}
1084
1085if (countPtr) *countPtr = map ? map->bvrcnt : 0;
1086
1087return map ? map->bvr : NULL;
1088}
1089
1090//==============================================================================
1091
1092static BVRef diskScanAPMBootVolumes( int biosdev, int * countPtr )
1093{
1094struct DiskBVMap * map;
1095struct Block0 *block0_p;
1096unsigned int blksize;
1097unsigned int factor;
1098void *buffer = malloc(BPS);
1099
1100if (!buffer)
1101{
1102return NULL;
1103}
1104bzero(buffer,BPS);
1105
1106/* Check for alternate block size */
1107if (readBytes( biosdev, 0, 0, BPS, buffer ) != 0)
1108{
1109return NULL;
1110}
1111block0_p = buffer;
1112if (OSSwapBigToHostInt16(block0_p->sbSig) == BLOCK0_SIGNATURE)
1113{
1114blksize = OSSwapBigToHostInt16(block0_p->sbBlkSize);
1115if (blksize != BPS)
1116{
1117free(buffer);
1118buffer = malloc(blksize);
1119if (!buffer)
1120{
1121return NULL;
1122}
1123bzero(buffer,BPS);
1124}
1125factor = blksize / BPS;
1126}
1127else
1128{
1129blksize = BPS;
1130factor = 1;
1131}
1132
1133do
1134{
1135// Create a new mapping.
1136
1137map = (struct DiskBVMap *) malloc( sizeof(*map) );
1138if ( map )
1139{
1140int error;
1141DPME *dpme_p = (DPME *)buffer;
1142UInt32 i, npart = UINT_MAX;
1143BVRef bvr;
1144
1145map->biosdev = biosdev;
1146map->bvr = NULL;
1147map->bvrcnt = 0;
1148map->next = gDiskBVMap;
1149gDiskBVMap = map;
1150
1151for (i=0; i<npart; i++)
1152{
1153error = readBytes( biosdev, (kAPMSector + i) * factor, 0, blksize, buffer );
1154
1155if (error || OSSwapBigToHostInt16(dpme_p->dpme_signature) != DPME_SIGNATURE)
1156{
1157break;
1158}
1159
1160if (i==0)
1161{
1162npart = OSSwapBigToHostInt32(dpme_p->dpme_map_entries);
1163}
1164/*
1165printf("name = %s, %s%s %d -> %d [%d -> %d] {%d}\n",
1166dpme.dpme_name, dpme.dpme_type, (dpme.dpme_flags & DPME_FLAGS_BOOTABLE) ? "(bootable)" : "",
1167dpme.dpme_pblock_start, dpme.dpme_pblocks,
1168dpme.dpme_lblock_start, dpme.dpme_lblocks,
1169dpme.dpme_boot_block);
1170*/
1171
1172if (strcmp(dpme_p->dpme_type, "Apple_HFS") == 0)
1173{
1174bvr = newAPMBVRef(biosdev,
1175i,
1176OSSwapBigToHostInt32(dpme_p->dpme_pblock_start) * factor,
1177dpme_p,
1178HFSInitPartition,
1179HFSLoadFile,
1180HFSReadFile,
1181HFSGetDirEntry,
1182HFSGetFileBlock,
1183HFSGetUUID,
1184HFSGetDescription,
1185HFSFree,
11860,
1187kBIOSDevTypeHardDrive, 0);
1188bvr->next = map->bvr;
1189map->bvr = bvr;
1190map->bvrcnt++;
1191}
1192}
1193}
1194} while (0);
1195
1196free(buffer);
1197
1198if (countPtr) *countPtr = map ? map->bvrcnt : 0;
1199
1200return map ? map->bvr : NULL;
1201}
1202
1203//==============================================================================
1204
1205/*
1206 * Trying to figure out the filsystem type of a given partition.
1207 */
1208static int probeFileSystem(int biosdev, unsigned int blkoff)
1209{
1210// detected filesystem type;
1211int result = -1;
1212int fatbits;
1213
1214// Allocating buffer for 4 sectors.
1215const void * probeBuffer = malloc(PROBEFS_SIZE);
1216if (probeBuffer == NULL)
1217{
1218goto exit;
1219}
1220
1221// Reading first 4 sectors of current partition
1222int error = readBytes(biosdev, blkoff, 0, PROBEFS_SIZE, (void *)probeBuffer);
1223
1224if (error)
1225{
1226goto exit;
1227}
1228
1229if (HFSProbe(probeBuffer))
1230{
1231result = FDISK_HFS;
1232}
1233else if (EX2Probe(probeBuffer))
1234{
1235result = FDISK_LINUX;
1236}
1237else if (FreeBSDProbe(probeBuffer))
1238{
1239result = FDISK_FREEBSD;
1240}
1241
1242else if (OpenBSDProbe(probeBuffer))
1243{
1244result = FDISK_OPENBSD;
1245}
1246
1247else if (BeFSProbe(probeBuffer))
1248{
1249result = FDISK_BEFS;
1250}
1251
1252else if (NTFSProbe(probeBuffer))
1253{
1254result = FDISK_NTFS;
1255}
1256
1257else if ( (fatbits = MSDOSProbe(probeBuffer)) )
1258{
1259switch (fatbits)
1260{
1261case 32:
1262default:
1263result = FDISK_FAT32;
1264break;
1265case 16:
1266result = FDISK_DOS16B;
1267break;
1268case 12:
1269result = FDISK_DOS12;
1270break;
1271}
1272}
1273else
1274{
1275// Couldn't detect filesystem type
1276result = 0;
1277}
1278
1279exit:
1280if (probeBuffer != NULL) free((void *)probeBuffer);
1281{
1282return result;
1283}
1284}
1285
1286//==============================================================================
1287
1288static bool isPartitionUsed(gpt_ent * partition)
1289{
1290
1291// Ask whether the given partition is used.
1292
1293return efi_guid_is_null((EFI_GUID const*)partition->ent_type) ? false : true;
1294}
1295
1296//==============================================================================
1297
1298static BVRef diskScanGPTBootVolumes(int biosdev, int * countPtr)
1299{
1300struct DiskBVMap *map = NULL;
1301
1302void *buffer = malloc(BPS);
1303
1304int error;
1305if ( (error = readBytes( biosdev, /*secno*/0, 0, BPS, buffer )) != 0)
1306{
1307verbose("Failed to read boot sector from BIOS device %02xh. Error=%d\n", biosdev, error);
1308goto scanErr;
1309}
1310struct REAL_disk_blk0 *fdiskMap = buffer;
1311if ( OSSwapLittleToHostInt16(fdiskMap->signature) != DISK_SIGNATURE )
1312{
1313verbose("Failed to find boot signature on BIOS device %02xh\n", biosdev);
1314goto scanErr;
1315}
1316
1317int fdiskID = 0;
1318unsigned index;
1319for ( index = 0; index < FDISK_NPART; index++ )
1320{
1321if ( fdiskMap->parts[index].systid )
1322{
1323if ( fdiskMap->parts[index].systid == 0xEE )
1324{
1325// Fail if two 0xEE partitions are present which
1326// means the FDISK code will wind up parsing it.
1327if ( fdiskID )
1328{
1329goto scanErr;
1330}
1331
1332fdiskID = index + 1;
1333}
1334}
1335}
1336
1337if ( fdiskID == 0 )
1338{
1339goto scanErr;
1340}
1341
1342verbose("Attempting to read GPT\n");
1343
1344if(readBytes(biosdev, 1, 0, BPS, buffer) != 0)
1345{
1346goto scanErr;
1347}
1348
1349gpt_hdr *headerMap = buffer;
1350
1351// Determine whether the partition header signature is present.
1352
1353if ( memcmp(headerMap->hdr_sig, GPT_HDR_SIG, strlen(GPT_HDR_SIG)) )
1354{
1355goto scanErr;
1356}
1357
1358// Determine whether the partition header size is valid.
1359
1360UInt32 headerCheck = OSSwapLittleToHostInt32(headerMap->hdr_crc_self);
1361UInt32 headerSize = OSSwapLittleToHostInt32(headerMap->hdr_size);
1362
1363if ( headerSize < offsetof(gpt_hdr, padding) )
1364{
1365goto scanErr;
1366}
1367
1368if ( headerSize > BPS )
1369{
1370goto scanErr;
1371}
1372
1373// Determine whether the partition header checksum is valid.
1374
1375headerMap->hdr_crc_self = 0;
1376
1377if ( crc32(0, headerMap, headerSize) != headerCheck )
1378{
1379goto scanErr;
1380}
1381
1382// Determine whether the partition entry size is valid.
1383
1384UInt64 gptBlock = 0;
1385UInt32 gptCheck = 0;
1386UInt32 gptCount = 0;
1387UInt32 gptID = 0;
1388gpt_ent * gptMap = 0;
1389UInt32 gptSize = 0;
1390
1391gptBlock = OSSwapLittleToHostInt64(headerMap->hdr_lba_table);
1392gptCheck = OSSwapLittleToHostInt32(headerMap->hdr_crc_table);
1393gptCount = OSSwapLittleToHostInt32(headerMap->hdr_entries);
1394gptSize = OSSwapLittleToHostInt32(headerMap->hdr_entsz);
1395
1396if ( gptSize < sizeof(gpt_ent) )
1397{
1398goto scanErr;
1399}
1400
1401// Allocate a buffer large enough to hold one map, rounded to a media block.
1402free(buffer);
1403buffer = NULL;
1404
1405UInt32 bufferSize = IORound(gptCount * gptSize, BPS);
1406if (bufferSize == 0)
1407{
1408goto scanErr;
1409}
1410buffer = malloc(bufferSize);
1411if (!buffer)
1412{
1413 goto scanErr;
1414}
1415
1416if (readBytes(biosdev, gptBlock, 0, bufferSize, buffer) != 0)
1417{
1418goto scanErr;
1419}
1420verbose("Read GPT\n");
1421
1422// Allocate a new map for this BIOS device and insert it into the chain
1423map = malloc(sizeof(*map));
1424if (!map)
1425{
1426goto scanErr;
1427}
1428map->biosdev = biosdev;
1429map->bvr = NULL;
1430map->bvrcnt = 0;
1431map->next = gDiskBVMap;
1432gDiskBVMap = map;
1433
1434// fdisk like partition type id.
1435int fsType = 0;
1436
1437for(gptID = 1; gptID <= gptCount; ++gptID) {
1438BVRef bvr = NULL;
1439unsigned int bvrFlags = 0;
1440
1441// size on disk can be larger than sizeof(gpt_ent)
1442gptMap = (gpt_ent *) ( buffer + ( (gptID - 1) * gptSize) );
1443
1444// NOTE: EFI_GUID's are in LE and we know we're on an x86.
1445// The IOGUIDPartitionScheme.cpp code uses byte-based UUIDs, we don't.
1446
1447if (isPartitionUsed(gptMap)) {
1448char stringuuid[100];
1449efi_guid_unparse_upper((EFI_GUID*)gptMap->ent_type, stringuuid);
1450verbose("Reading GPT partition %d, type %s\n", gptID, stringuuid);
1451
1452// Getting fdisk like partition type.
1453fsType = probeFileSystem(biosdev, gptMap->ent_lba_start);
1454
1455if ( (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) ) {
1456bvrFlags = (efi_guid_compare(&GPT_BOOT_GUID, (EFI_GUID const*)gptMap->ent_type) == 0) ? kBVFlagBooter : 0;
1457bvr = newGPTBVRef(biosdev,
1458gptID,
1459gptMap->ent_lba_start,
1460gptMap,
1461HFSInitPartition,
1462HFSLoadFile,
1463HFSReadFile,
1464HFSGetDirEntry,
1465HFSGetFileBlock,
1466HFSGetUUID,
1467HFSGetDescription,
1468HFSFree,
14690,
1470kBIOSDevTypeHardDrive, bvrFlags);
1471}
1472
1473// zef - foreign OS support
1474if ( (efi_guid_compare(&GPT_BASICDATA_GUID, (EFI_GUID const*)gptMap->ent_type) == 0) ||
1475(efi_guid_compare(&GPT_BASICDATA2_GUID, (EFI_GUID const*)gptMap->ent_type) == 0) ) {
1476switch (fsType)
1477{
1478case FDISK_NTFS:
1479bvr = newGPTBVRef(biosdev, gptID, gptMap->ent_lba_start, gptMap,
14800, 0, 0, 0, 0, 0, NTFSGetDescription,
1481(BVFree)free, 0, kBIOSDevTypeHardDrive, 0);
1482break;
1483
1484case FDISK_LINUX:
1485bvr = newGPTBVRef(biosdev, gptID, gptMap->ent_lba_start, gptMap,
14860, 0, 0, 0, 0, 0, EX2GetDescription,
1487(BVFree)free, 0, kBIOSDevTypeHardDrive, 0);
1488break;
1489
1490default:
1491bvr = newGPTBVRef(biosdev, gptID, gptMap->ent_lba_start, gptMap,
14920, 0, 0, 0, 0, 0, 0,
1493(BVFree)free, 0, kBIOSDevTypeHardDrive, 0);
1494break;
1495}
1496
1497}
1498
1499// turbo - save our booter partition
1500// zef - only on original boot device
1501if ( (efi_guid_compare(&GPT_EFISYS_GUID, (EFI_GUID const*)gptMap->ent_type) == 0) ) {
1502switch (fsType) {
1503case FDISK_HFS:
1504if (readBootSector( biosdev, gptMap->ent_lba_start, (void *)0x7e00 ) == 0) {
1505bvr = newGPTBVRef(biosdev, gptID, gptMap->ent_lba_start, gptMap,
1506HFSInitPartition,
1507HFSLoadFile,
1508HFSReadFile,
1509HFSGetDirEntry,
1510HFSGetFileBlock,
1511HFSGetUUID,
1512HFSGetDescription,
1513HFSFree,
15140, kBIOSDevTypeHardDrive, kBVFlagEFISystem);
1515}
1516break;
1517
1518case FDISK_FAT32:
1519if (testFAT32EFIBootSector( biosdev, gptMap->ent_lba_start, (void *)0x7e00 ) == 0) {
1520bvr = newGPTBVRef(biosdev, gptID, gptMap->ent_lba_start, gptMap,
1521MSDOSInitPartition,
1522MSDOSLoadFile,
1523MSDOSReadFile,
1524MSDOSGetDirEntry,
1525MSDOSGetFileBlock,
1526MSDOSGetUUID,
1527MSDOSGetDescription,
1528MSDOSFree,
15290, kBIOSDevTypeHardDrive, kBVFlagEFISystem);
1530}
1531break;
1532
1533default:
1534if (biosdev == gBIOSDev) {
1535gBIOSBootVolume = bvr;
1536}
1537break;
1538}
1539}
1540
1541if (bvr)
1542{
1543// Fixup bvr with the fake fdisk partition type.
1544if (fsType > 0) {
1545bvr->part_type = fsType;
1546}
1547
1548bvr->next = map->bvr;
1549map->bvr = bvr;
1550++map->bvrcnt;
1551}
1552
1553}
1554}
1555
1556scanErr:
1557if (buffer) {
1558free(buffer);
1559}
1560
1561if(map) {
1562if(countPtr) *countPtr = map->bvrcnt;
1563{
1564return map->bvr;
1565}
1566
1567} else {
1568if(countPtr) *countPtr = 0;
1569{
1570return NULL;
1571}
1572}
1573}
1574
1575//==============================================================================
1576
1577static bool getOSVersion(BVRef bvr, char *str)
1578{
1579bool valid = false;
1580config_file_t systemVersion;
1581char dirSpec[512];
1582
1583sprintf(dirSpec, "hd(%d,%d)/System/Library/CoreServices/SystemVersion.plist", BIOS_DEV_UNIT(bvr), bvr->part_no);
1584
1585if (!loadConfigFile(dirSpec, &systemVersion))
1586{
1587valid = true;
1588}
1589else
1590{
1591sprintf(dirSpec, "hd(%d,%d)/System/Library/CoreServices/ServerVersion.plist", BIOS_DEV_UNIT(bvr), bvr->part_no);
1592
1593if (!loadConfigFile(dirSpec, &systemVersion))
1594{
1595bvr->OSisServer = true;
1596valid = true;
1597}
1598}
1599
1600if (valid)
1601{
1602const char *val;
1603int len;
1604
1605if (getValueForKey(kProductVersion, &val, &len, &systemVersion))
1606{
1607// getValueForKey uses const char for val
1608// so copy it and trim
1609*str = '\0';
1610// crazybirdy
1611if (len > 4 && (val[3] == '1'))
1612{
1613strncat(str, val, MIN(len, 5));
1614}
1615else
1616{
1617strncat(str, val, MIN(len, 4));
1618}
1619}
1620else
1621{
1622valid = false;
1623}
1624}
1625
1626if(!valid)
1627{
1628int fh = -1;
1629sprintf(dirSpec, "hd(%d,%d)/.PhysicalMediaInstall", BIOS_DEV_UNIT(bvr), bvr->part_no);
1630fh = open(dirSpec, 0);
1631
1632if (fh >= 0)
1633{
1634valid = true;
1635bvr->OSisInstaller = true;
1636strcpy(bvr->OSVersion, "10.7"); // 10.7 +
1637close(fh);
1638}
1639else
1640{
1641close(fh);
1642}
1643}
1644return valid;
1645}
1646
1647//==============================================================================
1648
1649static void scanFSLevelBVRSettings(BVRef chain)
1650{
1651BVRef bvr;
1652char dirSpec[512], fileSpec[512];
1653char label[BVSTRLEN];
1654int ret;
1655long flags;
1656u_int32_t time;
1657int fh, fileSize, error;
1658
1659for (bvr = chain; bvr; bvr = bvr->next)
1660{
1661ret = -1;
1662error = 0;
1663
1664//
1665// Check for alternate volume label on boot helper partitions.
1666//
1667if (bvr->flags & kBVFlagBooter)
1668{
1669sprintf(dirSpec, "hd(%d,%d)/System/Library/CoreServices/", BIOS_DEV_UNIT(bvr), bvr->part_no);
1670strcpy(fileSpec, ".disk_label.contentDetails");
1671ret = GetFileInfo(dirSpec, fileSpec, &flags, &time);
1672if (!ret)
1673{
1674fh = open(strcat(dirSpec, fileSpec), 0);
1675fileSize = file_size(fh);
1676if (fileSize > 0 && fileSize < BVSTRLEN)
1677{
1678if (read(fh, label, fileSize) != fileSize)
1679{
1680error = -1;
1681}
1682}
1683else
1684{
1685error = -1;
1686}
1687
1688close(fh);
1689
1690if (!error)
1691{
1692label[fileSize] = '\0';
1693strcpy(bvr->altlabel, label);
1694}
1695}
1696}
1697
1698// Check for SystemVersion.plist or ServerVersion.plist to determine if a volume hosts an installed system.
1699
1700if (bvr->flags & kBVFlagNativeBoot)
1701{
1702if (getOSVersion(bvr,bvr->OSVersion) == true)
1703{
1704bvr->flags |= kBVFlagSystemVolume;
1705}
1706}
1707
1708}
1709}
1710
1711//==============================================================================
1712
1713void rescanBIOSDevice(int biosdev)
1714{
1715struct DiskBVMap *oldMap = diskResetBootVolumes(biosdev);
1716CacheReset();
1717diskFreeMap(oldMap);
1718oldMap = NULL;
1719scanBootVolumes(biosdev, 0);
1720}
1721
1722//==============================================================================
1723
1724struct DiskBVMap* diskResetBootVolumes(int biosdev)
1725{
1726struct DiskBVMap * map;
1727struct DiskBVMap *prevMap = NULL;
1728for ( map = gDiskBVMap; map; prevMap = map, map = map->next ) {
1729if ( biosdev == map->biosdev ) {
1730break;
1731}
1732}
1733
1734if(map != NULL) {
1735verbose("Resetting BIOS device %xh\n", biosdev);
1736// Reset the biosbuf cache
1737cache_valid = false;
1738if(map == gDiskBVMap)
1739{
1740gDiskBVMap = map->next;
1741}
1742else if(prevMap != NULL)
1743{
1744prevMap->next = map->next;
1745}
1746else
1747{
1748stop("");
1749}
1750}
1751// Return the old map, either to be freed, or reinserted later
1752return map;
1753}
1754
1755//==============================================================================
1756
1757// Frees a DiskBVMap and all of its BootVolume's
1758void diskFreeMap(struct DiskBVMap *map)
1759{
1760if(map != NULL)
1761{
1762while(map->bvr != NULL)
1763{
1764BVRef bvr = map->bvr;
1765map->bvr = bvr->next;
1766(*bvr->bv_free)(bvr);
1767}
1768
1769free(map);
1770}
1771}
1772
1773//==============================================================================
1774
1775BVRef diskScanBootVolumes(int biosdev, int * countPtr)
1776{
1777struct DiskBVMap *map;
1778BVRef bvr;
1779int count = 0;
1780
1781// Find an existing mapping for this device.
1782
1783for (map = gDiskBVMap; map; map = map->next)
1784{
1785if (biosdev == map->biosdev)
1786{
1787count = map->bvrcnt;
1788break;
1789}
1790}
1791
1792if (map == NULL)
1793{
1794bvr = diskScanGPTBootVolumes(biosdev, &count);
1795if (bvr == NULL)
1796{
1797bvr = diskScanFDiskBootVolumes(biosdev, &count);
1798}
1799if (bvr == NULL)
1800{
1801bvr = diskScanAPMBootVolumes(biosdev, &count);
1802}
1803if (bvr)
1804{
1805scanFSLevelBVRSettings(bvr);
1806}
1807}
1808else
1809{
1810bvr = map->bvr;
1811}
1812if (countPtr)
1813{
1814*countPtr += count;
1815}
1816return bvr;
1817}
1818
1819//==============================================================================
1820
1821BVRef getBVChainForBIOSDev(int biosdev)
1822{
1823BVRef chain = NULL;
1824struct DiskBVMap * map = NULL;
1825
1826for (map = gDiskBVMap; map; map = map->next)
1827{
1828if (map->biosdev == biosdev)
1829{
1830chain = map->bvr;
1831break;
1832}
1833}
1834
1835return chain;
1836}
1837
1838//==============================================================================
1839
1840BVRef newFilteredBVChain(int minBIOSDev, int maxBIOSDev, unsigned int allowFlags, unsigned int denyFlags, int *count)
1841{
1842BVRef chain = NULL;
1843BVRef bvr = NULL;
1844BVRef newBVR = NULL;
1845BVRef prevBVR = NULL;
1846
1847struct DiskBVMap * map = NULL;
1848int bvCount = 0;
1849
1850const char *raw = 0;
1851char* val = 0;
1852int len;
1853
1854getValueForKey(kHidePartition, &raw, &len, &bootInfo->chameleonConfig);
1855if(raw)
1856{
1857val = XMLDecode(raw);
1858}
1859
1860/*
1861 * Traverse gDISKBVmap to get references for
1862 * individual bvr chains of each drive.
1863 */
1864for (map = gDiskBVMap; map; map = map->next)
1865{
1866for (bvr = map->bvr; bvr; bvr = bvr->next)
1867{
1868/*
1869 * Save the last bvr.
1870 */
1871if (newBVR)
1872{
1873prevBVR = newBVR;
1874}
1875
1876/*
1877 * Allocate and copy the matched bvr entry into a new one.
1878 */
1879newBVR = (BVRef) malloc(sizeof(*newBVR));
1880if (!newBVR)
1881{
1882continue;
1883}
1884bcopy(bvr, newBVR, sizeof(*newBVR));
1885
1886/*
1887 * Adjust the new bvr's fields.
1888 */
1889newBVR->next = NULL;
1890newBVR->filtered = true;
1891
1892if ( (!allowFlags || newBVR->flags & allowFlags)
1893&& (!denyFlags || !(newBVR->flags & denyFlags) )
1894&& (newBVR->biosdev >= minBIOSDev && newBVR->biosdev <= maxBIOSDev)
1895)
1896{
1897newBVR->visible = true;
1898}
1899
1900/*
1901 * Looking for "Hide Partition" entries in 'hd(x,y)|uuid|"label" hd(m,n)|uuid|"label"' format,
1902 * to be able to hide foreign partitions from the boot menu.
1903 *
1904 */
1905if ( (newBVR->flags & kBVFlagForeignBoot) )
1906{
1907char *start, *next = val;
1908long len = 0;
1909do
1910{
1911start = strbreak(next, &next, &len);
1912if(len && matchVolumeToString(newBVR, start, len) )
1913{
1914newBVR->visible = false;
1915}
1916}
1917while ( next && *next );
1918}
1919
1920/*
1921 * Use the first bvr entry as the starting chain pointer.
1922 */
1923if (!chain)
1924{
1925chain = newBVR;
1926}
1927
1928/*
1929 * Update the previous bvr's link pointer to use the new memory area.
1930 */
1931if (prevBVR)
1932{
1933prevBVR->next = newBVR;
1934}
1935
1936if (newBVR->visible)
1937{
1938bvCount++;
1939}
1940}
1941}
1942
1943#if DEBUG //Azi: warning - too big for boot-log.. far too big.. i mean HUGE!! :P
1944for (bvr = chain; bvr; bvr = bvr->next)
1945{
1946printf(" bvr: %d, dev: %d, part: %d, flags: %d, vis: %d\n", bvr, bvr->biosdev, bvr->part_no, bvr->flags, bvr->visible);
1947}
1948printf("count: %d\n", bvCount);
1949getchar();
1950#endif
1951
1952*count = bvCount;
1953
1954free(val);
1955return chain;
1956}
1957
1958//==============================================================================
1959
1960int freeFilteredBVChain(const BVRef chain)
1961{
1962int ret = 1;
1963BVRef bvr = chain;
1964BVRef nextBVR = NULL;
1965
1966while (bvr)
1967{
1968nextBVR = bvr->next;
1969
1970if (bvr->filtered)
1971{
1972free(bvr);
1973}
1974else
1975{
1976ret = 0;
1977break;
1978}
1979
1980bvr = nextBVR;
1981}
1982
1983return ret;
1984}
1985
1986//==============================================================================
1987
1988static const struct NamedValue fdiskTypes[] =
1989{
1990{ FDISK_NTFS,"Windows NTFS" },
1991{ FDISK_DOS12,"Windows FAT12" },
1992{ FDISK_DOS16B,"Windows FAT16" },
1993{ FDISK_DOS16S,"Windows FAT16" },
1994{ FDISK_DOS16SLBA,"Windows FAT16" },
1995{ FDISK_SMALLFAT32,"Windows FAT32" },
1996{ FDISK_FAT32,"Windows FAT32" },
1997{ FDISK_FREEBSD,"FreeBSD" },
1998{ FDISK_OPENBSD,"OpenBSD" },
1999{ FDISK_LINUX,"Linux" },
2000{ FDISK_UFS,"Apple UFS" },
2001{ FDISK_HFS,"Apple HFS" },
2002{ FDISK_BOOTER,"Apple Boot/UFS" },
2003{ FDISK_BEFS,"Haiku" },
2004{ 0xCD,"CD-ROM" },
2005{ 0x00,0 } /* must be last */
2006};
2007
2008//==============================================================================
2009
2010bool matchVolumeToString( BVRef bvr, const char* match, long matchLen)
2011{
2012char testStr[128];
2013
2014if ( !bvr || !match || !*match)
2015{
2016return 0;
2017}
2018
2019if ( bvr->biosdev < 0x80 || bvr->biosdev >= 0x100 )
2020{
2021 return 0;
2022}
2023
2024// Try to match hd(x,y) first.
2025sprintf(testStr, "hd(%d,%d)", BIOS_DEV_UNIT(bvr), bvr->part_no);
2026if ( matchLen ? !strncmp(match, testStr, matchLen) : !strcmp(match, testStr) )
2027{
2028return true;
2029}
2030
2031// Try to match volume UUID.
2032if ( bvr->fs_getuuid && bvr->fs_getuuid(bvr, testStr) == 0)
2033{
2034if ( matchLen ? !strncmp(match, testStr, matchLen) : !strcmp(match, testStr) )
2035{
2036return true;
2037}
2038}
2039
2040// Try to match volume label (always quoted).
2041if ( bvr->description )
2042{
2043bvr->description(bvr, testStr, sizeof(testStr)-1);
2044if ( matchLen ? !strncmp(match, testStr, matchLen) : !strcmp(match, testStr) )
2045{
2046return true;
2047}
2048}
2049
2050return false;
2051}
2052
2053//==============================================================================
2054
2055/* If Rename Partition has defined an alias, then extract it for description purpose.
2056 * The format for the rename string is the following:
2057 * hd(x,y)|uuid|"label" "alias";hd(m,n)|uuid|"label" "alias"; etc...
2058 */
2059
2060bool getVolumeLabelAlias(BVRef bvr, char* str, long strMaxLen)
2061{
2062char *aliasList, *entryStart, *entryNext;
2063
2064if ( !str || strMaxLen <= 0)
2065{
2066return false;
2067}
2068
2069aliasList = XMLDecode(getStringForKey(kRenamePartition, &bootInfo->chameleonConfig));
2070if ( !aliasList )
2071{
2072return false;
2073}
2074
2075for ( entryStart = entryNext = aliasList; entryNext && *entryNext; entryStart = entryNext )
2076{
2077char *volStart, *volEnd, *aliasStart;
2078long volLen, aliasLen;
2079
2080// Delimit current entry
2081entryNext = strchr(entryStart, ';');
2082if ( entryNext )
2083{
2084*entryNext = '\0';
2085entryNext++;
2086}
2087
2088volStart = strbreak(entryStart, &volEnd, &volLen);
2089if(!volLen)
2090{
2091continue;
2092}
2093
2094aliasStart = strbreak(volEnd, 0, &aliasLen);
2095if(!aliasLen)
2096{
2097continue;
2098}
2099
2100if ( matchVolumeToString(bvr, volStart, volLen) )
2101{
2102strncat(str, aliasStart, MIN(strMaxLen, aliasLen));
2103free(aliasList);
2104
2105return true;
2106}
2107}
2108
2109free(aliasList);
2110return false;
2111}
2112
2113//==============================================================================
2114
2115void getBootVolumeDescription( BVRef bvr, char * str, long strMaxLen, bool useDeviceDescription )
2116{
2117unsigned char type;
2118char *p = str;
2119
2120if(!bvr || !p || strMaxLen <= 0)
2121{
2122return;
2123}
2124
2125type = (unsigned char) bvr->part_type;
2126
2127if (useDeviceDescription)
2128{
2129int len = getDeviceDescription(bvr, str);
2130if(len >= strMaxLen)
2131{
2132return;
2133}
2134
2135strcpy(str + len, bvr->OSisInstaller ? " (Installer) " : " ");
2136len += bvr->OSisInstaller ? 13 : 1;
2137strMaxLen -= len;
2138p += len;
2139}
2140
2141/* See if a partition rename is preferred */
2142if (getVolumeLabelAlias(bvr, p, strMaxLen))
2143{
2144strncpy(bvr->label, p, strMaxLen);
2145return; // we're done here no need to seek for real name
2146}
2147
2148// Get the volume label using filesystem specific functions or use the alternate volume label if available.
2149
2150if (*bvr->altlabel != '\0')
2151{
2152strncpy(p, bvr->altlabel, strMaxLen);
2153}
2154else if (bvr->description)
2155{
2156bvr->description(bvr, p, strMaxLen);
2157}
2158
2159if (*p == '\0')
2160{
2161const char * name = getNameForValue( fdiskTypes, type );
2162
2163if (name == NULL)
2164{
2165name = bvr->type_name;
2166}
2167
2168if (name == NULL)
2169{
2170sprintf(p, "TYPE %02x", type);
2171}
2172else
2173{
2174strncpy(p, name, strMaxLen);
2175}
2176}
2177
2178// Set the devices label
2179sprintf(bvr->label, p);
2180}
2181
2182
2183//==============================================================================
2184
2185int readBootSector(int biosdev, unsigned int secno, void * buffer)
2186{
2187int error;
2188struct disk_blk0 * bootSector = (struct disk_blk0 *) buffer;
2189
2190if (bootSector == NULL)
2191{
2192if (gBootSector == NULL)
2193{
2194gBootSector = (struct disk_blk0 *) malloc(sizeof(*gBootSector));
2195
2196if (gBootSector == NULL)
2197{
2198return -1;
2199}
2200}
2201
2202bootSector = gBootSector;
2203}
2204
2205error = readBytes(biosdev, secno, 0, BPS, bootSector);
2206
2207if (error || bootSector->signature != DISK_SIGNATURE)
2208{
2209return -1;
2210}
2211return 0;
2212}
2213
2214//==============================================================================
2215
2216/*
2217 * Format of boot1f32 block.
2218 */
2219
2220#define BOOT1F32_MAGIC "BOOT "
2221#define BOOT1F32_MAGICLEN 11
2222
2223struct disk_boot1f32_blk
2224{
2225unsigned char init[3];
2226unsigned char fsheader[87];
2227unsigned char magic[BOOT1F32_MAGICLEN];
2228unsigned char bootcode[409];
2229unsigned short signature;
2230};
2231
2232//==============================================================================
2233
2234int testFAT32EFIBootSector( int biosdev, unsigned int secno, void * buffer )
2235{
2236struct disk_boot1f32_blk * bootSector = (struct disk_boot1f32_blk *) buffer;
2237int error;
2238
2239if ( bootSector == NULL )
2240{
2241if ( gBootSector == NULL )
2242{
2243gBootSector = (struct disk_blk0 *) malloc(sizeof(*gBootSector));
2244if ( gBootSector == NULL )
2245{
2246return -1;
2247}
2248}
2249bootSector = (struct disk_boot1f32_blk *) gBootSector;
2250}
2251
2252error = readBytes( biosdev, secno, 0, BPS, bootSector );
2253if ( error || bootSector->signature != DISK_SIGNATURE || strncmp((const char *)bootSector->magic, BOOT1F32_MAGIC, BOOT1F32_MAGICLEN) )
2254{
2255return -1;
2256}
2257return 0;
2258}
2259
2260
2261//==============================================================================
2262// Handle seek request from filesystem modules.
2263
2264void diskSeek(BVRef bvr, long long position)
2265{
2266bvr->fs_boff = position / BPS;
2267bvr->fs_byteoff = position % BPS;
2268}
2269
2270
2271//==============================================================================
2272// Handle read request from filesystem modules.
2273
2274int diskRead(BVRef bvr, long addr, long length)
2275{
2276return readBytes(bvr->biosdev, bvr->fs_boff + bvr->part_boff, bvr->fs_byteoff, length, (void *) addr);
2277}
2278
2279//==============================================================================
2280
2281int rawDiskRead( BVRef bvr, unsigned int secno, void *buffer, unsigned int len )
2282{
2283int secs;
2284unsigned char *cbuf = (unsigned char *)buffer;
2285unsigned int copy_len;
2286int rc;
2287
2288if ((len & (BPS-1)) != 0)
2289{
2290error("raw disk read not sector aligned");
2291return -1;
2292}
2293secno += bvr->part_boff;
2294
2295cache_valid = false;
2296
2297while (len > 0)
2298{
2299secs = len / BPS;
2300if (secs > N_CACHE_SECS)
2301{
2302secs = N_CACHE_SECS;
2303}
2304copy_len = secs * BPS;
2305
2306//printf("rdr: ebiosread(%d, %d, %d)\n", bvr->biosdev, secno, secs);
2307if ((rc = ebiosread(bvr->biosdev, secno, secs)) != 0)
2308{
2309/* Ignore corrected ECC errors */
2310if (rc != ECC_CORRECTED_ERR)
2311{
2312error(" EBIOS read error: %s\n", bios_error(rc), rc);
2313error(" Block %d Sectors %d\n", secno, secs);
2314return rc;
2315}
2316}
2317bcopy( trackbuf, cbuf, copy_len );
2318len -= copy_len;
2319cbuf += copy_len;
2320secno += secs;
2321spinActivityIndicator(secs);
2322}
2323
2324return 0;
2325}
2326
2327//==============================================================================
2328
2329int rawDiskWrite( BVRef bvr, unsigned int secno, void *buffer, unsigned int len )
2330{
2331 int secs;
2332 unsigned char *cbuf = (unsigned char *)buffer;
2333 unsigned int copy_len;
2334 int rc;
2335
2336if ((len & (BPS-1)) != 0)
2337{
2338error("raw disk write not sector aligned");
2339return -1;
2340}
2341secno += bvr->part_boff;
2342
2343cache_valid = false;
2344
2345while (len > 0)
2346{
2347secs = len / BPS;
2348if (secs > N_CACHE_SECS)
2349{
2350secs = N_CACHE_SECS;
2351}
2352copy_len = secs * BPS;
2353
2354bcopy( cbuf, trackbuf, copy_len );
2355//printf("rdr: ebioswrite(%d, %d, %d)\n", bvr->biosdev, secno, secs);
2356if ((rc = ebioswrite(bvr->biosdev, secno, secs)) != 0)
2357{
2358error(" EBIOS write error: %s\n", bios_error(rc), rc);
2359error(" Block %d Sectors %d\n", secno, secs);
2360return rc;
2361}
2362
2363len -= copy_len;
2364cbuf += copy_len;
2365secno += secs;
2366spinActivityIndicator(secs);
2367}
2368
2369return 0;
2370}
2371
2372//==============================================================================
2373
2374int diskIsCDROM(BVRef bvr)
2375{
2376struct driveInfo di;
2377
2378if (getDriveInfo(bvr->biosdev, &di) == 0 && di.no_emulation)
2379{
2380return 1;
2381}
2382return 0;
2383}
2384
2385//==============================================================================
2386
2387int biosDevIsCDROM(int biosdev)
2388{
2389struct driveInfo di;
2390
2391if (getDriveInfo(biosdev, &di) == 0 && di.no_emulation)
2392{
2393return 1;
2394}
2395return 0;
2396}
2397

Archive Download this file

Revision: 2479