#!/bin/bash

# Copyright (c) 2013-2019 Huawei Technologies Co., Ltd. All rights reserved.
# Copyright (c) 2020-2022 Huawei Device Co., Ltd. All rights reserved.
#
# Redistribution and use in source and binary forms, with or without modification,
# are permitted provided that the following conditions are met:
#
# 1. Redistributions of source code must retain the above copyright notice, this list of
#    conditions and the following disclaimer.
#
# 2. Redistributions in binary form must reproduce the above copyright notice, this list
#    of conditions and the following disclaimer in the documentation and/or other materials
#    provided with the distribution.
#
# 3. Neither the name of the copyright holder nor the names of its contributors may be used
#    to endorse or promote products derived from this software without specific prior written
#    permission.
#
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
# THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
# PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR
# CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
# EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
# PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
# OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
# WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
# OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
# ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
set -e

LOAD_BASE="0x2000000"
LLVM_ADDR2LINE=llvm-addr2line
GCC_ADDR2LINE=addr2line

get_line()
{
    SYM_ADDR=$(echo $1 | awk '{print $2}')
    ELF_OFFSET=$(echo ${SYM_ADDR} | cut -d '[' -f2 | cut -d ']' -f1)
    FILE_LINE=$(${ADDR2LINE} -f -e $2 ${ELF_OFFSET} | awk 'NR==2')
    if [[ "${FILE_LINE}" == *"?"* ]]; then
        typeset ELF_OFFSET=$((ELF_OFFSET+LOAD_BASE))
        ELF_OFFSET=$(echo "obase=16;${ELF_OFFSET}" | bc)
        FILE_LINE=$(${ADDR2LINE} -f -e $2 ${ELF_OFFSET} | awk 'NR==2')
    fi
    echo ${FILE_LINE}
}

parse_line()
{
    FILE_PATH=$(echo $1 | awk '{print $4}')
    ELF_FILE=${FILE_PATH##*/}
    ELF_FILE=${ELF_FILE//[$'\t\r\n']}
    FLAG=false
    for i in $2; do
        if [ "${ELF_FILE}" = "$i" ]; then
            if [ ! -f "$i" ]; then
                echo "Error: no such file: $i"
                exit 1
            fi
            ELF_FILE_LINE=$(get_line "$1" $i)
            if [[ "${ELF_FILE_LINE}" == *"?"* ]] || [ -z "${ELF_FILE_LINE}" ]; then
                echo "        * Error: you need ensure whether file: "$i" was compiled with -g or not! *"
                exit 1
            fi
            LINE=$(echo $1 | tr -d '\r')
            LINE=$(echo ${LINE} | awk '{print $1,$2}')
            echo "        "${LINE}" at "${ELF_FILE_LINE}
            FLAG=true
            break
        fi
    done
    if [[ "${FLAG}" == "false" ]]; then
        echo "        "$1
    fi
}

if [ $# -le 1 ]; then
    echo "Usage: ./out2line.sh text.txt elf1 elf2 ..."
    exit 1
fi

read -n5 -p "Compiler is [gcc/llvm]: " ANSWER
case "${ANSWER}" in
(gcc | GCC)
    which ${GCC_ADDR2LINE} >/dev/null 2>&1
    if [ $? -eq 0 ]; then
        ADDR2LINE=${GCC_ADDR2LINE}
    else
        echo "${GCC_ADDR2LINE} command not found!"
        exit 1
    fi;;
(llvm | LLVM)
    which ${LLVM_ADDR2LINE} >/dev/null 2>&1
    if [ $? -eq 0 ]; then
        ADDR2LINE=${LLVM_ADDR2LINE}
    else
        echo "${LLVM_ADDR2LINE} command not found! Trying to use ${GCC_ADDR2LINE}!"
        which ${GCC_ADDR2LINE} >/dev/null 2>&1
        if [ $? -eq 0 ]; then
            ADDR2LINE=${GCC_ADDR2LINE}
        else
            echo "${GCC_ADDR2LINE} command not found!"
            exit 1
        fi
    fi;;
(*)
    echo "Error choise!"
    exit 1
esac
echo -e "Now using ${ADDR2LINE} ...\n"

PARAM_LIST="${*:2}"
cat $1 | while read line; do
    HEAD_STRING=$(echo ${line} | awk '{print $1}')
    if [[ "${HEAD_STRING}" == *"#"* ]]; then
        parse_line "${line}" "${PARAM_LIST}"
    else
        if [[ "${HEAD_STRING}" == *"["* ]]; then
            echo "    "${line}
        else
            echo ${line}
        fi
    fi
done