GasMeter.h

Go to the documentation of this file.

#pragma once

#include <evm.h>
#include "CompilerHelper.h"
#include "Instruction.h"

namespace dev
{
namespace eth
{
namespace jit
{
class RuntimeManager;
using namespace evmjit;

class GasMeter : public CompilerHelper // TODO: Use RuntimeHelper
{
public:
        GasMeter(IRBuilder& _builder, RuntimeManager& _runtimeManager, evm_mode mode);

        void count(Instruction _inst);

        void count(llvm::Value* _cost, llvm::Value* _jmpBuf = nullptr, llvm::Value* _gasPtr = nullptr);

        void countSStore(class Ext& _ext, llvm::Value* _index, llvm::Value* _newValue);

        void countExp(llvm::Value* _exponent);

        void countLogData(llvm::Value* _dataLength);

        void countSha3Data(llvm::Value* _dataLength);

        void commitCostBlock();

        void giveBack(llvm::Value* _gas);

        void countMemory(llvm::Value* _additionalMemoryInWords, llvm::Value* _jmpBuf, llvm::Value* _gasPtr);

        void countCopy(llvm::Value* _copyWords);

private:
        int64_t getStepCost(Instruction inst) const;

        int64_t m_blockCost = 0;

        llvm::CallInst* m_checkCall = nullptr;
        llvm::Function* m_gasCheckFunc = nullptr;

        RuntimeManager& m_runtimeManager;

        evm_mode m_mode;
};

}
}
}