diff options
author | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-27 14:19:18 +0000 |
---|---|---|
committer | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-27 14:19:18 +0000 |
commit | 4035b1bfb1e5843a539a8b624d21952b756974d1 (patch) | |
tree | f1e9cd5bf548cbc57ff2fddfb2b4aa9ae95587e2 /src/VBox/Runtime/common/string/strprintf2.cpp | |
parent | Initial commit. (diff) | |
download | virtualbox-upstream.tar.xz virtualbox-upstream.zip |
Adding upstream version 6.1.22-dfsg.upstream/6.1.22-dfsgupstream
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'src/VBox/Runtime/common/string/strprintf2.cpp')
-rw-r--r-- | src/VBox/Runtime/common/string/strprintf2.cpp | 163 |
1 files changed, 163 insertions, 0 deletions
diff --git a/src/VBox/Runtime/common/string/strprintf2.cpp b/src/VBox/Runtime/common/string/strprintf2.cpp new file mode 100644 index 00000000..c39d6d84 --- /dev/null +++ b/src/VBox/Runtime/common/string/strprintf2.cpp @@ -0,0 +1,163 @@ +/* $Id: strprintf2.cpp $ */ +/** @file + * IPRT - String Formatters, alternative. + */ + +/* + * Copyright (C) 2006-2020 Oracle Corporation + * + * This file is part of VirtualBox Open Source Edition (OSE), as + * available from http://www.virtualbox.org. This file is free software; + * you can redistribute it and/or modify it under the terms of the GNU + * General Public License (GPL) as published by the Free Software + * Foundation, in version 2 as it comes in the "COPYING" file of the + * VirtualBox OSE distribution. VirtualBox OSE is distributed in the + * hope that it will be useful, but WITHOUT ANY WARRANTY of any kind. + * + * The contents of this file may alternatively be used under the terms + * of the Common Development and Distribution License Version 1.0 + * (CDDL) only, as it comes in the "COPYING.CDDL" file of the + * VirtualBox OSE distribution, in which case the provisions of the + * CDDL are applicable instead of those of the GPL. + * + * You may elect to license modified versions of this file under the + * terms and conditions of either the GPL or the CDDL or both. + */ + + +/********************************************************************************************************************************* +* Header Files * +*********************************************************************************************************************************/ +#include <iprt/string.h> +#include "internal/iprt.h" + +#include <iprt/assert.h> + + +/********************************************************************************************************************************* +* Structures and Typedefs * +*********************************************************************************************************************************/ +/** rtStrPrintf2Output() argument structure. */ +typedef struct STRPRINTF2OUTPUTARGS +{ + /** Pointer to current buffer position. */ + char *pszCur; + /** Number of bytes left in the buffer (including the trailing zero). */ + size_t cbLeft; + /** Set if we overflowed. */ + bool fOverflowed; +} STRPRINTF2OUTPUTARGS; +/** Pointer to a rtStrPrintf2Output() argument structure. */ +typedef STRPRINTF2OUTPUTARGS *PSTRPRINTF2OUTPUTARGS; + + +/** + * Output callback. + * + * @returns cbChars + * + * @param pvArg Pointer to a STRBUFARG structure. + * @param pachChars Pointer to an array of utf-8 characters. + * @param cbChars Number of bytes in the character array pointed to by pachChars. + */ +static DECLCALLBACK(size_t) rtStrPrintf2Output(void *pvArg, const char *pachChars, size_t cbChars) +{ + PSTRPRINTF2OUTPUTARGS pArgs = (PSTRPRINTF2OUTPUTARGS)pvArg; + char *pszCur = pArgs->pszCur; /* We actually have to spell this out for VS2010, or it will load it for each case. */ + + if (cbChars < pArgs->cbLeft) + { + pArgs->cbLeft -= cbChars; + + /* Note! For VS2010/64 we need at least 7 case statements before it generates a jump table. */ + switch (cbChars) + { + default: + memcpy(pszCur, pachChars, cbChars); + break; + case 8: pszCur[7] = pachChars[7]; RT_FALL_THRU(); + case 7: pszCur[6] = pachChars[6]; RT_FALL_THRU(); + case 6: pszCur[5] = pachChars[5]; RT_FALL_THRU(); + case 5: pszCur[4] = pachChars[4]; RT_FALL_THRU(); + case 4: pszCur[3] = pachChars[3]; RT_FALL_THRU(); + case 3: pszCur[2] = pachChars[2]; RT_FALL_THRU(); + case 2: pszCur[1] = pachChars[1]; RT_FALL_THRU(); + case 1: pszCur[0] = pachChars[0]; RT_FALL_THRU(); + case 0: + break; + } + pArgs->pszCur = pszCur += cbChars; + *pszCur = '\0'; + } + else + { + size_t cbLeft = pArgs->cbLeft; + if (cbLeft-- > 1) + { + memcpy(pszCur, pachChars, cbLeft); + pArgs->pszCur = pszCur += cbLeft; + *pszCur = '\0'; + pArgs->cbLeft = 1; + } + pArgs->fOverflowed = true; + } + + return cbChars; +} + + +RTDECL(ssize_t) RTStrPrintf2(char *pszBuffer, size_t cchBuffer, const char *pszFormat, ...) +{ + /* Explicitly inline RTStrPrintf2V + RTStrPrintf2ExV here because this is a frequently use API. */ + STRPRINTF2OUTPUTARGS Args; + size_t cchRet; + va_list args; + AssertMsg(cchBuffer > 0, ("Excellent idea! Format a string with no space for the output!\n")); + + Args.pszCur = pszBuffer; + Args.cbLeft = cchBuffer; + Args.fOverflowed = false; + + va_start(args, pszFormat); + cchRet = RTStrFormatV(rtStrPrintf2Output, &Args, NULL, NULL, pszFormat, args); + va_end(args); + + return !Args.fOverflowed ? (ssize_t)cchRet : -(ssize_t)cchRet - 1; +} +RT_EXPORT_SYMBOL(RTStrPrintf2); + + +RTDECL(ssize_t) RTStrPrintf2ExV(PFNSTRFORMAT pfnFormat, void *pvArg, char *pszBuffer, size_t cchBuffer, + const char *pszFormat, va_list args) +{ + STRPRINTF2OUTPUTARGS Args; + size_t cchRet; + AssertMsg(cchBuffer > 0, ("Excellent idea! Format a string with no space for the output!\n")); + + Args.pszCur = pszBuffer; + Args.cbLeft = cchBuffer; + Args.fOverflowed = false; + cchRet = RTStrFormatV(rtStrPrintf2Output, &Args, pfnFormat, pvArg, pszFormat, args); + return !Args.fOverflowed ? (ssize_t)cchRet : -(ssize_t)cchRet - 1; +} +RT_EXPORT_SYMBOL(RTStrPrintf2ExV); + + +RTDECL(ssize_t) RTStrPrintf2V(char *pszBuffer, size_t cchBuffer, const char *pszFormat, va_list args) +{ + return RTStrPrintf2ExV(NULL, NULL, pszBuffer, cchBuffer, pszFormat, args); +} +RT_EXPORT_SYMBOL(RTStrPrintf2V); + + +RTDECL(ssize_t) RTStrPrintf2Ex(PFNSTRFORMAT pfnFormat, void *pvArg, char *pszBuffer, size_t cchBuffer, const char *pszFormat, ...) +{ + va_list args; + ssize_t cbRet; + va_start(args, pszFormat); + cbRet = RTStrPrintf2ExV(pfnFormat, pvArg, pszBuffer, cchBuffer, pszFormat, args); + va_end(args); + return cbRet; +} +RT_EXPORT_SYMBOL(RTStrPrintf2Ex); + |