From 5a516a195c300764e5aa8c7f1c56d9b2eb7dcf53 Mon Sep 17 00:00:00 2001 From: Luna Yao <40349250+ZnqbuZ@users.noreply.github.com> Date: Sat, 18 Apr 2026 22:01:24 +0200 Subject: [PATCH] move common::dns to utils::dns move dns move --- easytier/src/common/mod.rs | 1 - easytier/src/common/stun.rs | 2 +- easytier/src/connector/direct.rs | 6 ++---- easytier/src/connector/dns_connector.rs | 4 ++-- easytier/src/connector/manual.rs | 6 +++--- easytier/src/dns/zone.rs | 2 +- easytier/src/tunnel/mod.rs | 6 ++---- easytier/src/{common => utils}/dns.rs | 0 easytier/src/utils/mod.rs | 1 + 9 files changed, 12 insertions(+), 16 deletions(-) rename easytier/src/{common => utils}/dns.rs (100%) diff --git a/easytier/src/common/mod.rs b/easytier/src/common/mod.rs index 1b8e303c..b81599a0 100644 --- a/easytier/src/common/mod.rs +++ b/easytier/src/common/mod.rs @@ -15,7 +15,6 @@ pub mod compressor; pub mod config; pub mod constants; pub mod defer; -pub mod dns; pub mod env_parser; pub mod error; pub mod global_ctx; diff --git a/easytier/src/common/stun.rs b/easytier/src/common/stun.rs index 94a36479..f2f16b23 100644 --- a/easytier/src/common/stun.rs +++ b/easytier/src/common/stun.rs @@ -22,7 +22,7 @@ use stun_codec::{Message, MessageClass, MessageDecoder, MessageEncoder}; use crate::common::error::Error; -use super::dns::resolve_txt_record; +use crate::utils::dns::resolve_txt_record; use super::stun_codec_ext::*; const DEFAULT_UDP_STUN_SERVERS: &[&str] = &[ diff --git a/easytier/src/connector/direct.rs b/easytier/src/connector/direct.rs index 6a5b63e1..6ae85eb2 100644 --- a/easytier/src/connector/direct.rs +++ b/easytier/src/connector/direct.rs @@ -12,10 +12,7 @@ use std::{ }; use crate::{ - common::{ - PeerId, dns::socket_addrs, error::Error, global_ctx::ArcGlobalCtx, - stun::StunInfoCollectorTrait, - }, + common::{PeerId, error::Error, global_ctx::ArcGlobalCtx, stun::StunInfoCollectorTrait}, connector::udp_hole_punch::handle_rpc_result, peers::{ peer_conn::PeerConnId, @@ -40,6 +37,7 @@ use super::{ udp_hole_punch, }; use crate::tunnel::{FromUrl, IpScheme, TunnelScheme, matches_scheme}; +use crate::utils::dns::socket_addrs; use anyhow::Context; use rand::Rng; use socket2::Protocol; diff --git a/easytier/src/connector/dns_connector.rs b/easytier/src/connector/dns_connector.rs index 59c289a6..f4fb357a 100644 --- a/easytier/src/connector/dns_connector.rs +++ b/easytier/src/connector/dns_connector.rs @@ -3,7 +3,6 @@ use std::{net::SocketAddr, sync::Arc}; use super::{create_connector_by_url, http_connector::TunnelWithInfo}; use crate::{ common::{ - dns::{RESOLVER, resolve_txt_record}, error::Error, global_ctx::ArcGlobalCtx, log, @@ -15,8 +14,9 @@ use anyhow::Context; use dashmap::DashSet; use hickory_proto::rr::RData; use hickory_resolver::proto::rr::rdata::SRV; -use rand::{Rng as _, seq::SliceRandom}; +use rand::{seq::SliceRandom, Rng as _}; use strum::VariantArray; +use crate::utils::dns::{resolve_txt_record, RESOLVER}; fn weighted_choice(options: &[(T, u64)]) -> Option<&T> { let total_weight = options.iter().map(|(_, weight)| *weight).sum(); diff --git a/easytier/src/connector/manual.rs b/easytier/src/connector/manual.rs index 8e9fd902..b4a0ef4d 100644 --- a/easytier/src/connector/manual.rs +++ b/easytier/src/connector/manual.rs @@ -7,7 +7,7 @@ use dashmap::DashSet; use tokio::{sync::mpsc, task::JoinSet, time::timeout}; use crate::{ - common::{PeerId, dns::socket_addrs, join_joinset_background}, + common::{PeerId, join_joinset_background}, peers::peer_conn::PeerConnId, proto::{ api::instance::{ @@ -20,6 +20,8 @@ use crate::{ utils::weak_upgrade, }; +use super::create_connector_by_url; +use crate::utils::dns::socket_addrs; use crate::{ common::{ error::Error, @@ -30,8 +32,6 @@ use crate::{ use_global_var, }; -use super::create_connector_by_url; - type ConnectorMap = Arc>; #[derive(Debug, Clone)] diff --git a/easytier/src/dns/zone.rs b/easytier/src/dns/zone.rs index 3cdd6eb6..42f92cac 100644 --- a/easytier/src/dns/zone.rs +++ b/easytier/src/dns/zone.rs @@ -1,4 +1,4 @@ -use crate::common::dns::get_default_resolver_config; +use crate::utils::dns::get_default_resolver_config; use crate::dns::utils::addr::{NameServerAddr, NameServerAddrGroup}; use crate::dns::utils::zone_handler::{ArcZoneHandler, ChainedZoneHandler}; use crate::proto; diff --git a/easytier/src/tunnel/mod.rs b/easytier/src/tunnel/mod.rs index 36feb753..f8bdcaa9 100644 --- a/easytier/src/tunnel/mod.rs +++ b/easytier/src/tunnel/mod.rs @@ -2,10 +2,7 @@ use std::{ collections::hash_map::DefaultHasher, hash::Hasher, net::SocketAddr, pin::Pin, sync::Arc, }; -use crate::{ - common::{dns::socket_addrs, error::Error}, - proto::common::TunnelInfo, -}; +use crate::{common::error::Error, proto::common::TunnelInfo}; use async_trait::async_trait; use derive_more::{From, TryInto}; use futures::{Sink, Stream}; @@ -392,4 +389,5 @@ macro_rules! __matches_protocol__ { }; } +use crate::utils::dns::socket_addrs; pub(crate) use __matches_protocol__ as matches_protocol; diff --git a/easytier/src/common/dns.rs b/easytier/src/utils/dns.rs similarity index 100% rename from easytier/src/common/dns.rs rename to easytier/src/utils/dns.rs diff --git a/easytier/src/utils/mod.rs b/easytier/src/utils/mod.rs index 1b7c07c7..0b2a6ad9 100644 --- a/easytier/src/utils/mod.rs +++ b/easytier/src/utils/mod.rs @@ -1,4 +1,5 @@ pub mod dirty; +pub mod dns; pub mod panic; pub mod string; pub mod task;