# Copyright (C) 2015-2022 Virgil Security, Inc.
#
# All rights reserved.
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions are
# met:
#
#     (1) Redistributions of source code must retain the above copyright
#     notice, this list of conditions and the following disclaimer.
#
#     (2) Redistributions in binary form must reproduce the above copyright
#     notice, this list of conditions and the following disclaimer in
#     the documentation and/or other materials provided with the
#     distribution.
#
#     (3) Neither the name of the copyright holder nor the names of its
#     contributors may be used to endorse or promote products derived from
#     this software without specific prior written permission.
#
# THIS SOFTWARE IS PROVIDED BY THE AUTHOR ''AS IS'' AND ANY EXPRESS OR
# IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
# DISCLAIMED. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT,
# INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
# (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
# SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
# HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
# STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING
# IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
# POSSIBILITY OF SUCH DAMAGE.
#
# Lead Maintainer: Virgil Security Inc. <support@virgilsecurity.com>


from virgil_crypto_lib._libs import *
from ctypes import *
from ._vscf_impl import vscf_impl_t
from ._vscf_ecies import vscf_ecies_t
from ._vscf_error import vscf_error_t
from ._vscf_raw_public_key import vscf_raw_public_key_t
from ._vscf_raw_private_key import vscf_raw_private_key_t
from virgil_crypto_lib.common._c_bridge import vsc_data_t
from virgil_crypto_lib.common._c_bridge import vsc_buffer_t


class vscf_curve25519_t(Structure):
    pass


