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

// -*- mode: rust; -*-
//
// This file is part of subhd/solo, substrate.
// Copyright (c) 2017-2019 Chester Li and extropies.com
// See LICENSE for licensing information.
//
// Authors:
// - Chester Li<chester@lichester.com>

//! C API to RUST for solo
extern {
    fn solo_pubkey( pk: *mut u8, path:*const u8, path_level:usize ) -> u32;
    fn solo_sign( message: *const u8, len: usize, path:*const u8, path_level:usize, sig: *mut u8 ) -> u32;
    fn solo_generate( seed:*mut u8 ) -> u32;
    fn solo_format() -> u32;
    fn solo_import( seed:*const u8, len:usize ) -> u32;
}


/// get pulic key by path
pub fn wk_getpub( pk: &mut [u8], path:&Vec<&[u8;32]>) -> u32 {
    if !path.is_empty(){
        let v_bytes: &[u8] = unsafe {
            std::slice::from_raw_parts(
                path.get(0).unwrap().as_ptr(),
                path.len() * 32,
            )
        };
         unsafe { solo_pubkey(pk.as_mut_ptr(),v_bytes.as_ptr(),path.len()*32) }
    }else{
        let empty:[u8;32] = [0u8;32];
         unsafe { solo_pubkey(pk.as_mut_ptr(),empty.as_ptr(),0) }
    }
}
/// sign the message
pub fn wk_sign( message:&[u8], sig: &mut [u8], path:&Vec<&[u8;32]>) -> u32 {
    if !path.is_empty(){
        let v_bytes: &[u8] = unsafe {
            std::slice::from_raw_parts(
                path.get(0).unwrap().as_ptr(),
                path.len() * 32,
            )
        };
        unsafe { solo_sign( message.as_ptr(), message.len(), v_bytes.as_ptr(), path.len() * 32, sig.as_mut_ptr()) }
    }else{
        let empty:[u8;32] = [0u8;32];
        unsafe { solo_sign( message.as_ptr(), message.len(), empty.as_ptr(), 0, sig.as_mut_ptr()) }
    }
        
}
    
/// gen key paird and return seed
pub fn wk_generate( seed: &mut [u8] ) -> u32 {
    unsafe { solo_generate( seed.as_mut_ptr() ) }
}
///clear all data
pub fn wk_format() -> u32{
    unsafe { solo_format() }
}
///import seed to device
pub fn wk_import( seed:&[u8] ) -> u32 {
     unsafe { solo_import( seed.as_ptr(), seed.len() ) }
}