#pragma once

#ifdef __ELF__

#include <vector>
#include <string>
#include <ext/singleton.h>
#include <Common/Elf.h>


namespace DB
{

/** Allow to quickly find symbol name from address.
  * Used as a replacement for "dladdr" function which is extremely slow.
  * It works better than "dladdr" because it also allows to search private symbols, that are not participated in shared linking.
  */
class SymbolIndex : public ext::singleton<SymbolIndex>
{
protected:
    friend class ext::singleton<SymbolIndex>;
    SymbolIndex() { update(); }

public:
    struct Symbol
    {
        const void * address_begin;
        const void * address_end;
        const char * name;
    };

    struct Object
    {
        const void * address_begin;
        const void * address_end;
        std::string name;
        std::unique_ptr<Elf> elf;
    };

    const Symbol * findSymbol(const void * address) const;
    const Object * findObject(const void * address) const;

    const std::vector<Symbol> & symbols() const { return data.symbols; }
    const std::vector<Object> & objects() const { return data.objects; }

    struct Data
    {
        std::vector<Symbol> symbols;
        std::vector<Object> objects;
    };
private:
    Data data;

    void update();
};

}

#endif