/** * This file has no copyright assigned and is placed in the Public Domain. * This file is part of the mingw-w64 runtime package. * No warranty is given; refer to the file DISCLAIMER.PD within this package. */ #ifndef __STRALIGN_H_ #define __STRALIGN_H_ #ifndef _STRALIGN_USE_SECURE_CRT #define _STRALIGN_USE_SECURE_CRT 0 #endif #ifdef __cplusplus extern "C" { #endif #if defined (__amd64__) || defined (__arm__) #define WSTR_ALIGNED(s) TRUE #else #define WSTR_ALIGNED(s) (((DWORD_PTR)(s) & 1) == 0) #endif #if defined(_X86_) #define ua_CharUpperW CharUpperW #define ua_lstrcmpiW lstrcmpiW #define ua_lstrcmpW lstrcmpW #define ua_lstrlenW lstrlenW #define ua_wcschr wcschr #define ua_wcsicmp wcsicmp #define ua_wcslen wcslen #define ua_wcsrchr wcsrchr PUWSTR ua_wcscpy(PUWSTR Destination,PCUWSTR Source); #if !defined (__CRT__NO_INLINE) && !defined (__CYGWIN__) __CRT_INLINE PUWSTR ua_wcscpy(PUWSTR Destination,PCUWSTR Source) { return wcscpy(Destination,Source); } #else #define ua_wcscpy wcscpy #endif #else /* not _X86_ : */ #define WSTR_ALIGNED(s) (((DWORD_PTR)(s) & (sizeof(WCHAR)-1))==0) /* TODO: This method seems to be not present for amd64. */ LPUWSTR WINAPI uaw_CharUpperW(LPUWSTR String); int WINAPI uaw_lstrcmpW(PCUWSTR String1,PCUWSTR String2); int WINAPI uaw_lstrcmpiW(PCUWSTR String1,PCUWSTR String2); int WINAPI uaw_lstrlenW(LPCUWSTR String); PUWSTR __cdecl uaw_wcschr(PCUWSTR String,WCHAR Character); PUWSTR __cdecl uaw_wcscpy(PUWSTR Destination,PCUWSTR Source); int __cdecl uaw_wcsicmp(PCUWSTR String1,PCUWSTR String2); size_t __cdecl uaw_wcslen(PCUWSTR String); PUWSTR __cdecl uaw_wcsrchr(PCUWSTR String,WCHAR Character); #ifdef CharUpper LPUWSTR ua_CharUpperW(LPUWSTR String); #ifndef __CRT__NO_INLINE __CRT_INLINE LPUWSTR ua_CharUpperW(LPUWSTR String) { if(WSTR_ALIGNED(String)) return CharUpperW((PWSTR)String); return uaw_CharUpperW(String); } #endif /* !__CRT__NO_INLINE */ #endif /* CharUpper */ #ifdef lstrcmp int ua_lstrcmpW(LPCUWSTR String1,LPCUWSTR String2); #endif #ifdef lstrcmpi int ua_lstrcmpiW(LPCUWSTR String1,LPCUWSTR String2); #endif #ifdef lstrlen int ua_lstrlenW(LPCUWSTR String); #endif #ifndef __CRT__NO_INLINE #ifdef lstrcmp __CRT_INLINE int ua_lstrcmpW(LPCUWSTR String1,LPCUWSTR String2) { if(WSTR_ALIGNED(String1) && WSTR_ALIGNED(String2)) return lstrcmpW((LPCWSTR)String1,(LPCWSTR)String2); return uaw_lstrcmpW(String1,String2); } #endif #ifdef lstrcmpi __CRT_INLINE int ua_lstrcmpiW(LPCUWSTR String1,LPCUWSTR String2) { if(WSTR_ALIGNED(String1) && WSTR_ALIGNED(String2)) return lstrcmpiW((LPCWSTR)String1,(LPCWSTR)String2); return uaw_lstrcmpiW(String1,String2); } #endif #ifdef lstrlen __CRT_INLINE int ua_lstrlenW(LPCUWSTR String) { if(WSTR_ALIGNED(String)) return lstrlenW((PCWSTR)String); return uaw_lstrlenW(String); } #endif #endif /* !__CRT__NO_INLINE */ #if defined(_WSTRING_DEFINED) #ifdef _WConst_return typedef _WConst_return WCHAR UNALIGNED *PUWSTR_C; #else typedef WCHAR UNALIGNED *PUWSTR_C; #endif PUWSTR_C ua_wcschr(PCUWSTR String,WCHAR Character); PUWSTR_C ua_wcsrchr(PCUWSTR String,WCHAR Character); #if defined(__cplusplus) && defined(_WConst_Return) PUWSTR ua_wcschr(PUWSTR String,WCHAR Character); PUWSTR ua_wcsrchr(PUWSTR String,WCHAR Character); #endif PUWSTR ua_wcscpy(PUWSTR Destination,PCUWSTR Source); size_t ua_wcslen(PCUWSTR String); #ifndef __CRT__NO_INLINE __CRT_INLINE PUWSTR_C ua_wcschr(PCUWSTR String,WCHAR Character) { if(WSTR_ALIGNED(String)) return wcschr((PCWSTR)String,Character); return (PUWSTR_C)uaw_wcschr(String,Character); } __CRT_INLINE PUWSTR_C ua_wcsrchr(PCUWSTR String,WCHAR Character) { if(WSTR_ALIGNED(String)) return wcsrchr((PCWSTR)String,Character); return (PUWSTR_C)uaw_wcsrchr(String,Character); } #if defined(__cplusplus) && defined(_WConst_Return) __CRT_INLINE PUWSTR ua_wcschr(PUWSTR String,WCHAR Character) { if(WSTR_ALIGNED(String)) return wcscpy((PWSTR)Destination,(PCWSTR)Source); return uaw_wcscpy(Destination,Source); } __CRT_INLINE PUWSTR ua_wcsrchr(PUWSTR String,WCHAR Character) { if(WSTR_ALIGNED(String)) return wcsrchr(String,Character); return uaw_wcsrchr((PCUWSTR)String,Character); } #endif __CRT_INLINE PUWSTR ua_wcscpy(PUWSTR Destination,PCUWSTR Source) { if(WSTR_ALIGNED(Source) && WSTR_ALIGNED(Destination)) return wcscpy((PWSTR)Destination,(PCWSTR)Source); return uaw_wcscpy(Destination,Source); } __CRT_INLINE size_t ua_wcslen(PCUWSTR String) { if(WSTR_ALIGNED(String)) return wcslen((PCWSTR)String); return uaw_wcslen(String); } #endif /* !__CRT__NO_INLINE */ #endif /* _X86_ */ int ua_wcsicmp(LPCUWSTR String1,LPCUWSTR String2); #ifndef __CRT__NO_INLINE __CRT_INLINE int ua_wcsicmp(LPCUWSTR String1,LPCUWSTR String2) { if(WSTR_ALIGNED(String1) && WSTR_ALIGNED(String2)) return _wcsicmp((LPCWSTR)String1,(LPCWSTR)String2); return uaw_wcsicmp(String1,String2); } #endif /* !__CRT__NO_INLINE */ #endif /* _WSTRING_DEFINED */ #ifndef __UA_WCSLEN #define __UA_WCSLEN ua_wcslen #endif #define __UA_WSTRSIZE(s) ((__UA_WCSLEN(s)+1)*sizeof(WCHAR)) #define __UA_STACKCOPY(p,s) memcpy(_alloca(s),p,s) #if defined (__amd64__) || defined (__arm__) || defined (_X86_) #define WSTR_ALIGNED_STACK_COPY(d,s) (*(d) = (PCWSTR)(s)) #else #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; } #endif #define ASTR_ALIGNED_STACK_COPY(d,s) (*(d) = (PCSTR)(s)) #if !defined (_X86_) && !defined (__amd64__) && !defined (__arm__) #define __UA_STRUC_ALIGNED(t,s) (((DWORD_PTR)(s) & (TYPE_ALIGNMENT(t)-1))==0) #define STRUC_ALIGNED_STACK_COPY(t,s) __UA_STRUC_ALIGNED(t,s) ? ((t const *)(s)) : ((t const *)__UA_STACKCOPY((s),sizeof(t))) #else #define STRUC_ALIGNED_STACK_COPY(t,s) ((CONST t *)(s)) #endif #if defined(UNICODE) #define TSTR_ALIGNED_STACK_COPY(d,s) WSTR_ALIGNED_STACK_COPY(d,s) #define TSTR_ALIGNED(x) WSTR_ALIGNED(x) #define ua_CharUpper ua_CharUpperW #define ua_lstrcmp ua_lstrcmpW #define ua_lstrcmpi ua_lstrcmpiW #define ua_lstrlen ua_lstrlenW #define ua_tcscpy ua_wcscpy #else #define TSTR_ALIGNED_STACK_COPY(d,s) ASTR_ALIGNED_STACK_COPY(d,s) #define TSTR_ALIGNED(x) TRUE #define ua_CharUpper CharUpperA #define ua_lstrcmp lstrcmpA #define ua_lstrcmpi lstrcmpiA #define ua_lstrlen lstrlenA #define ua_tcscpy strcpy #endif #ifdef __cplusplus } #endif #include <sec_api/stralign_s.h> #endif