| Index: net/quic/quic_data_writer.h
|
| diff --git a/net/quic/quic_data_writer.h b/net/quic/quic_data_writer.h
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..2db5eca16c4d18fd1dc2b9b80352884bdd77483a
|
| --- /dev/null
|
| +++ b/net/quic/quic_data_writer.h
|
| @@ -0,0 +1,94 @@
|
| +// Copyright (c) 2012 The Chromium Authors. All rights reserved.
|
| +// Use of this source code is governed by a BSD-style license that can be
|
| +// found in the LICENSE file.
|
| +
|
| +#ifndef NET_QUIC_QUIC_DATA_WRITER_H_
|
| +#define NET_QUIC_QUIC_DATA_WRITER_H_
|
| +
|
| +#include <string>
|
| +
|
| +#include "base/basictypes.h"
|
| +#include "base/logging.h"
|
| +#include "base/string_piece.h"
|
| +#include "net/base/net_export.h"
|
| +#include "net/quic/quic_protocol.h"
|
| +#include "net/quic/uint128.h"
|
| +
|
| +namespace net {
|
| +
|
| +// This class provides facilities for packing QUIC data.
|
| +//
|
| +// The QuicDataWriter supports appending primitive values (int, string, etc)
|
| +// to a frame instance. The QuicDataWriter grows its internal memory buffer
|
| +// dynamically to hold the sequence of primitive values. The internal memory
|
| +// buffer is exposed as the "data" of the QuicDataWriter.
|
| +class NET_EXPORT_PRIVATE QuicDataWriter {
|
| + public:
|
| + explicit QuicDataWriter(size_t length);
|
| + ~QuicDataWriter();
|
| +
|
| + // Returns the size of the QuicDataWriter's data.
|
| + size_t length() const { return length_; }
|
| +
|
| + // Takes the buffer from the QuicDataWriter.
|
| + // TODO(rch): move non-trivial methods into .cc file.
|
| + char* take() {
|
| + char* rv = buffer_;
|
| + buffer_ = NULL;
|
| + capacity_ = 0;
|
| + length_ = 0;
|
| + return rv;
|
| + }
|
| +
|
| + // Methods for adding to the payload. These values are appended to the end
|
| + // of the QuicDataWriter payload. Note - binary integers are written in
|
| + // host byte order (little endian) not network byte order (big endian).
|
| + bool WriteUInt8(uint8 value) {
|
| + return WriteBytes(&value, sizeof(value));
|
| + }
|
| + bool WriteUInt16(uint16 value) {
|
| + return WriteBytes(&value, sizeof(value));
|
| + }
|
| + bool WriteUInt32(uint32 value) {
|
| + return WriteBytes(&value, sizeof(value));
|
| + }
|
| + bool WriteUInt48(uint64 value) {
|
| + uint32 hi = value >> 32;
|
| + uint32 lo = value & 0x00000000FFFFFFFF;
|
| + return WriteUInt32(lo) && WriteUInt16(hi);
|
| + }
|
| + bool WriteUInt64(uint64 value) {
|
| + return WriteBytes(&value, sizeof(value));
|
| + }
|
| + bool WriteUInt128(uint128 value) {
|
| + return WriteUInt64(value.lo) && WriteUInt64(value.hi);
|
| + }
|
| +
|
| + bool AdvancePointer(uint32 len);
|
| +
|
| + bool WriteBytes(const void* data, uint32 data_len);
|
| +
|
| + static void WriteUint64ToBuffer(uint64 value, char* buffer);
|
| + static void WriteUint128ToBuffer(uint128 value, char* buffer);
|
| +
|
| + size_t capacity() const {
|
| + return capacity_;
|
| + }
|
| +
|
| + protected:
|
| + const char* end_of_payload() const { return buffer_ + length_; }
|
| +
|
| + private:
|
| + // Returns the location that the data should be written at, or NULL if there
|
| + // is not enough room. Call EndWrite with the returned offset and the given
|
| + // length to pad out for the next write.
|
| + char* BeginWrite(size_t length);
|
| +
|
| + char* buffer_;
|
| + size_t capacity_; // Allocation size of payload (or -1 if buffer is const).
|
| + size_t length_; // Current length of the buffer.
|
| +};
|
| +
|
| +} // namespace net
|
| +
|
| +#endif // NET_QUIC_QUIC_DATA_WRITER_H_
|
|
|