Changeset 30270 in vbox for trunk/src/VBox/Runtime/r3/socket.cpp

Timestamp:

Jun 17, 2010 6:56:26 AM (14 years ago)

Author:

vboxsync

Message:

Runtime/tcp+socket: Add function to write to a socket using a scatter/gather buffer

File:

• trunk/src/VBox/Runtime/r3/socket.cpp (modified) (3 diffs)

trunk/src/VBox/Runtime/r3/socket.cpp

@@ -45 +45 @@
 # include <unistd.h>
 # include <fcntl.h>
+# include <sys/uio.h>
 #endif /* !RT_OS_WINDOWS */
 #include <limits.h>
@@ -59 +60 @@
 #include <iprt/thread.h>
 #include <iprt/time.h>
+#include <iprt/mem.h>
+#include <iprt/sg.h>
 
 #include "internal/magics.h"
@@ -574 +577 @@
 
 
+RTDECL(int) RTSocketSgWrite(RTSOCKET hSocket, PCRTSGBUF pSgBuf)
+{
+    /*
+     * Validate input.
+     */
+    RTSOCKETINT *pThis = hSocket;
+    AssertPtrReturn(pThis, VERR_INVALID_HANDLE);
+    AssertReturn(pThis->u32Magic == RTSOCKET_MAGIC, VERR_INVALID_HANDLE);
+    AssertPtrReturn(pSgBuf, VERR_INVALID_PARAMETER);
+    AssertReturn(pSgBuf->cSeg > 0, VERR_INVALID_PARAMETER);
+    AssertReturn(rtSocketTryLock(pThis), VERR_CONCURRENT_ACCESS);
+
+    /*
+     * Construct message descriptor (translate pSgBuf) and send it.
+     */
+    int rc = VINF_SUCCESS;
+    do
+    {
+#ifdef RT_OS_WINDOWS
+        AssertCompileSize(WSABUF, sizeof(RTSGSEG));
+        AssertCompileMemberSize(WSABUF, buf, RT_SIZEOFMEMB(RTSGSEG, pvSeg));
+        AssertCompileMemberSize(WSABUF, len, RT_SIZEOFMEMB(RTSGSEG, cbSeg));
+
+        LPWSABUF *paMsg = (LPWSABUF)RTMemTmpAllocZ(pSgBuf->cSeg * sizeof(WSABUF));
+        AssertPtrBreakStmt(paMsg, rc = VERR_NO_MEMORY);
+        for (unsigned i = 0; i < pSgBuf->cSeg; i++)
+        {
+            paMsg[i].buf = pSgBuf->pcaSeg[i].pvSeg;
+            paMsg[i].len = pSgBuf->pcaSeg[i].cbSeg;
+        }
+
+        WSAMSG msgHdr;
+        DWORD dwSent;
+        memset(&msgHdr, '\0', sizeof(msgHdr));
+        msgHdr.lpBuffers = paMsg;
+        msgHdr.dwBufferCount = pSgBuf->cSeg;
+        int hrc = WSASendMsg(pThis->hNative, &msgHdr, MSG_NOSIGNAL, &dwSent,
+                             NULL, NULL);
+        ssize_t cbWritten;
+        if (!hrc)
+        {
+            /* avoid overflowing ssize_t, the exact value isn't important */
+            cbWritten = RT_MIN(dwSent, INT_MAX);
+        }
+        else
+            cbWritten = -1;
+#else
+        AssertCompileSize(struct iovec, sizeof(RTSGSEG));
+        AssertCompileMemberSize(struct iovec, iov_base, RT_SIZEOFMEMB(RTSGSEG, pvSeg));
+        AssertCompileMemberSize(struct iovec, iov_len, RT_SIZEOFMEMB(RTSGSEG, cbSeg));
+
+        struct iovec *paMsg = (struct iovec *)RTMemTmpAllocZ(pSgBuf->cSeg * sizeof(struct iovec));
+        AssertPtrBreakStmt(paMsg, rc = VERR_NO_MEMORY);
+        for (unsigned i = 0; i < pSgBuf->cSeg; i++)
+        {
+            paMsg[i].iov_base = pSgBuf->pcaSeg[i].pvSeg;
+            paMsg[i].iov_len = pSgBuf->pcaSeg[i].cbSeg;
+        }
+
+        struct msghdr msgHdr;
+        memset(&msgHdr, '\0', sizeof(msgHdr));
+        msgHdr.msg_iov = paMsg;
+        msgHdr.msg_iovlen = pSgBuf->cSeg;
+        ssize_t cbWritten = sendmsg(pThis->hNative, &msgHdr, MSG_NOSIGNAL);
+#endif
+
+        RTMemTmpFree(paMsg);
+        if (RT_LIKELY(cbWritten >= 0))
+            rc = VINF_SUCCESS;
+        else
+            rc = rtSocketError();
+    } while (0);
+
+    rtSocketUnlock(pThis);
+    return rc;
+}
+
+
 RTDECL(int) RTSocketSelectOne(RTSOCKET hSocket, RTMSINTERVAL cMillies)
 {