4

バイトを削除せずにストリームバッファから読み取る方法はありますか?

メッセージ全体が受信されたかどうかを確認するために、バッファから「メッセージ サイズ」フィールドを読み取っています。

そうでない場合は、それを取得するために別の非同期読み取りを投稿していますが、サイズ フィールドが削除されたため、ハンドラーはメッセージの長さを知る方法がありません。

どんな助けでも大歓迎です！

例えば

boost::asio::streambuf _buffer;

void onReceive(const boost::system::error_code& e, std::size_t bytesTransferred)
{
  if(e) return;

  if(_buffer.size() > 0)
  {
    // Partial message was previously received, but I don't know how long.
  }
  else
  {
    _buffer.commit(bytesTransferred);

    /* Read the size (and remove it from the stream) */
    unsigned short size = 0;
    std::istream in(&_buffer);
    in.read((char*)&size, sizeof(unsigned short);

    /* Got the whole message? */
    if(_buffer.size() > size)
    {
      /* Yes. */
    }
    else
    {
      /* No - read the rest. */
      boost::asio::async_read(/*...*/);
    }
  }
}
4

3 に答える 3

2

read_async を使用して、メッセージ ヘッダーのサイズを使用して読み取りを開始し、次のように「完了条件」コールバックで調整できます。

typedef boost::system::error_code error_code;

template <typename Stream, typename Message>
void MessageReader<Stream, Message>::startRead()
{
  readBuffer = allocateMsg();
  async_read(stream, 
             boost::asio::buffer(readBuffer.get(), sizeof(*readBuffer)),
             boost::bind(&MessageReader<Stream, Message>::bytesToRead, this,
                         boost::asio::placeholders::error, 
                         boost::asio::placeholders::bytes_transferred),
             boost::bind(&MessageReader<Stream, Message>::readDone, this, 
                         boost::asio::placeholders::error, 
                         boost::asio::placeholders::bytes_transferred));
}

template <typename Stream, typename Message>
size_t MessageReader<Stream, Message>::bytesToRead(const error_code& error, 
                                                   size_t bytes_read)
{
  size_t result;

  if (error)
    result = 0;                                         // error - stop reading

  else if (bytes_read < sizeof(CmnMessageHeader))
    result = sizeof(CmnMessageHeader) - bytes_read;     // read rest of header

  else if (readBuffer->header.byteCount > sizeof(*readBuffer))
    result = 0;                                         // bad byte count

  else
    result = readBuffer->header.byteCount - bytes_read; // read message body

  return result;
}

template <typename Stream, typename Message>
void MessageReader<Stream, Message>::readDone(const error_code& error, 
                                              size_t bytes_read)
{
  if (error)
  {
    if (error.value() == boost::system::errc::no_such_file_or_directory)
    {
      notifyStop();
    }

    else if (error.value() != boost::system::errc::operation_canceled)
    {
      notifyStop();
    }

    // else the operation was cancelled, thus no stop notification is needed and
    // we can merely return
  }

  else if (bytes_read != readBuffer->header.byteCount)
  {
    LOG4CXX_ERROR(logger, "Message byte count mismatch");
    notifyStop();
  }

  else
  {
    handleMsg(readBuffer);
    startRead();
  }
}

編集: error_code の typedef を追加しました。

于 2012-02-05T04:03:09.900 に答える
0

採用できるアプローチは 2 つあります。

  1. サイズ (たとえば 4) のバイト数を読み取るために単一の読み取りを発行し、必要なサイズの読み取りを発行します。

  2. read some 呼び出しを使用して、コード内のバイトをバッファリングし、たとえばベクトルで分析して、そのように分析します。

オプション 2 を選択します。これはバッファーをコピーすることを意味しますが、複数の read some 呼び出しよりも安価であるという危険があります。

于 2012-02-05T00:02:38.557 に答える