apple.cc

/*
 * Copyright (c) 2008-2014, Pedigree Developers
 *
 * Please see the CONTRIB file in the root of the source tree for a full
 * list of contributors.
 *
 * Permission to use, copy, modify, and distribute this software for any
 * purpose with or without fee is hereby granted, provided that the above
 * copyright notice and this permission notice appear in all copies.
 *
 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 */

#include "apple.h"
#include "Partition.h"
#include "pedigree/kernel/Log.h"
#include "pedigree/kernel/machine/Disk.h"
#include "pedigree/kernel/processor/types.h"
#include "pedigree/kernel/utilities/String.h"
#include "pedigree/kernel/utilities/utility.h"

class Device;

bool appleProbeDisk(Disk *pDisk)
{
    // Read the second sector (512 bytes) of the disk into a buffer.
    uintptr_t buff;
    if ((buff = pDisk->read(512ULL)) == 0)
    {
        WARNING("Disk read failure during Apple partition table search.");
        return false;
    }

    ApplePartitionMap *pMap = reinterpret_cast<ApplePartitionMap *>(buff);

    String diskName;
    pDisk->getName(diskName);

    // Check for the magic signature.
    if (pMap->pmSig != BIG_TO_HOST16(APPLE_PM_SIG))
    {
        NOTICE("Apple partition map not found on disk " << diskName);
        return false;
    }

    NOTICE("Apple partition map found on disk " << diskName);

    // Cache the number of partition table entries.
    size_t nEntries = BIG_TO_HOST32(pMap->pmMapBlkCnt);
    for (size_t i = 0; i < nEntries; i++)
    {
        if (i > 0)  // We don't need to load anything in for the first pmap -
                    // already done!
        {
            if ((buff = pDisk->read(512ULL + i * 512ULL)) == 0)
            {
                WARNING(
                    "Disk read failure during partition table recognition.");
                return false;
            }
            pMap = reinterpret_cast<ApplePartitionMap *>(buff);
        }

        NOTICE(
            "Detected partition '" << pMap->pmPartName << "', type '"
                                   << pMap->pmParType << "'");

        // Create a partition object.
        Partition *pObj = new Partition(
            String(pMap->pmParType),
            static_cast<uint64_t>(BIG_TO_HOST32(pMap->pmPyPartStart)) * 512ULL,
            static_cast<uint64_t>(BIG_TO_HOST32(pMap->pmPartBlkCnt)) * 512ULL);
        pObj->setParent(static_cast<Device *>(pDisk));
        pDisk->addChild(static_cast<Device *>(pObj));
    }
    return true;
}