Changeset 26482 in vbox for trunk/src/VBox/Runtime

Timestamp:

Feb 14, 2010 1:38:44 AM (15 years ago)

Author:

vboxsync

svn:sync-xref-src-repo-rev:

57651

Message:

IPRT: Added RTStrATruncate.

Location:

trunk/src/VBox/Runtime

Files:

• common/string/stringalloc.cpp (modified) (1 diff)
• testcase/tstRTStrAlloc.cpp (modified) (1 diff)

trunk/src/VBox/Runtime/common/string/stringalloc.cpp

@@ -249 +249 @@
 RT_EXPORT_SYMBOL(RTStrAAppendExN);
 
+
+RTDECL(int) RTStrATruncate(char **ppsz, size_t cchNew)
+{
+    char *pszOld = *ppsz;
+    if (!cchNew)
+    {
+        if (pszOld && *pszOld)
+        {
+            *pszOld = '\0';
+            char *pszNew = (char *)RTMemRealloc(pszOld, 1);
+            if (!pszNew)
+                *ppsz = pszNew;
+        }
+    }
+    else
+    {
+        AssertPtrReturn(pszOld, VERR_OUT_OF_RANGE);
+        char *pszZero = (char *)memchr(pszOld, '\0', cchNew + 63);
+        AssertReturn(!pszZero || (size_t)(pszZero - pszOld) >= cchNew, VERR_OUT_OF_RANGE);
+        pszOld[cchNew] = '\0';
+        if (!pszZero)
+        {
+            char *pszNew = (char *)RTMemRealloc(pszOld,  cchNew + 1);
+            if (pszNew)
+                *ppsz = pszNew;
+        }
+    }
+    return VINF_SUCCESS;
+}
+RT_EXPORT_SYMBOL(RTStrATruncate);
+

trunk/src/VBox/Runtime/testcase/tstRTStrAlloc.cpp

@@ -203 +203 @@
     RTTESTI_CHECK(!RTStrCmp(psz, "abcdefghijklmnopqrstuvwxyz-"));
     RTStrFree(psz);
+
+    /* RTStrATruncate */
+    psz = NULL;
+    RTTESTI_CHECK_RC(RTStrATruncate(&psz, 0), VINF_SUCCESS);
+    RTTESTI_CHECK(!psz);
+
+    RTTESTI_CHECK(psz = RTStrDup(""));
+    RTTESTI_CHECK_RC(RTStrATruncate(&psz, 0), VINF_SUCCESS);
+    RTStrFree(psz);
+
+    RTTESTI_CHECK(psz = RTStrDup("1234567890"));
+    RTTESTI_CHECK_RC(RTStrATruncate(&psz, 5), VINF_SUCCESS);
+    RTTESTI_CHECK(!RTStrCmp(psz, "12345"));
+    RTStrFree(psz);
+
+    psz = NULL;
+    for (uint32_t i = 0; i < 128; i++)
+        RTTESTI_CHECK_RC_RETV(RTStrAAppend(&psz, "abcdefghijklmnopqrstuvwxyz"), VINF_SUCCESS);
+    RTTESTI_CHECK_RC(RTStrATruncate(&psz, sizeof("abcdefghijklmnopqrstuvwxyz") - 1), VINF_SUCCESS);
+    RTTESTI_CHECK(!RTStrCmp(psz, "abcdefghijklmnopqrstuvwxyz"));
+    RTTESTI_CHECK_RC(RTStrATruncate(&psz, 6), VINF_SUCCESS);
+    RTTESTI_CHECK(!RTStrCmp(psz, "abcdef"));
+    RTTESTI_CHECK_RC(RTStrATruncate(&psz, 1), VINF_SUCCESS);
+    RTTESTI_CHECK(!RTStrCmp(psz, "a"));
+    RTTESTI_CHECK_RC(RTStrATruncate(&psz, 0), VINF_SUCCESS);
+    RTTESTI_CHECK(!RTStrCmp(psz, ""));
+    RTStrFree(psz);
+
 }