如何从套接字读取并返回单独的行？

Question

我是否应该读取每个字符，直到它到达 \n 字符，将它们全部连接在一起并返回，或者是否有更好的方法？我应该使用 std::string 还是 char 来实现此目的？

我尝试了以下两个示例，但我需要将它们作为单独的行来读取

示例 1:

std::string sockread()
{
    std::string s;
    s.resize(DEFAULT_BUFLEN);
    int result = recv(m_socket, &s[0], DEFAULT_BUFLEN, 0);

    if (result > 0) {
        return s;
    } else if (result == 0) {
        connected = false;
    } else {
        std::cout << "recv failed with error " << WSAGetLastError() << "\n";
    }
    throw std::runtime_error("Socket connection failed!");
}

示例 2:

char sockread(void)

  {
    int result;
    char buffer[DEFAULT_BUFLEN];
        result = recv(m_socket, buffer, DEFAULT_BUFLEN, 0);

        if (result > 0) {
             return *buffer;
              }
          else if (result == 0)
              {
             connected = false;
                 return *buffer;
              }
          else {
         printf("recv failed with error: %d\n", WSAGetLastError());
         return *buffer;
        }

   }

Answer 1

您有几个选择，具体取决于套接字代码的其余部分的布局方式。

从编码的角度来看，最简单的方法是一次只读取 1 个字符，直到遇到要查找的字符。从性能角度来看，这不是最好的方法，尽管您可以使用本地缓冲区来帮助您至少避免内存碎片，例如：

std::string sockread(void) 
{ 
    char buffer[DEFAULT_BUFLEN]; 
    int buflen = 0;
    char c;
    std::string s;

    do
    {
        int result = recv(m_socket, &c, 1, 0); 
        if (result > 0)
        { 
            if (c == '\n')
                break;

            if (buflen == DEFAULT_BUFLEN)
            {
                s += std::string(buffer, buflen);
                buflen = 0;
            }

            buffer[buflen] = c;
            ++buflen;

            continue;
        } 

        if (result == SOCKET_ERROR) 
        {
            if (WSAGetLastError() == WSAEWOULDBLOCK)
                continue;

            std::cout << "recv failed with error " << WSAGetLastError() << "\n";
        }
        else
            connected = false; 

        throw std::runtime_error("Socket connection failed!");
    }
    while (true);

    if (buflen > 0)
        s += std::string(buffer, buflen);

    return s;
}

另一方面，将原始套接字数据读入中间缓冲区，其余的读取功能将在中间缓冲区中进行需要时访问可以更有效地读取套接字，以便更快地从套接字缓冲区中获取数据（减少另一端的阻塞），例如：

std::vector<unsigned char> buffer;

std::string sockread(void) 
{ 
    unsigned char buf[DEFAULT_BUFLEN];
    int result;
    std:vector<unsigned char>::iterator it;

    do
    {
        it = std::find(buffer.begin(), buffer.end(), '\n');
        if (it != buffer.end())
            break;

        result = recv(m_socket, buf, DEFAULT_BUFLEN, 0); 
        if (result > 0)
        { 
            std::vector<unsigned char>::size_type pos = buffer.size();
            buffer.resize(pos + result);
            memcpy(&buffer[pos], buf, result);
            continue;
        } 

        if (result == SOCKET_ERROR) 
        {
            if (WSAGetLastError() == WSAEWOULDBLOCK)
                continue;

            std::cout << "recv failed with error " << WSAGetLastError() << "\n";
        }
        else
            connected = false; 

        throw std::runtime_error("Socket connection failed!");
    }
    while (true);

    std::string s((char*)&buffer[0], std::distance(buffer.begin(), it));
    buffer.erase(buffer.begin(), it);
    return s;
}

Answer 2

使用Boost.ASIO - 基于行的操作这里。

许多常用的互联网协议
是基于线路的，这意味着它们
具有以下协议元素
由字符序列分隔
“\r\n”。示例包括 HTTP、SMTP
和FTP。为了更容易地允许
基于线路的实施
协议以及其他协议
使用分隔符，Boost.Asio
包括函数 read_until()
和 async_read_until()。