| OLD | NEW |
| (Empty) |
| 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 #ifndef NET_QUIC_CRYPTO_CRYPTO_FRAMER_H_ | |
| 6 #define NET_QUIC_CRYPTO_CRYPTO_FRAMER_H_ | |
| 7 | |
| 8 #include <utility> | |
| 9 #include <vector> | |
| 10 | |
| 11 #include "base/basictypes.h" | |
| 12 #include "base/logging.h" | |
| 13 #include "base/strings/string_piece.h" | |
| 14 #include "net/base/net_export.h" | |
| 15 #include "net/quic/crypto/crypto_handshake_message.h" | |
| 16 #include "net/quic/quic_protocol.h" | |
| 17 | |
| 18 namespace net { | |
| 19 | |
| 20 class CryptoFramer; | |
| 21 class QuicData; | |
| 22 class QuicDataReader; | |
| 23 class QuicDataWriter; | |
| 24 | |
| 25 class NET_EXPORT_PRIVATE CryptoFramerVisitorInterface { | |
| 26 public: | |
| 27 virtual ~CryptoFramerVisitorInterface() {} | |
| 28 | |
| 29 // Called if an error is detected. | |
| 30 virtual void OnError(CryptoFramer* framer) = 0; | |
| 31 | |
| 32 // Called when a complete handshake message has been parsed. | |
| 33 virtual void OnHandshakeMessage( | |
| 34 const CryptoHandshakeMessage& message) = 0; | |
| 35 }; | |
| 36 | |
| 37 // A class for framing the crypto messages that are exchanged in a QUIC | |
| 38 // session. | |
| 39 class NET_EXPORT_PRIVATE CryptoFramer { | |
| 40 public: | |
| 41 CryptoFramer(); | |
| 42 | |
| 43 virtual ~CryptoFramer(); | |
| 44 | |
| 45 // ParseMessage parses exactly one message from the given StringPiece. If | |
| 46 // there is an error, the message is truncated, or the message has trailing | |
| 47 // garbage then nullptr will be returned. | |
| 48 static CryptoHandshakeMessage* ParseMessage(base::StringPiece in); | |
| 49 | |
| 50 // Set callbacks to be called from the framer. A visitor must be set, or | |
| 51 // else the framer will crash. It is acceptable for the visitor to do | |
| 52 // nothing. If this is called multiple times, only the last visitor | |
| 53 // will be used. |visitor| will be owned by the framer. | |
| 54 void set_visitor(CryptoFramerVisitorInterface* visitor) { | |
| 55 visitor_ = visitor; | |
| 56 } | |
| 57 | |
| 58 QuicErrorCode error() const { return error_; } | |
| 59 | |
| 60 // Processes input data, which must be delivered in order. Returns | |
| 61 // false if there was an error, and true otherwise. | |
| 62 bool ProcessInput(base::StringPiece input); | |
| 63 | |
| 64 // Returns the number of bytes of buffered input data remaining to be | |
| 65 // parsed. | |
| 66 size_t InputBytesRemaining() const { return buffer_.length(); } | |
| 67 | |
| 68 // Returns a new QuicData owned by the caller that contains a serialized | |
| 69 // |message|, or nullptr if there was an error. | |
| 70 static QuicData* ConstructHandshakeMessage( | |
| 71 const CryptoHandshakeMessage& message); | |
| 72 | |
| 73 private: | |
| 74 // Clears per-message state. Does not clear the visitor. | |
| 75 void Clear(); | |
| 76 | |
| 77 // Process does does the work of |ProcessInput|, but returns an error code, | |
| 78 // doesn't set error_ and doesn't call |visitor_->OnError()|. | |
| 79 QuicErrorCode Process(base::StringPiece input); | |
| 80 | |
| 81 static bool WritePadTag(QuicDataWriter* writer, | |
| 82 size_t pad_length, | |
| 83 uint32* end_offset); | |
| 84 | |
| 85 // Represents the current state of the parsing state machine. | |
| 86 enum CryptoFramerState { | |
| 87 STATE_READING_TAG, | |
| 88 STATE_READING_NUM_ENTRIES, | |
| 89 STATE_READING_TAGS_AND_LENGTHS, | |
| 90 STATE_READING_VALUES | |
| 91 }; | |
| 92 | |
| 93 // Visitor to invoke when messages are parsed. | |
| 94 CryptoFramerVisitorInterface* visitor_; | |
| 95 // Last error. | |
| 96 QuicErrorCode error_; | |
| 97 // Remaining unparsed data. | |
| 98 std::string buffer_; | |
| 99 // Current state of the parsing. | |
| 100 CryptoFramerState state_; | |
| 101 // The message currently being parsed. | |
| 102 CryptoHandshakeMessage message_; | |
| 103 // Number of entires in the message currently being parsed. | |
| 104 uint16 num_entries_; | |
| 105 // tags_and_lengths_ contains the tags that are currently being parsed and | |
| 106 // their lengths. | |
| 107 std::vector<std::pair<QuicTag, size_t> > tags_and_lengths_; | |
| 108 // Cumulative length of all values in the message currently being parsed. | |
| 109 size_t values_len_; | |
| 110 }; | |
| 111 | |
| 112 } // namespace net | |
| 113 | |
| 114 #endif // NET_QUIC_CRYPTO_CRYPTO_FRAMER_H_ | |
| OLD | NEW |