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
127
128
129
130
131
132
133
134
135
136
137
138
|
// 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 CHROME_BROWSER_EXTENSIONS_API_GCD_PRIVATE_PRIVET_V3_SESSION_H_
#define CHROME_BROWSER_EXTENSIONS_API_GCD_PRIVATE_PRIVET_V3_SESSION_H_
#include <stdint.h>
#include <memory>
#include <string>
#include "base/callback.h"
#include "base/gtest_prod_util.h"
#include "base/macros.h"
#include "base/memory/weak_ptr.h"
#include "chrome/common/extensions/api/gcd_private.h"
#include "net/url_request/url_fetcher.h"
namespace base {
class DictionaryValue;
}
namespace crypto {
class P224EncryptedKeyExchange;
}
namespace extensions {
class PrivetV3ContextGetter;
// Manages secure communication between browser and local Privet device.
class PrivetV3Session {
private:
class FetcherDelegate;
public:
typedef extensions::api::gcd_private::PairingType PairingType;
typedef extensions::api::gcd_private::Status Result;
typedef base::Callback<
void(Result result, const base::DictionaryValue& response)> InitCallback;
typedef base::Callback<void(Result result)> ResultCallback;
typedef base::Callback<void(Result result,
const base::DictionaryValue& response)>
MessageCallback;
PrivetV3Session(const scoped_refptr<PrivetV3ContextGetter>& context_getter,
const net::HostPortPair& host_port);
~PrivetV3Session();
// Initializes session. Queries /privet/info and returns supported pairing
// types in callback.
void Init(const InitCallback& callback);
// Starts pairing by calling /privet/v3/pairing/start.
void StartPairing(PairingType pairing_type, const ResultCallback& callback);
// Confirms pairing code by calling /privet/v3/pairing/confirm.
// Calls /privet/v3/pairing/auth after pairing.
void ConfirmCode(const std::string& code, const ResultCallback& callback);
// TODO(vitalybuka): Make HTTPS request to device with certificate validation.
void SendMessage(const std::string& api,
const base::DictionaryValue& input,
const MessageCallback& callback);
private:
friend class PrivetV3SessionTest;
FRIEND_TEST_ALL_PREFIXES(PrivetV3SessionTest, Pairing);
FRIEND_TEST_ALL_PREFIXES(PrivetV3SessionTest, SendMessage);
void OnInfoDone(const InitCallback& callback,
Result result,
const base::DictionaryValue& response);
void OnPairingStartDone(const ResultCallback& callback,
Result result,
const base::DictionaryValue& response);
void OnPairingConfirmDone(const ResultCallback& callback,
Result result,
const base::DictionaryValue& response);
void OnPairedHostAddedToContext(const std::string& auth_code,
const ResultCallback& callback);
void OnAuthenticateDone(const ResultCallback& callback,
Result result,
const base::DictionaryValue& response);
void RunCallback(const base::Closure& callback);
void StartGetRequest(const std::string& api, const MessageCallback& callback);
void StartPostRequest(const std::string& api,
const base::DictionaryValue& input,
const MessageCallback& callback);
void SwitchToHttps();
GURL CreatePrivetURL(const std::string& path) const;
net::URLFetcher* CreateFetcher(const std::string& api,
net::URLFetcher::RequestType request_type,
const MessageCallback& callback);
void DeleteFetcher(const FetcherDelegate* fetcher);
void Cancel();
// Endpoint for the current session.
net::HostPortPair host_port_;
// Provides context for client_.
scoped_refptr<PrivetV3ContextGetter> context_getter_;
// Current authentication token.
std::string privet_auth_token_;
// ID of the session received from pairing/start.
std::string session_id_;
// Device commitment received from pairing/start.
std::string commitment_;
// Key exchange algorithm for pairing.
std::unique_ptr<crypto::P224EncryptedKeyExchange> spake_;
// HTTPS port of the device.
uint16_t https_port_ = 0;
// If true, HTTPS will be used.
bool use_https_ = false;
// List of fetches to cancel when session is destroyed.
std::vector<std::unique_ptr<FetcherDelegate>> fetchers_;
// Intercepts POST requests. Used by tests only.
base::Callback<void(const base::DictionaryValue&)> on_post_data_;
base::WeakPtrFactory<PrivetV3Session> weak_ptr_factory_;
DISALLOW_COPY_AND_ASSIGN(PrivetV3Session);
};
} // namespace extensions
#endif // CHROME_BROWSER_EXTENSIONS_API_GCD_PRIVATE_PRIVET_V3_SESSION_H_
|
