1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
|
// 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.
#include <limits>
#include "net/spdy/spdy_frame_builder.h"
#include "net/spdy/spdy_protocol.h"
namespace net {
namespace {
// Creates a FlagsAndLength.
FlagsAndLength CreateFlagsAndLength(SpdyControlFlags flags, size_t length) {
DCHECK_EQ(0u, length & ~static_cast<size_t>(kLengthMask));
FlagsAndLength flags_length;
flags_length.length_ = htonl(static_cast<uint32>(length));
DCHECK_EQ(0, flags & ~kControlFlagsMask);
flags_length.flags_[0] = flags;
return flags_length;
}
} // namespace
SpdyFrameBuilder::SpdyFrameBuilder(size_t size)
: buffer_(new char[size]),
capacity_(size),
length_(0) {
}
SpdyFrameBuilder::SpdyFrameBuilder(SpdyControlType type,
SpdyControlFlags flags,
int spdy_version,
size_t size)
: buffer_(new char[size]),
capacity_(size),
length_(0) {
FlagsAndLength flags_length = CreateFlagsAndLength(
flags, size - SpdyFrame::kHeaderSize);
WriteUInt16(kControlFlagMask | spdy_version);
WriteUInt16(type);
WriteBytes(&flags_length, sizeof(flags_length));
}
SpdyFrameBuilder::SpdyFrameBuilder(SpdyStreamId stream_id,
SpdyDataFlags flags,
size_t size)
: buffer_(new char[size]),
capacity_(size),
length_(0) {
DCHECK_EQ(0u, stream_id & ~kStreamIdMask);
WriteUInt32(stream_id);
size_t length = size - SpdyFrame::kHeaderSize;
DCHECK_EQ(0u, length & ~static_cast<size_t>(kLengthMask));
FlagsAndLength flags_length;
flags_length.length_ = htonl(length);
DCHECK_EQ(0, flags & ~kDataFlagsMask);
flags_length.flags_[0] = flags;
WriteBytes(&flags_length, sizeof(flags_length));
}
SpdyFrameBuilder::~SpdyFrameBuilder() {
}
char* SpdyFrameBuilder::BeginWrite(size_t length) {
size_t offset = length_;
size_t needed_size = length_ + length;
if (needed_size > capacity_)
return NULL;
#ifdef ARCH_CPU_64_BITS
DCHECK_LE(length, std::numeric_limits<uint32>::max());
#endif
return buffer_.get() + offset;
}
void SpdyFrameBuilder::EndWrite(char* dest, int length) {
}
bool SpdyFrameBuilder::WriteBytes(const void* data, uint32 data_len) {
if (data_len > kLengthMask) {
return false;
}
char* dest = BeginWrite(data_len);
if (!dest)
return false;
memcpy(dest, data, data_len);
EndWrite(dest, data_len);
length_ += data_len;
return true;
}
bool SpdyFrameBuilder::WriteString(const std::string& value) {
if (value.size() > 0xffff)
return false;
if (!WriteUInt16(static_cast<int>(value.size())))
return false;
return WriteBytes(value.data(), static_cast<uint16>(value.size()));
}
bool SpdyFrameBuilder::WriteStringPiece32(const base::StringPiece& value) {
if (!WriteUInt32(value.size())) {
return false;
}
return WriteBytes(value.data(), value.size());
}
} // namespace net
|
