Changeset 89264 in vbox for trunk/src/VBox/Runtime/common/misc

Timestamp:

May 25, 2021 11:09:36 AM (4 years ago)

Author:

vboxsync

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

144607

Message:

Runtime/sg.h: Add copy variants which take a callback instead of a buffer to do the actual copying for single segments, bugref:4841

File:

• trunk/src/VBox/Runtime/common/misc/sg.cpp (modified) (1 diff)

trunk/src/VBox/Runtime/common/misc/sg.cpp

@@ -347 +347 @@
 
 
+RTDECL(size_t) RTSgBufCopyToFn(PRTSGBUF pSgBuf, size_t cbCopy, PFNRTSGBUFCOPYTO pfnCopyTo, void *pvUser)
+{
+    AssertPtrReturn(pSgBuf, 0);
+    AssertPtrReturn(pfnCopyTo, 0);
+
+    size_t cbLeft = cbCopy;
+
+    while (cbLeft)
+    {
+        size_t cbThisCopy = cbLeft;
+        void *pvSrc = rtSgBufGet(pSgBuf, &cbThisCopy);
+
+        if (!cbThisCopy)
+            break;
+
+        size_t cbThisCopied = pfnCopyTo(pSgBuf, pvSrc, cbThisCopy, pvUser);
+        cbLeft -= cbThisCopied;
+        if (cbThisCopied < cbThisCopy)
+            break;
+    }
+
+    return cbCopy - cbLeft;
+}
+
+
+RTDECL(size_t) RTSgBufCopyFromFn(PRTSGBUF pSgBuf, size_t cbCopy, PFNRTSGBUFCOPYFROM pfnCopyFrom, void *pvUser)
+{
+    AssertPtrReturn(pSgBuf, 0);
+    AssertPtrReturn(pfnCopyFrom, 0);
+
+    size_t cbLeft = cbCopy;
+
+    while (cbLeft)
+    {
+        size_t cbThisCopy = cbLeft;
+        void *pvDst = rtSgBufGet(pSgBuf, &cbThisCopy);
+
+        if (!cbThisCopy)
+            break;
+
+        size_t cbThisCopied = pfnCopyFrom(pSgBuf, pvDst, cbThisCopy, pvUser);
+        cbLeft -= cbThisCopied;
+        if (cbThisCopied < cbThisCopy)
+            break;
+    }
+
+    return cbCopy - cbLeft;
+}
+
+
 RTDECL(size_t) RTSgBufAdvance(PRTSGBUF pSgBuf, size_t cbAdvance)
 {