OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "remoting/protocol/message_decoder.h" | 5 #include "remoting/protocol/message_decoder.h" |
6 | 6 |
7 #include "base/logging.h" | 7 #include "base/logging.h" |
8 #include "net/base/io_buffer.h" | 8 #include "net/base/io_buffer.h" |
9 #include "remoting/base/compound_buffer.h" | 9 #include "remoting/base/compound_buffer.h" |
10 #include "remoting/proto/internal.pb.h" | 10 #include "remoting/proto/internal.pb.h" |
(...skipping 21 matching lines...) Expand all Loading... |
32 int next_payload = -1; | 32 int next_payload = -1; |
33 if (!next_payload_known_ && GetPayloadSize(&next_payload)) { | 33 if (!next_payload_known_ && GetPayloadSize(&next_payload)) { |
34 DCHECK_NE(-1, next_payload); | 34 DCHECK_NE(-1, next_payload); |
35 next_payload_ = next_payload; | 35 next_payload_ = next_payload; |
36 next_payload_known_ = true; | 36 next_payload_known_ = true; |
37 } | 37 } |
38 | 38 |
39 // If the next payload size is still not known or we don't have enough | 39 // If the next payload size is still not known or we don't have enough |
40 // data for parsing then exit. | 40 // data for parsing then exit. |
41 if (!next_payload_known_ || buffer_.total_bytes() < next_payload_) | 41 if (!next_payload_known_ || buffer_.total_bytes() < next_payload_) |
42 return NULL; | 42 return nullptr; |
43 | 43 |
44 CompoundBuffer* message_buffer = new CompoundBuffer(); | 44 CompoundBuffer* message_buffer = new CompoundBuffer(); |
45 message_buffer->CopyFrom(buffer_, 0, next_payload_); | 45 message_buffer->CopyFrom(buffer_, 0, next_payload_); |
46 message_buffer->Lock(); | 46 message_buffer->Lock(); |
47 buffer_.CropFront(next_payload_); | 47 buffer_.CropFront(next_payload_); |
48 next_payload_known_ = false; | 48 next_payload_known_ = false; |
49 | 49 |
50 return message_buffer; | 50 return message_buffer; |
51 } | 51 } |
52 | 52 |
53 bool MessageDecoder::GetPayloadSize(int* size) { | 53 bool MessageDecoder::GetPayloadSize(int* size) { |
54 // The header has a size of 4 bytes. | 54 // The header has a size of 4 bytes. |
55 const int kHeaderSize = sizeof(int32); | 55 const int kHeaderSize = sizeof(int32); |
56 | 56 |
57 if (buffer_.total_bytes() < kHeaderSize) | 57 if (buffer_.total_bytes() < kHeaderSize) |
58 return false; | 58 return false; |
59 | 59 |
60 CompoundBuffer header_buffer; | 60 CompoundBuffer header_buffer; |
61 char header[kHeaderSize]; | 61 char header[kHeaderSize]; |
62 header_buffer.CopyFrom(buffer_, 0, kHeaderSize); | 62 header_buffer.CopyFrom(buffer_, 0, kHeaderSize); |
63 header_buffer.CopyTo(header, kHeaderSize); | 63 header_buffer.CopyTo(header, kHeaderSize); |
64 *size = rtc::GetBE32(header); | 64 *size = rtc::GetBE32(header); |
65 buffer_.CropFront(kHeaderSize); | 65 buffer_.CropFront(kHeaderSize); |
66 return true; | 66 return true; |
67 } | 67 } |
68 | 68 |
69 } // namespace protocol | 69 } // namespace protocol |
70 } // namespace remoting | 70 } // namespace remoting |
OLD | NEW |