def load_disassembly_hunk(f, program_data):
    program_data.branch_addresses = persistence.read_dict_uint32_to_set_of_uint32s(f)
    program_data.reference_addresses = persistence.read_dict_uint32_to_set_of_uint32s(f)
    program_data.symbols_by_address = persistence.read_dict_uint32_to_string(f)
    program_data.post_segment_addresses = persistence.read_dict_uint32_to_list_of_uint32s(f)
    program_data.flags = persistence.read_uint32(f)
    program_data.processor_id = persistence.read_uint32(f)

    # Reconstitute the segment block list.
    num_blocks = persistence.read_uint32(f)
    program_data.blocks = [ None ] * num_blocks
    for i in xrange(num_blocks):
        program_data.blocks[i] = read_SegmentBlock(f)

    ## POST PROCESSING
    # Rebuild the segment block list indexing lists.
    program_data.block_addresses = [ 0 ] * num_blocks
    program_data.block_line0s_dirtyidx = 0
    program_data.block_line0s = program_data.block_addresses[:]
    for i in xrange(num_blocks):
        program_data.block_addresses[i] = program_data.blocks[i].address
Exemple #2
0
def load_disassembly_hunk(f, program_data):
    program_data.branch_addresses = persistence.read_dict_uint32_to_set_of_uint32s(
        f)
    program_data.reference_addresses = persistence.read_dict_uint32_to_set_of_uint32s(
        f)
    program_data.symbols_by_address = persistence.read_dict_uint32_to_string(f)
    program_data.post_segment_addresses = persistence.read_dict_uint32_to_list_of_uint32s(
        f)
    program_data.flags = persistence.read_uint32(f)
    program_data.processor_id = persistence.read_uint32(f)

    # Reconstitute the segment block list.
    num_blocks = persistence.read_uint32(f)
    program_data.blocks = [None] * num_blocks
    for i in range(num_blocks):
        program_data.blocks[i] = read_SegmentBlock(f)

    ## POST PROCESSING
    # Rebuild the segment block list indexing lists.
    program_data.block_addresses = [0] * num_blocks
    program_data.block_line0s_dirtyidx = 0
    program_data.block_line0s = program_data.block_addresses[:]
    for i in range(num_blocks):
        program_data.block_addresses[i] = program_data.blocks[i].address
