close.c

Go to the documentation of this file.

/*
 * This file is part of the Aaru Data Preservation Suite.
 * Copyright (c) 2019-2025 Natalia Portillo.
 *
 * 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/>.
 */
 
#include <errno.h>
#include <stdio.h>
#include <stdlib.h>
 
#ifdef __linux__
#include <sys/mman.h>
#endif
 
#include <aaruformat.h>
 
#include "internal.h"
#include "log.h"

static int32_t write_cached_secondary_ddt(aaruformat_context *ctx)
{
    // Write cached secondary table to file end and update primary table entry with its position
    // Check if we have a cached table that needs to be written (either it has an offset or exists in memory)
    bool has_cached_secondary_ddt =
        ctx->user_data_ddt_header.tableShift > 0 && (ctx->cached_ddt_offset != 0 || ctx->cached_secondary_ddt2 != NULL);
 
    if(!has_cached_secondary_ddt) return AARUF_STATUS_OK;
 
    TRACE("Writing cached secondary DDT table to file");
 
    fseek(ctx->imageStream, 0, SEEK_END);
    long end_of_file = ftell(ctx->imageStream);
 
    // Align the position according to block alignment shift
    uint64_t alignment_mask = (1ULL << ctx->user_data_ddt_header.blockAlignmentShift) - 1;
    if(end_of_file & alignment_mask)
    {
        // Calculate the next aligned position
        uint64_t aligned_position = end_of_file + alignment_mask & ~alignment_mask;
 
        // Seek to the aligned position and pad with zeros if necessary
        fseek(ctx->imageStream, aligned_position, SEEK_SET);
        end_of_file = aligned_position;
 
        TRACE("Aligned DDT write position from %ld to %" PRIu64 " (alignment shift: %d)",
              ftell(ctx->imageStream) - (aligned_position - end_of_file), aligned_position,
              ctx->user_data_ddt_header.blockAlignmentShift);
    }
 
    // Prepare DDT header for the cached table
    DdtHeader2 ddt_header          = {0};
    ddt_header.identifier          = DeDuplicationTableSecondary;
    ddt_header.type                = UserData;
    ddt_header.compression         = ctx->compression_enabled ? Lzma : None;
    ddt_header.levels              = ctx->user_data_ddt_header.levels;
    ddt_header.tableLevel          = ctx->user_data_ddt_header.tableLevel + 1;
    ddt_header.previousLevelOffset = ctx->primary_ddt_offset;
    ddt_header.negative            = ctx->user_data_ddt_header.negative;
    ddt_header.overflow            = ctx->user_data_ddt_header.overflow;
    ddt_header.blockAlignmentShift = ctx->user_data_ddt_header.blockAlignmentShift;
    ddt_header.dataShift           = ctx->user_data_ddt_header.dataShift;
    ddt_header.tableShift          = 0;  // Secondary tables are single level
 
    uint64_t items_per_ddt_entry = 1 << ctx->user_data_ddt_header.tableShift;
    ddt_header.blocks            = items_per_ddt_entry;
    ddt_header.entries           = items_per_ddt_entry;
    ddt_header.start             = ctx->cached_ddt_position * items_per_ddt_entry;
 
    // Calculate data size
    ddt_header.length = items_per_ddt_entry * sizeof(uint64_t);
 
    // Calculate CRC64 of the data
    crc64_ctx *crc64_context = aaruf_crc64_init();
    if(crc64_context != NULL)
    {
        aaruf_crc64_update(crc64_context, (uint8_t *)ctx->cached_secondary_ddt2, (uint32_t)ddt_header.length);
 
        uint64_t crc64;
        aaruf_crc64_final(crc64_context, &crc64);
        ddt_header.crc64 = crc64;
    }
 
    uint8_t *buffer                                  = NULL;
    uint8_t  lzma_properties[LZMA_PROPERTIES_LENGTH] = {0};
 
    if(ddt_header.compression == None)
    {
        buffer              = (uint8_t *)ctx->cached_secondary_ddt2;
        ddt_header.cmpCrc64 = ddt_header.crc64;
    }
    else
    {
        buffer = malloc((size_t)ddt_header.length * 2);  // Allocate double size for compression
        if(buffer == NULL)
        {
            TRACE("Failed to allocate memory for secondary DDT v2 compression");
            return AARUF_ERROR_NOT_ENOUGH_MEMORY;
        }
 
        size_t dst_size   = (size_t)ddt_header.length * 2 * 2;
        size_t props_size = LZMA_PROPERTIES_LENGTH;
        aaruf_lzma_encode_buffer(buffer, &dst_size,
 
                                 (uint8_t *)ctx->cached_secondary_ddt2, ddt_header.length, lzma_properties, &props_size,
                                 9, ctx->lzma_dict_size, 4, 0, 2, 273, 8);
 
        ddt_header.cmpLength = (uint32_t)dst_size;
 
        if(ddt_header.cmpLength >= ddt_header.length)
        {
            ddt_header.compression = None;
            free(buffer);
            buffer = (uint8_t *)ctx->cached_secondary_ddt2;
        }
    }
 
    if(ddt_header.compression == None)
    {
        ddt_header.cmpLength = ddt_header.length;
        ddt_header.cmpCrc64  = ddt_header.crc64;
    }
    else
        ddt_header.cmpCrc64 = aaruf_crc64_data(buffer, ddt_header.cmpLength);
 
    if(ddt_header.compression == Lzma) ddt_header.cmpLength += LZMA_PROPERTIES_LENGTH;
 
    // Write header
    if(fwrite(&ddt_header, sizeof(DdtHeader2), 1, ctx->imageStream) == 1)
    {
        if(ddt_header.compression == Lzma) fwrite(lzma_properties, LZMA_PROPERTIES_LENGTH, 1, ctx->imageStream);
 
        // Write data
        if(fwrite(buffer, ddt_header.cmpLength, 1, ctx->imageStream) == 1)
        {
            // Update primary table entry to point to new location
            const uint64_t new_secondary_table_block_offset =
                end_of_file >> ctx->user_data_ddt_header.blockAlignmentShift;
 
            ctx->user_data_ddt2[ctx->cached_ddt_position] = (uint64_t)new_secondary_table_block_offset;
 
            // Update index: remove old entry for cached DDT and add new one
            TRACE("Updating index for cached secondary DDT");
 
            // Remove old index entry for the cached DDT
            if(ctx->cached_ddt_offset != 0)
            {
                TRACE("Removing old index entry for DDT at offset %" PRIu64, ctx->cached_ddt_offset);
                const IndexEntry *entry = NULL;
 
                // Find and remove the old index entry
                for(unsigned int k = 0; k < utarray_len(ctx->index_entries); k++)
                {
                    entry = (IndexEntry *)utarray_eltptr(ctx->index_entries, k);
                    if(entry && entry->offset == ctx->cached_ddt_offset &&
                       entry->blockType == DeDuplicationTableSecondary)
                    {
                        TRACE("Found old DDT index entry at position %u, removing", k);
                        utarray_erase(ctx->index_entries, k, 1);
                        break;
                    }
                }
            }
 
            // Add new index entry for the newly written secondary DDT
            IndexEntry new_ddt_entry;
            new_ddt_entry.blockType = DeDuplicationTableSecondary;
            new_ddt_entry.dataType  = UserData;
            new_ddt_entry.offset    = end_of_file;
 
            utarray_push_back(ctx->index_entries, &new_ddt_entry);
            TRACE("Added new DDT index entry at offset %" PRIu64, end_of_file);
 
            // Write the updated primary table back to its original position in the file
            long saved_pos = ftell(ctx->imageStream);
            fseek(ctx->imageStream, ctx->primary_ddt_offset + sizeof(DdtHeader2), SEEK_SET);
 
            size_t primary_table_size = ctx->user_data_ddt_header.entries * sizeof(uint64_t);
 
            size_t primary_written_bytes = 0;
            primary_written_bytes        = fwrite(ctx->user_data_ddt2, primary_table_size, 1, ctx->imageStream);
 
            if(primary_written_bytes != 1)
            {
                TRACE("Could not flush primary DDT table to file.");
                return AARUF_ERROR_CANNOT_WRITE_HEADER;
            }
 
            fseek(ctx->imageStream, saved_pos, SEEK_SET);
        }
        else
            TRACE("Failed to write cached secondary DDT data");
    }
    else
        TRACE("Failed to write cached secondary DDT header");
 
    // Free the cached table
    free(ctx->cached_secondary_ddt2);
    ctx->cached_secondary_ddt2 = NULL;
    ctx->cached_ddt_offset     = 0;
 
    // Set position
    fseek(ctx->imageStream, 0, SEEK_END);
 
    if(ddt_header.compression == Lzma) free(buffer);
 
    return AARUF_STATUS_OK;
}

static int32_t write_primary_ddt(aaruformat_context *ctx)
{
    // Write the cached primary DDT table back to its position in the file
    if(ctx->user_data_ddt_header.tableShift <= 0 || ctx->user_data_ddt2 == NULL) return AARUF_STATUS_OK;
 
    TRACE("Writing cached primary DDT table back to file");
 
    // Calculate CRC64 of the primary DDT table data first
    crc64_ctx *crc64_context = aaruf_crc64_init();
    if(crc64_context != NULL)
    {
        size_t primary_table_size = ctx->user_data_ddt_header.entries * sizeof(uint64_t);
 
        aaruf_crc64_update(crc64_context, (uint8_t *)ctx->user_data_ddt2, primary_table_size);
 
        uint64_t crc64;
        aaruf_crc64_final(crc64_context, &crc64);
 
        // Properly populate all header fields for multi-level DDT primary table
        ctx->user_data_ddt_header.identifier  = DeDuplicationTable2;
        ctx->user_data_ddt_header.type        = UserData;
        ctx->user_data_ddt_header.compression = None;
        // levels, tableLevel, previousLevelOffset, negative, overflow, blockAlignmentShift,
        // dataShift, tableShift, sizeType, entries, blocks, start are already set during creation
        ctx->user_data_ddt_header.crc64       = crc64;
        ctx->user_data_ddt_header.cmpCrc64    = crc64;
        ctx->user_data_ddt_header.length      = primary_table_size;
        ctx->user_data_ddt_header.cmpLength   = primary_table_size;
 
        TRACE("Calculated CRC64 for primary DDT: 0x%16lX", crc64);
    }
 
    // First write the DDT header
    fseek(ctx->imageStream, ctx->primary_ddt_offset, SEEK_SET);
 
    size_t headerWritten = fwrite(&ctx->user_data_ddt_header, sizeof(DdtHeader2), 1, ctx->imageStream);
    if(headerWritten != 1)
    {
        TRACE("Failed to write primary DDT header to file");
        return AARUF_ERROR_CANNOT_WRITE_HEADER;
    }
 
    // Then write the table data (position is already after the header)
    size_t primary_table_size = ctx->user_data_ddt_header.entries * sizeof(uint64_t);
 
    // Write the primary table data
    size_t written_bytes = 0;
    written_bytes        = fwrite(ctx->user_data_ddt2, primary_table_size, 1, ctx->imageStream);
 
    if(written_bytes == 1)
    {
        TRACE("Successfully wrote primary DDT header and table to file (%" PRIu64 " entries, %zu bytes)",
              ctx->user_data_ddt_header.entries, primary_table_size);
 
        // Add primary DDT to index
        TRACE("Adding primary DDT to index");
        IndexEntry primary_ddt_entry;
        primary_ddt_entry.blockType = DeDuplicationTable2;
        primary_ddt_entry.dataType  = UserData;
        primary_ddt_entry.offset    = ctx->primary_ddt_offset;
 
        utarray_push_back(ctx->index_entries, &primary_ddt_entry);
        TRACE("Added primary DDT index entry at offset %" PRIu64, ctx->primary_ddt_offset);
    }
    else
        TRACE("Failed to write primary DDT table to file");
 
    return AARUF_STATUS_OK;
}

