https://git.reactos.org/?p=reactos.git;a=commitdiff;h=eaa3692f7d1743f588e38c9d9b35978ac0cb32e5
commit eaa3692f7d1743f588e38c9d9b35978ac0cb32e5 Author: Katayama Hirofumi MZ <[email protected]> AuthorDate: Wed Jan 29 22:16:58 2020 +0900 Commit: GitHub <[email protected]> CommitDate: Wed Jan 29 22:16:58 2020 +0900 [SDK][ATL] Implement CStringT::FormatMessage (#2286) CStringT::FormatMessage in ATL is a method to build a formatted string. CORE-16666 --- sdk/lib/atl/cstringt.h | 50 ++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 50 insertions(+) diff --git a/sdk/lib/atl/cstringt.h b/sdk/lib/atl/cstringt.h index 87ac52aac6f..ad89dc6270a 100644 --- a/sdk/lib/atl/cstringt.h +++ b/sdk/lib/atl/cstringt.h @@ -150,6 +150,16 @@ public: return ::vswprintf(pszDest, pszFormat, args); } + static LPWSTR + FormatMessageV(_In_z_ LPCWSTR pszFormat, _In_opt_ va_list *pArgList) + { + LPWSTR psz; + ::FormatMessageW( + FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_STRING, pszFormat, 0, 0, + reinterpret_cast<LPWSTR>(&psz), 0, pArgList); + return psz; + } + static BSTR __cdecl AllocSysString( _In_z_ LPCWSTR pszSource, _In_ int nLength) @@ -289,6 +299,16 @@ public: return ::vsprintf(pszDest, pszFormat, args); } + static LPSTR + FormatMessageV(_In_z_ LPCSTR pszFormat, _In_opt_ va_list *pArgList) + { + LPSTR psz; + ::FormatMessageA( + FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_STRING, pszFormat, 0, 0, reinterpret_cast<LPSTR>(&psz), + 0, pArgList); + return psz; + } + static BSTR __cdecl AllocSysString( _In_z_ LPCSTR pszSource, _In_ int nLength) @@ -692,6 +712,36 @@ public: CThisSimpleString::ReleaseBufferSetLength(nLength); } + void __cdecl FormatMessage(UINT nFormatID, ...) + { + va_list va; + va_start(va, nFormatID); + + CStringT str; + if (str.LoadString(nFormatID)) + FormatMessageV(str, &va); + + va_end(va); + } + + void __cdecl FormatMessage(PCXSTR pszFormat, ...) + { + va_list va; + va_start(va, pszFormat); + FormatMessageV(pszFormat, &va); + va_end(va); + } + + void + FormatMessageV(PCXSTR pszFormat, va_list *pArgList) + { + PXSTR psz = StringTraits::FormatMessageV(pszFormat, pArgList); + if (!psz) + CThisSimpleString::ThrowMemoryException(); + + *this = psz; + ::LocalFree(psz); + } int Replace(PCXSTR pszOld, PCXSTR pszNew) {
