ntstrsafe.h 64 KB

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