| [1166] | 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 | 
|---|