Archives/Alfis
2
0
Fork 0
mirror of https://github.com/Revertron/Alfis synced 2024-11-17 03:25:43 +00:00
Code Issues Projects Releases Wiki Activity

Fixed DNS answers for provided zones.

Browse Source
This commit is contained in:
Revertron 2021-03-30 19:10:26 +02:00
parent bbcfa102b1
commit 7e742c1553
1 changed files with 28 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
src/blockchain/filter.rs
View File

@ -1,10 +1,11 @@
use crate::Context; use crate::Context;
use std::sync::{Mutex, Arc}; use std::sync::{Mutex, Arc};
use crate::dns::filter::DnsFilter; use crate::dns::filter::DnsFilter;
use crate::dns::protocol::{DnsPacket, QueryType, DnsRecord, DnsQuestion, ResultCode}; use crate::dns::protocol::{DnsPacket, QueryType, DnsRecord, DnsQuestion, ResultCode, TransientTtl};
#[allow(unused_imports)] #[allow(unused_imports)]
use log::{trace, debug, info, warn, error}; use log::{trace, debug, info, warn, error};
use crate::blockchain::transaction::DomainData; use crate::blockchain::transaction::DomainData;
use chrono::Utc;
pub struct BlockchainFilter { pub struct BlockchainFilter {
context: Arc<Mutex<Context>> context: Arc<Mutex<Context>>
@ -16,6 +17,9 @@ impl BlockchainFilter {
} }
} }
const NAME_SERVER: & str = "ns.alfis.name";
const SERVER_ADMIN: & str = "admin.alfis.name";
impl DnsFilter for BlockchainFilter { impl DnsFilter for BlockchainFilter {
fn lookup(&self, qname: &str, qtype: QueryType) -> Option<DnsPacket> { fn lookup(&self, qname: &str, qtype: QueryType) -> Option<DnsPacket> {
let search; let search;
@ -35,14 +39,16 @@ impl DnsFilter for BlockchainFilter {
debug!("Searching record type '{:?}', name '{}' for domain '{}'", &qtype, &subdomain, &search); debug!("Searching record type '{:?}', name '{}' for domain '{}'", &qtype, &subdomain, &search);
let data = self.context.lock().unwrap().chain.get_domain_info(&search); let data = self.context.lock().unwrap().chain.get_domain_info(&search);
let zone = parts[0].to_owned();
match data { match data {
None => { None => {
debug!("Not found data for domain {}", &search); debug!("Not found data for domain {}", &search);
if self.context.lock().unwrap().chain.is_zone_in_blockchain(parts[0]) { if self.context.lock().unwrap().chain.is_zone_in_blockchain(&zone) {
// Create DnsPacket // Create DnsPacket
let mut packet = DnsPacket::new(); let mut packet = DnsPacket::new();
packet.questions.push(DnsQuestion::new(String::from(qname), qtype)); packet.questions.push(DnsQuestion::new(String::from(qname), qtype));
packet.header.rescode = ResultCode::SERVFAIL; packet.header.rescode = ResultCode::NXDOMAIN;
packet.header.authoritative_answer = true;
//trace!("Returning packet: {:?}", &packet); //trace!("Returning packet: {:?}", &packet);
return Some(packet); return Some(packet);
} }
@ -133,17 +139,35 @@ impl DnsFilter for BlockchainFilter {
return if !answers.is_empty() { return if !answers.is_empty() {
// Create DnsPacket // Create DnsPacket
let mut packet = DnsPacket::new(); let mut packet = DnsPacket::new();
packet.header.authoritative_answer = true;
packet.questions.push(DnsQuestion::new(String::from(qname), qtype)); packet.questions.push(DnsQuestion::new(String::from(qname), qtype));
for answer in answers { for answer in answers {
packet.answers.push(answer); packet.answers.push(answer);
} }
packet.authorities.push( DnsRecord::NS {
domain: zone,
host: String::from(NAME_SERVER),
ttl: TransientTtl(600)
});
//trace!("Returning packet: {:?}", &packet); //trace!("Returning packet: {:?}", &packet);
Some(packet) Some(packet)
} else { } else {
// Create DnsPacket // Create DnsPacket
let mut packet = DnsPacket::new(); let mut packet = DnsPacket::new();
packet.header.authoritative_answer = true;
packet.header.rescode = ResultCode::NXDOMAIN;
packet.questions.push(DnsQuestion::new(String::from(qname), qtype)); packet.questions.push(DnsQuestion::new(String::from(qname), qtype));
packet.header.rescode = ResultCode::SERVFAIL; packet.authorities.push(DnsRecord::SOA {
domain: zone,
m_name: String::from(NAME_SERVER),
r_name: String::from(SERVER_ADMIN),
serial: Utc::now().timestamp() as u32,
refresh: 3600,
retry: 300,
expire: 604800,
minimum: 60,
ttl: TransientTtl(600),
});
//trace!("Returning packet: {:?}", &packet); //trace!("Returning packet: {:?}", &packet);
Some(packet) Some(packet)
} }