#!/bin/bash

###############################################################################
USAGE="$0 <binary_input_file> [<binary_output_file>]"

HELP_TEXT="synthesize_dwarf.sh
Script that sticks all the parts of dwarf-synthesis together.

Usage: $USAGE

The provided <binary_input_file> is expected **NOT** to have any .eh_frame
section.

If <binary_output_file> is provided, a binary file equivalent to
<binary_input_file> that contains an .eh_frame ELF section will be written as
<binary_output_file>.

If not, the input file will be overwriten with such a file."
###############################################################################

function find_ml_dwarf_write {
    out=$(which "ml_dwarf_write.bin" 2>/dev/null)
    if [ -n "$out" ] ; then
        echo $out
        return 0
    fi
    for location in . "$(dirname $0)" ; do
        out="$location/ml_dwarf_write.bin"
        if [ -x "$out" ] ; then
            echo $out
            return 0
        fi
    done
    return 1
}

function bap_synth {
    bap "$INPUT_FILE" \
        --no-byteweight -p dwarfsynth \
        --dwarfsynth-output "$TMP_DIR/marshal" $BAP_ARGS \
        > /dev/null
    return $?
}

function dwarf_write_synth {
    $ML_DWARF_WRITE "$TMP_DIR/marshal" "$INPUT_FILE" "$TMP_DIR/eh_frame" \
        > /dev/null
    return $?
}

function dwarf_plug {
    objcopy \
        --add-section .eh_frame="$TMP_DIR/eh_frame" \
        "$INPUT_FILE" "$OUTPUT_FILE"
    return $?
}

if [ "$#" -lt "1" ] ; then
    >&2 echo -e "Missing argument.\n\n$HELP_TEXT"
    exit 1
fi

ML_DWARF_WRITE=$(find_ml_dwarf_write)
if [ "$?" -ne "0" ] ; then
    >&2 echo -e "Cannot find ml_dwarf_write"
    exit 1
fi

INPUT_FILE="$1"
OUTPUT_FILE="$2"

if ! [ -f "$INPUT_FILE" ] ; then
    >&2 echo -e "$INPUT_FILE: no such file.\n\n$HELP_TEXT"
fi

TMP_DIR="$(mktemp -d)"

bap_synth \
    && dwarf_write_synth \
    && dwarf_plug

rm -rf "$TMP_DIR"