# Copyright 2017 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

import functools
import logging
import os
import subprocess
import threading

_CHROME_SRC = os.path.join(os.path.dirname(__file__), os.pardir, os.pardir)
_LLVM_SYMBOLIZER_PATH = os.path.join(
    _CHROME_SRC,
    'third_party',
    'llvm-build',
    'Release+Asserts',
    'bin',
    'llvm-symbolizer',
)

_UNKNOWN = '<UNKNOWN>'

_ELF_MAGIC_HEADER_BYTES = b'\x7f\x45\x4c\x46'


@functools.lru_cache
def IsValidLLVMSymbolizerTarget(file_path):
    """Verify the passed file is a valid target for llvm-symbolization

    Args:
      file_path: Path to a file to be checked

    Return:
      True if the file exists and has the correct ELF header, False otherwise
    """
    try:
        with open(file_path, 'rb') as f:
            header_bytes = f.read(4)
            return header_bytes == _ELF_MAGIC_HEADER_BYTES
    except IOError:
        return False


class LLVMSymbolizer(object):
    def __init__(self):
        """Create a LLVMSymbolizer instance that interacts with the llvm symbolizer.

        The purpose of the LLVMSymbolizer is to get function names and line
        numbers of an address from the symbols library.
        """
        self._llvm_symbolizer_subprocess = None
        self._llvm_symbolizer_parameters = [
            '--functions',
            '--demangle',
            '--inlines',
        ]

        # Allow only one thread to call GetSymbolInformation at a time.
        self._lock = threading.Lock()

    def Start(self):
        """Start the llvm symbolizer subprocess.

        Create a subprocess of the llvm symbolizer executable, which will be used
        to retrieve function names etc.
        """
        if os.path.isfile(_LLVM_SYMBOLIZER_PATH):
            self._llvm_symbolizer_subprocess = subprocess.Popen(
                [_LLVM_SYMBOLIZER_PATH] + self._llvm_symbolizer_parameters,
                stdout=subprocess.PIPE,
                stdin=subprocess.PIPE,
                universal_newlines=True,
            )
        else:
            logging.error(
                'Cannot find llvm_symbolizer here: %s.' % _LLVM_SYMBOLIZER_PATH
            )
            self._llvm_symbolizer_subprocess = None

    def Close(self):
        """Close the llvm symbolizer subprocess.

        Close the subprocess by closing stdin, stdout and killing the subprocess.
        """
        with self._lock:
            if self._llvm_symbolizer_subprocess:
                self._llvm_symbolizer_subprocess.kill()
                self._llvm_symbolizer_subprocess = None

    def __enter__(self):
        """Start the llvm symbolizer subprocess."""
        self.Start()
        return self

    def __exit__(self, exc_type, exc_val, exc_tb):
        """Close the llvm symbolizer subprocess."""
        self.Close()

    def GetSymbolInformation(self, lib, addr):
        """Return the corresponding function names and line numbers.

        Args:
          lib: library to search for info.
          addr: address to look for info.

        Returns:
          A triplet of address, module-name and list of symbols
        """
        if self._llvm_symbolizer_subprocess is None:
            logging.error(
                'Can\'t run llvm-symbolizer! '
                + 'Subprocess for llvm-symbolizer has not been started!'
            )
            return [(_UNKNOWN, lib)]

        if not lib:
            logging.error('Can\'t run llvm-symbolizer! No target is given!')
            return [(_UNKNOWN, lib)]

        if not IsValidLLVMSymbolizerTarget(lib):
            logging.error(
                'Can\'t run llvm-symbolizer! '
                + 'Given binary is not a valid target. path=%s',
                lib,
            )
            return [(_UNKNOWN, lib)]

        proc = self._llvm_symbolizer_subprocess
        with self._lock:
            proc.stdin.write('%s %s\n' % (lib, hex(addr)))
            proc.stdin.flush()
            result = []
            # Read until an empty line is observed, which indicates the end of the
            # output. Each line with a function name is always followed by one line
            # with the corresponding line number.
            while True:
                line = proc.stdout.readline()
                if line != '\n':
                    line_numbers = proc.stdout.readline()
                    result.append((line[:-1], line_numbers[:-1]))
                else:
                    return result

    @staticmethod
    def IsValidTarget(path):
        return IsValidLLVMSymbolizerTarget(path)