53 lines
1.2 KiB
Rust
53 lines
1.2 KiB
Rust
use std::net::Ipv4Addr;
|
|
use chrono::{Timelike, Utc};
|
|
|
|
pub struct UDPVpnPacket {
|
|
pub nonce: Vec<u8>, // [u8; 12]
|
|
pub data: Vec<u8>
|
|
}
|
|
|
|
pub struct UDPKeepAlive {
|
|
|
|
}
|
|
|
|
impl UDPSerializable for UDPKeepAlive {
|
|
fn serialize(&self) -> Vec<u8> {
|
|
let h: &[u8] = &[2];
|
|
[h, &[Utc::now().second() as u8]].concat()
|
|
}
|
|
}
|
|
|
|
impl UDPSerializable for UDPVpnPacket {
|
|
fn serialize(&self) -> Vec<u8> {
|
|
let h: &[u8] = &[1];
|
|
[h, &self.nonce, &self.data[..]].concat()
|
|
}
|
|
}
|
|
|
|
impl UDPVpnPacket {
|
|
pub fn deserialize(data: &Vec<u8>) -> Self {
|
|
UDPVpnPacket { nonce: data[1..=12].to_vec(), data: data[13..].to_vec() }
|
|
}
|
|
}
|
|
|
|
pub struct UDPVpnHandshake {
|
|
pub public_key: Vec<u8>,
|
|
pub request_ip: Ipv4Addr // [u8; 4]
|
|
}
|
|
|
|
impl UDPSerializable for UDPVpnHandshake {
|
|
fn serialize(&self) -> Vec<u8> {
|
|
let h: &[u8] = &[0];
|
|
[h, &self.public_key[..], &self.request_ip.octets()].concat()
|
|
}
|
|
}
|
|
|
|
impl UDPVpnHandshake {
|
|
pub fn deserialize(data: &Vec<u8>) -> Self {
|
|
UDPVpnHandshake { public_key: data[1..=32].to_vec(), request_ip: Ipv4Addr::new(data[33], data[34], data[35], data[36]) }
|
|
}
|
|
}
|
|
|
|
pub trait UDPSerializable {
|
|
fn serialize(&self) -> Vec<u8>;
|
|
} |