/*
 * Copyright 2019, 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.
 */

#ifndef ANDROID_HARDWARE_IDENTITY_WRITABLEIDENTITYCREDENTIAL_H
#define ANDROID_HARDWARE_IDENTITY_WRITABLEIDENTITYCREDENTIAL_H

#include <aidl/android/hardware/identity/BnWritableIdentityCredential.h>
#include <android/hardware/identity/support/IdentityCredentialSupport.h>

#include <cppbor.h>
#include <set>

#include "IdentityCredentialStore.h"
#include "SecureHardwareProxy.h"

namespace aidl::android::hardware::identity {

using ::android::sp;
using ::android::hardware::identity::SecureHardwareProvisioningProxy;
using ::std::optional;
using ::std::set;
using ::std::string;
using ::std::vector;

class WritableIdentityCredential : public BnWritableIdentityCredential {
  public:
    // For a new credential, call initialize() right after construction.
    //
    // For an updated credential, call initializeForUpdate() right after construction.
    //
    WritableIdentityCredential(sp<SecureHardwareProvisioningProxy> hwProxy, const string& docType,
                               bool testCredential, HardwareInformation hardwareInformation)
        : hwProxy_(hwProxy),
          docType_(docType),
          testCredential_(testCredential),
          hardwareInformation_(std::move(hardwareInformation)) {}

    ~WritableIdentityCredential();

    // Creates the Credential Key. Returns false on failure.
    bool initialize();

    // Used when updating a credential. Returns false on failure.
    bool initializeForUpdate(const vector<uint8_t>& encryptedCredentialKeys);

    // Methods from IWritableIdentityCredential follow.
    ndk::ScopedAStatus getAttestationCertificate(const vector<uint8_t>& attestationApplicationId,
                                                 const vector<uint8_t>& attestationChallenge,
                                                 vector<Certificate>* outCertificateChain) override;

    ndk::ScopedAStatus setExpectedProofOfProvisioningSize(
            int32_t expectedProofOfProvisioningSize) override;

    ndk::ScopedAStatus startPersonalization(int32_t accessControlProfileCount,
                                            const vector<int32_t>& entryCounts) override;

    ndk::ScopedAStatus addAccessControlProfile(
            int32_t id, const Certificate& readerCertificate, bool userAuthenticationRequired,
            int64_t timeoutMillis, int64_t secureUserId,
            SecureAccessControlProfile* outSecureAccessControlProfile) override;

    ndk::ScopedAStatus beginAddEntry(const vector<int32_t>& accessControlProfileIds,
                                     const string& nameSpace, const string& name,
                                     int32_t entrySize) override;
    ndk::ScopedAStatus addEntryValue(const vector<uint8_t>& content,
                                     vector<uint8_t>* outEncryptedContent) override;

    ndk::ScopedAStatus finishAddingEntries(
            vector<uint8_t>* outCredentialData,
            vector<uint8_t>* outProofOfProvisioningSignature) override;

    ndk::ScopedAStatus setRemotelyProvisionedAttestationKey(
            const vector<uint8_t>& attestationKeyBlob,
            const vector<uint8_t>& attestationCertificateChain) override;

  private:
    // Set by constructor.
    sp<SecureHardwareProvisioningProxy> hwProxy_;
    string docType_;
    bool testCredential_;
    HardwareInformation hardwareInformation_;

    // This is set in initialize().
    bool startPersonalizationCalled_;
    bool firstEntry_;

    // This is set in getAttestationCertificate().
    bool getAttestationCertificateAlreadyCalled_ = false;

    // These fields are initialized during startPersonalization()
    size_t numAccessControlProfileRemaining_;
    vector<int32_t> remainingEntryCounts_;
    cppbor::Array signedDataAccessControlProfiles_;
    cppbor::Map signedDataNamespaces_;
    cppbor::Array signedDataCurrentNamespace_;
    size_t expectedProofOfProvisioningSize_;

    // This field is initialized in addAccessControlProfile
    set<int32_t> accessControlProfileIds_;

    // These fields are initialized during beginAddEntry()
    size_t entryRemainingBytes_;
    string entryNameSpace_;
    string entryName_;
    vector<int32_t> entryAccessControlProfileIds_;
    vector<uint8_t> entryBytes_;
    set<string> allNameSpaces_;

    // Remotely provisioned attestation data, set via setRemotelyProvisionedAttestationKey
    optional<vector<uint8_t>> attestationKeyBlob_;
    optional<vector<vector<uint8_t>>> attestationCertificateChain_;
};

}  // namespace aidl::android::hardware::identity

#endif  // ANDROID_HARDWARE_IDENTITY_WRITABLEIDENTITYCREDENTIAL_H