/*
 * Copyright (C) 2018 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 <gtest/gtest.h>

#include <string>
#include <utils/String16.h>

#include <keymaster/logger.h>
#include <keystore/keystore_attestation_id.h>

#include <android/security/keystore/KeyAttestationApplicationId.h>
#include <android/security/keystore/KeyAttestationPackageInfo.h>
#include <android/security/keystore/Signature.h>

using ::android::String16;
using ::android::security::KEY_ATTESTATION_APPLICATION_ID_MAX_SIZE;
using ::android::security::keystore::KeyAttestationApplicationId;
using ::android::security::keystore::KeyAttestationPackageInfo;
using std::vector;

namespace keystore {

namespace test {

namespace {

constexpr const char* kDummyPackageName = "DummyPackage";

constexpr const char* kLongPackageName =
    "a.long.package.name"
    "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa"
    "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa"
    "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa"
    "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa"
    "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa"
    "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa"
    "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa"
    "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa"
    "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa"
    "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa"
    "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa"
    "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa";

constexpr const char* kReasonablePackageName =
    "a.reasonable.length.package.name"
    "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa"
    "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa"
    "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa"
    "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa";

constexpr const size_t kTooManyPackages = 4;

// Signatures should be 32 bytes
constexpr const uint8_t kDummySignature[32] = {
    0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08, 0x09, 0x0a, 0x0b, 0x0c, 0x0d, 0x0e, 0x0f,
    0x10, 0x11, 0x12, 0x13, 0x14, 0x15, 0x16, 0x17, 0x18, 0x19, 0x1a, 0x1b, 0x1c, 0x1d, 0x1e, 0x1f};
constexpr const size_t kTooManySignatures = 35;

}  // namespace

using ::android::security::build_attestation_application_id;
using ::android::security::keystore::Signature;

KeyAttestationPackageInfo make_package_info_with_signatures(const char* package_name,
                                                            std::vector<Signature> signatures) {
    auto pInfo = KeyAttestationPackageInfo();
    pInfo.packageName = String16(package_name);
    pInfo.versionCode = 1;
    std::move(signatures.begin(), signatures.end(), std::back_inserter(pInfo.signatures));

    return pInfo;
}

KeyAttestationPackageInfo make_package_info(const char* package_name) {
    return make_package_info_with_signatures(package_name, std::vector<Signature>());
}

TEST(AaidTruncationTest, shortPackageInfoTest) {
    KeyAttestationApplicationId app_id;
    auto pInfo = make_package_info(kDummyPackageName);
    app_id.packageInfos.push_back(std::move(pInfo));

    auto result = build_attestation_application_id(app_id);
    ASSERT_TRUE(result.isOk());
    std::vector<uint8_t>& encoded_app_id = result;
    ASSERT_LT(encoded_app_id.size(), KEY_ATTESTATION_APPLICATION_ID_MAX_SIZE);
}

TEST(AaidTruncationTest, tooLongPackageNameTest) {
    KeyAttestationApplicationId app_id;
    auto pInfo = make_package_info(kLongPackageName);
    app_id.packageInfos.push_back(std::move(pInfo));

    auto result = build_attestation_application_id(app_id);
    ASSERT_TRUE(result.isOk());
    std::vector<uint8_t>& encoded_app_id = result;
    ASSERT_LT(encoded_app_id.size(), KEY_ATTESTATION_APPLICATION_ID_MAX_SIZE);
}

TEST(AaidTruncationTest, tooManySignaturesTest) {
    std::vector<uint8_t> dummy_sig_data(kDummySignature, kDummySignature + 32);
    std::vector<Signature> signatures;
    // Add 35 signatures which will surely exceed the 1K limit.
    for (size_t i = 0; i < kTooManySignatures; ++i) {
        auto sign = Signature();
        sign.data = dummy_sig_data;
        signatures.push_back(std::move(sign));
    }

    auto pInfo = make_package_info_with_signatures(kDummyPackageName, std::move(signatures));
    KeyAttestationApplicationId app_id;
    app_id.packageInfos.push_back(std::move(pInfo));

    auto result = build_attestation_application_id(app_id);
    ASSERT_TRUE(result.isOk());
    std::vector<uint8_t>& encoded_app_id = result;
    ASSERT_LT(encoded_app_id.size(), KEY_ATTESTATION_APPLICATION_ID_MAX_SIZE);
}

TEST(AaidTruncationTest, combinedPackagesAndSignaturesTest) {
    std::vector<uint8_t> dummy_sig_data(kDummySignature, kDummySignature + 32);
    ::std::vector<KeyAttestationPackageInfo> packages;

    for (size_t i = 0; i < kTooManyPackages; ++i) {
        std::vector<Signature> signatures;
        // Add a few signatures for each package
        for (int j = 0; j < 3; ++j) {
            auto sign = Signature();
            sign.data = dummy_sig_data;
            signatures.push_back(std::move(sign));
        }
        packages.push_back(std::move(
            make_package_info_with_signatures(kReasonablePackageName, std::move(signatures))));
    }
    KeyAttestationApplicationId app_id;
    std::move(packages.begin(), packages.end(), std::back_inserter(app_id.packageInfos));

    auto result = build_attestation_application_id(app_id);
    ASSERT_TRUE(result.isOk());
    std::vector<uint8_t>& encoded_app_id = result;
    ASSERT_LT(encoded_app_id.size(), KEY_ATTESTATION_APPLICATION_ID_MAX_SIZE);
}

}  // namespace test
}  // namespace keystore