static int32_t write_single_level_ddt(aaruformat_context *ctx)
{
    // Write the single level DDT table block aligned just after the header
    if(ctx->user_data_ddt_header.tableShift != 0 || ctx->user_data_ddt2 == NULL) return AARUF_STATUS_OK;
 
    TRACE("Writing single-level DDT table to file");
 
    // Calculate CRC64 of the primary DDT table data
    const size_t primary_table_size = ctx->user_data_ddt_header.entries * sizeof(uint64_t);
 
    // Properly populate all header fields
    ctx->user_data_ddt_header.identifier          = DeDuplicationTable2;
    ctx->user_data_ddt_header.type                = UserData;
    ctx->user_data_ddt_header.compression         = ctx->compression_enabled ? Lzma : None;
    ctx->user_data_ddt_header.levels              = 1;  // Single level
    ctx->user_data_ddt_header.tableLevel          = 0;  // Top level
    ctx->user_data_ddt_header.previousLevelOffset = 0;  // No previous level for single-level DDT
    // negative and overflow are already set during creation
    // blockAlignmentShift, dataShift, tableShift, sizeType, entries, blocks, start are already set
    ctx->user_data_ddt_header.length              = primary_table_size;
    ctx->user_data_ddt_header.cmpLength           = primary_table_size;
 
    ctx->user_data_ddt_header.crc64 = aaruf_crc64_data((uint8_t *)ctx->user_data_ddt2, primary_table_size);
 
    TRACE("Calculated CRC64 for single-level DDT: 0x%16lX", ctx->user_data_ddt_header.crc64);
 
    uint8_t *cmp_buffer                              = NULL;
    uint8_t  lzma_properties[LZMA_PROPERTIES_LENGTH] = {0};
 
    if(ctx->user_data_ddt_header.compression == None)
    {
 
        cmp_buffer                         = (uint8_t *)ctx->user_data_ddt2;
        ctx->user_data_ddt_header.cmpCrc64 = ctx->user_data_ddt_header.crc64;
    }
    else
    {
        cmp_buffer = malloc((size_t)ctx->user_data_ddt_header.length * 2);  // Allocate double size for compression
        if(cmp_buffer == NULL)
        {
            TRACE("Failed to allocate memory for secondary DDT v2 compression");
            return AARUF_ERROR_NOT_ENOUGH_MEMORY;
        }
 
        size_t dst_size   = (size_t)ctx->user_data_ddt_header.length * 2 * 2;
        size_t props_size = LZMA_PROPERTIES_LENGTH;
        aaruf_lzma_encode_buffer(cmp_buffer, &dst_size, (uint8_t *)ctx->user_data_ddt2,
                                 ctx->user_data_ddt_header.length, lzma_properties, &props_size, 9, ctx->lzma_dict_size,
                                 4, 0, 2, 273, 8);
 
        ctx->user_data_ddt_header.cmpLength = (uint32_t)dst_size;
 
        if(ctx->user_data_ddt_header.cmpLength >= ctx->user_data_ddt_header.length)
        {
            ctx->user_data_ddt_header.compression = None;
            free(cmp_buffer);
 
            cmp_buffer = (uint8_t *)ctx->user_data_ddt2;
        }
    }
 
    if(ctx->user_data_ddt_header.compression == None)
    {
        ctx->user_data_ddt_header.cmpLength = ctx->user_data_ddt_header.length;
        ctx->user_data_ddt_header.cmpCrc64  = ctx->user_data_ddt_header.crc64;
    }
    else
        ctx->user_data_ddt_header.cmpCrc64 =
            aaruf_crc64_data(cmp_buffer, (uint32_t)ctx->user_data_ddt_header.cmpLength);
 
    if(ctx->user_data_ddt_header.compression == Lzma) ctx->user_data_ddt_header.cmpLength += LZMA_PROPERTIES_LENGTH;
 
    // Write the DDT header first
    fseek(ctx->imageStream, 0, SEEK_END);
    long           ddt_position   = ftell(ctx->imageStream);
    // Align index position to block boundary if needed
    const uint64_t alignment_mask = (1ULL << ctx->user_data_ddt_header.blockAlignmentShift) - 1;
    if(ddt_position & alignment_mask)
    {
        const uint64_t aligned_position = ddt_position + alignment_mask & ~alignment_mask;
        fseek(ctx->imageStream, aligned_position, SEEK_SET);
        ddt_position = aligned_position;
    }
 
    const size_t header_written = fwrite(&ctx->user_data_ddt_header, sizeof(DdtHeader2), 1, ctx->imageStream);
    if(header_written != 1)
    {
        TRACE("Failed to write single-level DDT header to file");
        return AARUF_ERROR_CANNOT_WRITE_HEADER;
    }
 
    // Write the primary table data
    size_t written_bytes = 0;
    if(ctx->user_data_ddt_header.compression == Lzma)
        fwrite(lzma_properties, LZMA_PROPERTIES_LENGTH, 1, ctx->imageStream);
 
    written_bytes = fwrite(cmp_buffer, ctx->user_data_ddt_header.cmpLength, 1, ctx->imageStream);
 
    if(written_bytes == 1)
    {
        TRACE("Successfully wrote single-level DDT header and table to file (%" PRIu64
              " entries, %zu bytes, %zu compressed bytes)",
              ctx->user_data_ddt_header.entries, ctx->user_data_ddt_header.length, ctx->user_data_ddt_header.cmpLength);
 
        // Add single-level DDT to index
        TRACE("Adding single-level DDT to index");
        IndexEntry single_ddt_entry;
        single_ddt_entry.blockType = DeDuplicationTable2;
        single_ddt_entry.dataType  = UserData;
        single_ddt_entry.offset    = ddt_position;
 
        utarray_push_back(ctx->index_entries, &single_ddt_entry);
        TRACE("Added single-level DDT index entry at offset %" PRIu64, ddt_position);
    }
    else
        TRACE("Failed to write single-level DDT table data to file");
 
    return AARUF_STATUS_OK;
}

static int32_t write_tape_ddt(aaruformat_context *ctx)
{
    if(!ctx->is_tape) return AARUF_STATUS_INVALID_CONTEXT;
 
    // Traverse the tape DDT uthash and find the biggest key
    uint64_t          max_key = 0;
    TapeDdtHashEntry *entry, *tmp;
    HASH_ITER(hh, ctx->tape_ddt, entry, tmp)
    if(entry->key > max_key) max_key = entry->key;
 
    // Initialize context user data DDT header
    ctx->user_data_ddt_header.identifier          = DeDuplicationTable2;
    ctx->user_data_ddt_header.type                = UserData;
    ctx->user_data_ddt_header.compression         = ctx->compression_enabled ? Lzma : None;
    ctx->user_data_ddt_header.levels              = 1;  // Single level
    ctx->user_data_ddt_header.tableLevel          = 0;  // Top level
    ctx->user_data_ddt_header.previousLevelOffset = 0;  // No previous level for single-level DDT
    ctx->user_data_ddt_header.negative            = 0;
    ctx->user_data_ddt_header.overflow            = 0;
    ctx->user_data_ddt_header.tableShift          = 0;  // Single level
    ctx->user_data_ddt_header.entries             = max_key + 1;
    ctx->user_data_ddt_header.blocks              = max_key + 1;
    ctx->user_data_ddt_header.start               = 0;
    ctx->user_data_ddt_header.length              = ctx->user_data_ddt_header.entries * sizeof(uint64_t);
    ctx->user_data_ddt_header.cmpLength           = ctx->user_data_ddt_header.length;
 
    // Initialize memory for user data DDT
    ctx->user_data_ddt2 = calloc(ctx->user_data_ddt_header.entries, sizeof(uint64_t));
    if(ctx->user_data_ddt2 == NULL)
    {
        TRACE("Failed to allocate memory for tape DDT table");
        return AARUF_ERROR_NOT_ENOUGH_MEMORY;
    }
 
    // Populate user data DDT from tape DDT uthash
    HASH_ITER(hh, ctx->tape_ddt, entry, tmp)
    if(entry->key < ctx->user_data_ddt_header.blocks) ctx->user_data_ddt2[entry->key] = entry->value;
 
    // Do not repeat code
    return write_single_level_ddt(ctx);
}

static void write_checksum_block(aaruformat_context *ctx)
{
    uint64_t alignment_mask;
    uint64_t aligned_position;
 
    // Finalize pending checksums
    if(ctx->calculating_md5)
    {
        ctx->checksums.hasMd5 = true;
        aaruf_md5_final(&ctx->md5_context, ctx->checksums.md5);
    }
    if(ctx->calculating_sha1)
    {
        ctx->checksums.hasSha1 = true;
        aaruf_sha1_final(&ctx->sha1_context, ctx->checksums.sha1);
    }
    if(ctx->calculating_sha256)
    {
        ctx->checksums.hasSha256 = true;
        aaruf_sha256_final(&ctx->sha256_context, ctx->checksums.sha256);
    }
    if(ctx->calculating_spamsum)
    {
        ctx->checksums.hasSpamSum = true;
        ctx->checksums.spamsum    = calloc(1, FUZZY_MAX_RESULT);
        aaruf_spamsum_final(ctx->spamsum_context, ctx->checksums.spamsum);
        aaruf_spamsum_free(ctx->spamsum_context);
    }
    if(ctx->calculating_blake3)
    {
        ctx->checksums.hasBlake3 = true;
        blake3_hasher_finalize(ctx->blake3_context, ctx->checksums.blake3, BLAKE3_OUT_LEN);
        free(ctx->blake3_context);
    }
 
    // Write the checksums block
    bool has_checksums = ctx->checksums.hasMd5 || ctx->checksums.hasSha1 || ctx->checksums.hasSha256 ||
                         ctx->checksums.hasSpamSum || ctx->checksums.hasBlake3;
 
    if(!has_checksums) return;
 
    ChecksumHeader checksum_header = {0};
    checksum_header.identifier     = ChecksumBlock;
 
    fseek(ctx->imageStream, 0, SEEK_END);
    long checksum_position = ftell(ctx->imageStream);
    // Align index position to block boundary if needed
    alignment_mask         = (1ULL << ctx->user_data_ddt_header.blockAlignmentShift) - 1;
    if(checksum_position & alignment_mask)
    {
        aligned_position = checksum_position + alignment_mask & ~alignment_mask;
        fseek(ctx->imageStream, aligned_position, SEEK_SET);
        checksum_position = aligned_position;
    }
 
    // Skip checksum_header
    fseek(ctx->imageStream, sizeof(checksum_header), SEEK_CUR);
 
    if(ctx->checksums.hasMd5)
    {
        TRACE("Writing MD5 checksum entry");
        ChecksumEntry md5_entry = {0};
        md5_entry.length        = MD5_DIGEST_LENGTH;
        md5_entry.type          = Md5;
        fwrite(&md5_entry, sizeof(ChecksumEntry), 1, ctx->imageStream);
        fwrite(&ctx->checksums.md5, MD5_DIGEST_LENGTH, 1, ctx->imageStream);
        checksum_header.length += sizeof(ChecksumEntry) + MD5_DIGEST_LENGTH;
        checksum_header.entries++;
    }
 
    if(ctx->checksums.hasSha1)
    {
        TRACE("Writing SHA1 checksum entry");
        ChecksumEntry sha1_entry = {0};
        sha1_entry.length        = SHA1_DIGEST_LENGTH;
        sha1_entry.type          = Sha1;
        fwrite(&sha1_entry, sizeof(ChecksumEntry), 1, ctx->imageStream);
        fwrite(&ctx->checksums.sha1, SHA1_DIGEST_LENGTH, 1, ctx->imageStream);
        checksum_header.length += sizeof(ChecksumEntry) + SHA1_DIGEST_LENGTH;
        checksum_header.entries++;
    }
 
    if(ctx->checksums.hasSha256)
    {
        TRACE("Writing SHA256 checksum entry");
        ChecksumEntry sha256_entry = {0};
        sha256_entry.length        = SHA256_DIGEST_LENGTH;
        sha256_entry.type          = Sha256;
        fwrite(&sha256_entry, sizeof(ChecksumEntry), 1, ctx->imageStream);
        fwrite(&ctx->checksums.sha256, SHA256_DIGEST_LENGTH, 1, ctx->imageStream);
        checksum_header.length += sizeof(ChecksumEntry) + SHA256_DIGEST_LENGTH;
        checksum_header.entries++;
    }
 
    if(ctx->checksums.hasSpamSum)
    {
        TRACE("Writing SpamSum checksum entry");
        ChecksumEntry spamsum_entry = {0};
        spamsum_entry.length        = strlen((const char *)ctx->checksums.spamsum);
        spamsum_entry.type          = SpamSum;
        fwrite(&spamsum_entry, sizeof(ChecksumEntry), 1, ctx->imageStream);
        fwrite(ctx->checksums.spamsum, spamsum_entry.length, 1, ctx->imageStream);
        checksum_header.length += sizeof(ChecksumEntry) + spamsum_entry.length;
        checksum_header.entries++;
    }
 
    if(ctx->checksums.hasBlake3)
    {
        TRACE("Writing BLAKE3 checksum entry");
        ChecksumEntry blake3_entry = {0};
        blake3_entry.length        = BLAKE3_OUT_LEN;
        blake3_entry.type          = Blake3;
        fwrite(&blake3_entry, sizeof(ChecksumEntry), 1, ctx->imageStream);
        fwrite(&ctx->checksums.blake3, BLAKE3_OUT_LEN, 1, ctx->imageStream);
        checksum_header.length += sizeof(ChecksumEntry) + BLAKE3_OUT_LEN;
        checksum_header.entries++;
        ctx->header.featureCompatible |= AARU_FEATURE_RW_BLAKE3;
    }
 
    fseek(ctx->imageStream, checksum_position, SEEK_SET);
    TRACE("Writing checksum header");
    fwrite(&checksum_header, sizeof(ChecksumHeader), 1, ctx->imageStream);
 
    // Add checksum block to index
    TRACE("Adding checksum block to index");
    IndexEntry checksum_index_entry;
    checksum_index_entry.blockType = ChecksumBlock;
    checksum_index_entry.dataType  = 0;
    checksum_index_entry.offset    = checksum_position;
 
    utarray_push_back(ctx->index_entries, &checksum_index_entry);
    TRACE("Added checksum block index entry at offset %" PRIu64, checksum_position);
}

static void write_tracks_block(aaruformat_context *ctx)
{
    // Write tracks block
    if(ctx->tracks_header.entries <= 0 || ctx->track_entries == NULL) return;
 
    fseek(ctx->imageStream, 0, SEEK_END);
    long     tracks_position = ftell(ctx->imageStream);
    // Align index position to block boundary if needed
    uint64_t alignment_mask  = (1ULL << ctx->user_data_ddt_header.blockAlignmentShift) - 1;
    if(tracks_position & alignment_mask)
    {
        uint64_t aligned_position = tracks_position + alignment_mask & ~alignment_mask;
        fseek(ctx->imageStream, aligned_position, SEEK_SET);
        tracks_position = aligned_position;
    }
 
    TRACE("Writing tracks block at position %ld", tracks_position);
    // Write header
    if(fwrite(&ctx->tracks_header, sizeof(TracksHeader), 1, ctx->imageStream) == 1)
    {
        // Write entries
        size_t written_entries =
            fwrite(ctx->track_entries, sizeof(TrackEntry), ctx->tracks_header.entries, ctx->imageStream);
 
        if(written_entries == ctx->tracks_header.entries)
        {
            TRACE("Successfully wrote tracks block with %u entries", ctx->tracks_header.entries);
            // Add tracks block to index
            TRACE("Adding tracks block to index");
 
            IndexEntry tracks_index_entry;
            tracks_index_entry.blockType = TracksBlock;
            tracks_index_entry.dataType  = 0;
            tracks_index_entry.offset    = tracks_position;
            utarray_push_back(ctx->index_entries, &tracks_index_entry);
            TRACE("Added tracks block index entry at offset %" PRIu64, tracks_position);
        }
    }
}