class VscfCurve25519(object):
    """This is implementation of Curve25519 elliptic curve algorithms."""

    # Defines whether a public key can be imported or not.
    CAN_IMPORT_PUBLIC_KEY = True
    # Define whether a public key can be exported or not.
    CAN_EXPORT_PUBLIC_KEY = True
    # Define whether a private key can be imported or not.
    CAN_IMPORT_PRIVATE_KEY = True
    # Define whether a private key can be exported or not.
    CAN_EXPORT_PRIVATE_KEY = True

    def __init__(self):
        """Create underlying C context."""
        self._ll = LowLevelLibs()
        self._lib = self._ll.foundation

    def vscf_curve25519_new(self):
        vscf_curve25519_new = self._lib.vscf_curve25519_new
        vscf_curve25519_new.argtypes = []
        vscf_curve25519_new.restype = POINTER(vscf_curve25519_t)
        return vscf_curve25519_new()

    def vscf_curve25519_delete(self, ctx):
        vscf_curve25519_delete = self._lib.vscf_curve25519_delete
        vscf_curve25519_delete.argtypes = [POINTER(vscf_curve25519_t)]
        vscf_curve25519_delete.restype = None
        return vscf_curve25519_delete(ctx)

    def vscf_curve25519_use_random(self, ctx, random):
        vscf_curve25519_use_random = self._lib.vscf_curve25519_use_random
        vscf_curve25519_use_random.argtypes = [POINTER(vscf_curve25519_t), POINTER(vscf_impl_t)]
        vscf_curve25519_use_random.restype = None
        return vscf_curve25519_use_random(ctx, random)

    def vscf_curve25519_use_ecies(self, ctx, ecies):
        vscf_curve25519_use_ecies = self._lib.vscf_curve25519_use_ecies
        vscf_curve25519_use_ecies.argtypes = [POINTER(vscf_curve25519_t), POINTER(vscf_ecies_t)]
        vscf_curve25519_use_ecies.restype = None
        return vscf_curve25519_use_ecies(ctx, ecies)

    def vscf_curve25519_generate_ephemeral_key(self, ctx, key, error):
        """Generate ephemeral private key of the same type.
        Note, this operation might be slow."""
        vscf_curve25519_generate_ephemeral_key = self._lib.vscf_curve25519_generate_ephemeral_key
        vscf_curve25519_generate_ephemeral_key.argtypes = [POINTER(vscf_curve25519_t), POINTER(vscf_impl_t), POINTER(vscf_error_t)]
        vscf_curve25519_generate_ephemeral_key.restype = POINTER(vscf_impl_t)
        return vscf_curve25519_generate_ephemeral_key(ctx, key, error)

    def vscf_curve25519_import_public_key(self, ctx, raw_key, error):
        """Import public key from the raw binary format.

        Return public key that is adopted and optimized to be used
        with this particular algorithm.

        Binary format must be defined in the key specification.
        For instance, RSA public key must be imported from the format defined in
        RFC 3447 Appendix A.1.1."""
        vscf_curve25519_import_public_key = self._lib.vscf_curve25519_import_public_key
        vscf_curve25519_import_public_key.argtypes = [POINTER(vscf_curve25519_t), POINTER(vscf_raw_public_key_t), POINTER(vscf_error_t)]
        vscf_curve25519_import_public_key.restype = POINTER(vscf_impl_t)
        return vscf_curve25519_import_public_key(ctx, raw_key, error)

    def vscf_curve25519_export_public_key(self, ctx, public_key, error):
        """Export public key to the raw binary format.

        Binary format must be defined in the key specification.
        For instance, RSA public key must be exported in format defined in
        RFC 3447 Appendix A.1.1."""
        vscf_curve25519_export_public_key = self._lib.vscf_curve25519_export_public_key
        vscf_curve25519_export_public_key.argtypes = [POINTER(vscf_curve25519_t), POINTER(vscf_impl_t), POINTER(vscf_error_t)]
        vscf_curve25519_export_public_key.restype = POINTER(vscf_raw_public_key_t)
        return vscf_curve25519_export_public_key(ctx, public_key, error)

    def vscf_curve25519_import_private_key(self, ctx, raw_key, error):
        """Import private key from the raw binary format.

        Return private key that is adopted and optimized to be used
        with this particular algorithm.

        Binary format must be defined in the key specification.
        For instance, RSA private key must be imported from the format defined in
        RFC 3447 Appendix A.1.2."""
        vscf_curve25519_import_private_key = self._lib.vscf_curve25519_import_private_key
        vscf_curve25519_import_private_key.argtypes = [POINTER(vscf_curve25519_t), POINTER(vscf_raw_private_key_t), POINTER(vscf_error_t)]
        vscf_curve25519_import_private_key.restype = POINTER(vscf_impl_t)
        return vscf_curve25519_import_private_key(ctx, raw_key, error)

    def vscf_curve25519_export_private_key(self, ctx, private_key, error):
        """Export private key in the raw binary format.

        Binary format must be defined in the key specification.
        For instance, RSA private key must be exported in format defined in
        RFC 3447 Appendix A.1.2."""
        vscf_curve25519_export_private_key = self._lib.vscf_curve25519_export_private_key
        vscf_curve25519_export_private_key.argtypes = [POINTER(vscf_curve25519_t), POINTER(vscf_impl_t), POINTER(vscf_error_t)]
        vscf_curve25519_export_private_key.restype = POINTER(vscf_raw_private_key_t)
        return vscf_curve25519_export_private_key(ctx, private_key, error)

    def vscf_curve25519_can_encrypt(self, ctx, public_key, data_len):
        """Check if algorithm can encrypt data with a given key."""
        vscf_curve25519_can_encrypt = self._lib.vscf_curve25519_can_encrypt
        vscf_curve25519_can_encrypt.argtypes = [POINTER(vscf_curve25519_t), POINTER(vscf_impl_t), c_size_t]
        vscf_curve25519_can_encrypt.restype = c_bool
        return vscf_curve25519_can_encrypt(ctx, public_key, data_len)

    def vscf_curve25519_encrypted_len(self, ctx, public_key, data_len):
        """Calculate required buffer length to hold the encrypted data."""
        vscf_curve25519_encrypted_len = self._lib.vscf_curve25519_encrypted_len
        vscf_curve25519_encrypted_len.argtypes = [POINTER(vscf_curve25519_t), POINTER(vscf_impl_t), c_size_t]
        vscf_curve25519_encrypted_len.restype = c_size_t
        return vscf_curve25519_encrypted_len(ctx, public_key, data_len)

    def vscf_curve25519_encrypt(self, ctx, public_key, data, out):
        """Encrypt data with a given public key."""
        vscf_curve25519_encrypt = self._lib.vscf_curve25519_encrypt
        vscf_curve25519_encrypt.argtypes = [POINTER(vscf_curve25519_t), POINTER(vscf_impl_t), vsc_data_t, POINTER(vsc_buffer_t)]
        vscf_curve25519_encrypt.restype = c_int
        return vscf_curve25519_encrypt(ctx, public_key, data, out)

    def vscf_curve25519_can_decrypt(self, ctx, private_key, data_len):
        """Check if algorithm can decrypt data with a given key.
        However, success result of decryption is not guaranteed."""
        vscf_curve25519_can_decrypt = self._lib.vscf_curve25519_can_decrypt
        vscf_curve25519_can_decrypt.argtypes = [POINTER(vscf_curve25519_t), POINTER(vscf_impl_t), c_size_t]
        vscf_curve25519_can_decrypt.restype = c_bool
        return vscf_curve25519_can_decrypt(ctx, private_key, data_len)

    def vscf_curve25519_decrypted_len(self, ctx, private_key, data_len):
        """Calculate required buffer length to hold the decrypted data."""
        vscf_curve25519_decrypted_len = self._lib.vscf_curve25519_decrypted_len
        vscf_curve25519_decrypted_len.argtypes = [POINTER(vscf_curve25519_t), POINTER(vscf_impl_t), c_size_t]
        vscf_curve25519_decrypted_len.restype = c_size_t
        return vscf_curve25519_decrypted_len(ctx, private_key, data_len)

    def vscf_curve25519_decrypt(self, ctx, private_key, data, out):
        """Decrypt given data."""
        vscf_curve25519_decrypt = self._lib.vscf_curve25519_decrypt
        vscf_curve25519_decrypt.argtypes = [POINTER(vscf_curve25519_t), POINTER(vscf_impl_t), vsc_data_t, POINTER(vsc_buffer_t)]
        vscf_curve25519_decrypt.restype = c_int
        return vscf_curve25519_decrypt(ctx, private_key, data, out)

    def vscf_curve25519_compute_shared_key(self, ctx, public_key, private_key, shared_key):
        """Compute shared key for 2 asymmetric keys.
        Note, computed shared key can be used only within symmetric cryptography."""
        vscf_curve25519_compute_shared_key = self._lib.vscf_curve25519_compute_shared_key
        vscf_curve25519_compute_shared_key.argtypes = [POINTER(vscf_curve25519_t), POINTER(vscf_impl_t), POINTER(vscf_impl_t), POINTER(vsc_buffer_t)]
        vscf_curve25519_compute_shared_key.restype = c_int
        return vscf_curve25519_compute_shared_key(ctx, public_key, private_key, shared_key)

    def vscf_curve25519_shared_key_len(self, ctx, key):
        """Return number of bytes required to hold shared key.
        Expect Public Key or Private Key."""
        vscf_curve25519_shared_key_len = self._lib.vscf_curve25519_shared_key_len
        vscf_curve25519_shared_key_len.argtypes = [POINTER(vscf_curve25519_t), POINTER(vscf_impl_t)]
        vscf_curve25519_shared_key_len.restype = c_size_t
        return vscf_curve25519_shared_key_len(ctx, key)

    def vscf_curve25519_kem_shared_key_len(self, ctx, key):
        """Return length in bytes required to hold encapsulated shared key."""
        vscf_curve25519_kem_shared_key_len = self._lib.vscf_curve25519_kem_shared_key_len
        vscf_curve25519_kem_shared_key_len.argtypes = [POINTER(vscf_curve25519_t), POINTER(vscf_impl_t)]
        vscf_curve25519_kem_shared_key_len.restype = c_size_t
        return vscf_curve25519_kem_shared_key_len(ctx, key)

    def vscf_curve25519_kem_encapsulated_key_len(self, ctx, public_key):
        """Return length in bytes required to hold encapsulated key."""
        vscf_curve25519_kem_encapsulated_key_len = self._lib.vscf_curve25519_kem_encapsulated_key_len
        vscf_curve25519_kem_encapsulated_key_len.argtypes = [POINTER(vscf_curve25519_t), POINTER(vscf_impl_t)]
        vscf_curve25519_kem_encapsulated_key_len.restype = c_size_t
        return vscf_curve25519_kem_encapsulated_key_len(ctx, public_key)

    def vscf_curve25519_kem_encapsulate(self, ctx, public_key, shared_key, encapsulated_key):
        """Generate a shared key and a key encapsulated message."""
        vscf_curve25519_kem_encapsulate = self._lib.vscf_curve25519_kem_encapsulate
        vscf_curve25519_kem_encapsulate.argtypes = [POINTER(vscf_curve25519_t), POINTER(vscf_impl_t), POINTER(vsc_buffer_t), POINTER(vsc_buffer_t)]
        vscf_curve25519_kem_encapsulate.restype = c_int
        return vscf_curve25519_kem_encapsulate(ctx, public_key, shared_key, encapsulated_key)

    def vscf_curve25519_kem_decapsulate(self, ctx, encapsulated_key, private_key, shared_key):
        """Decapsulate the shared key."""
        vscf_curve25519_kem_decapsulate = self._lib.vscf_curve25519_kem_decapsulate
        vscf_curve25519_kem_decapsulate.argtypes = [POINTER(vscf_curve25519_t), vsc_data_t, POINTER(vscf_impl_t), POINTER(vsc_buffer_t)]
        vscf_curve25519_kem_decapsulate.restype = c_int
        return vscf_curve25519_kem_decapsulate(ctx, encapsulated_key, private_key, shared_key)

    def vscf_curve25519_setup_defaults(self, ctx):
        """Setup predefined values to the uninitialized class dependencies."""
        vscf_curve25519_setup_defaults = self._lib.vscf_curve25519_setup_defaults
        vscf_curve25519_setup_defaults.argtypes = [POINTER(vscf_curve25519_t)]
        vscf_curve25519_setup_defaults.restype = c_int
        return vscf_curve25519_setup_defaults(ctx)

    def vscf_curve25519_generate_key(self, ctx, error):
        """Generate new private key.
        Note, this operation might be slow."""
        vscf_curve25519_generate_key = self._lib.vscf_curve25519_generate_key
        vscf_curve25519_generate_key.argtypes = [POINTER(vscf_curve25519_t), POINTER(vscf_error_t)]
        vscf_curve25519_generate_key.restype = POINTER(vscf_impl_t)
        return vscf_curve25519_generate_key(ctx, error)

    def vscf_curve25519_shallow_copy(self, ctx):
        vscf_curve25519_shallow_copy = self._lib.vscf_curve25519_shallow_copy
        vscf_curve25519_shallow_copy.argtypes = [POINTER(vscf_curve25519_t)]
        vscf_curve25519_shallow_copy.restype = POINTER(vscf_curve25519_t)
        return vscf_curve25519_shallow_copy(ctx)

    def vscf_curve25519_impl(self, ctx):
        vscf_curve25519_impl = self._lib.vscf_curve25519_impl
        vscf_curve25519_impl.argtypes = [POINTER(vscf_curve25519_t)]
        vscf_curve25519_impl.restype = POINTER(vscf_impl_t)
        return vscf_curve25519_impl(ctx)
