| /** | |
| * The runtime module exposes information specific to the D runtime code. | |
| * | |
| * Copyright: Copyright Sean Kelly 2005 - 2009. | |
| * License: $(LINK2 http://www.boost.org/LICENSE_1_0.txt, Boost License 1.0) | |
| * Authors: Sean Kelly | |
| * Source: $(DRUNTIMESRC core/_runtime.d) | |
| * Documentation: https://dlang.org/phobos/core_runtime.html | |
| */ | |
| /* NOTE: This file has been patched from the original DMD distribution to | |
| * work with the GDC compiler. | |
| */ | |
| module core.runtime; | |
| version (OSX) | |
| version = Darwin; | |
| else version (iOS) | |
| version = Darwin; | |
| else version (TVOS) | |
| version = Darwin; | |
| else version (WatchOS) | |
| version = Darwin; | |
| version (GNU) | |
| { | |
| import gcc.backtrace; | |
| // This shouldn't be necessary but ensure that code doesn't get mixed | |
| // It does however prevent the unittest SEGV handler to be installed, | |
| // which is desireable as it uses backtrace directly. | |
| private enum hasExecinfo = false; | |
| version = LibBacktrace_TraceHandler; | |
| } | |
| else version (DRuntime_Use_Libunwind) | |
| { | |
| version = DefineBacktrace_using_UnwindBacktrace; | |
| // This shouldn't be necessary but ensure that code doesn't get mixed | |
| // It does however prevent the unittest SEGV handler to be installed, | |
| // which is desireable as it uses backtrace directly. | |
| private enum hasExecinfo = false; | |
| } | |
| else | |
| import core.internal.execinfo; | |
| /// C interface for Runtime.loadLibrary | |
| extern (C) void* rt_loadLibrary(const char* name); | |
| /// ditto | |
| version (Windows) extern (C) void* rt_loadLibraryW(const wchar* name); | |
| /// C interface for Runtime.unloadLibrary, returns 1/0 instead of bool | |
| extern (C) int rt_unloadLibrary(void* ptr); | |
| /// C interface for Runtime.initialize, returns 1/0 instead of bool | |
| extern(C) int rt_init(); | |
| /// C interface for Runtime.terminate, returns 1/0 instead of bool | |
| extern(C) int rt_term(); | |
| /** | |
| * This type is returned by the module unit test handler to indicate testing | |
| * results. | |
| */ | |
| struct UnitTestResult | |
| { | |
| /** | |
| * Number of modules which were tested | |
| */ | |
| size_t executed; | |
| /** | |
| * Number of modules passed the unittests | |
| */ | |
| size_t passed; | |
| /** | |
| * Should the main function be run or not? This is ignored if any tests | |
| * failed. | |
| */ | |
| bool runMain; | |
| /** | |
| * Should we print a summary of the results? | |
| */ | |
| bool summarize; | |
| /** | |
| * Simple check for whether execution should continue after unit tests | |
| * have been run. Works with legacy code that expected a bool return. | |
| * | |
| * Returns: | |
| * true if execution should continue after testing is complete, false if | |
| * not. | |
| */ | |
| bool opCast(T : bool)() const | |
| { | |
| return runMain && (executed == passed); | |
| } | |
| /// Simple return code that says unit tests pass, and main should be run | |
| enum UnitTestResult pass = UnitTestResult(0, 0, true, false); | |
| /// Simple return code that says unit tests failed. | |
| enum UnitTestResult fail = UnitTestResult(1, 0, false, false); | |
| } | |
| /// Legacy module unit test handler | |
| alias ModuleUnitTester = bool function(); | |
| /// Module unit test handler | |
| alias ExtendedModuleUnitTester = UnitTestResult function(); | |
| private | |
| { | |
| alias CollectHandler = bool function(Object); | |
| alias TraceHandler = Throwable.TraceInfo function(void* ptr); | |
| alias ExceptionHandler = void delegate(Throwable); | |
| extern (C) void _d_print_throwable(Throwable t); | |
| extern (C) void* thread_stackBottom() nothrow @nogc; | |
| } | |
| shared static this() | |
| { | |
| // NOTE: Some module ctors will run before this handler is set, so it's | |
| // still possible the app could exit without a stack trace. If | |
| // this becomes an issue, the handler could be set in C main | |
| // before the module ctors are run. | |
| Runtime.traceHandler(&defaultTraceHandler, &defaultTraceDeallocator); | |
| } | |
| /////////////////////////////////////////////////////////////////////////////// | |
| // Runtime | |
| /////////////////////////////////////////////////////////////////////////////// | |
| /** | |
| * Stores the unprocessed arguments supplied when the | |
| * process was started. | |
| */ | |
| struct CArgs | |
| { | |
| int argc; /// The argument count. | |
| char** argv; /// The arguments as a C array of strings. | |
| } | |
| /** | |
| * This struct encapsulates all functionality related to the underlying runtime | |
| * module for the calling context. | |
| */ | |
| struct Runtime | |
| { | |
| /** | |
| * Initializes the runtime. This call is to be used in instances where the | |
| * standard program initialization process is not executed. This is most | |
| * often in shared libraries or in libraries linked to a C program. | |
| * If the runtime was already successfully initialized this returns true. | |
| * Each call to initialize must be paired by a call to $(LREF terminate). | |
| * | |
| * Returns: | |
| * true if initialization succeeded or false if initialization failed. | |
| */ | |
| static bool initialize() | |
| { | |
| return !!rt_init(); | |
| } | |
| /** | |
| * Terminates the runtime. This call is to be used in instances where the | |
| * standard program termination process will not be not executed. This is | |
| * most often in shared libraries or in libraries linked to a C program. | |
| * If the runtime was not successfully initialized the function returns false. | |
| * | |
| * Returns: | |
| * true if termination succeeded or false if termination failed. | |
| */ | |
| static bool terminate() | |
| { | |
| return !!rt_term(); | |
| } | |
| /** | |
| * Returns the arguments supplied when the process was started. | |
| * | |
| * Returns: | |
| * The arguments supplied when this process was started. | |
| */ | |
| extern(C) pragma(mangle, "rt_args") static @property string[] args(); | |
| /** | |
| * Returns the unprocessed C arguments supplied when the process was started. | |
| * Use this when you need to supply argc and argv to C libraries. | |
| * | |
| * Returns: | |
| * A $(LREF CArgs) struct with the arguments supplied when this process was started. | |
| * | |
| * Example: | |
| * --- | |
| * import core.runtime; | |
| * | |
| * // A C library function requiring char** arguments | |
| * extern(C) void initLibFoo(int argc, char** argv); | |
| * | |
| * void main() | |
| * { | |
| * auto args = Runtime.cArgs; | |
| * initLibFoo(args.argc, args.argv); | |
| * } | |
| * --- | |
| */ | |
| extern(C) pragma(mangle, "rt_cArgs") static @property CArgs cArgs() @nogc; | |
| /** | |
| * Locates a dynamic library with the supplied library name and dynamically | |
| * loads it into the caller's address space. If the library contains a D | |
| * runtime it will be integrated with the current runtime. | |
| * | |
| * Params: | |
| * name = The name of the dynamic library to load. | |
| * | |
| * Returns: | |
| * A reference to the library or null on error. | |
| */ | |
| static void* loadLibrary()(const scope char[] name) | |
| { | |
| import core.stdc.stdlib : free, malloc; | |
| version (Windows) | |
| { | |
| import core.sys.windows.winnls : CP_UTF8, MultiByteToWideChar; | |
| import core.sys.windows.winnt : WCHAR; | |
| if (name.length == 0) return null; | |
| // Load a DLL at runtime | |
| auto len = MultiByteToWideChar( | |
| CP_UTF8, 0, name.ptr, cast(int)name.length, null, 0); | |
| if (len == 0) | |
| return null; | |
| auto buf = cast(WCHAR*)malloc((len+1) * WCHAR.sizeof); | |
| if (buf is null) return null; | |
| scope (exit) free(buf); | |
| len = MultiByteToWideChar( | |
| CP_UTF8, 0, name.ptr, cast(int)name.length, buf, len); | |
| if (len == 0) | |
| return null; | |
| buf[len] = '\0'; | |
| return rt_loadLibraryW(buf); | |
| } | |
| else version (Posix) | |
| { | |
| /* Need a 0-terminated C string for the dll name | |
| */ | |
| immutable len = name.length; | |
| auto buf = cast(char*)malloc(len + 1); | |
| if (!buf) return null; | |
| scope (exit) free(buf); | |
| buf[0 .. len] = name[]; | |
| buf[len] = 0; | |
| return rt_loadLibrary(buf); | |
| } | |
| } | |
| /** | |
| * Unloads the dynamic library referenced by p. If this library contains a | |
| * D runtime then any necessary finalization or cleanup of that runtime | |
| * will be performed. | |
| * | |
| * Params: | |
| * p = A reference to the library to unload. | |
| */ | |
| static bool unloadLibrary()(void* p) | |
| { | |
| return !!rt_unloadLibrary(p); | |
| } | |
| /** | |
| * Overrides the default trace mechanism with a user-supplied version. A | |
| * trace represents the context from which an exception was thrown, and the | |
| * trace handler will be called when this occurs. The pointer supplied to | |
| * this routine indicates the base address from which tracing should occur. | |
| * If the supplied pointer is null then the trace routine should determine | |
| * an appropriate calling context from which to begin the trace. | |
| * | |
| * If the deallocator is set, then it is called with the traceinfo when the | |
| * exception is finalized. The deallocator is only set in the exception if | |
| * the default handler is used to generate the trace info. | |
| * | |
| * Params: | |
| * h = The new trace handler. Set to null to disable exception backtracing. | |
| * d = The new trace deallocator. If non-null, this will be called on | |
| * exception destruction with the trace info, only when the trace | |
| * handler is used to generate TraceInfo. | |
| */ | |
| extern(C) pragma(mangle, "rt_setTraceHandler") static @property void traceHandler(TraceHandler h, | |
| Throwable.TraceDeallocator d = null); | |
| /** | |
| * Gets the current trace handler. | |
| * | |
| * Returns: | |
| * The current trace handler or null if none has been set. | |
| */ | |
| extern(C) pragma(mangle, "rt_getTraceHandler") static @property TraceHandler traceHandler(); | |
| /** | |
| * Gets the current trace deallocator. | |
| * | |
| * Returns: | |
| * The current trace deallocator or null if none has been set. | |
| */ | |
| extern(C) pragma(mangle, "rt_getTraceDeallocator") static @property Throwable.TraceDeallocator traceDeallocator(); | |
| /** | |
| * Overrides the default collect hander with a user-supplied version. This | |
| * routine will be called for each resource object that is finalized in a | |
| * non-deterministic manner--typically during a garbage collection cycle. | |
| * If the supplied routine returns true then the object's dtor will called | |
| * as normal, but if the routine returns false than the dtor will not be | |
| * called. The default behavior is for all object dtors to be called. | |
| * | |
| * Params: | |
| * h = The new collect handler. Set to null to use the default handler. | |
| */ | |
| extern(C) pragma(mangle, "rt_setCollectHandler") static @property void collectHandler( CollectHandler h ); | |
| /** | |
| * Gets the current collect handler. | |
| * | |
| * Returns: | |
| * The current collect handler or null if none has been set. | |
| */ | |
| extern(C) pragma(mangle, "rt_getCollectHandler") static @property CollectHandler collectHandler(); | |
| /** | |
| * Overrides the default module unit tester with a user-supplied version. | |
| * This routine will be called once on program initialization. The return | |
| * value of this routine indicates to the runtime whether the tests ran | |
| * without error. | |
| * | |
| * There are two options for handlers. The `bool` version is deprecated but | |
| * will be kept for legacy support. Returning `true` from the handler is | |
| * equivalent to returning `UnitTestResult.pass` from the extended version. | |
| * Returning `false` from the handler is equivalent to returning | |
| * `UnitTestResult.fail` from the extended version. | |
| * | |
| * See the documentation for `UnitTestResult` to see how you should set up | |
| * the return structure. | |
| * | |
| * See the documentation for `runModuleUnitTests` for how the default | |
| * algorithm works, or read the example below. | |
| * | |
| * Params: | |
| * h = The new unit tester. Set both to null to use the default unit | |
| * tester. | |
| * | |
| * Example: | |
| * --------- | |
| * shared static this() | |
| * { | |
| * import core.runtime; | |
| * | |
| * Runtime.extendedModuleUnitTester = &customModuleUnitTester; | |
| * } | |
| * | |
| * UnitTestResult customModuleUnitTester() | |
| * { | |
| * import std.stdio; | |
| * | |
| * writeln("Using customModuleUnitTester"); | |
| * | |
| * // Do the same thing as the default moduleUnitTester: | |
| * UnitTestResult result; | |
| * foreach (m; ModuleInfo) | |
| * { | |
| * if (m) | |
| * { | |
| * auto fp = m.unitTest; | |
| * | |
| * if (fp) | |
| * { | |
| * ++result.executed; | |
| * try | |
| * { | |
| * fp(); | |
| * ++result.passed; | |
| * } | |
| * catch (Throwable e) | |
| * { | |
| * writeln(e); | |
| * } | |
| * } | |
| * } | |
| * } | |
| * if (result.executed != result.passed) | |
| * { | |
| * result.runMain = false; // don't run main | |
| * result.summarize = true; // print failure | |
| * } | |
| * else | |
| * { | |
| * result.runMain = true; // all UT passed | |
| * result.summarize = false; // be quiet about it. | |
| * } | |
| * return result; | |
| * } | |
| * --------- | |
| */ | |
| static @property void extendedModuleUnitTester( ExtendedModuleUnitTester h ) | |
| { | |
| sm_extModuleUnitTester = h; | |
| } | |
| /// Ditto | |
| static @property void moduleUnitTester( ModuleUnitTester h ) | |
| { | |
| sm_moduleUnitTester = h; | |
| } | |
| /** | |
| * Gets the current legacy module unit tester. | |
| * | |
| * This property should not be used, but is supported for legacy purposes. | |
| * | |
| * Note that if the extended unit test handler is set, this handler will | |
| * be ignored. | |
| * | |
| * Returns: | |
| * The current legacy module unit tester handler or null if none has been | |
| * set. | |
| */ | |
| static @property ModuleUnitTester moduleUnitTester() | |
| { | |
| return sm_moduleUnitTester; | |
| } | |
| /** | |
| * Gets the current module unit tester. | |
| * | |
| * This handler overrides any legacy module unit tester set by the | |
| * moduleUnitTester property. | |
| * | |
| * Returns: | |
| * The current module unit tester handler or null if none has been | |
| * set. | |
| */ | |
| static @property ExtendedModuleUnitTester extendedModuleUnitTester() | |
| { | |
| return sm_extModuleUnitTester; | |
| } | |
| private: | |
| // NOTE: This field will only ever be set in a static ctor and should | |
| // never occur within any but the main thread, so it is safe to | |
| // make it __gshared. | |
| __gshared ExtendedModuleUnitTester sm_extModuleUnitTester = null; | |
| __gshared ModuleUnitTester sm_moduleUnitTester = null; | |
| } | |
| /** | |
| * Set source file path for coverage reports. | |
| * | |
| * Params: | |
| * path = The new path name. | |
| * Note: | |
| * This is a dmd specific setting. | |
| */ | |
| extern (C) void dmd_coverSourcePath(string path); | |
| /** | |
| * Set output path for coverage reports. | |
| * | |
| * Params: | |
| * path = The new path name. | |
| * Note: | |
| * This is a dmd specific setting. | |
| */ | |
| extern (C) void dmd_coverDestPath(string path); | |
| /** | |
| * Enable merging of coverage reports with existing data. | |
| * | |
| * Params: | |
| * flag = enable/disable coverage merge mode | |
| * Note: | |
| * This is a dmd specific setting. | |
| */ | |
| extern (C) void dmd_coverSetMerge(bool flag); | |
| /** | |
| * Set the output file name for profile reports (-profile switch). | |
| * An empty name will set the output to stdout. | |
| * | |
| * Params: | |
| * name = file name | |
| * Note: | |
| * This is a dmd specific setting. | |
| */ | |
| extern (C) void trace_setlogfilename(string name); | |
| /** | |
| * Set the output file name for the optimized profile linker DEF file (-profile switch). | |
| * An empty name will set the output to stdout. | |
| * | |
| * Params: | |
| * name = file name | |
| * Note: | |
| * This is a dmd specific setting. | |
| */ | |
| extern (C) void trace_setdeffilename(string name); | |
| /** | |
| * Set the output file name for memory profile reports (-profile=gc switch). | |
| * An empty name will set the output to stdout. | |
| * | |
| * Params: | |
| * name = file name | |
| * Note: | |
| * This is a dmd specific setting. | |
| */ | |
| extern (C) void profilegc_setlogfilename(string name); | |
| /////////////////////////////////////////////////////////////////////////////// | |
| // Overridable Callbacks | |
| /////////////////////////////////////////////////////////////////////////////// | |
| /** | |
| * This routine is called by the runtime to run module unit tests on startup. | |
| * The user-supplied unit tester will be called if one has been set, | |
| * otherwise all unit tests will be run in sequence. | |
| * | |
| * If the extended unittest handler is registered, this function returns the | |
| * result from that handler directly. | |
| * | |
| * If a legacy boolean returning custom handler is used, `false` maps to | |
| * `UnitTestResult.fail`, and `true` maps to `UnitTestResult.pass`. This was | |
| * the original behavior of the unit testing system. | |
| * | |
| * If no unittest custom handlers are registered, the following algorithm is | |
| * executed (the behavior can be affected by the `--DRT-testmode` switch | |
| * below): | |
| * 1. Execute any unittests present. For each that fails, print the stack | |
| * trace and continue. | |
| * 2. If no unittests were present, set summarize to false, and runMain to | |
| * true. | |
| * 3. Otherwise, set summarize to true, and runMain to false. | |
| * | |
| * See the documentation for `UnitTestResult` for details on how the runtime | |
| * treats the return value from this function. | |
| * | |
| * If the switch `--DRT-testmode` is passed to the executable, it can have | |
| * one of 3 values: | |
| * 1. "run-main": even if unit tests are run (and all pass), runMain is set | |
| to true. | |
| * 2. "test-or-main": any unit tests present will cause the program to | |
| * summarize the results and exit regardless of the result. This is the | |
| * default. | |
| * 3. "test-only", runMain is set to false, even with no tests present. | |
| * | |
| * This command-line parameter does not affect custom unit test handlers. | |
| * | |
| * Returns: | |
| * A `UnitTestResult` struct indicating the result of running unit tests. | |
| */ | |
| extern (C) UnitTestResult runModuleUnitTests() | |
| { | |
| version (Windows) | |
| import core.sys.windows.stacktrace; | |
| static if (__traits(compiles, new LibBacktrace(0))) | |
| { | |
| // segv handler | |
| import core.sys.posix.signal : SA_RESETHAND, SA_SIGINFO, sigaction, sigaction_t, SIGBUS, sigfillset, siginfo_t, | |
| SIGSEGV; | |
| static extern (C) void unittestSegvHandler(int signum, siginfo_t* info, void* ptr) | |
| { | |
| import core.stdc.stdio; | |
| fprintf(stderr, "Segmentation fault while running unittests:\n"); | |
| fprintf(stderr, "----------------\n"); | |
| // First frame is LibBacktrace ctor. Second is signal handler, | |
| // but include that for now | |
| scope bt = new LibBacktrace(1); | |
| foreach (size_t i, const(char[]) msg; bt) | |
| fprintf(stderr, "%s\n", msg.ptr ? msg.ptr : "???"); | |
| } | |
| sigaction_t action = void; | |
| sigaction_t oldseg = void; | |
| sigaction_t oldbus = void; | |
| (cast(byte*) &action)[0 .. action.sizeof] = 0; | |
| sigfillset(&action.sa_mask); // block other signals | |
| action.sa_flags = SA_SIGINFO | SA_RESETHAND; | |
| action.sa_sigaction = &unittestSegvHandler; | |
| sigaction(SIGSEGV, &action, &oldseg); | |
| sigaction(SIGBUS, &action, &oldbus); | |
| scope (exit) | |
| { | |
| sigaction(SIGSEGV, &oldseg, null); | |
| sigaction(SIGBUS, &oldbus, null); | |
| } | |
| } | |
| else static if (hasExecinfo) | |
| { | |
| // segv handler | |
| import core.sys.posix.signal : SA_RESETHAND, SA_SIGINFO, sigaction, sigaction_t, SIGBUS, sigfillset, siginfo_t, | |
| SIGSEGV; | |
| static extern (C) void unittestSegvHandler( int signum, siginfo_t* info, void* ptr ) nothrow | |
| { | |
| enum MAXFRAMES = 128; | |
| void*[MAXFRAMES] callstack; | |
| auto numframes = backtrace( callstack.ptr, MAXFRAMES ); | |
| backtrace_symbols_fd( callstack.ptr, numframes, 2 ); | |
| } | |
| sigaction_t action = void; | |
| sigaction_t oldseg = void; | |
| sigaction_t oldbus = void; | |
| (cast(byte*) &action)[0 .. action.sizeof] = 0; | |
| sigfillset( &action.sa_mask ); // block other signals | |
| action.sa_flags = SA_SIGINFO | SA_RESETHAND; | |
| action.sa_sigaction = &unittestSegvHandler; | |
| sigaction( SIGSEGV, &action, &oldseg ); | |
| sigaction( SIGBUS, &action, &oldbus ); | |
| scope( exit ) | |
| { | |
| sigaction( SIGSEGV, &oldseg, null ); | |
| sigaction( SIGBUS, &oldbus, null ); | |
| } | |
| } | |
| if (Runtime.sm_extModuleUnitTester !is null) | |
| return Runtime.sm_extModuleUnitTester(); | |
| else if (Runtime.sm_moduleUnitTester !is null) | |
| return Runtime.sm_moduleUnitTester() ? UnitTestResult.pass : UnitTestResult.fail; | |
| UnitTestResult results; | |
| foreach ( m; ModuleInfo ) | |
| { | |
| if ( !m ) | |
| continue; | |
| auto fp = m.unitTest; | |
| if ( !fp ) | |
| continue; | |
| import core.exception; | |
| ++results.executed; | |
| try | |
| { | |
| fp(); | |
| ++results.passed; | |
| } | |
| catch ( Throwable e ) | |
| { | |
| if ( typeid(e) == typeid(AssertError) ) | |
| { | |
| // Crude heuristic to figure whether the assertion originates in | |
| // the unittested module. TODO: improve. | |
| auto moduleName = m.name; | |
| if (moduleName.length && e.file.length > moduleName.length | |
| && e.file[0 .. moduleName.length] == moduleName) | |
| { | |
| import core.stdc.stdio : printf; | |
| printf("%.*s(%llu): [unittest] %.*s\n", | |
| cast(int) e.file.length, e.file.ptr, cast(ulong) e.line, | |
| cast(int) e.message.length, e.message.ptr); | |
| // Exception originates in the same module, don't print | |
| // the stack trace. | |
| // TODO: omit stack trace only if assert was thrown | |
| // directly by the unittest. | |
| continue; | |
| } | |
| } | |
| // TODO: perhaps indent all of this stuff. | |
| _d_print_throwable(e); | |
| } | |
| } | |
| import core.internal.parseoptions : rt_configOption; | |
| if (results.passed != results.executed) | |
| { | |
| // by default, we always print a summary if there are failures. | |
| results.summarize = true; | |
| } | |
| else switch (rt_configOption("testmode", null, false)) | |
| { | |
| case "run-main": | |
| results.runMain = true; | |
| break; | |
| case "test-only": | |
| // Never run main, always summarize | |
| results.summarize = true; | |
| break; | |
| case "": | |
| // By default, do not run main if tests are present. | |
| case "test-or-main": | |
| // only run main if there were no tests. Only summarize if we are not | |
| // running main. | |
| results.runMain = (results.executed == 0); | |
| results.summarize = !results.runMain; | |
| break; | |
| default: | |
| assert(0, "Unknown --DRT-testmode option: " ~ rt_configOption("testmode", null, false)); | |
| } | |
| return results; | |
| } | |
| version (DefineBacktrace_using_UnwindBacktrace) | |
| { | |
| import core.internal.backtrace.unwind; | |
| private int backtrace(void** buffer, int maxSize) nothrow | |
| { | |
| if (maxSize < 0) return 0; | |
| struct State | |
| { | |
| void** buffer; | |
| int maxSize; | |
| int entriesWritten = 0; | |
| } | |
| static extern(C) int handler(_Unwind_Context* context, void* statePtr) | |
| { | |
| auto state = cast(State*)statePtr; | |
| if (state.entriesWritten >= state.maxSize) return _URC_END_OF_STACK; | |
| auto instructionPtr = _Unwind_GetIP(context); | |
| if (!instructionPtr) return _URC_END_OF_STACK; | |
| state.buffer[state.entriesWritten] = cast(void*)instructionPtr; | |
| ++state.entriesWritten; | |
| return _URC_NO_REASON; | |
| } | |
| State state; | |
| state.buffer = buffer; | |
| state.maxSize = maxSize; | |
| _Unwind_Backtrace(&handler, &state); | |
| return state.entriesWritten; | |
| } | |
| } | |
| /** | |
| * Get the default `Throwable.TraceInfo` implementation for the platform | |
| * | |
| * This functions returns a trace handler, allowing to inspect the | |
| * current stack trace. | |
| * | |
| * IMPORTANT NOTE! the returned trace is potentially not GC allocated, and so | |
| * you must call `defaultTraceDeallocator` when you are finished with the | |
| * `TraceInfo` | |
| * | |
| * Params: | |
| * ptr = (Windows only) The context to get the stack trace from. | |
| * When `null` (the default), start from the current frame. | |
| * | |
| * Returns: | |
| * A `Throwable.TraceInfo` implementation suitable to iterate over the stack, | |
| * or `null`. If called from a finalizer (destructor), always returns `null` | |
| * as trace handlers allocate. | |
| */ | |
| Throwable.TraceInfo defaultTraceHandler( void* ptr = null ) // @nogc | |
| { | |
| // NOTE: with traces now being allocated using C malloc, no need to worry | |
| // about GC reentrancy. This code left commented out for reference. | |
| // | |
| // avoid recursive GC calls in finalizer, trace handlers should be made @nogc instead | |
| /*import core.memory : GC; | |
| if (GC.inFinalizer) | |
| return null;*/ | |
| static T allocate(T, Args...)(auto ref Args args) @nogc | |
| { | |
| import core.lifetime : emplace; | |
| import core.stdc.stdlib : malloc; | |
| auto result = cast(T)malloc(__traits(classInstanceSize, T)); | |
| return emplace(result, args); | |
| } | |
| static if (__traits(compiles, allocate!LibBacktrace(0))) | |
| { | |
| version (Posix) | |
| static enum FIRSTFRAME = 4; | |
| else version (Win64) | |
| static enum FIRSTFRAME = 4; | |
| else | |
| static enum FIRSTFRAME = 0; | |
| return allocate!LibBacktrace(FIRSTFRAME); | |
| } | |
| else static if (__traits(compiles, allocate!UnwindBacktrace(0))) | |
| { | |
| version (Posix) | |
| static enum FIRSTFRAME = 5; | |
| else version (Win64) | |
| static enum FIRSTFRAME = 4; | |
| else | |
| static enum FIRSTFRAME = 0; | |
| return allocate!UnwindBacktrace(FIRSTFRAME); | |
| } | |
| else version (Windows) | |
| { | |
| import core.sys.windows.stacktrace; | |
| static if (__traits(compiles, allocate!StackTrace(0, null))) | |
| { | |
| import core.sys.windows.winnt : CONTEXT; | |
| version (Win64) | |
| enum FIRSTFRAME = 4; | |
| else version (Win32) | |
| enum FIRSTFRAME = 0; | |
| return allocate!StackTrace(FIRSTFRAME, cast(CONTEXT*)ptr); | |
| } | |
| else | |
| return null; | |
| } | |
| else static if (__traits(compiles, allocate!DefaultTraceInfo())) | |
| return allocate!DefaultTraceInfo(); | |
| else | |
| return null; | |
| } | |
| /// Example of a simple program printing its stack trace | |
| unittest | |
| { | |
| import core.runtime; | |
| import core.stdc.stdio : printf; | |
| void main() | |
| { | |
| auto trace = defaultTraceHandler(null); | |
| foreach (line; trace) | |
| { | |
| printf("%.*s\n", cast(int)line.length, line.ptr); | |
| } | |
| defaultTraceDeallocator(trace); | |
| } | |
| } | |
| /*** | |
| * Deallocate a traceinfo generated by deaultTraceHander. | |
| * | |
| * Call this function on a TraceInfo generated via `defaultTraceHandler` when | |
| * you are done with it. If necessary, this cleans up any manually managed | |
| * resources from the `TraceInfo`, and invalidates it. After this, the object | |
| * is no longer valid. | |
| * | |
| * Params: | |
| * info = The `TraceInfo` to deallocate. This should only be a value that | |
| * was returned by `defaultTraceHandler`. | |
| */ | |
| void defaultTraceDeallocator(Throwable.TraceInfo info) nothrow | |
| { | |
| if (info is null) | |
| return; | |
| auto obj = cast(Object)info; | |
| destroy(obj); | |
| import core.stdc.stdlib : free; | |
| free(cast(void *)obj); | |
| } | |
| version (LibBacktrace_TraceHandler) | |
| { | |
| } | |
| /// Default implementation for most POSIX systems | |
| else version (Posix) private class DefaultTraceInfo : Throwable.TraceInfo | |
| { | |
| import core.demangle; | |
| import core.stdc.stdlib : free; | |
| import core.stdc.string : strlen, memchr, memmove; | |
| this() @nogc | |
| { | |
| // it may not be 1 but it is good enough to get | |
| // in CALL instruction address range for backtrace | |
| enum CALL_INSTRUCTION_SIZE = 1; | |
| static if (__traits(compiles, backtrace((void**).init, int.init))) | |
| numframes = cast(int) backtrace(this.callstack.ptr, MAXFRAMES); | |
| // Backtrace succeeded, adjust the frame to point to the caller | |
| if (numframes >= 2) | |
| foreach (ref elem; this.callstack) | |
| elem -= CALL_INSTRUCTION_SIZE; | |
| else // backtrace() failed, do it ourselves | |
| { | |
| static void** getBasePtr() @nogc | |
| { | |
| version (D_InlineAsm_X86) | |
| asm @nogc { naked; mov EAX, EBP; ret; } | |
| else | |
| version (D_InlineAsm_X86_64) | |
| asm @nogc { naked; mov RAX, RBP; ret; } | |
| else | |
| return null; | |
| } | |
| auto stackTop = getBasePtr(); | |
| auto stackBottom = cast(void**) thread_stackBottom(); | |
| void* dummy; | |
| if ( stackTop && &dummy < stackTop && stackTop < stackBottom ) | |
| { | |
| auto stackPtr = stackTop; | |
| for ( numframes = 0; stackTop <= stackPtr && | |
| stackPtr < stackBottom && | |
| numframes < MAXFRAMES; ) | |
| { | |
| callstack[numframes++] = *(stackPtr + 1) - CALL_INSTRUCTION_SIZE; | |
| stackPtr = cast(void**) *stackPtr; | |
| } | |
| } | |
| } | |
| } | |
| override int opApply( scope int delegate(ref const(char[])) dg ) const | |
| { | |
| return opApply( (ref size_t, ref const(char[]) buf) | |
| { | |
| return dg( buf ); | |
| } ); | |
| } | |
| override int opApply( scope int delegate(ref size_t, ref const(char[])) dg ) const | |
| { | |
| version (linux) enum enableDwarf = true; | |
| else version (FreeBSD) enum enableDwarf = true; | |
| else version (DragonFlyBSD) enum enableDwarf = true; | |
| else version (OpenBSD) enum enableDwarf = true; | |
| else version (Darwin) enum enableDwarf = true; | |
| else enum enableDwarf = false; | |
| static if (hasExecinfo) | |
| { | |
| const framelist = backtrace_symbols( callstack.ptr, numframes ); | |
| scope(exit) free(cast(void*) framelist); | |
| static if (enableDwarf) | |
| { | |
| import core.internal.backtrace.dwarf; | |
| return traceHandlerOpApplyImpl(numframes, | |
| i => callstack[i], | |
| (i) { auto str = framelist[i][0 .. strlen(framelist[i])]; return getMangledSymbolName(str); }, | |
| dg); | |
| } | |
| else | |
| { | |
| int ret = 0; | |
| for (size_t pos = 0; pos < numframes; ++pos) | |
| { | |
| char[4096] fixbuf = void; | |
| auto buf = framelist[pos][0 .. strlen(framelist[pos])]; | |
| buf = fixline( buf, fixbuf ); | |
| ret = dg( pos, buf ); | |
| if ( ret ) | |
| break; | |
| } | |
| return ret; | |
| } | |
| } | |
| else | |
| { | |
| // https://code.woboq.org/userspace/glibc/debug/backtracesyms.c.html | |
| // The logic that glibc's backtrace use is to check for for `dli_fname`, | |
| // the file name, and error if not present, then check for `dli_sname`. | |
| // In case `dli_fname` is present but not `dli_sname`, the address is | |
| // printed related to the file. We just print the file. | |
| static const(char)[] getFrameName (const(void)* ptr) | |
| { | |
| import core.sys.posix.dlfcn; | |
| Dl_info info = void; | |
| // Note: See the module documentation about `-L--export-dynamic` | |
| if (dladdr(ptr, &info)) | |
| { | |
| // Return symbol name if possible | |
| if (info.dli_sname !is null && info.dli_sname[0] != '\0') | |
| return info.dli_sname[0 .. strlen(info.dli_sname)]; | |
| // Fall back to file name | |
| if (info.dli_fname !is null && info.dli_fname[0] != '\0') | |
| return info.dli_fname[0 .. strlen(info.dli_fname)]; | |
| } | |
| // `dladdr` failed | |
| return "<ERROR: Unable to retrieve function name>"; | |
| } | |
| static if (enableDwarf) | |
| { | |
| import core.internal.backtrace.dwarf; | |
| return traceHandlerOpApplyImpl(numframes, | |
| i => callstack[i], | |
| i => getFrameName(callstack[i]), | |
| dg); | |
| } | |
| else | |
| { | |
| // Poor man solution. Does not show line numbers, but does (potentially) show a backtrace of function names. | |
| import core.internal.container.array; | |
| Array!(const(char)[]) frameNames; | |
| frameNames.length = numframes; | |
| size_t startIdx; | |
| foreach (idx; 0 .. numframes) | |
| { | |
| frameNames[idx] = getFrameName(callstack[idx]); | |
| // NOTE: The first few frames with the current implementation are | |
| // inside core.runtime and the object code, so eliminate | |
| // these for readability. | |
| // They also might depend on build parameters, which would make | |
| // using a fixed number of frames otherwise brittle. | |
| version (LDC) enum BaseExceptionFunctionName = "_d_throw_exception"; | |
| else enum BaseExceptionFunctionName = "_d_throwdwarf"; | |
| if (!startIdx && frameNames[idx] == BaseExceptionFunctionName) | |
| startIdx = idx + 1; | |
| } | |
| int ret = 0; | |
| foreach (idx; startIdx .. numframes) | |
| { | |
| ret = dg( idx, frameNames[idx] ); | |
| if ( ret ) | |
| break; | |
| } | |
| return ret; | |
| } | |
| } | |
| } | |
| override string toString() const | |
| { | |
| string buf; | |
| foreach ( i, line; this ) | |
| buf ~= i ? "\n" ~ line : line; | |
| return buf; | |
| } | |
| private: | |
| int numframes; | |
| enum MAXFRAMES = 128; | |
| void*[MAXFRAMES] callstack = void; | |
| private: | |
| static if (hasExecinfo) | |
| { | |
| const(char)[] fixline( const(char)[] buf, return ref char[4096] fixbuf ) const | |
| { | |
| size_t symBeg, symEnd; | |
| getMangledSymbolName(buf, symBeg, symEnd); | |
| enum min = (size_t a, size_t b) => a <= b ? a : b; | |
| if (symBeg == symEnd || symBeg >= fixbuf.length) | |
| { | |
| immutable len = min(buf.length, fixbuf.length); | |
| fixbuf[0 .. len] = buf[0 .. len]; | |
| return fixbuf[0 .. len]; | |
| } | |
| else | |
| { | |
| fixbuf[0 .. symBeg] = buf[0 .. symBeg]; | |
| auto sym = demangle(buf[symBeg .. symEnd], fixbuf[symBeg .. $], getCXXDemangler()); | |
| if (sym.ptr !is fixbuf.ptr + symBeg) | |
| { | |
| // demangle reallocated the buffer, copy the symbol to fixbuf | |
| immutable len = min(fixbuf.length - symBeg, sym.length); | |
| memmove(fixbuf.ptr + symBeg, sym.ptr, len); | |
| if (symBeg + len == fixbuf.length) | |
| return fixbuf[]; | |
| } | |
| immutable pos = symBeg + sym.length; | |
| assert(pos < fixbuf.length); | |
| immutable tail = buf.length - symEnd; | |
| immutable len = min(fixbuf.length - pos, tail); | |
| fixbuf[pos .. pos + len] = buf[symEnd .. symEnd + len]; | |
| return fixbuf[0 .. pos + len]; | |
| } | |
| } | |
| } | |
| } |