libcrux/hacl/
curve25519.rs

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
use libcrux_hacl::{Hacl_Curve25519_51_ecdh, Hacl_Curve25519_51_secret_to_public};

#[derive(Debug, PartialEq, Eq, Clone, Copy)]
pub enum Error {
    InvalidInput,
}

/// Compute the ECDH with the `private_key` and `public_key`.
///
/// Returns the 32 bytes shared key.

#[must_use]
#[inline(always)]
pub fn ecdh(
    private_key: impl AsRef<[u8; 32]>,
    public_key: impl AsRef<[u8; 32]>,
) -> Result<[u8; 32], Error> {
    let mut shared = [0u8; 32];
    let ok = unsafe {
        Hacl_Curve25519_51_ecdh(
            shared.as_mut_ptr(),
            private_key.as_ref().as_ptr() as _,
            public_key.as_ref().as_ptr() as _,
        )
    };
    if !ok {
        Err(Error::InvalidInput)
    } else {
        Ok(shared)
    }
}

/// Compute the public key for the provided `private_key` (scalar multiplication
/// with the base point).
///
/// Returns the 32 bytes shared key.

#[must_use]
#[inline(always)]
pub fn secret_to_public(private_key: impl AsRef<[u8; 32]>) -> [u8; 32] {
    let mut public = [0u8; 32];
    unsafe {
        Hacl_Curve25519_51_secret_to_public(public.as_mut_ptr(), private_key.as_ref().as_ptr() as _)
    };
    public
}

#[cfg(all(bmi2, adx, target_arch = "x86_64"))]
pub mod vale {
    use super::Error;
    use libcrux_hacl::Hacl_Curve25519_64_ecdh;

    /// Compute the ECDH with the `private_key` and `public_key`.
    ///
    /// Returns the 32 bytes shared key.

    #[must_use]
    #[inline(always)]
    pub fn ecdh(
        private_key: impl AsRef<[u8; 32]>,
        public_key: impl AsRef<[u8; 32]>,
    ) -> Result<[u8; 32], Error> {
        let mut shared = [0u8; 32];
        let ok = unsafe {
            Hacl_Curve25519_64_ecdh(
                shared.as_mut_ptr(),
                private_key.as_ref().as_ptr() as _,
                public_key.as_ref().as_ptr() as _,
            )
        };
        if !ok {
            Err(Error::InvalidInput)
        } else {
            Ok(shared)
        }
    }
}