| OLD | NEW |
| 1 // Copyright (c) 2010 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2010 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 #ifndef REMOTING_PROTOCOL_MESSAGES_DECODER_H_ | 5 #ifndef REMOTING_PROTOCOL_MESSAGES_DECODER_H_ |
| 6 #define REMOTING_PROTOCOL_MESSAGES_DECODER_H_ | 6 #define REMOTING_PROTOCOL_MESSAGES_DECODER_H_ |
| 7 | 7 |
| 8 #include <deque> | 8 #include <deque> |
| 9 #include <list> | 9 #include <list> |
| 10 | 10 |
| (...skipping 17 matching lines...) Expand all Loading... |
| 28 // message_data in bytes. message_data - content of the message. | 28 // message_data in bytes. message_data - content of the message. |
| 29 class MessageDecoder { | 29 class MessageDecoder { |
| 30 public: | 30 public: |
| 31 MessageDecoder(); | 31 MessageDecoder(); |
| 32 virtual ~MessageDecoder(); | 32 virtual ~MessageDecoder(); |
| 33 | 33 |
| 34 // Add next chunk of data. MessageDecoder retains |data| until all | 34 // Add next chunk of data. MessageDecoder retains |data| until all |
| 35 // its bytes are consumed. | 35 // its bytes are consumed. |
| 36 void AddData(scoped_refptr<net::IOBuffer> data, int data_size); | 36 void AddData(scoped_refptr<net::IOBuffer> data, int data_size); |
| 37 | 37 |
| 38 // Get next message from the stream and puts it in | 38 // Returns next message from the stream. Ownership of the result is |
| 39 // |message_buffer|. Returns false if there are no complete messages | 39 // passed to the caller. Returns NULL if there are no complete |
| 40 // yet. | 40 // messages yet, otherwise returns a buffer that contains one |
| 41 bool GetNextMessage(CompoundBuffer* message_buffer); | 41 // message. |
| 42 CompoundBuffer* GetNextMessage(); |
| 42 | 43 |
| 43 private: | 44 private: |
| 44 // Retrieves the read payload size of the current protocol buffer via |size|. | 45 // Retrieves the read payload size of the current protocol buffer via |size|. |
| 45 // Returns false and leaves |size| unmodified, if we do not have enough data | 46 // Returns false and leaves |size| unmodified, if we do not have enough data |
| 46 // to retrieve the current size. | 47 // to retrieve the current size. |
| 47 bool GetPayloadSize(int* size); | 48 bool GetPayloadSize(int* size); |
| 48 | 49 |
| 49 CompoundBuffer buffer_; | 50 CompoundBuffer buffer_; |
| 50 | 51 |
| 51 // |next_payload_| stores the size of the next payload if known. | 52 // |next_payload_| stores the size of the next payload if known. |
| 52 // |next_payload_known_| is true if the size of the next payload is known. | 53 // |next_payload_known_| is true if the size of the next payload is known. |
| 53 // After one payload is read this is reset to false. | 54 // After one payload is read this is reset to false. |
| 54 int next_payload_; | 55 int next_payload_; |
| 55 bool next_payload_known_; | 56 bool next_payload_known_; |
| 56 }; | 57 }; |
| 57 | 58 |
| 58 } // namespace protocol | 59 } // namespace protocol |
| 59 } // namespace remoting | 60 } // namespace remoting |
| 60 | 61 |
| 61 #endif // REMOTING_PROTOCOL_MESSAGES_DECODER_H_ | 62 #endif // REMOTING_PROTOCOL_MESSAGES_DECODER_H_ |
| OLD | NEW |