| 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 _NTSTRSAFE_H_INCLUDED_ | 
 
 
 
 
 | 7 | #define _NTSTRSAFE_H_INCLUDED_ | 
 
 
 
 
 | 8 |  | 
 
 
 
 
 | 9 | #include <_mingw_unicode.h> | 
 
 
 
 
 | 10 | #include <stdio.h> | 
 
 
 
 
 | 11 | #include <string.h> | 
 
 
 
 
 | 12 | #include <stdarg.h> | 
 
 
 
 
 | 13 | #include <specstrings.h> | 
 
 
 
 
 | 14 |  | 
 
 
 
 
 | 15 | #if defined(__CRT__NO_INLINE) && !defined(__CRT_STRSAFE_IMPL) | 
 
 
 
 
 | 16 | #define __STRSAFE__NO_INLINE | 
 
 
 
 
 | 17 | #endif | 
 
 
 
 
 | 18 |  | 
 
 
 
 
 | 19 | #ifndef _SIZE_T_DEFINED | 
 
 
 
 
 | 20 | #define _SIZE_T_DEFINED | 
 
 
 
 
 | 21 | #undef size_t | 
 
 
 
 
 | 22 | #ifdef _WIN64 | 
 
 
 
 
 | 23 | __MINGW_EXTENSION typedef unsigned __int64 size_t; | 
 
 
 
 
 | 24 | #else | 
 
 
 
 
 | 25 | typedef unsigned int size_t; | 
 
 
 
 
 | 26 | #endif | 
 
 
 
 
 | 27 | #endif | 
 
 
 
 
 | 28 |  | 
 
 
 
 
 | 29 | #ifndef _WCHAR_T_DEFINED | 
 
 
 
 
 | 30 | #define _WCHAR_T_DEFINED | 
 
 
 
 
 | 31 | typedef unsigned short wchar_t; | 
 
 
 
 
 | 32 | #endif | 
 
 
 
 
 | 33 |  | 
 
 
 
 
 | 34 | #ifndef _NTSTATUS_DEFINED | 
 
 
 
 
 | 35 | #define _NTSTATUS_DEFINED | 
 
 
 
 
 | 36 | typedef __LONG32 NTSTATUS; | 
 
 
 
 
 | 37 | #endif | 
 
 
 
 
 | 38 |  | 
 
 
 
 
 | 39 | #ifndef C_ASSERT | 
 
 
 
 
 | 40 | #ifdef _MSC_VER | 
 
 
 
 
 | 41 | # define C_ASSERT(e) typedef char __C_ASSERT__[(e)?1:-1] | 
 
 
 
 
 | 42 | #else | 
 
 
 
 
 | 43 | # define C_ASSERT(e) extern void __C_ASSERT__(int [(e)?1:-1]) | 
 
 
 
 
 | 44 | #endif | 
 
 
 
 
 | 45 | #endif /* C_ASSERT */ | 
 
 
 
 
 | 46 |  | 
 
 
 
 
 | 47 | /* extern removed for C mode to avoid double extern qualifier from __CRT_INLINE */ | 
 
 
 
 
 | 48 | #ifdef __cplusplus | 
 
 
 
 
 | 49 | #define _STRSAFE_EXTERN_C extern "C" | 
 
 
 
 
 | 50 | #else | 
 
 
 
 
 | 51 | #define _STRSAFE_EXTERN_C extern | 
 
 
 
 
 | 52 | #endif | 
 
 
 
 
 | 53 |  | 
 
 
 
 
 | 54 | #ifndef WINAPI | 
 
 
 
 
 | 55 | #if defined(_ARM_) | 
 
 
 
 
 | 56 | #define WINAPI | 
 
 
 
 
 | 57 | #else | 
 
 
 
 
 | 58 | #define WINAPI __stdcall | 
 
 
 
 
 | 59 | #endif | 
 
 
 
 
 | 60 | #endif | 
 
 
 
 
 | 61 |  | 
 
 
 
 
 | 62 | #if !defined(__CRT__NO_INLINE) && !defined(__CRT_STRSAFE_IMPL) | 
 
 
 
 
 | 63 | #define NTSTRSAFEDDI _STRSAFE_EXTERN_C __inline NTSTATUS WINAPI | 
 
 
 
 
 | 64 | /* Variadic functions can't be __stdcall.  */ | 
 
 
 
 
 | 65 | #define NTSTRSAFEDDIV _STRSAFE_EXTERN_C __inline NTSTATUS | 
 
 
 
 
 | 66 | #else | 
 
 
 
 
 | 67 | #define NTSTRSAFEDDI _STRSAFE_EXTERN_C NTSTATUS WINAPI | 
 
 
 
 
 | 68 | /* Variadic functions can't be __stdcall.  */ | 
 
 
 
 
 | 69 | #define NTSTRSAFEDDIV _STRSAFE_EXTERN_C NTSTATUS | 
 
 
 
 
 | 70 | #endif | 
 
 
 
 
 | 71 |  | 
 
 
 
 
 | 72 | #define NTSTRSAFE_MAX_CCH 2147483647 | 
 
 
 
 
 | 73 |  | 
 
 
 
 
 | 74 | #define STRSAFE_IGNORE_NULLS 0x00000100 | 
 
 
 
 
 | 75 | #define STRSAFE_FILL_BEHIND_NULL 0x00000200 | 
 
 
 
 
 | 76 | #define STRSAFE_FILL_ON_FAILURE 0x00000400 | 
 
 
 
 
 | 77 | #define STRSAFE_NULL_ON_FAILURE 0x00000800 | 
 
 
 
 
 | 78 | #define STRSAFE_NO_TRUNCATION 0x00001000 | 
 
 
 
 
 | 79 | #define STRSAFE_IGNORE_NULL_UNICODE_STRINGS 0x00010000 | 
 
 
 
 
 | 80 | #define STRSAFE_UNICODE_STRING_DEST_NULL_TERMINATED 0x00020000 | 
 
 
 
 
 | 81 |  | 
 
 
 
 
 | 82 | #define STRSAFE_VALID_FLAGS (0x000000FF | STRSAFE_IGNORE_NULLS | STRSAFE_FILL_BEHIND_NULL | STRSAFE_FILL_ON_FAILURE | STRSAFE_NULL_ON_FAILURE | STRSAFE_NO_TRUNCATION) | 
 
 
 
 
 | 83 | #define STRSAFE_UNICODE_STRING_VALID_FLAGS (STRSAFE_VALID_FLAGS | STRSAFE_IGNORE_NULL_UNICODE_STRINGS | STRSAFE_UNICODE_STRING_DEST_NULL_TERMINATED) | 
 
 
 
 
 | 84 |  | 
 
 
 
 
 | 85 | #define STRSAFE_FILL_BYTE(x) ((unsigned __LONG32)((x & 0x000000FF) | STRSAFE_FILL_BEHIND_NULL)) | 
 
 
 
 
 | 86 | #define STRSAFE_FAILURE_BYTE(x) ((unsigned __LONG32)((x & 0x000000FF) | STRSAFE_FILL_ON_FAILURE)) | 
 
 
 
 
 | 87 |  | 
 
 
 
 
 | 88 | #define STRSAFE_GET_FILL_PATTERN(dwFlags) ((int)(dwFlags & 0x000000FF)) | 
 
 
 
 
 | 89 |  | 
 
 
 
 
 | 90 | typedef char *NTSTRSAFE_PSTR; | 
 
 
 
 
 | 91 | typedef const char *NTSTRSAFE_PCSTR; | 
 
 
 
 
 | 92 | typedef wchar_t *NTSTRSAFE_PWSTR; | 
 
 
 
 
 | 93 | typedef const wchar_t *NTSTRSAFE_PCWSTR; | 
 
 
 
 
 | 94 |  | 
 
 
 
 
 | 95 | NTSTRSAFEDDI RtlStringCopyWorkerA(NTSTRSAFE_PSTR pszDest,size_t cchDest,NTSTRSAFE_PCSTR pszSrc); | 
 
 
 
 
 | 96 | NTSTRSAFEDDI RtlStringCopyWorkerW(NTSTRSAFE_PWSTR pszDest,size_t cchDest,NTSTRSAFE_PCWSTR pszSrc); | 
 
 
 
 
 | 97 | NTSTRSAFEDDI RtlStringCopyExWorkerA(NTSTRSAFE_PSTR pszDest,size_t cchDest,size_t cbDest,NTSTRSAFE_PCSTR pszSrc,NTSTRSAFE_PSTR *ppszDestEnd,size_t *pcchRemaining,unsigned __LONG32 dwFlags); | 
 
 
 
 
 | 98 | NTSTRSAFEDDI RtlStringCopyExWorkerW(NTSTRSAFE_PWSTR pszDest,size_t cchDest,size_t cbDest,NTSTRSAFE_PCWSTR pszSrc,NTSTRSAFE_PWSTR *ppszDestEnd,size_t *pcchRemaining,unsigned __LONG32 dwFlags); | 
 
 
 
 
 | 99 | NTSTRSAFEDDI RtlStringCopyNWorkerA(NTSTRSAFE_PSTR pszDest,size_t cchDest,NTSTRSAFE_PCSTR pszSrc,size_t cchToCopy); | 
 
 
 
 
 | 100 | NTSTRSAFEDDI RtlStringCopyNWorkerW(NTSTRSAFE_PWSTR pszDest,size_t cchDest,NTSTRSAFE_PCWSTR pszSrc,size_t cchToCopy); | 
 
 
 
 
 | 101 | NTSTRSAFEDDI RtlStringCopyNExWorkerA(NTSTRSAFE_PSTR pszDest,size_t cchDest,size_t cbDest,NTSTRSAFE_PCSTR pszSrc,size_t cchToCopy,NTSTRSAFE_PSTR *ppszDestEnd,size_t *pcchRemaining,unsigned __LONG32 dwFlags); | 
 
 
 
 
 | 102 | NTSTRSAFEDDI RtlStringCopyNExWorkerW(NTSTRSAFE_PWSTR pszDest,size_t cchDest,size_t cbDest,NTSTRSAFE_PCWSTR pszSrc,size_t cchToCopy,NTSTRSAFE_PWSTR *ppszDestEnd,size_t *pcchRemaining,unsigned __LONG32 dwFlags); | 
 
 
 
 
 | 103 | NTSTRSAFEDDI RtlStringCatWorkerA(NTSTRSAFE_PSTR pszDest,size_t cchDest,NTSTRSAFE_PCSTR pszSrc); | 
 
 
 
 
 | 104 | NTSTRSAFEDDI RtlStringCatWorkerW(NTSTRSAFE_PWSTR pszDest,size_t cchDest,NTSTRSAFE_PCWSTR pszSrc); | 
 
 
 
 
 | 105 | NTSTRSAFEDDI RtlStringCatExWorkerA(NTSTRSAFE_PSTR pszDest,size_t cchDest,size_t cbDest,NTSTRSAFE_PCSTR pszSrc,NTSTRSAFE_PSTR *ppszDestEnd,size_t *pcchRemaining,unsigned __LONG32 dwFlags); | 
 
 
 
 
 | 106 | NTSTRSAFEDDI RtlStringCatExWorkerW(NTSTRSAFE_PWSTR pszDest,size_t cchDest,size_t cbDest,NTSTRSAFE_PCWSTR pszSrc,NTSTRSAFE_PWSTR *ppszDestEnd,size_t *pcchRemaining,unsigned __LONG32 dwFlags); | 
 
 
 
 
 | 107 | NTSTRSAFEDDI RtlStringCatNWorkerA(NTSTRSAFE_PSTR pszDest,size_t cchDest,NTSTRSAFE_PCSTR pszSrc,size_t cchToAppend); | 
 
 
 
 
 | 108 | NTSTRSAFEDDI RtlStringCatNWorkerW(NTSTRSAFE_PWSTR pszDest,size_t cchDest,NTSTRSAFE_PCWSTR pszSrc,size_t cchToAppend); | 
 
 
 
 
 | 109 | NTSTRSAFEDDI RtlStringCatNExWorkerA(NTSTRSAFE_PSTR pszDest,size_t cchDest,size_t cbDest,NTSTRSAFE_PCSTR pszSrc,size_t cchToAppend,NTSTRSAFE_PSTR *ppszDestEnd,size_t *pcchRemaining,unsigned __LONG32 dwFlags); | 
 
 
 
 
 | 110 | NTSTRSAFEDDI RtlStringCatNExWorkerW(NTSTRSAFE_PWSTR pszDest,size_t cchDest,size_t cbDest,NTSTRSAFE_PCWSTR pszSrc,size_t cchToAppend,NTSTRSAFE_PWSTR *ppszDestEnd,size_t *pcchRemaining,unsigned __LONG32 dwFlags); | 
 
 
 
 
 | 111 | NTSTRSAFEDDI RtlStringVPrintfWorkerA(NTSTRSAFE_PSTR pszDest,size_t cchDest,NTSTRSAFE_PCSTR pszFormat,va_list argList); | 
 
 
 
 
 | 112 | NTSTRSAFEDDI RtlStringVPrintfWorkerW(NTSTRSAFE_PWSTR pszDest,size_t cchDest,NTSTRSAFE_PCWSTR pszFormat,va_list argList); | 
 
 
 
 
 | 113 | NTSTRSAFEDDI RtlStringVPrintfExWorkerA(NTSTRSAFE_PSTR pszDest,size_t cchDest,size_t cbDest,NTSTRSAFE_PSTR *ppszDestEnd,size_t *pcchRemaining,unsigned __LONG32 dwFlags,NTSTRSAFE_PCSTR pszFormat,va_list argList); | 
 
 
 
 
 | 114 | NTSTRSAFEDDI RtlStringVPrintfExWorkerW(NTSTRSAFE_PWSTR pszDest,size_t cchDest,size_t cbDest,NTSTRSAFE_PWSTR *ppszDestEnd,size_t *pcchRemaining,unsigned __LONG32 dwFlags,NTSTRSAFE_PCWSTR pszFormat,va_list argList); | 
 
 
 
 
 | 115 | NTSTRSAFEDDI RtlStringLengthWorkerA(NTSTRSAFE_PCSTR psz,size_t cchMax,size_t *pcchLength); | 
 
 
 
 
 | 116 | NTSTRSAFEDDI RtlStringLengthWorkerW(NTSTRSAFE_PCWSTR psz,size_t cchMax,size_t *pcchLength); | 
 
 
 
 
 | 117 |  | 
 
 
 
 
 | 118 | #define RtlStringCchCopy __MINGW_NAME_AW(RtlStringCchCopy) | 
 
 
 
 
 | 119 |  | 
 
 
 
 
 | 120 | NTSTRSAFEDDI RtlStringCchCopyA(NTSTRSAFE_PSTR pszDest,size_t cchDest,NTSTRSAFE_PCSTR pszSrc); | 
 
 
 
 
 | 121 | NTSTRSAFEDDI RtlStringCchCopyW(NTSTRSAFE_PWSTR pszDest,size_t cchDest,NTSTRSAFE_PCWSTR pszSrc); | 
 
 
 
 
 | 122 |  | 
 
 
 
 
 | 123 | #ifndef __STRSAFE__NO_INLINE | 
 
 
 
 
 | 124 | NTSTRSAFEDDI RtlStringCchCopyA(NTSTRSAFE_PSTR pszDest,size_t cchDest,NTSTRSAFE_PCSTR pszSrc) { | 
 
 
 
 
 | 125 | return (cchDest > NTSTRSAFE_MAX_CCH ? STATUS_INVALID_PARAMETER : RtlStringCopyWorkerA(pszDest,cchDest,pszSrc)); | 
 
 
 
 
 | 126 | } | 
 
 
 
 
 | 127 |  | 
 
 
 
 
 | 128 | NTSTRSAFEDDI RtlStringCchCopyW(NTSTRSAFE_PWSTR pszDest,size_t cchDest,NTSTRSAFE_PCWSTR pszSrc) { | 
 
 
 
 
 | 129 | if(cchDest > NTSTRSAFE_MAX_CCH) return STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 130 | return RtlStringCopyWorkerW(pszDest,cchDest,pszSrc); | 
 
 
 
 
 | 131 | } | 
 
 
 
 
 | 132 | #endif /* !__STRSAFE__NO_INLINE */ | 
 
 
 
 
 | 133 |  | 
 
 
 
 
 | 134 | #define RtlStringCbCopy __MINGW_NAME_AW(RtlStringCbCopy) | 
 
 
 
 
 | 135 |  | 
 
 
 
 
 | 136 | NTSTRSAFEDDI RtlStringCbCopyA(NTSTRSAFE_PSTR pszDest,size_t cbDest,NTSTRSAFE_PCSTR pszSrc); | 
 
 
 
 
 | 137 | NTSTRSAFEDDI RtlStringCbCopyW(NTSTRSAFE_PWSTR pszDest,size_t cbDest,NTSTRSAFE_PCWSTR pszSrc); | 
 
 
 
 
 | 138 |  | 
 
 
 
 
 | 139 | #ifndef __STRSAFE__NO_INLINE | 
 
 
 
 
 | 140 | NTSTRSAFEDDI RtlStringCbCopyA(NTSTRSAFE_PSTR pszDest,size_t cbDest,NTSTRSAFE_PCSTR pszSrc) { | 
 
 
 
 
 | 141 | if(cbDest > NTSTRSAFE_MAX_CCH) return STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 142 | return RtlStringCopyWorkerA(pszDest,cbDest,pszSrc); | 
 
 
 
 
 | 143 | } | 
 
 
 
 
 | 144 |  | 
 
 
 
 
 | 145 | NTSTRSAFEDDI RtlStringCbCopyW(NTSTRSAFE_PWSTR pszDest,size_t cbDest,NTSTRSAFE_PCWSTR pszSrc) { | 
 
 
 
 
 | 146 | size_t cchDest = cbDest / sizeof(wchar_t); | 
 
 
 
 
 | 147 | if(cchDest > NTSTRSAFE_MAX_CCH) return STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 148 | return RtlStringCopyWorkerW(pszDest,cchDest,pszSrc); | 
 
 
 
 
 | 149 | } | 
 
 
 
 
 | 150 | #endif /* !__STRSAFE__NO_INLINE */ | 
 
 
 
 
 | 151 |  | 
 
 
 
 
 | 152 | #define RtlStringCchCopyEx __MINGW_NAME_AW(RtlStringCchCopyEx) | 
 
 
 
 
 | 153 |  | 
 
 
 
 
 | 154 | NTSTRSAFEDDI RtlStringCchCopyExA(NTSTRSAFE_PSTR pszDest,size_t cchDest,NTSTRSAFE_PCSTR pszSrc,NTSTRSAFE_PSTR *ppszDestEnd,size_t *pcchRemaining,unsigned __LONG32 dwFlags); | 
 
 
 
 
 | 155 | NTSTRSAFEDDI RtlStringCchCopyExW(NTSTRSAFE_PWSTR pszDest,size_t cchDest,NTSTRSAFE_PCWSTR pszSrc,NTSTRSAFE_PWSTR *ppszDestEnd,size_t *pcchRemaining,unsigned __LONG32 dwFlags); | 
 
 
 
 
 | 156 |  | 
 
 
 
 
 | 157 | #ifndef __STRSAFE__NO_INLINE | 
 
 
 
 
 | 158 | NTSTRSAFEDDI RtlStringCchCopyExA(NTSTRSAFE_PSTR pszDest,size_t cchDest,NTSTRSAFE_PCSTR pszSrc,NTSTRSAFE_PSTR *ppszDestEnd,size_t *pcchRemaining,unsigned __LONG32 dwFlags) { | 
 
 
 
 
 | 159 | if(cchDest > NTSTRSAFE_MAX_CCH) return STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 160 | return RtlStringCopyExWorkerA(pszDest,cchDest,cchDest,pszSrc,ppszDestEnd,pcchRemaining,dwFlags); | 
 
 
 
 
 | 161 | } | 
 
 
 
 
 | 162 |  | 
 
 
 
 
 | 163 | NTSTRSAFEDDI RtlStringCchCopyExW(NTSTRSAFE_PWSTR pszDest,size_t cchDest,NTSTRSAFE_PCWSTR pszSrc,NTSTRSAFE_PWSTR *ppszDestEnd,size_t *pcchRemaining,unsigned __LONG32 dwFlags) { | 
 
 
 
 
 | 164 | size_t cbDest; | 
 
 
 
 
 | 165 | if(cchDest > NTSTRSAFE_MAX_CCH) return STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 166 | cbDest = cchDest * sizeof(wchar_t); | 
 
 
 
 
 | 167 | return RtlStringCopyExWorkerW(pszDest,cchDest,cbDest,pszSrc,ppszDestEnd,pcchRemaining,dwFlags); | 
 
 
 
 
 | 168 | } | 
 
 
 
 
 | 169 | #endif /* !__STRSAFE__NO_INLINE */ | 
 
 
 
 
 | 170 |  | 
 
 
 
 
 | 171 | #define RtlStringCbCopyEx __MINGW_NAME_AW(RtlStringCbCopyEx) | 
 
 
 
 
 | 172 |  | 
 
 
 
 
 | 173 | NTSTRSAFEDDI RtlStringCbCopyExA(NTSTRSAFE_PSTR pszDest,size_t cbDest,NTSTRSAFE_PCSTR pszSrc,NTSTRSAFE_PSTR *ppszDestEnd,size_t *pcbRemaining,unsigned __LONG32 dwFlags); | 
 
 
 
 
 | 174 | NTSTRSAFEDDI RtlStringCbCopyExW(NTSTRSAFE_PWSTR pszDest,size_t cbDest,NTSTRSAFE_PCWSTR pszSrc,NTSTRSAFE_PWSTR *ppszDestEnd,size_t *pcbRemaining,unsigned __LONG32 dwFlags); | 
 
 
 
 
 | 175 |  | 
 
 
 
 
 | 176 | #ifndef __STRSAFE__NO_INLINE | 
 
 
 
 
 | 177 | NTSTRSAFEDDI RtlStringCbCopyExA(NTSTRSAFE_PSTR pszDest,size_t cbDest,NTSTRSAFE_PCSTR pszSrc,NTSTRSAFE_PSTR *ppszDestEnd,size_t *pcbRemaining,unsigned __LONG32 dwFlags) { | 
 
 
 
 
 | 178 | NTSTATUS hr; | 
 
 
 
 
 | 179 | size_t cchRemaining = 0; | 
 
 
 
 
 | 180 | if(cbDest > NTSTRSAFE_MAX_CCH) return STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 181 | hr = RtlStringCopyExWorkerA(pszDest,cbDest,cbDest,pszSrc,ppszDestEnd,&cchRemaining,dwFlags); | 
 
 
 
 
 | 182 | if(NT_SUCCESS(hr) || hr == STATUS_BUFFER_OVERFLOW) { | 
 
 
 
 
 | 183 | if(pcbRemaining) | 
 
 
 
 
 | 184 | *pcbRemaining = (cchRemaining*sizeof(char)) + (cbDest % sizeof(char)); | 
 
 
 
 
 | 185 | } | 
 
 
 
 
 | 186 | return hr; | 
 
 
 
 
 | 187 | } | 
 
 
 
 
 | 188 |  | 
 
 
 
 
 | 189 | NTSTRSAFEDDI RtlStringCbCopyExW(NTSTRSAFE_PWSTR pszDest,size_t cbDest,NTSTRSAFE_PCWSTR pszSrc,NTSTRSAFE_PWSTR *ppszDestEnd,size_t *pcbRemaining,unsigned __LONG32 dwFlags) { | 
 
 
 
 
 | 190 | NTSTATUS hr; | 
 
 
 
 
 | 191 | size_t cchDest = cbDest / sizeof(wchar_t); | 
 
 
 
 
 | 192 | size_t cchRemaining = 0; | 
 
 
 
 
 | 193 |  | 
 
 
 
 
 | 194 | if(cchDest > NTSTRSAFE_MAX_CCH) return STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 195 | hr = RtlStringCopyExWorkerW(pszDest,cchDest,cbDest,pszSrc,ppszDestEnd,&cchRemaining,dwFlags); | 
 
 
 
 
 | 196 | if(NT_SUCCESS(hr) || (hr==STATUS_BUFFER_OVERFLOW)) { | 
 
 
 
 
 | 197 | if(pcbRemaining) | 
 
 
 
 
 | 198 | *pcbRemaining = (cchRemaining*sizeof(wchar_t)) + (cbDest % sizeof(wchar_t)); | 
 
 
 
 
 | 199 | } | 
 
 
 
 
 | 200 | return hr; | 
 
 
 
 
 | 201 | } | 
 
 
 
 
 | 202 | #endif /* !__STRSAFE__NO_INLINE */ | 
 
 
 
 
 | 203 |  | 
 
 
 
 
 | 204 | NTSTRSAFEDDI RtlStringCchCopyNA(NTSTRSAFE_PSTR pszDest,size_t cchDest,NTSTRSAFE_PCSTR pszSrc,size_t cchToCopy); | 
 
 
 
 
 | 205 | NTSTRSAFEDDI RtlStringCchCopyNW(NTSTRSAFE_PWSTR pszDest,size_t cchDest,NTSTRSAFE_PCWSTR pszSrc,size_t cchToCopy); | 
 
 
 
 
 | 206 | #define RtlStringCchCopyN __MINGW_NAME_AW(RtlStringCchCopyN) | 
 
 
 
 
 | 207 |  | 
 
 
 
 
 | 208 | #ifndef __STRSAFE__NO_INLINE | 
 
 
 
 
 | 209 | NTSTRSAFEDDI RtlStringCchCopyNA(NTSTRSAFE_PSTR pszDest,size_t cchDest,NTSTRSAFE_PCSTR pszSrc,size_t cchToCopy) { | 
 
 
 
 
 | 210 | if(cchDest > NTSTRSAFE_MAX_CCH || cchToCopy > NTSTRSAFE_MAX_CCH) | 
 
 
 
 
 | 211 | return STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 212 | return RtlStringCopyNWorkerA(pszDest,cchDest,pszSrc,cchToCopy); | 
 
 
 
 
 | 213 | } | 
 
 
 
 
 | 214 |  | 
 
 
 
 
 | 215 | NTSTRSAFEDDI RtlStringCchCopyNW(NTSTRSAFE_PWSTR pszDest,size_t cchDest,NTSTRSAFE_PCWSTR pszSrc,size_t cchToCopy) { | 
 
 
 
 
 | 216 | if(cchDest > NTSTRSAFE_MAX_CCH || cchToCopy > NTSTRSAFE_MAX_CCH) | 
 
 
 
 
 | 217 | return STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 218 | return RtlStringCopyNWorkerW(pszDest,cchDest,pszSrc,cchToCopy); | 
 
 
 
 
 | 219 | } | 
 
 
 
 
 | 220 | #endif /* !__STRSAFE__NO_INLINE */ | 
 
 
 
 
 | 221 |  | 
 
 
 
 
 | 222 | NTSTRSAFEDDI RtlStringCbCopyNA(NTSTRSAFE_PSTR pszDest,size_t cbDest,NTSTRSAFE_PCSTR pszSrc,size_t cbToCopy); | 
 
 
 
 
 | 223 | NTSTRSAFEDDI RtlStringCbCopyNW(NTSTRSAFE_PWSTR pszDest,size_t cbDest,NTSTRSAFE_PCWSTR pszSrc,size_t cbToCopy); | 
 
 
 
 
 | 224 |  | 
 
 
 
 
 | 225 | #define RtlStringCbCopyN __MINGW_NAME_AW(RtlStringCbCopyN) | 
 
 
 
 
 | 226 |  | 
 
 
 
 
 | 227 | #ifndef __STRSAFE__NO_INLINE | 
 
 
 
 
 | 228 | NTSTRSAFEDDI RtlStringCbCopyNA(NTSTRSAFE_PSTR pszDest,size_t cbDest,NTSTRSAFE_PCSTR pszSrc,size_t cbToCopy) { | 
 
 
 
 
 | 229 | if(cbDest > NTSTRSAFE_MAX_CCH || cbToCopy > NTSTRSAFE_MAX_CCH) | 
 
 
 
 
 | 230 | return STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 231 | return RtlStringCopyNWorkerA(pszDest,cbDest,pszSrc,cbToCopy); | 
 
 
 
 
 | 232 | } | 
 
 
 
 
 | 233 |  | 
 
 
 
 
 | 234 | NTSTRSAFEDDI RtlStringCbCopyNW(NTSTRSAFE_PWSTR pszDest,size_t cbDest,NTSTRSAFE_PCWSTR pszSrc,size_t cbToCopy) { | 
 
 
 
 
 | 235 | size_t cchDest  = cbDest / sizeof(wchar_t); | 
 
 
 
 
 | 236 | size_t cchToCopy = cbToCopy / sizeof(wchar_t); | 
 
 
 
 
 | 237 | if(cchDest > NTSTRSAFE_MAX_CCH || cchToCopy > NTSTRSAFE_MAX_CCH) | 
 
 
 
 
 | 238 | return STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 239 | return RtlStringCopyNWorkerW(pszDest,cchDest,pszSrc,cchToCopy); | 
 
 
 
 
 | 240 | } | 
 
 
 
 
 | 241 | #endif /* !__STRSAFE__NO_INLINE */ | 
 
 
 
 
 | 242 |  | 
 
 
 
 
 | 243 | NTSTRSAFEDDI RtlStringCchCopyNExA(NTSTRSAFE_PSTR pszDest,size_t cchDest,NTSTRSAFE_PCSTR pszSrc,size_t cchToCopy,NTSTRSAFE_PSTR *ppszDestEnd,size_t *pcchRemaining,unsigned __LONG32 dwFlags); | 
 
 
 
 
 | 244 | NTSTRSAFEDDI RtlStringCchCopyNExW(NTSTRSAFE_PWSTR pszDest,size_t cchDest,NTSTRSAFE_PCWSTR pszSrc,size_t cchToCopy,NTSTRSAFE_PWSTR *ppszDestEnd,size_t *pcchRemaining,unsigned __LONG32 dwFlags); | 
 
 
 
 
 | 245 |  | 
 
 
 
 
 | 246 | #define RtlStringCchCopyNEx __MINGW_NAME_AW(RtlStringCchCopyNEx) | 
 
 
 
 
 | 247 |  | 
 
 
 
 
 | 248 | #ifndef __STRSAFE__NO_INLINE | 
 
 
 
 
 | 249 | NTSTRSAFEDDI RtlStringCchCopyNExA(NTSTRSAFE_PSTR pszDest,size_t cchDest,NTSTRSAFE_PCSTR pszSrc,size_t cchToCopy,NTSTRSAFE_PSTR *ppszDestEnd,size_t *pcchRemaining,unsigned __LONG32 dwFlags) { | 
 
 
 
 
 | 250 | if(cchDest > NTSTRSAFE_MAX_CCH) return STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 251 | return RtlStringCopyNExWorkerA(pszDest,cchDest,cchDest,pszSrc,cchToCopy,ppszDestEnd,pcchRemaining,dwFlags); | 
 
 
 
 
 | 252 | } | 
 
 
 
 
 | 253 |  | 
 
 
 
 
 | 254 | NTSTRSAFEDDI RtlStringCchCopyNExW(NTSTRSAFE_PWSTR pszDest,size_t cchDest,NTSTRSAFE_PCWSTR pszSrc,size_t cchToCopy,NTSTRSAFE_PWSTR *ppszDestEnd,size_t *pcchRemaining,unsigned __LONG32 dwFlags) { | 
 
 
 
 
 | 255 | if(cchDest > NTSTRSAFE_MAX_CCH) return STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 256 | return RtlStringCopyNExWorkerW(pszDest,cchDest,cchDest * sizeof(wchar_t),pszSrc,cchToCopy,ppszDestEnd,pcchRemaining,dwFlags); | 
 
 
 
 
 | 257 | } | 
 
 
 
 
 | 258 | #endif /* !__STRSAFE__NO_INLINE */ | 
 
 
 
 
 | 259 |  | 
 
 
 
 
 | 260 | NTSTRSAFEDDI RtlStringCbCopyNExA(NTSTRSAFE_PSTR pszDest,size_t cbDest,NTSTRSAFE_PCSTR pszSrc,size_t cbToCopy,NTSTRSAFE_PSTR *ppszDestEnd,size_t *pcbRemaining,unsigned __LONG32 dwFlags); | 
 
 
 
 
 | 261 | NTSTRSAFEDDI RtlStringCbCopyNExW(NTSTRSAFE_PWSTR pszDest,size_t cbDest,NTSTRSAFE_PCWSTR pszSrc,size_t cbToCopy,NTSTRSAFE_PWSTR *ppszDestEnd,size_t *pcbRemaining,unsigned __LONG32 dwFlags); | 
 
 
 
 
 | 262 |  | 
 
 
 
 
 | 263 | #define RtlStringCbCopyNEx __MINGW_NAME_AW(RtlStringCbCopyNEx) | 
 
 
 
 
 | 264 |  | 
 
 
 
 
 | 265 | #ifndef __STRSAFE__NO_INLINE | 
 
 
 
 
 | 266 | NTSTRSAFEDDI RtlStringCbCopyNExA(NTSTRSAFE_PSTR pszDest,size_t cbDest,NTSTRSAFE_PCSTR pszSrc,size_t cbToCopy,NTSTRSAFE_PSTR *ppszDestEnd,size_t *pcbRemaining,unsigned __LONG32 dwFlags) { | 
 
 
 
 
 | 267 | NTSTATUS hr; | 
 
 
 
 
 | 268 | size_t cchRemaining = 0; | 
 
 
 
 
 | 269 | if(cbDest > NTSTRSAFE_MAX_CCH) | 
 
 
 
 
 | 270 | hr = STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 271 | else | 
 
 
 
 
 | 272 | hr = RtlStringCopyNExWorkerA(pszDest,cbDest,cbDest,pszSrc,cbToCopy,ppszDestEnd,&cchRemaining,dwFlags); | 
 
 
 
 
 | 273 | if((NT_SUCCESS(hr) || hr == STATUS_BUFFER_OVERFLOW) && pcbRemaining) | 
 
 
 
 
 | 274 | *pcbRemaining = cchRemaining; | 
 
 
 
 
 | 275 | return hr; | 
 
 
 
 
 | 276 | } | 
 
 
 
 
 | 277 |  | 
 
 
 
 
 | 278 | NTSTRSAFEDDI RtlStringCbCopyNExW(NTSTRSAFE_PWSTR pszDest,size_t cbDest,NTSTRSAFE_PCWSTR pszSrc,size_t cbToCopy,NTSTRSAFE_PWSTR *ppszDestEnd,size_t *pcbRemaining,unsigned __LONG32 dwFlags) { | 
 
 
 
 
 | 279 | NTSTATUS hr; | 
 
 
 
 
 | 280 | size_t cchDest; | 
 
 
 
 
 | 281 | size_t cchToCopy; | 
 
 
 
 
 | 282 | size_t cchRemaining = 0; | 
 
 
 
 
 | 283 | cchDest = cbDest / sizeof(wchar_t); | 
 
 
 
 
 | 284 | cchToCopy = cbToCopy / sizeof(wchar_t); | 
 
 
 
 
 | 285 | if(cchDest > NTSTRSAFE_MAX_CCH) hr = STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 286 | else hr = RtlStringCopyNExWorkerW(pszDest,cchDest,cbDest,pszSrc,cchToCopy,ppszDestEnd,&cchRemaining,dwFlags); | 
 
 
 
 
 | 287 | if((NT_SUCCESS(hr) || hr == STATUS_BUFFER_OVERFLOW) && pcbRemaining) | 
 
 
 
 
 | 288 | *pcbRemaining = (cchRemaining*sizeof(wchar_t)) + (cbDest % sizeof(wchar_t)); | 
 
 
 
 
 | 289 | return hr; | 
 
 
 
 
 | 290 | } | 
 
 
 
 
 | 291 | #endif /* !__STRSAFE__NO_INLINE */ | 
 
 
 
 
 | 292 |  | 
 
 
 
 
 | 293 | NTSTRSAFEDDI RtlStringCchCatA(NTSTRSAFE_PSTR pszDest,size_t cchDest,NTSTRSAFE_PCSTR pszSrc); | 
 
 
 
 
 | 294 | NTSTRSAFEDDI RtlStringCchCatW(NTSTRSAFE_PWSTR pszDest,size_t cchDest,NTSTRSAFE_PCWSTR pszSrc); | 
 
 
 
 
 | 295 |  | 
 
 
 
 
 | 296 | #define RtlStringCchCat __MINGW_NAME_AW(RtlStringCchCat) | 
 
 
 
 
 | 297 |  | 
 
 
 
 
 | 298 | #ifndef __STRSAFE__NO_INLINE | 
 
 
 
 
 | 299 | NTSTRSAFEDDI RtlStringCchCatA(NTSTRSAFE_PSTR pszDest,size_t cchDest,NTSTRSAFE_PCSTR pszSrc) { | 
 
 
 
 
 | 300 | if(cchDest > NTSTRSAFE_MAX_CCH) return STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 301 | return RtlStringCatWorkerA(pszDest,cchDest,pszSrc); | 
 
 
 
 
 | 302 | } | 
 
 
 
 
 | 303 |  | 
 
 
 
 
 | 304 | NTSTRSAFEDDI RtlStringCchCatW(NTSTRSAFE_PWSTR pszDest,size_t cchDest,NTSTRSAFE_PCWSTR pszSrc) { | 
 
 
 
 
 | 305 | if(cchDest > NTSTRSAFE_MAX_CCH) return STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 306 | return RtlStringCatWorkerW(pszDest,cchDest,pszSrc); | 
 
 
 
 
 | 307 | } | 
 
 
 
 
 | 308 | #endif /* !__STRSAFE__NO_INLINE */ | 
 
 
 
 
 | 309 |  | 
 
 
 
 
 | 310 | NTSTRSAFEDDI RtlStringCbCatA(NTSTRSAFE_PSTR pszDest,size_t cbDest,NTSTRSAFE_PCSTR pszSrc); | 
 
 
 
 
 | 311 | NTSTRSAFEDDI RtlStringCbCatW(NTSTRSAFE_PWSTR pszDest,size_t cbDest,NTSTRSAFE_PCWSTR pszSrc); | 
 
 
 
 
 | 312 |  | 
 
 
 
 
 | 313 | #define RtlStringCbCat __MINGW_NAME_AW(RtlStringCbCat) | 
 
 
 
 
 | 314 |  | 
 
 
 
 
 | 315 | #ifndef __STRSAFE__NO_INLINE | 
 
 
 
 
 | 316 | NTSTRSAFEDDI RtlStringCbCatA(NTSTRSAFE_PSTR pszDest,size_t cbDest,NTSTRSAFE_PCSTR pszSrc) { | 
 
 
 
 
 | 317 | if(cbDest > NTSTRSAFE_MAX_CCH) return STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 318 | return RtlStringCatWorkerA(pszDest,cbDest,pszSrc); | 
 
 
 
 
 | 319 | } | 
 
 
 
 
 | 320 |  | 
 
 
 
 
 | 321 | NTSTRSAFEDDI RtlStringCbCatW(NTSTRSAFE_PWSTR pszDest,size_t cbDest,NTSTRSAFE_PCWSTR pszSrc) { | 
 
 
 
 
 | 322 | size_t cchDest = cbDest / sizeof(wchar_t); | 
 
 
 
 
 | 323 | if(cchDest > NTSTRSAFE_MAX_CCH) return STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 324 | return RtlStringCatWorkerW(pszDest,cchDest,pszSrc); | 
 
 
 
 
 | 325 | } | 
 
 
 
 
 | 326 | #endif /* !__STRSAFE__NO_INLINE */ | 
 
 
 
 
 | 327 |  | 
 
 
 
 
 | 328 | NTSTRSAFEDDI RtlStringCchCatExA(NTSTRSAFE_PSTR pszDest,size_t cchDest,NTSTRSAFE_PCSTR pszSrc,NTSTRSAFE_PSTR *ppszDestEnd,size_t *pcchRemaining,unsigned __LONG32 dwFlags); | 
 
 
 
 
 | 329 | NTSTRSAFEDDI RtlStringCchCatExW(NTSTRSAFE_PWSTR pszDest,size_t cchDest,NTSTRSAFE_PCWSTR pszSrc,NTSTRSAFE_PWSTR *ppszDestEnd,size_t *pcchRemaining,unsigned __LONG32 dwFlags); | 
 
 
 
 
 | 330 |  | 
 
 
 
 
 | 331 | #define RtlStringCchCatEx __MINGW_NAME_AW(RtlStringCchCatEx) | 
 
 
 
 
 | 332 |  | 
 
 
 
 
 | 333 | #ifndef __STRSAFE__NO_INLINE | 
 
 
 
 
 | 334 | NTSTRSAFEDDI RtlStringCchCatExA(NTSTRSAFE_PSTR pszDest,size_t cchDest,NTSTRSAFE_PCSTR pszSrc,NTSTRSAFE_PSTR *ppszDestEnd,size_t *pcchRemaining,unsigned __LONG32 dwFlags) { | 
 
 
 
 
 | 335 | if(cchDest > NTSTRSAFE_MAX_CCH) return STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 336 | return RtlStringCatExWorkerA(pszDest,cchDest,cchDest,pszSrc,ppszDestEnd,pcchRemaining,dwFlags); | 
 
 
 
 
 | 337 | } | 
 
 
 
 
 | 338 |  | 
 
 
 
 
 | 339 | NTSTRSAFEDDI RtlStringCchCatExW(NTSTRSAFE_PWSTR pszDest,size_t cchDest,NTSTRSAFE_PCWSTR pszSrc,NTSTRSAFE_PWSTR *ppszDestEnd,size_t *pcchRemaining,unsigned __LONG32 dwFlags) { | 
 
 
 
 
 | 340 | size_t cbDest = cchDest*sizeof(wchar_t); | 
 
 
 
 
 | 341 | if(cchDest > NTSTRSAFE_MAX_CCH) return STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 342 | return RtlStringCatExWorkerW(pszDest,cchDest,cbDest,pszSrc,ppszDestEnd,pcchRemaining,dwFlags); | 
 
 
 
 
 | 343 | } | 
 
 
 
 
 | 344 | #endif /* !__STRSAFE__NO_INLINE */ | 
 
 
 
 
 | 345 |  | 
 
 
 
 
 | 346 | NTSTRSAFEDDI RtlStringCbCatExA(NTSTRSAFE_PSTR pszDest,size_t cbDest,NTSTRSAFE_PCSTR pszSrc,NTSTRSAFE_PSTR *ppszDestEnd,size_t *pcbRemaining,unsigned __LONG32 dwFlags); | 
 
 
 
 
 | 347 | NTSTRSAFEDDI RtlStringCbCatExW(NTSTRSAFE_PWSTR pszDest,size_t cbDest,NTSTRSAFE_PCWSTR pszSrc,NTSTRSAFE_PWSTR *ppszDestEnd,size_t *pcbRemaining,unsigned __LONG32 dwFlags); | 
 
 
 
 
 | 348 |  | 
 
 
 
 
 | 349 | #define RtlStringCbCatEx __MINGW_NAME_AW(RtlStringCbCatEx) | 
 
 
 
 
 | 350 |  | 
 
 
 
 
 | 351 | #ifndef __STRSAFE__NO_INLINE | 
 
 
 
 
 | 352 | NTSTRSAFEDDI RtlStringCbCatExA(NTSTRSAFE_PSTR pszDest,size_t cbDest,NTSTRSAFE_PCSTR pszSrc,NTSTRSAFE_PSTR *ppszDestEnd,size_t *pcbRemaining,unsigned __LONG32 dwFlags) { | 
 
 
 
 
 | 353 | NTSTATUS hr; | 
 
 
 
 
 | 354 | size_t cchRemaining = 0; | 
 
 
 
 
 | 355 | if(cbDest > NTSTRSAFE_MAX_CCH) hr = STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 356 | else hr = RtlStringCatExWorkerA(pszDest,cbDest,cbDest,pszSrc,ppszDestEnd,&cchRemaining,dwFlags); | 
 
 
 
 
 | 357 | if((NT_SUCCESS(hr) || hr == STATUS_BUFFER_OVERFLOW) && pcbRemaining) | 
 
 
 
 
 | 358 | *pcbRemaining = (cchRemaining*sizeof(char)) + (cbDest % sizeof(char)); | 
 
 
 
 
 | 359 | return hr; | 
 
 
 
 
 | 360 | } | 
 
 
 
 
 | 361 |  | 
 
 
 
 
 | 362 | NTSTRSAFEDDI RtlStringCbCatExW(NTSTRSAFE_PWSTR pszDest,size_t cbDest,NTSTRSAFE_PCWSTR pszSrc,NTSTRSAFE_PWSTR *ppszDestEnd,size_t *pcbRemaining,unsigned __LONG32 dwFlags) { | 
 
 
 
 
 | 363 | NTSTATUS hr; | 
 
 
 
 
 | 364 | size_t cchDest = cbDest / sizeof(wchar_t); | 
 
 
 
 
 | 365 | size_t cchRemaining = 0; | 
 
 
 
 
 | 366 |  | 
 
 
 
 
 | 367 | if(cchDest > NTSTRSAFE_MAX_CCH) hr = STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 368 | else hr = RtlStringCatExWorkerW(pszDest,cchDest,cbDest,pszSrc,ppszDestEnd,&cchRemaining,dwFlags); | 
 
 
 
 
 | 369 | if((NT_SUCCESS(hr) || hr == STATUS_BUFFER_OVERFLOW) && pcbRemaining) | 
 
 
 
 
 | 370 | *pcbRemaining = (cchRemaining*sizeof(wchar_t)) + (cbDest % sizeof(wchar_t)); | 
 
 
 
 
 | 371 | return hr; | 
 
 
 
 
 | 372 | } | 
 
 
 
 
 | 373 | #endif /* !__STRSAFE__NO_INLINE */ | 
 
 
 
 
 | 374 |  | 
 
 
 
 
 | 375 | NTSTRSAFEDDI RtlStringCchCatNA(NTSTRSAFE_PSTR pszDest,size_t cchDest,NTSTRSAFE_PCSTR pszSrc,size_t cchToAppend); | 
 
 
 
 
 | 376 | NTSTRSAFEDDI RtlStringCchCatNW(NTSTRSAFE_PWSTR pszDest,size_t cchDest,NTSTRSAFE_PCWSTR pszSrc,size_t cchToAppend); | 
 
 
 
 
 | 377 |  | 
 
 
 
 
 | 378 | #define RtlStringCchCatN __MINGW_NAME_AW(RtlStringCchCatN) | 
 
 
 
 
 | 379 |  | 
 
 
 
 
 | 380 | #ifndef __STRSAFE__NO_INLINE | 
 
 
 
 
 | 381 | NTSTRSAFEDDI RtlStringCchCatNA(NTSTRSAFE_PSTR pszDest,size_t cchDest,NTSTRSAFE_PCSTR pszSrc,size_t cchToAppend) { | 
 
 
 
 
 | 382 | if(cchDest > NTSTRSAFE_MAX_CCH) return STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 383 | return RtlStringCatNWorkerA(pszDest,cchDest,pszSrc,cchToAppend); | 
 
 
 
 
 | 384 | } | 
 
 
 
 
 | 385 |  | 
 
 
 
 
 | 386 | NTSTRSAFEDDI RtlStringCchCatNW(NTSTRSAFE_PWSTR pszDest,size_t cchDest,NTSTRSAFE_PCWSTR pszSrc,size_t cchToAppend) { | 
 
 
 
 
 | 387 | if(cchDest > NTSTRSAFE_MAX_CCH) return STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 388 | return RtlStringCatNWorkerW(pszDest,cchDest,pszSrc,cchToAppend); | 
 
 
 
 
 | 389 | } | 
 
 
 
 
 | 390 | #endif /* !__STRSAFE__NO_INLINE */ | 
 
 
 
 
 | 391 |  | 
 
 
 
 
 | 392 | NTSTRSAFEDDI RtlStringCbCatNA(NTSTRSAFE_PSTR pszDest,size_t cbDest,NTSTRSAFE_PCSTR pszSrc,size_t cbToAppend); | 
 
 
 
 
 | 393 | NTSTRSAFEDDI RtlStringCbCatNW(NTSTRSAFE_PWSTR pszDest,size_t cbDest,NTSTRSAFE_PCWSTR pszSrc,size_t cbToAppend); | 
 
 
 
 
 | 394 |  | 
 
 
 
 
 | 395 | #define RtlStringCbCatN __MINGW_NAME_AW(RtlStringCbCatN) | 
 
 
 
 
 | 396 |  | 
 
 
 
 
 | 397 | #ifndef __STRSAFE__NO_INLINE | 
 
 
 
 
 | 398 | NTSTRSAFEDDI RtlStringCbCatNA(NTSTRSAFE_PSTR pszDest,size_t cbDest,NTSTRSAFE_PCSTR pszSrc,size_t cbToAppend) { | 
 
 
 
 
 | 399 | if(cbDest > NTSTRSAFE_MAX_CCH) return STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 400 | return RtlStringCatNWorkerA(pszDest,cbDest,pszSrc,cbToAppend); | 
 
 
 
 
 | 401 | } | 
 
 
 
 
 | 402 |  | 
 
 
 
 
 | 403 | NTSTRSAFEDDI RtlStringCbCatNW(NTSTRSAFE_PWSTR pszDest,size_t cbDest,NTSTRSAFE_PCWSTR pszSrc,size_t cbToAppend) { | 
 
 
 
 
 | 404 | size_t cchDest = cbDest / sizeof(wchar_t); | 
 
 
 
 
 | 405 | size_t cchToAppend = cbToAppend / sizeof(wchar_t); | 
 
 
 
 
 | 406 |  | 
 
 
 
 
 | 407 | if(cchDest > NTSTRSAFE_MAX_CCH) return STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 408 | return RtlStringCatNWorkerW(pszDest,cchDest,pszSrc,cchToAppend); | 
 
 
 
 
 | 409 | } | 
 
 
 
 
 | 410 | #endif /* !__STRSAFE__NO_INLINE */ | 
 
 
 
 
 | 411 |  | 
 
 
 
 
 | 412 | NTSTRSAFEDDI RtlStringCchCatNExA(NTSTRSAFE_PSTR pszDest,size_t cchDest,NTSTRSAFE_PCSTR pszSrc,size_t cchToAppend,NTSTRSAFE_PSTR *ppszDestEnd,size_t *pcchRemaining,unsigned __LONG32 dwFlags); | 
 
 
 
 
 | 413 | NTSTRSAFEDDI RtlStringCchCatNExW(NTSTRSAFE_PWSTR pszDest,size_t cchDest,NTSTRSAFE_PCWSTR pszSrc,size_t cchToAppend,NTSTRSAFE_PWSTR *ppszDestEnd,size_t *pcchRemaining,unsigned __LONG32 dwFlags); | 
 
 
 
 
 | 414 |  | 
 
 
 
 
 | 415 | #define RtlStringCchCatNEx __MINGW_NAME_AW(RtlStringCchCatNEx) | 
 
 
 
 
 | 416 |  | 
 
 
 
 
 | 417 | #ifndef __STRSAFE__NO_INLINE | 
 
 
 
 
 | 418 | NTSTRSAFEDDI RtlStringCchCatNExA(NTSTRSAFE_PSTR pszDest,size_t cchDest,NTSTRSAFE_PCSTR pszSrc,size_t cchToAppend,NTSTRSAFE_PSTR *ppszDestEnd,size_t *pcchRemaining,unsigned __LONG32 dwFlags) { | 
 
 
 
 
 | 419 | if(cchDest > NTSTRSAFE_MAX_CCH) return STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 420 | return RtlStringCatNExWorkerA(pszDest,cchDest,cchDest,pszSrc,cchToAppend,ppszDestEnd,pcchRemaining,dwFlags); | 
 
 
 
 
 | 421 | } | 
 
 
 
 
 | 422 |  | 
 
 
 
 
 | 423 | NTSTRSAFEDDI RtlStringCchCatNExW(NTSTRSAFE_PWSTR pszDest,size_t cchDest,NTSTRSAFE_PCWSTR pszSrc,size_t cchToAppend,NTSTRSAFE_PWSTR *ppszDestEnd,size_t *pcchRemaining,unsigned __LONG32 dwFlags) { | 
 
 
 
 
 | 424 | if(cchDest > NTSTRSAFE_MAX_CCH) return STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 425 | return RtlStringCatNExWorkerW(pszDest,cchDest,(cchDest*sizeof(wchar_t)),pszSrc,cchToAppend,ppszDestEnd,pcchRemaining,dwFlags); | 
 
 
 
 
 | 426 | } | 
 
 
 
 
 | 427 | #endif | 
 
 
 
 
 | 428 |  | 
 
 
 
 
 | 429 | NTSTRSAFEDDI RtlStringCbCatNExA(NTSTRSAFE_PSTR pszDest,size_t cbDest,NTSTRSAFE_PCSTR pszSrc,size_t cbToAppend,NTSTRSAFE_PSTR *ppszDestEnd,size_t *pcbRemaining,unsigned __LONG32 dwFlags); | 
 
 
 
 
 | 430 | NTSTRSAFEDDI RtlStringCbCatNExW(NTSTRSAFE_PWSTR pszDest,size_t cbDest,NTSTRSAFE_PCWSTR pszSrc,size_t cbToAppend,NTSTRSAFE_PWSTR *ppszDestEnd,size_t *pcbRemaining,unsigned __LONG32 dwFlags); | 
 
 
 
 
 | 431 |  | 
 
 
 
 
 | 432 | #define RtlStringCbCatNEx __MINGW_NAME_AW(RtlStringCbCatNEx) | 
 
 
 
 
 | 433 |  | 
 
 
 
 
 | 434 | #ifndef __STRSAFE__NO_INLINE | 
 
 
 
 
 | 435 | NTSTRSAFEDDI RtlStringCbCatNExA(NTSTRSAFE_PSTR pszDest,size_t cbDest,NTSTRSAFE_PCSTR pszSrc,size_t cbToAppend,NTSTRSAFE_PSTR *ppszDestEnd,size_t *pcbRemaining,unsigned __LONG32 dwFlags) { | 
 
 
 
 
 | 436 | NTSTATUS hr; | 
 
 
 
 
 | 437 | size_t cchRemaining = 0; | 
 
 
 
 
 | 438 | if(cbDest > NTSTRSAFE_MAX_CCH) hr = STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 439 | else hr = RtlStringCatNExWorkerA(pszDest,cbDest,cbDest,pszSrc,cbToAppend,ppszDestEnd,&cchRemaining,dwFlags); | 
 
 
 
 
 | 440 | if((NT_SUCCESS(hr) || hr == STATUS_BUFFER_OVERFLOW) && pcbRemaining) | 
 
 
 
 
 | 441 | *pcbRemaining = (cchRemaining*sizeof(char)) + (cbDest % sizeof(char)); | 
 
 
 
 
 | 442 | return hr; | 
 
 
 
 
 | 443 | } | 
 
 
 
 
 | 444 |  | 
 
 
 
 
 | 445 | NTSTRSAFEDDI RtlStringCbCatNExW(NTSTRSAFE_PWSTR pszDest,size_t cbDest,NTSTRSAFE_PCWSTR pszSrc,size_t cbToAppend,NTSTRSAFE_PWSTR *ppszDestEnd,size_t *pcbRemaining,unsigned __LONG32 dwFlags) { | 
 
 
 
 
 | 446 | NTSTATUS hr; | 
 
 
 
 
 | 447 | size_t cchDest = cbDest / sizeof(wchar_t); | 
 
 
 
 
 | 448 | size_t cchToAppend = cbToAppend / sizeof(wchar_t); | 
 
 
 
 
 | 449 | size_t cchRemaining = 0; | 
 
 
 
 
 | 450 | if(cchDest > NTSTRSAFE_MAX_CCH) hr = STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 451 | else hr = RtlStringCatNExWorkerW(pszDest,cchDest,cbDest,pszSrc,cchToAppend,ppszDestEnd,&cchRemaining,dwFlags); | 
 
 
 
 
 | 452 | if((NT_SUCCESS(hr) || hr == STATUS_BUFFER_OVERFLOW) && pcbRemaining) | 
 
 
 
 
 | 453 | *pcbRemaining = (cchRemaining*sizeof(wchar_t)) + (cbDest % sizeof(wchar_t)); | 
 
 
 
 
 | 454 | return hr; | 
 
 
 
 
 | 455 | } | 
 
 
 
 
 | 456 | #endif /* !__STRSAFE__NO_INLINE */ | 
 
 
 
 
 | 457 |  | 
 
 
 
 
 | 458 | NTSTRSAFEDDI RtlStringCchVPrintfA(NTSTRSAFE_PSTR pszDest,size_t cchDest,NTSTRSAFE_PCSTR pszFormat,va_list argList); | 
 
 
 
 
 | 459 | NTSTRSAFEDDI RtlStringCchVPrintfW(NTSTRSAFE_PWSTR pszDest,size_t cchDest,NTSTRSAFE_PCWSTR pszFormat,va_list argList); | 
 
 
 
 
 | 460 |  | 
 
 
 
 
 | 461 | #define RtlStringCchVPrintf __MINGW_NAME_AW(RtlStringCchVPrintf) | 
 
 
 
 
 | 462 |  | 
 
 
 
 
 | 463 | #ifndef __STRSAFE__NO_INLINE | 
 
 
 
 
 | 464 | NTSTRSAFEDDI RtlStringCchVPrintfA(NTSTRSAFE_PSTR pszDest,size_t cchDest,NTSTRSAFE_PCSTR pszFormat,va_list argList) { | 
 
 
 
 
 | 465 | if(cchDest > NTSTRSAFE_MAX_CCH) return STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 466 | return RtlStringVPrintfWorkerA(pszDest,cchDest,pszFormat,argList); | 
 
 
 
 
 | 467 | } | 
 
 
 
 
 | 468 |  | 
 
 
 
 
 | 469 | NTSTRSAFEDDI RtlStringCchVPrintfW(NTSTRSAFE_PWSTR pszDest,size_t cchDest,NTSTRSAFE_PCWSTR pszFormat,va_list argList) { | 
 
 
 
 
 | 470 | if(cchDest > NTSTRSAFE_MAX_CCH) return STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 471 | return RtlStringVPrintfWorkerW(pszDest,cchDest,pszFormat,argList); | 
 
 
 
 
 | 472 | } | 
 
 
 
 
 | 473 | #endif /* !__STRSAFE__NO_INLINE */ | 
 
 
 
 
 | 474 |  | 
 
 
 
 
 | 475 | NTSTRSAFEDDI RtlStringCbVPrintfA(NTSTRSAFE_PSTR pszDest,size_t cbDest,NTSTRSAFE_PCSTR pszFormat,va_list argList); | 
 
 
 
 
 | 476 | NTSTRSAFEDDI RtlStringCbVPrintfW(NTSTRSAFE_PWSTR pszDest,size_t cbDest,NTSTRSAFE_PCWSTR pszFormat,va_list argList); | 
 
 
 
 
 | 477 |  | 
 
 
 
 
 | 478 | #define RtlStringCbVPrintf __MINGW_NAME_AW(RtlStringCbVPrintf) | 
 
 
 
 
 | 479 |  | 
 
 
 
 
 | 480 | #ifndef __STRSAFE__NO_INLINE | 
 
 
 
 
 | 481 | NTSTRSAFEDDI RtlStringCbVPrintfA(NTSTRSAFE_PSTR pszDest,size_t cbDest,NTSTRSAFE_PCSTR pszFormat,va_list argList) { | 
 
 
 
 
 | 482 | if(cbDest > NTSTRSAFE_MAX_CCH) return STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 483 | return RtlStringVPrintfWorkerA(pszDest,cbDest,pszFormat,argList); | 
 
 
 
 
 | 484 | } | 
 
 
 
 
 | 485 |  | 
 
 
 
 
 | 486 | NTSTRSAFEDDI RtlStringCbVPrintfW(NTSTRSAFE_PWSTR pszDest,size_t cbDest,NTSTRSAFE_PCWSTR pszFormat,va_list argList) { | 
 
 
 
 
 | 487 | size_t cchDest = cbDest / sizeof(wchar_t); | 
 
 
 
 
 | 488 | if(cchDest > NTSTRSAFE_MAX_CCH) return STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 489 | return RtlStringVPrintfWorkerW(pszDest,cchDest,pszFormat,argList); | 
 
 
 
 
 | 490 | } | 
 
 
 
 
 | 491 | #endif /* !__STRSAFE__NO_INLINE */ | 
 
 
 
 
 | 492 |  | 
 
 
 
 
 | 493 | NTSTRSAFEDDIV RtlStringCchPrintfA(NTSTRSAFE_PSTR pszDest,size_t cchDest,NTSTRSAFE_PCSTR pszFormat,...); | 
 
 
 
 
 | 494 | NTSTRSAFEDDIV RtlStringCchPrintfW(NTSTRSAFE_PWSTR pszDest,size_t cchDest,NTSTRSAFE_PCWSTR pszFormat,...); | 
 
 
 
 
 | 495 |  | 
 
 
 
 
 | 496 | #define RtlStringCchPrintf __MINGW_NAME_AW(RtlStringCchPrintf) | 
 
 
 
 
 | 497 |  | 
 
 
 
 
 | 498 | #ifndef __STRSAFE__NO_INLINE | 
 
 
 
 
 | 499 | NTSTRSAFEDDIV RtlStringCchPrintfA(NTSTRSAFE_PSTR pszDest,size_t cchDest,NTSTRSAFE_PCSTR pszFormat,...) { | 
 
 
 
 
 | 500 | NTSTATUS hr; | 
 
 
 
 
 | 501 | va_list argList; | 
 
 
 
 
 | 502 | if(cchDest > NTSTRSAFE_MAX_CCH) return STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 503 | va_start(argList,pszFormat); | 
 
 
 
 
 | 504 | hr = RtlStringVPrintfWorkerA(pszDest,cchDest,pszFormat,argList); | 
 
 
 
 
 | 505 | va_end(argList); | 
 
 
 
 
 | 506 | return hr; | 
 
 
 
 
 | 507 | } | 
 
 
 
 
 | 508 |  | 
 
 
 
 
 | 509 | NTSTRSAFEDDIV RtlStringCchPrintfW(NTSTRSAFE_PWSTR pszDest,size_t cchDest,NTSTRSAFE_PCWSTR pszFormat,...) { | 
 
 
 
 
 | 510 | NTSTATUS hr; | 
 
 
 
 
 | 511 | va_list argList; | 
 
 
 
 
 | 512 | if(cchDest > NTSTRSAFE_MAX_CCH) return STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 513 | va_start(argList,pszFormat); | 
 
 
 
 
 | 514 | hr = RtlStringVPrintfWorkerW(pszDest,cchDest,pszFormat,argList); | 
 
 
 
 
 | 515 | va_end(argList); | 
 
 
 
 
 | 516 | return hr; | 
 
 
 
 
 | 517 | } | 
 
 
 
 
 | 518 | #endif /* !__STRSAFE__NO_INLINE */ | 
 
 
 
 
 | 519 |  | 
 
 
 
 
 | 520 | NTSTRSAFEDDIV RtlStringCbPrintfA(NTSTRSAFE_PSTR pszDest,size_t cbDest,NTSTRSAFE_PCSTR pszFormat,...); | 
 
 
 
 
 | 521 | NTSTRSAFEDDIV RtlStringCbPrintfW(NTSTRSAFE_PWSTR pszDest,size_t cbDest,NTSTRSAFE_PCWSTR pszFormat,...); | 
 
 
 
 
 | 522 |  | 
 
 
 
 
 | 523 | #define RtlStringCbPrintf __MINGW_NAME_AW(RtlStringCbPrintf) | 
 
 
 
 
 | 524 |  | 
 
 
 
 
 | 525 | #ifndef __STRSAFE__NO_INLINE | 
 
 
 
 
 | 526 | NTSTRSAFEDDIV RtlStringCbPrintfA(NTSTRSAFE_PSTR pszDest,size_t cbDest,NTSTRSAFE_PCSTR pszFormat,...) { | 
 
 
 
 
 | 527 | NTSTATUS hr; | 
 
 
 
 
 | 528 | va_list argList; | 
 
 
 
 
 | 529 | if(cbDest > NTSTRSAFE_MAX_CCH) return STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 530 | va_start(argList,pszFormat); | 
 
 
 
 
 | 531 | hr = RtlStringVPrintfWorkerA(pszDest,cbDest,pszFormat,argList); | 
 
 
 
 
 | 532 | va_end(argList); | 
 
 
 
 
 | 533 | return hr; | 
 
 
 
 
 | 534 | } | 
 
 
 
 
 | 535 |  | 
 
 
 
 
 | 536 | NTSTRSAFEDDIV RtlStringCbPrintfW(NTSTRSAFE_PWSTR pszDest,size_t cbDest,NTSTRSAFE_PCWSTR pszFormat,...) { | 
 
 
 
 
 | 537 | NTSTATUS hr; | 
 
 
 
 
 | 538 | va_list argList; | 
 
 
 
 
 | 539 | size_t cchDest = cbDest / sizeof(wchar_t); | 
 
 
 
 
 | 540 | if(cchDest > NTSTRSAFE_MAX_CCH) return STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 541 | va_start(argList,pszFormat); | 
 
 
 
 
 | 542 | hr = RtlStringVPrintfWorkerW(pszDest,cchDest,pszFormat,argList); | 
 
 
 
 
 | 543 | va_end(argList); | 
 
 
 
 
 | 544 | return hr; | 
 
 
 
 
 | 545 | } | 
 
 
 
 
 | 546 | #endif /* !__STRSAFE__NO_INLINE */ | 
 
 
 
 
 | 547 |  | 
 
 
 
 
 | 548 | NTSTRSAFEDDIV RtlStringCchPrintfExA(NTSTRSAFE_PSTR pszDest,size_t cchDest,NTSTRSAFE_PSTR *ppszDestEnd,size_t *pcchRemaining,unsigned __LONG32 dwFlags,NTSTRSAFE_PCSTR pszFormat,...); | 
 
 
 
 
 | 549 | NTSTRSAFEDDIV RtlStringCchPrintfExW(NTSTRSAFE_PWSTR pszDest,size_t cchDest,NTSTRSAFE_PWSTR *ppszDestEnd,size_t *pcchRemaining,unsigned __LONG32 dwFlags,NTSTRSAFE_PCWSTR pszFormat,...); | 
 
 
 
 
 | 550 |  | 
 
 
 
 
 | 551 | #define RtlStringCchPrintfEx __MINGW_NAME_AW(RtlStringCchPrintfEx) | 
 
 
 
 
 | 552 |  | 
 
 
 
 
 | 553 | #ifndef __STRSAFE__NO_INLINE | 
 
 
 
 
 | 554 | NTSTRSAFEDDIV RtlStringCchPrintfExA(NTSTRSAFE_PSTR pszDest,size_t cchDest,NTSTRSAFE_PSTR *ppszDestEnd,size_t *pcchRemaining,unsigned __LONG32 dwFlags,NTSTRSAFE_PCSTR pszFormat,...) { | 
 
 
 
 
 | 555 | NTSTATUS hr; | 
 
 
 
 
 | 556 | va_list argList; | 
 
 
 
 
 | 557 | if(cchDest > NTSTRSAFE_MAX_CCH) return STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 558 | va_start(argList,pszFormat); | 
 
 
 
 
 | 559 | hr = RtlStringVPrintfExWorkerA(pszDest,cchDest,cchDest,ppszDestEnd,pcchRemaining,dwFlags,pszFormat,argList); | 
 
 
 
 
 | 560 | va_end(argList); | 
 
 
 
 
 | 561 | return hr; | 
 
 
 
 
 | 562 | } | 
 
 
 
 
 | 563 |  | 
 
 
 
 
 | 564 | NTSTRSAFEDDIV RtlStringCchPrintfExW(NTSTRSAFE_PWSTR pszDest,size_t cchDest,NTSTRSAFE_PWSTR *ppszDestEnd,size_t *pcchRemaining,unsigned __LONG32 dwFlags,NTSTRSAFE_PCWSTR pszFormat,...) { | 
 
 
 
 
 | 565 | NTSTATUS hr; | 
 
 
 
 
 | 566 | size_t cbDest = cchDest * sizeof(wchar_t); | 
 
 
 
 
 | 567 | va_list argList; | 
 
 
 
 
 | 568 | if(cchDest > NTSTRSAFE_MAX_CCH) return STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 569 | va_start(argList,pszFormat); | 
 
 
 
 
 | 570 | hr = RtlStringVPrintfExWorkerW(pszDest,cchDest,cbDest,ppszDestEnd,pcchRemaining,dwFlags,pszFormat,argList); | 
 
 
 
 
 | 571 | va_end(argList); | 
 
 
 
 
 | 572 | return hr; | 
 
 
 
 
 | 573 | } | 
 
 
 
 
 | 574 | #endif /* !__STRSAFE__NO_INLINE */ | 
 
 
 
 
 | 575 |  | 
 
 
 
 
 | 576 | NTSTRSAFEDDIV RtlStringCbPrintfExA(NTSTRSAFE_PSTR pszDest,size_t cbDest,NTSTRSAFE_PSTR *ppszDestEnd,size_t *pcbRemaining,unsigned __LONG32 dwFlags,NTSTRSAFE_PCSTR pszFormat,...); | 
 
 
 
 
 | 577 | NTSTRSAFEDDIV RtlStringCbPrintfExW(NTSTRSAFE_PWSTR pszDest,size_t cbDest,NTSTRSAFE_PWSTR *ppszDestEnd,size_t *pcbRemaining,unsigned __LONG32 dwFlags,NTSTRSAFE_PCWSTR pszFormat,...); | 
 
 
 
 
 | 578 |  | 
 
 
 
 
 | 579 | #define RtlStringCbPrintfEx __MINGW_NAME_AW(RtlStringCbPrintfEx) | 
 
 
 
 
 | 580 |  | 
 
 
 
 
 | 581 | #ifndef __STRSAFE__NO_INLINE | 
 
 
 
 
 | 582 | NTSTRSAFEDDIV RtlStringCbPrintfExA(NTSTRSAFE_PSTR pszDest,size_t cbDest,NTSTRSAFE_PSTR *ppszDestEnd,size_t *pcbRemaining,unsigned __LONG32 dwFlags,NTSTRSAFE_PCSTR pszFormat,...) { | 
 
 
 
 
 | 583 | NTSTATUS hr; | 
 
 
 
 
 | 584 | size_t cchDest; | 
 
 
 
 
 | 585 | size_t cchRemaining = 0; | 
 
 
 
 
 | 586 | cchDest = cbDest / sizeof(char); | 
 
 
 
 
 | 587 | if(cchDest > NTSTRSAFE_MAX_CCH) hr = STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 588 | else { | 
 
 
 
 
 | 589 | va_list argList; | 
 
 
 
 
 | 590 | va_start(argList,pszFormat); | 
 
 
 
 
 | 591 | hr = RtlStringVPrintfExWorkerA(pszDest,cchDest,cbDest,ppszDestEnd,&cchRemaining,dwFlags,pszFormat,argList); | 
 
 
 
 
 | 592 | va_end(argList); | 
 
 
 
 
 | 593 | } | 
 
 
 
 
 | 594 | if(NT_SUCCESS(hr) || (hr==STATUS_BUFFER_OVERFLOW)) { | 
 
 
 
 
 | 595 | if(pcbRemaining) { | 
 
 
 
 
 | 596 | *pcbRemaining = (cchRemaining*sizeof(char)) + (cbDest % sizeof(char)); | 
 
 
 
 
 | 597 | } | 
 
 
 
 
 | 598 | } | 
 
 
 
 
 | 599 | return hr; | 
 
 
 
 
 | 600 | } | 
 
 
 
 
 | 601 |  | 
 
 
 
 
 | 602 | NTSTRSAFEDDIV RtlStringCbPrintfExW(NTSTRSAFE_PWSTR pszDest,size_t cbDest,NTSTRSAFE_PWSTR *ppszDestEnd,size_t *pcbRemaining,unsigned __LONG32 dwFlags,NTSTRSAFE_PCWSTR pszFormat,...) { | 
 
 
 
 
 | 603 | NTSTATUS hr; | 
 
 
 
 
 | 604 | size_t cchDest; | 
 
 
 
 
 | 605 | size_t cchRemaining = 0; | 
 
 
 
 
 | 606 | cchDest = cbDest / sizeof(wchar_t); | 
 
 
 
 
 | 607 | if(cchDest > NTSTRSAFE_MAX_CCH) hr = STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 608 | else { | 
 
 
 
 
 | 609 | va_list argList; | 
 
 
 
 
 | 610 | va_start(argList,pszFormat); | 
 
 
 
 
 | 611 | hr = RtlStringVPrintfExWorkerW(pszDest,cchDest,cbDest,ppszDestEnd,&cchRemaining,dwFlags,pszFormat,argList); | 
 
 
 
 
 | 612 | va_end(argList); | 
 
 
 
 
 | 613 | } | 
 
 
 
 
 | 614 | if(NT_SUCCESS(hr) || (hr==STATUS_BUFFER_OVERFLOW)) { | 
 
 
 
 
 | 615 | if(pcbRemaining) { | 
 
 
 
 
 | 616 | *pcbRemaining = (cchRemaining*sizeof(wchar_t)) + (cbDest % sizeof(wchar_t)); | 
 
 
 
 
 | 617 | } | 
 
 
 
 
 | 618 | } | 
 
 
 
 
 | 619 | return hr; | 
 
 
 
 
 | 620 | } | 
 
 
 
 
 | 621 | #endif /* !__STRSAFE__NO_INLINE */ | 
 
 
 
 
 | 622 |  | 
 
 
 
 
 | 623 | NTSTRSAFEDDI RtlStringCchVPrintfExA(NTSTRSAFE_PSTR pszDest,size_t cchDest,NTSTRSAFE_PSTR *ppszDestEnd,size_t *pcchRemaining,unsigned __LONG32 dwFlags,NTSTRSAFE_PCSTR pszFormat,va_list argList); | 
 
 
 
 
 | 624 | NTSTRSAFEDDI RtlStringCchVPrintfExW(NTSTRSAFE_PWSTR pszDest,size_t cchDest,NTSTRSAFE_PWSTR *ppszDestEnd,size_t *pcchRemaining,unsigned __LONG32 dwFlags,NTSTRSAFE_PCWSTR pszFormat,va_list argList); | 
 
 
 
 
 | 625 |  | 
 
 
 
 
 | 626 | #define RtlStringCchVPrintfEx __MINGW_NAME_AW(RtlStringCchVPrintfEx) | 
 
 
 
 
 | 627 |  | 
 
 
 
 
 | 628 | #ifndef __STRSAFE__NO_INLINE | 
 
 
 
 
 | 629 | NTSTRSAFEDDI RtlStringCchVPrintfExA(NTSTRSAFE_PSTR pszDest,size_t cchDest,NTSTRSAFE_PSTR *ppszDestEnd,size_t *pcchRemaining,unsigned __LONG32 dwFlags,NTSTRSAFE_PCSTR pszFormat,va_list argList) { | 
 
 
 
 
 | 630 | NTSTATUS hr; | 
 
 
 
 
 | 631 | if(cchDest > NTSTRSAFE_MAX_CCH) hr = STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 632 | else { | 
 
 
 
 
 | 633 | size_t cbDest; | 
 
 
 
 
 | 634 | cbDest = cchDest*sizeof(char); | 
 
 
 
 
 | 635 | hr = RtlStringVPrintfExWorkerA(pszDest,cchDest,cbDest,ppszDestEnd,pcchRemaining,dwFlags,pszFormat,argList); | 
 
 
 
 
 | 636 | } | 
 
 
 
 
 | 637 | return hr; | 
 
 
 
 
 | 638 | } | 
 
 
 
 
 | 639 |  | 
 
 
 
 
 | 640 | NTSTRSAFEDDI RtlStringCchVPrintfExW(NTSTRSAFE_PWSTR pszDest,size_t cchDest,NTSTRSAFE_PWSTR *ppszDestEnd,size_t *pcchRemaining,unsigned __LONG32 dwFlags,NTSTRSAFE_PCWSTR pszFormat,va_list argList) { | 
 
 
 
 
 | 641 | NTSTATUS hr; | 
 
 
 
 
 | 642 | if(cchDest > NTSTRSAFE_MAX_CCH) hr = STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 643 | else { | 
 
 
 
 
 | 644 | size_t cbDest; | 
 
 
 
 
 | 645 | cbDest = cchDest*sizeof(wchar_t); | 
 
 
 
 
 | 646 | hr = RtlStringVPrintfExWorkerW(pszDest,cchDest,cbDest,ppszDestEnd,pcchRemaining,dwFlags,pszFormat,argList); | 
 
 
 
 
 | 647 | } | 
 
 
 
 
 | 648 | return hr; | 
 
 
 
 
 | 649 | } | 
 
 
 
 
 | 650 | #endif /* !__STRSAFE__NO_INLINE */ | 
 
 
 
 
 | 651 |  | 
 
 
 
 
 | 652 | NTSTRSAFEDDI RtlStringCbVPrintfExA(NTSTRSAFE_PSTR pszDest,size_t cbDest,NTSTRSAFE_PSTR *ppszDestEnd,size_t *pcbRemaining,unsigned __LONG32 dwFlags,NTSTRSAFE_PCSTR pszFormat,va_list argList); | 
 
 
 
 
 | 653 | NTSTRSAFEDDI RtlStringCbVPrintfExW(NTSTRSAFE_PWSTR pszDest,size_t cbDest,NTSTRSAFE_PWSTR *ppszDestEnd,size_t *pcbRemaining,unsigned __LONG32 dwFlags,NTSTRSAFE_PCWSTR pszFormat,va_list argList); | 
 
 
 
 
 | 654 |  | 
 
 
 
 
 | 655 | #define RtlStringCbVPrintfEx __MINGW_NAME_AW(RtlStringCbVPrintfEx) | 
 
 
 
 
 | 656 |  | 
 
 
 
 
 | 657 | #ifndef __STRSAFE__NO_INLINE | 
 
 
 
 
 | 658 | NTSTRSAFEDDI RtlStringCbVPrintfExA(NTSTRSAFE_PSTR pszDest,size_t cbDest,NTSTRSAFE_PSTR *ppszDestEnd,size_t *pcbRemaining,unsigned __LONG32 dwFlags,NTSTRSAFE_PCSTR pszFormat,va_list argList) { | 
 
 
 
 
 | 659 | NTSTATUS hr; | 
 
 
 
 
 | 660 | size_t cchDest; | 
 
 
 
 
 | 661 | size_t cchRemaining = 0; | 
 
 
 
 
 | 662 | cchDest = cbDest / sizeof(char); | 
 
 
 
 
 | 663 | if(cchDest > NTSTRSAFE_MAX_CCH) hr = STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 664 | else hr = RtlStringVPrintfExWorkerA(pszDest,cchDest,cbDest,ppszDestEnd,&cchRemaining,dwFlags,pszFormat,argList); | 
 
 
 
 
 | 665 | if(NT_SUCCESS(hr) || (hr==STATUS_BUFFER_OVERFLOW)) { | 
 
 
 
 
 | 666 | if(pcbRemaining) { | 
 
 
 
 
 | 667 | *pcbRemaining = (cchRemaining*sizeof(char)) + (cbDest % sizeof(char)); | 
 
 
 
 
 | 668 | } | 
 
 
 
 
 | 669 | } | 
 
 
 
 
 | 670 | return hr; | 
 
 
 
 
 | 671 | } | 
 
 
 
 
 | 672 |  | 
 
 
 
 
 | 673 | NTSTRSAFEDDI RtlStringCbVPrintfExW(NTSTRSAFE_PWSTR pszDest,size_t cbDest,NTSTRSAFE_PWSTR *ppszDestEnd,size_t *pcbRemaining,unsigned __LONG32 dwFlags,NTSTRSAFE_PCWSTR pszFormat,va_list argList) { | 
 
 
 
 
 | 674 | NTSTATUS hr; | 
 
 
 
 
 | 675 | size_t cchDest; | 
 
 
 
 
 | 676 | size_t cchRemaining = 0; | 
 
 
 
 
 | 677 | cchDest = cbDest / sizeof(wchar_t); | 
 
 
 
 
 | 678 | if(cchDest > NTSTRSAFE_MAX_CCH) hr = STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 679 | else hr = RtlStringVPrintfExWorkerW(pszDest,cchDest,cbDest,ppszDestEnd,&cchRemaining,dwFlags,pszFormat,argList); | 
 
 
 
 
 | 680 | if(NT_SUCCESS(hr) || (hr==STATUS_BUFFER_OVERFLOW)) { | 
 
 
 
 
 | 681 | if(pcbRemaining) { | 
 
 
 
 
 | 682 | *pcbRemaining = (cchRemaining*sizeof(wchar_t)) + (cbDest % sizeof(wchar_t)); | 
 
 
 
 
 | 683 | } | 
 
 
 
 
 | 684 | } | 
 
 
 
 
 | 685 | return hr; | 
 
 
 
 
 | 686 | } | 
 
 
 
 
 | 687 | #endif /* !__STRSAFE__NO_INLINE */ | 
 
 
 
 
 | 688 |  | 
 
 
 
 
 | 689 | NTSTRSAFEDDI RtlStringCchLengthA(NTSTRSAFE_PCSTR psz,size_t cchMax,size_t *pcchLength); | 
 
 
 
 
 | 690 | NTSTRSAFEDDI RtlStringCchLengthW(NTSTRSAFE_PCWSTR psz,size_t cchMax,size_t *pcchLength); | 
 
 
 
 
 | 691 |  | 
 
 
 
 
 | 692 | #define RtlStringCchLength __MINGW_NAME_AW(RtlStringCchLength) | 
 
 
 
 
 | 693 |  | 
 
 
 
 
 | 694 | #ifndef __STRSAFE__NO_INLINE | 
 
 
 
 
 | 695 | NTSTRSAFEDDI RtlStringCchLengthA(NTSTRSAFE_PCSTR psz,size_t cchMax,size_t *pcchLength) { | 
 
 
 
 
 | 696 | NTSTATUS hr; | 
 
 
 
 
 | 697 | if(!psz || (cchMax > NTSTRSAFE_MAX_CCH)) hr = STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 698 | else hr = RtlStringLengthWorkerA(psz,cchMax,pcchLength); | 
 
 
 
 
 | 699 | if(!NT_SUCCESS(hr) && pcchLength) { | 
 
 
 
 
 | 700 | *pcchLength = 0; | 
 
 
 
 
 | 701 | } | 
 
 
 
 
 | 702 | return hr; | 
 
 
 
 
 | 703 | } | 
 
 
 
 
 | 704 |  | 
 
 
 
 
 | 705 | NTSTRSAFEDDI RtlStringCchLengthW(NTSTRSAFE_PCWSTR psz,size_t cchMax,size_t *pcchLength) { | 
 
 
 
 
 | 706 | NTSTATUS hr; | 
 
 
 
 
 | 707 | if(!psz || (cchMax > NTSTRSAFE_MAX_CCH)) hr = STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 708 | else hr = RtlStringLengthWorkerW(psz,cchMax,pcchLength); | 
 
 
 
 
 | 709 | if(!NT_SUCCESS(hr) && pcchLength) { | 
 
 
 
 
 | 710 | *pcchLength = 0; | 
 
 
 
 
 | 711 | } | 
 
 
 
 
 | 712 | return hr; | 
 
 
 
 
 | 713 | } | 
 
 
 
 
 | 714 | #endif /* !__STRSAFE__NO_INLINE */ | 
 
 
 
 
 | 715 |  | 
 
 
 
 
 | 716 | NTSTRSAFEDDI RtlStringCbLengthA(NTSTRSAFE_PCSTR psz,size_t cbMax,size_t *pcbLength); | 
 
 
 
 
 | 717 | NTSTRSAFEDDI RtlStringCbLengthW(NTSTRSAFE_PCWSTR psz,size_t cbMax,size_t *pcbLength); | 
 
 
 
 
 | 718 |  | 
 
 
 
 
 | 719 | #define RtlStringCbLength __MINGW_NAME_AW(RtlStringCbLength) | 
 
 
 
 
 | 720 |  | 
 
 
 
 
 | 721 | #ifndef __STRSAFE__NO_INLINE | 
 
 
 
 
 | 722 | NTSTRSAFEDDI RtlStringCbLengthA(NTSTRSAFE_PCSTR psz,size_t cbMax,size_t *pcbLength) { | 
 
 
 
 
 | 723 | NTSTATUS hr; | 
 
 
 
 
 | 724 | size_t cchMax; | 
 
 
 
 
 | 725 | size_t cchLength = 0; | 
 
 
 
 
 | 726 | cchMax = cbMax / sizeof(char); | 
 
 
 
 
 | 727 | if(!psz || (cchMax > NTSTRSAFE_MAX_CCH)) hr = STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 728 | else hr = RtlStringLengthWorkerA(psz,cchMax,&cchLength); | 
 
 
 
 
 | 729 | if(pcbLength) { | 
 
 
 
 
 | 730 | if(NT_SUCCESS(hr)) { | 
 
 
 
 
 | 731 | *pcbLength = cchLength*sizeof(char); | 
 
 
 
 
 | 732 | } else { | 
 
 
 
 
 | 733 | *pcbLength = 0; | 
 
 
 
 
 | 734 | } | 
 
 
 
 
 | 735 | } | 
 
 
 
 
 | 736 | return hr; | 
 
 
 
 
 | 737 | } | 
 
 
 
 
 | 738 |  | 
 
 
 
 
 | 739 | NTSTRSAFEDDI RtlStringCbLengthW(NTSTRSAFE_PCWSTR psz,size_t cbMax,size_t *pcbLength) { | 
 
 
 
 
 | 740 | NTSTATUS hr; | 
 
 
 
 
 | 741 | size_t cchMax; | 
 
 
 
 
 | 742 | size_t cchLength = 0; | 
 
 
 
 
 | 743 | cchMax = cbMax / sizeof(wchar_t); | 
 
 
 
 
 | 744 | if(!psz || (cchMax > NTSTRSAFE_MAX_CCH)) hr = STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 745 | else hr = RtlStringLengthWorkerW(psz,cchMax,&cchLength); | 
 
 
 
 
 | 746 | if(pcbLength) { | 
 
 
 
 
 | 747 | if(NT_SUCCESS(hr)) { | 
 
 
 
 
 | 748 | *pcbLength = cchLength*sizeof(wchar_t); | 
 
 
 
 
 | 749 | } else { | 
 
 
 
 
 | 750 | *pcbLength = 0; | 
 
 
 
 
 | 751 | } | 
 
 
 
 
 | 752 | } | 
 
 
 
 
 | 753 | return hr; | 
 
 
 
 
 | 754 | } | 
 
 
 
 
 | 755 |  | 
 
 
 
 
 | 756 | NTSTRSAFEDDI RtlStringCopyWorkerA(NTSTRSAFE_PSTR pszDest,size_t cchDest,NTSTRSAFE_PCSTR pszSrc) { | 
 
 
 
 
 | 757 | NTSTATUS hr = STATUS_SUCCESS; | 
 
 
 
 
 | 758 | if(cchDest==0) hr = STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 759 | else { | 
 
 
 
 
 | 760 | while(cchDest && (*pszSrc!='\0')) { | 
 
 
 
 
 | 761 | *pszDest++ = *pszSrc++; | 
 
 
 
 
 | 762 | cchDest--; | 
 
 
 
 
 | 763 | } | 
 
 
 
 
 | 764 | if(cchDest==0) { | 
 
 
 
 
 | 765 | pszDest--; | 
 
 
 
 
 | 766 | hr = STATUS_BUFFER_OVERFLOW; | 
 
 
 
 
 | 767 | } | 
 
 
 
 
 | 768 | *pszDest= '\0'; | 
 
 
 
 
 | 769 | } | 
 
 
 
 
 | 770 | return hr; | 
 
 
 
 
 | 771 | } | 
 
 
 
 
 | 772 |  | 
 
 
 
 
 | 773 | NTSTRSAFEDDI RtlStringCopyWorkerW(NTSTRSAFE_PWSTR pszDest,size_t cchDest,NTSTRSAFE_PCWSTR pszSrc) { | 
 
 
 
 
 | 774 | NTSTATUS hr = STATUS_SUCCESS; | 
 
 
 
 
 | 775 | if(cchDest==0) hr = STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 776 | else { | 
 
 
 
 
 | 777 | while(cchDest && (*pszSrc!=L'\0')) { | 
 
 
 
 
 | 778 | *pszDest++ = *pszSrc++; | 
 
 
 
 
 | 779 | cchDest--; | 
 
 
 
 
 | 780 | } | 
 
 
 
 
 | 781 | if(cchDest==0) { | 
 
 
 
 
 | 782 | pszDest--; | 
 
 
 
 
 | 783 | hr = STATUS_BUFFER_OVERFLOW; | 
 
 
 
 
 | 784 | } | 
 
 
 
 
 | 785 | *pszDest= L'\0'; | 
 
 
 
 
 | 786 | } | 
 
 
 
 
 | 787 | return hr; | 
 
 
 
 
 | 788 | } | 
 
 
 
 
 | 789 |  | 
 
 
 
 
 | 790 | NTSTRSAFEDDI RtlStringCopyExWorkerA(NTSTRSAFE_PSTR pszDest,size_t cchDest,size_t cbDest,NTSTRSAFE_PCSTR pszSrc,NTSTRSAFE_PSTR *ppszDestEnd,size_t *pcchRemaining,unsigned __LONG32 dwFlags) { | 
 
 
 
 
 | 791 | NTSTATUS hr = STATUS_SUCCESS; | 
 
 
 
 
 | 792 | NTSTRSAFE_PSTR pszDestEnd = pszDest; | 
 
 
 
 
 | 793 | size_t cchRemaining = 0; | 
 
 
 
 
 | 794 | if(dwFlags & (~STRSAFE_VALID_FLAGS)) hr = STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 795 | else { | 
 
 
 
 
 | 796 | if(dwFlags & STRSAFE_IGNORE_NULLS) { | 
 
 
 
 
 | 797 | if(!pszDest) { | 
 
 
 
 
 | 798 | if((cchDest!=0) || (cbDest!=0)) hr = STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 799 | } | 
 
 
 
 
 | 800 | if(!pszSrc) pszSrc = ""; | 
 
 
 
 
 | 801 | } | 
 
 
 
 
 | 802 | if(NT_SUCCESS(hr)) { | 
 
 
 
 
 | 803 | if(cchDest==0) { | 
 
 
 
 
 | 804 | pszDestEnd = pszDest; | 
 
 
 
 
 | 805 | cchRemaining = 0; | 
 
 
 
 
 | 806 | if(*pszSrc!='\0') { | 
 
 
 
 
 | 807 | if(!pszDest) hr = STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 808 | else hr = STATUS_BUFFER_OVERFLOW; | 
 
 
 
 
 | 809 | } | 
 
 
 
 
 | 810 | } else { | 
 
 
 
 
 | 811 | pszDestEnd = pszDest; | 
 
 
 
 
 | 812 | cchRemaining = cchDest; | 
 
 
 
 
 | 813 | while(cchRemaining && (*pszSrc!='\0')) { | 
 
 
 
 
 | 814 | *pszDestEnd++ = *pszSrc++; | 
 
 
 
 
 | 815 | cchRemaining--; | 
 
 
 
 
 | 816 | } | 
 
 
 
 
 | 817 | if(cchRemaining > 0) { | 
 
 
 
 
 | 818 | if(dwFlags & STRSAFE_FILL_BEHIND_NULL) { | 
 
 
 
 
 | 819 | memset(pszDestEnd + 1,STRSAFE_GET_FILL_PATTERN(dwFlags),((cchRemaining - 1)*sizeof(char)) + (cbDest % sizeof(char))); | 
 
 
 
 
 | 820 | } | 
 
 
 
 
 | 821 | } else { | 
 
 
 
 
 | 822 | pszDestEnd--; | 
 
 
 
 
 | 823 | cchRemaining++; | 
 
 
 
 
 | 824 | hr = STATUS_BUFFER_OVERFLOW; | 
 
 
 
 
 | 825 | } | 
 
 
 
 
 | 826 | *pszDestEnd = '\0'; | 
 
 
 
 
 | 827 | } | 
 
 
 
 
 | 828 | } | 
 
 
 
 
 | 829 | } | 
 
 
 
 
 | 830 | if(!NT_SUCCESS(hr)) { | 
 
 
 
 
 | 831 | if(pszDest) { | 
 
 
 
 
 | 832 | if(dwFlags & STRSAFE_FILL_ON_FAILURE) { | 
 
 
 
 
 | 833 | memset(pszDest,STRSAFE_GET_FILL_PATTERN(dwFlags),cbDest); | 
 
 
 
 
 | 834 | if(STRSAFE_GET_FILL_PATTERN(dwFlags)==0) { | 
 
 
 
 
 | 835 | pszDestEnd = pszDest; | 
 
 
 
 
 | 836 | cchRemaining = cchDest; | 
 
 
 
 
 | 837 | } else if(cchDest > 0) { | 
 
 
 
 
 | 838 | pszDestEnd = pszDest + cchDest - 1; | 
 
 
 
 
 | 839 | cchRemaining = 1; | 
 
 
 
 
 | 840 | *pszDestEnd = '\0'; | 
 
 
 
 
 | 841 | } | 
 
 
 
 
 | 842 | } | 
 
 
 
 
 | 843 | if(dwFlags & (STRSAFE_NULL_ON_FAILURE | STRSAFE_NO_TRUNCATION)) { | 
 
 
 
 
 | 844 | if(cchDest > 0) { | 
 
 
 
 
 | 845 | pszDestEnd = pszDest; | 
 
 
 
 
 | 846 | cchRemaining = cchDest; | 
 
 
 
 
 | 847 | *pszDestEnd = '\0'; | 
 
 
 
 
 | 848 | } | 
 
 
 
 
 | 849 | } | 
 
 
 
 
 | 850 | } | 
 
 
 
 
 | 851 | } | 
 
 
 
 
 | 852 | if(NT_SUCCESS(hr) || (hr==STATUS_BUFFER_OVERFLOW)) { | 
 
 
 
 
 | 853 | if(ppszDestEnd) *ppszDestEnd = pszDestEnd; | 
 
 
 
 
 | 854 | if(pcchRemaining) *pcchRemaining = cchRemaining; | 
 
 
 
 
 | 855 | } | 
 
 
 
 
 | 856 | return hr; | 
 
 
 
 
 | 857 | } | 
 
 
 
 
 | 858 |  | 
 
 
 
 
 | 859 | NTSTRSAFEDDI RtlStringCopyExWorkerW(NTSTRSAFE_PWSTR pszDest,size_t cchDest,size_t cbDest,NTSTRSAFE_PCWSTR pszSrc,NTSTRSAFE_PWSTR *ppszDestEnd,size_t *pcchRemaining,unsigned __LONG32 dwFlags) { | 
 
 
 
 
 | 860 | NTSTATUS hr = STATUS_SUCCESS; | 
 
 
 
 
 | 861 | NTSTRSAFE_PWSTR pszDestEnd = pszDest; | 
 
 
 
 
 | 862 | size_t cchRemaining = 0; | 
 
 
 
 
 | 863 | if(dwFlags & (~STRSAFE_VALID_FLAGS)) hr = STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 864 | else { | 
 
 
 
 
 | 865 | if(dwFlags & STRSAFE_IGNORE_NULLS) { | 
 
 
 
 
 | 866 | if(!pszDest) { | 
 
 
 
 
 | 867 | if((cchDest!=0) || (cbDest!=0)) hr = STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 868 | } | 
 
 
 
 
 | 869 | if(!pszSrc) pszSrc = L""; | 
 
 
 
 
 | 870 | } | 
 
 
 
 
 | 871 | if(NT_SUCCESS(hr)) { | 
 
 
 
 
 | 872 | if(cchDest==0) { | 
 
 
 
 
 | 873 | pszDestEnd = pszDest; | 
 
 
 
 
 | 874 | cchRemaining = 0; | 
 
 
 
 
 | 875 | if(*pszSrc!=L'\0') { | 
 
 
 
 
 | 876 | if(!pszDest) hr = STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 877 | else hr = STATUS_BUFFER_OVERFLOW; | 
 
 
 
 
 | 878 | } | 
 
 
 
 
 | 879 | } else { | 
 
 
 
 
 | 880 | pszDestEnd = pszDest; | 
 
 
 
 
 | 881 | cchRemaining = cchDest; | 
 
 
 
 
 | 882 | while(cchRemaining && (*pszSrc!=L'\0')) { | 
 
 
 
 
 | 883 | *pszDestEnd++ = *pszSrc++; | 
 
 
 
 
 | 884 | cchRemaining--; | 
 
 
 
 
 | 885 | } | 
 
 
 
 
 | 886 | if(cchRemaining > 0) { | 
 
 
 
 
 | 887 | if(dwFlags & STRSAFE_FILL_BEHIND_NULL) { | 
 
 
 
 
 | 888 | memset(pszDestEnd + 1,STRSAFE_GET_FILL_PATTERN(dwFlags),((cchRemaining - 1)*sizeof(wchar_t)) + (cbDest % sizeof(wchar_t))); | 
 
 
 
 
 | 889 | } | 
 
 
 
 
 | 890 | } else { | 
 
 
 
 
 | 891 | pszDestEnd--; | 
 
 
 
 
 | 892 | cchRemaining++; | 
 
 
 
 
 | 893 | hr = STATUS_BUFFER_OVERFLOW; | 
 
 
 
 
 | 894 | } | 
 
 
 
 
 | 895 | *pszDestEnd = L'\0'; | 
 
 
 
 
 | 896 | } | 
 
 
 
 
 | 897 | } | 
 
 
 
 
 | 898 | } | 
 
 
 
 
 | 899 | if(!NT_SUCCESS(hr)) { | 
 
 
 
 
 | 900 | if(pszDest) { | 
 
 
 
 
 | 901 | if(dwFlags & STRSAFE_FILL_ON_FAILURE) { | 
 
 
 
 
 | 902 | memset(pszDest,STRSAFE_GET_FILL_PATTERN(dwFlags),cbDest); | 
 
 
 
 
 | 903 | if(STRSAFE_GET_FILL_PATTERN(dwFlags)==0) { | 
 
 
 
 
 | 904 | pszDestEnd = pszDest; | 
 
 
 
 
 | 905 | cchRemaining = cchDest; | 
 
 
 
 
 | 906 | } else if(cchDest > 0) { | 
 
 
 
 
 | 907 | pszDestEnd = pszDest + cchDest - 1; | 
 
 
 
 
 | 908 | cchRemaining = 1; | 
 
 
 
 
 | 909 | *pszDestEnd = L'\0'; | 
 
 
 
 
 | 910 | } | 
 
 
 
 
 | 911 | } | 
 
 
 
 
 | 912 | if(dwFlags & (STRSAFE_NULL_ON_FAILURE | STRSAFE_NO_TRUNCATION)) { | 
 
 
 
 
 | 913 | if(cchDest > 0) { | 
 
 
 
 
 | 914 | pszDestEnd = pszDest; | 
 
 
 
 
 | 915 | cchRemaining = cchDest; | 
 
 
 
 
 | 916 | *pszDestEnd = L'\0'; | 
 
 
 
 
 | 917 | } | 
 
 
 
 
 | 918 | } | 
 
 
 
 
 | 919 | } | 
 
 
 
 
 | 920 | } | 
 
 
 
 
 | 921 | if(NT_SUCCESS(hr) || (hr==STATUS_BUFFER_OVERFLOW)) { | 
 
 
 
 
 | 922 | if(ppszDestEnd) *ppszDestEnd = pszDestEnd; | 
 
 
 
 
 | 923 | if(pcchRemaining) *pcchRemaining = cchRemaining; | 
 
 
 
 
 | 924 | } | 
 
 
 
 
 | 925 | return hr; | 
 
 
 
 
 | 926 | } | 
 
 
 
 
 | 927 |  | 
 
 
 
 
 | 928 | NTSTRSAFEDDI RtlStringCopyNWorkerA(NTSTRSAFE_PSTR pszDest,size_t cchDest,NTSTRSAFE_PCSTR pszSrc,size_t cchSrc) { | 
 
 
 
 
 | 929 | NTSTATUS hr = STATUS_SUCCESS; | 
 
 
 
 
 | 930 | if(cchDest==0) hr = STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 931 | else { | 
 
 
 
 
 | 932 | while(cchDest && cchSrc && (*pszSrc!='\0')) { | 
 
 
 
 
 | 933 | *pszDest++ = *pszSrc++; | 
 
 
 
 
 | 934 | cchDest--; | 
 
 
 
 
 | 935 | cchSrc--; | 
 
 
 
 
 | 936 | } | 
 
 
 
 
 | 937 | if(cchDest==0) { | 
 
 
 
 
 | 938 | pszDest--; | 
 
 
 
 
 | 939 | hr = STATUS_BUFFER_OVERFLOW; | 
 
 
 
 
 | 940 | } | 
 
 
 
 
 | 941 | *pszDest= '\0'; | 
 
 
 
 
 | 942 | } | 
 
 
 
 
 | 943 | return hr; | 
 
 
 
 
 | 944 | } | 
 
 
 
 
 | 945 |  | 
 
 
 
 
 | 946 | NTSTRSAFEDDI RtlStringCopyNWorkerW(NTSTRSAFE_PWSTR pszDest,size_t cchDest,NTSTRSAFE_PCWSTR pszSrc,size_t cchToCopy) { | 
 
 
 
 
 | 947 | NTSTATUS hr = STATUS_SUCCESS; | 
 
 
 
 
 | 948 | if(cchDest==0) hr = STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 949 | else { | 
 
 
 
 
 | 950 | while(cchDest && cchToCopy && (*pszSrc!=L'\0')) { | 
 
 
 
 
 | 951 | *pszDest++ = *pszSrc++; | 
 
 
 
 
 | 952 | cchDest--; | 
 
 
 
 
 | 953 | cchToCopy--; | 
 
 
 
 
 | 954 | } | 
 
 
 
 
 | 955 | if(cchDest==0) { | 
 
 
 
 
 | 956 | pszDest--; | 
 
 
 
 
 | 957 | hr = STATUS_BUFFER_OVERFLOW; | 
 
 
 
 
 | 958 | } | 
 
 
 
 
 | 959 | *pszDest= L'\0'; | 
 
 
 
 
 | 960 | } | 
 
 
 
 
 | 961 | return hr; | 
 
 
 
 
 | 962 | } | 
 
 
 
 
 | 963 |  | 
 
 
 
 
 | 964 | NTSTRSAFEDDI RtlStringCopyNExWorkerA(NTSTRSAFE_PSTR pszDest,size_t cchDest,size_t cbDest,NTSTRSAFE_PCSTR pszSrc,size_t cchToCopy,NTSTRSAFE_PSTR *ppszDestEnd,size_t *pcchRemaining,unsigned __LONG32 dwFlags) { | 
 
 
 
 
 | 965 | NTSTATUS hr = STATUS_SUCCESS; | 
 
 
 
 
 | 966 | NTSTRSAFE_PSTR pszDestEnd = pszDest; | 
 
 
 
 
 | 967 | size_t cchRemaining = 0; | 
 
 
 
 
 | 968 | if(dwFlags & (~STRSAFE_VALID_FLAGS)) hr = STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 969 | else if(cchToCopy > NTSTRSAFE_MAX_CCH) hr = STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 970 | else { | 
 
 
 
 
 | 971 | if(dwFlags & STRSAFE_IGNORE_NULLS) { | 
 
 
 
 
 | 972 | if(!pszDest) { | 
 
 
 
 
 | 973 | if((cchDest!=0) || (cbDest!=0)) hr = STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 974 | } | 
 
 
 
 
 | 975 | if(!pszSrc) pszSrc = ""; | 
 
 
 
 
 | 976 | } | 
 
 
 
 
 | 977 | if(NT_SUCCESS(hr)) { | 
 
 
 
 
 | 978 | if(cchDest==0) { | 
 
 
 
 
 | 979 | pszDestEnd = pszDest; | 
 
 
 
 
 | 980 | cchRemaining = 0; | 
 
 
 
 
 | 981 | if((cchToCopy!=0) && (*pszSrc!='\0')) { | 
 
 
 
 
 | 982 | if(!pszDest) hr = STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 983 | else hr = STATUS_BUFFER_OVERFLOW; | 
 
 
 
 
 | 984 | } | 
 
 
 
 
 | 985 | } else { | 
 
 
 
 
 | 986 | pszDestEnd = pszDest; | 
 
 
 
 
 | 987 | cchRemaining = cchDest; | 
 
 
 
 
 | 988 | while(cchRemaining && cchToCopy && (*pszSrc!='\0')) { | 
 
 
 
 
 | 989 | *pszDestEnd++ = *pszSrc++; | 
 
 
 
 
 | 990 | cchRemaining--; | 
 
 
 
 
 | 991 | cchToCopy--; | 
 
 
 
 
 | 992 | } | 
 
 
 
 
 | 993 | if(cchRemaining > 0) { | 
 
 
 
 
 | 994 | if(dwFlags & STRSAFE_FILL_BEHIND_NULL) { | 
 
 
 
 
 | 995 | memset(pszDestEnd + 1,STRSAFE_GET_FILL_PATTERN(dwFlags),((cchRemaining - 1)*sizeof(char)) + (cbDest % sizeof(char))); | 
 
 
 
 
 | 996 | } | 
 
 
 
 
 | 997 | } else { | 
 
 
 
 
 | 998 | pszDestEnd--; | 
 
 
 
 
 | 999 | cchRemaining++; | 
 
 
 
 
 | 1000 | hr = STATUS_BUFFER_OVERFLOW; | 
 
 
 
 
 | 1001 | } | 
 
 
 
 
 | 1002 | *pszDestEnd = '\0'; | 
 
 
 
 
 | 1003 | } | 
 
 
 
 
 | 1004 | } | 
 
 
 
 
 | 1005 | } | 
 
 
 
 
 | 1006 | if(!NT_SUCCESS(hr)) { | 
 
 
 
 
 | 1007 | if(pszDest) { | 
 
 
 
 
 | 1008 | if(dwFlags & STRSAFE_FILL_ON_FAILURE) { | 
 
 
 
 
 | 1009 | memset(pszDest,STRSAFE_GET_FILL_PATTERN(dwFlags),cbDest); | 
 
 
 
 
 | 1010 | if(STRSAFE_GET_FILL_PATTERN(dwFlags)==0) { | 
 
 
 
 
 | 1011 | pszDestEnd = pszDest; | 
 
 
 
 
 | 1012 | cchRemaining = cchDest; | 
 
 
 
 
 | 1013 | } else if(cchDest > 0) { | 
 
 
 
 
 | 1014 | pszDestEnd = pszDest + cchDest - 1; | 
 
 
 
 
 | 1015 | cchRemaining = 1; | 
 
 
 
 
 | 1016 | *pszDestEnd = '\0'; | 
 
 
 
 
 | 1017 | } | 
 
 
 
 
 | 1018 | } | 
 
 
 
 
 | 1019 | if(dwFlags & (STRSAFE_NULL_ON_FAILURE | STRSAFE_NO_TRUNCATION)) { | 
 
 
 
 
 | 1020 | if(cchDest > 0) { | 
 
 
 
 
 | 1021 | pszDestEnd = pszDest; | 
 
 
 
 
 | 1022 | cchRemaining = cchDest; | 
 
 
 
 
 | 1023 | *pszDestEnd = '\0'; | 
 
 
 
 
 | 1024 | } | 
 
 
 
 
 | 1025 | } | 
 
 
 
 
 | 1026 | } | 
 
 
 
 
 | 1027 | } | 
 
 
 
 
 | 1028 | if(NT_SUCCESS(hr) || (hr==STATUS_BUFFER_OVERFLOW)) { | 
 
 
 
 
 | 1029 | if(ppszDestEnd) *ppszDestEnd = pszDestEnd; | 
 
 
 
 
 | 1030 | if(pcchRemaining) *pcchRemaining = cchRemaining; | 
 
 
 
 
 | 1031 | } | 
 
 
 
 
 | 1032 | return hr; | 
 
 
 
 
 | 1033 | } | 
 
 
 
 
 | 1034 |  | 
 
 
 
 
 | 1035 | NTSTRSAFEDDI RtlStringCopyNExWorkerW(NTSTRSAFE_PWSTR pszDest,size_t cchDest,size_t cbDest,NTSTRSAFE_PCWSTR pszSrc,size_t cchToCopy,NTSTRSAFE_PWSTR *ppszDestEnd,size_t *pcchRemaining,unsigned __LONG32 dwFlags) { | 
 
 
 
 
 | 1036 | NTSTATUS hr = STATUS_SUCCESS; | 
 
 
 
 
 | 1037 | NTSTRSAFE_PWSTR pszDestEnd = pszDest; | 
 
 
 
 
 | 1038 | size_t cchRemaining = 0; | 
 
 
 
 
 | 1039 | if(dwFlags & (~STRSAFE_VALID_FLAGS)) hr = STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 1040 | else if(cchToCopy > NTSTRSAFE_MAX_CCH) hr = STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 1041 | else { | 
 
 
 
 
 | 1042 | if(dwFlags & STRSAFE_IGNORE_NULLS) { | 
 
 
 
 
 | 1043 | if(!pszDest) { | 
 
 
 
 
 | 1044 | if((cchDest!=0) || (cbDest!=0)) hr = STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 1045 | } | 
 
 
 
 
 | 1046 | if(!pszSrc) pszSrc = L""; | 
 
 
 
 
 | 1047 | } | 
 
 
 
 
 | 1048 | if(NT_SUCCESS(hr)) { | 
 
 
 
 
 | 1049 | if(cchDest==0) { | 
 
 
 
 
 | 1050 | pszDestEnd = pszDest; | 
 
 
 
 
 | 1051 | cchRemaining = 0; | 
 
 
 
 
 | 1052 | if((cchToCopy!=0) && (*pszSrc!=L'\0')) { | 
 
 
 
 
 | 1053 | if(!pszDest) hr = STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 1054 | else hr = STATUS_BUFFER_OVERFLOW; | 
 
 
 
 
 | 1055 | } | 
 
 
 
 
 | 1056 | } else { | 
 
 
 
 
 | 1057 | pszDestEnd = pszDest; | 
 
 
 
 
 | 1058 | cchRemaining = cchDest; | 
 
 
 
 
 | 1059 | while(cchRemaining && cchToCopy && (*pszSrc!=L'\0')) { | 
 
 
 
 
 | 1060 | *pszDestEnd++ = *pszSrc++; | 
 
 
 
 
 | 1061 | cchRemaining--; | 
 
 
 
 
 | 1062 | cchToCopy--; | 
 
 
 
 
 | 1063 | } | 
 
 
 
 
 | 1064 | if(cchRemaining > 0) { | 
 
 
 
 
 | 1065 | if(dwFlags & STRSAFE_FILL_BEHIND_NULL) { | 
 
 
 
 
 | 1066 | memset(pszDestEnd + 1,STRSAFE_GET_FILL_PATTERN(dwFlags),((cchRemaining - 1)*sizeof(wchar_t)) + (cbDest % sizeof(wchar_t))); | 
 
 
 
 
 | 1067 | } | 
 
 
 
 
 | 1068 | } else { | 
 
 
 
 
 | 1069 | pszDestEnd--; | 
 
 
 
 
 | 1070 | cchRemaining++; | 
 
 
 
 
 | 1071 | hr = STATUS_BUFFER_OVERFLOW; | 
 
 
 
 
 | 1072 | } | 
 
 
 
 
 | 1073 | *pszDestEnd = L'\0'; | 
 
 
 
 
 | 1074 | } | 
 
 
 
 
 | 1075 | } | 
 
 
 
 
 | 1076 | } | 
 
 
 
 
 | 1077 | if(!NT_SUCCESS(hr)) { | 
 
 
 
 
 | 1078 | if(pszDest) { | 
 
 
 
 
 | 1079 | if(dwFlags & STRSAFE_FILL_ON_FAILURE) { | 
 
 
 
 
 | 1080 | memset(pszDest,STRSAFE_GET_FILL_PATTERN(dwFlags),cbDest); | 
 
 
 
 
 | 1081 | if(STRSAFE_GET_FILL_PATTERN(dwFlags)==0) { | 
 
 
 
 
 | 1082 | pszDestEnd = pszDest; | 
 
 
 
 
 | 1083 | cchRemaining = cchDest; | 
 
 
 
 
 | 1084 | } else if(cchDest > 0) { | 
 
 
 
 
 | 1085 | pszDestEnd = pszDest + cchDest - 1; | 
 
 
 
 
 | 1086 | cchRemaining = 1; | 
 
 
 
 
 | 1087 | *pszDestEnd = L'\0'; | 
 
 
 
 
 | 1088 | } | 
 
 
 
 
 | 1089 | } | 
 
 
 
 
 | 1090 | if(dwFlags & (STRSAFE_NULL_ON_FAILURE | STRSAFE_NO_TRUNCATION)) { | 
 
 
 
 
 | 1091 | if(cchDest > 0) { | 
 
 
 
 
 | 1092 | pszDestEnd = pszDest; | 
 
 
 
 
 | 1093 | cchRemaining = cchDest; | 
 
 
 
 
 | 1094 | *pszDestEnd = L'\0'; | 
 
 
 
 
 | 1095 | } | 
 
 
 
 
 | 1096 | } | 
 
 
 
 
 | 1097 | } | 
 
 
 
 
 | 1098 | } | 
 
 
 
 
 | 1099 | if(NT_SUCCESS(hr) || (hr==STATUS_BUFFER_OVERFLOW)) { | 
 
 
 
 
 | 1100 | if(ppszDestEnd) *ppszDestEnd = pszDestEnd; | 
 
 
 
 
 | 1101 | if(pcchRemaining) *pcchRemaining = cchRemaining; | 
 
 
 
 
 | 1102 | } | 
 
 
 
 
 | 1103 | return hr; | 
 
 
 
 
 | 1104 | } | 
 
 
 
 
 | 1105 |  | 
 
 
 
 
 | 1106 | NTSTRSAFEDDI RtlStringCatWorkerA(NTSTRSAFE_PSTR pszDest,size_t cchDest,NTSTRSAFE_PCSTR pszSrc) { | 
 
 
 
 
 | 1107 | NTSTATUS hr; | 
 
 
 
 
 | 1108 | size_t cchDestLength; | 
 
 
 
 
 | 1109 | hr = RtlStringLengthWorkerA(pszDest,cchDest,&cchDestLength); | 
 
 
 
 
 | 1110 | if(NT_SUCCESS(hr)) hr = RtlStringCopyWorkerA(pszDest + cchDestLength,cchDest - cchDestLength,pszSrc); | 
 
 
 
 
 | 1111 | return hr; | 
 
 
 
 
 | 1112 | } | 
 
 
 
 
 | 1113 |  | 
 
 
 
 
 | 1114 | NTSTRSAFEDDI RtlStringCatWorkerW(NTSTRSAFE_PWSTR pszDest,size_t cchDest,NTSTRSAFE_PCWSTR pszSrc) { | 
 
 
 
 
 | 1115 | NTSTATUS hr; | 
 
 
 
 
 | 1116 | size_t cchDestLength; | 
 
 
 
 
 | 1117 | hr = RtlStringLengthWorkerW(pszDest,cchDest,&cchDestLength); | 
 
 
 
 
 | 1118 | if(NT_SUCCESS(hr)) hr = RtlStringCopyWorkerW(pszDest + cchDestLength,cchDest - cchDestLength,pszSrc); | 
 
 
 
 
 | 1119 | return hr; | 
 
 
 
 
 | 1120 | } | 
 
 
 
 
 | 1121 |  | 
 
 
 
 
 | 1122 | NTSTRSAFEDDI RtlStringCatExWorkerA(NTSTRSAFE_PSTR pszDest,size_t cchDest,size_t cbDest,NTSTRSAFE_PCSTR pszSrc,NTSTRSAFE_PSTR *ppszDestEnd,size_t *pcchRemaining,unsigned __LONG32 dwFlags) { | 
 
 
 
 
 | 1123 | NTSTATUS hr = STATUS_SUCCESS; | 
 
 
 
 
 | 1124 | NTSTRSAFE_PSTR pszDestEnd = pszDest; | 
 
 
 
 
 | 1125 | size_t cchRemaining = 0; | 
 
 
 
 
 | 1126 | if(dwFlags & (~STRSAFE_VALID_FLAGS)) hr = STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 1127 | else { | 
 
 
 
 
 | 1128 | size_t cchDestLength; | 
 
 
 
 
 | 1129 | if(dwFlags & STRSAFE_IGNORE_NULLS) { | 
 
 
 
 
 | 1130 | if(!pszDest) { | 
 
 
 
 
 | 1131 | if((cchDest==0) && (cbDest==0)) cchDestLength = 0; | 
 
 
 
 
 | 1132 | else hr = STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 1133 | } else { | 
 
 
 
 
 | 1134 | hr = RtlStringLengthWorkerA(pszDest,cchDest,&cchDestLength); | 
 
 
 
 
 | 1135 | if(NT_SUCCESS(hr)) { | 
 
 
 
 
 | 1136 | pszDestEnd = pszDest + cchDestLength; | 
 
 
 
 
 | 1137 | cchRemaining = cchDest - cchDestLength; | 
 
 
 
 
 | 1138 | } | 
 
 
 
 
 | 1139 | } | 
 
 
 
 
 | 1140 | if(!pszSrc) pszSrc = ""; | 
 
 
 
 
 | 1141 | } else { | 
 
 
 
 
 | 1142 | hr = RtlStringLengthWorkerA(pszDest,cchDest,&cchDestLength); | 
 
 
 
 
 | 1143 | if(NT_SUCCESS(hr)) { | 
 
 
 
 
 | 1144 | pszDestEnd = pszDest + cchDestLength; | 
 
 
 
 
 | 1145 | cchRemaining = cchDest - cchDestLength; | 
 
 
 
 
 | 1146 | } | 
 
 
 
 
 | 1147 | } | 
 
 
 
 
 | 1148 | if(NT_SUCCESS(hr)) { | 
 
 
 
 
 | 1149 | if(cchDest==0) { | 
 
 
 
 
 | 1150 | if(*pszSrc!='\0') { | 
 
 
 
 
 | 1151 | if(!pszDest) hr = STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 1152 | else hr = STATUS_BUFFER_OVERFLOW; | 
 
 
 
 
 | 1153 | } | 
 
 
 
 
 | 1154 | } else hr = RtlStringCopyExWorkerA(pszDestEnd,cchRemaining,(cchRemaining*sizeof(char)) + (cbDest % sizeof(char)),pszSrc,&pszDestEnd,&cchRemaining,dwFlags & (~(STRSAFE_FILL_ON_FAILURE | STRSAFE_NULL_ON_FAILURE))); | 
 
 
 
 
 | 1155 | } | 
 
 
 
 
 | 1156 | } | 
 
 
 
 
 | 1157 | if(!NT_SUCCESS(hr)) { | 
 
 
 
 
 | 1158 | if(pszDest) { | 
 
 
 
 
 | 1159 | if(dwFlags & STRSAFE_FILL_ON_FAILURE) { | 
 
 
 
 
 | 1160 | memset(pszDest,STRSAFE_GET_FILL_PATTERN(dwFlags),cbDest); | 
 
 
 
 
 | 1161 | if(STRSAFE_GET_FILL_PATTERN(dwFlags)==0) { | 
 
 
 
 
 | 1162 | pszDestEnd = pszDest; | 
 
 
 
 
 | 1163 | cchRemaining = cchDest; | 
 
 
 
 
 | 1164 | } else if(cchDest > 0) { | 
 
 
 
 
 | 1165 | pszDestEnd = pszDest + cchDest - 1; | 
 
 
 
 
 | 1166 | cchRemaining = 1; | 
 
 
 
 
 | 1167 | *pszDestEnd = '\0'; | 
 
 
 
 
 | 1168 | } | 
 
 
 
 
 | 1169 | } | 
 
 
 
 
 | 1170 | if(dwFlags & STRSAFE_NULL_ON_FAILURE) { | 
 
 
 
 
 | 1171 | if(cchDest > 0) { | 
 
 
 
 
 | 1172 | pszDestEnd = pszDest; | 
 
 
 
 
 | 1173 | cchRemaining = cchDest; | 
 
 
 
 
 | 1174 | *pszDestEnd = '\0'; | 
 
 
 
 
 | 1175 | } | 
 
 
 
 
 | 1176 | } | 
 
 
 
 
 | 1177 | } | 
 
 
 
 
 | 1178 | } | 
 
 
 
 
 | 1179 | if(NT_SUCCESS(hr) || (hr==STATUS_BUFFER_OVERFLOW)) { | 
 
 
 
 
 | 1180 | if(ppszDestEnd) *ppszDestEnd = pszDestEnd; | 
 
 
 
 
 | 1181 | if(pcchRemaining) *pcchRemaining = cchRemaining; | 
 
 
 
 
 | 1182 | } | 
 
 
 
 
 | 1183 | return hr; | 
 
 
 
 
 | 1184 | } | 
 
 
 
 
 | 1185 |  | 
 
 
 
 
 | 1186 | NTSTRSAFEDDI RtlStringCatExWorkerW(NTSTRSAFE_PWSTR pszDest,size_t cchDest,size_t cbDest,NTSTRSAFE_PCWSTR pszSrc,NTSTRSAFE_PWSTR *ppszDestEnd,size_t *pcchRemaining,unsigned __LONG32 dwFlags) { | 
 
 
 
 
 | 1187 | NTSTATUS hr = STATUS_SUCCESS; | 
 
 
 
 
 | 1188 | NTSTRSAFE_PWSTR pszDestEnd = pszDest; | 
 
 
 
 
 | 1189 | size_t cchRemaining = 0; | 
 
 
 
 
 | 1190 | if(dwFlags & (~STRSAFE_VALID_FLAGS)) hr = STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 1191 | else { | 
 
 
 
 
 | 1192 | size_t cchDestLength; | 
 
 
 
 
 | 1193 | if(dwFlags & STRSAFE_IGNORE_NULLS) { | 
 
 
 
 
 | 1194 | if(!pszDest) { | 
 
 
 
 
 | 1195 | if((cchDest==0) && (cbDest==0)) cchDestLength = 0; | 
 
 
 
 
 | 1196 | else hr = STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 1197 | } else { | 
 
 
 
 
 | 1198 | hr = RtlStringLengthWorkerW(pszDest,cchDest,&cchDestLength); | 
 
 
 
 
 | 1199 | if(NT_SUCCESS(hr)) { | 
 
 
 
 
 | 1200 | pszDestEnd = pszDest + cchDestLength; | 
 
 
 
 
 | 1201 | cchRemaining = cchDest - cchDestLength; | 
 
 
 
 
 | 1202 | } | 
 
 
 
 
 | 1203 | } | 
 
 
 
 
 | 1204 | if(!pszSrc) pszSrc = L""; | 
 
 
 
 
 | 1205 | } else { | 
 
 
 
 
 | 1206 | hr = RtlStringLengthWorkerW(pszDest,cchDest,&cchDestLength); | 
 
 
 
 
 | 1207 | if(NT_SUCCESS(hr)) { | 
 
 
 
 
 | 1208 | pszDestEnd = pszDest + cchDestLength; | 
 
 
 
 
 | 1209 | cchRemaining = cchDest - cchDestLength; | 
 
 
 
 
 | 1210 | } | 
 
 
 
 
 | 1211 | } | 
 
 
 
 
 | 1212 | if(NT_SUCCESS(hr)) { | 
 
 
 
 
 | 1213 | if(cchDest==0) { | 
 
 
 
 
 | 1214 | if(*pszSrc!=L'\0') { | 
 
 
 
 
 | 1215 | if(!pszDest) hr = STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 1216 | else hr = STATUS_BUFFER_OVERFLOW; | 
 
 
 
 
 | 1217 | } | 
 
 
 
 
 | 1218 | } else hr = RtlStringCopyExWorkerW(pszDestEnd,cchRemaining,(cchRemaining*sizeof(wchar_t)) + (cbDest % sizeof(wchar_t)),pszSrc,&pszDestEnd,&cchRemaining,dwFlags & (~(STRSAFE_FILL_ON_FAILURE | STRSAFE_NULL_ON_FAILURE))); | 
 
 
 
 
 | 1219 | } | 
 
 
 
 
 | 1220 | } | 
 
 
 
 
 | 1221 | if(!NT_SUCCESS(hr)) { | 
 
 
 
 
 | 1222 | if(pszDest) { | 
 
 
 
 
 | 1223 | if(dwFlags & STRSAFE_FILL_ON_FAILURE) { | 
 
 
 
 
 | 1224 | memset(pszDest,STRSAFE_GET_FILL_PATTERN(dwFlags),cbDest); | 
 
 
 
 
 | 1225 | if(STRSAFE_GET_FILL_PATTERN(dwFlags)==0) { | 
 
 
 
 
 | 1226 | pszDestEnd = pszDest; | 
 
 
 
 
 | 1227 | cchRemaining = cchDest; | 
 
 
 
 
 | 1228 | } else if(cchDest > 0) { | 
 
 
 
 
 | 1229 | pszDestEnd = pszDest + cchDest - 1; | 
 
 
 
 
 | 1230 | cchRemaining = 1; | 
 
 
 
 
 | 1231 | *pszDestEnd = L'\0'; | 
 
 
 
 
 | 1232 | } | 
 
 
 
 
 | 1233 | } | 
 
 
 
 
 | 1234 | if(dwFlags & STRSAFE_NULL_ON_FAILURE) { | 
 
 
 
 
 | 1235 | if(cchDest > 0) { | 
 
 
 
 
 | 1236 | pszDestEnd = pszDest; | 
 
 
 
 
 | 1237 | cchRemaining = cchDest; | 
 
 
 
 
 | 1238 | *pszDestEnd = L'\0'; | 
 
 
 
 
 | 1239 | } | 
 
 
 
 
 | 1240 | } | 
 
 
 
 
 | 1241 | } | 
 
 
 
 
 | 1242 | } | 
 
 
 
 
 | 1243 | if(NT_SUCCESS(hr) || (hr==STATUS_BUFFER_OVERFLOW)) { | 
 
 
 
 
 | 1244 | if(ppszDestEnd) *ppszDestEnd = pszDestEnd; | 
 
 
 
 
 | 1245 | if(pcchRemaining) *pcchRemaining = cchRemaining; | 
 
 
 
 
 | 1246 | } | 
 
 
 
 
 | 1247 | return hr; | 
 
 
 
 
 | 1248 | } | 
 
 
 
 
 | 1249 |  | 
 
 
 
 
 | 1250 | NTSTRSAFEDDI RtlStringCatNWorkerA(NTSTRSAFE_PSTR pszDest,size_t cchDest,NTSTRSAFE_PCSTR pszSrc,size_t cchToAppend) { | 
 
 
 
 
 | 1251 | NTSTATUS hr; | 
 
 
 
 
 | 1252 | size_t cchDestLength; | 
 
 
 
 
 | 1253 | hr = RtlStringLengthWorkerA(pszDest,cchDest,&cchDestLength); | 
 
 
 
 
 | 1254 | if(NT_SUCCESS(hr)) hr = RtlStringCopyNWorkerA(pszDest + cchDestLength,cchDest - cchDestLength,pszSrc,cchToAppend); | 
 
 
 
 
 | 1255 | return hr; | 
 
 
 
 
 | 1256 | } | 
 
 
 
 
 | 1257 |  | 
 
 
 
 
 | 1258 | NTSTRSAFEDDI RtlStringCatNWorkerW(NTSTRSAFE_PWSTR pszDest,size_t cchDest,NTSTRSAFE_PCWSTR pszSrc,size_t cchToAppend) { | 
 
 
 
 
 | 1259 | NTSTATUS hr; | 
 
 
 
 
 | 1260 | size_t cchDestLength; | 
 
 
 
 
 | 1261 | hr = RtlStringLengthWorkerW(pszDest,cchDest,&cchDestLength); | 
 
 
 
 
 | 1262 | if(NT_SUCCESS(hr)) hr = RtlStringCopyNWorkerW(pszDest + cchDestLength,cchDest - cchDestLength,pszSrc,cchToAppend); | 
 
 
 
 
 | 1263 | return hr; | 
 
 
 
 
 | 1264 | } | 
 
 
 
 
 | 1265 |  | 
 
 
 
 
 | 1266 | NTSTRSAFEDDI RtlStringCatNExWorkerA(NTSTRSAFE_PSTR pszDest,size_t cchDest,size_t cbDest,NTSTRSAFE_PCSTR pszSrc,size_t cchToAppend,NTSTRSAFE_PSTR *ppszDestEnd,size_t *pcchRemaining,unsigned __LONG32 dwFlags) { | 
 
 
 
 
 | 1267 | NTSTATUS hr = STATUS_SUCCESS; | 
 
 
 
 
 | 1268 | NTSTRSAFE_PSTR pszDestEnd = pszDest; | 
 
 
 
 
 | 1269 | size_t cchRemaining = 0; | 
 
 
 
 
 | 1270 | size_t cchDestLength = 0; | 
 
 
 
 
 | 1271 | if(dwFlags & (~STRSAFE_VALID_FLAGS)) hr = STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 1272 | else if(cchToAppend > NTSTRSAFE_MAX_CCH) hr = STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 1273 | else { | 
 
 
 
 
 | 1274 | if(dwFlags & STRSAFE_IGNORE_NULLS) { | 
 
 
 
 
 | 1275 | if(!pszDest) { | 
 
 
 
 
 | 1276 | if((cchDest==0) && (cbDest==0)) cchDestLength = 0; | 
 
 
 
 
 | 1277 | else hr = STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 1278 | } else { | 
 
 
 
 
 | 1279 | hr = RtlStringLengthWorkerA(pszDest,cchDest,&cchDestLength); | 
 
 
 
 
 | 1280 | if(NT_SUCCESS(hr)) { | 
 
 
 
 
 | 1281 | pszDestEnd = pszDest + cchDestLength; | 
 
 
 
 
 | 1282 | cchRemaining = cchDest - cchDestLength; | 
 
 
 
 
 | 1283 | } | 
 
 
 
 
 | 1284 | } | 
 
 
 
 
 | 1285 | if(!pszSrc) pszSrc = ""; | 
 
 
 
 
 | 1286 | } else { | 
 
 
 
 
 | 1287 | hr = RtlStringLengthWorkerA(pszDest,cchDest,&cchDestLength); | 
 
 
 
 
 | 1288 | if(NT_SUCCESS(hr)) { | 
 
 
 
 
 | 1289 | pszDestEnd = pszDest + cchDestLength; | 
 
 
 
 
 | 1290 | cchRemaining = cchDest - cchDestLength; | 
 
 
 
 
 | 1291 | } | 
 
 
 
 
 | 1292 | } | 
 
 
 
 
 | 1293 | if(NT_SUCCESS(hr)) { | 
 
 
 
 
 | 1294 | if(cchDest==0) { | 
 
 
 
 
 | 1295 | if((cchToAppend!=0) && (*pszSrc!='\0')) { | 
 
 
 
 
 | 1296 | if(!pszDest) hr = STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 1297 | else hr = STATUS_BUFFER_OVERFLOW; | 
 
 
 
 
 | 1298 | } | 
 
 
 
 
 | 1299 | } else hr = RtlStringCopyNExWorkerA(pszDestEnd,cchRemaining,(cchRemaining*sizeof(char)) + (cbDest % sizeof(char)),pszSrc,cchToAppend,&pszDestEnd,&cchRemaining,dwFlags & (~(STRSAFE_FILL_ON_FAILURE | STRSAFE_NULL_ON_FAILURE))); | 
 
 
 
 
 | 1300 | } | 
 
 
 
 
 | 1301 | } | 
 
 
 
 
 | 1302 | if(!NT_SUCCESS(hr)) { | 
 
 
 
 
 | 1303 | if(pszDest) { | 
 
 
 
 
 | 1304 | if(dwFlags & STRSAFE_FILL_ON_FAILURE) { | 
 
 
 
 
 | 1305 | memset(pszDest,STRSAFE_GET_FILL_PATTERN(dwFlags),cbDest); | 
 
 
 
 
 | 1306 | if(STRSAFE_GET_FILL_PATTERN(dwFlags)==0) { | 
 
 
 
 
 | 1307 | pszDestEnd = pszDest; | 
 
 
 
 
 | 1308 | cchRemaining = cchDest; | 
 
 
 
 
 | 1309 | } else if(cchDest > 0) { | 
 
 
 
 
 | 1310 | pszDestEnd = pszDest + cchDest - 1; | 
 
 
 
 
 | 1311 | cchRemaining = 1; | 
 
 
 
 
 | 1312 | *pszDestEnd = '\0'; | 
 
 
 
 
 | 1313 | } | 
 
 
 
 
 | 1314 | } | 
 
 
 
 
 | 1315 | if(dwFlags & (STRSAFE_NULL_ON_FAILURE)) { | 
 
 
 
 
 | 1316 | if(cchDest > 0) { | 
 
 
 
 
 | 1317 | pszDestEnd = pszDest; | 
 
 
 
 
 | 1318 | cchRemaining = cchDest; | 
 
 
 
 
 | 1319 | *pszDestEnd = '\0'; | 
 
 
 
 
 | 1320 | } | 
 
 
 
 
 | 1321 | } | 
 
 
 
 
 | 1322 | } | 
 
 
 
 
 | 1323 | } | 
 
 
 
 
 | 1324 | if(NT_SUCCESS(hr) || (hr==STATUS_BUFFER_OVERFLOW)) { | 
 
 
 
 
 | 1325 | if(ppszDestEnd) *ppszDestEnd = pszDestEnd; | 
 
 
 
 
 | 1326 | if(pcchRemaining) *pcchRemaining = cchRemaining; | 
 
 
 
 
 | 1327 | } | 
 
 
 
 
 | 1328 | return hr; | 
 
 
 
 
 | 1329 | } | 
 
 
 
 
 | 1330 |  | 
 
 
 
 
 | 1331 | NTSTRSAFEDDI RtlStringCatNExWorkerW(NTSTRSAFE_PWSTR pszDest,size_t cchDest,size_t cbDest,NTSTRSAFE_PCWSTR pszSrc,size_t cchToAppend,NTSTRSAFE_PWSTR *ppszDestEnd,size_t *pcchRemaining,unsigned __LONG32 dwFlags) { | 
 
 
 
 
 | 1332 | NTSTATUS hr = STATUS_SUCCESS; | 
 
 
 
 
 | 1333 | NTSTRSAFE_PWSTR pszDestEnd = pszDest; | 
 
 
 
 
 | 1334 | size_t cchRemaining = 0; | 
 
 
 
 
 | 1335 | size_t cchDestLength = 0; | 
 
 
 
 
 | 1336 | if(dwFlags & (~STRSAFE_VALID_FLAGS)) hr = STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 1337 | else if(cchToAppend > NTSTRSAFE_MAX_CCH) hr = STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 1338 | else { | 
 
 
 
 
 | 1339 | if(dwFlags & STRSAFE_IGNORE_NULLS) { | 
 
 
 
 
 | 1340 | if(!pszDest) { | 
 
 
 
 
 | 1341 | if((cchDest==0) && (cbDest==0)) cchDestLength = 0; | 
 
 
 
 
 | 1342 | else hr = STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 1343 | } else { | 
 
 
 
 
 | 1344 | hr = RtlStringLengthWorkerW(pszDest,cchDest,&cchDestLength); | 
 
 
 
 
 | 1345 | if(NT_SUCCESS(hr)) { | 
 
 
 
 
 | 1346 | pszDestEnd = pszDest + cchDestLength; | 
 
 
 
 
 | 1347 | cchRemaining = cchDest - cchDestLength; | 
 
 
 
 
 | 1348 | } | 
 
 
 
 
 | 1349 | } | 
 
 
 
 
 | 1350 | if(!pszSrc) pszSrc = L""; | 
 
 
 
 
 | 1351 | } else { | 
 
 
 
 
 | 1352 | hr = RtlStringLengthWorkerW(pszDest,cchDest,&cchDestLength); | 
 
 
 
 
 | 1353 | if(NT_SUCCESS(hr)) { | 
 
 
 
 
 | 1354 | pszDestEnd = pszDest + cchDestLength; | 
 
 
 
 
 | 1355 | cchRemaining = cchDest - cchDestLength; | 
 
 
 
 
 | 1356 | } | 
 
 
 
 
 | 1357 | } | 
 
 
 
 
 | 1358 | if(NT_SUCCESS(hr)) { | 
 
 
 
 
 | 1359 | if(cchDest==0) { | 
 
 
 
 
 | 1360 | if((cchToAppend!=0) && (*pszSrc!=L'\0')) { | 
 
 
 
 
 | 1361 | if(!pszDest) hr = STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 1362 | else hr = STATUS_BUFFER_OVERFLOW; | 
 
 
 
 
 | 1363 | } | 
 
 
 
 
 | 1364 | } else hr = RtlStringCopyNExWorkerW(pszDestEnd,cchRemaining,(cchRemaining*sizeof(wchar_t)) + (cbDest % sizeof(wchar_t)),pszSrc,cchToAppend,&pszDestEnd,&cchRemaining,dwFlags & (~(STRSAFE_FILL_ON_FAILURE | STRSAFE_NULL_ON_FAILURE))); | 
 
 
 
 
 | 1365 | } | 
 
 
 
 
 | 1366 | } | 
 
 
 
 
 | 1367 | if(!NT_SUCCESS(hr)) { | 
 
 
 
 
 | 1368 | if(pszDest) { | 
 
 
 
 
 | 1369 | if(dwFlags & STRSAFE_FILL_ON_FAILURE) { | 
 
 
 
 
 | 1370 | memset(pszDest,STRSAFE_GET_FILL_PATTERN(dwFlags),cbDest); | 
 
 
 
 
 | 1371 | if(STRSAFE_GET_FILL_PATTERN(dwFlags)==0) { | 
 
 
 
 
 | 1372 | pszDestEnd = pszDest; | 
 
 
 
 
 | 1373 | cchRemaining = cchDest; | 
 
 
 
 
 | 1374 | } else if(cchDest > 0) { | 
 
 
 
 
 | 1375 | pszDestEnd = pszDest + cchDest - 1; | 
 
 
 
 
 | 1376 | cchRemaining = 1; | 
 
 
 
 
 | 1377 | *pszDestEnd = L'\0'; | 
 
 
 
 
 | 1378 | } | 
 
 
 
 
 | 1379 | } | 
 
 
 
 
 | 1380 | if(dwFlags & (STRSAFE_NULL_ON_FAILURE)) { | 
 
 
 
 
 | 1381 | if(cchDest > 0) { | 
 
 
 
 
 | 1382 | pszDestEnd = pszDest; | 
 
 
 
 
 | 1383 | cchRemaining = cchDest; | 
 
 
 
 
 | 1384 | *pszDestEnd = L'\0'; | 
 
 
 
 
 | 1385 | } | 
 
 
 
 
 | 1386 | } | 
 
 
 
 
 | 1387 | } | 
 
 
 
 
 | 1388 | } | 
 
 
 
 
 | 1389 | if(NT_SUCCESS(hr) || (hr==STATUS_BUFFER_OVERFLOW)) { | 
 
 
 
 
 | 1390 | if(ppszDestEnd) *ppszDestEnd = pszDestEnd; | 
 
 
 
 
 | 1391 | if(pcchRemaining) *pcchRemaining = cchRemaining; | 
 
 
 
 
 | 1392 | } | 
 
 
 
 
 | 1393 | return hr; | 
 
 
 
 
 | 1394 | } | 
 
 
 
 
 | 1395 |  | 
 
 
 
 
 | 1396 | NTSTRSAFEDDI RtlStringVPrintfWorkerA(NTSTRSAFE_PSTR pszDest,size_t cchDest,NTSTRSAFE_PCSTR pszFormat,va_list argList) { | 
 
 
 
 
 | 1397 | NTSTATUS hr = STATUS_SUCCESS; | 
 
 
 
 
 | 1398 | if(cchDest==0) hr = STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 1399 | else { | 
 
 
 
 
 | 1400 | int iRet; | 
 
 
 
 
 | 1401 | size_t cchMax; | 
 
 
 
 
 | 1402 | cchMax = cchDest - 1; | 
 
 
 
 
 | 1403 | iRet = _vsnprintf(pszDest,cchMax,pszFormat,argList); | 
 
 
 
 
 | 1404 | if((iRet < 0) || (((size_t)iRet) > cchMax)) { | 
 
 
 
 
 | 1405 | pszDest += cchMax; | 
 
 
 
 
 | 1406 | *pszDest = '\0'; | 
 
 
 
 
 | 1407 | hr = STATUS_BUFFER_OVERFLOW; | 
 
 
 
 
 | 1408 | } else if(((size_t)iRet)==cchMax) { | 
 
 
 
 
 | 1409 | pszDest += cchMax; | 
 
 
 
 
 | 1410 | *pszDest = '\0'; | 
 
 
 
 
 | 1411 | } | 
 
 
 
 
 | 1412 | } | 
 
 
 
 
 | 1413 | return hr; | 
 
 
 
 
 | 1414 | } | 
 
 
 
 
 | 1415 |  | 
 
 
 
 
 | 1416 | NTSTRSAFEDDI RtlStringVPrintfWorkerW(NTSTRSAFE_PWSTR pszDest,size_t cchDest,NTSTRSAFE_PCWSTR pszFormat,va_list argList) { | 
 
 
 
 
 | 1417 | NTSTATUS hr = STATUS_SUCCESS; | 
 
 
 
 
 | 1418 | if(cchDest==0) hr = STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 1419 | else { | 
 
 
 
 
 | 1420 | int iRet; | 
 
 
 
 
 | 1421 | size_t cchMax; | 
 
 
 
 
 | 1422 | cchMax = cchDest - 1; | 
 
 
 
 
 | 1423 | iRet = _vsnwprintf(pszDest,cchMax,pszFormat,argList); | 
 
 
 
 
 | 1424 | if((iRet < 0) || (((size_t)iRet) > cchMax)) { | 
 
 
 
 
 | 1425 | pszDest += cchMax; | 
 
 
 
 
 | 1426 | *pszDest = L'\0'; | 
 
 
 
 
 | 1427 | hr = STATUS_BUFFER_OVERFLOW; | 
 
 
 
 
 | 1428 | } else if(((size_t)iRet)==cchMax) { | 
 
 
 
 
 | 1429 | pszDest += cchMax; | 
 
 
 
 
 | 1430 | *pszDest = L'\0'; | 
 
 
 
 
 | 1431 | } | 
 
 
 
 
 | 1432 | } | 
 
 
 
 
 | 1433 | return hr; | 
 
 
 
 
 | 1434 | } | 
 
 
 
 
 | 1435 |  | 
 
 
 
 
 | 1436 | NTSTRSAFEDDI RtlStringVPrintfExWorkerA(NTSTRSAFE_PSTR pszDest,size_t cchDest,size_t cbDest,NTSTRSAFE_PSTR *ppszDestEnd,size_t *pcchRemaining,unsigned __LONG32 dwFlags,NTSTRSAFE_PCSTR pszFormat,va_list argList) { | 
 
 
 
 
 | 1437 | NTSTATUS hr = STATUS_SUCCESS; | 
 
 
 
 
 | 1438 | NTSTRSAFE_PSTR pszDestEnd = pszDest; | 
 
 
 
 
 | 1439 | size_t cchRemaining = 0; | 
 
 
 
 
 | 1440 | if(dwFlags & (~STRSAFE_VALID_FLAGS)) hr = STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 1441 | else { | 
 
 
 
 
 | 1442 | if(dwFlags & STRSAFE_IGNORE_NULLS) { | 
 
 
 
 
 | 1443 | if(!pszDest) { | 
 
 
 
 
 | 1444 | if((cchDest!=0) || (cbDest!=0)) hr = STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 1445 | } | 
 
 
 
 
 | 1446 | if(!pszFormat) pszFormat = ""; | 
 
 
 
 
 | 1447 | } | 
 
 
 
 
 | 1448 | if(NT_SUCCESS(hr)) { | 
 
 
 
 
 | 1449 | if(cchDest==0) { | 
 
 
 
 
 | 1450 | pszDestEnd = pszDest; | 
 
 
 
 
 | 1451 | cchRemaining = 0; | 
 
 
 
 
 | 1452 | if(*pszFormat!='\0') { | 
 
 
 
 
 | 1453 | if(!pszDest) hr = STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 1454 | else hr = STATUS_BUFFER_OVERFLOW; | 
 
 
 
 
 | 1455 | } | 
 
 
 
 
 | 1456 | } else { | 
 
 
 
 
 | 1457 | int iRet; | 
 
 
 
 
 | 1458 | size_t cchMax; | 
 
 
 
 
 | 1459 | cchMax = cchDest - 1; | 
 
 
 
 
 | 1460 | iRet = _vsnprintf(pszDest,cchMax,pszFormat,argList); | 
 
 
 
 
 | 1461 | if((iRet < 0) || (((size_t)iRet) > cchMax)) { | 
 
 
 
 
 | 1462 | pszDestEnd = pszDest + cchMax; | 
 
 
 
 
 | 1463 | cchRemaining = 1; | 
 
 
 
 
 | 1464 | *pszDestEnd = '\0'; | 
 
 
 
 
 | 1465 | hr = STATUS_BUFFER_OVERFLOW; | 
 
 
 
 
 | 1466 | } else if(((size_t)iRet)==cchMax) { | 
 
 
 
 
 | 1467 | pszDestEnd = pszDest + cchMax; | 
 
 
 
 
 | 1468 | cchRemaining = 1; | 
 
 
 
 
 | 1469 | *pszDestEnd = '\0'; | 
 
 
 
 
 | 1470 | } else if(((size_t)iRet) < cchMax) { | 
 
 
 
 
 | 1471 | pszDestEnd = pszDest + iRet; | 
 
 
 
 
 | 1472 | cchRemaining = cchDest - iRet; | 
 
 
 
 
 | 1473 | if(dwFlags & STRSAFE_FILL_BEHIND_NULL) { | 
 
 
 
 
 | 1474 | memset(pszDestEnd + 1,STRSAFE_GET_FILL_PATTERN(dwFlags),((cchRemaining - 1)*sizeof(char)) + (cbDest % sizeof(char))); | 
 
 
 
 
 | 1475 | } | 
 
 
 
 
 | 1476 | } | 
 
 
 
 
 | 1477 | } | 
 
 
 
 
 | 1478 | } | 
 
 
 
 
 | 1479 | } | 
 
 
 
 
 | 1480 | if(!NT_SUCCESS(hr)) { | 
 
 
 
 
 | 1481 | if(pszDest) { | 
 
 
 
 
 | 1482 | if(dwFlags & STRSAFE_FILL_ON_FAILURE) { | 
 
 
 
 
 | 1483 | memset(pszDest,STRSAFE_GET_FILL_PATTERN(dwFlags),cbDest); | 
 
 
 
 
 | 1484 | if(STRSAFE_GET_FILL_PATTERN(dwFlags)==0) { | 
 
 
 
 
 | 1485 | pszDestEnd = pszDest; | 
 
 
 
 
 | 1486 | cchRemaining = cchDest; | 
 
 
 
 
 | 1487 | } else if(cchDest > 0) { | 
 
 
 
 
 | 1488 | pszDestEnd = pszDest + cchDest - 1; | 
 
 
 
 
 | 1489 | cchRemaining = 1; | 
 
 
 
 
 | 1490 | *pszDestEnd = '\0'; | 
 
 
 
 
 | 1491 | } | 
 
 
 
 
 | 1492 | } | 
 
 
 
 
 | 1493 | if(dwFlags & (STRSAFE_NULL_ON_FAILURE | STRSAFE_NO_TRUNCATION)) { | 
 
 
 
 
 | 1494 | if(cchDest > 0) { | 
 
 
 
 
 | 1495 | pszDestEnd = pszDest; | 
 
 
 
 
 | 1496 | cchRemaining = cchDest; | 
 
 
 
 
 | 1497 | *pszDestEnd = '\0'; | 
 
 
 
 
 | 1498 | } | 
 
 
 
 
 | 1499 | } | 
 
 
 
 
 | 1500 | } | 
 
 
 
 
 | 1501 | } | 
 
 
 
 
 | 1502 | if(NT_SUCCESS(hr) || (hr==STATUS_BUFFER_OVERFLOW)) { | 
 
 
 
 
 | 1503 | if(ppszDestEnd) *ppszDestEnd = pszDestEnd; | 
 
 
 
 
 | 1504 | if(pcchRemaining) *pcchRemaining = cchRemaining; | 
 
 
 
 
 | 1505 | } | 
 
 
 
 
 | 1506 | return hr; | 
 
 
 
 
 | 1507 | } | 
 
 
 
 
 | 1508 |  | 
 
 
 
 
 | 1509 | NTSTRSAFEDDI RtlStringVPrintfExWorkerW(NTSTRSAFE_PWSTR pszDest,size_t cchDest,size_t cbDest,NTSTRSAFE_PWSTR *ppszDestEnd,size_t *pcchRemaining,unsigned __LONG32 dwFlags,NTSTRSAFE_PCWSTR pszFormat,va_list argList) { | 
 
 
 
 
 | 1510 | NTSTATUS hr = STATUS_SUCCESS; | 
 
 
 
 
 | 1511 | NTSTRSAFE_PWSTR pszDestEnd = pszDest; | 
 
 
 
 
 | 1512 | size_t cchRemaining = 0; | 
 
 
 
 
 | 1513 | if(dwFlags & (~STRSAFE_VALID_FLAGS)) hr = STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 1514 | else { | 
 
 
 
 
 | 1515 | if(dwFlags & STRSAFE_IGNORE_NULLS) { | 
 
 
 
 
 | 1516 | if(!pszDest) { | 
 
 
 
 
 | 1517 | if((cchDest!=0) || (cbDest!=0)) hr = STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 1518 | } | 
 
 
 
 
 | 1519 | if(!pszFormat) pszFormat = L""; | 
 
 
 
 
 | 1520 | } | 
 
 
 
 
 | 1521 | if(NT_SUCCESS(hr)) { | 
 
 
 
 
 | 1522 | if(cchDest==0) { | 
 
 
 
 
 | 1523 | pszDestEnd = pszDest; | 
 
 
 
 
 | 1524 | cchRemaining = 0; | 
 
 
 
 
 | 1525 | if(*pszFormat!=L'\0') { | 
 
 
 
 
 | 1526 | if(!pszDest) hr = STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 1527 | else hr = STATUS_BUFFER_OVERFLOW; | 
 
 
 
 
 | 1528 | } | 
 
 
 
 
 | 1529 | } else { | 
 
 
 
 
 | 1530 | int iRet; | 
 
 
 
 
 | 1531 | size_t cchMax; | 
 
 
 
 
 | 1532 | cchMax = cchDest - 1; | 
 
 
 
 
 | 1533 | iRet = _vsnwprintf(pszDest,cchMax,pszFormat,argList); | 
 
 
 
 
 | 1534 | if((iRet < 0) || (((size_t)iRet) > cchMax)) { | 
 
 
 
 
 | 1535 | pszDestEnd = pszDest + cchMax; | 
 
 
 
 
 | 1536 | cchRemaining = 1; | 
 
 
 
 
 | 1537 | *pszDestEnd = L'\0'; | 
 
 
 
 
 | 1538 | hr = STATUS_BUFFER_OVERFLOW; | 
 
 
 
 
 | 1539 | } else if(((size_t)iRet)==cchMax) { | 
 
 
 
 
 | 1540 | pszDestEnd = pszDest + cchMax; | 
 
 
 
 
 | 1541 | cchRemaining = 1; | 
 
 
 
 
 | 1542 | *pszDestEnd = L'\0'; | 
 
 
 
 
 | 1543 | } else if(((size_t)iRet) < cchMax) { | 
 
 
 
 
 | 1544 | pszDestEnd = pszDest + iRet; | 
 
 
 
 
 | 1545 | cchRemaining = cchDest - iRet; | 
 
 
 
 
 | 1546 | if(dwFlags & STRSAFE_FILL_BEHIND_NULL) { | 
 
 
 
 
 | 1547 | memset(pszDestEnd + 1,STRSAFE_GET_FILL_PATTERN(dwFlags),((cchRemaining - 1)*sizeof(wchar_t)) + (cbDest % sizeof(wchar_t))); | 
 
 
 
 
 | 1548 | } | 
 
 
 
 
 | 1549 | } | 
 
 
 
 
 | 1550 | } | 
 
 
 
 
 | 1551 | } | 
 
 
 
 
 | 1552 | } | 
 
 
 
 
 | 1553 | if(!NT_SUCCESS(hr)) { | 
 
 
 
 
 | 1554 | if(pszDest) { | 
 
 
 
 
 | 1555 | if(dwFlags & STRSAFE_FILL_ON_FAILURE) { | 
 
 
 
 
 | 1556 | memset(pszDest,STRSAFE_GET_FILL_PATTERN(dwFlags),cbDest); | 
 
 
 
 
 | 1557 | if(STRSAFE_GET_FILL_PATTERN(dwFlags)==0) { | 
 
 
 
 
 | 1558 | pszDestEnd = pszDest; | 
 
 
 
 
 | 1559 | cchRemaining = cchDest; | 
 
 
 
 
 | 1560 | } else if(cchDest > 0) { | 
 
 
 
 
 | 1561 | pszDestEnd = pszDest + cchDest - 1; | 
 
 
 
 
 | 1562 | cchRemaining = 1; | 
 
 
 
 
 | 1563 | *pszDestEnd = L'\0'; | 
 
 
 
 
 | 1564 | } | 
 
 
 
 
 | 1565 | } | 
 
 
 
 
 | 1566 | if(dwFlags & (STRSAFE_NULL_ON_FAILURE | STRSAFE_NO_TRUNCATION)) { | 
 
 
 
 
 | 1567 | if(cchDest > 0) { | 
 
 
 
 
 | 1568 | pszDestEnd = pszDest; | 
 
 
 
 
 | 1569 | cchRemaining = cchDest; | 
 
 
 
 
 | 1570 | *pszDestEnd = L'\0'; | 
 
 
 
 
 | 1571 | } | 
 
 
 
 
 | 1572 | } | 
 
 
 
 
 | 1573 | } | 
 
 
 
 
 | 1574 | } | 
 
 
 
 
 | 1575 | if(NT_SUCCESS(hr) || (hr==STATUS_BUFFER_OVERFLOW)) { | 
 
 
 
 
 | 1576 | if(ppszDestEnd) *ppszDestEnd = pszDestEnd; | 
 
 
 
 
 | 1577 | if(pcchRemaining) *pcchRemaining = cchRemaining; | 
 
 
 
 
 | 1578 | } | 
 
 
 
 
 | 1579 | return hr; | 
 
 
 
 
 | 1580 | } | 
 
 
 
 
 | 1581 |  | 
 
 
 
 
 | 1582 | NTSTRSAFEDDI RtlStringLengthWorkerA(NTSTRSAFE_PCSTR psz,size_t cchMax,size_t *pcchLength) { | 
 
 
 
 
 | 1583 | NTSTATUS hr = STATUS_SUCCESS; | 
 
 
 
 
 | 1584 | size_t cchMaxPrev = cchMax; | 
 
 
 
 
 | 1585 | while(cchMax && (*psz!='\0')) { | 
 
 
 
 
 | 1586 | psz++; | 
 
 
 
 
 | 1587 | cchMax--; | 
 
 
 
 
 | 1588 | } | 
 
 
 
 
 | 1589 | if(cchMax==0) hr = STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 1590 | if(pcchLength) { | 
 
 
 
 
 | 1591 | if(NT_SUCCESS(hr)) *pcchLength = cchMaxPrev - cchMax; | 
 
 
 
 
 | 1592 | else *pcchLength = 0; | 
 
 
 
 
 | 1593 | } | 
 
 
 
 
 | 1594 | return hr; | 
 
 
 
 
 | 1595 | } | 
 
 
 
 
 | 1596 |  | 
 
 
 
 
 | 1597 | NTSTRSAFEDDI RtlStringLengthWorkerW(NTSTRSAFE_PCWSTR psz,size_t cchMax,size_t *pcchLength) { | 
 
 
 
 
 | 1598 | NTSTATUS hr = STATUS_SUCCESS; | 
 
 
 
 
 | 1599 | size_t cchMaxPrev = cchMax; | 
 
 
 
 
 | 1600 | while(cchMax && (*psz!=L'\0')) { | 
 
 
 
 
 | 1601 | psz++; | 
 
 
 
 
 | 1602 | cchMax--; | 
 
 
 
 
 | 1603 | } | 
 
 
 
 
 | 1604 | if(cchMax==0) hr = STATUS_INVALID_PARAMETER; | 
 
 
 
 
 | 1605 | if(pcchLength) { | 
 
 
 
 
 | 1606 | if(NT_SUCCESS(hr)) *pcchLength = cchMaxPrev - cchMax; | 
 
 
 
 
 | 1607 | else *pcchLength = 0; | 
 
 
 
 
 | 1608 | } | 
 
 
 
 
 | 1609 | return hr; | 
 
 
 
 
 | 1610 | } | 
 
 
 
 
 | 1611 |  | 
 
 
 
 
 | 1612 | #endif /* !__STRSAFE__NO_INLINE */ | 
 
 
 
 
 | 1613 |  | 
 
 
 
 
 | 1614 | #define RtlStringCopyWorkerA RtlStringCopyWorkerA_instead_use_RtlStringCchCopyA_or_RtlStringCchCopyExA; | 
 
 
 
 
 | 1615 | #define RtlStringCopyWorkerW RtlStringCopyWorkerW_instead_use_RtlStringCchCopyW_or_RtlStringCchCopyExW; | 
 
 
 
 
 | 1616 | #define RtlStringCopyExWorkerA RtlStringCopyExWorkerA_instead_use_RtlStringCchCopyA_or_RtlStringCchCopyExA; | 
 
 
 
 
 | 1617 | #define RtlStringCopyExWorkerW RtlStringCopyExWorkerW_instead_use_RtlStringCchCopyW_or_RtlStringCchCopyExW; | 
 
 
 
 
 | 1618 | #define RtlStringCatWorkerA RtlStringCatWorkerA_instead_use_RtlStringCchCatA_or_RtlStringCchCatExA; | 
 
 
 
 
 | 1619 | #define RtlStringCatWorkerW RtlStringCatWorkerW_instead_use_RtlStringCchCatW_or_RtlStringCchCatExW; | 
 
 
 
 
 | 1620 | #define RtlStringCatExWorkerA RtlStringCatExWorkerA_instead_use_RtlStringCchCatA_or_RtlStringCchCatExA; | 
 
 
 
 
 | 1621 | #define RtlStringCatExWorkerW RtlStringCatExWorkerW_instead_use_RtlStringCchCatW_or_RtlStringCchCatExW; | 
 
 
 
 
 | 1622 | #define RtlStringCatNWorkerA RtlStringCatNWorkerA_instead_use_RtlStringCchCatNA_or_StrincCbCatNA; | 
 
 
 
 
 | 1623 | #define RtlStringCatNWorkerW RtlStringCatNWorkerW_instead_use_RtlStringCchCatNW_or_RtlStringCbCatNW; | 
 
 
 
 
 | 1624 | #define RtlStringCatNExWorkerA RtlStringCatNExWorkerA_instead_use_RtlStringCchCatNExA_or_RtlStringCbCatNExA; | 
 
 
 
 
 | 1625 | #define RtlStringCatNExWorkerW RtlStringCatNExWorkerW_instead_use_RtlStringCchCatNExW_or_RtlStringCbCatNExW; | 
 
 
 
 
 | 1626 | #define RtlStringVPrintfWorkerA RtlStringVPrintfWorkerA_instead_use_RtlStringCchVPrintfA_or_RtlStringCchVPrintfExA; | 
 
 
 
 
 | 1627 | #define RtlStringVPrintfWorkerW RtlStringVPrintfWorkerW_instead_use_RtlStringCchVPrintfW_or_RtlStringCchVPrintfExW; | 
 
 
 
 
 | 1628 | #define RtlStringVPrintfExWorkerA RtlStringVPrintfExWorkerA_instead_use_RtlStringCchVPrintfA_or_RtlStringCchVPrintfExA; | 
 
 
 
 
 | 1629 | #define RtlStringVPrintfExWorkerW RtlStringVPrintfExWorkerW_instead_use_RtlStringCchVPrintfW_or_RtlStringCchVPrintfExW; | 
 
 
 
 
 | 1630 | #define RtlStringLengthWorkerA RtlStringLengthWorkerA_instead_use_RtlStringCchLengthA_or_RtlStringCbLengthA; | 
 
 
 
 
 | 1631 | #define RtlStringLengthWorkerW RtlStringLengthWorkerW_instead_use_RtlStringCchLengthW_or_RtlStringCbLengthW; | 
 
 
 
 
 | 1632 |  | 
 
 
 
 
 | 1633 | #endif |