1

我对何时需要将数据存储在等待缓冲区（等待 FD_WRITE 事件）有一些疑问。

这是我的发送功能（固定）：

bool MyClass::DataSend(char *buf, int len)
{
    if (len <=0 || m_Socket == INVALID_SOCKET) return false;

    if (m_SendBufferLen > 0)
    {
        if ((m_SendBufferLen + len) < MAX_BUFF_SIZE)
        {
            memcpy((m_SendBuffer + m_SendBufferLen), buf, len);
            m_SendBufferLen += len;
            return true;
        }
        else
        {
            // close the connection and log insuficient wait buffer size
            return false;
        }
    }

    int iResult;
    int nPosition = 0;
    int nLeft = len;

    while (true)
    {
        iResult = send(m_Socket, (char*)(buf + nPosition), nLeft, 0);

        if (iResult != SOCKET_ERROR)
        {
            if (iResult > 0)
            {
                nPosition   += iResult;
                nLeft       -= iResult;

            }
            else
            {
                // log 0 bytes sent
                break;
            }
        }
        else
        {
            if (WSAGetLastError() == WSAEWOULDBLOCK)
            {
                if ((m_SendBufferLen + nLeft) < MAX_BUFF_SIZE)
                {
                    // log data copied to the wait buffer
                    memcpy((m_SendBuffer + m_SendBufferLen), (buf + nPosition), nLeft);
                    m_SendBufferLen += nLeft;
                    return true;
                }
                else
                {
                    // close the connection and log insuficient wait buffer size
                    return false;
                }
            }
            else
            {
                // close the connection and log winsock error
                return false;
            }
        }

        if (nLeft <= 0) break;
    }

    return true;
}

我的发送（FD_WRITE 事件）功能（已修复）：

bool MyClass::DataSendEvent()
{
    if (m_SendBufferLen < 1) return true;

    int iResult;
    int nPosition = 0;
    int nLeft = m_SendBufferLen;

    while (true)
    {
        iResult = send(m_Socket, (char*)(m_SendBuffer + nPosition), nLeft, 0);

        if (iResult != SOCKET_ERROR)
        {
            if (iResult > 0)
            {
                nPosition   += iResult;
                nLeft       -= iResult;
            }
            else
            {
                // log 0 bytes sent
                break;
            }
        }
        else
        {
            if (WSAGetLastError() == WSAEWOULDBLOCK)
            {
                if (nPosition > 0)
                {
                    memmove(m_SendBuffer, (m_SendBuffer + nPosition), (m_SendBufferLen - nPosition));
                    m_SendBufferLen -= nPosition;
                }

                break;
            }
            else
            {
                // close the connection and log winsock error
                return false;
            }
        }

        if (nLeft <= 0)
        {
            if (m_SendBufferLen == nPosition)
            {
                m_SendBufferLen = 0;
                break;
            }
            else
            {
                memmove(m_SendBuffer, (m_SendBuffer + nPosition), (m_SendBufferLen - nPosition));
                m_SendBufferLen -= nPosition;
                nPosition   = 0;
                nLeft       = m_SendBufferLen;
            }
        }
    }

    return true;
}

我真的需要if (nPosition > 0)还是不需要？如何模拟这种情况？非阻塞模式下的 send() 发送的字节数是否可能少于请求的字节数？如果不是，为什么要使用 while() 循环？

  • 这是最终代码（感谢@user315052）
4

1 回答 1

0

在你的 while 循环的顶部，你已经在 decrementing nLeft，所以你不需要从中减去nPosition。

iResult = send(m_Socket, (char*)(buf + nPosition), nLeft, 0);

在您的第二个函数中，当您将未发送的字节移动到数组的开头时，您应该使用memmove，因为您有重叠的区域（您正在将一个区域复制m_SendBuffer到自身中）。重叠如下图所示，其中一些A' 将被复制到自身上。

m_SendBuffer: [XXXXAAAAAAAAAAAA]
mPosition: 4
nLeft: 12

我有点困惑，为什么DataSend()要实现允许调用者继续成功调用它，即使WSAEWOULDBLOCK遇到 。我建议修改接口以返回一个结果，让调用者知道它应该停止发送，并等待恢复发送的指示。

你不需要nPosition > 0支票。

您可以通过让数据的接收者不读取任何内容来强制发生这种情况。

绝对有可能send在非阻塞模式下发送比请求更少的字节。

于 2012-06-10T23:09:04.407 回答