1// Copyright 2014 The Chromium Authors. All rights reserved.
2// Use of this source code is governed by a BSD-style license that can be
3// found in the LICENSE file.
4
5#include "remoting/protocol/message_serialization.h"
6
7#include "base/basictypes.h"
8#include "base/containers/hash_tables.h"
9#include "base/logging.h"
10#include "net/base/io_buffer.h"
11#include "third_party/webrtc/base/byteorder.h"
12
13namespace remoting {
14namespace protocol {
15
16scoped_refptr<net::IOBufferWithSize> SerializeAndFrameMessage(
17    const google::protobuf::MessageLite& msg) {
18  // Create a buffer with 4 extra bytes. This is used as prefix to write an
19  // int32 of the serialized message size for framing.
20  const int kExtraBytes = sizeof(int32);
21  int size = msg.ByteSize() + kExtraBytes;
22  scoped_refptr<net::IOBufferWithSize> buffer(new net::IOBufferWithSize(size));
23  rtc::SetBE32(buffer->data(), msg.GetCachedSize());
24  msg.SerializeWithCachedSizesToArray(
25      reinterpret_cast<uint8*>(buffer->data()) + kExtraBytes);
26  return buffer;
27}
28
29}  // namespace protocol
30}  // namespace remoting
31