Commit 6bcade87 authored by David Goulet's avatar David Goulet
Browse files

Minor refactor 



Signed-off-by: default avatarDavid Goulet <dgoulet@ev0ke.net>
parent 918adf80

crates/onion-tunnel/src/dns.rs

+4 −4
Original line number Diff line number Diff line
@@ -79,7 +79,7 @@ impl DnsManager { 
        self.cookies.clone()

    }



    fn build_answer(&mut self, query: &dns_message_parser::Dns) -> dns_message_parser::Dns {

    fn build_answer(&self, query: &dns_message_parser::Dns) -> dns_message_parser::Dns {

        let q = query.questions.get(0).unwrap();

        let cookie = self

            .cookies
@@ -113,7 +113,7 @@ impl DnsManager { 
        }

    }



    fn decode_and_answer(&mut self, payload: Bytes) -> Option<dns_message_parser::Dns> {

    fn decode(&self, payload: Bytes) -> Option<dns_message_parser::Dns> {

        let ret = dns_message_parser::Dns::decode(payload);

        match ret {

            Ok(p) => {
@@ -134,7 +134,7 @@ impl DnsManager { 
            tokio::select! {

                r = self.listener_udp_v4.next() => match r {

                    Some((payload, endpoint)) => {

                        if let Some(reply) = self.decode_and_answer(payload) {

                        if let Some(reply) = self.decode(payload) {

                            let _ = self.listener_udp_v4.send(reply.encode().unwrap().as_ref(), endpoint);

                        }

                    }
@@ -142,7 +142,7 @@ impl DnsManager { 
                },

                r = self.listener_udp_v6.next() => match r {

                    Some((payload, endpoint)) => {

                        if let Some(reply) = self.decode_and_answer(payload) {

                        if let Some(reply) = self.decode(payload) {

                            let _ = self.listener_udp_v6.send(reply.encode().unwrap().as_ref(), endpoint);

                        }

                    }