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
120
121
122
123
124
125
126
|
// Copyright 2014 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 COMPONENTS_PROXIMITY_AUTH_MESSENGER_IMPL_H
#define COMPONENTS_PROXIMITY_AUTH_MESSENGER_IMPL_H
#include <deque>
#include <memory>
#include "base/macros.h"
#include "base/memory/weak_ptr.h"
#include "base/observer_list.h"
#include "components/cryptauth/connection.h"
#include "components/cryptauth/connection_observer.h"
#include "components/proximity_auth/messenger.h"
namespace base {
class DictionaryValue;
}
namespace proximity_auth {
class SecureContext;
// Concrete implementation of the Messenger interface.
class MessengerImpl : public Messenger, public cryptauth::ConnectionObserver {
public:
// Constructs a messenger that sends and receives messages over the given
// |connection|, using the |secure_context| to encrypt and decrypt the
// messages. The |connection| must be connected. The messenger begins
// observing messages as soon as it is constructed.
MessengerImpl(std::unique_ptr<cryptauth::Connection> connection,
std::unique_ptr<SecureContext> secure_context);
~MessengerImpl() override;
// Messenger:
void AddObserver(MessengerObserver* observer) override;
void RemoveObserver(MessengerObserver* observer) override;
bool SupportsSignIn() const override;
void DispatchUnlockEvent() override;
void RequestDecryption(const std::string& challenge) override;
void RequestUnlock() override;
SecureContext* GetSecureContext() const override;
// Exposed for testing.
cryptauth::Connection* connection() { return connection_.get(); }
private:
// Internal data structure to represent a pending message that either hasn't
// been sent yet or is waiting for a response from the remote device.
struct PendingMessage {
PendingMessage();
PendingMessage(const base::DictionaryValue& message);
PendingMessage(const std::string& message);
~PendingMessage();
// The message, serialized as JSON.
const std::string json_message;
// The message type. This is possible to parse from the |json_message|; it's
// stored redundantly for convenience.
const std::string type;
};
// Pops the first of the |queued_messages_| and sends it to the remote device.
void ProcessMessageQueue();
// Called when the message is encoded so it can be sent over the connection.
void OnMessageEncoded(const std::string& encoded_message);
// Called when the message is decoded so it can be parsed.
void OnMessageDecoded(const std::string& decoded_message);
// Handles an incoming "status_update" |message|, parsing and notifying
// observers of the content.
void HandleRemoteStatusUpdateMessage(const base::DictionaryValue& message);
// Handles an incoming "decrypt_response" message, parsing and notifying
// observers of the decrypted content.
void HandleDecryptResponseMessage(const base::DictionaryValue& message);
// Handles an incoming "unlock_response" message, notifying observers of the
// response.
void HandleUnlockResponseMessage(const base::DictionaryValue& message);
// For iOS, we need to poll the phone every few seconds to keep the app alive
// in the background. This function starts the poll loop.
void PollScreenStateForIOS();
// cryptauth::ConnectionObserver:
void OnConnectionStatusChanged(
cryptauth::Connection* connection,
cryptauth::Connection::Status old_status,
cryptauth::Connection::Status new_status) override;
void OnMessageReceived(const cryptauth::Connection& connection,
const cryptauth::WireMessage& wire_message) override;
void OnSendCompleted(const cryptauth::Connection& connection,
const cryptauth::WireMessage& wire_message,
bool success) override;
// The connection used to send and receive events and status updates.
std::unique_ptr<cryptauth::Connection> connection_;
// Used to encrypt and decrypt payloads sent and received over the
// |connection_|.
std::unique_ptr<SecureContext> secure_context_;
// The registered observers of |this_| messenger.
base::ObserverList<MessengerObserver> observers_;
// Queue of messages to send to the remote device.
std::deque<PendingMessage> queued_messages_;
// The current message being sent or waiting on the remote device for a
// response. Null if there is no message currently in this state.
std::unique_ptr<PendingMessage> pending_message_;
base::WeakPtrFactory<MessengerImpl> weak_ptr_factory_;
DISALLOW_COPY_AND_ASSIGN(MessengerImpl);
};
} // namespace proximity_auth
#endif // COMPONENTS_PROXIMITY_AUTH_MESSENGER_IMPL_H
|
