/*
 * Copyright 2014 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 SYSTEM_KEYMASTER_RSA_KEY_H_
#define SYSTEM_KEYMASTER_RSA_KEY_H_

#include <openssl/rsa.h>

#include "asymmetric_key.h"

namespace keymaster {

class RsaKeyFactory : public AsymmetricKeyFactory {
  public:
    keymaster_algorithm_t registry_key() const override { return KM_ALGORITHM_RSA; }
    Key* GenerateKey(const AuthorizationSet& key_description, keymaster_error_t* error);
    Key* ImportKey(const AuthorizationSet& key_description, keymaster_key_format_t key_format,
                   const uint8_t* key_data, size_t key_data_length,
                   keymaster_error_t* error) override;
    Key* LoadKey(const UnencryptedKeyBlob& blob, keymaster_error_t* error) override;
};

class RsaOperationFactory;

class RsaKey : public AsymmetricKey {
  private:
    friend class RsaKeyFactory;
    friend class RsaOperationFactory;

    RsaKey(const UnencryptedKeyBlob& blob, keymaster_error_t* error);
    RsaKey(RSA* rsa_key, const AuthorizationSet& auths) : AsymmetricKey(auths), rsa_key_(rsa_key) {}

    int evp_key_type() override { return EVP_PKEY_RSA; }
    bool InternalToEvp(EVP_PKEY* pkey) const override;
    bool EvpToInternal(const EVP_PKEY* pkey) override;

    bool SupportedMode(keymaster_purpose_t purpose, keymaster_padding_t padding);
    bool SupportedMode(keymaster_purpose_t purpose, keymaster_digest_t digest);

    struct RSA_Delete {
        void operator()(RSA* p) { RSA_free(p); }
    };

    RSA* key() const;

    mutable UniquePtr<RSA, RSA_Delete> rsa_key_;
};

}  // namespace keymaster

#endif  // SYSTEM_KEYMASTER_RSA_KEY_H_