Changeset 74823 in vbox for trunk/src/VBox/Main

Timestamp:

Oct 12, 2018 6:42:01 PM (6 years ago)

Author:

vboxsync

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

125776

Message:

Main: bugref:9152 Implement the convertToStream API [missing file]

File:

• trunk/src/VBox/Main/src-server/DataStreamImpl.cpp (modified) (9 diffs)

trunk/src/VBox/Main/src-server/DataStreamImpl.cpp

@@ -78 +78 @@
     HRESULT hrc = S_OK;
 
-    m_aBufferSize        = aBufferSize;
     m_hSemEvtDataAvail   = NIL_RTSEMEVENT;
     m_hSemEvtBufSpcAvail = NIL_RTSEMEVENT;
+    m_pBuffer            = NULL;
     int vrc = RTSemEventCreate(&m_hSemEvtDataAvail);
     if (RT_SUCCESS(vrc))
         vrc = RTSemEventCreate(&m_hSemEvtBufSpcAvail);
+    if (RT_SUCCESS(vrc))
+        vrc = RTCircBufCreate(&m_pBuffer, aBufferSize);
 
     if (RT_FAILURE(vrc))
@@ -111 +113 @@
     if (!autoUninitSpan.uninitDone())
     {
-        m_aBuffer.clear();
         if (m_hSemEvtDataAvail != NIL_RTSEMEVENT)
             RTSemEventDestroy(m_hSemEvtDataAvail);
         if (m_hSemEvtBufSpcAvail != NIL_RTSEMEVENT)
             RTSemEventDestroy(m_hSemEvtBufSpcAvail);
+        if (m_pBuffer != NULL)
+            RTCircBufDestroy(m_pBuffer);
         m_hSemEvtDataAvail = NIL_RTSEMEVENT;
         m_hSemEvtBufSpcAvail = NIL_RTSEMEVENT;
@@ -131 +134 @@
 {
     AutoReadLock alock(this COMMA_LOCKVAL_SRC_POS);
-    *aReadSize = (ULONG)m_aBuffer.size();
+    *aReadSize = (ULONG)RTCircBufUsed(m_pBuffer);
     return S_OK;
 }
@@ -160 +163 @@
 
     int vrc = VINF_SUCCESS;
-    while (   !m_aBuffer.size()
+    while (   !RTCircBufUsed(m_pBuffer)
            && !m_fEos
            && RT_SUCCESS(vrc))
@@ -175 +178 @@
     HRESULT hrc = S_OK;
     if (   RT_SUCCESS(vrc)
-        && m_aBuffer.size())
-    {
-        size_t cbCopy = RT_MIN(aSize, m_aBuffer.size());
+        && RTCircBufUsed(m_pBuffer))
+    {
+
+        size_t off = 0;
+        size_t cbCopy = RT_MIN(aSize, RTCircBufUsed(m_pBuffer));
         if (cbCopy != aSize)
         {
@@ -183 +188 @@
             aData.resize(cbCopy);
         }
-        aData.insert(aData.begin(), m_aBuffer.begin(), m_aBuffer.begin() + cbCopy);
-        m_aBuffer.erase(m_aBuffer.begin(), m_aBuffer.begin() + cbCopy);
+
+        while (cbCopy)
+        {
+            void *pvSrc = NULL;
+            size_t cbThisCopy = 0;
+
+            RTCircBufAcquireReadBlock(m_pBuffer, cbCopy, &pvSrc, &cbThisCopy);
+            memcpy(&aData.front() + off, pvSrc, cbThisCopy);
+            RTCircBufReleaseReadBlock(m_pBuffer, cbThisCopy);
+
+            cbCopy -= cbThisCopy;
+            off    += cbThisCopy;
+        }
         vrc = RTSemEventSignal(m_hSemEvtBufSpcAvail);
         AssertRC(vrc);
@@ -192 +208 @@
         Assert(   RT_FAILURE(vrc)
                || (   m_fEos
-                   && !m_aBuffer.size()));
+                   && !RTCircBufUsed(m_pBuffer)));
 
         aData.resize(0);
@@ -222 +238 @@
     AssertReturn(!m_fEos, VERR_INVALID_STATE);
 
+    *pcbWritten = 0;
+
     int vrc = VINF_SUCCESS;
-    while (   m_aBuffer.size() == m_aBufferSize
+    while (   !RTCircBufFree(m_pBuffer)
            && RT_SUCCESS(vrc))
     {
@@ -234 +252 @@
     if (RT_SUCCESS(vrc))
     {
-        Assert(m_aBuffer.size() < m_aBufferSize);
-        size_t cbCopy = RT_MIN(cbWrite, m_aBufferSize - m_aBuffer.size());
-        if (m_aBuffer.capacity() < m_aBuffer.size() + cbWrite)
-            m_aBuffer.reserve(m_aBuffer.size() + cbWrite);
-        memcpy(&m_aBuffer.front() + m_aBuffer.size(), pvBuf, cbCopy);
+        const uint8_t *pbBuf = (const uint8_t *)pvBuf;
+        size_t cbCopy = RT_MIN(cbWrite, RTCircBufFree(m_pBuffer));
+
         *pcbWritten = cbCopy;
+
+        while (cbCopy)
+        {
+            void *pvDst = NULL;
+            size_t cbThisCopy = 0;
+
+            RTCircBufAcquireWriteBlock(m_pBuffer, cbCopy, &pvDst, &cbThisCopy);
+            memcpy(pvDst, pbBuf, cbThisCopy);
+            RTCircBufReleaseWriteBlock(m_pBuffer, cbThisCopy);
+
+            cbCopy -= cbThisCopy;
+            pbBuf  += cbThisCopy;
+        }
+
         RTSemEventSignal(m_hSemEvtDataAvail);
     }