| // Copyright (c) 2012 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 CRYPTO_MOCK_APPLE_KEYCHAIN_H_ |
| #define CRYPTO_MOCK_APPLE_KEYCHAIN_H_ |
| |
| #include <stddef.h> |
| #include <stdint.h> |
| |
| #include <map> |
| #include <set> |
| #include <string> |
| #include <vector> |
| |
| #include "base/compiler_specific.h" |
| #include "crypto/apple_keychain.h" |
| |
| namespace crypto { |
| |
| // Mock Keychain wrapper for testing code that interacts with the OS X |
| // Keychain. |
| // |
| // Note that "const" is pretty much meaningless for this class; the const-ness |
| // of AppleKeychain doesn't apply to the actual keychain data, so all of the |
| // Mock data is mutable; don't assume that it won't change over the life of |
| // tests. |
| class CRYPTO_EXPORT MockAppleKeychain : public AppleKeychain { |
| public: |
| MockAppleKeychain(); |
| ~MockAppleKeychain() override; |
| |
| // AppleKeychain implementation. |
| OSStatus FindGenericPassword(UInt32 serviceNameLength, |
| const char* serviceName, |
| UInt32 accountNameLength, |
| const char* accountName, |
| UInt32* passwordLength, |
| void** passwordData, |
| AppleSecKeychainItemRef* itemRef) const override; |
| OSStatus ItemFreeContent(void* data) const override; |
| OSStatus AddGenericPassword(UInt32 serviceNameLength, |
| const char* serviceName, |
| UInt32 accountNameLength, |
| const char* accountName, |
| UInt32 passwordLength, |
| const void* passwordData, |
| AppleSecKeychainItemRef* itemRef) const override; |
| |
| // Returns the password that OSCrypt uses to generate its encryption key. |
| std::string GetEncryptionPassword() const; |
| |
| #if !defined(OS_IOS) |
| OSStatus ItemDelete(SecKeychainItemRef itemRef) const override; |
| #endif // !defined(OS_IOS) |
| |
| // |FindGenericPassword()| can return different results depending on user |
| // interaction with the system Keychain. For mocking purposes we allow the |
| // user of this class to specify the result code of the |
| // |FindGenericPassword()| call so we can simulate the result of different |
| // user interactions. |
| void set_find_generic_result(OSStatus result) { |
| find_generic_result_ = result; |
| } |
| |
| // Returns the true if |AddGenericPassword()| was called. |
| bool called_add_generic() const { return called_add_generic_; } |
| |
| // Returns the number of allocations - deallocations for password data. |
| int password_data_count() const { return password_data_count_; } |
| |
| private: |
| // Result code for the |FindGenericPassword()| method. |
| OSStatus find_generic_result_; |
| |
| // Records whether |AddGenericPassword()| gets called. |
| mutable bool called_add_generic_; |
| |
| // Tracks the allocations and frees of password data in |FindGenericPassword| |
| // and |ItemFreeContent|. |
| mutable int password_data_count_; |
| |
| DISALLOW_COPY_AND_ASSIGN(MockAppleKeychain); |
| }; |
| |
| } // namespace crypto |
| |
| #endif // CRYPTO_MOCK_APPLE_KEYCHAIN_H_ |