/*
 * Copyright 2009, Johannes Wischert, johanneswi@gmail.com. All rights reserved.
 * Distributed under the terms of the MIT License.
 */


#include <KernelExport.h>

#include <boot/disk_identifier.h>
#include <boot/vfs.h>
#include <boot/platform.h>
#include <boot/partitions.h>
#include <boot/stage2.h>
#include <boot/stdio.h>

#define TRACE_DEVICES
#ifdef TRACE_DEVICES
#       define TRACE(x...) dprintf(x)
#else
#       define TRACE(x...)
#endif




status_t 
platform_add_boot_device(struct stage2_args *args, NodeList *devicesList)
{
        TRACE("platform_add_boot_device\n");

        if (!args->platform.boot_tgz_data || !args->platform.boot_tgz_size)
                return B_DEVICE_NOT_FOUND;

        TRACE("Memory Disk at: %p size: %lx\n", args->platform.boot_tgz_data,
                args->platform.boot_tgz_size);

        MemoryDisk* disk = new(nothrow) MemoryDisk(
                (const uint8 *)args->platform.boot_tgz_data,
                args->platform.boot_tgz_size, "boot.tgz");
        if (!disk) {
                dprintf("platform_add_boot_device(): Could not create MemoryDisk !\n");
                return B_NO_MEMORY;
        }

        devicesList->Add(disk);
        return B_OK;
}


status_t
platform_get_boot_partitions(struct stage2_args *args, Node *device,
        NodeList *list, NodeList *partitionList)
{
        TRACE("platform_get_boot_partition\n");

        NodeIterator iterator = list->GetIterator();
        boot::Partition *partition = NULL;
        while ((partition = (boot::Partition *)iterator.Next()) != NULL) {
                // ToDo: just take the first partition for now
                partitionList->Insert(partition);
                return B_OK;
        }

        return B_ENTRY_NOT_FOUND;
}


status_t
platform_add_block_devices(stage2_args *args, NodeList *devicesList)
{
        TRACE("platform_add_block_devices\n");
        return B_OK;
}


status_t 
platform_register_boot_device(Node *device)
{
        disk_identifier disk_ident;
        disk_ident.bus_type = UNKNOWN_BUS;
        disk_ident.device_type = UNKNOWN_DEVICE;
        disk_ident.device.unknown.size = device->Size();

        for (int32 i = 0; i < NUM_DISK_CHECK_SUMS; i++) {
                disk_ident.device.unknown.check_sums[i].offset = -1;
                disk_ident.device.unknown.check_sums[i].sum = 0;
        }

        gBootParams.SetData(BOOT_VOLUME_DISK_IDENTIFIER, B_RAW_TYPE,
                &disk_ident, sizeof(disk_ident));

        return B_OK;
}


void
platform_cleanup_devices()
{
}