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 #ifndef NET_QUIC_QUIC_DATA_WRITER_H_ | 5 #ifndef NET_QUIC_QUIC_DATA_WRITER_H_ |
6 #define NET_QUIC_QUIC_DATA_WRITER_H_ | 6 #define NET_QUIC_QUIC_DATA_WRITER_H_ |
7 | 7 |
8 #include <string> | 8 #include <string> |
9 | 9 |
10 #include "base/basictypes.h" | 10 #include "base/basictypes.h" |
11 #include "base/logging.h" | 11 #include "base/logging.h" |
12 #include "base/port.h" | 12 #include "base/port.h" |
13 #include "base/string_piece.h" | 13 #include "base/string_piece.h" |
14 #include "net/base/net_export.h" | 14 #include "net/base/net_export.h" |
15 #include "net/quic/quic_protocol.h" | 15 #include "net/quic/quic_protocol.h" |
16 #include "net/quic/uint128.h" | 16 #include "net/quic/uint128.h" |
17 | 17 |
18 namespace net { | 18 namespace net { |
19 | 19 |
20 // This class provides facilities for packing QUIC data. | 20 // This class provides facilities for packing QUIC data. |
21 // | 21 // |
22 // The QuicDataWriter supports appending primitive values (int, string, etc) | 22 // The QuicDataWriter supports appending primitive values (int, string, etc) |
23 // to a frame instance. The QuicDataWriter grows its internal memory buffer | 23 // to a frame instance. The QuicDataWriter grows its internal memory buffer |
24 // dynamically to hold the sequence of primitive values. The internal memory | 24 // dynamically to hold the sequence of primitive values. The internal memory |
25 // buffer is exposed as the "data" of the QuicDataWriter. | 25 // buffer is exposed as the "data" of the QuicDataWriter. |
26 class NET_EXPORT_PRIVATE QuicDataWriter { | 26 class NET_EXPORT_PRIVATE QuicDataWriter { |
27 public: | 27 public: |
28 explicit QuicDataWriter(size_t length); | 28 explicit QuicDataWriter(size_t length); |
| 29 |
29 ~QuicDataWriter(); | 30 ~QuicDataWriter(); |
30 | 31 |
31 // Returns the size of the QuicDataWriter's data. | 32 // Returns the size of the QuicDataWriter's data. |
32 size_t length() const { return length_; } | 33 size_t length() const { return length_; } |
33 | 34 |
34 // Takes the buffer from the QuicDataWriter. | 35 // Takes the buffer from the QuicDataWriter. |
35 // TODO(rch): move non-trivial methods into .cc file. | 36 char* take(); |
36 char* take() { | |
37 char* rv = buffer_; | |
38 buffer_ = NULL; | |
39 capacity_ = 0; | |
40 length_ = 0; | |
41 return rv; | |
42 } | |
43 | 37 |
44 // Methods for adding to the payload. These values are appended to the end | 38 // Methods for adding to the payload. These values are appended to the end |
45 // of the QuicDataWriter payload. Note - binary integers are written in | 39 // of the QuicDataWriter payload. Note - binary integers are written in |
46 // host byte order (little endian) not network byte order (big endian). | 40 // host byte order (little endian) not network byte order (big endian). |
47 bool WriteUInt8(uint8 value) { | 41 bool WriteUInt8(uint8 value); |
48 return WriteBytes(&value, sizeof(value)); | 42 bool WriteUInt16(uint16 value); |
49 } | 43 bool WriteUInt32(uint32 value); |
50 bool WriteUInt16(uint16 value) { | 44 bool WriteUInt48(uint64 value); |
51 return WriteBytes(&value, sizeof(value)); | 45 bool WriteUInt64(uint64 value); |
52 } | 46 bool WriteUInt128(uint128 value); |
53 bool WriteUInt32(uint32 value) { | 47 bool WriteStringPiece16(base::StringPiece val); |
54 return WriteBytes(&value, sizeof(value)); | |
55 } | |
56 bool WriteUInt48(uint64 value) { | |
57 uint32 hi = value >> 32; | |
58 uint32 lo = value & GG_UINT64_C(0x00000000FFFFFFFF); | |
59 return WriteUInt32(lo) && WriteUInt16(hi); | |
60 } | |
61 bool WriteUInt64(uint64 value) { | |
62 return WriteBytes(&value, sizeof(value)); | |
63 } | |
64 bool WriteUInt128(uint128 value) { | |
65 return WriteUInt64(value.lo) && WriteUInt64(value.hi); | |
66 } | |
67 bool WriteStringPiece16(base::StringPiece val) { | |
68 if (val.length() > std::numeric_limits<uint16>::max()) { | |
69 return false; | |
70 } | |
71 if (!WriteUInt16(val.size())) { | |
72 return false; | |
73 } | |
74 return WriteBytes(val.data(), val.size()); | |
75 } | |
76 | |
77 bool AdvancePointer(uint32 len); | |
78 | |
79 bool WriteBytes(const void* data, uint32 data_len); | 48 bool WriteBytes(const void* data, uint32 data_len); |
80 | 49 |
81 static void WriteUint64ToBuffer(uint64 value, char* buffer); | 50 static void WriteUint64ToBuffer(uint64 value, char* buffer); |
82 static void WriteUint128ToBuffer(uint128 value, char* buffer); | 51 static void WriteUint128ToBuffer(uint128 value, char* buffer); |
83 | 52 |
84 size_t capacity() const { | 53 size_t capacity() const { |
85 return capacity_; | 54 return capacity_; |
86 } | 55 } |
87 | 56 |
88 protected: | 57 protected: |
89 const char* end_of_payload() const { return buffer_ + length_; } | 58 const char* end_of_payload() const { return buffer_ + length_; } |
90 | 59 |
91 private: | 60 private: |
92 // Returns the location that the data should be written at, or NULL if there | 61 // Returns the location that the data should be written at, or NULL if there |
93 // is not enough room. Call EndWrite with the returned offset and the given | 62 // is not enough room. Call EndWrite with the returned offset and the given |
94 // length to pad out for the next write. | 63 // length to pad out for the next write. |
95 char* BeginWrite(size_t length); | 64 char* BeginWrite(size_t length); |
96 | 65 |
97 char* buffer_; | 66 char* buffer_; |
98 size_t capacity_; // Allocation size of payload (or -1 if buffer is const). | 67 size_t capacity_; // Allocation size of payload (or -1 if buffer is const). |
99 size_t length_; // Current length of the buffer. | 68 size_t length_; // Current length of the buffer. |
100 }; | 69 }; |
101 | 70 |
102 } // namespace net | 71 } // namespace net |
103 | 72 |
104 #endif // NET_QUIC_QUIC_DATA_WRITER_H_ | 73 #endif // NET_QUIC_QUIC_DATA_WRITER_H_ |
OLD | NEW |