SysHelper.h 5.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183
  1. /*
  2. * Copyright: JessMA Open Source (ldcsaa@gmail.com)
  3. *
  4. * Author : Bruce Liang
  5. * Website : https://github.com/ldcsaa
  6. * Project : https://github.com/ldcsaa/HP-Socket
  7. * Blog : http://www.cnblogs.com/ldcsaa
  8. * Wiki : http://www.oschina.net/p/hp-socket
  9. * QQ Group : 44636872, 75375912
  10. *
  11. * Licensed under the Apache License, Version 2.0 (the "License");
  12. * you may not use this file except in compliance with the License.
  13. * You may obtain a copy of the License at
  14. *
  15. * http://www.apache.org/licenses/LICENSE-2.0
  16. *
  17. * Unless required by applicable law or agreed to in writing, software
  18. * distributed under the License is distributed on an "AS IS" BASIS,
  19. * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  20. * See the License for the specific language governing permissions and
  21. * limitations under the License.
  22. */
  23. #pragma once
  24. #include "../hpsocket/GlobalDef.h"
  25. #include <unistd.h>
  26. #include <sched.h>
  27. #include <pthread.h>
  28. #include <sys/syscall.h>
  29. #include <sys/sysinfo.h>
  30. using namespace std;
  31. /* 最大工作线程数 */
  32. #define MAX_WORKER_THREAD_COUNT 512
  33. /* 默认对象缓存锁定时间 */
  34. #define DEFAULT_OBJECT_CACHE_LOCK_TIME (30 * 1000)
  35. /* 默认对象缓存池大小 */
  36. #define DEFAULT_OBJECT_CACHE_POOL_SIZE 600
  37. /* 默认对象缓存池回收阀值 */
  38. #define DEFAULT_OBJECT_CACHE_POOL_HOLD 600
  39. /* 默认内存块缓存容量 */
  40. #define DEFAULT_BUFFER_CACHE_CAPACITY 4096
  41. /* 默认内存块缓存池大小 */
  42. #define DEFAULT_BUFFER_CACHE_POOL_SIZE 1024
  43. /* 默认内存块缓存池回收阀值 */
  44. #define DEFAULT_BUFFER_CACHE_POOL_HOLD 1024
  45. #define SysGetSystemConfig sysconf
  46. #define SysGetSystemInfo sysinfo
  47. #if !defined(__ANDROID__)
  48. #define SysGetPageSize getpagesize
  49. #define SysGetNumberOfProcessors get_nprocs
  50. #else
  51. #define SysGetPageSize() sysconf(_SC_PAGESIZE)
  52. #define SysGetNumberOfProcessors() sysconf(_SC_NPROCESSORS_ONLN)
  53. #endif
  54. #define SYS_PAGE_SIZE GetSysPageSize()
  55. #define PROCESSOR_COUNT (::SysGetNumberOfProcessors())
  56. #define GetCurrentProcessId getpid
  57. #define SELF_PROCESS_ID (::GetCurrentProcessId())
  58. #define gettid() syscall(__NR_gettid)
  59. #define GetCurrentNativeThreadId() gettid()
  60. #define SELF_NATIVE_THREAD_ID (::GetCurrentNativeThreadId())
  61. #define GetCurrentThreadId pthread_self
  62. #define SELF_THREAD_ID (::GetCurrentThreadId())
  63. #define IsSameThread(tid1, tid2) pthread_equal((tid1), (tid2))
  64. #define IsSelfThread(tid) IsSameThread((tid), SELF_THREAD_ID)
  65. inline BOOL IsSameNativeThread(pid_t pid1, pid_t pid2)
  66. {return (pid1 == pid2);}
  67. #define IsSelfNativeThread(pid) IsSameNativeThread((pid), SELF_PROCESS_ID)
  68. #define DEFAULT_WORKER_THREAD_COUNT GetDefaultWorkerThreadCount()
  69. // Yield
  70. #if defined(__cplusplus)
  71. #include <thread>
  72. static inline void __atomic_yield()
  73. {
  74. std::this_thread::yield();
  75. }
  76. #elif defined(_WIN32)
  77. #define WIN32_LEAN_AND_MEAN
  78. #include <windows.h>
  79. static inline void __atomic_yield()
  80. {
  81. YieldProcessor();
  82. }
  83. #elif defined(__SSE2__)
  84. #include <emmintrin.h>
  85. static inline void __atomic_yield()
  86. {
  87. _mm_pause();
  88. }
  89. #elif (defined(__GNUC__) || defined(__clang__)) && \
  90. (defined(__x86_64__) || defined(__i386__) || defined(__arm__) || defined(__armel__) || defined(__ARMEL__) || \
  91. defined(__aarch64__) || defined(__powerpc__) || defined(__ppc__) || defined(__PPC__))
  92. #if defined(__x86_64__) || defined(__i386__)
  93. static inline void __atomic_yield()
  94. {
  95. __asm__ volatile ("pause" ::: "memory");
  96. }
  97. #elif defined(__aarch64__)
  98. static inline void __atomic_yield()
  99. {
  100. __asm__ volatile("wfe");
  101. }
  102. #elif (defined(__arm__) && __ARM_ARCH__ >= 7)
  103. static inline void __atomic_yield()
  104. {
  105. __asm__ volatile("yield" ::: "memory");
  106. }
  107. #elif defined(__powerpc__) || defined(__ppc__) || defined(__PPC__)
  108. static inline void __atomic_yield()
  109. {
  110. __asm__ __volatile__ ("or 27,27,27" ::: "memory");
  111. }
  112. #elif defined(__armel__) || defined(__ARMEL__)
  113. static inline void __atomic_yield()
  114. {
  115. __asm__ volatile ("nop" ::: "memory");
  116. }
  117. #endif
  118. #elif defined(__sun)
  119. // Fallback for other archs
  120. #include <synch.h>
  121. static inline void __atomic_yield()
  122. {
  123. smt_pause();
  124. }
  125. #elif defined(__wasi__)
  126. #include <sched.h>
  127. static inline void __atomic_yield()
  128. {
  129. sched_yield();
  130. }
  131. #else
  132. #include <unistd.h>
  133. static inline void __atomic_yield()
  134. {
  135. sleep(0);
  136. }
  137. #endif // Yield
  138. #define YieldProcessor __atomic_yield
  139. #define SwitchToThread sched_yield
  140. inline void __asm_nop() {__asm__ __volatile__("nop" : : : "memory");}
  141. inline void __asm_rep_nop() {__asm__ __volatile__("rep; nop" : : : "memory");}
  142. DWORD GetSysPageSize();
  143. DWORD GetKernelVersion();
  144. BOOL IsKernelVersionAbove(BYTE major, BYTE minor, BYTE revise);
  145. DWORD GetDefaultWorkerThreadCount();
  146. #if defined(__ANDROID__)
  147. #include<android/api-level.h>
  148. #if !defined(EFD_SEMAPHORE)
  149. #define EFD_SEMAPHORE 00000001
  150. #endif
  151. #define pthread_cancel(t)
  152. #if defined(__ANDROID_API__)
  153. #if (__ANDROID_API__ < 21)
  154. #define ppoll(fd, nfds, ptmspec, sig) poll((fd), (nfds), ((ptmspec) == nullptr) ? -1 : ((ptmspec)->tv_sec * 1000 + (ptmspec)->tv_nsec / 1000000))
  155. #define epoll_create1(flag) epoll_create(32)
  156. #define epoll_pwait(epfd, events, maxevents, timeout, sigmask) epoll_wait((epfd), (events), (maxevents), (timeout))
  157. #endif
  158. #endif
  159. #endif