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
3 // found in the LICENSE file.
4
5 #include "net/tools/quic/quic_spdy_client_stream.h"
6
7 #include "net/spdy/spdy_framer.h"
8 #include "net/tools/quic/quic_client_session.h"
9 #include "net/tools/quic/spdy_utils.h"
10
11 using base::StringPiece;
12 using std::string;
13
14 namespace net {
15 namespace tools {
16
17 static const size_t kHeaderBufInitialSize = 4096;
18
QuicSpdyClientStream(QuicStreamId id,QuicClientSession * session)19 QuicSpdyClientStream::QuicSpdyClientStream(QuicStreamId id,
20 QuicClientSession* session)
21 : QuicDataStream(id, session),
22 read_buf_(new GrowableIOBuffer()),
23 response_headers_received_(false),
24 header_bytes_read_(0),
25 header_bytes_written_(0) {
26 }
27
~QuicSpdyClientStream()28 QuicSpdyClientStream::~QuicSpdyClientStream() {
29 }
30
OnStreamFrame(const QuicStreamFrame & frame)31 void QuicSpdyClientStream::OnStreamFrame(const QuicStreamFrame& frame) {
32 if (!write_side_closed()) {
33 DVLOG(1) << "Got a response before the request was complete. "
34 << "Aborting request.";
35 CloseWriteSide();
36 }
37 QuicDataStream::OnStreamFrame(frame);
38 }
39
OnStreamHeadersComplete(bool fin,size_t frame_len)40 void QuicSpdyClientStream::OnStreamHeadersComplete(bool fin,
41 size_t frame_len) {
42 header_bytes_read_ = frame_len;
43 QuicDataStream::OnStreamHeadersComplete(fin, frame_len);
44 }
45
ProcessData(const char * data,uint32 data_len)46 uint32 QuicSpdyClientStream::ProcessData(const char* data,
47 uint32 data_len) {
48 int total_bytes_processed = 0;
49
50 // Are we still reading the response headers.
51 if (!response_headers_received_) {
52 // Grow the read buffer if necessary.
53 if (read_buf_->RemainingCapacity() < (int)data_len) {
54 read_buf_->SetCapacity(read_buf_->capacity() + kHeaderBufInitialSize);
55 }
56 memcpy(read_buf_->data(), data, data_len);
57 read_buf_->set_offset(read_buf_->offset() + data_len);
58 ParseResponseHeaders();
59 } else {
60 data_.append(data + total_bytes_processed,
61 data_len - total_bytes_processed);
62 }
63 return data_len;
64 }
65
OnFinRead()66 void QuicSpdyClientStream::OnFinRead() {
67 ReliableQuicStream::OnFinRead();
68 if (!response_headers_received_) {
69 Reset(QUIC_BAD_APPLICATION_PAYLOAD);
70 } else if ((headers().content_length_status() ==
71 BalsaHeadersEnums::VALID_CONTENT_LENGTH) &&
72 data_.size() != headers().content_length()) {
73 Reset(QUIC_BAD_APPLICATION_PAYLOAD);
74 }
75 }
76
SendRequest(const BalsaHeaders & headers,StringPiece body,bool fin)77 ssize_t QuicSpdyClientStream::SendRequest(const BalsaHeaders& headers,
78 StringPiece body,
79 bool fin) {
80 SpdyHeaderBlock header_block =
81 SpdyUtils::RequestHeadersToSpdyHeaders(headers);
82
83 bool send_fin_with_headers = fin && body.empty();
84 size_t bytes_sent = body.size();
85 header_bytes_written_ = WriteHeaders(
86 header_block, send_fin_with_headers, NULL);
87 bytes_sent += header_bytes_written_;
88
89 if (!body.empty()) {
90 WriteOrBufferData(body, fin, NULL);
91 }
92
93 return bytes_sent;
94 }
95
ParseResponseHeaders()96 int QuicSpdyClientStream::ParseResponseHeaders() {
97 size_t read_buf_len = static_cast<size_t>(read_buf_->offset());
98 SpdyFramer framer(SPDY3);
99 SpdyHeaderBlock headers;
100 char* data = read_buf_->StartOfBuffer();
101 size_t len = framer.ParseHeaderBlockInBuffer(data, read_buf_->offset(),
102 &headers);
103 if (len == 0) {
104 return -1;
105 }
106
107 if (!SpdyUtils::FillBalsaResponseHeaders(headers, &headers_)) {
108 Reset(QUIC_BAD_APPLICATION_PAYLOAD);
109 return -1;
110 }
111 response_headers_received_ = true;
112
113 size_t delta = read_buf_len - len;
114 if (delta > 0) {
115 data_.append(data + len, delta);
116 }
117
118 return len;
119 }
120
121 // Sends body data to the server and returns the number of bytes sent.
SendBody(const string & data,bool fin)122 void QuicSpdyClientStream::SendBody(const string& data, bool fin) {
123 WriteOrBufferData(data, fin, NULL);
124 }
125
126 } // namespace tools
127 } // namespace net
128