From b6afe6305a86562631b80600b5686761ca3d7e02 Mon Sep 17 00:00:00 2001 From: Luna Yao <40349250+ZnqbuZ@users.noreply.github.com> Date: Fri, 3 Apr 2026 16:08:22 +0200 Subject: [PATCH] format --- easytier/src/dns/utils/response.rs | 12 ++++++------ easytier/src/dns/zone.rs | 29 ++++++----------------------- easytier/src/launcher.rs | 4 +++- 3 files changed, 15 insertions(+), 30 deletions(-) diff --git a/easytier/src/dns/utils/response.rs b/easytier/src/dns/utils/response.rs index 80a65c08..36cf590c 100644 --- a/easytier/src/dns/utils/response.rs +++ b/easytier/src/dns/utils/response.rs @@ -1,10 +1,10 @@ +use hickory_proto::rr::Record; +use hickory_proto::serialize::binary::BinEncoder; +use hickory_server::authority::MessageResponse; +use hickory_server::server::{ResponseHandler, ResponseInfo}; +use parking_lot::Mutex; use std::io; use std::sync::Arc; -use hickory_proto::rr::Record; -use parking_lot::Mutex; -use hickory_server::server::{ResponseHandler, ResponseInfo}; -use hickory_server::authority::MessageResponse; -use hickory_proto::serialize::binary::BinEncoder; // ResponseWrapper for serializing DNS responses into a byte buffer. // Used by the address hijacking NIC packet filter to produce DNS replies in-place. @@ -55,4 +55,4 @@ impl ResponseHandler for ResponseHandle { .destructive_emit(&mut encoder) .map_err(io::Error::other) } -} \ No newline at end of file +} diff --git a/easytier/src/dns/zone.rs b/easytier/src/dns/zone.rs index e1e8b822..e2ea54d2 100644 --- a/easytier/src/dns/zone.rs +++ b/easytier/src/dns/zone.rs @@ -194,7 +194,6 @@ mod tests { use super::*; use crate::common::log; use crate::dns::config::DnsConfig; - use crate::dns::server::DynamicCatalog; use crate::dns::utils::response::ResponseHandle; use hickory_client::client::{Client, ClientHandle}; use hickory_proto::op::{Message, MessageType, OpCode, Query, ResponseCode}; @@ -203,7 +202,7 @@ mod tests { use hickory_proto::serialize::binary::{BinDecodable, BinEncodable, BinEncoder}; use hickory_proto::udp::UdpClientStream; use hickory_proto::xfer::Protocol; - use hickory_server::authority::MessageRequest; + use hickory_server::authority::{Catalog, MessageRequest}; use hickory_server::server::Request; use hickory_server::ServerFuture; use std::net::{Ipv4Addr, SocketAddrV4}; @@ -259,7 +258,7 @@ mod tests { async fn test_config() -> anyhow::Result<()> { log::tests::init(); - let catalog = DynamicCatalog::new(); + let mut catalog = Catalog::new(); let sep = "=".repeat(80); let config = toml::from_str::(CONFIG)?; @@ -280,11 +279,7 @@ mod tests { let mut authorities = Vec::new(); authorities.extend(zone.create_memory_authority().into_iter()); authorities.extend(zone.create_forward_authority().into_iter()); - catalog - .inner - .write() - .await - .upsert(zone.origin.clone().into(), authorities); + catalog.upsert(zone.origin.clone().into(), authorities); let mut record = Record::update0(zone.origin.clone().into(), 60, RecordType::A); record.set_data(RData::A(rdata::a::A("100.100.100.100".parse()?))); @@ -346,21 +341,14 @@ mod tests { let mut authorities = Vec::new(); authorities.extend(zone.create_memory_authority().into_iter()); authorities.extend(zone.create_forward_authority().into_iter()); - catalog - .inner - .write() - .await - .upsert(zone.origin.clone().into(), authorities); + catalog.upsert(zone.origin.clone().into(), authorities); let mut query = Message::new(); query.set_id(0x1234); query.set_message_type(MessageType::Query); query.set_op_code(OpCode::Query); query.set_recursion_desired(true); - query.add_query(Query::query( - Name::from_ascii("et.top.")?, - RecordType::A, - )); + query.add_query(Query::query(Name::from_ascii("et.top.")?, RecordType::A)); let mut request = Vec::new(); let mut encoder = BinEncoder::new(&mut request); @@ -373,12 +361,7 @@ mod tests { ); let response = ResponseHandle::new(512); - let info = catalog - .inner - .read() - .await - .lookup(&request, None, response.clone()) - .await; + let info = catalog.lookup(&request, None, response.clone()).await; assert_eq!(info.response_code(), ResponseCode::NoError); diff --git a/easytier/src/launcher.rs b/easytier/src/launcher.rs index 78dc8ae3..4296ac1f 100644 --- a/easytier/src/launcher.rs +++ b/easytier/src/launcher.rs @@ -492,7 +492,9 @@ impl NetworkConfig { .parse() .with_context(|| format!("failed to parse instance id: {:?}", self.instance_id))?, ); - if let Some(n) = self.hostname.as_ref() { cfg.set_hostname(n) } + if let Some(n) = self.hostname.as_ref() { + cfg.set_hostname(n) + } cfg.set_dhcp(self.dhcp.unwrap_or_default()); cfg.set_inst_name(self.network_name.clone().unwrap_or_default());