/* +----------------------------------------------------------------------+ | 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 | | Twosee | +----------------------------------------------------------------------+ */ #pragma once #ifdef HAVE_CONFIG_H #include "config.h" #elif defined(ENABLE_PHP_SWOOLE) #include "php_config.h" #endif #ifdef __cplusplus #define SW_EXTERN_C_BEGIN extern "C" { #define SW_EXTERN_C_END } #else #define SW_EXTERN_C_BEGIN #define SW_EXTERN_C_END #endif #ifndef _GNU_SOURCE #define _GNU_SOURCE #endif #ifndef _PTHREAD_PSHARED #define _PTHREAD_PSHARED #endif /*--- C standard library ---*/ #include #include #include #include #include #include #include #include #include #include #include /* sched_yield() */ #include #include #include #include #include #include #include typedef unsigned long ulong_t; #ifndef PRId64 #define PRId64 "lld" #endif #ifndef PRIu64 #define PRIu64 "llu" #endif #ifndef PRIx64 #define PRIx64 "llx" #endif #if defined(__GNUC__) #if __GNUC__ >= 3 #define sw_inline inline __attribute__((always_inline)) #else #define sw_inline inline #endif #elif defined(_MSC_VER) #define sw_inline __forceinline #else #define sw_inline inline #endif #if defined(__GNUC__) && __GNUC__ >= 4 #define SW_API __attribute__((visibility("default"))) #else #define SW_API #endif #if !defined(__GNUC__) || __GNUC__ < 3 #define __builtin_expect(x, expected_value) (x) #endif #define sw_likely(x) __builtin_expect(!!(x), 1) #define sw_unlikely(x) __builtin_expect(!!(x), 0) #define SW_START_LINE "-------------------------START----------------------------" #define SW_END_LINE "--------------------------END-----------------------------" #define SW_ECHO_RED "\e[31m%s\e[0m" #define SW_ECHO_GREEN "\e[32m%s\e[0m" #define SW_ECHO_YELLOW "\e[33m%s\e[0m" #define SW_ECHO_BLUE "\e[34m%s\e[0m" #define SW_ECHO_MAGENTA "\e[35m%s\e[0m" #define SW_ECHO_CYAN "\e[36m%s\e[0m" #define SW_ECHO_WHITE "\e[37m%s\e[0m" #define SW_ECHO_LEN_RED "\e[31m%.*s\e[0m" #define SW_ECHO_LEN_GREEN "\e[32m%.*s\e[0m" #define SW_ECHO_LEN_YELLOW "\e[33m%.*s\e[0m" #define SW_ECHO_LEN_BLUE "\e[34m%.*s\e[0m" #define SW_ECHO_LEN_MAGENTA "\e[35m%.*s\e[0m" #define SW_ECHO_LEN_CYAN "\e[36m%.*s\e[0m" #define SW_ECHO_LEN_WHITE "\e[37m%.*s\e[0m" #define SW_ECHO_RED_BG "\e[41m%s\e[0m" #define SW_ECHO_GREEN_BG "\e[42m%s\e[0m" #define SW_COLOR_RED 1 #define SW_COLOR_GREEN 2 #define SW_COLOR_YELLOW 3 #define SW_COLOR_BLUE 4 #define SW_COLOR_MAGENTA 5 #define SW_COLOR_CYAN 6 #define SW_COLOR_WHITE 7 #define SW_SPACE ' ' #define SW_CRLF "\r\n" #define SW_CRLF_LEN 2 #define SW_ASCII_CODE_0 64 #define SW_ASCII_CODE_Z 106 /*----------------------------------------------------------------------------*/ #include "swoole_config.h" #include "swoole_version.h" #include "swoole_log.h" #include "swoole_atomic.h" #include "swoole_error.h" #define SW_MAX(A, B) ((A) > (B) ? (A) : (B)) #define SW_MIN(A, B) ((A) < (B) ? (A) : (B)) #define SW_LOOP_N(n) for (decltype(n) i = 0; i < n; i++) #define SW_LOOP for (;;) #ifndef MAX #define MAX(A, B) SW_MAX(A, B) #endif #ifndef MIN #define MIN(A, B) SW_MIN(A, B) #endif #define SW_NUM_BILLION (1000 * 1000 * 1000) #define SW_NUM_MILLION (1000 * 1000) #ifdef SW_DEBUG #define SW_ASSERT(e) assert(e) #define SW_ASSERT_1BYTE(v) \ do { \ size_t i = 0, n = 0; \ for (; i < sizeof(v); i++) { \ n += ((v >> i) & 1) ? 1 : 0; \ } \ assert(n == 1); \ } while (0) #else #define SW_ASSERT(e) #define SW_ASSERT_1BYTE(v) #endif #define SW_START_SLEEP usleep(100000) // sleep 1s,wait fork and pthread_create /*-----------------------------------Memory------------------------------------*/ void *sw_malloc(size_t size); void sw_free(void *ptr); void *sw_calloc(size_t nmemb, size_t size); void *sw_realloc(void *ptr, size_t size); // Evaluates to the number of elements in 'array' #define SW_ARRAY_SIZE(array) (sizeof(array) / sizeof(array[0])) #define SW_DEFAULT_ALIGNMENT sizeof(unsigned long) #define SW_MEM_ALIGNED_SIZE(size) SW_MEM_ALIGNED_SIZE_EX(size, SW_DEFAULT_ALIGNMENT) #define SW_MEM_ALIGNED_SIZE_EX(size, alignment) (((size) + ((alignment) -1LL)) & ~((alignment) -1LL)) /*-------------------------------Declare Struct--------------------------------*/ namespace swoole { class MemoryPool; class Reactor; class String; class Timer; struct TimerNode; struct Event; class Pipe; namespace network { struct Socket; struct Address; } // namespace network class AsyncThreads; namespace async { class ThreadPool; } struct Protocol; struct EventData; struct DataHead; typedef int (*ReactorHandler)(Reactor *reactor, Event *event); typedef std::function Callback; typedef std::function TimerCallback; } // namespace swoole typedef swoole::Reactor swReactor; typedef swoole::String swString; typedef swoole::Timer swTimer; typedef swoole::network::Socket swSocket; typedef swoole::Protocol swProtocol; typedef swoole::EventData swEventData; typedef swoole::DataHead swDataHead; typedef swoole::Event swEvent; /*----------------------------------String-------------------------------------*/ #define SW_STRS(s) s, sizeof(s) #define SW_STRL(s) s, sizeof(s) - 1 #define SW_STREQ(str, len, const_str) swoole_streq(str, len, SW_STRL(const_str)) #define SW_STRCASEEQ(str, len, const_str) swoole_strcaseeq(str, len, SW_STRL(const_str)) #define SW_STR_STARTS_WITH(str, len, const_sub_str) swoole_str_starts_with(str, len, SW_STRL(const_sub_str)) #define SW_STR_ISTARTS_WITH(str, len, const_sub_str) swoole_str_istarts_with(str, len, SW_STRL(const_sub_str)) #if defined(SW_USE_JEMALLOC) || defined(SW_USE_TCMALLOC) #define sw_strdup swoole_strdup #define sw_strndup swoole_strndup #else #define sw_strdup strdup #define sw_strndup strndup #endif /** always return less than size, zero termination */ size_t sw_snprintf(char *buf, size_t size, const char *format, ...) __attribute__((format(printf, 3, 4))); size_t sw_vsnprintf(char *buf, size_t size, const char *format, va_list args); #define sw_memset_zero(s, n) memset(s, '\0', n) static sw_inline int sw_mem_equal(const void *v1, size_t s1, const void *v2, size_t s2) { return s1 == s2 && memcmp(v1, v2, s2) == 0; } static inline size_t swoole_strlcpy(char *dest, const char *src, size_t size) { const size_t len = strlen(src); if (size != 0) { const size_t n = std::min(len, size - 1); memcpy(dest, src, n); dest[n] = '\0'; } return len; } static inline char *swoole_strdup(const char *s) { size_t l = strlen(s) + 1; char *p = (char *) sw_malloc(l); if (sw_likely(p)) { memcpy(p, s, l); } return p; } static inline char *swoole_strndup(const char *s, size_t n) { char *p = (char *) sw_malloc(n + 1); if (sw_likely(p)) { strncpy(p, s, n)[n] = '\0'; } return p; } /* string equal */ static inline unsigned int swoole_streq(const char *str1, size_t len1, const char *str2, size_t len2) { return (len1 == len2) && (strncmp(str1, str2, len1) == 0); } static inline unsigned int swoole_strcaseeq(const char *str1, size_t len1, const char *str2, size_t len2) { return (len1 == len2) && (strncasecmp(str1, str2, len1) == 0); } static inline unsigned int swoole_str_starts_with(const char *pstr, size_t plen, const char *sstr, size_t slen) { return (plen >= slen) && (strncmp(pstr, sstr, slen) == 0); } static inline unsigned int swoole_str_istarts_with(const char *pstr, size_t plen, const char *sstr, size_t slen) { return (plen >= slen) && (strncasecmp(pstr, sstr, slen) == 0); } static inline const char *swoole_strnstr(const char *haystack, uint32_t haystack_length, const char *needle, uint32_t needle_length) { assert(needle_length > 0); uint32_t i; if (sw_likely(needle_length <= haystack_length)) { for (i = 0; i < haystack_length - needle_length + 1; i++) { if ((haystack[0] == needle[0]) && (0 == memcmp(haystack, needle, needle_length))) { return haystack; } haystack++; } } return NULL; } static inline const char *swoole_strncasestr(const char *haystack, uint32_t haystack_length, const char *needle, uint32_t needle_length) { assert(needle_length > 0); uint32_t i; if (sw_likely(needle_length <= haystack_length)) { for (i = 0; i < haystack_length - needle_length + 1; i++) { if ((haystack[0] == needle[0]) && (0 == strncasecmp(haystack, needle, needle_length))) { return haystack; } haystack++; } } return NULL; } static inline ssize_t swoole_strnpos(const char *haystack, uint32_t haystack_length, const char *needle, uint32_t needle_length) { assert(needle_length > 0); const char *pos; pos = swoole_strnstr(haystack, haystack_length, needle, needle_length); return pos == NULL ? -1 : pos - haystack; } static inline ssize_t swoole_strrnpos(const char *haystack, const char *needle, uint32_t length) { uint32_t needle_length = strlen(needle); assert(needle_length > 0); uint32_t i; haystack += (length - needle_length); for (i = length - needle_length; i > 0; i--) { if ((haystack[0] == needle[0]) && (0 == memcmp(haystack, needle, needle_length))) { return i; } haystack--; } return -1; } static inline void swoole_strtolower(char *str, int length) { char *c, *e; c = str; e = c + length; while (c < e) { *c = tolower(*c); c++; } } /*--------------------------------Constants------------------------------------*/ enum swResultCode { SW_OK = 0, SW_ERR = -1, }; enum swReturnCode { SW_CONTINUE = 1, SW_WAIT = 2, SW_CLOSE = 3, SW_ERROR = 4, SW_READY = 5, SW_INVALID = 6, SW_REDUCE_SIZE = 7, }; enum swFdType { SW_FD_SESSION, // server stream session SW_FD_STREAM_SERVER, // server stream port SW_FD_DGRAM_SERVER, // server dgram port SW_FD_PIPE, SW_FD_STREAM, SW_FD_AIO, /** * Coroutine Socket */ SW_FD_CO_SOCKET, /** * socket poll fd [coroutine::socket_poll] */ SW_FD_CO_POLL, /** * event waiter */ SW_FD_CO_EVENT, /** * signalfd */ SW_FD_SIGNAL, SW_FD_DNS_RESOLVER, SW_FD_CARES, /** * SW_FD_USER or SW_FD_USER+n: for custom event */ SW_FD_USER = 16, SW_FD_STREAM_CLIENT, SW_FD_DGRAM_CLIENT, }; enum swSocketFlag { SW_SOCK_NONBLOCK = 1 << 2, SW_SOCK_CLOEXEC = 1 << 3, SW_SOCK_SSL = (1u << 9), }; enum swSocketType { SW_SOCK_TCP = 1, SW_SOCK_UDP = 2, SW_SOCK_TCP6 = 3, SW_SOCK_UDP6 = 4, SW_SOCK_UNIX_STREAM = 5, // unix sock stream SW_SOCK_UNIX_DGRAM = 6, // unix sock dgram SW_SOCK_RAW = 7, }; enum swEventType { SW_EVENT_NULL = 0, SW_EVENT_DEAULT = 1u << 8, SW_EVENT_READ = 1u << 9, SW_EVENT_WRITE = 1u << 10, SW_EVENT_RDWR = SW_EVENT_READ | SW_EVENT_WRITE, SW_EVENT_ERROR = 1u << 11, SW_EVENT_ONCE = 1u << 12, }; enum swForkType { SW_FORK_SPAWN = 0, SW_FORK_EXEC = 1 << 1, SW_FORK_DAEMON = 1 << 2, SW_FORK_PRECHECK = 1 << 3, }; enum swTraverseOperation { SW_TRAVERSE_KEEP = 0, SW_TRAVERSE_REMOVE = 1, SW_TRAVERSE_STOP = 2, }; //------------------------------------------------------------------------------- #define sw_yield() sched_yield() //------------------------------Base-------------------------------- #ifndef uchar typedef unsigned char uchar; #endif #define swoole_tolower(c) (uchar)((c >= 'A' && c <= 'Z') ? (c | 0x20) : c) #define swoole_toupper(c) (uchar)((c >= 'a' && c <= 'z') ? (c & ~0x20) : c) void swoole_random_string(char *buf, size_t size); void swoole_random_string(std::string &str, size_t size); size_t swoole_random_bytes(char *buf, size_t size); static sw_inline char *swoole_strlchr(char *p, char *last, char c) { while (p < last) { if (*p == c) { return p; } p++; } return NULL; } static sw_inline size_t swoole_size_align(size_t size, int pagesize) { return size + (pagesize - (size % pagesize)); } //------------------------------Base-------------------------------- enum swEventDataFlag { SW_EVENT_DATA_NORMAL, SW_EVENT_DATA_PTR = 1u << 1, SW_EVENT_DATA_CHUNK = 1u << 2, SW_EVENT_DATA_BEGIN = 1u << 3, SW_EVENT_DATA_END = 1u << 4, SW_EVENT_DATA_OBJ_PTR = 1u << 5, SW_EVENT_DATA_POP_PTR = 1u << 6, }; enum swTaskFlag { SW_TASK_TMPFILE = 1, SW_TASK_SERIALIZE = 1u << 1, SW_TASK_NONBLOCK = 1u << 2, SW_TASK_CALLBACK = 1u << 3, SW_TASK_WAITALL = 1u << 4, SW_TASK_COROUTINE = 1u << 5, SW_TASK_PEEK = 1u << 6, SW_TASK_NOREPLY = 1u << 7, }; enum swDNSLookupFlag { SW_DNS_LOOKUP_RANDOM = (1u << 11), }; #ifdef __MACH__ char *sw_error_(); #define sw_error sw_error_() #else extern __thread char sw_error[SW_ERROR_MSG_SIZE]; #endif enum swProcessType { SW_PROCESS_MASTER = 1, SW_PROCESS_WORKER = 2, SW_PROCESS_MANAGER = 3, SW_PROCESS_TASKWORKER = 4, SW_PROCESS_USERWORKER = 5, }; enum swPipeType { SW_PIPE_WORKER = 0, SW_PIPE_MASTER = 1, SW_PIPE_READ = 0, SW_PIPE_WRITE = 1, SW_PIPE_NONBLOCK = 2, }; //----------------------Tool Function--------------------- uint32_t swoole_common_multiple(uint32_t u, uint32_t v); uint32_t swoole_common_divisor(uint32_t u, uint32_t v); int swoole_itoa(char *buf, long value); bool swoole_mkdir_recursive(const std::string &dir); int swoole_rand(int min, int max); int swoole_system_random(int min, int max); int swoole_version_compare(const char *version1, const char *version2); void swoole_print_backtrace(void); char *swoole_string_format(size_t n, const char *format, ...); bool swoole_get_env(const char *name, int *value); int swoole_get_systemd_listen_fds(); void swoole_init(void); void swoole_clean(void); pid_t swoole_fork(int flags); pid_t swoole_fork_exec(const std::function &child_fn); void swoole_redirect_stdout(int new_fd); int swoole_shell_exec(const char *command, pid_t *pid, bool get_error_stream); int swoole_daemon(int nochdir, int noclose); bool swoole_set_task_tmpdir(const std::string &dir); int swoole_tmpfile(char *filename); #ifdef HAVE_CPU_AFFINITY #ifdef __FreeBSD__ #include #include #include typedef cpuset_t cpu_set_t; #endif int swoole_set_cpu_affinity(cpu_set_t *set); #endif #if defined(_POSIX_TIMERS) && ((_POSIX_TIMERS > 0) || defined(__OpenBSD__)) && defined(_POSIX_MONOTONIC_CLOCK) && defined(CLOCK_MONOTONIC) #ifndef HAVE_CLOCK_GETTIME #define HAVE_CLOCK_GETTIME #endif #define swoole_clock_realtime(t) clock_gettime(CLOCK_REALTIME, t) #elif defined(__APPLE__) int swoole_clock_realtime(struct timespec *t); #endif static inline struct timespec swoole_time_until(int milliseconds) { struct timespec t; swoole_clock_realtime(&t); int sec = milliseconds / 1000; int msec = milliseconds - (sec * 1000); t.tv_sec += sec; t.tv_nsec += msec * 1000 * 1000; if (t.tv_nsec > SW_NUM_BILLION) { int _sec = t.tv_nsec / SW_NUM_BILLION; t.tv_sec += _sec; t.tv_nsec -= _sec * SW_NUM_BILLION; } return t; } namespace swoole { typedef long SessionId; typedef long TaskId; typedef uint8_t ReactorId; typedef uint32_t WorkerId; typedef enum swEventType EventType; typedef enum swSocketType SocketType; typedef enum swFdType FdType; typedef enum swReturnCode ReturnCode; typedef enum swResultCode ResultCode; struct Event { int fd; int16_t reactor_id; FdType type; network::Socket *socket; }; struct DataHead { SessionId fd; uint64_t msg_id; uint32_t len; int16_t reactor_id; uint8_t type; uint8_t flags; uint16_t server_fd; uint16_t ext_flags; uint32_t reserved; double time; size_t dump(char *buf, size_t len); void print(); }; struct EventData { DataHead info; char data[SW_IPC_BUFFER_SIZE]; }; struct SendData { DataHead info; const char *data; }; struct RecvData { DataHead info; const char *data; }; struct ThreadGlobal { uint16_t id; uint8_t type; String *buffer_stack; Reactor *reactor; Timer *timer; AsyncThreads *async_threads; uint32_t signal_listener_num; uint32_t co_signal_listener_num; int error; }; struct Allocator { void *(*malloc)(size_t size); void *(*calloc)(size_t nmemb, size_t size); void *(*realloc)(void *ptr, size_t size); void (*free)(void *ptr); }; struct NameResolver { enum Type { TYPE_KERNEL, TYPE_PHP, TYPE_USER, }; struct Context { int type; double timeout; void *private_data; bool with_port; bool cluster_; bool final_; std::function dtor; ~Context() { if (private_data && dtor) { dtor(this); } } }; std::function resolve; void *private_data; enum Type type; }; struct Global { uchar init : 1; uchar running : 1; uchar wait_signal : 1; uchar enable_signalfd : 1; uchar socket_dontwait : 1; uchar dns_lookup_random : 1; uchar use_async_resolver : 1; uchar use_name_resolver : 1; int process_type; uint32_t process_id; TaskId current_task_id; pid_t pid; int signal_fd; bool signal_alarm; long trace_flags; void (*fatal_error)(int code, const char *str, ...); //-----------------------[System]-------------------------- uint16_t cpu_num; uint32_t pagesize; struct utsname uname; uint32_t max_sockets; //-----------------------[Memory]-------------------------- MemoryPool *memory_pool; Allocator std_allocator; std::string task_tmpfile; //-----------------------[DNS]-------------------------- std::string dns_server_host; int dns_server_port; double dns_cache_refresh_time; int dns_tries; std::string dns_resolvconf_path; std::string dns_hosts_path; std::list name_resolvers; //-----------------------[AIO]-------------------------- uint32_t aio_core_worker_num; uint32_t aio_worker_num; double aio_max_wait_time; double aio_max_idle_time; network::Socket *aio_default_socket; //-----------------------[Hook]-------------------------- void *hooks[SW_MAX_HOOK_TYPE]; std::function user_exit_condition; // bug report message std::string bug_report_message; }; std::string dirname(const std::string &file); int hook_add(void **hooks, int type, const Callback &func, int push_back); void hook_call(void **hooks, int type, void *arg); double microtime(void); } // namespace swoole extern swoole::Global SwooleG; // Local Global Variable extern __thread swoole::ThreadGlobal SwooleTG; // Thread Global Variable #define SW_CPU_NUM (SwooleG.cpu_num) static inline void swoole_set_last_error(int error) { SwooleTG.error = error; } static inline int swoole_get_last_error() { return SwooleTG.error; } static inline int swoole_get_thread_id() { return SwooleTG.id; } static inline int swoole_get_process_type() { return SwooleG.process_type; } static inline int swoole_get_process_id() { return SwooleG.process_id; } static inline uint32_t swoole_pagesize() { return SwooleG.pagesize; } SW_API const char *swoole_strerror(int code); SW_API void swoole_throw_error(int code); SW_API void swoole_ignore_error(int code); SW_API bool swoole_is_ignored_error(int code); SW_API void swoole_set_log_level(int level); SW_API void swoole_set_log_file(const char *file); SW_API void swoole_set_trace_flags(long flags); SW_API void swoole_set_dns_server(const std::string &server); SW_API void swoole_set_hosts_path(const std::string &hosts_file); SW_API std::pair swoole_get_dns_server(); SW_API bool swoole_load_resolv_conf(); SW_API void swoole_name_resolver_add(const swoole::NameResolver &resolver, bool append = true); SW_API void swoole_name_resolver_each( const std::function::iterator &iter)> &fn); SW_API std::string swoole_name_resolver_lookup(const std::string &host_name, swoole::NameResolver::Context *ctx); SW_API int swoole_get_log_level(); //----------------------------------------------- static sw_inline void sw_spinlock(sw_atomic_t *lock) { uint32_t i, n; while (1) { if (*lock == 0 && sw_atomic_cmp_set(lock, 0, 1)) { return; } if (SW_CPU_NUM > 1) { for (n = 1; n < SW_SPINLOCK_LOOP_N; n <<= 1) { for (i = 0; i < n; i++) { sw_atomic_cpu_pause(); } if (*lock == 0 && sw_atomic_cmp_set(lock, 0, 1)) { return; } } } sw_yield(); } } static sw_inline swoole::String *sw_tg_buffer() { return SwooleTG.buffer_stack; } static sw_inline swoole::MemoryPool *sw_mem_pool() { return SwooleG.memory_pool; } static sw_inline const swoole::Allocator *sw_std_allocator() { return &SwooleG.std_allocator; }