name : swoole_coroutine_context.h /* +----------------------------------------------------------------------+ | Swoole | +----------------------------------------------------------------------+ | This source file is subject to version 2.0 of the Apache license, | | that is bundled with this package in the file LICENSE, and is | | available through the world-wide-web at the following url: | | http://www.apache.org/licenses/LICENSE-2.0.html | | If you did not receive a copy of the Apache2.0 license and are unable| | to obtain it through the world-wide-web, please send a note to | | license@swoole.com so we can mail you a copy immediately. | +----------------------------------------------------------------------+ | Author: Tianfeng Han <rango@swoole.com> | | Twosee <twose@qq.com> | +----------------------------------------------------------------------+ */ #pragma once #include "swoole.h" #ifdef SW_USE_THREAD_CONTEXT #include <thread> #include <mutex> #elif !defined(SW_USE_ASM_CONTEXT) #define USE_UCONTEXT 1 #ifndef _XOPEN_SOURCE #define _XOPEN_SOURCE #endif #include <ucontext.h> #else #define USE_ASM_CONTEXT 1 #include "swoole_asm_context.h" #endif #if defined(HAVE_VALGRIND) && !defined(HAVE_KQUEUE) #define USE_VALGRIND 1 #include <valgrind/valgrind.h> #endif #ifdef USE_UCONTEXT typedef ucontext_t coroutine_context_t; #elif defined(USE_ASM_CONTEXT) typedef fcontext_t coroutine_context_t; #endif typedef std::function<void(void *)> CoroutineFunc; namespace swoole { namespace coroutine { class Context { public: Context(size_t stack_size, CoroutineFunc fn, void *private_data); ~Context(); bool swap_in(); bool swap_out(); #if !defined(SW_USE_THREAD_CONTEXT) && defined(SW_CONTEXT_DETECT_STACK_USAGE) ssize_t get_stack_usage(); #endif #ifndef SW_USE_THREAD_CONTEXT char *get_stack() const { return stack_; } size_t get_stack_size() const { return stack_size_; } #endif bool is_end() const { return end_; } protected: CoroutineFunc fn_; #ifdef SW_USE_THREAD_CONTEXT std::thread thread_; std::mutex lock_; std::mutex *swap_lock_; #else coroutine_context_t ctx_; coroutine_context_t swap_ctx_; char *stack_; uint32_t stack_size_; #endif #ifdef USE_VALGRIND uint32_t valgrind_stack_id; #endif void *private_data_; bool end_; static void context_func(void *arg); }; } // namespace coroutine } // namespace swoole