static void write_mode2_subheaders_block(aaruformat_context *ctx)
{
    // Write MODE 2 subheader data block
    if(ctx->mode2_subheaders == NULL) return;
 
    fseek(ctx->imageStream, 0, SEEK_END);
    long     mode2_subheaders_position = ftell(ctx->imageStream);
    // Align index position to block boundary if needed
    uint64_t alignment_mask            = (1ULL << ctx->user_data_ddt_header.blockAlignmentShift) - 1;
    if(mode2_subheaders_position & alignment_mask)
    {
        uint64_t aligned_position = mode2_subheaders_position + alignment_mask & ~alignment_mask;
        fseek(ctx->imageStream, aligned_position, SEEK_SET);
        mode2_subheaders_position = aligned_position;
    }
 
    TRACE("Writing MODE 2 subheaders block at position %ld", mode2_subheaders_position);
    BlockHeader subheaders_block = {0};
    subheaders_block.identifier  = DataBlock;
    subheaders_block.type        = CompactDiscMode2Subheader;
    subheaders_block.compression = ctx->compression_enabled ? Lzma : None;
    subheaders_block.length =
        (uint32_t)(ctx->user_data_ddt_header.negative + ctx->image_info.Sectors + ctx->user_data_ddt_header.overflow) *
        8;
 
    // Calculate CRC64
    subheaders_block.crc64 = aaruf_crc64_data(ctx->mode2_subheaders, subheaders_block.length);
 
    uint8_t *buffer                                  = NULL;
    uint8_t  lzma_properties[LZMA_PROPERTIES_LENGTH] = {0};
 
    if(subheaders_block.compression == None)
    {
        buffer                    = ctx->mode2_subheaders;
        subheaders_block.cmpCrc64 = subheaders_block.crc64;
    }
    else
    {
        buffer = malloc((size_t)subheaders_block.length * 2);  // Allocate double size for compression
        if(buffer == NULL)
        {
            TRACE("Failed to allocate memory for MODE 2 subheaders compression");
            return;
        }
 
        size_t dst_size   = (size_t)subheaders_block.length * 2 * 2;
        size_t props_size = LZMA_PROPERTIES_LENGTH;
        aaruf_lzma_encode_buffer(buffer, &dst_size, ctx->mode2_subheaders, subheaders_block.length, lzma_properties,
                                 &props_size, 9, ctx->lzma_dict_size, 4, 0, 2, 273, 8);
 
        subheaders_block.cmpLength = (uint32_t)dst_size;
 
        if(subheaders_block.cmpLength >= subheaders_block.length)
        {
            subheaders_block.compression = None;
            free(buffer);
            buffer = ctx->mode2_subheaders;
        }
    }
 
    if(subheaders_block.compression == None)
    {
        subheaders_block.cmpLength = subheaders_block.length;
        subheaders_block.cmpCrc64  = subheaders_block.crc64;
    }
    else
        subheaders_block.cmpCrc64 = aaruf_crc64_data(buffer, subheaders_block.cmpLength);
 
    if(subheaders_block.compression == Lzma) subheaders_block.cmpLength += LZMA_PROPERTIES_LENGTH;
 
    // Write header
    if(fwrite(&subheaders_block, sizeof(BlockHeader), 1, ctx->imageStream) == 1)
    {
        if(subheaders_block.compression == Lzma) fwrite(lzma_properties, LZMA_PROPERTIES_LENGTH, 1, ctx->imageStream);
 
        // Write data
        size_t written_bytes = fwrite(buffer, subheaders_block.cmpLength, 1, ctx->imageStream);
        if(written_bytes == 1)
        {
            TRACE("Successfully wrote MODE 2 subheaders block (%" PRIu64 " bytes)", subheaders_block.cmpLength);
            // Add MODE 2 subheaders block to index
            TRACE("Adding MODE 2 subheaders block to index");
            IndexEntry mode2_subheaders_index_entry;
            mode2_subheaders_index_entry.blockType = DataBlock;
            mode2_subheaders_index_entry.dataType  = CompactDiscMode2Subheader;
            mode2_subheaders_index_entry.offset    = mode2_subheaders_position;
            utarray_push_back(ctx->index_entries, &mode2_subheaders_index_entry);
            TRACE("Added MODE 2 subheaders block index entry at offset %" PRIu64, mode2_subheaders_position);
        }
    }
 
    if(subheaders_block.compression == Lzma) free(buffer);
}

static void write_sector_prefix(aaruformat_context *ctx)
{
    if(ctx->sector_prefix == NULL) return;
 
    fseek(ctx->imageStream, 0, SEEK_END);
    long     prefix_position = ftell(ctx->imageStream);
    // Align index position to block boundary if needed
    uint64_t alignment_mask  = (1ULL << ctx->user_data_ddt_header.blockAlignmentShift) - 1;
    if(prefix_position & alignment_mask)
    {
        uint64_t aligned_position = prefix_position + alignment_mask & ~alignment_mask;
        fseek(ctx->imageStream, aligned_position, SEEK_SET);
        prefix_position = aligned_position;
    }
 
    TRACE("Writing sector prefix block at position %ld", prefix_position);
    BlockHeader prefix_block = {0};
    prefix_block.identifier  = DataBlock;
    prefix_block.type        = CdSectorPrefix;
    prefix_block.compression = ctx->compression_enabled ? Lzma : None;
    prefix_block.length      = (uint32_t)ctx->sector_prefix_offset;
 
    // Calculate CRC64
    prefix_block.crc64 = aaruf_crc64_data(ctx->sector_prefix, prefix_block.length);
 
    uint8_t *buffer                                  = NULL;
    uint8_t  lzma_properties[LZMA_PROPERTIES_LENGTH] = {0};
 
    if(prefix_block.compression == None)
    {
        buffer                = ctx->sector_prefix;
        prefix_block.cmpCrc64 = prefix_block.crc64;
    }
    else
    {
        buffer = malloc((size_t)prefix_block.length * 2);  // Allocate double size for compression
        if(buffer == NULL)
        {
            TRACE("Failed to allocate memory for CD sector prefix compression");
            return;
        }
 
        size_t dst_size   = (size_t)prefix_block.length * 2 * 2;
        size_t props_size = LZMA_PROPERTIES_LENGTH;
        aaruf_lzma_encode_buffer(buffer, &dst_size, ctx->sector_prefix, prefix_block.length, lzma_properties,
                                 &props_size, 9, ctx->lzma_dict_size, 4, 0, 2, 273, 8);
 
        prefix_block.cmpLength = (uint32_t)dst_size;
 
        if(prefix_block.cmpLength >= prefix_block.length)
        {
            prefix_block.compression = None;
            free(buffer);
            buffer = ctx->sector_prefix;
        }
    }
 
    if(prefix_block.compression == None)
    {
        prefix_block.cmpLength = prefix_block.length;
        prefix_block.cmpCrc64  = prefix_block.crc64;
    }
    else
        prefix_block.cmpCrc64 = aaruf_crc64_data(buffer, prefix_block.cmpLength);
 
    if(prefix_block.compression == Lzma) prefix_block.cmpLength += LZMA_PROPERTIES_LENGTH;
 
    // Write header
    if(fwrite(&prefix_block, sizeof(BlockHeader), 1, ctx->imageStream) == 1)
    {
        if(prefix_block.compression == Lzma) fwrite(lzma_properties, LZMA_PROPERTIES_LENGTH, 1, ctx->imageStream);
 
        // Write data
        const size_t written_bytes = fwrite(buffer, prefix_block.cmpLength, 1, ctx->imageStream);
        if(written_bytes == 1)
        {
            TRACE("Successfully wrote CD sector prefix block (%" PRIu64 " bytes)", prefix_block.cmpLength);
            // Add prefix block to index
            TRACE("Adding CD sector prefix block to index");
            IndexEntry prefix_index_entry;
            prefix_index_entry.blockType = DataBlock;
            prefix_index_entry.dataType  = CdSectorPrefix;
            prefix_index_entry.offset    = prefix_position;
            utarray_push_back(ctx->index_entries, &prefix_index_entry);
            TRACE("Added CD sector prefix block index entry at offset %" PRIu64, prefix_position);
        }
    }
 
    if(prefix_block.compression == Lzma) free(buffer);
}

static void write_sector_suffix(aaruformat_context *ctx)
{
    if(ctx->sector_suffix == NULL) return;
 
    fseek(ctx->imageStream, 0, SEEK_END);
    long           suffix_position = ftell(ctx->imageStream);
    // Align index position to block boundary if needed
    const uint64_t alignment_mask  = (1ULL << ctx->user_data_ddt_header.blockAlignmentShift) - 1;
    if(suffix_position & alignment_mask)
    {
        const uint64_t aligned_position = suffix_position + alignment_mask & ~alignment_mask;
        fseek(ctx->imageStream, aligned_position, SEEK_SET);
        suffix_position = aligned_position;
    }
 
    TRACE("Writing sector suffix block at position %ld", suffix_position);
    BlockHeader suffix_block = {0};
    suffix_block.identifier  = DataBlock;
    suffix_block.type        = CdSectorSuffix;
    suffix_block.compression = ctx->compression_enabled ? Lzma : None;
    suffix_block.length      = (uint32_t)ctx->sector_suffix_offset;
 
    // Calculate CRC64
    suffix_block.crc64 = aaruf_crc64_data(ctx->sector_suffix, suffix_block.length);
 
    uint8_t *buffer                                  = NULL;
    uint8_t  lzma_properties[LZMA_PROPERTIES_LENGTH] = {0};
 
    if(suffix_block.compression == None)
    {
        buffer                = ctx->sector_suffix;
        suffix_block.cmpCrc64 = suffix_block.crc64;
    }
    else
    {
        buffer = malloc((size_t)suffix_block.length * 2);  // Allocate double size for compression
        if(buffer == NULL)
        {
            TRACE("Failed to allocate memory for CD sector suffix compression");
            return;
        }
 
        size_t dst_size   = (size_t)suffix_block.length * 2 * 2;
        size_t props_size = LZMA_PROPERTIES_LENGTH;
        aaruf_lzma_encode_buffer(buffer, &dst_size, ctx->sector_suffix, suffix_block.length, lzma_properties,
                                 &props_size, 9, ctx->lzma_dict_size, 4, 0, 2, 273, 8);
 
        suffix_block.cmpLength = (uint32_t)dst_size;
 
        if(suffix_block.cmpLength >= suffix_block.length)
        {
            suffix_block.compression = None;
            free(buffer);
            buffer = ctx->sector_suffix;
        }
    }
 
    if(suffix_block.compression == None)
    {
        suffix_block.cmpLength = suffix_block.length;
        suffix_block.cmpCrc64  = suffix_block.crc64;
    }
    else
        suffix_block.cmpCrc64 = aaruf_crc64_data(buffer, suffix_block.cmpLength);
 
    if(suffix_block.compression == Lzma) suffix_block.cmpLength += LZMA_PROPERTIES_LENGTH;
 
    // Write header
    if(fwrite(&suffix_block, sizeof(BlockHeader), 1, ctx->imageStream) == 1)
    {
        if(suffix_block.compression == Lzma) fwrite(lzma_properties, LZMA_PROPERTIES_LENGTH, 1, ctx->imageStream);
 
        // Write data
        const size_t written_bytes = fwrite(buffer, suffix_block.cmpLength, 1, ctx->imageStream);
        if(written_bytes == 1)
        {
            TRACE("Successfully wrote CD sector suffix block (%" PRIu64 " bytes)", suffix_block.cmpLength);
            // Add suffix block to index
            TRACE("Adding CD sector suffix block to index");
            IndexEntry suffix_index_entry;
            suffix_index_entry.blockType = DataBlock;
            suffix_index_entry.dataType  = CdSectorSuffix;
            suffix_index_entry.offset    = suffix_position;
            utarray_push_back(ctx->index_entries, &suffix_index_entry);
            TRACE("Added CD sector suffix block index entry at offset %" PRIu64, suffix_position);
        }
    }
 
    if(suffix_block.compression == Lzma) free(buffer);
}

static void write_sector_prefix_ddt(aaruformat_context *ctx)
{
    if(ctx->sector_prefix_ddt2 == NULL) return;
 
    fseek(ctx->imageStream, 0, SEEK_END);
    long           prefix_ddt_position = ftell(ctx->imageStream);
    // Align index position to block boundary if needed
    const uint64_t alignment_mask      = (1ULL << ctx->user_data_ddt_header.blockAlignmentShift) - 1;
    if(prefix_ddt_position & alignment_mask)
    {
        const uint64_t aligned_position = prefix_ddt_position + alignment_mask & ~alignment_mask;
        fseek(ctx->imageStream, aligned_position, SEEK_SET);
        prefix_ddt_position = aligned_position;
    }
 
    TRACE("Writing sector prefix DDT v2 at position %ld", prefix_ddt_position);
    DdtHeader2 ddt_header2          = {0};
    ddt_header2.identifier          = DeDuplicationTable2;
    ddt_header2.type                = CdSectorPrefix;
    ddt_header2.compression         = ctx->compression_enabled ? Lzma : None;
    ddt_header2.levels              = 1;
    ddt_header2.tableLevel          = 0;
    ddt_header2.negative            = ctx->user_data_ddt_header.negative;
    ddt_header2.overflow            = ctx->user_data_ddt_header.overflow;
    ddt_header2.blockAlignmentShift = ctx->user_data_ddt_header.blockAlignmentShift;
    ddt_header2.dataShift           = ctx->user_data_ddt_header.dataShift;
    ddt_header2.tableShift          = 0;  // Single-level DDT
    ddt_header2.entries =
        ctx->image_info.Sectors + ctx->user_data_ddt_header.negative + ctx->user_data_ddt_header.overflow;
    ddt_header2.blocks = ctx->user_data_ddt_header.blocks;
    ddt_header2.start  = 0;
    ddt_header2.length = ddt_header2.entries * sizeof(uint64_t);
    // Calculate CRC64
    ddt_header2.crc64  = aaruf_crc64_data((uint8_t *)ctx->sector_prefix_ddt2, (uint32_t)ddt_header2.length);
 
    uint8_t *buffer                                  = NULL;
    uint8_t  lzma_properties[LZMA_PROPERTIES_LENGTH] = {0};
 
    if(ddt_header2.compression == None)
    {
        buffer               = (uint8_t *)ctx->sector_prefix_ddt2;
        ddt_header2.cmpCrc64 = ddt_header2.crc64;
    }
    else
    {
        buffer = malloc((size_t)ddt_header2.length * 2);  // Allocate double size for compression
        if(buffer == NULL)
        {
            TRACE("Failed to allocate memory for sector prefix DDT v2 compression");
            return;
        }
 
        size_t dst_size   = (size_t)ddt_header2.length * 2 * 2;
        size_t props_size = LZMA_PROPERTIES_LENGTH;
        aaruf_lzma_encode_buffer(buffer, &dst_size, (uint8_t *)ctx->sector_prefix_ddt2, ddt_header2.length,
                                 lzma_properties, &props_size, 9, ctx->lzma_dict_size, 4, 0, 2, 273, 8);
 
        ddt_header2.cmpLength = (uint32_t)dst_size;
 
        if(ddt_header2.cmpLength >= ddt_header2.length)
        {
            ddt_header2.compression = None;
            free(buffer);
            buffer = (uint8_t *)ctx->sector_prefix_ddt2;
        }
    }
 
    if(ddt_header2.compression == None)
    {
        ddt_header2.cmpLength = ddt_header2.length;
        ddt_header2.cmpCrc64  = ddt_header2.crc64;
    }
    else
        ddt_header2.cmpCrc64 = aaruf_crc64_data(buffer, (uint32_t)ddt_header2.cmpLength);
 
    if(ddt_header2.compression == Lzma) ddt_header2.cmpLength += LZMA_PROPERTIES_LENGTH;
 
    // Write header
    if(fwrite(&ddt_header2, sizeof(DdtHeader2), 1, ctx->imageStream) == 1)
    {
        if(ddt_header2.compression == Lzma) fwrite(lzma_properties, LZMA_PROPERTIES_LENGTH, 1, ctx->imageStream);
 
        // Write data
        const size_t written_bytes = fwrite(buffer, ddt_header2.cmpLength, 1, ctx->imageStream);
        if(written_bytes == 1)
        {
            TRACE("Successfully wrote sector prefix DDT v2 (%" PRIu64 " bytes)", ddt_header2.cmpLength);
            // Add prefix block to index
            TRACE("Adding sector prefix DDT v2 to index");
            IndexEntry prefix_ddt_index_entry;
            prefix_ddt_index_entry.blockType = DeDuplicationTable2;
            prefix_ddt_index_entry.dataType  = CdSectorPrefix;
            prefix_ddt_index_entry.offset    = prefix_ddt_position;
            utarray_push_back(ctx->index_entries, &prefix_ddt_index_entry);
            TRACE("Added sector prefix DDT v2 index entry at offset %" PRIu64, prefix_ddt_position);
        }
    }
 
    if(ddt_header2.compression == Lzma) free(buffer);
}

