73 lines
		
	
	
		
			2.8 KiB
		
	
	
	
		
			C++
		
	
	
	
			
		
		
	
	
			73 lines
		
	
	
		
			2.8 KiB
		
	
	
	
		
			C++
		
	
	
	
| /*
 | |
|  * Copyright 2015 The Android Open Source Project
 | |
|  *
 | |
|  * Licensed under the Apache License, Version 2.0 (the "License");
 | |
|  * you may not use this file except in compliance with the License.
 | |
|  * You may obtain a copy of the License at
 | |
|  *
 | |
|  *      http://www.apache.org/licenses/LICENSE-2.0
 | |
|  *
 | |
|  * Unless required by applicable law or agreed to in writing, software
 | |
|  * distributed under the License is distributed on an "AS IS" BASIS,
 | |
|  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 | |
|  * See the License for the specific language governing permissions and
 | |
|  * limitations under the License.
 | |
|  */
 | |
| 
 | |
| #include <keymaster/km_openssl/ecies_kem.h>
 | |
| 
 | |
| #include <gtest/gtest.h>
 | |
| #include <openssl/evp.h>
 | |
| 
 | |
| #include <hardware/keymaster_defs.h>
 | |
| 
 | |
| #include <keymaster/android_keymaster_utils.h>
 | |
| #include <keymaster/km_openssl/nist_curve_key_exchange.h>
 | |
| 
 | |
| #include "android_keymaster_test_utils.h"
 | |
| 
 | |
| using std::string;
 | |
| 
 | |
| namespace keymaster {
 | |
| namespace test {
 | |
| 
 | |
| static const keymaster_ec_curve_t kEcCurves[] = {KM_EC_CURVE_P_224, KM_EC_CURVE_P_256,
 | |
|                                                  KM_EC_CURVE_P_384, KM_EC_CURVE_P_521};
 | |
| 
 | |
| /**
 | |
|  * TestConsistency just tests that the basic key encapsulation hold.
 | |
|  */
 | |
| TEST(EciesKem, TestConsistency) {
 | |
|     static const uint32_t kKeyLen = 32;
 | |
|     for (auto& curve : kEcCurves) {
 | |
|         AuthorizationSet kem_description(AuthorizationSetBuilder()
 | |
|                                              .Authorization(TAG_EC_CURVE, curve)
 | |
|                                              .Authorization(TAG_KDF, KM_KDF_RFC5869_SHA256)
 | |
|                                              .Authorization(TAG_ECIES_SINGLE_HASH_MODE)
 | |
|                                              .Authorization(TAG_KEY_SIZE, kKeyLen));
 | |
|         keymaster_error_t error;
 | |
|         EciesKem* kem = new EciesKem(kem_description, &error);
 | |
|         ASSERT_EQ(KM_ERROR_OK, error);
 | |
| 
 | |
|         NistCurveKeyExchange* key_exchange = NistCurveKeyExchange::GenerateKeyExchange(curve);
 | |
|         Buffer peer_public_value;
 | |
|         ASSERT_TRUE(key_exchange->public_value(&peer_public_value));
 | |
| 
 | |
|         Buffer output_clear_key;
 | |
|         Buffer output_encrypted_key;
 | |
|         ASSERT_TRUE(kem->Encrypt(peer_public_value, &output_clear_key, &output_encrypted_key));
 | |
|         ASSERT_EQ(kKeyLen, output_clear_key.available_read());
 | |
|         ASSERT_EQ(peer_public_value.available_read(), output_encrypted_key.available_read());
 | |
| 
 | |
|         Buffer decrypted_clear_key;
 | |
|         ASSERT_TRUE(
 | |
|             kem->Decrypt(key_exchange->private_key(), output_encrypted_key, &decrypted_clear_key));
 | |
|         ASSERT_EQ(kKeyLen, decrypted_clear_key.available_read());
 | |
|         EXPECT_EQ(0, memcmp(output_clear_key.peek_read(), decrypted_clear_key.peek_read(),
 | |
|                             output_clear_key.available_read()));
 | |
|     }
 | |
| }
 | |
| 
 | |
| }  // namespace test
 | |
| }  // namespace keymaster
 |