66 lines
2.1 KiB
C++
66 lines
2.1 KiB
C++
/*
|
|
* Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License").
|
|
* You may not use this file except in compliance with the License.
|
|
* A copy of the License is located at
|
|
*
|
|
* http://aws.amazon.com/apache2.0/
|
|
*
|
|
* or in the "license" file accompanying this file. This file 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 <string>
|
|
|
|
#include <gtest/gtest.h>
|
|
|
|
#include <acsdkPkcs11/private/PKCS11Functions.h>
|
|
#include <acsdkPkcs11/private/PKCS11Slot.h>
|
|
#include <acsdkPkcs11/private/PKCS11Session.h>
|
|
#include <acsdkPkcs11/private/PKCS11Key.h>
|
|
|
|
namespace alexaClientSDK {
|
|
namespace acsdkPkcs11 {
|
|
namespace test {
|
|
|
|
using namespace ::testing;
|
|
using namespace ::alexaClientSDK::acsdkCryptoInterfaces;
|
|
|
|
/// Test that the constructor with a nullptr doesn't segfault.
|
|
TEST(PKCS11SessionTest, test_loginLogout) {
|
|
auto functions = PKCS11Functions::create(PKCS11_LIBRARY);
|
|
ASSERT_NE(nullptr, functions);
|
|
std::shared_ptr<PKCS11Slot> slot;
|
|
ASSERT_TRUE(functions->findSlotByTokenName(PKCS11_TOKEN_NAME, slot));
|
|
ASSERT_NE(nullptr, slot);
|
|
auto session = slot->openSession();
|
|
ASSERT_NE(nullptr, session);
|
|
|
|
ASSERT_TRUE(session->logIn(PKCS11_PIN));
|
|
ASSERT_TRUE(session->logOut());
|
|
}
|
|
|
|
TEST(PKCS11SessionTest, test_keySessionOps) {
|
|
auto functions = PKCS11Functions::create(PKCS11_LIBRARY);
|
|
ASSERT_NE(nullptr, functions);
|
|
std::shared_ptr<PKCS11Slot> slot;
|
|
ASSERT_TRUE(functions->findSlotByTokenName(PKCS11_TOKEN_NAME, slot));
|
|
ASSERT_NE(nullptr, slot);
|
|
auto session = slot->openSession();
|
|
ASSERT_NE(nullptr, session);
|
|
|
|
ASSERT_EQ(true, session->logIn(PKCS11_PIN));
|
|
|
|
// Delete test key if it exists
|
|
auto key = session->findKey({PKCS11_KEY_NAME, AlgorithmType::AES_256_CBC});
|
|
ASSERT_NE(nullptr, key);
|
|
ASSERT_TRUE(session->logOut());
|
|
}
|
|
|
|
} // namespace test
|
|
} // namespace acsdkPkcs11
|
|
} // namespace alexaClientSDK
|