blob: 9167e6f78ca099bb5773067488d069bd643a801a (
plain)
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
|
// 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 "net/quic/quic_crypto_stream.h"
#include <string>
#include "base/strings/string_piece.h"
#include "net/quic/crypto/crypto_handshake.h"
#include "net/quic/quic_connection.h"
#include "net/quic/quic_session.h"
using std::string;
using base::StringPiece;
namespace net {
QuicCryptoStream::QuicCryptoStream(QuicSession* session)
: ReliableQuicStream(kCryptoStreamId, session),
handshake_complete_(false) {
crypto_framer_.set_visitor(this);
}
void QuicCryptoStream::OnError(CryptoFramer* framer) {
session()->ConnectionClose(framer->error(), false);
}
uint32 QuicCryptoStream::ProcessData(const char* data,
uint32 data_len) {
// Do not process handshake messages after the handshake is complete.
if (handshake_complete()) {
CloseConnection(QUIC_CRYPTO_MESSAGE_AFTER_HANDSHAKE_COMPLETE);
return 0;
}
if (!crypto_framer_.ProcessInput(StringPiece(data, data_len))) {
CloseConnection(crypto_framer_.error());
return 0;
}
return data_len;
}
void QuicCryptoStream::CloseConnection(QuicErrorCode error) {
session()->connection()->SendConnectionClose(error);
}
void QuicCryptoStream::CloseConnectionWithDetails(QuicErrorCode error,
const string& details) {
session()->connection()->SendConnectionCloseWithDetails(error, details);
}
void QuicCryptoStream::SetHandshakeComplete(QuicErrorCode error) {
handshake_complete_ = true;
session()->OnCryptoHandshakeComplete(error);
}
void QuicCryptoStream::SendHandshakeMessage(
const CryptoHandshakeMessage& message) {
const QuicData& data = message.GetSerialized();
// TODO(wtc): check the return value.
WriteData(string(data.data(), data.length()), false);
}
} // namespace net
|
