| Index: net/quic/crypto/crypto_framer.h
|
| diff --git a/net/quic/crypto/crypto_framer.h b/net/quic/crypto/crypto_framer.h
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..e52b26458cd31986ea0a339987ef647ffb02f791
|
| --- /dev/null
|
| +++ b/net/quic/crypto/crypto_framer.h
|
| @@ -0,0 +1,106 @@
|
| +// 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_CRYPTO_CRYPTO_FRAMER_H_
|
| +#define NET_QUIC_CRYPTO_CRYPTO_FRAMER_H_
|
| +
|
| +#include <map>
|
| +#include <vector>
|
| +
|
| +#include "base/basictypes.h"
|
| +#include "base/logging.h"
|
| +#include "base/memory/scoped_ptr.h"
|
| +#include "base/string_piece.h"
|
| +#include "net/base/net_export.h"
|
| +#include "net/quic/crypto/crypto_protocol.h"
|
| +#include "net/quic/quic_protocol.h"
|
| +
|
| +namespace net {
|
| +
|
| +class CryptoFramer;
|
| +class QuicDataReader;
|
| +class QuicData;
|
| +
|
| +class NET_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 message that are exchanged in a QUIC session.
|
| +class NET_EXPORT_PRIVATE CryptoFramer {
|
| + public:
|
| + CryptoFramer();
|
| +
|
| + virtual ~CryptoFramer();
|
| +
|
| + // Set callbacks to be called from the framer. A visitor must be set, or
|
| + // else the framer will likely 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 {
|
| + return error_;
|
| + }
|
| +
|
| + // Processes input data, which must be delievered in order. Returns
|
| + // false if there was an error, and true otherwise.
|
| + bool ProcessInput(base::StringPiece input);
|
| +
|
| + // Serializes |message| into |packet|. Returns false if there was an
|
| + // error, and true otherwise.
|
| + bool ConstructHandshakeMessage(const CryptoHandshakeMessage& message,
|
| + QuicData** packet);
|
| +
|
| + private:
|
| + // Clears per-message state. Does not clear the visitor.
|
| + void Clear();
|
| +
|
| + void set_error(QuicErrorCode error) {
|
| + error_ = error;
|
| + }
|
| +
|
| + enum CryptoFramerState {
|
| + STATE_READING_TAG,
|
| + STATE_READING_NUM_ENTRIES,
|
| + STATE_READING_KEY_TAGS,
|
| + STATE_READING_LENGTHS,
|
| + STATE_READING_VALUES
|
| + };
|
| +
|
| + // Visitor to send invoke when messages are parsed.
|
| + CryptoFramerVisitorInterface* visitor_;
|
| + // Last error.
|
| + QuicErrorCode error_;
|
| + // Remaining unparsed data.
|
| + std::string buffer_;
|
| + // Current state of the parsing.
|
| + CryptoFramerState state_;
|
| + // Tag of the message currently being parsed.
|
| + CryptoTag message_tag_;
|
| + // Number of entires in the message currently being parsed.
|
| + uint16 num_entries_;
|
| + // Vector of tags in the message currently being parsed.
|
| + std::vector<CryptoTag> tags_;
|
| + // Length of the data associated with each tag in the message currently
|
| + // being parsed.
|
| + std::map<CryptoTag, size_t> tag_length_map_;
|
| + // Length of the data associated with each tag in the message currently
|
| + // being parsed.
|
| + CryptoTagValueMap tag_value_map_;
|
| + // Cumulative length of all values in the message currently being parsed.
|
| + size_t values_len_;
|
| +};
|
| +
|
| +} // namespace net
|
| +
|
| +#endif // NET_QUIC_CRYPTO_CRYPTO_FRAMER_H_
|
|
|