luckyimnotanigger/tun2proxy
1
0
Fork 0
mirror of https://github.com/tun2proxy/tun2proxy.git synced 2025-06-23 09:30:54 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

Improve code elegance in DNS cache removal function

Browse source
This commit is contained in:
B. Blechschmidt 2023-03-25 22:23:46 +01:00
parent 58faf871c3
commit 33892e28dc
1 changed files with 11 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

22
src/virtdns.rs
View file

@ -1,7 +1,7 @@
use hashlink::linked_hash_map::RawEntryMut;
use hashlink::LruCache;
use smoltcp::wire::Ipv4Cidr;
use std::collections::{HashMap, LinkedList};
use std::collections::HashMap;
use std::convert::{TryFrom, TryInto};
use std::net::{IpAddr, Ipv4Addr, Ipv6Addr};
use std::str::FromStr;
@ -188,19 +188,19 @@ impl VirtualDns {
fn allocate_ip(&mut self, name: String) -> Option<IpAddr> {
let now = Instant::now();
// Building the to_remove list seems to be a bit clunky.
// But removing inside the loop does not immediately work due to borrow rules.
// TODO: Is there a better solution?
let mut to_remove = LinkedList::<IpAddr>::new();
for (ip, entry) in self.lru_cache.iter() {
loop {
let p = self.lru_cache.iter().next();
if p.is_none() {
break;
}
let (ip, entry) = p.unwrap();
if now > entry.expiry {
to_remove.push_back(*ip);
self.name_to_ip.remove(&entry.name);
let name = entry.name.clone();
self.lru_cache.remove(&ip.clone());
self.name_to_ip.remove(&name);
continue;
}
}
for ip in to_remove {
self.lru_cache.remove(&ip);
break;
}
if let Some(ip) = self.name_to_ip.get(&name) {