static void write_sector_suffix_ddt(aaruformat_context *ctx)
{
    if(ctx->sector_suffix_ddt2 == NULL) return;
 
    fseek(ctx->imageStream, 0, SEEK_END);
    long           suffix_ddt_position = ftell(ctx->imageStream);
    // Align index position to block boundary if needed
    const uint64_t alignment_mask      = (1ULL << ctx->user_data_ddt_header.blockAlignmentShift) - 1;
    if(suffix_ddt_position & alignment_mask)
    {
        const uint64_t aligned_position = suffix_ddt_position + alignment_mask & ~alignment_mask;
        fseek(ctx->imageStream, aligned_position, SEEK_SET);
        suffix_ddt_position = aligned_position;
    }
 
    TRACE("Writing sector suffix DDT v2 at position %ld", suffix_ddt_position);
    DdtHeader2 ddt_header2          = {0};
    ddt_header2.identifier          = DeDuplicationTable2;
    ddt_header2.type                = CdSectorSuffix;
    ddt_header2.compression         = ctx->compression_enabled ? Lzma : None;
    ddt_header2.levels              = 1;
    ddt_header2.tableLevel          = 0;
    ddt_header2.negative            = ctx->user_data_ddt_header.negative;
    ddt_header2.overflow            = ctx->user_data_ddt_header.overflow;
    ddt_header2.blockAlignmentShift = ctx->user_data_ddt_header.blockAlignmentShift;
    ddt_header2.dataShift           = ctx->user_data_ddt_header.dataShift;
    ddt_header2.tableShift          = 0;  // Single-level DDT
    ddt_header2.entries =
        ctx->image_info.Sectors + ctx->user_data_ddt_header.negative + ctx->user_data_ddt_header.overflow;
    ddt_header2.blocks = ctx->user_data_ddt_header.blocks;
    ddt_header2.start  = 0;
    ddt_header2.length = ddt_header2.entries * sizeof(uint64_t);
    // Calculate CRC64
    ddt_header2.crc64  = aaruf_crc64_data((uint8_t *)ctx->sector_suffix_ddt2, (uint32_t)ddt_header2.length);
 
    uint8_t *buffer                                  = NULL;
    uint8_t  lzma_properties[LZMA_PROPERTIES_LENGTH] = {0};
 
    if(ddt_header2.compression == None)
    {
        buffer               = (uint8_t *)ctx->sector_suffix_ddt2;
        ddt_header2.cmpCrc64 = ddt_header2.crc64;
    }
    else
    {
        buffer = malloc((size_t)ddt_header2.length * 2);  // Allocate double size for compression
        if(buffer == NULL)
        {
            TRACE("Failed to allocate memory for sector suffix DDT v2 compression");
            return;
        }
 
        size_t dst_size   = (size_t)ddt_header2.length * 2 * 2;
        size_t props_size = LZMA_PROPERTIES_LENGTH;
        aaruf_lzma_encode_buffer(buffer, &dst_size, (uint8_t *)ctx->sector_suffix_ddt2, ddt_header2.length,
                                 lzma_properties, &props_size, 9, ctx->lzma_dict_size, 4, 0, 2, 273, 8);
 
        ddt_header2.cmpLength = (uint32_t)dst_size;
 
        if(ddt_header2.cmpLength >= ddt_header2.length)
        {
            ddt_header2.compression = None;
            free(buffer);
            buffer = (uint8_t *)ctx->sector_suffix_ddt2;
        }
    }
 
    if(ddt_header2.compression == None)
    {
        ddt_header2.cmpLength = ddt_header2.length;
        ddt_header2.cmpCrc64  = ddt_header2.crc64;
    }
    else
        ddt_header2.cmpCrc64 = aaruf_crc64_data(buffer, (uint32_t)ddt_header2.cmpLength);
 
    if(ddt_header2.compression == Lzma) ddt_header2.cmpLength += LZMA_PROPERTIES_LENGTH;
 
    // Write header
    if(fwrite(&ddt_header2, sizeof(DdtHeader2), 1, ctx->imageStream) == 1)
    {
        if(ddt_header2.compression == Lzma) fwrite(lzma_properties, LZMA_PROPERTIES_LENGTH, 1, ctx->imageStream);
 
        // Write data
        const size_t written_bytes = fwrite(buffer, ddt_header2.cmpLength, 1, ctx->imageStream);
        if(written_bytes == 1)
        {
            TRACE("Successfully wrote sector suffix DDT v2 (%" PRIu64 " bytes)", ddt_header2.cmpLength);
            // Add suffix block to index
            TRACE("Adding sector suffix DDT v2 to index");
            IndexEntry suffix_ddt_index_entry;
            suffix_ddt_index_entry.blockType = DeDuplicationTable2;
            suffix_ddt_index_entry.dataType  = CdSectorSuffix;
            suffix_ddt_index_entry.offset    = suffix_ddt_position;
            utarray_push_back(ctx->index_entries, &suffix_ddt_index_entry);
            TRACE("Added sector suffix DDT v2 index entry at offset %" PRIu64, suffix_ddt_position);
        }
    }
 
    if(ddt_header2.compression == Lzma) free(buffer);
}

static void write_sector_subchannel(const aaruformat_context *ctx)
{
    if(ctx->sector_subchannel == NULL) return;
 
    fseek(ctx->imageStream, 0, SEEK_END);
    long           block_position = ftell(ctx->imageStream);
    // Align index position to block boundary if needed
    const uint64_t alignment_mask = (1ULL << ctx->user_data_ddt_header.blockAlignmentShift) - 1;
    if(block_position & alignment_mask)
    {
        const uint64_t aligned_position = block_position + alignment_mask & ~alignment_mask;
        fseek(ctx->imageStream, aligned_position, SEEK_SET);
        block_position = aligned_position;
    }
 
    TRACE("Writing sector subchannel block at position %ld", block_position);
    BlockHeader subchannel_block = {0};
    subchannel_block.identifier  = DataBlock;
    subchannel_block.compression = None;
 
    uint8_t *buffer                                  = ctx->sector_subchannel;
    bool     owns_buffer                             = false;
    uint8_t  lzma_properties[LZMA_PROPERTIES_LENGTH] = {0};
 
    subchannel_block.cmpLength = subchannel_block.length;
 
    if(ctx->image_info.MetadataMediaType == OpticalDisc)
    {
        subchannel_block.type   = CdSectorSubchannel;
        subchannel_block.length = (uint32_t)(ctx->user_data_ddt_header.negative + ctx->image_info.Sectors +
                                             ctx->user_data_ddt_header.overflow) *
                                  96;
 
        if(ctx->compression_enabled)
        {
            uint8_t *cst_buffer = malloc(subchannel_block.length);
 
            if(cst_buffer == NULL)
            {
                TRACE("Failed to allocate memory for Claunia Subchannel Transform output");
                return;
            }
 
            uint8_t *dst_buffer = malloc(subchannel_block.length);
 
            if(dst_buffer == NULL)
            {
                TRACE("Failed to allocate memory for LZMA output");
                free(cst_buffer);
                return;
            }
 
            aaruf_cst_transform(ctx->sector_subchannel, cst_buffer, subchannel_block.length);
            size_t dst_size   = subchannel_block.length;
            size_t props_size = LZMA_PROPERTIES_LENGTH;
 
            aaruf_lzma_encode_buffer(dst_buffer, &dst_size, cst_buffer, subchannel_block.length, lzma_properties,
                                     &props_size, 9, ctx->lzma_dict_size, 4, 0, 2, 273, 8);
 
            free(cst_buffer);
 
            if(dst_size < subchannel_block.length)
            {
                subchannel_block.compression = LzmaClauniaSubchannelTransform;
                subchannel_block.cmpLength   = (uint32_t)dst_size;
                buffer                       = dst_buffer;
                owns_buffer                  = true;
            }
            else
            {
                subchannel_block.compression = None;
                free(dst_buffer);
                subchannel_block.cmpLength = subchannel_block.length;
            }
        }
    }
    else if(ctx->image_info.MetadataMediaType == BlockMedia)
    {
        switch(ctx->image_info.MediaType)
        {
            case AppleProfile:
            case AppleFileWare:
                subchannel_block.type   = AppleProfileTag;
                subchannel_block.length = (uint32_t)(ctx->image_info.Sectors + ctx->user_data_ddt_header.overflow) * 20;
                break;
            case AppleSonyDS:
            case AppleSonySS:
                subchannel_block.type   = AppleSonyTag;
                subchannel_block.length = (uint32_t)(ctx->image_info.Sectors + ctx->user_data_ddt_header.overflow) * 12;
                break;
            case PriamDataTower:
                subchannel_block.type   = PriamDataTowerTag;
                subchannel_block.length = (uint32_t)(ctx->image_info.Sectors + ctx->user_data_ddt_header.overflow) * 24;
                break;
            default:
                TRACE("Incorrect media type, not writing sector subchannel block");
                return;  // Incorrect media type
        }
        subchannel_block.compression = Lzma;
 
        uint8_t *dst_buffer = malloc(subchannel_block.length);
 
        if(dst_buffer == NULL)
        {
            TRACE("Failed to allocate memory for LZMA output");
            return;
        }
 
        size_t dst_size   = subchannel_block.length;
        size_t props_size = LZMA_PROPERTIES_LENGTH;
 
        aaruf_lzma_encode_buffer(dst_buffer, &dst_size, ctx->sector_subchannel, subchannel_block.length,
                                 lzma_properties, &props_size, 9, ctx->lzma_dict_size, 4, 0, 2, 273, 8);
 
        if(dst_size < subchannel_block.length)
        {
            subchannel_block.cmpLength = (uint32_t)dst_size;
            buffer                     = dst_buffer;
            owns_buffer                = true;
        }
        else
        {
            subchannel_block.compression = None;
            free(dst_buffer);
            subchannel_block.cmpLength = subchannel_block.length;
        }
    }
    else
    {
        TRACE("Incorrect media type, not writing sector subchannel block");
        return;  // Incorrect media type
    }
 
    // Calculate CRC64 for raw subchannel data and compressed payload when present
    subchannel_block.crc64 = aaruf_crc64_data(ctx->sector_subchannel, subchannel_block.length);
    if(subchannel_block.compression == None)
        subchannel_block.cmpCrc64 = subchannel_block.crc64;
    else
        subchannel_block.cmpCrc64 = aaruf_crc64_data(buffer, subchannel_block.cmpLength);
 
    // Write header
    if(fwrite(&subchannel_block, sizeof(BlockHeader), 1, ctx->imageStream) == 1)
    {
        if(subchannel_block.compression != None) fwrite(lzma_properties, LZMA_PROPERTIES_LENGTH, 1, ctx->imageStream);
 
        // Write data
        const size_t written_bytes = fwrite(buffer, subchannel_block.cmpLength, 1, ctx->imageStream);
        if(written_bytes == 1)
        {
            TRACE("Successfully wrote sector subchannel block (%" PRIu64 " bytes)", subchannel_block.cmpLength);
            // Add subchannel block to index
            TRACE("Adding sector subchannel block to index");
            IndexEntry subchannel_index_entry;
            subchannel_index_entry.blockType = DataBlock;
            subchannel_index_entry.dataType  = subchannel_block.type;
            subchannel_index_entry.offset    = block_position;
            utarray_push_back(ctx->index_entries, &subchannel_index_entry);
            TRACE("Added sector subchannel block index entry at offset %" PRIu64, block_position);
        }
    }
 
    if(owns_buffer) free(buffer);
}

