Home | History | Annotate | Download | only in quic 
      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 
     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 
     28 QuicSpdyClientStream::~QuicSpdyClientStream() {
     29 }
     30 
     31 bool 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   return QuicDataStream::OnStreamFrame(frame);
     38 }
     39 
     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 
     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 
     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 
     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 
     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.
    122 void QuicSpdyClientStream::SendBody(const string& data, bool fin) {
    123   WriteOrBufferData(data, fin, NULL);
    124 }
    125 
    126 }  // namespace tools
    127 }  // namespace net
    128