Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(110)

Side by Side Diff: chrome/browser/chromeos/login/mock_owner_key_utils.h

Issue 6538098: [Chrome OS] Owner keys are generated outside Chrome now; handle appropriately (Closed) Base URL: svn://svn.chromium.org/chrome/trunk/src
Patch Set: Created 9 years, 10 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch | Annotate | Revision Log
OLDNEW
1 // Copyright (c) 2010 The Chromium Authors. All rights reserved. 1 // Copyright (c) 2010 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be 2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file. 3 // found in the LICENSE file.
4 4
5 #ifndef CHROME_BROWSER_CHROMEOS_LOGIN_MOCK_OWNER_KEY_UTILS_H_ 5 #ifndef CHROME_BROWSER_CHROMEOS_LOGIN_MOCK_OWNER_KEY_UTILS_H_
6 #define CHROME_BROWSER_CHROMEOS_LOGIN_MOCK_OWNER_KEY_UTILS_H_ 6 #define CHROME_BROWSER_CHROMEOS_LOGIN_MOCK_OWNER_KEY_UTILS_H_
7 #pragma once 7 #pragma once
8 8
9 #include <vector> 9 #include <vector>
10 10
11 #include "base/basictypes.h" 11 #include "base/basictypes.h"
12 #include "base/crypto/rsa_private_key.h" 12 #include "base/crypto/rsa_private_key.h"
13 #include "base/file_path.h" 13 #include "base/file_path.h"
14 #include "chrome/browser/browser_thread.h" 14 #include "chrome/browser/browser_thread.h"
15 #include "chrome/browser/chromeos/cros/login_library.h" 15 #include "chrome/browser/chromeos/cros/login_library.h"
16 #include "chrome/browser/chromeos/login/owner_key_utils.h" 16 #include "chrome/browser/chromeos/login/owner_key_utils.h"
17 #include "testing/gmock/include/gmock/gmock.h" 17 #include "testing/gmock/include/gmock/gmock.h"
18 #include "testing/gtest/include/gtest/gtest.h" 18 #include "testing/gtest/include/gtest/gtest.h"
19 19
20 using ::base::RSAPrivateKey; 20 using ::base::RSAPrivateKey;
21 21
22 namespace chromeos { 22 namespace chromeos {
23 23
24 class MockKeyUtils : public OwnerKeyUtils { 24 class MockKeyUtils : public OwnerKeyUtils {
25 public: 25 public:
26 MockKeyUtils() {} 26 MockKeyUtils() {}
27 MOCK_METHOD0(GenerateKeyPair, RSAPrivateKey*());
28 MOCK_METHOD2(ExportPublicKeyViaDbus, bool(RSAPrivateKey* pair,
29 LoginLibrary::Delegate*));
30 MOCK_METHOD2(ExportPublicKeyToFile, bool(RSAPrivateKey* pair,
31 const FilePath& key_file));
32 MOCK_METHOD2(ImportPublicKey, bool(const FilePath& key_file, 27 MOCK_METHOD2(ImportPublicKey, bool(const FilePath& key_file,
33 std::vector<uint8>* output)); 28 std::vector<uint8>* output));
34 MOCK_METHOD3(Verify, bool(const std::string& data, 29 MOCK_METHOD3(Verify, bool(const std::string& data,
35 const std::vector<uint8> signature, 30 const std::vector<uint8> signature,
36 const std::vector<uint8> public_key)); 31 const std::vector<uint8> public_key));
37 MOCK_METHOD3(Sign, bool(const std::string& data, 32 MOCK_METHOD3(Sign, bool(const std::string& data,
38 std::vector<uint8>* OUT_signature, 33 std::vector<uint8>* OUT_signature,
39 base::RSAPrivateKey* key)); 34 base::RSAPrivateKey* key));
40 MOCK_METHOD1(FindPrivateKey, RSAPrivateKey*(const std::vector<uint8>& key)); 35 MOCK_METHOD1(FindPrivateKey, RSAPrivateKey*(const std::vector<uint8>& key));
41 MOCK_METHOD0(GetOwnerKeyFilePath, FilePath()); 36 MOCK_METHOD0(GetOwnerKeyFilePath, FilePath());
42 37 MOCK_METHOD2(ExportPublicKeyToFile, bool(base::RSAPrivateKey* pair,
43 // To simulate doing a LoginLibrary::SetOwnerKey call 38 const FilePath& key_file));
44 static void SetOwnerKeyCallback(LoginLibrary::Delegate* callback,
45 bool value) {
46 callback->OnComplete(value);
47 }
48
49 static bool ExportPublicKeyViaDbusWin(RSAPrivateKey* key,
50 LoginLibrary::Delegate* d) {
51 BrowserThread::PostTask(
52 BrowserThread::UI, FROM_HERE,
53 NewRunnableFunction(&SetOwnerKeyCallback, d, true));
54 return true;
55 }
56
57 static bool ExportPublicKeyViaDbusFail(RSAPrivateKey* key,
58 LoginLibrary::Delegate* d) {
59 BrowserThread::PostTask(
60 BrowserThread::UI, FROM_HERE,
61 NewRunnableFunction(&SetOwnerKeyCallback, d, false));
62 return false;
63 }
64
65 protected: 39 protected:
66 virtual ~MockKeyUtils() {} 40 virtual ~MockKeyUtils() {}
67 41
68 private: 42 private:
69 DISALLOW_COPY_AND_ASSIGN(MockKeyUtils); 43 DISALLOW_COPY_AND_ASSIGN(MockKeyUtils);
70 }; 44 };
71 45
72 class MockInjector : public OwnerKeyUtils::Factory { 46 class MockInjector : public OwnerKeyUtils::Factory {
73 public: 47 public:
74 // Takes ownership of |mock|. 48 // Takes ownership of |mock|.
75 explicit MockInjector(MockKeyUtils* mock) : transient_(mock) { 49 explicit MockInjector(MockKeyUtils* mock) : transient_(mock) {
76 } 50 }
77 51
78 virtual ~MockInjector() {} 52 virtual ~MockInjector() {}
79 53
80 // If this is called, its caller takes ownership of |transient_|. 54 // If this is called, its caller takes ownership of |transient_|.
81 // If it's never called, |transient_| remains our problem. 55 // If it's never called, |transient_| remains our problem.
82 OwnerKeyUtils* CreateOwnerKeyUtils() { 56 OwnerKeyUtils* CreateOwnerKeyUtils() {
83 return transient_.get(); 57 return transient_.get();
84 } 58 }
85 59
86 private: 60 private:
87 scoped_refptr<MockKeyUtils> transient_; 61 scoped_refptr<MockKeyUtils> transient_;
88 DISALLOW_COPY_AND_ASSIGN(MockInjector); 62 DISALLOW_COPY_AND_ASSIGN(MockInjector);
89 }; 63 };
90 64
91 } // namespace chromeos 65 } // namespace chromeos
92 66
93 #endif // CHROME_BROWSER_CHROMEOS_LOGIN_MOCK_OWNER_KEY_UTILS_H_ 67 #endif // CHROME_BROWSER_CHROMEOS_LOGIN_MOCK_OWNER_KEY_UTILS_H_
OLDNEW
« no previous file with comments | « chrome/browser/chromeos/login/login_utils.cc ('k') | chrome/browser/chromeos/login/mock_ownership_service.h » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698