Chameleon

Chameleon Svn Source Tree

Root/branches/ErmaC/Enoch/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}
768else if ( 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 && 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
1074// (unless malloc failed above)
1075if(map && map == gDiskBVMap)
1076{
1077// Don't leave a null map in the chain
1078if(map->bvrcnt == 0 && map->bvr == NULL)
1079{
1080gDiskBVMap = map->next;
1081free(map);
1082map = NULL;
1083}
1084}
1085
1086if (countPtr) *countPtr = map ? map->bvrcnt : 0;
1087
1088return map ? map->bvr : NULL;
1089}
1090
1091//==============================================================================
1092
1093static BVRef diskScanAPMBootVolumes( int biosdev, int * countPtr )
1094{
1095struct DiskBVMap * map;
1096struct Block0 *block0_p;
1097unsigned int blksize;
1098unsigned int factor;
1099void *buffer = malloc(BPS);
1100
1101if (!buffer)
1102{
1103return NULL;
1104}
1105bzero(buffer,BPS);
1106
1107/* Check for alternate block size */
1108if (readBytes( biosdev, 0, 0, BPS, buffer ) != 0)
1109{
1110return NULL;
1111}
1112block0_p = buffer;
1113if (OSSwapBigToHostInt16(block0_p->sbSig) == BLOCK0_SIGNATURE)
1114{
1115blksize = OSSwapBigToHostInt16(block0_p->sbBlkSize);
1116if (blksize != BPS)
1117{
1118free(buffer);
1119buffer = malloc(blksize);
1120if (!buffer)
1121{
1122return NULL;
1123}
1124bzero(buffer,BPS);
1125}
1126factor = blksize / BPS;
1127}
1128else
1129{
1130blksize = BPS;
1131factor = 1;
1132}
1133
1134do
1135{
1136// Create a new mapping.
1137
1138map = (struct DiskBVMap *) malloc( sizeof(*map) );
1139if ( map )
1140{
1141int error;
1142DPME *dpme_p = (DPME *)buffer;
1143UInt32 i, npart = UINT_MAX;
1144BVRef bvr;
1145
1146map->biosdev = biosdev;
1147map->bvr = NULL;
1148map->bvrcnt = 0;
1149map->next = gDiskBVMap;
1150gDiskBVMap = map;
1151
1152for (i=0; i<npart; i++)
1153{
1154error = readBytes( biosdev, (kAPMSector + i) * factor, 0, blksize, buffer );
1155
1156if (error || OSSwapBigToHostInt16(dpme_p->dpme_signature) != DPME_SIGNATURE)
1157{
1158break;
1159}
1160
1161if (i==0)
1162{
1163npart = OSSwapBigToHostInt32(dpme_p->dpme_map_entries);
1164}
1165/*
1166printf("name = %s, %s%s %d -> %d [%d -> %d] {%d}\n",
1167dpme.dpme_name, dpme.dpme_type, (dpme.dpme_flags & DPME_FLAGS_BOOTABLE) ? "(bootable)" : "",
1168dpme.dpme_pblock_start, dpme.dpme_pblocks,
1169dpme.dpme_lblock_start, dpme.dpme_lblocks,
1170dpme.dpme_boot_block);
1171*/
1172
1173if (strcmp(dpme_p->dpme_type, "Apple_HFS") == 0)
1174{
1175bvr = newAPMBVRef(biosdev,
1176i,
1177OSSwapBigToHostInt32(dpme_p->dpme_pblock_start) * factor,
1178dpme_p,
1179HFSInitPartition,
1180HFSLoadFile,
1181HFSReadFile,
1182HFSGetDirEntry,
1183HFSGetFileBlock,
1184HFSGetUUID,
1185HFSGetDescription,
1186HFSFree,
11870,
1188kBIOSDevTypeHardDrive, 0);
1189bvr->next = map->bvr;
1190map->bvr = bvr;
1191map->bvrcnt++;
1192}
1193}
1194}
1195} while (0);
1196
1197free(buffer);
1198
1199if (countPtr) *countPtr = map ? map->bvrcnt : 0;
1200
1201return map ? map->bvr : NULL;
1202}
1203
1204//==============================================================================
1205
1206/*
1207 * Trying to figure out the filsystem type of a given partition.
1208 */
1209static int probeFileSystem(int biosdev, unsigned int blkoff)
1210{
1211// detected filesystem type;
1212int result = -1;
1213int fatbits;
1214
1215// Allocating buffer for 4 sectors.
1216const void * probeBuffer = malloc(PROBEFS_SIZE);
1217if (probeBuffer == NULL)
1218{
1219goto exit;
1220}
1221
1222// Reading first 4 sectors of current partition
1223int error = readBytes(biosdev, blkoff, 0, PROBEFS_SIZE, (void *)probeBuffer);
1224
1225if (error)
1226{
1227goto exit;
1228}
1229
1230if (HFSProbe(probeBuffer))
1231{
1232result = FDISK_HFS;
1233}
1234else if (EX2Probe(probeBuffer))
1235{
1236result = FDISK_LINUX;
1237}
1238else if (FreeBSDProbe(probeBuffer))
1239{
1240result = FDISK_FREEBSD;
1241}
1242
1243else if (OpenBSDProbe(probeBuffer))
1244{
1245result = FDISK_OPENBSD;
1246}
1247
1248else if (BeFSProbe(probeBuffer))
1249{
1250result = FDISK_BEFS;
1251}
1252
1253else if (NTFSProbe(probeBuffer))
1254{
1255result = FDISK_NTFS;
1256}
1257
1258else if ( (fatbits = MSDOSProbe(probeBuffer)) )
1259{
1260switch (fatbits)
1261{
1262case 32:
1263default:
1264result = FDISK_FAT32;
1265break;
1266case 16:
1267result = FDISK_DOS16B;
1268break;
1269case 12:
1270result = FDISK_DOS12;
1271break;
1272}
1273}
1274else
1275{
1276// Couldn't detect filesystem type
1277result = 0;
1278}
1279
1280exit:
1281if (probeBuffer != NULL) free((void *)probeBuffer);
1282{
1283return result;
1284}
1285}
1286
1287//==============================================================================
1288
1289static bool isPartitionUsed(gpt_ent * partition)
1290{
1291
1292// Ask whether the given partition is used.
1293
1294return efi_guid_is_null((EFI_GUID const*)partition->ent_type) ? false : true;
1295}
1296
1297//==============================================================================
1298
1299static BVRef diskScanGPTBootVolumes(int biosdev, int * countPtr)
1300{
1301struct DiskBVMap *map = NULL;
1302
1303void *buffer = malloc(BPS);
1304
1305int error;
1306if ( (error = readBytes( biosdev, /*secno*/0, 0, BPS, buffer )) != 0)
1307{
1308verbose("Failed to read boot sector from BIOS device %02xh. Error=%d\n", biosdev, error);
1309goto scanErr;
1310}
1311struct REAL_disk_blk0 *fdiskMap = buffer;
1312if ( OSSwapLittleToHostInt16(fdiskMap->signature) != DISK_SIGNATURE )
1313{
1314verbose("Failed to find boot signature on BIOS device %02xh\n", biosdev);
1315goto scanErr;
1316}
1317
1318int fdiskID = 0;
1319unsigned index;
1320for ( index = 0; index < FDISK_NPART; index++ )
1321{
1322if ( fdiskMap->parts[index].systid )
1323{
1324if ( fdiskMap->parts[index].systid == 0xEE )
1325{
1326// Fail if two 0xEE partitions are present which
1327// means the FDISK code will wind up parsing it.
1328if ( fdiskID )
1329{
1330goto scanErr;
1331}
1332
1333fdiskID = index + 1;
1334}
1335}
1336}
1337
1338if ( fdiskID == 0 )
1339{
1340goto scanErr;
1341}
1342
1343verbose("Attempting to read GPT\n");
1344
1345if(readBytes(biosdev, 1, 0, BPS, buffer) != 0)
1346{
1347goto scanErr;
1348}
1349
1350gpt_hdr *headerMap = buffer;
1351
1352// Determine whether the partition header signature is present.
1353
1354if ( memcmp(headerMap->hdr_sig, GPT_HDR_SIG, strlen(GPT_HDR_SIG)) )
1355{
1356goto scanErr;
1357}
1358
1359// Determine whether the partition header size is valid.
1360
1361UInt32 headerCheck = OSSwapLittleToHostInt32(headerMap->hdr_crc_self);
1362UInt32 headerSize = OSSwapLittleToHostInt32(headerMap->hdr_size);
1363
1364if ( headerSize < offsetof(gpt_hdr, padding) )
1365{
1366goto scanErr;
1367}
1368
1369if ( headerSize > BPS )
1370{
1371goto scanErr;
1372}
1373
1374// Determine whether the partition header checksum is valid.
1375
1376headerMap->hdr_crc_self = 0;
1377
1378if ( crc32(0, headerMap, headerSize) != headerCheck )
1379{
1380goto scanErr;
1381}
1382
1383// Determine whether the partition entry size is valid.
1384
1385UInt64 gptBlock = 0;
1386UInt32 gptCheck = 0;
1387UInt32 gptCount = 0;
1388UInt32 gptID = 0;
1389gpt_ent * gptMap = 0;
1390UInt32 gptSize = 0;
1391
1392gptBlock = OSSwapLittleToHostInt64(headerMap->hdr_lba_table);
1393gptCheck = OSSwapLittleToHostInt32(headerMap->hdr_crc_table);
1394gptCount = OSSwapLittleToHostInt32(headerMap->hdr_entries);
1395gptSize = OSSwapLittleToHostInt32(headerMap->hdr_entsz);
1396
1397if ( gptSize < sizeof(gpt_ent) )
1398{
1399goto scanErr;
1400}
1401
1402// Allocate a buffer large enough to hold one map, rounded to a media block.
1403free(buffer);
1404buffer = NULL;
1405
1406UInt32 bufferSize = IORound(gptCount * gptSize, BPS);
1407if (bufferSize == 0)
1408{
1409goto scanErr;
1410}
1411buffer = malloc(bufferSize);
1412if (!buffer)
1413{
1414 goto scanErr;
1415}
1416
1417if (readBytes(biosdev, gptBlock, 0, bufferSize, buffer) != 0)
1418{
1419goto scanErr;
1420}
1421verbose("Read GPT\n");
1422
1423// Allocate a new map for this BIOS device and insert it into the chain
1424map = malloc(sizeof(*map));
1425if (!map)
1426{
1427goto scanErr;
1428}
1429map->biosdev = biosdev;
1430map->bvr = NULL;
1431map->bvrcnt = 0;
1432map->next = gDiskBVMap;
1433gDiskBVMap = map;
1434
1435// fdisk like partition type id.
1436int fsType = 0;
1437
1438for(gptID = 1; gptID <= gptCount; ++gptID) {
1439BVRef bvr = NULL;
1440unsigned int bvrFlags = 0;
1441
1442// size on disk can be larger than sizeof(gpt_ent)
1443gptMap = (gpt_ent *) ( buffer + ( (gptID - 1) * gptSize) );
1444
1445// NOTE: EFI_GUID's are in LE and we know we're on an x86.
1446// The IOGUIDPartitionScheme.cpp code uses byte-based UUIDs, we don't.
1447
1448if (isPartitionUsed(gptMap)) {
1449char stringuuid[100];
1450efi_guid_unparse_upper((EFI_GUID*)gptMap->ent_type, stringuuid);
1451verbose("Reading GPT partition %d, type %s\n", gptID, stringuuid);
1452
1453// Getting fdisk like partition type.
1454fsType = probeFileSystem(biosdev, gptMap->ent_lba_start);
1455
1456if ( (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) ) {
1457bvrFlags = (efi_guid_compare(&GPT_BOOT_GUID, (EFI_GUID const*)gptMap->ent_type) == 0) ? kBVFlagBooter : 0;
1458bvr = newGPTBVRef(biosdev,
1459gptID,
1460gptMap->ent_lba_start,
1461gptMap,
1462HFSInitPartition,
1463HFSLoadFile,
1464HFSReadFile,
1465HFSGetDirEntry,
1466HFSGetFileBlock,
1467HFSGetUUID,
1468HFSGetDescription,
1469HFSFree,
14700,
1471kBIOSDevTypeHardDrive, bvrFlags);
1472}
1473
1474// zef - foreign OS support
1475if ( (efi_guid_compare(&GPT_BASICDATA_GUID, (EFI_GUID const*)gptMap->ent_type) == 0) ||
1476(efi_guid_compare(&GPT_BASICDATA2_GUID, (EFI_GUID const*)gptMap->ent_type) == 0) ) {
1477switch (fsType)
1478{
1479case FDISK_NTFS:
1480bvr = newGPTBVRef(biosdev, gptID, gptMap->ent_lba_start, gptMap,
14810, 0, 0, 0, 0, 0, NTFSGetDescription,
1482(BVFree)free, 0, kBIOSDevTypeHardDrive, 0);
1483break;
1484
1485case FDISK_LINUX:
1486bvr = newGPTBVRef(biosdev, gptID, gptMap->ent_lba_start, gptMap,
14870, 0, 0, 0, 0, 0, EX2GetDescription,
1488(BVFree)free, 0, kBIOSDevTypeHardDrive, 0);
1489break;
1490
1491default:
1492bvr = newGPTBVRef(biosdev, gptID, gptMap->ent_lba_start, gptMap,
14930, 0, 0, 0, 0, 0, 0,
1494(BVFree)free, 0, kBIOSDevTypeHardDrive, 0);
1495break;
1496}
1497
1498}
1499
1500// turbo - save our booter partition
1501// zef - only on original boot device
1502if ( (efi_guid_compare(&GPT_EFISYS_GUID, (EFI_GUID const*)gptMap->ent_type) == 0) ) {
1503switch (fsType) {
1504case FDISK_HFS:
1505if (readBootSector( biosdev, gptMap->ent_lba_start, (void *)0x7e00 ) == 0) {
1506bvr = newGPTBVRef(biosdev, gptID, gptMap->ent_lba_start, gptMap,
1507HFSInitPartition,
1508HFSLoadFile,
1509HFSReadFile,
1510HFSGetDirEntry,
1511HFSGetFileBlock,
1512HFSGetUUID,
1513HFSGetDescription,
1514HFSFree,
15150, kBIOSDevTypeHardDrive, kBVFlagEFISystem);
1516}
1517break;
1518
1519case FDISK_FAT32:
1520if (testFAT32EFIBootSector( biosdev, gptMap->ent_lba_start, (void *)0x7e00 ) == 0) {
1521bvr = newGPTBVRef(biosdev, gptID, gptMap->ent_lba_start, gptMap,
1522MSDOSInitPartition,
1523MSDOSLoadFile,
1524MSDOSReadFile,
1525MSDOSGetDirEntry,
1526MSDOSGetFileBlock,
1527MSDOSGetUUID,
1528MSDOSGetDescription,
1529MSDOSFree,
15300, kBIOSDevTypeHardDrive, kBVFlagEFISystem);
1531}
1532break;
1533
1534default:
1535if (biosdev == gBIOSDev) {
1536gBIOSBootVolume = bvr;
1537}
1538break;
1539}
1540}
1541
1542if (bvr)
1543{
1544// Fixup bvr with the fake fdisk partition type.
1545if (fsType > 0) {
1546bvr->part_type = fsType;
1547}
1548
1549bvr->next = map->bvr;
1550map->bvr = bvr;
1551++map->bvrcnt;
1552}
1553
1554}
1555}
1556
1557scanErr:
1558if (buffer) {
1559free(buffer);
1560}
1561
1562if(map) {
1563if(countPtr) *countPtr = map->bvrcnt;
1564{
1565return map->bvr;
1566}
1567
1568} else {
1569if(countPtr) *countPtr = 0;
1570{
1571return NULL;
1572}
1573}
1574}
1575
1576//==============================================================================
1577
1578static bool getOSVersion(BVRef bvr, char *str)
1579{
1580bool valid = false;
1581config_file_t systemVersion;
1582char dirSpec[512];
1583
1584sprintf(dirSpec, "hd(%d,%d)/com.apple.recovery.boot/SystemVersion.plist", BIOS_DEV_UNIT(bvr), bvr->part_no);
1585
1586if (!loadConfigFile(dirSpec, &systemVersion))
1587{
1588bvr->OSisInstaller = true;
1589valid = true;
1590}
1591
1592if (!valid)
1593{
1594sprintf(dirSpec, "hd(%d,%d)/System/Library/CoreServices/SystemVersion.plist", BIOS_DEV_UNIT(bvr), bvr->part_no);
1595
1596if (!loadConfigFile(dirSpec, &systemVersion))
1597{
1598bvr->OSisInstaller = true;
1599valid = true;
1600}
1601else
1602{
1603sprintf(dirSpec, "hd(%d,%d)/System/Library/CoreServices/ServerVersion.plist", BIOS_DEV_UNIT(bvr), bvr->part_no);
1604
1605if (!loadConfigFile(dirSpec, &systemVersion))
1606{
1607bvr->OSisServer = true;
1608valid = true;
1609}
1610/*else
1611{
1612sprintf(dirSpec, "hd(%d,%d)/.IAProductInfo", BIOS_DEV_UNIT(bvr), bvr->part_no);
1613
1614if (!loadConfigFile(dirSpec, &systemVersion))
1615{
1616
1617}
1618}
1619*/
1620}
1621}
1622
1623if (valid)
1624{
1625const char *val;
1626int len;
1627
1628if (getValueForKey(kProductVersion, &val, &len, &systemVersion))
1629{
1630// getValueForKey uses const char for val
1631// so copy it and trim
1632*str = '\0';
1633// crazybirdy
1634if (len > 4 && (val[3] == '1'))
1635{
1636strncat(str, val, MIN(len, 5));
1637}
1638else
1639{
1640strncat(str, val, MIN(len, 4));
1641}
1642}
1643else
1644{
1645valid = false;
1646}
1647}
1648
1649if(!valid)
1650{
1651int fh = -1;
1652sprintf(dirSpec, "hd(%d,%d)/.PhysicalMediaInstall", BIOS_DEV_UNIT(bvr), bvr->part_no);
1653fh = open(dirSpec, 0);
1654
1655if (fh >= 0)
1656{
1657valid = true;
1658bvr->OSisInstaller = true;
1659strcpy(bvr->OSVersion, "10.7"); // 10.7 +
1660close(fh);
1661}
1662else
1663{
1664sprintf(dirSpec, "hd(%d,%d)/.IAPhysicalMedia", BIOS_DEV_UNIT(bvr), bvr->part_no);
1665fh = open(dirSpec, 0);
1666
1667if (fh >= 0)
1668{
1669valid = true;
1670bvr->OSisInstaller = true;
1671strcpy(bvr->OSVersion, "10.9"); // 10.9 +
1672}
1673else
1674{
1675close(fh);
1676}
1677}
1678}
1679return valid;
1680}
1681
1682//==============================================================================
1683
1684static void scanFSLevelBVRSettings(BVRef chain)
1685{
1686BVRef bvr;
1687char dirSpec[512], fileSpec[512];
1688char label[BVSTRLEN];
1689int ret;
1690long flags;
1691u_int32_t time;
1692int fh, fileSize, error;
1693
1694for (bvr = chain; bvr; bvr = bvr->next)
1695{
1696ret = -1;
1697error = 0;
1698
1699//
1700// Check for alternate volume label on boot helper partitions.
1701//
1702if (bvr->flags & kBVFlagBooter)
1703{
1704sprintf(dirSpec, "hd(%d,%d)/System/Library/CoreServices/", BIOS_DEV_UNIT(bvr), bvr->part_no);
1705strcpy(fileSpec, ".disk_label.contentDetails");
1706ret = GetFileInfo(dirSpec, fileSpec, &flags, &time);
1707if (!ret)
1708{
1709fh = open(strcat(dirSpec, fileSpec), 0);
1710fileSize = file_size(fh);
1711if (fileSize > 0 && fileSize < BVSTRLEN)
1712{
1713if (read(fh, label, fileSize) != fileSize)
1714{
1715error = -1;
1716}
1717}
1718else
1719{
1720error = -1;
1721}
1722
1723close(fh);
1724
1725if (!error)
1726{
1727label[fileSize] = '\0';
1728strcpy(bvr->altlabel, label);
1729}
1730}
1731}
1732
1733// Check for SystemVersion.plist or ServerVersion.plist to determine if a volume hosts an installed system.
1734
1735if (bvr->flags & kBVFlagNativeBoot)
1736{
1737if (getOSVersion(bvr,bvr->OSVersion) == true)
1738{
1739bvr->flags |= kBVFlagSystemVolume;
1740}
1741}
1742
1743}
1744}
1745
1746//==============================================================================
1747
1748void rescanBIOSDevice(int biosdev)
1749{
1750struct DiskBVMap *oldMap = diskResetBootVolumes(biosdev);
1751CacheReset();
1752diskFreeMap(oldMap);
1753oldMap = NULL;
1754scanBootVolumes(biosdev, 0);
1755}
1756
1757//==============================================================================
1758
1759struct DiskBVMap* diskResetBootVolumes(int biosdev)
1760{
1761struct DiskBVMap * map;
1762struct DiskBVMap *prevMap = NULL;
1763for ( map = gDiskBVMap; map; prevMap = map, map = map->next ) {
1764if ( biosdev == map->biosdev ) {
1765break;
1766}
1767}
1768
1769if(map != NULL) {
1770verbose("Resetting BIOS device %xh\n", biosdev);
1771// Reset the biosbuf cache
1772cache_valid = false;
1773if(map == gDiskBVMap)
1774{
1775gDiskBVMap = map->next;
1776}
1777else if(prevMap != NULL)
1778{
1779prevMap->next = map->next;
1780}
1781else
1782{
1783stop("");
1784}
1785}
1786// Return the old map, either to be freed, or reinserted later
1787return map;
1788}
1789
1790//==============================================================================
1791
1792// Frees a DiskBVMap and all of its BootVolume's
1793void diskFreeMap(struct DiskBVMap *map)
1794{
1795if(map != NULL)
1796{
1797while(map->bvr != NULL)
1798{
1799BVRef bvr = map->bvr;
1800map->bvr = bvr->next;
1801(*bvr->bv_free)(bvr);
1802}
1803
1804free(map);
1805}
1806}
1807
1808//==============================================================================
1809
1810BVRef diskScanBootVolumes(int biosdev, int * countPtr)
1811{
1812struct DiskBVMap *map;
1813BVRef bvr;
1814int count = 0;
1815
1816// Find an existing mapping for this device.
1817
1818for (map = gDiskBVMap; map; map = map->next)
1819{
1820if (biosdev == map->biosdev)
1821{
1822count = map->bvrcnt;
1823break;
1824}
1825}
1826
1827if (map == NULL)
1828{
1829bvr = diskScanGPTBootVolumes(biosdev, &count);
1830if (bvr == NULL)
1831{
1832bvr = diskScanFDiskBootVolumes(biosdev, &count);
1833}
1834if (bvr == NULL)
1835{
1836bvr = diskScanAPMBootVolumes(biosdev, &count);
1837}
1838if (bvr)
1839{
1840scanFSLevelBVRSettings(bvr);
1841}
1842}
1843else
1844{
1845bvr = map->bvr;
1846}
1847if (countPtr)
1848{
1849*countPtr += count;
1850}
1851return bvr;
1852}
1853
1854//==============================================================================
1855
1856BVRef getBVChainForBIOSDev(int biosdev)
1857{
1858BVRef chain = NULL;
1859struct DiskBVMap * map = NULL;
1860
1861for (map = gDiskBVMap; map; map = map->next)
1862{
1863if (map->biosdev == biosdev)
1864{
1865chain = map->bvr;
1866break;
1867}
1868}
1869
1870return chain;
1871}
1872
1873//==============================================================================
1874
1875BVRef newFilteredBVChain(int minBIOSDev, int maxBIOSDev, unsigned int allowFlags, unsigned int denyFlags, int *count)
1876{
1877BVRef chain = NULL;
1878BVRef bvr = NULL;
1879BVRef newBVR = NULL;
1880BVRef prevBVR = NULL;
1881
1882struct DiskBVMap * map = NULL;
1883int bvCount = 0;
1884
1885const char *raw = 0;
1886char* val = 0;
1887int len;
1888
1889getValueForKey(kHidePartition, &raw, &len, &bootInfo->chameleonConfig);
1890if(raw)
1891{
1892val = XMLDecode(raw);
1893}
1894
1895/*
1896 * Traverse gDISKBVmap to get references for
1897 * individual bvr chains of each drive.
1898 */
1899for (map = gDiskBVMap; map; map = map->next)
1900{
1901for (bvr = map->bvr; bvr; bvr = bvr->next)
1902{
1903/*
1904 * Save the last bvr.
1905 */
1906if (newBVR)
1907{
1908prevBVR = newBVR;
1909}
1910
1911/*
1912 * Allocate and copy the matched bvr entry into a new one.
1913 */
1914newBVR = (BVRef) malloc(sizeof(*newBVR));
1915if (!newBVR)
1916{
1917continue;
1918}
1919bcopy(bvr, newBVR, sizeof(*newBVR));
1920
1921/*
1922 * Adjust the new bvr's fields.
1923 */
1924newBVR->next = NULL;
1925newBVR->filtered = true;
1926
1927if ( (!allowFlags || newBVR->flags & allowFlags)
1928&& (!denyFlags || !(newBVR->flags & denyFlags) )
1929&& (newBVR->biosdev >= minBIOSDev && newBVR->biosdev <= maxBIOSDev)
1930)
1931{
1932newBVR->visible = true;
1933}
1934
1935/*
1936 * Looking for "Hide Partition" entries in 'hd(x,y)|uuid|"label" hd(m,n)|uuid|"label"' format,
1937 * to be able to hide foreign partitions from the boot menu.
1938 *
1939 */
1940if ( (newBVR->flags & kBVFlagForeignBoot) )
1941{
1942char *start, *next = val;
1943long len = 0;
1944do
1945{
1946start = strbreak(next, &next, &len);
1947if(len && matchVolumeToString(newBVR, start, len) )
1948{
1949newBVR->visible = false;
1950}
1951}
1952while ( next && *next );
1953}
1954
1955/*
1956 * Use the first bvr entry as the starting chain pointer.
1957 */
1958if (!chain)
1959{
1960chain = newBVR;
1961}
1962
1963/*
1964 * Update the previous bvr's link pointer to use the new memory area.
1965 */
1966if (prevBVR)
1967{
1968prevBVR->next = newBVR;
1969}
1970
1971if (newBVR->visible)
1972{
1973bvCount++;
1974}
1975}
1976}
1977
1978#if DEBUG //Azi: warning - too big for boot-log.. far too big.. i mean HUGE!! :P
1979for (bvr = chain; bvr; bvr = bvr->next)
1980{
1981printf(" bvr: %d, dev: %d, part: %d, flags: %d, vis: %d\n", bvr, bvr->biosdev, bvr->part_no, bvr->flags, bvr->visible);
1982}
1983printf("count: %d\n", bvCount);
1984getchar();
1985#endif
1986
1987*count = bvCount;
1988
1989free(val);
1990return chain;
1991}
1992
1993//==============================================================================
1994
1995int freeFilteredBVChain(const BVRef chain)
1996{
1997int ret = 1;
1998BVRef bvr = chain;
1999BVRef nextBVR = NULL;
2000
2001while (bvr)
2002{
2003nextBVR = bvr->next;
2004
2005if (bvr->filtered)
2006{
2007free(bvr);
2008}
2009else
2010{
2011ret = 0;
2012break;
2013}
2014
2015bvr = nextBVR;
2016}
2017
2018return ret;
2019}
2020
2021//==============================================================================
2022
2023static const struct NamedValue fdiskTypes[] =
2024{
2025{ FDISK_NTFS,"Windows NTFS" },
2026{ FDISK_DOS12,"Windows FAT12" },
2027{ FDISK_DOS16B,"Windows FAT16" },
2028{ FDISK_DOS16S,"Windows FAT16" },
2029{ FDISK_DOS16SLBA,"Windows FAT16" },
2030{ FDISK_SMALLFAT32,"Windows FAT32" },
2031{ FDISK_FAT32,"Windows FAT32" },
2032{ FDISK_FREEBSD,"FreeBSD" },
2033{ FDISK_OPENBSD,"OpenBSD" },
2034{ FDISK_LINUX,"Linux" },
2035{ FDISK_UFS,"Apple UFS" },
2036{ FDISK_HFS,"Apple HFS" },
2037{ FDISK_BOOTER,"Apple Boot/UFS" },
2038{ FDISK_BEFS,"Haiku" },
2039{ 0xCD,"CD-ROM" },
2040{ 0x00,0 } /* must be last */
2041};
2042
2043//==============================================================================
2044
2045bool matchVolumeToString( BVRef bvr, const char* match, long matchLen)
2046{
2047char testStr[128];
2048
2049if ( !bvr || !match || !*match)
2050{
2051return 0;
2052}
2053
2054if ( bvr->biosdev < 0x80 || bvr->biosdev >= 0x100 )
2055{
2056 return 0;
2057}
2058
2059// Try to match hd(x,y) first.
2060sprintf(testStr, "hd(%d,%d)", BIOS_DEV_UNIT(bvr), bvr->part_no);
2061if ( matchLen ? !strncmp(match, testStr, matchLen) : !strcmp(match, testStr) )
2062{
2063return true;
2064}
2065
2066// Try to match volume UUID.
2067if ( bvr->fs_getuuid && bvr->fs_getuuid(bvr, testStr) == 0)
2068{
2069if ( matchLen ? !strncmp(match, testStr, matchLen) : !strcmp(match, testStr) )
2070{
2071return true;
2072}
2073}
2074
2075// Try to match volume label (always quoted).
2076if ( bvr->description )
2077{
2078bvr->description(bvr, testStr, sizeof(testStr)-1);
2079if ( matchLen ? !strncmp(match, testStr, matchLen) : !strcmp(match, testStr) )
2080{
2081return true;
2082}
2083}
2084
2085return false;
2086}
2087
2088//==============================================================================
2089
2090/* If Rename Partition has defined an alias, then extract it for description purpose.
2091 * The format for the rename string is the following:
2092 * hd(x,y)|uuid|"label" "alias";hd(m,n)|uuid|"label" "alias"; etc...
2093 */
2094
2095bool getVolumeLabelAlias(BVRef bvr, char* str, long strMaxLen)
2096{
2097char *aliasList, *entryStart, *entryNext;
2098
2099if ( !str || strMaxLen <= 0)
2100{
2101return false;
2102}
2103
2104aliasList = XMLDecode(getStringForKey(kRenamePartition, &bootInfo->chameleonConfig));
2105if ( !aliasList )
2106{
2107return false;
2108}
2109
2110for ( entryStart = entryNext = aliasList; entryNext && *entryNext; entryStart = entryNext )
2111{
2112char *volStart, *volEnd, *aliasStart;
2113long volLen, aliasLen;
2114
2115// Delimit current entry
2116entryNext = strchr(entryStart, ';');
2117if ( entryNext )
2118{
2119*entryNext = '\0';
2120entryNext++;
2121}
2122
2123volStart = strbreak(entryStart, &volEnd, &volLen);
2124if(!volLen)
2125{
2126continue;
2127}
2128
2129aliasStart = strbreak(volEnd, 0, &aliasLen);
2130if(!aliasLen)
2131{
2132continue;
2133}
2134
2135if ( matchVolumeToString(bvr, volStart, volLen) )
2136{
2137strncat(str, aliasStart, MIN(strMaxLen, aliasLen));
2138free(aliasList);
2139
2140return true;
2141}
2142}
2143
2144free(aliasList);
2145return false;
2146}
2147
2148//==============================================================================
2149
2150void getBootVolumeDescription( BVRef bvr, char * str, long strMaxLen, bool useDeviceDescription )
2151{
2152unsigned char type;
2153char *p = str;
2154
2155if(!bvr || !p || strMaxLen <= 0)
2156{
2157return;
2158}
2159
2160type = (unsigned char) bvr->part_type;
2161
2162if (useDeviceDescription)
2163{
2164int len = getDeviceDescription(bvr, str);
2165if(len >= strMaxLen)
2166{
2167return;
2168}
2169
2170strcpy(str + len, bvr->OSisInstaller ? " (Installer) " : " ");
2171len += bvr->OSisInstaller ? 13 : 1;
2172strMaxLen -= len;
2173p += len;
2174}
2175
2176/* See if a partition rename is preferred */
2177if (getVolumeLabelAlias(bvr, p, strMaxLen))
2178{
2179strncpy(bvr->label, p, strMaxLen);
2180return; // we're done here no need to seek for real name
2181}
2182
2183// Get the volume label using filesystem specific functions or use the alternate volume label if available.
2184
2185if (*bvr->altlabel != '\0')
2186{
2187strncpy(p, bvr->altlabel, strMaxLen);
2188}
2189else if (bvr->description)
2190{
2191bvr->description(bvr, p, strMaxLen);
2192}
2193
2194if (*p == '\0')
2195{
2196const char * name = getNameForValue( fdiskTypes, type );
2197
2198if (name == NULL)
2199{
2200name = bvr->type_name;
2201}
2202
2203if (name == NULL)
2204{
2205sprintf(p, "TYPE %02x", type);
2206}
2207else
2208{
2209strncpy(p, name, strMaxLen);
2210}
2211}
2212
2213// Set the devices label
2214sprintf(bvr->label, p);
2215}
2216
2217
2218//==============================================================================
2219
2220int readBootSector(int biosdev, unsigned int secno, void * buffer)
2221{
2222int error;
2223struct disk_blk0 * bootSector = (struct disk_blk0 *) buffer;
2224
2225if (bootSector == NULL)
2226{
2227if (gBootSector == NULL)
2228{
2229gBootSector = (struct disk_blk0 *) malloc(sizeof(*gBootSector));
2230
2231if (gBootSector == NULL)
2232{
2233return -1;
2234}
2235}
2236
2237bootSector = gBootSector;
2238}
2239
2240error = readBytes(biosdev, secno, 0, BPS, bootSector);
2241
2242if (error || bootSector->signature != DISK_SIGNATURE)
2243{
2244return -1;
2245}
2246return 0;
2247}
2248
2249//==============================================================================
2250
2251/*
2252 * Format of boot1f32 block.
2253 */
2254
2255#define BOOT1F32_MAGIC "BOOT "
2256#define BOOT1F32_MAGICLEN 11
2257
2258struct disk_boot1f32_blk
2259{
2260unsigned char init[3];
2261unsigned char fsheader[87];
2262unsigned char magic[BOOT1F32_MAGICLEN];
2263unsigned char bootcode[409];
2264unsigned short signature;
2265};
2266
2267//==============================================================================
2268
2269int testFAT32EFIBootSector( int biosdev, unsigned int secno, void * buffer )
2270{
2271struct disk_boot1f32_blk * bootSector = (struct disk_boot1f32_blk *) buffer;
2272int error;
2273
2274if ( bootSector == NULL )
2275{
2276if ( gBootSector == NULL )
2277{
2278gBootSector = (struct disk_blk0 *) malloc(sizeof(*gBootSector));
2279if ( gBootSector == NULL )
2280{
2281return -1;
2282}
2283}
2284bootSector = (struct disk_boot1f32_blk *) gBootSector;
2285}
2286
2287error = readBytes( biosdev, secno, 0, BPS, bootSector );
2288if ( error || bootSector->signature != DISK_SIGNATURE || strncmp((const char *)bootSector->magic, BOOT1F32_MAGIC, BOOT1F32_MAGICLEN) )
2289{
2290return -1;
2291}
2292return 0;
2293}
2294
2295
2296//==============================================================================
2297// Handle seek request from filesystem modules.
2298
2299void diskSeek(BVRef bvr, long long position)
2300{
2301bvr->fs_boff = position / BPS;
2302bvr->fs_byteoff = position % BPS;
2303}
2304
2305
2306//==============================================================================
2307// Handle read request from filesystem modules.
2308
2309int diskRead(BVRef bvr, long addr, long length)
2310{
2311return readBytes(bvr->biosdev, bvr->fs_boff + bvr->part_boff, bvr->fs_byteoff, length, (void *) addr);
2312}
2313
2314//==============================================================================
2315
2316int rawDiskRead( BVRef bvr, unsigned int secno, void *buffer, unsigned int len )
2317{
2318int secs;
2319unsigned char *cbuf = (unsigned char *)buffer;
2320unsigned int copy_len;
2321int rc;
2322
2323if ((len & (BPS-1)) != 0)
2324{
2325error("raw disk read not sector aligned");
2326return -1;
2327}
2328secno += bvr->part_boff;
2329
2330cache_valid = false;
2331
2332while (len > 0)
2333{
2334secs = len / BPS;
2335if (secs > N_CACHE_SECS)
2336{
2337secs = N_CACHE_SECS;
2338}
2339copy_len = secs * BPS;
2340
2341//printf("rdr: ebiosread(%d, %d, %d)\n", bvr->biosdev, secno, secs);
2342if ((rc = ebiosread(bvr->biosdev, secno, secs)) != 0)
2343{
2344/* Ignore corrected ECC errors */
2345if (rc != ECC_CORRECTED_ERR)
2346{
2347error(" EBIOS read error: %s\n", bios_error(rc), rc);
2348error(" Block %d Sectors %d\n", secno, secs);
2349return rc;
2350}
2351}
2352bcopy( trackbuf, cbuf, copy_len );
2353len -= copy_len;
2354cbuf += copy_len;
2355secno += secs;
2356spinActivityIndicator(secs);
2357}
2358
2359return 0;
2360}
2361
2362//==============================================================================
2363
2364int rawDiskWrite( BVRef bvr, unsigned int secno, void *buffer, unsigned int len )
2365{
2366 int secs;
2367 unsigned char *cbuf = (unsigned char *)buffer;
2368 unsigned int copy_len;
2369 int rc;
2370
2371if ((len & (BPS-1)) != 0)
2372{
2373error("raw disk write not sector aligned");
2374return -1;
2375}
2376secno += bvr->part_boff;
2377
2378cache_valid = false;
2379
2380while (len > 0)
2381{
2382secs = len / BPS;
2383if (secs > N_CACHE_SECS)
2384{
2385secs = N_CACHE_SECS;
2386}
2387copy_len = secs * BPS;
2388
2389bcopy( cbuf, trackbuf, copy_len );
2390//printf("rdr: ebioswrite(%d, %d, %d)\n", bvr->biosdev, secno, secs);
2391if ((rc = ebioswrite(bvr->biosdev, secno, secs)) != 0)
2392{
2393error(" EBIOS write error: %s\n", bios_error(rc), rc);
2394error(" Block %d Sectors %d\n", secno, secs);
2395return rc;
2396}
2397
2398len -= copy_len;
2399cbuf += copy_len;
2400secno += secs;
2401spinActivityIndicator(secs);
2402}
2403
2404return 0;
2405}
2406
2407//==============================================================================
2408
2409int diskIsCDROM(BVRef bvr)
2410{
2411struct driveInfo di;
2412
2413if (getDriveInfo(bvr->biosdev, &di) == 0 && di.no_emulation)
2414{
2415return 1;
2416}
2417return 0;
2418}
2419
2420//==============================================================================
2421
2422int biosDevIsCDROM(int biosdev)
2423{
2424struct driveInfo di;
2425
2426if (getDriveInfo(biosdev, &di) == 0 && di.no_emulation)
2427{
2428return 1;
2429}
2430return 0;
2431}
2432

Archive Download this file

Revision: 2471