From 5f94c8d6a08aa8929036dea0db254c092297cec7 Mon Sep 17 00:00:00 2001 From: asxalex Date: Sat, 30 Mar 2024 16:16:19 +0800 Subject: [PATCH] net_bit_len_to_mask --- src/utils/myrsa.rs | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/src/utils/myrsa.rs b/src/utils/myrsa.rs index c5735dd..675c7c9 100644 --- a/src/utils/myrsa.rs +++ b/src/utils/myrsa.rs @@ -109,7 +109,7 @@ pub fn rsa_decrypt(privkey: &RsaPrivateKey, cipherd: &[u8]) -> Result> { #[cfg(test)] mod tests { use super::*; - use crate::utils::test_utils::generate_info; + use crate::utils::{ip_to_string, net_bit_len_to_mask, test_utils::generate_info}; fn generate_key() { if std::fs::File::open(".data/id_rsa").is_ok() { @@ -144,4 +144,12 @@ mod tests { Ok(()) } + + #[test] + fn test_net_bit_to_ip() { + assert_eq!(ip_to_string(net_bit_len_to_mask(8)), "255.0.0.0"); + assert_eq!(ip_to_string(net_bit_len_to_mask(16)), "255.255.0.0"); + assert_eq!(ip_to_string(net_bit_len_to_mask(24)), "255.255.255.0"); + assert_eq!(ip_to_string(net_bit_len_to_mask(32)), "255.255.255.255"); + } }