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
use crate::chain::address::Address;
use crate::sigma_protocol::private_input::{DlogProverInput, PrivateInput};
#[derive(PartialEq, Debug, Clone)]
pub enum SecretKey {
DlogSecretKey(DlogProverInput),
}
impl SecretKey {
pub fn random_dlog() -> SecretKey {
SecretKey::DlogSecretKey(DlogProverInput::random())
}
pub fn dlog_from_bytes(bytes: &[u8; DlogProverInput::SIZE_BYTES]) -> Option<SecretKey> {
DlogProverInput::from_bytes(bytes).map(SecretKey::DlogSecretKey)
}
pub fn get_address_from_public_image(&self) -> Address {
match self {
SecretKey::DlogSecretKey(dpi) => Address::P2PK(dpi.public_image()),
}
}
pub fn to_bytes(&self) -> Vec<u8> {
match self {
SecretKey::DlogSecretKey(key) => key.to_bytes().to_vec(),
}
}
}
impl From<SecretKey> for PrivateInput {
fn from(s: SecretKey) -> Self {
match s {
SecretKey::DlogSecretKey(dpi) => PrivateInput::DlogProverInput(dpi),
}
}
}
impl From<DlogProverInput> for SecretKey {
fn from(pi: DlogProverInput) -> Self {
SecretKey::DlogSecretKey(pi)
}
}
#[cfg(test)]
mod tests {
use super::*;
use std::convert::TryInto;
#[test]
fn dlog_roundtrip() {
let sk = SecretKey::random_dlog();
let sk_copy =
SecretKey::dlog_from_bytes(&sk.to_bytes().as_slice().try_into().unwrap()).unwrap();
assert_eq!(sk, sk_copy);
}
}