void write_dvd_long_sector_blocks(aaruformat_context *ctx)
{
    if(ctx->sector_id == NULL || ctx->sector_ied == NULL || ctx->sector_cpr_mai == NULL || ctx->sector_edc == NULL)
        return;
 
    uint64_t total_sectors =
        ctx->user_data_ddt_header.negative + ctx->image_info.Sectors + ctx->user_data_ddt_header.overflow;
 
    // Write DVD sector ID block
    fseek(ctx->imageStream, 0, SEEK_END);
    long           id_position    = ftell(ctx->imageStream);
    const uint64_t alignment_mask = (1ULL << ctx->user_data_ddt_header.blockAlignmentShift) - 1;
    if(id_position & alignment_mask)
    {
        const uint64_t aligned_position = id_position + alignment_mask & ~alignment_mask;
        fseek(ctx->imageStream, aligned_position, SEEK_SET);
        id_position = aligned_position;
    }
    TRACE("Writing DVD sector ID block at position %ld", id_position);
    BlockHeader id_block = {0};
    id_block.identifier  = DataBlock;
    id_block.type        = DvdSectorId;
    id_block.compression = ctx->compression_enabled ? Lzma : None;
    id_block.length      = (uint32_t)total_sectors * 4;
 
    // Calculate CRC64
    id_block.crc64 = aaruf_crc64_data(ctx->sector_id, id_block.length);
 
    uint8_t *buffer                                  = NULL;
    uint8_t  lzma_properties[LZMA_PROPERTIES_LENGTH] = {0};
 
    if(id_block.compression == None)
    {
        buffer            = ctx->sector_id;
        id_block.cmpCrc64 = id_block.crc64;
    }
    else
    {
        buffer = malloc((size_t)id_block.length * 2);  // Allocate double size for compression
        if(buffer == NULL)
        {
            TRACE("Failed to allocate memory for DVD sector ID compression");
            return;
        }
 
        size_t dst_size   = (size_t)id_block.length * 2 * 2;
        size_t props_size = LZMA_PROPERTIES_LENGTH;
        aaruf_lzma_encode_buffer(buffer, &dst_size, ctx->sector_id, id_block.length, lzma_properties, &props_size, 9,
                                 ctx->lzma_dict_size, 4, 0, 2, 273, 8);
 
        id_block.cmpLength = (uint32_t)dst_size;
 
        if(id_block.cmpLength >= id_block.length)
        {
            id_block.compression = None;
            free(buffer);
            buffer = ctx->sector_id;
        }
    }
 
    if(id_block.compression == None)
    {
        id_block.cmpLength = id_block.length;
        id_block.cmpCrc64  = id_block.crc64;
    }
    else
        id_block.cmpCrc64 = aaruf_crc64_data(buffer, id_block.cmpLength);
 
    if(id_block.compression == Lzma) id_block.cmpLength += LZMA_PROPERTIES_LENGTH;
 
    // Write header
    if(fwrite(&id_block, sizeof(BlockHeader), 1, ctx->imageStream) == 1)
    {
        if(id_block.compression == Lzma) fwrite(lzma_properties, LZMA_PROPERTIES_LENGTH, 1, ctx->imageStream);
 
        // Write data
        const size_t written_bytes = fwrite(buffer, id_block.cmpLength, 1, ctx->imageStream);
        if(written_bytes == 1)
        {
            TRACE("Successfully wrote DVD sector ID block (%" PRIu64 " bytes)", id_block.cmpLength);
            // Add ID block to index
            TRACE("Adding DVD sector ID block to index");
            IndexEntry id_index_entry;
            id_index_entry.blockType = DataBlock;
            id_index_entry.dataType  = DvdSectorId;
            id_index_entry.offset    = id_position;
            utarray_push_back(ctx->index_entries, &id_index_entry);
            TRACE("Added DVD sector ID block index entry at offset %" PRIu64, id_position);
        }
    }
 
    if(id_block.compression == Lzma) free(buffer);
 
    // Write DVD sector IED block
    fseek(ctx->imageStream, 0, SEEK_END);
    long ied_position = ftell(ctx->imageStream);
    if(ied_position & alignment_mask)
    {
        const uint64_t aligned_position = ied_position + alignment_mask & ~alignment_mask;
        fseek(ctx->imageStream, aligned_position, SEEK_SET);
        ied_position = aligned_position;
    }
    TRACE("Writing DVD sector IED block at position %ld", ied_position);
    BlockHeader ied_block = {0};
    ied_block.identifier  = DataBlock;
    ied_block.type        = DvdSectorIed;
    ied_block.compression = ctx->compression_enabled ? Lzma : None;
    ied_block.length      = (uint32_t)total_sectors * 2;
    // Calculate CRC64
    ied_block.crc64       = aaruf_crc64_data(ctx->sector_ied, ied_block.length);
 
    buffer = NULL;
 
    if(ied_block.compression == None)
    {
        buffer             = ctx->sector_ied;
        ied_block.cmpCrc64 = ied_block.crc64;
    }
    else
    {
        buffer = malloc((size_t)ied_block.length * 2);  // Allocate double size for compression
        if(buffer == NULL)
        {
            TRACE("Failed to allocate memory for DVD sector IED compression");
            return;
        }
 
        size_t dst_size   = (size_t)ied_block.length * 2 * 2;
        size_t props_size = LZMA_PROPERTIES_LENGTH;
        aaruf_lzma_encode_buffer(buffer, &dst_size, ctx->sector_ied, ied_block.length, lzma_properties, &props_size, 9,
                                 ctx->lzma_dict_size, 4, 0, 2, 273, 8);
 
        ied_block.cmpLength = (uint32_t)dst_size;
 
        if(ied_block.cmpLength >= ied_block.length)
        {
            ied_block.compression = None;
            free(buffer);
            buffer = ctx->sector_ied;
        }
    }
 
    if(ied_block.compression == None)
    {
        ied_block.cmpLength = ied_block.length;
        ied_block.cmpCrc64  = ied_block.crc64;
    }
    else
        ied_block.cmpCrc64 = aaruf_crc64_data(buffer, ied_block.cmpLength);
 
    if(ied_block.compression == Lzma) ied_block.cmpLength += LZMA_PROPERTIES_LENGTH;
 
    // Write header
    if(fwrite(&ied_block, sizeof(BlockHeader), 1, ctx->imageStream) == 1)
    {
        if(ied_block.compression == Lzma) fwrite(lzma_properties, LZMA_PROPERTIES_LENGTH, 1, ctx->imageStream);
 
        // Write data
        const size_t written_bytes = fwrite(buffer, ied_block.cmpLength, 1, ctx->imageStream);
        if(written_bytes == 1)
        {
            TRACE("Successfully wrote DVD sector IED block (%" PRIu64 " bytes)", ied_block.cmpLength);
            // Add IED block to index
            TRACE("Adding DVD sector IED block to index");
            IndexEntry ied_index_entry;
            ied_index_entry.blockType = DataBlock;
            ied_index_entry.dataType  = DvdSectorIed;
            ied_index_entry.offset    = ied_position;
            utarray_push_back(ctx->index_entries, &ied_index_entry);
            TRACE("Added DVD sector IED block index entry at offset %" PRIu64, ied_position);
        }
    }
 
    if(ied_block.compression == Lzma) free(buffer);
 
    // Write DVD sector CPR/MAI block
    fseek(ctx->imageStream, 0, SEEK_END);
    long cpr_mai_position = ftell(ctx->imageStream);
    if(cpr_mai_position & alignment_mask)
    {
        const uint64_t aligned_position = cpr_mai_position + alignment_mask & ~alignment_mask;
        fseek(ctx->imageStream, aligned_position, SEEK_SET);
        cpr_mai_position = aligned_position;
    }
    TRACE("Writing DVD sector CPR/MAI block at position %ld", cpr_mai_position);
    BlockHeader cpr_mai_block = {0};
    cpr_mai_block.identifier  = DataBlock;
    cpr_mai_block.type        = DvdSectorCprMai;
    cpr_mai_block.compression = ctx->compression_enabled ? Lzma : None;
    cpr_mai_block.length      = (uint32_t)total_sectors * 6;
    // Calculate CRC64
    cpr_mai_block.crc64       = aaruf_crc64_data(ctx->sector_cpr_mai, cpr_mai_block.length);
 
    buffer = NULL;
 
    if(cpr_mai_block.compression == None)
    {
        buffer                 = ctx->sector_cpr_mai;
        cpr_mai_block.cmpCrc64 = cpr_mai_block.crc64;
    }
    else
    {
        buffer = malloc((size_t)cpr_mai_block.length * 2);  // Allocate double size for compression
        if(buffer == NULL)
        {
            TRACE("Failed to allocate memory for DVD sector CPR/MAI compression");
            return;
        }
 
        size_t dst_size   = (size_t)cpr_mai_block.length * 2 * 2;
        size_t props_size = LZMA_PROPERTIES_LENGTH;
        aaruf_lzma_encode_buffer(buffer, &dst_size, ctx->sector_cpr_mai, cpr_mai_block.length, lzma_properties,
                                 &props_size, 9, ctx->lzma_dict_size, 4, 0, 2, 273, 8);
 
        cpr_mai_block.cmpLength = (uint32_t)dst_size;
 
        if(cpr_mai_block.cmpLength >= cpr_mai_block.length)
        {
            cpr_mai_block.compression = None;
            free(buffer);
            buffer = ctx->sector_cpr_mai;
        }
    }
 
    if(cpr_mai_block.compression == None)
    {
        cpr_mai_block.cmpLength = cpr_mai_block.length;
        cpr_mai_block.cmpCrc64  = cpr_mai_block.crc64;
    }
    else
        cpr_mai_block.cmpCrc64 = aaruf_crc64_data(buffer, cpr_mai_block.cmpLength);
 
    if(cpr_mai_block.compression == Lzma) cpr_mai_block.cmpLength += LZMA_PROPERTIES_LENGTH;
 
    // Write header
    if(fwrite(&cpr_mai_block, sizeof(BlockHeader), 1, ctx->imageStream) == 1)
    {
        if(cpr_mai_block.compression == Lzma) fwrite(lzma_properties, LZMA_PROPERTIES_LENGTH, 1, ctx->imageStream);
 
        // Write data
        const size_t written_bytes = fwrite(buffer, cpr_mai_block.cmpLength, 1, ctx->imageStream);
        if(written_bytes == 1)
        {
            TRACE("Successfully wrote DVD sector CPR/MAI block (%" PRIu64 " bytes)", cpr_mai_block.cmpLength);
            // Add CPR/MAI block to index
            TRACE("Adding DVD sector CPR/MAI block to index");
            IndexEntry cpr_mai_index_entry;
            cpr_mai_index_entry.blockType = DataBlock;
            cpr_mai_index_entry.dataType  = DvdSectorCprMai;
            cpr_mai_index_entry.offset    = cpr_mai_position;
            utarray_push_back(ctx->index_entries, &cpr_mai_index_entry);
            TRACE("Added DVD sector CPR/MAI block index entry at offset %" PRIu64, cpr_mai_position);
        }
    }
 
    if(cpr_mai_block.compression == Lzma) free(buffer);
 
    // Write DVD sector EDC block
    fseek(ctx->imageStream, 0, SEEK_END);
    long edc_position = ftell(ctx->imageStream);
    if(edc_position & alignment_mask)
    {
        const uint64_t aligned_position = edc_position + alignment_mask & ~alignment_mask;
        fseek(ctx->imageStream, aligned_position, SEEK_SET);
        edc_position = aligned_position;
    }
    TRACE("Writing DVD sector EDC block at position %ld", edc_position);
    BlockHeader edc_block = {0};
    edc_block.identifier  = DataBlock;
    edc_block.type        = DvdSectorEdc;
    edc_block.compression = ctx->compression_enabled ? Lzma : None;
    edc_block.length      = (uint32_t)total_sectors * 4;
    // Calculate CRC64
    edc_block.crc64       = aaruf_crc64_data(ctx->sector_edc, edc_block.length);
 
    buffer = NULL;
 
    if(edc_block.compression == None)
    {
        buffer             = ctx->sector_edc;
        edc_block.cmpCrc64 = edc_block.crc64;
    }
    else
    {
        buffer = malloc((size_t)edc_block.length * 2);  // Allocate double size for compression
        if(buffer == NULL)
        {
            TRACE("Failed to allocate memory for DVD sector EDC compression");
            return;
        }
 
        size_t dst_size   = (size_t)edc_block.length * 2 * 2;
        size_t props_size = LZMA_PROPERTIES_LENGTH;
        aaruf_lzma_encode_buffer(buffer, &dst_size, ctx->sector_edc, edc_block.length, lzma_properties, &props_size, 9,
                                 ctx->lzma_dict_size, 4, 0, 2, 273, 8);
 
        edc_block.cmpLength = (uint32_t)dst_size;
 
        if(edc_block.cmpLength >= edc_block.length)
        {
            edc_block.compression = None;
            free(buffer);
            buffer = ctx->sector_edc;
        }
    }
 
    if(edc_block.compression == None)
    {
        edc_block.cmpLength = edc_block.length;
        edc_block.cmpCrc64  = edc_block.crc64;
    }
    else
        edc_block.cmpCrc64 = aaruf_crc64_data(buffer, edc_block.cmpLength);
 
    if(edc_block.compression == Lzma) edc_block.cmpLength += LZMA_PROPERTIES_LENGTH;
 
    // Write header
    if(fwrite(&edc_block, sizeof(BlockHeader), 1, ctx->imageStream) == 1)
    {
        if(edc_block.compression == Lzma) fwrite(lzma_properties, LZMA_PROPERTIES_LENGTH, 1, ctx->imageStream);
 
        // Write data
        const size_t written_bytes = fwrite(buffer, edc_block.cmpLength, 1, ctx->imageStream);
        if(written_bytes == 1)
        {
            TRACE("Successfully wrote DVD sector EDC block (%" PRIu64 " bytes)", edc_block.cmpLength);
            // Add EDC block to index
            TRACE("Adding DVD sector EDC block to index");
            IndexEntry edc_index_entry;
            edc_index_entry.blockType = DataBlock;
            edc_index_entry.dataType  = DvdSectorEdc;
            edc_index_entry.offset    = edc_position;
            utarray_push_back(ctx->index_entries, &edc_index_entry);
            TRACE("Added DVD sector EDC block index entry at offset %" PRIu64, edc_position);
        }
    }
 
    if(edc_block.compression == Lzma) free(buffer);
}

