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
|
// Copyright 2015 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef CHROMEOS_ASH_COMPONENTS_MULTIDEVICE_FAKE_SECURE_MESSAGE_DELEGATE_H_
#define CHROMEOS_ASH_COMPONENTS_MULTIDEVICE_FAKE_SECURE_MESSAGE_DELEGATE_H_
#include <memory>
#include "base/memory/raw_ptr.h"
#include "chromeos/ash/components/multidevice/secure_message_delegate.h"
#include "chromeos/ash/components/multidevice/secure_message_delegate_impl.h"
namespace ash::multidevice {
// Fake implementation of SecureMessageDelegate used in tests.
// For clarity in tests, all functions in this delegate will invoke their
// callback with the result before returning.
class FakeSecureMessageDelegate : public SecureMessageDelegate {
public:
FakeSecureMessageDelegate();
FakeSecureMessageDelegate(const FakeSecureMessageDelegate&) = delete;
FakeSecureMessageDelegate& operator=(const FakeSecureMessageDelegate&) =
delete;
~FakeSecureMessageDelegate() override;
// SecureMessageDelegate:
void GenerateKeyPair(GenerateKeyPairCallback callback) override;
void DeriveKey(const std::string& private_key,
const std::string& public_key,
DeriveKeyCallback callback) override;
void CreateSecureMessage(const std::string& payload,
const std::string& key,
const CreateOptions& create_options,
CreateSecureMessageCallback callback) override;
void UnwrapSecureMessage(const std::string& serialized_message,
const std::string& key,
const UnwrapOptions& unwrap_options,
UnwrapSecureMessageCallback callback) override;
// Returns the corresponding private key for the given |public_key|.
std::string GetPrivateKeyForPublicKey(const std::string& public_key);
// Sets the next public key to be returned by GenerateKeyPair(). The
// corresponding private key will be derived from this public key.
void set_next_public_key(const std::string& public_key) {
next_public_key_ = public_key;
}
private:
std::string next_public_key_;
};
class FakeSecureMessageDelegateFactory
: public multidevice::SecureMessageDelegateImpl::Factory {
public:
FakeSecureMessageDelegateFactory() = default;
FakeSecureMessageDelegateFactory(const FakeSecureMessageDelegateFactory&) =
delete;
FakeSecureMessageDelegateFactory& operator=(
const FakeSecureMessageDelegateFactory&) = delete;
~FakeSecureMessageDelegateFactory() override = default;
multidevice::FakeSecureMessageDelegate* instance() { return instance_; }
private:
// multidevice::SecureMessageDelegateImpl::Factory:
std::unique_ptr<multidevice::SecureMessageDelegate> CreateInstance() override;
raw_ptr<multidevice::FakeSecureMessageDelegate, DanglingUntriaged> instance_ =
nullptr;
};
} // namespace ash::multidevice
#endif // CHROMEOS_ASH_COMPONENTS_MULTIDEVICE_FAKE_SECURE_MESSAGE_DELEGATE_H_
|
