10 #ifndef KMP_SAFE_C_API_H 11 #define KMP_SAFE_C_API_H 13 #include <type_traits> 14 #include "kmp_platform.h" 21 #if KMP_OS_WINDOWS && KMP_MSVC_COMPAT 23 #define RSIZE_MAX_STR (4UL << 10) // 4KB 26 #define KMP_ALLOCA _alloca 28 #define KMP_MEMCPY_S memcpy_s 29 #define KMP_SNPRINTF sprintf_s 30 #define KMP_SSCANF sscanf_s 31 #define KMP_STRCPY_S strcpy_s 32 #define KMP_STRNCPY_S strncpy_s 33 #define KMP_STRNCAT_S strncat_s 36 #define KMP_MEMCPY(dst, src, cnt) memcpy_s(dst, cnt, src, cnt) 38 template <typename T, bool B = std::is_array<T>::value>
39 struct kmp_get_rmax_t {};
40 template <
typename T>
struct kmp_get_rmax_t<T, false> {
41 static const size_t value = RSIZE_MAX_STR;
43 template <
typename T>
struct kmp_get_rmax_t<T, true> {
44 static const size_t value =
sizeof(T);
46 #define KMP_STRLEN(str) strnlen_s(str, kmp_get_rmax_t<decltype(str)>::value) 49 #define KMP_STRNCPY(dst, src, cnt) strncpy_s(dst, cnt, src, cnt) 52 #define KMP_VSNPRINTF(dst, cnt, fmt, arg) \ 53 vsnprintf_s(dst, cnt, _TRUNCATE, fmt, arg) 55 #else // KMP_OS_WINDOWS 59 #define KMP_ALLOCA alloca 60 #define KMP_MEMCPY_S(dst, bsz, src, cnt) memcpy(dst, src, cnt) 61 #define KMP_SNPRINTF snprintf 62 #define KMP_SSCANF sscanf 63 #define KMP_STRCPY_S(dst, bsz, src) strcpy(dst, src) 64 #define KMP_STRNCPY_S(dst, bsz, src, cnt) strncpy(dst, src, cnt) 65 #define KMP_STRNCAT_S(dst, bsz, src, cnt) strncat(dst, src, cnt) 66 #define KMP_VSNPRINTF vsnprintf 67 #define KMP_STRNCPY strncpy 68 #define KMP_STRLEN strlen 69 #define KMP_MEMCPY memcpy 71 #endif // KMP_OS_WINDOWS 74 static inline void __kmp_strncpy_truncate(
char *buffer,
size_t buf_size,
75 char const *src,
size_t src_size) {
76 if (src_size >= buf_size) {
77 src_size = buf_size - 1;
79 KMP_STRNCPY_S(buffer, buf_size, src, src_size);
80 buffer[src_size] =
'\0';
83 #endif // KMP_SAFE_C_API_H