zydronium/internetee-registry
1
0
Fork 0
mirror of https://github.com/internetee/registry.git synced 2025-06-10 06:34:46 +02:00
Code Issues Projects Releases Packages Wiki Activity

Add creation of csync inputs based on name_puny

Browse source
This commit is contained in:
Alex Sherman 2021-05-25 13:28:16 +05:00
parent 82c048b316
commit d887bcd734
3 changed files with 39 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

19
app/jobs/csync_job.rb
View file

@ -83,9 +83,9 @@ class CsyncJob < ApplicationJob
# From this point we're working on generating input for cdnskey-scanner
def gather_pollable_domains
@logger.info 'CsyncJob Generate: Gathering current domain(s) data'
Nameserver.select(:hostname, :domain_id).all.each do |ns|
Nameserver.select(:hostname_puny, :domain_id).all.each do |ns|
%i[secure insecure].each do |i|
@input_store[i][ns.hostname] = [] unless @input_store[i].key? ns.hostname
@input_store[i][ns.hostname_puny] = [] unless @input_store[i].key? ns.hostname_puny
end
append_domains_to_list(ns)
@ -94,14 +94,16 @@ class CsyncJob < ApplicationJob
def append_domains_to_list(nameserver)
Domain.where(id: nameserver.domain_id).all.each do |domain|
@input_store[domain.dnskeys.any? ? :secure : :insecure][nameserver.hostname].push domain.name
key = domain.dnskeys.any? ? :secure : :insecure
hostname = nameserver.hostname_puny || nameserver.hostname
@input_store[key][hostname].push domain.name_puny
end
end
def generate_scanner_input
@logger.info 'CsyncJob Generate: Gathering current domain(s) data'
gather_pollable_domains
check_directory
out_file = File.new(ENV['cdns_scanner_input_file'], 'w+')
%i[secure insecure].each do |state|
@ -113,6 +115,15 @@ class CsyncJob < ApplicationJob
@logger.info 'CsyncJob Generate: Finished writing output to ' + ENV['cdns_scanner_input_file']
end
def check_directory
dirname = File.dirname(ENV['cdns_scanner_input_file'])
FileUtils.mkdir_p(dirname) unless File.directory?(dirname)
return if File.exist?(ENV['cdns_scanner_input_file'])
FileUtils.touch(ENV['cdns_scanner_input_file'])
end
def create_input_lines(out_file, state)
@input_store[state].keys.each do |nameserver|
domains = @input_store[state][nameserver].join(' ')