37 lines
No EOL
1.1 KiB
Rust
37 lines
No EOL
1.1 KiB
Rust
use std::io::Write;
|
|
use std::net::SocketAddrV4;
|
|
use hmac::{Hmac};
|
|
use md5::digest::Mac;
|
|
use macros::RmcSerialize;
|
|
use rnex_core::prudp::virtual_port::VirtualPort;
|
|
|
|
type Md5Hmac = Hmac<md5::Md5>;
|
|
|
|
#[derive(Eq, PartialEq, Hash, Debug, Copy, Clone, Ord, PartialOrd, RmcSerialize)]
|
|
#[rmc_struct(0)]
|
|
pub struct PRUDPSockAddr{
|
|
pub regular_socket_addr: SocketAddrV4,
|
|
pub virtual_port: VirtualPort
|
|
}
|
|
|
|
|
|
|
|
impl PRUDPSockAddr{
|
|
pub fn new(regular_socket_addr: SocketAddrV4, virtual_port: VirtualPort) -> Self{
|
|
Self{
|
|
regular_socket_addr,
|
|
virtual_port
|
|
}
|
|
}
|
|
|
|
pub fn calculate_connection_signature(&self) -> [u8; 16] {
|
|
let mut hmac = Md5Hmac::new_from_slice(&[0; 16]).expect("fuck");
|
|
|
|
let data = self.regular_socket_addr.ip().octets().to_vec();
|
|
//data.extend_from_slice(&self.regular_socket_addr.port().to_be_bytes());
|
|
|
|
hmac.write_all(&data).expect("figuring this out was complete ass");
|
|
let result: [u8; 16] = hmac.finalize().into_bytes()[0..16].try_into().expect("fuck");
|
|
result
|
|
}
|
|
} |