| /** |
| * Licensed to the Apache Software Foundation (ASF) under one or more |
| * contributor license agreements. See the NOTICE file distributed with |
| * this work for additional information regarding copyright ownership. |
| * The ASF licenses this file to You under the Apache License, Version 2.0 |
| * (the "License"); you may not use this file except in compliance with |
| * the License. You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| #ifndef LIBMINIFI_INCLUDE_UTILS_BACKTRACE_H_ |
| #define LIBMINIFI_INCLUDE_UTILS_BACKTRACE_H_ |
| |
| #ifdef HAS_EXECINFO |
| #include <execinfo.h> |
| #include <csignal> |
| #endif |
| #include <thread> |
| #include <utility> |
| #include <vector> |
| #include <mutex> |
| #include <condition_variable> |
| #include <iostream> |
| #include <sstream> |
| #include <memory> |
| |
| #define TRACE_BUFFER_SIZE 128 |
| |
| /** |
| * Forward declaration allows us to tightly couple TraceResolver |
| * with BackTrace. |
| */ |
| class TraceResolver; |
| |
| /** |
| * Purpose: Backtrace is a vector of trace lines. |
| * |
| */ |
| class BackTrace { |
| public: |
| BackTrace() = default; |
| |
| BackTrace(std::string name) |
| : name_(std::move(name)) { |
| } |
| |
| std::vector<std::string> getTraces() const { |
| return trace_; |
| } |
| |
| /** |
| * Return thread name of f this caller |
| * @returns name ; |
| */ |
| std::string getName() const { |
| return name_; |
| } |
| |
| protected: |
| void addLine(std::string symbol_line) { |
| trace_.emplace_back(std::move(symbol_line)); |
| } |
| |
| private: |
| std::string name_; |
| std::vector<std::string> trace_; |
| friend class TraceResolver; |
| }; |
| |
| /** |
| * Pulls the trace and places it onto the TraceResolver instance. |
| */ |
| void pull_trace(uint8_t frames_to_skip = 1); |
| |
| /** |
| * Emplaces a signal handler for SIGUSR2 |
| */ |
| void emplace_handler(); |
| |
| /** |
| * Purpose: Provides a singular instance to grab the call stack for thread(s). |
| * Design: is a singleton to avoid multiple signal handlers. |
| */ |
| class TraceResolver { |
| public: |
| /** |
| * Retrieves the backtrace for the provided thread reference |
| * @return BackTrace instance |
| */ |
| BackTrace getBackTrace(std::string thread_name, std::thread::native_handle_type thread); |
| |
| /** |
| * Retrieves the backtrace for the calling thread |
| * @returns BackTrace instance |
| */ |
| BackTrace getBackTrace(std::string thread_name) { |
| #ifdef WIN32 |
| // currrently not supported in windows |
| return BackTrace(std::move(thread_name)); |
| #else |
| return getBackTrace(std::move(thread_name), pthread_self()); |
| #endif |
| } |
| |
| /** |
| * Returns a static instance of the thread resolver. |
| */ |
| static TraceResolver &getResolver() { |
| static TraceResolver resolver; |
| return resolver; |
| } |
| |
| /** |
| * Adds a trace line with an optional function |
| * @param symbol_line symbol line that was produced |
| * @param func function name |
| */ |
| void addTraceLine(const char *symbol_line, const char *func = nullptr) { |
| std::stringstream line; |
| line << symbol_line; |
| if (nullptr != func) { |
| line << " @" << func; |
| } |
| trace_.addLine(line.str()); |
| } |
| |
| /** |
| * Adds a trace line in the format "<file_name> @ <symbol_name> + <symbol_offset>", if symbol_name is not nullptr, |
| * "<file_name>" otherwise. |
| * @param file_name the filename of the shared object |
| * @param symbol_name the name of the symbol (or of the shared object, if the offset is supplied from that) |
| * @param symbol_offset offset from the base address of the symbol (or the shared object) |
| */ |
| void addTraceLine(const char* file_name, const char* symbol_name, uintptr_t symbol_offset) { |
| std::stringstream line; |
| line << file_name; |
| if (symbol_name != nullptr) { |
| line << " @ " << symbol_name << " + " << symbol_offset; |
| } |
| trace_.addLine(line.str()); |
| } |
| |
| std::unique_lock<std::mutex> lock() { |
| return std::unique_lock<std::mutex>(trace_mutex_); |
| } |
| |
| void notifyPullTracesDone(std::unique_lock<std::mutex>& lock) { |
| std::unique_lock<std::mutex> tlock(std::move(lock)); |
| pull_traces_ = true; |
| trace_condition_.notify_one(); |
| } |
| |
| private: |
| TraceResolver() = default; |
| |
| BackTrace trace_; |
| mutable std::mutex mutex_; |
| |
| bool pull_traces_{false}; |
| mutable std::mutex trace_mutex_; |
| std::condition_variable trace_condition_; |
| }; |
| |
| #endif /* LIBMINIFI_INCLUDE_UTILS_BACKTRACE_H_ */ |
| |