OLD | NEW |
1 // Copyright (c) 2016 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2016 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_CORE_FRAMES_QUIC_STREAM_FRAME_H_ | 5 #ifndef NET_QUIC_CORE_FRAMES_QUIC_STREAM_FRAME_H_ |
6 #define NET_QUIC_CORE_FRAMES_QUIC_STREAM_FRAME_H_ | 6 #define NET_QUIC_CORE_FRAMES_QUIC_STREAM_FRAME_H_ |
7 | 7 |
8 #include <memory> | 8 #include <memory> |
9 | 9 |
10 #include "base/strings/string_piece.h" | 10 #include "base/strings/string_piece.h" |
11 #include "net/quic/core/quic_buffer_allocator.h" | 11 #include "net/quic/core/quic_buffer_allocator.h" |
12 #include "net/quic/core/quic_types.h" | 12 #include "net/quic/core/quic_types.h" |
| 13 #include "net/quic/platform/api/quic_export.h" |
13 | 14 |
14 namespace net { | 15 namespace net { |
15 | 16 |
16 // Deleter for stream buffers. Copyable to support platforms where the deleter | 17 // Deleter for stream buffers. Copyable to support platforms where the deleter |
17 // of a unique_ptr must be copyable. Otherwise it would be nice for this to be | 18 // of a unique_ptr must be copyable. Otherwise it would be nice for this to be |
18 // move-only. | 19 // move-only. |
19 class NET_EXPORT_PRIVATE StreamBufferDeleter { | 20 class QUIC_EXPORT_PRIVATE StreamBufferDeleter { |
20 public: | 21 public: |
21 StreamBufferDeleter() : allocator_(nullptr) {} | 22 StreamBufferDeleter() : allocator_(nullptr) {} |
22 explicit StreamBufferDeleter(QuicBufferAllocator* allocator) | 23 explicit StreamBufferDeleter(QuicBufferAllocator* allocator) |
23 : allocator_(allocator) {} | 24 : allocator_(allocator) {} |
24 | 25 |
25 // Deletes |buffer| using |allocator_|. | 26 // Deletes |buffer| using |allocator_|. |
26 void operator()(char* buffer) const; | 27 void operator()(char* buffer) const; |
27 | 28 |
28 private: | 29 private: |
29 // Not owned; must be valid so long as the buffer stored in the unique_ptr | 30 // Not owned; must be valid so long as the buffer stored in the unique_ptr |
30 // that owns |this| is valid. | 31 // that owns |this| is valid. |
31 QuicBufferAllocator* allocator_; | 32 QuicBufferAllocator* allocator_; |
32 }; | 33 }; |
33 | 34 |
34 using UniqueStreamBuffer = std::unique_ptr<char[], StreamBufferDeleter>; | 35 using UniqueStreamBuffer = std::unique_ptr<char[], StreamBufferDeleter>; |
35 | 36 |
36 // Allocates memory of size |size| using |allocator| for a QUIC stream buffer. | 37 // Allocates memory of size |size| using |allocator| for a QUIC stream buffer. |
37 NET_EXPORT_PRIVATE UniqueStreamBuffer | 38 QUIC_EXPORT_PRIVATE UniqueStreamBuffer |
38 NewStreamBuffer(QuicBufferAllocator* allocator, size_t size); | 39 NewStreamBuffer(QuicBufferAllocator* allocator, size_t size); |
39 | 40 |
40 struct NET_EXPORT_PRIVATE QuicStreamFrame { | 41 struct QUIC_EXPORT_PRIVATE QuicStreamFrame { |
41 QuicStreamFrame(); | 42 QuicStreamFrame(); |
42 QuicStreamFrame(QuicStreamId stream_id, | 43 QuicStreamFrame(QuicStreamId stream_id, |
43 bool fin, | 44 bool fin, |
44 QuicStreamOffset offset, | 45 QuicStreamOffset offset, |
45 base::StringPiece data); | 46 base::StringPiece data); |
46 QuicStreamFrame(QuicStreamId stream_id, | 47 QuicStreamFrame(QuicStreamId stream_id, |
47 bool fin, | 48 bool fin, |
48 QuicStreamOffset offset, | 49 QuicStreamOffset offset, |
49 QuicPacketLength data_length, | 50 QuicPacketLength data_length, |
50 UniqueStreamBuffer buffer); | 51 UniqueStreamBuffer buffer); |
51 ~QuicStreamFrame(); | 52 ~QuicStreamFrame(); |
52 | 53 |
53 friend NET_EXPORT_PRIVATE std::ostream& operator<<(std::ostream& os, | 54 friend QUIC_EXPORT_PRIVATE std::ostream& operator<<(std::ostream& os, |
54 const QuicStreamFrame& s); | 55 const QuicStreamFrame& s); |
55 | 56 |
56 QuicStreamId stream_id; | 57 QuicStreamId stream_id; |
57 bool fin; | 58 bool fin; |
58 QuicPacketLength data_length; | 59 QuicPacketLength data_length; |
59 const char* data_buffer; | 60 const char* data_buffer; |
60 QuicStreamOffset offset; // Location of this data in the stream. | 61 QuicStreamOffset offset; // Location of this data in the stream. |
61 // nullptr when the QuicStreamFrame is received, and non-null when sent. | 62 // nullptr when the QuicStreamFrame is received, and non-null when sent. |
62 UniqueStreamBuffer buffer; | 63 UniqueStreamBuffer buffer; |
63 | 64 |
64 private: | 65 private: |
65 QuicStreamFrame(QuicStreamId stream_id, | 66 QuicStreamFrame(QuicStreamId stream_id, |
66 bool fin, | 67 bool fin, |
67 QuicStreamOffset offset, | 68 QuicStreamOffset offset, |
68 const char* data_buffer, | 69 const char* data_buffer, |
69 QuicPacketLength data_length, | 70 QuicPacketLength data_length, |
70 UniqueStreamBuffer buffer); | 71 UniqueStreamBuffer buffer); |
71 | 72 |
72 DISALLOW_COPY_AND_ASSIGN(QuicStreamFrame); | 73 DISALLOW_COPY_AND_ASSIGN(QuicStreamFrame); |
73 }; | 74 }; |
74 static_assert(sizeof(QuicStreamFrame) <= 64, | 75 static_assert(sizeof(QuicStreamFrame) <= 64, |
75 "Keep the QuicStreamFrame size to a cacheline."); | 76 "Keep the QuicStreamFrame size to a cacheline."); |
76 | 77 |
77 } // namespace net | 78 } // namespace net |
78 | 79 |
79 #endif // NET_QUIC_CORE_FRAMES_QUIC_STREAM_FRAME_H_ | 80 #endif // NET_QUIC_CORE_FRAMES_QUIC_STREAM_FRAME_H_ |
OLD | NEW |