mirror of
https://github.com/postgres/postgres.git
synced 2026-06-22 06:59:16 -04:00
dccffd9a27
If an allocation failed within LLVM it is not safe to call back into LLVM as LLVM is not generally safe against exceptions / stack-unwinding. Thus errors while in LLVM code are promoted to FATAL. However llvm_shutdown() did call back into LLVM even in such cases, while llvm_release_context() was careful not to do so. We cannot generally skip shutting down LLVM, as that can break profiling. But it's OK to do so if there was an error from within LLVM. Reported-By: Jelte Fennema <Jelte.Fennema@microsoft.com> Author: Andres Freund <andres@anarazel.de> Author: Justin Pryzby <pryzby@telsasoft.com> Discussion: https://postgr.es/m/AM5PR83MB0178C52CCA0A8DEA0207DC14F7FF9@AM5PR83MB0178.EURPRD83.prod.outlook.com Backpatch: 11-, where jit was introduced
144 lines
4.4 KiB
C
144 lines
4.4 KiB
C
/*-------------------------------------------------------------------------
|
|
* llvmjit.h
|
|
* LLVM JIT provider.
|
|
*
|
|
* Copyright (c) 2016-2018, PostgreSQL Global Development Group
|
|
*
|
|
* src/include/jit/llvmjit.h
|
|
*
|
|
*-------------------------------------------------------------------------
|
|
*/
|
|
#ifndef LLVMJIT_H
|
|
#define LLVMJIT_H
|
|
|
|
/*
|
|
* To avoid breaking cpluspluscheck, allow including the file even when LLVM
|
|
* is not available.
|
|
*/
|
|
#ifdef USE_LLVM
|
|
|
|
#include <llvm-c/Types.h>
|
|
|
|
|
|
/*
|
|
* File needs to be includable by both C and C++ code, and include other
|
|
* headers doing the same. Therefore wrap C portion in our own extern "C" if
|
|
* in C++ mode.
|
|
*/
|
|
#ifdef __cplusplus
|
|
extern "C"
|
|
{
|
|
#endif
|
|
|
|
|
|
#include "fmgr.h"
|
|
#include "jit/jit.h"
|
|
#include "nodes/pg_list.h"
|
|
#include "access/tupdesc.h"
|
|
|
|
|
|
typedef struct LLVMJitContext
|
|
{
|
|
JitContext base;
|
|
|
|
/* number of modules created */
|
|
size_t module_generation;
|
|
|
|
/* current, "open for write", module */
|
|
LLVMModuleRef module;
|
|
|
|
/* is there any pending code that needs to be emitted */
|
|
bool compiled;
|
|
|
|
/* # of objects emitted, used to generate non-conflicting names */
|
|
int counter;
|
|
|
|
/* list of handles for code emitted via Orc */
|
|
List *handles;
|
|
} LLVMJitContext;
|
|
|
|
|
|
/* type and struct definitions */
|
|
extern LLVMTypeRef TypeParamBool;
|
|
extern LLVMTypeRef TypePGFunction;
|
|
extern LLVMTypeRef TypeSizeT;
|
|
extern LLVMTypeRef TypeStorageBool;
|
|
|
|
extern LLVMTypeRef StructtupleDesc;
|
|
extern LLVMTypeRef StructHeapTupleData;
|
|
extern LLVMTypeRef StructTupleTableSlot;
|
|
extern LLVMTypeRef StructMemoryContextData;
|
|
extern LLVMTypeRef StructFunctionCallInfoData;
|
|
extern LLVMTypeRef StructExprContext;
|
|
extern LLVMTypeRef StructExprEvalStep;
|
|
extern LLVMTypeRef StructExprState;
|
|
extern LLVMTypeRef StructAggState;
|
|
extern LLVMTypeRef StructAggStatePerTransData;
|
|
extern LLVMTypeRef StructAggStatePerGroupData;
|
|
|
|
extern LLVMValueRef AttributeTemplate;
|
|
extern LLVMValueRef FuncStrlen;
|
|
extern LLVMValueRef FuncVarsizeAny;
|
|
extern LLVMValueRef FuncSlotGetsomeattrs;
|
|
extern LLVMValueRef FuncSlotGetmissingattrs;
|
|
extern LLVMValueRef FuncHeapGetsysattr;
|
|
extern LLVMValueRef FuncMakeExpandedObjectReadOnlyInternal;
|
|
extern LLVMValueRef FuncExecEvalArrayRefSubscript;
|
|
extern LLVMValueRef FuncExecAggTransReparent;
|
|
extern LLVMValueRef FuncExecAggInitGroup;
|
|
|
|
|
|
extern void llvm_enter_fatal_on_oom(void);
|
|
extern void llvm_leave_fatal_on_oom(void);
|
|
extern bool llvm_in_fatal_on_oom(void);
|
|
extern void llvm_reset_after_error(void);
|
|
extern void llvm_assert_in_fatal_section(void);
|
|
|
|
extern LLVMJitContext *llvm_create_context(int jitFlags);
|
|
extern LLVMModuleRef llvm_mutable_module(LLVMJitContext *context);
|
|
extern char *llvm_expand_funcname(LLVMJitContext *context, const char *basename);
|
|
extern void *llvm_get_function(LLVMJitContext *context, const char *funcname);
|
|
extern void llvm_split_symbol_name(const char *name, char **modname, char **funcname);
|
|
extern LLVMValueRef llvm_get_decl(LLVMModuleRef mod, LLVMValueRef f);
|
|
extern void llvm_copy_attributes(LLVMValueRef from, LLVMValueRef to);
|
|
extern LLVMValueRef llvm_function_reference(LLVMJitContext *context,
|
|
LLVMBuilderRef builder,
|
|
LLVMModuleRef mod,
|
|
FunctionCallInfo fcinfo);
|
|
|
|
extern void llvm_inline(LLVMModuleRef mod);
|
|
|
|
/*
|
|
****************************************************************************
|
|
* Code generation functions.
|
|
****************************************************************************
|
|
*/
|
|
extern bool llvm_compile_expr(struct ExprState *state);
|
|
extern LLVMValueRef slot_compile_deform(struct LLVMJitContext *context, TupleDesc desc, int natts);
|
|
|
|
/*
|
|
****************************************************************************
|
|
* Extensions / Backward compatibility section of the LLVM C API
|
|
* Error handling related functions.
|
|
****************************************************************************
|
|
*/
|
|
#if defined(HAVE_DECL_LLVMGETHOSTCPUNAME) && !HAVE_DECL_LLVMGETHOSTCPUNAME
|
|
/** Get the host CPU as a string. The result needs to be disposed with
|
|
LLVMDisposeMessage. */
|
|
extern char *LLVMGetHostCPUName(void);
|
|
#endif
|
|
|
|
#if defined(HAVE_DECL_LLVMGETHOSTCPUFEATURES) && !HAVE_DECL_LLVMGETHOSTCPUFEATURES
|
|
/** Get the host CPU features as a string. The result needs to be disposed
|
|
with LLVMDisposeMessage. */
|
|
extern char *LLVMGetHostCPUFeatures(void);
|
|
#endif
|
|
|
|
extern unsigned LLVMGetAttributeCountAtIndexPG(LLVMValueRef F, uint32 Idx);
|
|
|
|
#ifdef __cplusplus
|
|
} /* extern "C" */
|
|
#endif
|
|
|
|
#endif /* USE_LLVM */
|
|
#endif /* LLVMJIT_H */
|