// /***************************************************************************
// The Disc Image Chef
// ----------------------------------------------------------------------------
//
// Filename       : open.c
// Author(s)      : Natalia Portillo <claunia@claunia.com>
//
// Component      : libdicformat.
//
// --[ Description ] ----------------------------------------------------------
//
//     Handles opening DiscImageChef format disk images.
//
// --[ License ] --------------------------------------------------------------
//
//     This library is free software; you can redistribute it and/or modify
//     it under the terms of the GNU Lesser General Public License as
//     published by the Free Software Foundation; either version 2.1 of the
//     License, or (at your option) any later version.
//
//     This library is distributed in the hope that it will be useful, but
//     WITHOUT ANY WARRANTY; without even the implied warranty of
//     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
//     Lesser General Public License for more details.
//
//     You should have received a copy of the GNU Lesser General Public
//     License along with this library; if not, see <http://www.gnu.org/licenses/>.
//
// ----------------------------------------------------------------------------
// Copyright © 2011-2019 Natalia Portillo
// ****************************************************************************/

#include <dicformat.h>
#include <malloc.h>
#include <string.h>

int32_t read_media_tag(void *context, uint8_t *data, int32_t tag, uint32_t *length)
{
    dicformatContext *ctx;
    dataLinkedList   *item;

    if(context == NULL)
        return DICF_ERROR_NOT_DICFORMAT;

    ctx = context;

    // TODO: Cast this field without casting the whole structure, as this can buffer overflow
    // Not a libdicformat context
    if(ctx->magic != DIC_MAGIC)
        return DICF_ERROR_NOT_DICFORMAT;

    item = ctx->mediaTagsHead;

    while(item != NULL)
    {
        if(item->type == tag)
        {
            if(data == NULL || *length < item->length)
            {
                *length = item->length;
                return DICF_ERROR_BUFFER_TOO_SMALL;
            }

            *length = item->length;
            memcpy(data, item->data, item->length);
        }

        item = item->next;
    }

    *length = 0;
    return DICF_ERROR_MEDIA_TAG_NOT_PRESENT;
}

int32_t read_sector(void *context, uint64_t sectorAddress, uint8_t *data, uint32_t *length)
{
    dicformatContext *ctx;
    uint64_t         ddtEntry;
    uint32_t         offsetMask;
    uint64_t         offset;
    uint64_t         blockOffset;
    BlockHeader      blockHeader;
    uint8_t          *block;
    size_t           readBytes;

    if(context == NULL)
        return DICF_ERROR_NOT_DICFORMAT;

    ctx = context;

    // TODO: Cast this field without casting the whole structure, as this can buffer overflow
    // Not a libdicformat context
    if(ctx->magic != DIC_MAGIC)
        return DICF_ERROR_NOT_DICFORMAT;

    if(sectorAddress > ctx->imageInfo.Sectors - 1)
        return DICF_ERROR_SECTOR_OUT_OF_BOUNDS;

    ddtEntry    = ctx->userDataDdt[sectorAddress];
    offsetMask  = (uint32_t)((1 << ctx->shift) - 1);
    offset      = ddtEntry & offsetMask;
    blockOffset = ddtEntry >> ctx->shift;

    // Partially written image... as we can't know the real sector size just assume it's common :/
    if(ddtEntry == 0)
    {
        memset(data, 0, ctx->imageInfo.SectorSize);
        *length = ctx->imageInfo.SectorSize;
        return DICF_STATUS_SECTOR_NEVER_WRITTEN;
    }

    // Check if block is cached
    // TODO: Caches

    // Read block header
    fseek(ctx->imageStream, blockOffset, SEEK_SET);
    readBytes = fread(&blockHeader, sizeof(BlockHeader), 1, ctx->imageStream);

    if(readBytes != sizeof(BlockHeader))
        return DICF_ERROR_CANNOT_READ_HEADER;

    if(data == NULL || *length < blockHeader.sectorSize)
    {
        *length = blockHeader.sectorSize;
        return DICF_ERROR_BUFFER_TOO_SMALL;
    }

    // Decompress block
    switch(blockHeader.compression)
    {
        case None:block = (uint8_t *)malloc(blockHeader.length);
            if(block == NULL)
                return DICF_ERROR_NOT_ENOUGH_MEMORY;

            readBytes = fread(block, blockHeader.length, 1, ctx->imageStream);

            if(readBytes != blockHeader.length)
            {
                free(block);
                return DICF_ERROR_CANNOT_READ_BLOCK;
            }

            break;
        default:return DICF_ERROR_UNSUPPORTED_COMPRESSION;
    }

    // Check if cache needs to be emptied
    // TODO: Caches

    // Add block to cache
    // TODO: Caches

    memcpy(data, block + offset, blockHeader.sectorSize);
    *length = blockHeader.sectorSize;
    free(block);
    return DICF_STATUS_OK;
}

int32_t read_track_sector(void *context, uint8_t *data, uint64_t sectorAddress, uint32_t *length, uint8_t track)
{
    dicformatContext *ctx;
    int              i;

    if(context == NULL)
        return DICF_ERROR_NOT_DICFORMAT;

    ctx = context;

    // TODO: Cast this field without casting the whole structure, as this can buffer overflow
    // Not a libdicformat context
    if(ctx->magic != DIC_MAGIC)
        return DICF_ERROR_NOT_DICFORMAT;

    if(ctx->imageInfo.XmlMediaType != OpticalDisc)
        return DICF_ERROR_INCORRECT_MEDIA_TYPE;

    for(i = 0; i < ctx->numberOfDataTracks; i++)
    {
        if(ctx->dataTracks[i].sequence == track)
        {
            return read_sector(context, ctx->dataTracks[i].start + sectorAddress, data, length);
        }
    }

    return DICF_ERROR_TRACK_NOT_FOUND;
}