72c29e83d5
Before this patch the function was using RZ_PREFIX even when IS_PORTABLE was set, making the function not working well because RZ_PREFIX might not exist in the system where the portable build is used (the path was passed to realpath(), which requires an existing path). This patch fixes the problem by searching for the BINDIR without using realpath() on the RZ_PREFIX. Moreover, it avoid retrieving the same prefix every time by saving it in a global variable. * Add rz_constructor.h to allow defining constructors/destructors on multiple compilers.
139 lines
4.4 KiB
C
139 lines
4.4 KiB
C
// SPDX-License-Identifier: LGPL-2.1-or-later
|
|
// SPDX-FileCopyrightText: 1995-1997 Peter Mattis, Spencer Kimball and Josh MacDonald
|
|
|
|
// Taken from https://github.com/GNOME/glib/blob/main/glib/gconstructor.h and adapted to Rizin
|
|
|
|
#ifndef __RZ_CONSTRUCTOR_H__
|
|
#define __RZ_CONSTRUCTOR_H__
|
|
|
|
/*
|
|
If RZ_HAS_CONSTRUCTORS is true then the compiler support *both* constructors and
|
|
destructors, in a usable way, including e.g. on library unload. If not you're on
|
|
your own.
|
|
Some compilers need #pragma to handle this, which does not work with macros,
|
|
so the way you need to use this is (for constructors):
|
|
#ifdef RZ_DEFINE_CONSTRUCTOR_NEEDS_PRAGMA
|
|
#pragma RZ_DEFINE_CONSTRUCTOR_PRAGMA_ARGS(my_constructor)
|
|
#endif
|
|
RZ_DEFINE_CONSTRUCTOR(my_constructor)
|
|
static void my_constructor(void) {
|
|
...
|
|
}
|
|
*/
|
|
|
|
#if __GNUC__ > 2 || (__GNUC__ == 2 && __GNUC_MINOR__ >= 7)
|
|
|
|
#define RZ_HAS_CONSTRUCTORS 1
|
|
|
|
#define RZ_DEFINE_CONSTRUCTOR(_func) static void __attribute__((constructor)) _func(void);
|
|
#define RZ_DEFINE_DESTRUCTOR(_func) static void __attribute__((destructor)) _func(void);
|
|
|
|
#elif defined(_MSC_VER) && (_MSC_VER >= 1500)
|
|
/* Visual studio 2008 and later has _Pragma */
|
|
|
|
#include <stdlib.h>
|
|
|
|
#define RZ_HAS_CONSTRUCTORS 1
|
|
|
|
/* We do some weird things to avoid the constructors being optimized
|
|
* away on VS2015 if WholeProgramOptimization is enabled. First we
|
|
* make a reference to the array from the wrapper to make sure its
|
|
* references. Then we use a pragma to make sure the wrapper function
|
|
* symbol is always included at the link stage. Also, the symbols
|
|
* need to be extern (but not dllexport), even though they are not
|
|
* really used from another object file.
|
|
*/
|
|
|
|
/* We need to account for differences between the mangling of symbols
|
|
* for x86 and x64/ARM/ARM64 programs, as symbols on x86 are prefixed
|
|
* with an underscore but symbols on x64/ARM/ARM64 are not.
|
|
*/
|
|
#ifdef _M_IX86
|
|
#define RZ_MSVC_SYMBOL_PREFIX "_"
|
|
#else
|
|
#define RZ_MSVC_SYMBOL_PREFIX ""
|
|
#endif
|
|
|
|
#define RZ_DEFINE_CONSTRUCTOR(_func) RZ_MSVC_CTOR(_func, RZ_MSVC_SYMBOL_PREFIX)
|
|
#define RZ_DEFINE_DESTRUCTOR(_func) RZ_MSVC_DTOR(_func, RZ_MSVC_SYMBOL_PREFIX)
|
|
|
|
#define RZ_MSVC_CTOR(_func, _sym_prefix) \
|
|
static void _func(void); \
|
|
extern int (*_array##_func)(void); \
|
|
int _func##_wrapper(void) { \
|
|
_func(); \
|
|
char *_func##_var = rz_str_new(""); \
|
|
free(_func##_var); \
|
|
return 0; \
|
|
} \
|
|
__pragma(comment(linker, "/include:" _sym_prefix #_func "_wrapper")) \
|
|
__pragma(section(".CRT$XCU", read)) __declspec(allocate(".CRT$XCU")) int (*_array##_func)(void) = _func##_wrapper;
|
|
|
|
#define RZ_MSVC_DTOR(_func, _sym_prefix) \
|
|
static void _func(void); \
|
|
extern int (*_array##_func)(void); \
|
|
int _func##_constructor(void) { \
|
|
atexit(_func); \
|
|
char *_func##_var = rz_str_new(""); \
|
|
free(_func##_var); \
|
|
return 0; \
|
|
} \
|
|
__pragma(comment(linker, "/include:" _sym_prefix #_func "_constructor")) \
|
|
__pragma(section(".CRT$XCU", read)) __declspec(allocate(".CRT$XCU")) int (*_array##_func)(void) = _func##_constructor;
|
|
|
|
#elif defined(_MSC_VER)
|
|
|
|
#define RZ_HAS_CONSTRUCTORS 1
|
|
|
|
/* Pre Visual studio 2008 must use #pragma section */
|
|
#define RZ_DEFINE_CONSTRUCTOR_NEEDS_PRAGMA 1
|
|
#define RZ_DEFINE_DESTRUCTOR_NEEDS_PRAGMA 1
|
|
|
|
#define RZ_DEFINE_CONSTRUCTOR_PRAGMA_ARGS(_func) \
|
|
section(".CRT$XCU", read)
|
|
#define RZ_DEFINE_CONSTRUCTOR(_func) \
|
|
static void _func(void); \
|
|
static int _func##_wrapper(void) { \
|
|
_func(); \
|
|
return 0; \
|
|
} \
|
|
__declspec(allocate(".CRT$XCU")) static int (*p)(void) = _func##_wrapper;
|
|
|
|
#define RZ_DEFINE_DESTRUCTOR_PRAGMA_ARGS(_func) \
|
|
section(".CRT$XCU", read)
|
|
#define RZ_DEFINE_DESTRUCTOR(_func) \
|
|
static void _func(void); \
|
|
static int _func##_constructor(void) { \
|
|
atexit(_func); \
|
|
return 0; \
|
|
} \
|
|
__declspec(allocate(".CRT$XCU")) static int (*_array##_func)(void) = _func##_constructor;
|
|
|
|
#elif defined(__SUNPRO_C)
|
|
|
|
/* This is not tested, but i believe it should work, based on:
|
|
* http://opensource.apple.com/source/OpenSSL098/OpenSSL098-35/src/fips/fips_premain.c
|
|
*/
|
|
|
|
#define RZ_HAS_CONSTRUCTORS 1
|
|
|
|
#define RZ_DEFINE_CONSTRUCTOR_NEEDS_PRAGMA 1
|
|
#define RZ_DEFINE_DESTRUCTOR_NEEDS_PRAGMA 1
|
|
|
|
#define RZ_DEFINE_CONSTRUCTOR_PRAGMA_ARGS(_func) \
|
|
init(_func)
|
|
#define RZ_DEFINE_CONSTRUCTOR(_func) \
|
|
static void _func(void);
|
|
|
|
#define RZ_DEFINE_DESTRUCTOR_PRAGMA_ARGS(_func) \
|
|
fini(_func)
|
|
#define RZ_DEFINE_DESTRUCTOR(_func) \
|
|
static void _func(void);
|
|
|
|
#else
|
|
|
|
/* constructors not supported for this compiler */
|
|
|
|
#endif
|
|
|
|
#endif /* __RZ_CONSTRUCTOR_H__ */
|