def convert_project_format_3_to_4(input_file):
    """
    This function should encapsulate all application-specific logic involved to
    make it independent of as many changes as possible.

    From version: 3.
    To version: 4.
    Modifications:
    - disassembly hunk processor id has changed from string to uint32.
    """
    SNAPSHOT_HUNK_VERSIONS = {
        SAVEFILE_HUNK_SOURCEDATA: 1,
        SAVEFILE_HUNK_SOURCEDATAINFO: 1,
        SAVEFILE_HUNK_LOADER: 1,
        SAVEFILE_HUNK_LOADERINTERNAL: 1,
        SAVEFILE_HUNK_DISASSEMBLY: 1,
    }

    input_file.seek(0, os.SEEK_END)
    file_size = input_file.tell()
    input_file.seek(0, os.SEEK_SET)

    savefile_id = persistence.read_uint32(input_file)
    savefile_version = persistence.read_uint16(input_file)
    if savefile_version != 3:
        return None

    logger.info("Upgrading save-file from version 3 to version 4: Processor id field..")
    save_count = persistence.read_uint32(input_file)

    output_file = tempfile.TemporaryFile()
    persistence.write_uint32(output_file, savefile_id)
    persistence.write_uint16(output_file, 4)
    persistence.write_uint32(output_file, save_count)

    while input_file.tell() < file_size:
        # This should be pretty straightforward.
        hunk_header_offset = input_file.tell()
        hunk_id = persistence.read_uint16(input_file)
        hunk_length = persistence.read_uint32(input_file)
        hunk_payload_offset = input_file.tell()

        actual_hunk_version = persistence.read_uint16(input_file)
        expected_hunk_version = SNAPSHOT_HUNK_VERSIONS[hunk_id]
        if expected_hunk_version != actual_hunk_version:
            logger.error("convert_project_format_3_to_4: hunk %d version mismatch %d != %d", hunk_id, expected_hunk_version, actual_hunk_version)
            return None
        logger.debug("convert_project_format_3_to_4: file hunk %d", hunk_id)

        # Copy unaffected hunks verbatim.
        if hunk_id != SAVEFILE_HUNK_DISASSEMBLY:
            input_file.seek(hunk_header_offset, os.SEEK_SET)
            raw_hunk_length = (hunk_payload_offset - hunk_header_offset) + hunk_length
            output_file.write(input_file.read(raw_hunk_length))
            continue

        ## 1. Load the hunk payload.
        branch_addresses = persistence.read_dict_uint32_to_set_of_uint32s(input_file)
        reference_addresses = persistence.read_dict_uint32_to_set_of_uint32s(input_file)
        symbols_by_address = persistence.read_dict_uint32_to_string(input_file)
        post_segment_addresses = persistence.read_dict_uint32_to_list_of_uint32s(input_file)
        flags = persistence.read_uint32(input_file)
        processor_name = persistence.read_string(input_file)

        # Reconstitute the segment block list.
        num_blocks = persistence.read_uint32(input_file)
        input_file_offset = input_file.tell()
        block_data_length = hunk_length - (input_file_offset - hunk_payload_offset)
        block_data_string = input_file.read(block_data_length)
        #blocks = [ None ] * num_blocks
        #for i in xrange(num_blocks):
        #    blocks[i] = read_SegmentBlock(input_file)
        ## 2. Write the generic hunk header, then the payload, then fill in the header.

        # Only these two are likely to have been in use.
        if processor_name == "m68k":
            processor_id = loaderlib.constants.PROCESSOR_M680x0
        elif processor_name == "mips":
            processor_id = loaderlib.constants.PROCESSOR_MIPS
        else:
            logger.error("convert_project_format_3_to_4: unrecognised arch name %s", processor_name)
            return None
        logger.debug("convert_project_format_3_to_4: arch name %s maps to processor id %d", processor_name, processor_id)

        # Write the as yet to be updated header.
        persistence.write_uint16(output_file, hunk_id)
        output_file_length_offset = output_file.tell()
        persistence.write_uint32(output_file, 0)
        output_file_payload_offset = output_file.tell()
        persistence.write_uint16(output_file, SNAPSHOT_HUNK_VERSIONS[SAVEFILE_HUNK_DISASSEMBLY] + 1)

        # Write the payload in the modified format.
        persistence.write_dict_uint32_to_set_of_uint32s(output_file, branch_addresses)
        persistence.write_dict_uint32_to_set_of_uint32s(output_file, reference_addresses)
        persistence.write_dict_uint32_to_string(output_file, symbols_by_address)
        persistence.write_dict_uint32_to_list_of_uint32s(output_file, post_segment_addresses)
        persistence.write_uint32(output_file, flags)
        persistence.write_uint32(output_file, processor_id)

        persistence.write_uint32(output_file, num_blocks)
        output_file_offset = output_file.tell()
        #for block in blocks:
        #    write_SegmentBlock(output_file, block)
        output_file.write(block_data_string)
        if output_file.tell() - output_file_offset != block_data_length:
            logger.error("convert_project_format_3_to_4: block length mismatch %d != %d", output_file.tell() - output_file_offset, block_data_length)
            return None

        # Update the header length field, then fast forward to the end of the hunk.
        new_hunk_length = output_file.tell() - output_file_payload_offset
        output_file.seek(output_file_length_offset, os.SEEK_SET)
        persistence.write_uint32(output_file, new_hunk_length)
        output_file.seek(new_hunk_length, os.SEEK_CUR)

        if output_file.tell() - output_file_payload_offset != new_hunk_length:
            logger.error("convert_project_format_3_to_4: block length mismatch %d != %d", output_file.tell() - output_file_payload_offset, new_hunk_length)
            return None

    return output_file