static void write_dvd_title_key_decrypted_block(const aaruformat_context *ctx)
{
    if(ctx->sector_decrypted_title_key == NULL) return;
 
    fseek(ctx->imageStream, 0, SEEK_END);
    long           block_position = ftell(ctx->imageStream);
    const uint64_t alignment_mask = (1ULL << ctx->user_data_ddt_header.blockAlignmentShift) - 1;
    if(block_position & alignment_mask)
    {
        const uint64_t aligned_position = block_position + alignment_mask & ~alignment_mask;
        fseek(ctx->imageStream, aligned_position, SEEK_SET);
        block_position = aligned_position;
    }
    TRACE("Writing DVD decrypted title key block at position %ld", block_position);
    BlockHeader decrypted_title_key_block = {0};
    decrypted_title_key_block.identifier  = DataBlock;
    decrypted_title_key_block.type        = DvdSectorTitleKeyDecrypted;
    decrypted_title_key_block.compression = ctx->compression_enabled ? Lzma : None;
    decrypted_title_key_block.length =
        (uint32_t)(ctx->user_data_ddt_header.negative + ctx->image_info.Sectors + ctx->user_data_ddt_header.overflow) *
        5;
    // Calculate CRC64
    decrypted_title_key_block.crc64 =
        aaruf_crc64_data(ctx->sector_decrypted_title_key, decrypted_title_key_block.length);
 
    uint8_t *buffer                                  = NULL;
    uint8_t  lzma_properties[LZMA_PROPERTIES_LENGTH] = {0};
 
    if(decrypted_title_key_block.compression == None)
    {
        buffer                             = ctx->sector_decrypted_title_key;
        decrypted_title_key_block.cmpCrc64 = decrypted_title_key_block.crc64;
    }
    else
    {
        buffer = malloc((size_t)decrypted_title_key_block.length * 2);  // Allocate double size for compression
        if(buffer == NULL)
        {
            TRACE("Failed to allocate memory for DVD decrypted title key compression");
            return;
        }
 
        size_t dst_size   = (size_t)decrypted_title_key_block.length * 2 * 2;
        size_t props_size = LZMA_PROPERTIES_LENGTH;
        aaruf_lzma_encode_buffer(buffer, &dst_size, ctx->sector_decrypted_title_key, decrypted_title_key_block.length,
                                 lzma_properties, &props_size, 9, ctx->lzma_dict_size, 4, 0, 2, 273, 8);
 
        decrypted_title_key_block.cmpLength = (uint32_t)dst_size;
 
        if(decrypted_title_key_block.cmpLength >= decrypted_title_key_block.length)
        {
            decrypted_title_key_block.compression = None;
            free(buffer);
            buffer = ctx->sector_decrypted_title_key;
        }
    }
 
    if(decrypted_title_key_block.compression == None)
    {
        decrypted_title_key_block.cmpLength = decrypted_title_key_block.length;
        decrypted_title_key_block.cmpCrc64  = decrypted_title_key_block.crc64;
    }
    else
        decrypted_title_key_block.cmpCrc64 = aaruf_crc64_data(buffer, decrypted_title_key_block.cmpLength);
 
    if(decrypted_title_key_block.compression == Lzma) decrypted_title_key_block.cmpLength += LZMA_PROPERTIES_LENGTH;
 
    // Write header
    if(fwrite(&decrypted_title_key_block, sizeof(BlockHeader), 1, ctx->imageStream) == 1)
    {
        if(decrypted_title_key_block.compression == Lzma)
            fwrite(lzma_properties, LZMA_PROPERTIES_LENGTH, 1, ctx->imageStream);
 
        // Write data
        const size_t written_bytes = fwrite(buffer, decrypted_title_key_block.cmpLength, 1, ctx->imageStream);
        if(written_bytes == 1)
        {
            TRACE("Successfully wrote DVD decrypted title key block (%" PRIu64 " bytes)",
                  decrypted_title_key_block.cmpLength);
            // Add decrypted title key block to index
            TRACE("Adding DVD decrypted title key block to index");
            IndexEntry decrypted_title_key_index_entry;
            decrypted_title_key_index_entry.blockType = DataBlock;
            decrypted_title_key_index_entry.dataType  = DvdSectorTitleKeyDecrypted;
            decrypted_title_key_index_entry.offset    = block_position;
            utarray_push_back(ctx->index_entries, &decrypted_title_key_index_entry);
            TRACE("Added DVD decrypted title key block index entry at offset %" PRIu64, block_position);
        }
    }
 
    if(decrypted_title_key_block.compression == Lzma) free(buffer);
}

static void write_media_tags(const aaruformat_context *ctx)
{
    if(ctx->mediaTags == NULL) return;
 
    mediaTagEntry *media_tag     = NULL;
    mediaTagEntry *tmp_media_tag = NULL;
 
    HASH_ITER(hh, ctx->mediaTags, media_tag, tmp_media_tag)
    {
        fseek(ctx->imageStream, 0, SEEK_END);
        long           tag_position   = ftell(ctx->imageStream);
        const uint64_t alignment_mask = (1ULL << ctx->user_data_ddt_header.blockAlignmentShift) - 1;
        if(tag_position & alignment_mask)
        {
            const uint64_t aligned_position = tag_position + alignment_mask & ~alignment_mask;
            fseek(ctx->imageStream, aligned_position, SEEK_SET);
            tag_position = aligned_position;
        }
 
        TRACE("Writing media tag block type %d at position %ld", aaruf_get_datatype_for_media_tag_type(media_tag->type),
              tag_position);
        BlockHeader tag_block = {0};
        tag_block.identifier  = DataBlock;
        tag_block.type        = (uint16_t)aaruf_get_datatype_for_media_tag_type(media_tag->type);
        tag_block.compression = ctx->compression_enabled ? Lzma : None;
        tag_block.length      = media_tag->length;
 
        // Calculate CRC64
        tag_block.crc64 = aaruf_crc64_data(media_tag->data, tag_block.length);
 
        uint8_t *buffer                                  = NULL;
        uint8_t  lzma_properties[LZMA_PROPERTIES_LENGTH] = {0};
 
        if(tag_block.compression == None)
        {
            buffer             = media_tag->data;
            tag_block.cmpCrc64 = tag_block.crc64;
        }
        else
        {
            buffer = malloc((size_t)tag_block.length * 2);  // Allocate double size for compression
            if(buffer == NULL)
            {
                TRACE("Failed to allocate memory for media tag compression");
                return;
            }
 
            size_t dst_size   = (size_t)tag_block.length * 2 * 2;
            size_t props_size = LZMA_PROPERTIES_LENGTH;
            aaruf_lzma_encode_buffer(buffer, &dst_size, media_tag->data, tag_block.length, lzma_properties, &props_size,
                                     9, ctx->lzma_dict_size, 4, 0, 2, 273, 8);
 
            tag_block.cmpLength = (uint32_t)dst_size;
 
            if(tag_block.cmpLength >= tag_block.length)
            {
                tag_block.compression = None;
                free(buffer);
                buffer = media_tag->data;
            }
        }
 
        if(tag_block.compression == None)
        {
            tag_block.cmpLength = tag_block.length;
            tag_block.cmpCrc64  = tag_block.crc64;
        }
        else
            tag_block.cmpCrc64 = aaruf_crc64_data(buffer, tag_block.cmpLength);
 
        if(tag_block.compression == Lzma) tag_block.cmpLength += LZMA_PROPERTIES_LENGTH;
 
        // Write header
        if(fwrite(&tag_block, sizeof(BlockHeader), 1, ctx->imageStream) == 1)
        {
            if(tag_block.compression == Lzma)
                fwrite(lzma_properties, LZMA_PROPERTIES_LENGTH, 1, ctx->imageStream);  // Write data
 
            const size_t written_bytes = fwrite(buffer, tag_block.cmpLength, 1, ctx->imageStream);
            if(written_bytes == 1)
            {
                TRACE("Successfully wrote media tag block type %d (%" PRIu64 " bytes)", tag_block.type,
                      tag_block.cmpLength);
                // Add media tag block to index
                TRACE("Adding media tag type %d block to index", tag_block.type);
                IndexEntry tag_index_entry;
                tag_index_entry.blockType = DataBlock;
                tag_index_entry.dataType  = tag_block.type;
                tag_index_entry.offset    = tag_position;
                utarray_push_back(ctx->index_entries, &tag_index_entry);
                TRACE("Added media tag block type %d index entry at offset %" PRIu64, tag_block.type, tag_position);
            }
        }
 
        if(tag_block.compression == Lzma) free(buffer);
    }
}

static void write_tape_file_block(const aaruformat_context *ctx)
{
    if(ctx->tape_files == NULL) return;
 
    // Iterate the uthash and count how many entries do we have
    const tapeFileHashEntry *tape_file       = NULL;
    const tapeFileHashEntry *tmp_tape_file   = NULL;
    size_t                   tape_file_count = 0;
    HASH_ITER(hh, ctx->tape_files, tape_file, tmp_tape_file) tape_file_count++;
 
    // Create a memory buffer to copy all the file entries
    const size_t   buffer_size = tape_file_count * sizeof(TapeFileEntry);
    TapeFileEntry *buffer      = malloc(buffer_size);
    if(buffer == NULL)
    {
        TRACE("Failed to allocate memory for tape file entries");
        return;
    }
    memset(buffer, 0, buffer_size);
    size_t index = 0;
    HASH_ITER(hh, ctx->tape_files, tape_file, tmp_tape_file)
    {
        if(index >= tape_file_count) break;
        memcpy(&buffer[index], &tape_file->fileEntry, sizeof(TapeFileEntry));
        index++;
    }
 
    // Create the tape file block in memory
    TapeFileHeader tape_file_block = {0};
    tape_file_block.identifier     = TapeFileBlock;
    tape_file_block.length         = (uint32_t)buffer_size;
    tape_file_block.crc64          = aaruf_crc64_data((uint8_t *)buffer, (uint32_t)tape_file_block.length);
 
    // Write tape file block to file, block aligned
    fseek(ctx->imageStream, 0, SEEK_END);
    long           block_position = ftell(ctx->imageStream);
    const uint64_t alignment_mask = (1ULL << ctx->user_data_ddt_header.blockAlignmentShift) - 1;
    if(block_position & alignment_mask)
    {
        const uint64_t aligned_position = block_position + alignment_mask & ~alignment_mask;
        fseek(ctx->imageStream, aligned_position, SEEK_SET);
        block_position = aligned_position;
    }
    TRACE("Writing tape file block at position %ld", block_position);
    if(fwrite(&tape_file_block, sizeof(TapeFileHeader), 1, ctx->imageStream) == 1)
    {
        const size_t written_bytes = fwrite(buffer, tape_file_block.length, 1, ctx->imageStream);
        if(written_bytes == 1)
        {
            TRACE("Successfully wrote tape file block (%" PRIu64 " bytes)", tape_file_block.length);
            // Add tape file block to index
            TRACE("Adding tape file block to index");
            IndexEntry index_entry;
            index_entry.blockType = TapeFileBlock;
            index_entry.dataType  = 0;
            index_entry.offset    = block_position;
            utarray_push_back(ctx->index_entries, &index_entry);
            TRACE("Added tape file block index entry at offset %" PRIu64, block_position);
        }
    }
 
    free(buffer);
}

static void write_tape_partition_block(const aaruformat_context *ctx)
{
    if(ctx->tape_partitions == NULL) return;
 
    // Iterate the uthash and count how many entries do we have
    const TapePartitionHashEntry *tape_partition       = NULL;
    const TapePartitionHashEntry *tmp_tape_partition   = NULL;
    size_t                        tape_partition_count = 0;
    HASH_ITER(hh, ctx->tape_partitions, tape_partition, tmp_tape_partition) tape_partition_count++;
 
    // Create a memory buffer to copy all the partition entries
    const size_t        buffer_size = tape_partition_count * sizeof(TapePartitionEntry);
    TapePartitionEntry *buffer      = malloc(buffer_size);
    if(buffer == NULL)
    {
        TRACE("Failed to allocate memory for tape partition entries");
        return;
    }
    memset(buffer, 0, buffer_size);
    size_t index = 0;
    HASH_ITER(hh, ctx->tape_partitions, tape_partition, tmp_tape_partition)
    {
        if(index >= tape_partition_count) break;
        memcpy(&buffer[index], &tape_partition->partitionEntry, sizeof(TapePartitionEntry));
        index++;
    }
 
    // Create the tape partition block in memory
    TapePartitionHeader tape_partition_block = {0};
    tape_partition_block.identifier          = TapePartitionBlock;
    tape_partition_block.length              = (uint32_t)buffer_size;
    tape_partition_block.crc64 = aaruf_crc64_data((uint8_t *)buffer, (uint32_t)tape_partition_block.length);
 
    // Write tape partition block to partition, block aligned
    fseek(ctx->imageStream, 0, SEEK_END);
    long           block_position = ftell(ctx->imageStream);
    const uint64_t alignment_mask = (1ULL << ctx->user_data_ddt_header.blockAlignmentShift) - 1;
    if(block_position & alignment_mask)
    {
        const uint64_t aligned_position = block_position + alignment_mask & ~alignment_mask;
        fseek(ctx->imageStream, aligned_position, SEEK_SET);
        block_position = aligned_position;
    }
    TRACE("Writing tape partition block at position %ld", block_position);
    if(fwrite(&tape_partition_block, sizeof(TapePartitionHeader), 1, ctx->imageStream) == 1)
    {
        const size_t written_bytes = fwrite(buffer, tape_partition_block.length, 1, ctx->imageStream);
        if(written_bytes == 1)
        {
            TRACE("Successfully wrote tape partition block (%" PRIu64 " bytes)", tape_partition_block.length);
            // Add tape partition block to index
            TRACE("Adding tape partition block to index");
            IndexEntry index_entry;
            index_entry.blockType = TapePartitionBlock;
            index_entry.dataType  = 0;
            index_entry.offset    = block_position;
            utarray_push_back(ctx->index_entries, &index_entry);
            TRACE("Added tape partition block index entry at offset %" PRIu64, block_position);
        }
    }
 
    free(buffer);
}

static void write_geometry_block(const aaruformat_context *ctx)
{
    if(ctx->geometry_block.identifier != GeometryBlock) return;
 
    fseek(ctx->imageStream, 0, SEEK_END);
    long           block_position = ftell(ctx->imageStream);
    const uint64_t alignment_mask = (1ULL << ctx->user_data_ddt_header.blockAlignmentShift) - 1;
    if(block_position & alignment_mask)
    {
        const uint64_t aligned_position = block_position + alignment_mask & ~alignment_mask;
        fseek(ctx->imageStream, aligned_position, SEEK_SET);
        block_position = aligned_position;
    }
 
    TRACE("Writing geometry block at position %ld", block_position);
 
    // Write header
    if(fwrite(&ctx->geometry_block, sizeof(GeometryBlockHeader), 1, ctx->imageStream) == 1)
    {
        TRACE("Successfully wrote geometry block");
 
        // Add geometry block to index
        TRACE("Adding geometry block to index");
        IndexEntry index_entry;
        index_entry.blockType = GeometryBlock;
        index_entry.dataType  = 0;
        index_entry.offset    = block_position;
        utarray_push_back(ctx->index_entries, &index_entry);
        TRACE("Added geometry block index entry at offset %" PRIu64, block_position);
    }
}

