/*
 * Copyright (c) 2022 Huawei Device Co., Ltd.
 * 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 <gtest/gtest.h>
#include <iostream>
#include <string>

#define private public
#include "tls.h"
#include "tls_key.h"

namespace OHOS {
namespace NetStack {
namespace TlsSocket {
namespace {
using namespace testing::ext;
constexpr int FILE_READ_KEY_LEN = 4096;
static char g_keyFile[] =
    "-----BEGIN RSA PRIVATE KEY-----\r\n"
    "MIIEowIBAAKCAQEAqVzrf6PkLu0uhp5yl2HPNm0vLyI1KLqgsdz5s+JvVdbPXNxD\r\n"
    "g6fmdwa64tJXZPKx7i1KwNs/Jx3xv1N6rqB0au+Ku0Zdq7zbMCqej63SbFW1XWvQ\r\n"
    "6RJ76GcitgrFMTlQN4AzfX0xLFaUJHRuDS4QC5UE9CmV3kD09BNgItu/hxPAHSwg\r\n"
    "q6myc1uufYCwCUIV3bzxd65M343zubTlwOSmsCSqQIl8C1Gd6NWT69tL4fq2hHc/\r\n"
    "09VAlcLvugztwM6NHwDCmRFEDz3RdRahAvCEde8OkY/Aor6UucYWzCJofLeyKVQg\r\n"
    "6J3CTsT/zUE6pdKTvuhQbpRCtWKWSa7qDv1WywIDAQABAoIBAFGpbCPvcmbuFjDy\r\n"
    "1W4Iy1EC9G1VoSwyUKlyUzRZSjWpjfLIggVJP+bEZ/hWU61pGEIvtIupK5pA5f/K\r\n"
    "0KzC0V9+gPYrx563QTjIVAwTVBLIgNq60dCQCQ7WK/Z62voRGIyqVCl94+ftFyE8\r\n"
    "wpO4UiRDhk/0fT7dMz882G32ZzNJmY9eHu+yOaRctJW2gRBROHpQfDGBCz7w8s2j\r\n"
    "ulIcnvwGOrvVllsL+vgY95M0LOq0W8ObbUSlawTnNTSRxFL68Hz5EaVJ19EYvEcC\r\n"
    "eWnpEqIfF8OhQ+mYbdrAutXCkqJLz3rdu5P2Lbk5Ht5ETfr7rtUzvb4+ExIcxVOs\r\n"
    "eys8EgECgYEA29tTxJOy2Cb4DKB9KwTErD1sFt9Ed+Z/A3RGmnM+/h75DHccqS8n\r\n"
    "g9DpvHVMcMWYFVYGlEHC1F+bupM9CgxqQcVhGk/ysJ5kXF6lSTnOQxORnku3HXnV\r\n"
    "4QzgKtLfHbukW1Y2RZM3aCz+Hg+bJrpacWyWZ4tRWNYsO58JRaubZjsCgYEAxTSP\r\n"
    "yUBleQejl5qO76PGUUs2W8+GPr492NJGb63mEiM1zTYLVN0uuDJ2JixzHb6o1NXZ\r\n"
    "6i00pSksT3+s0eiBTRnF6BJ0y/8J07ZnfQQXRAP8ypiZtd3jdOnUxEHfBw2QaIdP\r\n"
    "tVdUc2mpIhosAYT9sWpHYvlUqTCdeLwhkYfgeLECgYBoajjVcmQM3i0OKiZoCOKy\r\n"
    "/pTYI/8rho+p/04MylEPdXxIXEWDYD6/DrgDZh4ArQc2kt2bCcRTAnk+WfEyVYUd\r\n"
    "aXVdfry+/uqhJ94N8eMw3hlZeZIk8JkQQgIwtGd8goJjUoWB85Hr6vphIn5IHVcY\r\n"
    "6T5hPLxMmaL2SeioawDpwwKBgQCFXjDH6Hc3zQTEKND2HIqou/b9THH7yOlG056z\r\n"
    "NKZeKdXe/OfY8uT/yZDB7FnGCgVgO2huyTfLYvcGpNAZ/eZEYGPJuYGn3MmmlruS\r\n"
    "fsvFQfUahu2dY3zKusEcIXhV6sR5DNnJSFBi5VhvKcgNFwYDkF7K/thUu/4jgwgo\r\n"
    "xf33YQKBgDQffkP1jWqT/pzlVLFtF85/3eCC/uedBfxXknVMrWE+CM/Vsx9cvBZw\r\n"
    "hi15LA5+hEdbgvj87hmMiCOc75e0oz2Rd12ZoRlBVfbncH9ngfqBNQElM7Bueqoc\r\n"
    "JOpKV+gw0gQtiu4beIdFnYsdZoZwrTjC4rW7OI0WYoLJabMFFh3I\r\n"
    "-----END RSA PRIVATE KEY-----\r\n";
} // namespace

class TlsKeyTest : public testing::Test {
public:
    static void SetUpTestCase() {}

    static void TearDownTestCase() {}

    virtual void SetUp() {}

    virtual void TearDown() {}
};

HWTEST_F(TlsKeyTest, AlgorithmTest, TestSize.Level2)
{
    SecureData structureData(g_keyFile);
    std::string keyPassStr = "";
    SecureData keyPass(keyPassStr);
    TLSKey tlsKey = TLSKey(structureData, ALGORITHM_RSA, keyPass);
    KeyAlgorithm algorithm = tlsKey.Algorithm();
    EXPECT_EQ(algorithm, ALGORITHM_RSA);
}

HWTEST_F(TlsKeyTest, DecodePemTest, TestSize.Level2)
{
    std::string fileName = "";
    std::string keyPassStr = "";
    SecureData keyPass(keyPassStr);
    TLSKey tlsKey1 = TLSKey(fileName, ALGORITHM_RSA, keyPass);
    TLSKey tlsKey2 = TLSKey(fileName, ALGORITHM_RSA, keyPass, EncodingFormat::DER);
    std::string fileName2 = "/system/lib";
    TLSKey tlsKey3 = TLSKey(fileName2, ALGORITHM_RSA, keyPass);
    TLSKey tlsKey4 = TLSKey(fileName2, ALGORITHM_RSA, keyPass, EncodingFormat::DER);
    SecureData structureData(g_keyFile);
    structureData.length_ = sizeof(g_keyFile);
    TLSKey tlsKey = TLSKey(structureData, ALGORITHM_RSA, keyPass);
    EXPECT_EQ(tlsKey.keyIsNull_, false);
}

HWTEST_F(TlsKeyTest, CopyConstruction, TestSize.Level2)
{
    SecureData structureData(g_keyFile);
    std::string keyPassStr = "";
    SecureData keyPass(keyPassStr);
    TLSKey tlsKey = TLSKey(structureData, ALGORITHM_RSA, keyPass);
    TLSKey tlsKeyCopy = TLSKey(tlsKey);
    KeyAlgorithm algorithm = tlsKeyCopy.Algorithm();
    EXPECT_EQ(algorithm, ALGORITHM_RSA);
}

HWTEST_F(TlsKeyTest, AssignmentConstruction, TestSize.Level2)
{
    SecureData structureData(g_keyFile);
    std::string keyPassStr = "";
    SecureData keyPass(keyPassStr);
    TLSKey tlsKey = TLSKey(structureData, ALGORITHM_RSA, keyPass);
    TLSKey key = tlsKey;
    KeyAlgorithm algorithm = key.Algorithm();
    EXPECT_EQ(algorithm, ALGORITHM_RSA);
}

HWTEST_F(TlsKeyTest, HandleTest, TestSize.Level2)
{
    SecureData structureData(g_keyFile);
    std::string keyPassStr = "";
    SecureData keyPass(keyPassStr);
    TLSKey tlsKey = TLSKey(structureData, ALGORITHM_RSA, keyPass);
    Handle handle = tlsKey.handle();
    EXPECT_NE(handle, nullptr);
    tlsKey.Clear(true);
    TLSKey tlsKeyDsa = TLSKey(structureData, ALGORITHM_DSA, keyPass);
    Handle handleDsa = tlsKeyDsa.handle();
    EXPECT_EQ(handleDsa, nullptr);
    tlsKeyDsa.Clear(true);
    TLSKey tlsKeyEc = TLSKey(structureData, ALGORITHM_EC, keyPass);
    Handle handleEc = tlsKeyEc.handle();
    EXPECT_EQ(handleEc, nullptr);
    tlsKeyEc.Clear(true);
    TLSKey tlsKeyDh = TLSKey(structureData, ALGORITHM_DH, keyPass);
    Handle handleDh = tlsKeyDh.handle();
    EXPECT_EQ(handleDh, nullptr);
    tlsKeyDh.Clear(true);
    TLSKey tlsKeyOpaque = TLSKey(structureData, OPAQUE, keyPass);
    Handle handleOpaque = tlsKeyOpaque.handle();
    EXPECT_EQ(handleOpaque, nullptr);
    tlsKeyOpaque.Clear(true);
    TLSKey keyOpaque = tlsKeyOpaque;
    EXPECT_NE(handle, nullptr);
}

HWTEST_F(TlsKeyTest, GetKeyPassTest, TestSize.Level2)
{
    SecureData structureData(g_keyFile);
    std::string keyPassStr = "";
    SecureData keyPass(keyPassStr);
    TLSKey tlsKey = TLSKey(structureData, ALGORITHM_RSA, keyPass);
    SecureData getKeyPass = tlsKey.GetKeyPass();
    EXPECT_EQ(getKeyPass.Length(), keyPass.Length());
}

HWTEST_F(TlsKeyTest, GetKeyDataTest, TestSize.Level2)
{
    SecureData structureData(g_keyFile);
    std::string keyPassStr = "";
    SecureData keyPass(keyPassStr);
    TLSKey tlsKey = TLSKey(structureData, ALGORITHM_RSA, keyPass);
    SecureData getKeyData = tlsKey.GetKeyData();
    EXPECT_EQ(getKeyData.Length(), structureData.Length());
}

HWTEST_F(TlsKeyTest, AlgorithmTest2, TestSize.Level2)
{
    SecureData structureData(g_keyFile);
    std::string keyPassStr = "";
    SecureData keyPass(keyPassStr);
    TLSKey tlsKeyDsa = TLSKey(structureData, ALGORITHM_DSA, keyPass);
    TLSKey keyDsa = tlsKeyDsa;
    TLSKey tlsKeyEc = TLSKey(structureData, ALGORITHM_EC, keyPass);
    TLSKey keyEc = tlsKeyEc;
    TLSKey tlsKeyDh = TLSKey(structureData, ALGORITHM_DH, keyPass);
    TLSKey keyDh = tlsKeyDh;
    TLSKey tlsKeyOpaque = TLSKey(structureData, OPAQUE, keyPass);
    TLSKey keyOpaque = tlsKeyOpaque;
    SecureData getKeyData = tlsKeyDsa.GetKeyData();
    EXPECT_EQ(getKeyData.Length(), structureData.Length());
}

HWTEST_F(TlsKeyTest, SwitchAlgorithmTest, TestSize.Level2)
{
    SecureData structureData(g_keyFile);
    std::string keyPassStr = "";

    KeyType typePublic = KeyType::PUBLIC_KEY;
    SecureData keyPass(keyPassStr);
    KeyType typePrivate = KeyType::PRIVATE_KEY;
    char privateKey[FILE_READ_KEY_LEN] = {0};
    const char *privateKeyData = static_cast<const char *>(privateKey);
    BIO *bio = BIO_new_mem_buf(privateKeyData, -1);
    TLSKey tlsKeyRsa = TLSKey(structureData, ALGORITHM_RSA, keyPass);
    tlsKeyRsa.SwitchAlgorithm(typePrivate, ALGORITHM_RSA, bio);
    tlsKeyRsa.SwitchAlgorithm(typePublic, ALGORITHM_RSA, bio);
    TLSKey tlsKeyDsa = TLSKey(structureData, ALGORITHM_DSA, keyPass);
    tlsKeyDsa.SwitchAlgorithm(typePrivate, ALGORITHM_DSA, bio);
    tlsKeyDsa.SwitchAlgorithm(typePublic, ALGORITHM_DSA, bio);
    TLSKey tlsKeyEc = TLSKey(structureData, ALGORITHM_EC, keyPass);
    tlsKeyEc.SwitchAlgorithm(typePrivate, ALGORITHM_EC, bio);
    tlsKeyEc.SwitchAlgorithm(typePublic, ALGORITHM_EC, bio);
    TLSKey tlsKeyDh = TLSKey(structureData, ALGORITHM_DH, keyPass);
    tlsKeyDh.SwitchAlgorithm(typePrivate, ALGORITHM_DH, bio);
    tlsKeyDh.SwitchAlgorithm(typePublic, ALGORITHM_DH, bio);
    TLSKey tlsKeyOpaque = TLSKey(structureData, OPAQUE, keyPass);
    tlsKeyOpaque.SwitchAlgorithm(typePrivate, OPAQUE, bio);
    tlsKeyOpaque.SwitchAlgorithm(typePublic, OPAQUE, bio);
    SecureData getKeyData = tlsKeyDsa.GetKeyData();
    EXPECT_EQ(getKeyData.Length(), structureData.Length());
}

HWTEST_F(TlsKeyTest, ClearTest, TestSize.Level2)
{
    SecureData structureData(g_keyFile);
    std::string keyPassStr = "";
    SecureData keyPass(keyPassStr);
    TLSKey tlsKeyDsa = TLSKey(structureData, ALGORITHM_DSA, keyPass);
    tlsKeyDsa.dsa_ = DSA_new();
    tlsKeyDsa.Clear(true);
    TLSKey tlsKeyDh = TLSKey(structureData, ALGORITHM_DH, keyPass);
    tlsKeyDh.dh_ = DH_new();
    tlsKeyDh.Clear(true);
    TLSKey tlsKeyEc = TLSKey(structureData, ALGORITHM_EC, keyPass);
    tlsKeyEc.ec_ = EC_KEY_new();
    tlsKeyEc.Clear(true);
    TLSKey tlsKeyOpaque = TLSKey(structureData, OPAQUE, keyPass);
    tlsKeyOpaque.genericKey_ = EVP_PKEY_new();
    tlsKeyOpaque.Clear(true);
    EXPECT_EQ(tlsKeyOpaque.keyIsNull_, true);
}

HWTEST_F(TlsKeyTest, DecodeDataTest, TestSize.Level2)
{
    SecureData data;
    std::string keyPassStr = "";
    SecureData keyPass(keyPassStr);
    TLSKey tlsKey = TLSKey(data, ALGORITHM_DSA, keyPass);
    tlsKey.DecodeData(data, ALGORITHM_DSA, keyPass);
    SecureData getKeyData = tlsKey.GetKeyData();
    EXPECT_EQ(getKeyData.Length(), data.Length());
}
} // namespace TlsSocket
} // namespace NetStack
} // namespace OHOS