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
117
118
119
|
// 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_CORE_CRYPTO_CRYPTO_FRAMER_H_
#define NET_QUIC_CORE_CRYPTO_CRYPTO_FRAMER_H_
#include <cstddef>
#include <cstdint>
#include <memory>
#include <utility>
#include <vector>
#include "net/quic/core/crypto/crypto_handshake_message.h"
#include "net/quic/core/crypto/crypto_message_parser.h"
#include "net/quic/core/quic_packets.h"
#include "net/quic/platform/api/quic_export.h"
namespace net {
class CryptoFramer;
class QuicData;
class QuicDataWriter;
class QUIC_EXPORT_PRIVATE CryptoFramerVisitorInterface {
public:
virtual ~CryptoFramerVisitorInterface() {}
// Called if an error is detected.
virtual void OnError(CryptoFramer* framer) = 0;
// Called when a complete handshake message has been parsed.
virtual void OnHandshakeMessage(const CryptoHandshakeMessage& message) = 0;
};
// A class for framing the crypto messages that are exchanged in a QUIC
// session.
class QUIC_EXPORT_PRIVATE CryptoFramer : public CryptoMessageParser {
public:
CryptoFramer();
~CryptoFramer() override;
// ParseMessage parses exactly one message from the given QuicStringPiece. If
// there is an error, the message is truncated, or the message has trailing
// garbage then nullptr will be returned.
static std::unique_ptr<CryptoHandshakeMessage> ParseMessage(
QuicStringPiece in,
Perspective perspective);
// Set callbacks to be called from the framer. A visitor must be set, or
// else the framer will crash. It is acceptable for the visitor to do
// nothing. If this is called multiple times, only the last visitor
// will be used. |visitor| will be owned by the framer.
void set_visitor(CryptoFramerVisitorInterface* visitor) {
visitor_ = visitor;
}
QuicErrorCode error() const override;
const std::string& error_detail() const override;
// Processes input data, which must be delivered in order. Returns
// false if there was an error, and true otherwise.
bool ProcessInput(QuicStringPiece input, Perspective perspective) override;
// Returns the number of bytes of buffered input data remaining to be
// parsed.
size_t InputBytesRemaining() const override;
// Returns a new QuicData owned by the caller that contains a serialized
// |message|, or nullptr if there was an error.
static QuicData* ConstructHandshakeMessage(
const CryptoHandshakeMessage& message,
Perspective perspective);
private:
// Clears per-message state. Does not clear the visitor.
void Clear();
// Process does does the work of |ProcessInput|, but returns an error code,
// doesn't set error_ and doesn't call |visitor_->OnError()|.
QuicErrorCode Process(QuicStringPiece input, Perspective perspective);
static bool WritePadTag(QuicDataWriter* writer,
size_t pad_length,
uint32_t* end_offset);
// Represents the current state of the parsing state machine.
enum CryptoFramerState {
STATE_READING_TAG,
STATE_READING_NUM_ENTRIES,
STATE_READING_TAGS_AND_LENGTHS,
STATE_READING_VALUES
};
// Visitor to invoke when messages are parsed.
CryptoFramerVisitorInterface* visitor_;
// Last error.
QuicErrorCode error_;
// Remaining unparsed data.
std::string buffer_;
// Current state of the parsing.
CryptoFramerState state_;
// The message currently being parsed.
CryptoHandshakeMessage message_;
// The issue which caused |error_|
std::string error_detail_;
// Number of entires in the message currently being parsed.
uint16_t num_entries_;
// tags_and_lengths_ contains the tags that are currently being parsed and
// their lengths.
std::vector<std::pair<QuicTag, size_t>> tags_and_lengths_;
// Cumulative length of all values in the message currently being parsed.
size_t values_len_;
};
} // namespace net
#endif // NET_QUIC_CORE_CRYPTO_CRYPTO_FRAMER_H_
|
