1use crate::{EcParameters, Error, Result};
9use core::fmt;
10use der::{
11    asn1::{BitStringRef, ContextSpecific, ContextSpecificRef, OctetStringRef},
12    Decode, DecodeValue, Encode, EncodeValue, Header, Length, Reader, Sequence, Tag, TagMode,
13    TagNumber, Writer,
14};
15
16#[cfg(all(feature = "alloc", feature = "zeroize"))]
17use der::SecretDocument;
18
19#[cfg(feature = "pem")]
20use der::pem::PemLabel;
21
22const VERSION: u8 = 1;
32
33const EC_PARAMETERS_TAG: TagNumber = TagNumber::new(0);
35
36const PUBLIC_KEY_TAG: TagNumber = TagNumber::new(1);
38
39#[derive(Clone)]
62pub struct EcPrivateKey<'a> {
63    pub private_key: &'a [u8],
65
66    pub parameters: Option<EcParameters>,
68
69    pub public_key: Option<&'a [u8]>,
71}
72
73impl<'a> EcPrivateKey<'a> {
74    fn context_specific_parameters(&self) -> Option<ContextSpecificRef<'_, EcParameters>> {
75        self.parameters.as_ref().map(|params| ContextSpecificRef {
76            tag_number: EC_PARAMETERS_TAG,
77            tag_mode: TagMode::Explicit,
78            value: params,
79        })
80    }
81
82    fn context_specific_public_key(
83        &self,
84    ) -> der::Result<Option<ContextSpecific<BitStringRef<'a>>>> {
85        self.public_key
86            .map(|pk| {
87                BitStringRef::from_bytes(pk).map(|value| ContextSpecific {
88                    tag_number: PUBLIC_KEY_TAG,
89                    tag_mode: TagMode::Explicit,
90                    value,
91                })
92            })
93            .transpose()
94    }
95}
96
97impl<'a> DecodeValue<'a> for EcPrivateKey<'a> {
98    fn decode_value<R: Reader<'a>>(reader: &mut R, header: Header) -> der::Result<Self> {
99        reader.read_nested(header.length, |reader| {
100            if u8::decode(reader)? != VERSION {
101                return Err(der::Tag::Integer.value_error());
102            }
103
104            let private_key = OctetStringRef::decode(reader)?.as_bytes();
105            let parameters = reader.context_specific(EC_PARAMETERS_TAG, TagMode::Explicit)?;
106            let public_key = reader
107                .context_specific::<BitStringRef<'_>>(PUBLIC_KEY_TAG, TagMode::Explicit)?
108                .map(|bs| bs.as_bytes().ok_or_else(|| Tag::BitString.value_error()))
109                .transpose()?;
110
111            Ok(EcPrivateKey {
112                private_key,
113                parameters,
114                public_key,
115            })
116        })
117    }
118}
119
120impl EncodeValue for EcPrivateKey<'_> {
121    fn value_len(&self) -> der::Result<Length> {
122        VERSION.encoded_len()?
123            + OctetStringRef::new(self.private_key)?.encoded_len()?
124            + self.context_specific_parameters().encoded_len()?
125            + self.context_specific_public_key()?.encoded_len()?
126    }
127
128    fn encode_value(&self, writer: &mut impl Writer) -> der::Result<()> {
129        VERSION.encode(writer)?;
130        OctetStringRef::new(self.private_key)?.encode(writer)?;
131        self.context_specific_parameters().encode(writer)?;
132        self.context_specific_public_key()?.encode(writer)?;
133        Ok(())
134    }
135}
136
137impl<'a> Sequence<'a> for EcPrivateKey<'a> {}
138
139impl<'a> TryFrom<&'a [u8]> for EcPrivateKey<'a> {
140    type Error = Error;
141
142    fn try_from(bytes: &'a [u8]) -> Result<EcPrivateKey<'a>> {
143        Ok(Self::from_der(bytes)?)
144    }
145}
146
147impl<'a> fmt::Debug for EcPrivateKey<'a> {
148    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
149        f.debug_struct("EcPrivateKey")
150            .field("parameters", &self.parameters)
151            .field("public_key", &self.public_key)
152            .finish_non_exhaustive()
153    }
154}
155
156#[cfg(feature = "alloc")]
157impl TryFrom<EcPrivateKey<'_>> for SecretDocument {
158    type Error = Error;
159
160    fn try_from(private_key: EcPrivateKey<'_>) -> Result<Self> {
161        SecretDocument::try_from(&private_key)
162    }
163}
164
165#[cfg(feature = "alloc")]
166impl TryFrom<&EcPrivateKey<'_>> for SecretDocument {
167    type Error = Error;
168
169    fn try_from(private_key: &EcPrivateKey<'_>) -> Result<Self> {
170        Ok(Self::encode_msg(private_key)?)
171    }
172}
173
174#[cfg(feature = "pem")]
175impl PemLabel for EcPrivateKey<'_> {
176    const PEM_LABEL: &'static str = "EC PRIVATE KEY";
177}