rhoast_utils 0.1.0

The Rho vision utility library
Documentation 
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
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93

use crate::error::Error;
use bitcoin_hashes::{sha256, Hash};
use hex::FromHex;
use secp256k1::rand::thread_rng;
use secp256k1::{
    constants::SECRET_KEY_SIZE, ecdsa, Message, PublicKey, Secp256k1, SecretKey, Signing,
    Verification,
};

//use secret gotten from get_pri_pub_key_pair() to create new secret here
/// Get public key from a private key
///
/// ```ignore
/// use rhoast_utils::pub_from_private::{get_seckey_from_string,get_pub_key};
/// let seckey = get_seckey_from_string("secret key").unwrap();
/// //get pub key
/// let pub_key=get_pub_key(&seckey);
/// ```
///
pub fn get_pub_key(secret_key: &SecretKey) -> String {
    let secp = Secp256k1::new();

    let key = PublicKey::from_secret_key(&secp, secret_key);
    hex::encode(key.serialize_uncompressed())
}
/// Generate private key
/// ```ignore
/// use rhoast_utils::pub_from_private::get_pri_key;
/// let pri_key=get_pri_key();
/// //To get string value
/// let pri_string= pri_key.display_secret().to_string();
/// ````
///
pub fn get_pri_key() -> SecretKey {
    let secp = Secp256k1::new();
    let mut rng = thread_rng();
    let (seckey, _) = secp.generate_keypair(&mut rng);
    seckey
}

pub fn get_seckey_from_string(input: &str) -> Result<SecretKey, Error> {
    match <[u8; SECRET_KEY_SIZE]>::from_hex(input) {
        Ok(buffer) => match SecretKey::from_slice(&buffer) {
            Ok(sec_key) => return Ok(sec_key),
            Err(_) => Err(Error::PubFromPrivate("error writing slice to secret key ")),
        },
        Err(_) => Err(Error::PubFromPrivate("error writing hex to buffer ")),
    }
}

pub fn recover<C: Verification>(
    secp: &Secp256k1<C>,
    msg: &[u8],
    sig: &[u8],
    recovery_id: u8,
) -> Result<PublicKey, Error> {
    let msg = sha256::Hash::hash(msg);
    match Message::from_slice(&msg) {
        Ok(msg) => match ecdsa::RecoveryId::from_i32(recovery_id as i32) {
            Ok(id) => match ecdsa::RecoverableSignature::from_compact(&sig, id) {
                Ok(sig) => {
                    if let Ok(pub_key) = secp.recover_ecdsa(&msg, &sig) {
                        Ok(pub_key)
                    } else {
                        Err(Error::PubFromPrivate("error getting pub key "))
                    }
                }
                Err(_) => Err(Error::PubFromPrivate(
                    "error converting compact-encoded byte slice to a signature ",
                )),
            },
            Err(_) => return Err(Error::PubFromPrivate("error creating recovery id ")),
        },
        Err(_) => return Err(Error::PubFromPrivate("error getting message from slice ")),
    }
}

pub fn sign_recovery<C: Signing>(
    secp: &Secp256k1<C>,
    msg: &[u8],
    seckey: &[u8],
) -> Result<ecdsa::RecoverableSignature, Error> {
    let msg = sha256::Hash::hash(msg);
    match Message::from_slice(&msg) {
        Ok(msg) => match SecretKey::from_slice(&seckey) {
            Ok(seckey) => return Ok(secp.sign_ecdsa_recoverable(&msg, &seckey)),
            Err(_) => Err(Error::PubFromPrivate(
                "error getting secret key from slice ",
            )),
        },
        Err(_) => return Err(Error::PubFromPrivate("error getting message from slice ")),
    }
}