| 1 |
/** |
| 2 |
* This file has no copyright assigned and is placed in the Public Domain. |
| 3 |
* This file is part of the mingw-w64 runtime package. |
| 4 |
* No warranty is given; refer to the file DISCLAIMER.PD within this package. |
| 5 |
*/ |
| 6 |
#ifndef __STRALIGN_H_ |
| 7 |
#define __STRALIGN_H_ |
| 8 |
|
| 9 |
#ifndef _STRALIGN_USE_SECURE_CRT |
| 10 |
#define _STRALIGN_USE_SECURE_CRT 0 |
| 11 |
#endif |
| 12 |
|
| 13 |
#ifdef __cplusplus |
| 14 |
extern "C" { |
| 15 |
#endif |
| 16 |
|
| 17 |
#ifndef WSTR_ALIGNED |
| 18 |
#if defined (__x86_64__) || defined (__arm__) |
| 19 |
#define WSTR_ALIGNED(s) TRUE |
| 20 |
#else |
| 21 |
#define WSTR_ALIGNED(s) (((DWORD_PTR)(s) & 1) == 0) |
| 22 |
#endif |
| 23 |
#endif |
| 24 |
|
| 25 |
#if defined(_X86_) |
| 26 |
#define ua_CharUpperW CharUpperW |
| 27 |
#define ua_lstrcmpiW lstrcmpiW |
| 28 |
#define ua_lstrcmpW lstrcmpW |
| 29 |
#define ua_lstrlenW lstrlenW |
| 30 |
#define ua_wcschr wcschr |
| 31 |
#define ua_wcsicmp wcsicmp |
| 32 |
#define ua_wcslen wcslen |
| 33 |
#define ua_wcsrchr wcsrchr |
| 34 |
|
| 35 |
PUWSTR ua_wcscpy(PUWSTR Destination,PCUWSTR Source); |
| 36 |
#if !defined (__CRT__NO_INLINE) && !defined (__CYGWIN__) |
| 37 |
__CRT_INLINE PUWSTR ua_wcscpy(PUWSTR Destination,PCUWSTR Source) { return wcscpy(Destination,Source); } |
| 38 |
#else |
| 39 |
#define ua_wcscpy wcscpy |
| 40 |
#endif |
| 41 |
|
| 42 |
#else /* not _X86_ : */ |
| 43 |
|
| 44 |
#ifndef WSTR_ALIGNED |
| 45 |
#define WSTR_ALIGNED(s) (((DWORD_PTR)(s) & (sizeof(WCHAR)-1))==0) |
| 46 |
#endif |
| 47 |
|
| 48 |
/* TODO: This method seems to be not present for x86-64. */ |
| 49 |
LPUWSTR WINAPI uaw_CharUpperW(LPUWSTR String); |
| 50 |
int WINAPI uaw_lstrcmpW(PCUWSTR String1,PCUWSTR String2); |
| 51 |
int WINAPI uaw_lstrcmpiW(PCUWSTR String1,PCUWSTR String2); |
| 52 |
int WINAPI uaw_lstrlenW(LPCUWSTR String); |
| 53 |
PUWSTR __cdecl uaw_wcschr(PCUWSTR String,WCHAR Character); |
| 54 |
PUWSTR __cdecl uaw_wcscpy(PUWSTR Destination,PCUWSTR Source); |
| 55 |
int __cdecl uaw_wcsicmp(PCUWSTR String1,PCUWSTR String2); |
| 56 |
size_t __cdecl uaw_wcslen(PCUWSTR String); |
| 57 |
PUWSTR __cdecl uaw_wcsrchr(PCUWSTR String,WCHAR Character); |
| 58 |
#ifdef CharUpper |
| 59 |
LPUWSTR ua_CharUpperW(LPUWSTR String); |
| 60 |
#ifndef __CRT__NO_INLINE |
| 61 |
__CRT_INLINE LPUWSTR ua_CharUpperW(LPUWSTR String) { |
| 62 |
if(WSTR_ALIGNED(String)) return CharUpperW((PWSTR)String); |
| 63 |
return uaw_CharUpperW(String); |
| 64 |
} |
| 65 |
#endif /* !__CRT__NO_INLINE */ |
| 66 |
#endif /* CharUpper */ |
| 67 |
|
| 68 |
#ifdef lstrcmp |
| 69 |
int ua_lstrcmpW(LPCUWSTR String1,LPCUWSTR String2); |
| 70 |
#endif |
| 71 |
#ifdef lstrcmpi |
| 72 |
int ua_lstrcmpiW(LPCUWSTR String1,LPCUWSTR String2); |
| 73 |
#endif |
| 74 |
#ifdef lstrlen |
| 75 |
int ua_lstrlenW(LPCUWSTR String); |
| 76 |
#endif |
| 77 |
|
| 78 |
#ifndef __CRT__NO_INLINE |
| 79 |
#ifdef lstrcmp |
| 80 |
__CRT_INLINE int ua_lstrcmpW(LPCUWSTR String1,LPCUWSTR String2) { |
| 81 |
if(WSTR_ALIGNED(String1) && WSTR_ALIGNED(String2)) |
| 82 |
return lstrcmpW((LPCWSTR)String1,(LPCWSTR)String2); |
| 83 |
return uaw_lstrcmpW(String1,String2); |
| 84 |
} |
| 85 |
#endif |
| 86 |
|
| 87 |
#ifdef lstrcmpi |
| 88 |
__CRT_INLINE int ua_lstrcmpiW(LPCUWSTR String1,LPCUWSTR String2) { |
| 89 |
if(WSTR_ALIGNED(String1) && WSTR_ALIGNED(String2)) |
| 90 |
return lstrcmpiW((LPCWSTR)String1,(LPCWSTR)String2); |
| 91 |
return uaw_lstrcmpiW(String1,String2); |
| 92 |
} |
| 93 |
#endif |
| 94 |
|
| 95 |
#ifdef lstrlen |
| 96 |
__CRT_INLINE int ua_lstrlenW(LPCUWSTR String) { |
| 97 |
if(WSTR_ALIGNED(String)) return lstrlenW((PCWSTR)String); |
| 98 |
return uaw_lstrlenW(String); |
| 99 |
} |
| 100 |
#endif |
| 101 |
#endif /* !__CRT__NO_INLINE */ |
| 102 |
|
| 103 |
#if defined(_WSTRING_DEFINED) |
| 104 |
#ifdef _WConst_return |
| 105 |
typedef _WConst_return WCHAR UNALIGNED *PUWSTR_C; |
| 106 |
#else |
| 107 |
typedef WCHAR UNALIGNED *PUWSTR_C; |
| 108 |
#endif |
| 109 |
|
| 110 |
PUWSTR_C ua_wcschr(PCUWSTR String,WCHAR Character); |
| 111 |
PUWSTR_C ua_wcsrchr(PCUWSTR String,WCHAR Character); |
| 112 |
#if defined(__cplusplus) && defined(_WConst_Return) |
| 113 |
PUWSTR ua_wcschr(PUWSTR String,WCHAR Character); |
| 114 |
PUWSTR ua_wcsrchr(PUWSTR String,WCHAR Character); |
| 115 |
#endif |
| 116 |
PUWSTR ua_wcscpy(PUWSTR Destination,PCUWSTR Source); |
| 117 |
size_t ua_wcslen(PCUWSTR String); |
| 118 |
|
| 119 |
#ifndef __CRT__NO_INLINE |
| 120 |
__CRT_INLINE PUWSTR_C ua_wcschr(PCUWSTR String,WCHAR Character) { |
| 121 |
if(WSTR_ALIGNED(String)) return (PUWSTR_C)wcschr((PCWSTR)String,Character); |
| 122 |
return (PUWSTR_C)uaw_wcschr(String,Character); |
| 123 |
} |
| 124 |
__CRT_INLINE PUWSTR_C ua_wcsrchr(PCUWSTR String,WCHAR Character) { |
| 125 |
if(WSTR_ALIGNED(String)) return (PUWSTR_C)wcsrchr((PCWSTR)String,Character); |
| 126 |
return (PUWSTR_C)uaw_wcsrchr(String,Character); |
| 127 |
} |
| 128 |
#if defined(__cplusplus) && defined(_WConst_Return) |
| 129 |
__CRT_INLINE PUWSTR ua_wcschr(PUWSTR String,WCHAR Character) { |
| 130 |
if(WSTR_ALIGNED(String)) return wcscpy((PWSTR)Destination,(PCWSTR)Source); |
| 131 |
return uaw_wcscpy(Destination,Source); |
| 132 |
} |
| 133 |
__CRT_INLINE PUWSTR ua_wcsrchr(PUWSTR String,WCHAR Character) { |
| 134 |
if(WSTR_ALIGNED(String)) return wcsrchr(String,Character); |
| 135 |
return uaw_wcsrchr((PCUWSTR)String,Character); |
| 136 |
} |
| 137 |
#endif |
| 138 |
|
| 139 |
__CRT_INLINE PUWSTR ua_wcscpy(PUWSTR Destination,PCUWSTR Source) { |
| 140 |
if(WSTR_ALIGNED(Source) && WSTR_ALIGNED(Destination)) |
| 141 |
return wcscpy((PWSTR)Destination,(PCWSTR)Source); |
| 142 |
return uaw_wcscpy(Destination,Source); |
| 143 |
} |
| 144 |
__CRT_INLINE size_t ua_wcslen(PCUWSTR String) { |
| 145 |
if(WSTR_ALIGNED(String)) return wcslen((PCWSTR)String); |
| 146 |
return uaw_wcslen(String); |
| 147 |
} |
| 148 |
#endif /* !__CRT__NO_INLINE */ |
| 149 |
#endif /* _X86_ */ |
| 150 |
int ua_wcsicmp(LPCUWSTR String1,LPCUWSTR String2); |
| 151 |
|
| 152 |
#ifndef __CRT__NO_INLINE |
| 153 |
__CRT_INLINE int ua_wcsicmp(LPCUWSTR String1,LPCUWSTR String2) { |
| 154 |
if(WSTR_ALIGNED(String1) && WSTR_ALIGNED(String2)) |
| 155 |
return _wcsicmp((LPCWSTR)String1,(LPCWSTR)String2); |
| 156 |
return uaw_wcsicmp(String1,String2); |
| 157 |
} |
| 158 |
#endif /* !__CRT__NO_INLINE */ |
| 159 |
#endif /* _WSTRING_DEFINED */ |
| 160 |
|
| 161 |
#ifndef __UA_WCSLEN |
| 162 |
#define __UA_WCSLEN ua_wcslen |
| 163 |
#endif |
| 164 |
|
| 165 |
#define __UA_WSTRSIZE(s) ((__UA_WCSLEN(s)+1)*sizeof(WCHAR)) |
| 166 |
#define __UA_STACKCOPY(p,s) memcpy(_alloca(s),p,s) |
| 167 |
|
| 168 |
#if defined (__x86_64__) || defined (__arm__) || defined (_X86_) |
| 169 |
#define WSTR_ALIGNED_STACK_COPY(d,s) (*(d) = (PCWSTR)(s)) |
| 170 |
#else |
| 171 |
#define WSTR_ALIGNED_STACK_COPY(d,s) { PCUWSTR __ua_src; ULONG __ua_size; PWSTR __ua_dst; __ua_src = (s); if(WSTR_ALIGNED(__ua_src)) { __ua_dst = (PWSTR)__ua_src; } else { __ua_size = __UA_WSTRSIZE(__ua_src); __ua_dst = (PWSTR)_alloca(__ua_size); memcpy(__ua_dst,__ua_src,__ua_size); } *(d) = (PCWSTR)__ua_dst; } |
| 172 |
#endif |
| 173 |
|
| 174 |
#define ASTR_ALIGNED_STACK_COPY(d,s) (*(d) = (PCSTR)(s)) |
| 175 |
|
| 176 |
#if !defined (_X86_) && !defined (__x86_64__) && !defined (__arm__) |
| 177 |
#define __UA_STRUC_ALIGNED(t,s) (((DWORD_PTR)(s) & (TYPE_ALIGNMENT(t)-1))==0) |
| 178 |
#define STRUC_ALIGNED_STACK_COPY(t,s) __UA_STRUC_ALIGNED(t,s) ? ((t const *)(s)) : ((t const *)__UA_STACKCOPY((s),sizeof(t))) |
| 179 |
#else |
| 180 |
#define STRUC_ALIGNED_STACK_COPY(t,s) ((CONST t *)(s)) |
| 181 |
#endif |
| 182 |
|
| 183 |
#if defined(UNICODE) |
| 184 |
#define TSTR_ALIGNED_STACK_COPY(d,s) WSTR_ALIGNED_STACK_COPY(d,s) |
| 185 |
#define TSTR_ALIGNED(x) WSTR_ALIGNED(x) |
| 186 |
#define ua_CharUpper ua_CharUpperW |
| 187 |
#define ua_lstrcmp ua_lstrcmpW |
| 188 |
#define ua_lstrcmpi ua_lstrcmpiW |
| 189 |
#define ua_lstrlen ua_lstrlenW |
| 190 |
#define ua_tcscpy ua_wcscpy |
| 191 |
#else |
| 192 |
#define TSTR_ALIGNED_STACK_COPY(d,s) ASTR_ALIGNED_STACK_COPY(d,s) |
| 193 |
#define TSTR_ALIGNED(x) TRUE |
| 194 |
#define ua_CharUpper CharUpperA |
| 195 |
#define ua_lstrcmp lstrcmpA |
| 196 |
#define ua_lstrcmpi lstrcmpiA |
| 197 |
#define ua_lstrlen lstrlenA |
| 198 |
#define ua_tcscpy strcpy |
| 199 |
#endif |
| 200 |
|
| 201 |
#ifdef __cplusplus |
| 202 |
} |
| 203 |
#endif |
| 204 |
|
| 205 |
#include <sec_api/stralign_s.h> |
| 206 |
#endif |
