rustdi 0.1.0

Modern WhatsApp Client Library for Rust
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
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128

use std::collections::HashMap;
use serde::{Deserialize, Serialize};
use ring::rand::{SystemRandom, SecureRandom};

// Sesi koneksi WhatsApp
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct Session {
    pub client_id: String,
    pub server_token: String,
    pub client_token: String,
    pub enc_key: Vec<u8>,
    pub mac_key: Vec<u8>,
    pub wid: String,
    pub push_name: String,
    pub phone: Option<PhoneInfo>,
}

// Informasi perangkat pengguna
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct PhoneInfo {
    pubwa_version: String,
    pub mcc: String,
    pub mnc: String,
    pub os_version: String,
    pub device_manufacturer: String,
    pub device_model: String,
    pub os_build_number: String,
}

// Sesi multi-perangkat untuk WhatsApp
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct MultiDeviceSession {
    pub device_id: String,
    pub registration_id: u32,
    pub identity_id: Vec<u8>,
    pub identity_key_pair: KeyPair,
    pub signed_pre_key: SignedPreKey,
    pub one_time_keys: HashMap<u32, Key>,
    pub signed_device_keys: HashMap<String, Vec<u8>>,
    pub next_pre_key_id: u32,
    pub next_session_id: u32,
}

// Pasangan kunci kriptografi
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct KeyPair {
    pub public_key: Vec<u8>,
    pub private_key: Vec<u8>,
}

// Kunci pre-signed untuk pertukanan kunci
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct SignedPreKey {
    pub key_id: u32,
    pub public_key: Vec<u8>,
    pub signature: Vec<u8>,
    pub timestamp: u64,
}

// Kunci satu kali pakai
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct Key {
    pub key_id: u32,
    pub public_key: Vec<u8>,
}

// Informasi perangkat lain dalam sesi multi-perangkat
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct DeviceInfo {
    pub device_id: u32,
    pub jid: String,
    pub registration_id: u32,
    pub noise_key: KeyPair,
    pub identity_key: KeyPair,
    pub signed_pre_key: SignedPreKey,
    pub last_seen: Option<u64>,
}

impl Default for MultiDeviceSession {
    fn default() -> Self {
        let mut registration_id = [0u8; 2];
        SystemRandom::new().fill(&mut registration_id).unwrap();
        let reg_id = u32::from_le_bytes(registration_id);

        let (identity_private, identity_public) = crate::crypto::generate_keypair().unwrap();

        MultiDeviceSession {
            device_id: uuid::Uuid::new_v4().to_string(),
            registration_id: reg_id,
            identity_id: vec![0u8; 32],
            identity_key_pair: KeyPair {
                public_key: identity_public,
                private_key: vec![], // Didapatkan dari generate_keypair
            },
            signed_pre_key: SignedPreKey {
                key_id: 1,
                public_key: vec![],
                signature: vec![],
                timestamp: std::time::SystemTime::now()
                    .duration_since(std::time::UNIX_EPOCH)
                    .unwrap()
                    .as_secs(),
            },
            one_time_keys: HashMap::new(),
            signed_device_keys: HashMap::new(),
            next_pre_key_id: 1,
            next_session_id: 1,
        }
    }
}

impl Session {
    pub fn new() -> Self {
        let mut client_id = [0u8; 16];
        SystemRandom::new().fill(&mut client_id).unwrap();
        
        Session {
            client_id: base64::encode(&client_id),
            server_token: String::new(),
            client_token: String::new(),
            enc_key: vec![0; 32],
            mac_key: vec![0; 32],
            wid: String::new(),
            push_name: String::new(),
            phone: None,
        }
    }
}