static void write_metadata_block(aaruformat_context *ctx)
{
    if(ctx->metadata_block_header.identifier != MetadataBlock && ctx->metadata_block_header.mediaSequence == 0 &&
       ctx->metadata_block_header.lastMediaSequence == 0 && ctx->creator == NULL && ctx->comments == NULL &&
       ctx->media_title == NULL && ctx->media_manufacturer == NULL && ctx->media_model == NULL &&
       ctx->media_serial_number == NULL && ctx->media_barcode == NULL && ctx->media_part_number == NULL &&
       ctx->drive_manufacturer == NULL && ctx->drive_model == NULL && ctx->drive_serial_number == NULL &&
       ctx->drive_firmware_revision == NULL)
        return;
 
    ctx->metadata_block_header.blockSize =
        sizeof(MetadataBlockHeader) + ctx->metadata_block_header.creatorLength +
        ctx->metadata_block_header.commentsLength + ctx->metadata_block_header.mediaTitleLength +
        ctx->metadata_block_header.mediaManufacturerLength + ctx->metadata_block_header.mediaModelLength +
        ctx->metadata_block_header.mediaSerialNumberLength + ctx->metadata_block_header.mediaBarcodeLength +
        ctx->metadata_block_header.mediaPartNumberLength + ctx->metadata_block_header.driveManufacturerLength +
        ctx->metadata_block_header.driveModelLength + ctx->metadata_block_header.driveSerialNumberLength +
        ctx->metadata_block_header.driveFirmwareRevisionLength;
 
    ctx->metadata_block_header.identifier = MetadataBlock;
 
    int pos = sizeof(MetadataBlockHeader);
 
    uint8_t *buffer = calloc(1, ctx->metadata_block_header.blockSize);
    if(buffer == NULL) return;
 
    if(ctx->creator != NULL && ctx->metadata_block_header.creatorLength > 0)
    {
        memcpy(buffer + pos, ctx->creator, ctx->metadata_block_header.creatorLength);
        ctx->metadata_block_header.creatorOffset = pos;
        pos += ctx->metadata_block_header.creatorLength;
    }
 
    if(ctx->comments != NULL && ctx->metadata_block_header.commentsLength > 0)
    {
        memcpy(buffer + pos, ctx->comments, ctx->metadata_block_header.commentsLength);
        ctx->metadata_block_header.commentsOffset = pos;
        pos += ctx->metadata_block_header.commentsLength;
    }
 
    if(ctx->media_title != NULL && ctx->metadata_block_header.mediaTitleLength > 0)
    {
        memcpy(buffer + pos, ctx->media_title, ctx->metadata_block_header.mediaTitleLength);
        ctx->metadata_block_header.mediaTitleOffset = pos;
        pos += ctx->metadata_block_header.mediaTitleLength;
    }
 
    if(ctx->media_manufacturer != NULL && ctx->metadata_block_header.mediaManufacturerLength > 0)
    {
        memcpy(buffer + pos, ctx->media_manufacturer, ctx->metadata_block_header.mediaManufacturerLength);
        ctx->metadata_block_header.mediaManufacturerOffset = pos;
        pos += ctx->metadata_block_header.mediaManufacturerLength;
    }
 
    if(ctx->media_model != NULL && ctx->metadata_block_header.mediaModelLength > 0)
    {
        memcpy(buffer + pos, ctx->media_model, ctx->metadata_block_header.mediaModelLength);
        ctx->metadata_block_header.mediaModelOffset = pos;
        pos += ctx->metadata_block_header.mediaModelLength;
    }
 
    if(ctx->media_serial_number != NULL && ctx->metadata_block_header.mediaSerialNumberLength > 0)
    {
        memcpy(buffer + pos, ctx->media_serial_number, ctx->metadata_block_header.mediaSerialNumberLength);
        ctx->metadata_block_header.mediaSerialNumberOffset = pos;
        pos += ctx->metadata_block_header.mediaSerialNumberLength;
    }
 
    if(ctx->media_barcode != NULL && ctx->metadata_block_header.mediaBarcodeLength > 0)
    {
        memcpy(buffer + pos, ctx->media_barcode, ctx->metadata_block_header.mediaBarcodeLength);
        ctx->metadata_block_header.mediaBarcodeOffset = pos;
        pos += ctx->metadata_block_header.mediaBarcodeLength;
    }
 
    if(ctx->media_part_number != NULL && ctx->metadata_block_header.mediaPartNumberLength > 0)
    {
        memcpy(buffer + pos, ctx->media_part_number, ctx->metadata_block_header.mediaPartNumberLength);
        ctx->metadata_block_header.mediaPartNumberOffset = pos;
        pos += ctx->metadata_block_header.mediaPartNumberLength;
    }
 
    if(ctx->drive_manufacturer != NULL && ctx->metadata_block_header.driveManufacturerLength > 0)
    {
        memcpy(buffer + pos, ctx->drive_manufacturer, ctx->metadata_block_header.driveManufacturerLength);
        ctx->metadata_block_header.driveManufacturerOffset = pos;
        pos += ctx->metadata_block_header.driveManufacturerLength;
    }
 
    if(ctx->drive_model != NULL && ctx->metadata_block_header.driveModelLength > 0)
    {
        memcpy(buffer + pos, ctx->drive_model, ctx->metadata_block_header.driveModelLength);
        ctx->metadata_block_header.driveModelOffset = pos;
        pos += ctx->metadata_block_header.driveModelLength;
    }
 
    if(ctx->drive_serial_number != NULL && ctx->metadata_block_header.driveSerialNumberLength > 0)
    {
        memcpy(buffer + pos, ctx->drive_serial_number, ctx->metadata_block_header.driveSerialNumberLength);
        ctx->metadata_block_header.driveSerialNumberOffset = pos;
        pos += ctx->metadata_block_header.driveSerialNumberLength;
    }
 
    if(ctx->drive_firmware_revision != NULL && ctx->metadata_block_header.driveFirmwareRevisionLength > 0)
    {
        memcpy(buffer + pos, ctx->drive_firmware_revision, ctx->metadata_block_header.driveFirmwareRevisionLength);
        ctx->metadata_block_header.driveFirmwareRevisionOffset = pos;
    }
 
    fseek(ctx->imageStream, 0, SEEK_END);
    long           block_position = ftell(ctx->imageStream);
    const uint64_t alignment_mask = (1ULL << ctx->user_data_ddt_header.blockAlignmentShift) - 1;
    if(block_position & alignment_mask)
    {
        const uint64_t aligned_position = block_position + alignment_mask & ~alignment_mask;
        fseek(ctx->imageStream, aligned_position, SEEK_SET);
        block_position = aligned_position;
    }
 
    TRACE("Writing metadata block at position %ld", block_position);
 
    if(fwrite(buffer, ctx->metadata_block_header.blockSize, 1, ctx->imageStream) == 1)
    {
        TRACE("Successfully wrote metadata block");
 
        // Add metadata block to index
        TRACE("Adding metadata block to index");
        IndexEntry index_entry;
        index_entry.blockType = MetadataBlock;
        index_entry.dataType  = 0;
        index_entry.offset    = block_position;
        utarray_push_back(ctx->index_entries, &index_entry);
        TRACE("Added metadata block index entry at offset %" PRIu64, block_position);
    }
 
    free(buffer);
}

static void write_dumphw_block(aaruformat_context *ctx)
{
 
    if(ctx->dump_hardware_entries_with_data == NULL || ctx->dump_hardware_header.entries == 0 ||
       ctx->dump_hardware_header.identifier != DumpHardwareBlock)
        return;
 
    const size_t required_length = sizeof(DumpHardwareHeader) + ctx->dump_hardware_header.length;
 
    uint8_t *buffer = calloc(1, required_length);
 
    if(buffer == NULL) return;
 
    // Start to iterate and copy the data
    size_t offset = 0;
    for(int i = 0; i < ctx->dump_hardware_header.entries; i++)
    {
        size_t entry_size = sizeof(DumpHardwareEntry) +
                            ctx->dump_hardware_entries_with_data[i].entry.manufacturerLength +
                            ctx->dump_hardware_entries_with_data[i].entry.modelLength +
                            ctx->dump_hardware_entries_with_data[i].entry.revisionLength +
                            ctx->dump_hardware_entries_with_data[i].entry.firmwareLength +
                            ctx->dump_hardware_entries_with_data[i].entry.serialLength +
                            ctx->dump_hardware_entries_with_data[i].entry.softwareNameLength +
                            ctx->dump_hardware_entries_with_data[i].entry.softwareVersionLength +
                            ctx->dump_hardware_entries_with_data[i].entry.softwareOperatingSystemLength +
                            ctx->dump_hardware_entries_with_data[i].entry.extents * sizeof(DumpExtent);
 
        if(offset + entry_size > required_length)
        {
            FATAL("Calculated size exceeds provided buffer length");
            free(buffer);
            return;
        }
 
        memcpy(buffer + offset, &ctx->dump_hardware_entries_with_data[i].entry, sizeof(DumpHardwareEntry));
        offset += sizeof(DumpHardwareEntry);
        if(ctx->dump_hardware_entries_with_data[i].entry.manufacturerLength > 0 &&
           ctx->dump_hardware_entries_with_data[i].manufacturer != NULL)
        {
            memcpy(buffer + offset, ctx->dump_hardware_entries_with_data[i].manufacturer,
                   ctx->dump_hardware_entries_with_data[i].entry.manufacturerLength);
            offset += ctx->dump_hardware_entries_with_data[i].entry.manufacturerLength;
        }
        if(ctx->dump_hardware_entries_with_data[i].entry.modelLength > 0 &&
           ctx->dump_hardware_entries_with_data[i].model != NULL)
        {
            memcpy(buffer + offset, ctx->dump_hardware_entries_with_data[i].model,
                   ctx->dump_hardware_entries_with_data[i].entry.modelLength);
            offset += ctx->dump_hardware_entries_with_data[i].entry.modelLength;
        }
        if(ctx->dump_hardware_entries_with_data[i].entry.revisionLength > 0 &&
           ctx->dump_hardware_entries_with_data[i].revision != NULL)
        {
            memcpy(buffer + offset, ctx->dump_hardware_entries_with_data[i].revision,
                   ctx->dump_hardware_entries_with_data[i].entry.revisionLength);
            offset += ctx->dump_hardware_entries_with_data[i].entry.revisionLength;
        }
        if(ctx->dump_hardware_entries_with_data[i].entry.firmwareLength > 0 &&
           ctx->dump_hardware_entries_with_data[i].firmware != NULL)
        {
            memcpy(buffer + offset, ctx->dump_hardware_entries_with_data[i].firmware,
                   ctx->dump_hardware_entries_with_data[i].entry.firmwareLength);
            offset += ctx->dump_hardware_entries_with_data[i].entry.firmwareLength;
        }
        if(ctx->dump_hardware_entries_with_data[i].entry.serialLength > 0 &&
           ctx->dump_hardware_entries_with_data[i].serial != NULL)
        {
            memcpy(buffer + offset, ctx->dump_hardware_entries_with_data[i].serial,
                   ctx->dump_hardware_entries_with_data[i].entry.serialLength);
            offset += ctx->dump_hardware_entries_with_data[i].entry.serialLength;
        }
        if(ctx->dump_hardware_entries_with_data[i].entry.softwareNameLength > 0 &&
           ctx->dump_hardware_entries_with_data[i].softwareName != NULL)
        {
            memcpy(buffer + offset, ctx->dump_hardware_entries_with_data[i].softwareName,
                   ctx->dump_hardware_entries_with_data[i].entry.softwareNameLength);
            offset += ctx->dump_hardware_entries_with_data[i].entry.softwareNameLength;
        }
        if(ctx->dump_hardware_entries_with_data[i].entry.softwareVersionLength > 0 &&
           ctx->dump_hardware_entries_with_data[i].softwareVersion != NULL)
        {
            memcpy(buffer + offset, ctx->dump_hardware_entries_with_data[i].softwareVersion,
                   ctx->dump_hardware_entries_with_data[i].entry.softwareVersionLength);
            offset += ctx->dump_hardware_entries_with_data[i].entry.softwareVersionLength;
        }
        if(ctx->dump_hardware_entries_with_data[i].entry.softwareOperatingSystemLength > 0 &&
           ctx->dump_hardware_entries_with_data[i].softwareOperatingSystem != NULL)
        {
            memcpy(buffer + offset, ctx->dump_hardware_entries_with_data[i].softwareOperatingSystem,
                   ctx->dump_hardware_entries_with_data[i].entry.softwareOperatingSystemLength);
            offset += ctx->dump_hardware_entries_with_data[i].entry.softwareOperatingSystemLength;
        }
        if(ctx->dump_hardware_entries_with_data[i].entry.extents > 0 &&
           ctx->dump_hardware_entries_with_data[i].extents != NULL)
        {
            memcpy(buffer + offset, ctx->dump_hardware_entries_with_data[i].extents,
                   ctx->dump_hardware_entries_with_data[i].entry.extents * sizeof(DumpExtent));
            offset += ctx->dump_hardware_entries_with_data[i].entry.extents * sizeof(DumpExtent);
        }
    }
 
    // Calculate CRC64
    ctx->dump_hardware_header.crc64 =
        aaruf_crc64_data(buffer + sizeof(DumpHardwareHeader), ctx->dump_hardware_header.length);
 
    // Copy header
    memcpy(buffer, &ctx->dump_hardware_header, sizeof(DumpHardwareHeader));
 
    fseek(ctx->imageStream, 0, SEEK_END);
    long           block_position = ftell(ctx->imageStream);
    const uint64_t alignment_mask = (1ULL << ctx->user_data_ddt_header.blockAlignmentShift) - 1;
    if(block_position & alignment_mask)
    {
        const uint64_t aligned_position = block_position + alignment_mask & ~alignment_mask;
        fseek(ctx->imageStream, aligned_position, SEEK_SET);
        block_position = aligned_position;
    }
    TRACE("Writing dump hardware block at position %ld", block_position);
    if(fwrite(buffer, required_length, 1, ctx->imageStream) == 1)
    {
        TRACE("Successfully wrote dump hardware block");
 
        // Add dump hardware block to index
        TRACE("Adding dump hardware block to index");
        IndexEntry index_entry;
        index_entry.blockType = DumpHardwareBlock;
        index_entry.dataType  = 0;
        index_entry.offset    = block_position;
        utarray_push_back(ctx->index_entries, &index_entry);
        TRACE("Added dump hardware block index entry at offset %" PRIu64, block_position);
    }
 
    free(buffer);
}

