1d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens// Copyright 2016 The SwiftShader Authors. All Rights Reserved.
2d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens//
3d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens// Licensed under the Apache License, Version 2.0 (the "License");
4d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens// you may not use this file except in compliance with the License.
5d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens// You may obtain a copy of the License at
6d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens//
7d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens//    http://www.apache.org/licenses/LICENSE-2.0
8d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens//
9d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens// Unless required by applicable law or agreed to in writing, software
10d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens// distributed under the License is distributed on an "AS IS" BASIS,
11d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens// See the License for the specific language governing permissions and
13d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens// limitations under the License.
14d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens
15bc19941f85e8cfc75a860b8bd61789e6064e7f12Nicolas Capens#ifndef sw_LLVMRoutineManager_hpp
16bc19941f85e8cfc75a860b8bd61789e6064e7f12Nicolas Capens#define sw_LLVMRoutineManager_hpp
17d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens
18d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens#include "llvm/GlobalValue.h"
19d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens#include "llvm/ExecutionEngine/JITMemoryManager.h"
20d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens
21d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capensnamespace sw
22d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens{
23bc19941f85e8cfc75a860b8bd61789e6064e7f12Nicolas Capens	class LLVMRoutine;
24d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens
25bc19941f85e8cfc75a860b8bd61789e6064e7f12Nicolas Capens	class LLVMRoutineManager : public llvm::JITMemoryManager
26d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens	{
27d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens	public:
28bc19941f85e8cfc75a860b8bd61789e6064e7f12Nicolas Capens		LLVMRoutineManager();
29d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens
30bc19941f85e8cfc75a860b8bd61789e6064e7f12Nicolas Capens		virtual ~LLVMRoutineManager();
31d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens
32d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens		virtual void AllocateGOT();
33d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens
34d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens		virtual uint8_t *allocateStub(const llvm::GlobalValue *function, unsigned stubSize, unsigned alignment);
35d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens		virtual uint8_t *startFunctionBody(const llvm::Function *function, uintptr_t &actualSize);
36d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens		virtual void endFunctionBody(const llvm::Function *function, uint8_t *functionStart, uint8_t *functionEnd);
37d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens		virtual uint8_t *startExceptionTable(const llvm::Function *function, uintptr_t &ActualSize);
38d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens		virtual void endExceptionTable(const llvm::Function *function, uint8_t *tableStart, uint8_t *tableEnd, uint8_t *frameRegister);
39d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens		virtual uint8_t *getGOTBase() const;
40d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens		virtual uint8_t *allocateSpace(intptr_t Size, unsigned Alignment);
41d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens		virtual uint8_t *allocateGlobal(uintptr_t Size, unsigned int Alignment);
42d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens		virtual void deallocateFunctionBody(void *Body);
43d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens		virtual void deallocateExceptionTable(void *ET);
44d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens		virtual void setMemoryWritable();
45d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens		virtual void setMemoryExecutable();
46d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens		virtual void setPoisonMemory(bool poison);
47d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens
48bc19941f85e8cfc75a860b8bd61789e6064e7f12Nicolas Capens		LLVMRoutine *acquireRoutine(void *entry);
49d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens
50d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens	private:
51bc19941f85e8cfc75a860b8bd61789e6064e7f12Nicolas Capens		LLVMRoutine *routine;
52d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens
53d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens		static volatile int averageInstructionSize;
54d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens	};
55d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens}
56d999309b36cb3dceadd38217b322f0e96a06b202Nicolas Capens
57bc19941f85e8cfc75a860b8bd61789e6064e7f12Nicolas Capens#endif   // sw_LLVMRoutineManager_hpp
58