Exemple #4
0
def convert_project_format_3_to_4(input_file):
    """
    This function should encapsulate all application-specific logic involved to
    make it independent of as many changes as possible.

    From version: 3.
    To version: 4.
    Modifications:
    - disassembly hunk processor id has changed from string to uint32.
    """
    SNAPSHOT_HUNK_VERSIONS = {
        SAVEFILE_HUNK_SOURCEDATA: 1,
        SAVEFILE_HUNK_SOURCEDATAINFO: 1,
        SAVEFILE_HUNK_LOADER: 1,
        SAVEFILE_HUNK_LOADERINTERNAL: 1,
        SAVEFILE_HUNK_DISASSEMBLY: 1,
    }

    input_file.seek(0, os.SEEK_END)
    file_size = input_file.tell()
    input_file.seek(0, os.SEEK_SET)

    savefile_id = persistence.read_uint32(input_file)
    savefile_version = persistence.read_uint16(input_file)
    if savefile_version != 3:
        return None

    logger.info(
        "Upgrading save-file from version 3 to version 4: Processor id field.."
    )
    save_count = persistence.read_uint32(input_file)

    output_file = tempfile.TemporaryFile()
    persistence.write_uint32(output_file, savefile_id)
    persistence.write_uint16(output_file, 4)
    persistence.write_uint32(output_file, save_count)

    while input_file.tell() < file_size:
        # This should be pretty straightforward.
        hunk_header_offset = input_file.tell()
        hunk_id = persistence.read_uint16(input_file)
        hunk_length = persistence.read_uint32(input_file)
        hunk_payload_offset = input_file.tell()

        actual_hunk_version = persistence.read_uint16(input_file)
        expected_hunk_version = SNAPSHOT_HUNK_VERSIONS[hunk_id]
        if expected_hunk_version != actual_hunk_version:
            logger.error(
                "convert_project_format_3_to_4: hunk %d version mismatch %d != %d",
                hunk_id, expected_hunk_version, actual_hunk_version)
            return None
        logger.debug("convert_project_format_3_to_4: file hunk %d", hunk_id)

        # Copy unaffected hunks verbatim.
        if hunk_id != SAVEFILE_HUNK_DISASSEMBLY:
            input_file.seek(hunk_header_offset, os.SEEK_SET)
            raw_hunk_length = (hunk_payload_offset -
                               hunk_header_offset) + hunk_length
            output_file.write(input_file.read(raw_hunk_length))
            continue

        ## 1. Load the hunk payload.
        branch_addresses = persistence.read_dict_uint32_to_set_of_uint32s(
            input_file)
        reference_addresses = persistence.read_dict_uint32_to_set_of_uint32s(
            input_file)
        symbols_by_address = persistence.read_dict_uint32_to_string(input_file)
        post_segment_addresses = persistence.read_dict_uint32_to_list_of_uint32s(
            input_file)
        flags = persistence.read_uint32(input_file)
        processor_name = persistence.read_string(input_file)

        # Reconstitute the segment block list.
        num_blocks = persistence.read_uint32(input_file)
        input_file_offset = input_file.tell()
        block_data_length = hunk_length - (input_file_offset -
                                           hunk_payload_offset)
        block_data_string = input_file.read(block_data_length)
        #blocks = [ None ] * num_blocks
        #for i in range(num_blocks):
        #    blocks[i] = read_SegmentBlock(input_file)
        ## 2. Write the generic hunk header, then the payload, then fill in the header.

        # Only these two are likely to have been in use.
        if processor_name == "m68k":
            processor_id = loaderlib.constants.PROCESSOR_M680x0
        elif processor_name == "mips":
            processor_id = loaderlib.constants.PROCESSOR_MIPS
        else:
            logger.error(
                "convert_project_format_3_to_4: unrecognised arch name %s",
                processor_name)
            return None
        logger.debug(
            "convert_project_format_3_to_4: arch name %s maps to processor id %d",
            processor_name, processor_id)

        # Write the as yet to be updated header.
        persistence.write_uint16(output_file, hunk_id)
        output_file_length_offset = output_file.tell()
        persistence.write_uint32(output_file, 0)
        output_file_payload_offset = output_file.tell()
        persistence.write_uint16(
            output_file, SNAPSHOT_HUNK_VERSIONS[SAVEFILE_HUNK_DISASSEMBLY] + 1)

        # Write the payload in the modified format.
        persistence.write_dict_uint32_to_set_of_uint32s(
            output_file, branch_addresses)
        persistence.write_dict_uint32_to_set_of_uint32s(
            output_file, reference_addresses)
        persistence.write_dict_uint32_to_string(output_file,
                                                symbols_by_address)
        persistence.write_dict_uint32_to_list_of_uint32s(
            output_file, post_segment_addresses)
        persistence.write_uint32(output_file, flags)
        persistence.write_uint32(output_file, processor_id)

        persistence.write_uint32(output_file, num_blocks)
        output_file_offset = output_file.tell()
        #for block in blocks:
        #    write_SegmentBlock(output_file, block, program_data.dis_constant_pc_offset)
        output_file.write(block_data_string)
        if output_file.tell() - output_file_offset != block_data_length:
            logger.error(
                "convert_project_format_3_to_4: block length mismatch %d != %d",
                output_file.tell() - output_file_offset, block_data_length)
            return None

        # Update the header length field, then fast forward to the end of the hunk.
        new_hunk_length = output_file.tell() - output_file_payload_offset
        output_file.seek(output_file_length_offset, os.SEEK_SET)
        persistence.write_uint32(output_file, new_hunk_length)
        output_file.seek(new_hunk_length, os.SEEK_CUR)

        if output_file.tell() - output_file_payload_offset != new_hunk_length:
            logger.error(
                "convert_project_format_3_to_4: block length mismatch %d != %d",
                output_file.tell() - output_file_payload_offset,
                new_hunk_length)
            return None

    return output_file