static void write_cicm_block(const aaruformat_context *ctx)
{
    if(ctx->cicm_block == NULL || ctx->cicm_block_header.length == 0 || ctx->cicm_block_header.identifier != CicmBlock)
        return;
 
    fseek(ctx->imageStream, 0, SEEK_END);
    long           block_position = ftell(ctx->imageStream);
    const uint64_t alignment_mask = (1ULL << ctx->user_data_ddt_header.blockAlignmentShift) - 1;
 
    if(block_position & alignment_mask)
    {
        const uint64_t aligned_position = block_position + alignment_mask & ~alignment_mask;
        fseek(ctx->imageStream, aligned_position, SEEK_SET);
        block_position = aligned_position;
    }
 
    TRACE("Writing CICM XML block at position %ld", block_position);
    if(fwrite(&ctx->cicm_block_header, sizeof(CicmMetadataBlock), 1, ctx->imageStream) == 1)
        if(fwrite(ctx->cicm_block, ctx->cicm_block_header.length, 1, ctx->imageStream) == 1)
        {
            TRACE("Successfully wrote CICM XML block");
 
            // Add CICM block to index
            TRACE("Adding CICM XML block to index");
            IndexEntry index_entry;
            index_entry.blockType = CicmBlock;
            index_entry.dataType  = 0;
            index_entry.offset    = block_position;
            utarray_push_back(ctx->index_entries, &index_entry);
            TRACE("Added CICM XML block index entry at offset %" PRIu64, block_position);
        }
}

static void write_aaru_json_block(const aaruformat_context *ctx)
{
    if(ctx->json_block == NULL || ctx->json_block_header.length == 0 ||
       ctx->json_block_header.identifier != AaruMetadataJsonBlock)
        return;
 
    fseek(ctx->imageStream, 0, SEEK_END);
    long           block_position = ftell(ctx->imageStream);
    const uint64_t alignment_mask = (1ULL << ctx->user_data_ddt_header.blockAlignmentShift) - 1;
 
    if(block_position & alignment_mask)
    {
        const uint64_t aligned_position = block_position + alignment_mask & ~alignment_mask;
        fseek(ctx->imageStream, aligned_position, SEEK_SET);
        block_position = aligned_position;
    }
 
    TRACE("Writing Aaru metadata JSON block at position %ld", block_position);
    if(fwrite(&ctx->json_block_header, sizeof(AaruMetadataJsonBlockHeader), 1, ctx->imageStream) == 1)
        if(fwrite(ctx->json_block, ctx->json_block_header.length, 1, ctx->imageStream) == 1)
        {
            TRACE("Successfully wrote Aaru metadata JSON block");
 
            // Add Aaru metadata JSON block to index
            TRACE("Adding Aaru metadata JSON block to index");
            IndexEntry index_entry;
            index_entry.blockType = AaruMetadataJsonBlock;
            index_entry.dataType  = 0;
            index_entry.offset    = block_position;
            utarray_push_back(ctx->index_entries, &index_entry);
            TRACE("Added Aaru metadata JSON block index entry at offset %" PRIu64, block_position);
        }
}

static int32_t write_index_block(aaruformat_context *ctx)
{
    // Write the complete index at the end of the file
    TRACE("Writing index at the end of the file");
    fseek(ctx->imageStream, 0, SEEK_END);
    long index_position = ftell(ctx->imageStream);
 
    // Align index position to block boundary if needed
    uint64_t alignment_mask = (1ULL << ctx->user_data_ddt_header.blockAlignmentShift) - 1;
    if(index_position & alignment_mask)
    {
        uint64_t aligned_position = index_position + alignment_mask & ~alignment_mask;
        fseek(ctx->imageStream, aligned_position, SEEK_SET);
        index_position = aligned_position;
        TRACE("Aligned index position to %" PRIu64, aligned_position);
    }
 
    // Prepare index header
    IndexHeader3 index_header;
    index_header.identifier = IndexBlock3;
    index_header.entries    = utarray_len(ctx->index_entries);
    index_header.previous   = 0;  // No previous index for now
 
    TRACE("Writing index with %" PRIu64 " entries at position %ld", index_header.entries, index_position);
 
    // Calculate CRC64 of index entries
    crc64_ctx *index_crc64_context = aaruf_crc64_init();
    if(index_crc64_context != NULL && index_header.entries > 0)
    {
        size_t index_data_size = index_header.entries * sizeof(IndexEntry);
        aaruf_crc64_update(index_crc64_context, utarray_front(ctx->index_entries), index_data_size);
        aaruf_crc64_final(index_crc64_context, &index_header.crc64);
        TRACE("Calculated index CRC64: 0x%16lX", index_header.crc64);
    }
    else
        index_header.crc64 = 0;
 
    // Write index header
    if(fwrite(&index_header, sizeof(IndexHeader3), 1, ctx->imageStream) == 1)
    {
        TRACE("Successfully wrote index header");
 
        // Write index entries
        if(index_header.entries > 0)
        {
            size_t      entries_written = 0;
            IndexEntry *entry           = NULL;
 
            for(entry = (IndexEntry *)utarray_front(ctx->index_entries); entry != NULL;
                entry = (IndexEntry *)utarray_next(ctx->index_entries, entry))
                if(fwrite(entry, sizeof(IndexEntry), 1, ctx->imageStream) == 1)
                {
                    entries_written++;
                    TRACE("Wrote index entry: blockType=0x%08X dataType=%u offset=%" PRIu64, entry->blockType,
                          entry->dataType, entry->offset);
                }
                else
                {
                    TRACE("Failed to write index entry %zu", entries_written);
                    break;
                }
 
            if(entries_written == index_header.entries)
            {
                TRACE("Successfully wrote all %zu index entries", entries_written);
 
                // Update header with index offset and rewrite it
                ctx->header.indexOffset = index_position;
                TRACE("Updating header with index offset: %" PRIu64, ctx->header.indexOffset);
 
                // Seek back to beginning and rewrite header
                fseek(ctx->imageStream, 0, SEEK_SET);
                if(fwrite(&ctx->header, sizeof(AaruHeaderV2), 1, ctx->imageStream) == 1)
                    TRACE("Successfully updated header with index offset");
                else
                {
                    TRACE("Failed to update header with index offset");
                    return AARUF_ERROR_CANNOT_WRITE_HEADER;
                }
            }
            else
            {
                TRACE("Failed to write all index entries (wrote %zu of %" PRIu64 ")", entries_written,
                      index_header.entries);
                return AARUF_ERROR_CANNOT_WRITE_HEADER;
            }
        }
    }
    else
    {
        TRACE("Failed to write index header");
        return AARUF_ERROR_CANNOT_WRITE_HEADER;
    }
 
    return AARUF_STATUS_OK;
}

AARU_EXPORT int AARU_CALL aaruf_close(void *context)
{
    TRACE("Entering aaruf_close(%p)", context);
 
    mediaTagEntry *media_tag     = NULL;
    mediaTagEntry *tmp_media_tag = NULL;
 
    if(context == NULL)
    {
        FATAL("Invalid context");
        errno = EINVAL;
        return -1;
    }
 
    aaruformat_context *ctx = context;
 
    // Not a libaaruformat context
    if(ctx->magic != AARU_MAGIC)
    {
        FATAL("Invalid context");
        errno = EINVAL;
        return -1;
    }
 
    if(ctx->is_writing)
    {
        TRACE("File is writing");
 
        TRACE("Seeking to start of image");
        // Write the header at the beginning of the file
        fseek(ctx->imageStream, 0, SEEK_SET);
 
        TRACE("Writing header at position 0");
        if(fwrite(&ctx->header, sizeof(AaruHeaderV2), 1, ctx->imageStream) != 1)
        {
            fclose(ctx->imageStream);
            ctx->imageStream = NULL;
            errno            = AARUF_ERROR_CANNOT_WRITE_HEADER;
            return -1;
        }
 
        // Close current block first
        TRACE("Closing current block if any");
        if(ctx->writing_buffer != NULL)
        {
            int error = aaruf_close_current_block(ctx);
 
            if(error != AARUF_STATUS_OK) return error;
        }
 
        int32_t res;
        if(ctx->is_tape)
        {
            // Write tape DDT
            res = write_tape_ddt(ctx);
            if(res != AARUF_STATUS_OK) return res;
        }
        else
        {
            // Write cached secondary DDT table if any
            res = write_cached_secondary_ddt(ctx);
            if(res != AARUF_STATUS_OK) return res;
 
            // Write primary DDT table (multi-level) if applicable
            res = write_primary_ddt(ctx);
            if(res != AARUF_STATUS_OK) return res;
 
            // Write single-level DDT table if applicable
            res = write_single_level_ddt(ctx);
            if(res != AARUF_STATUS_OK) return res;
        }
 
        // Finalize checksums and write checksum block
        write_checksum_block(ctx);
 
        // Write tracks block
        write_tracks_block(ctx);
 
        // Write MODE 2 subheader data block
        write_mode2_subheaders_block(ctx);
 
        // Write CD sector prefix data block
        write_sector_prefix(ctx);
 
        // Write sector prefix DDT (statuses + optional indexes)
        write_sector_prefix_ddt(ctx);
 
        // Write CD sector suffix data block (EDC/ECC captures)
        write_sector_suffix(ctx);
 
        // Write sector prefix DDT (EDC/ECC captures)
        write_sector_suffix_ddt(ctx);
 
        // Write sector subchannel data block
        write_sector_subchannel(ctx);
 
        // Write DVD long sector data blocks
        write_dvd_long_sector_blocks(ctx);
 
        // Write DVD decrypted title keys
        write_dvd_title_key_decrypted_block(ctx);
 
        // Write media tags data blocks
        write_media_tags(ctx);
 
        // Write tape files
        write_tape_file_block(ctx);
 
        // Write tape partitions
        write_tape_partition_block(ctx);
 
        // Write geometry block if any
        write_geometry_block(ctx);
 
        // Write metadata block
        write_metadata_block(ctx);
 
        // Write dump hardware block if any
        write_dumphw_block(ctx);
 
        // Write CICM XML block if any
        write_cicm_block(ctx);
 
        // Write Aaru metadata JSON block if any
        write_aaru_json_block(ctx);
 
        // Write the complete index at the end of the file
        res = write_index_block(ctx);
        if(res != AARUF_STATUS_OK) return res;
 
        if(ctx->deduplicate && ctx->sector_hash_map != NULL)
        {
            TRACE("Clearing sector hash map");
            // Clear sector hash map
            free_map(ctx->sector_hash_map);
            ctx->sector_hash_map = NULL;
        }
    }
 
    TRACE("Freeing memory pointers");
    // This may do nothing if imageStream is NULL, but as the behaviour is undefined, better sure than sorry
    if(ctx->imageStream != NULL)
    {
        fclose(ctx->imageStream);
        ctx->imageStream = NULL;
    }
 
    // Free index entries array
    if(ctx->index_entries != NULL)
    {
        utarray_free(ctx->index_entries);
        ctx->index_entries = NULL;
    }
 
    free(ctx->sector_prefix);
    ctx->sector_prefix = NULL;
    free(ctx->sector_prefix_corrected);
    ctx->sector_prefix_corrected = NULL;
    free(ctx->sector_suffix);
    ctx->sector_suffix = NULL;
    free(ctx->sector_suffix_corrected);
    ctx->sector_suffix_corrected = NULL;
    free(ctx->sector_subchannel);
    ctx->sector_subchannel = NULL;
    free(ctx->mode2_subheaders);
    ctx->mode2_subheaders = NULL;
 
    TRACE("Freeing media tags");
    if(ctx->mediaTags != NULL) HASH_ITER(hh, ctx->mediaTags, media_tag, tmp_media_tag)
        {
            HASH_DEL(ctx->mediaTags, media_tag);
            free(media_tag->data);
            free(media_tag);
        }
 
#ifdef __linux__  // TODO: Implement
    TRACE("Unmapping user data DDT if it is not in memory");
    if(!ctx->in_memory_ddt)
    {
        munmap(ctx->user_data_ddt, ctx->mapped_memory_ddt_size);
        ctx->user_data_ddt = NULL;
    }
#endif
 
    free(ctx->sector_prefix_ddt2);
    ctx->sector_prefix_ddt2 = NULL;
    free(ctx->sector_prefix_ddt);
    ctx->sector_prefix_ddt = NULL;
    free(ctx->sector_suffix_ddt2);
    ctx->sector_suffix_ddt2 = NULL;
    free(ctx->sector_suffix_ddt);
    ctx->sector_suffix_ddt = NULL;
 
    free(ctx->metadata_block);
    ctx->metadata_block = NULL;
    free(ctx->track_entries);
    ctx->track_entries = NULL;
    free(ctx->cicm_block);
    ctx->cicm_block = NULL;
 
    if(ctx->dump_hardware_entries_with_data != NULL)
    {
        for(int i = 0; i < ctx->dump_hardware_header.entries; i++)
        {
            free(ctx->dump_hardware_entries_with_data[i].extents);
            ctx->dump_hardware_entries_with_data[i].extents = NULL;
            free(ctx->dump_hardware_entries_with_data[i].manufacturer);
            ctx->dump_hardware_entries_with_data[i].manufacturer = NULL;
            free(ctx->dump_hardware_entries_with_data[i].model);
            ctx->dump_hardware_entries_with_data[i].model = NULL;
            free(ctx->dump_hardware_entries_with_data[i].revision);
            ctx->dump_hardware_entries_with_data[i].revision = NULL;
            free(ctx->dump_hardware_entries_with_data[i].firmware);
            ctx->dump_hardware_entries_with_data[i].firmware = NULL;
            free(ctx->dump_hardware_entries_with_data[i].serial);
            ctx->dump_hardware_entries_with_data[i].serial = NULL;
            free(ctx->dump_hardware_entries_with_data[i].softwareName);
            ctx->dump_hardware_entries_with_data[i].softwareName = NULL;
            free(ctx->dump_hardware_entries_with_data[i].softwareVersion);
            ctx->dump_hardware_entries_with_data[i].softwareVersion = NULL;
            free(ctx->dump_hardware_entries_with_data[i].softwareOperatingSystem);
            ctx->dump_hardware_entries_with_data[i].softwareOperatingSystem = NULL;
        }
        ctx->dump_hardware_entries_with_data = NULL;
    }
 
    free(ctx->readableSectorTags);
    ctx->readableSectorTags = NULL;
 
    free(ctx->ecc_cd_context);
    ctx->ecc_cd_context = NULL;
 
    free(ctx->checksums.spamsum);
    ctx->checksums.spamsum = NULL;
 
    free(ctx->sector_id);
    free(ctx->sector_ied);
    free(ctx->sector_cpr_mai);
    free(ctx->sector_edc);
 
    // TODO: Free caches
 
    free(context);
 
    TRACE("Exiting aaruf_close() = 0");
    return 0;
}