Cache.h

Go to the documentation of this file.

#pragma once

#include <memory>
#include <unordered_map>

#include "preprocessor/llvm_includes_start.h"
#include <llvm/ExecutionEngine/ObjectCache.h>
#include <llvm/IR/LLVMContext.h>
#include "preprocessor/llvm_includes_end.h"

namespace llvm
{
        class ExecutionEngine;
}

namespace dev
{
namespace evmjit
{
class JITListener;

enum class CacheMode
{
        off,
        on,
        read,
        write,
        clear,
        preload
};

class ObjectCache : public llvm::ObjectCache
{
public:
        virtual void notifyObjectCompiled(llvm::Module const* _module, llvm::MemoryBufferRef _object) final override;

        virtual std::unique_ptr<llvm::MemoryBuffer> getObject(llvm::Module const* _module) final override;
};


class Cache
{
public:
        static ObjectCache* init(CacheMode _mode, JITListener* _listener);
        static std::unique_ptr<llvm::Module> getObject(std::string const& id, llvm::LLVMContext& _llvmContext);

        static void clear();

        static void preload(llvm::ExecutionEngine& _ee, std::unordered_map<std::string, uint64_t>& _funcCache,
                                                llvm::LLVMContext& _llvmContext);
};

}
}