root/lib/proxy/dns/dnscmd.rb @ 168ee4e6
168ee4e6 | Martin Matuska | require "proxy/dns"
|
|
require 'resolv'
|
|||
require 'checks'
|
|||
require 'rubygems' if USE_GEMS
|
|||
require 'win32/open3'
|
|||
module Proxy::DNS
|
|||
class Dnscmd < Record
|
|||
include Proxy::Log
|
|||
include Proxy::Util
|
|||
attr_reader :resolver
|
|||
def initialize options = {}
|
|||
super(options)
|
|||
end
|
|||
# create({ :fqdn => "node01.lab", :value => "192.168.100.2"}
|
|||
# create({ :fqdn => "node01.lab", :value => "3.100.168.192.in-addr.arpa",
|
|||
# :type => "PTR"}
|
|||
def create
|
|||
@resolver = Resolv::DNS.new(:nameserver => @server)
|
|||
case @type
|
|||
when "A"
|
|||
if ip = dns_find(@fqdn)
|
|||
raise(Proxy::DNS::Collision, "#{@fqdn} is already used by #{ip}") unless ip == @value
|
|||
else
|
|||
zone = @fqdn.sub(/[^.]+./,'')
|
|||
msg = "Added DNS entry #{@fqdn} => #{@value}"
|
|||
cmd = "/RecordAdd #{zone} #{@fqdn}. A #{@value}"
|
|||
execute(cmd, msg)
|
|||
end
|
|||
when "PTR"
|
|||
if name = dns_find(@value)
|
|||
raise(Proxy::DNS::Collision, "#{@value} is already used by #{name}") unless name == @fqdn
|
|||
else
|
|||
# TODO: determine reverse zone names, #4025
|
|||
return true
|
|||
end
|
|||
end
|
|||
end
|
|||
# remove({ :fqdn => "node01.lab", :value => "192.168.100.2"}
|
|||
# remove({ :fqdn => "node01.lab", :value => "3.100.168.192.in-addr.arpa"}
|
|||
def remove
|
|||
case @type
|
|||
when "A"
|
|||
zone = @fqdn.sub(/[^.]+./,'')
|
|||
msg = "Removed DNS entry #{@fqdn} => #{@value}"
|
|||
cmd = "/RecordDelete #{zone} #{@fqdn}. A /f"
|
|||
execute(cmd, msg)
|
|||
when "PTR"
|
|||
# TODO: determine reverse zone names, #4025
|
|||
return true
|
|||
end
|
|||
end
|
|||
private
|
|||
def execute cmd, msg=nil, error_only=false
|
|||
tsecs = 5
|
|||
response = nil
|
|||
interpreter = SETTINGS.x86_64 ? 'c:\windows\sysnative\cmd.exe' : 'c:\windows\system32\cmd.exe'
|
|||
command = interpreter + ' /c c:\Windows\System32\dnscmd.exe ' + "#{@server} #{cmd}"
|
|||
std_in = std_out = std_err = nil
|
|||
begin
|
|||
timeout(tsecs) do
|
|||
logger.debug "executing: #{command}"
|
|||
std_in, std_out, std_err = Open3.popen3(command)
|
|||
response = std_out.readlines
|
|||
response += std_err.readlines
|
|||
end
|
|||
rescue TimeoutError
|
|||
raise Proxy::DNS::Error.new("dnscmd did not respond within #{tsecs} seconds")
|
|||
ensure
|
|||
std_in.close unless std_in.nil?
|
|||
std_out.close unless std_in.nil?
|
|||
std_err.close unless std_in.nil?
|
|||
end
|
|||
report msg, response, error_only
|
|||
end
|
|||
def report msg, response, error_only
|
|||
if response.grep(/completed successfully/).empty?
|
|||
logger.error "Dnscmd failed:\n" + response.join("\n")
|
|||
msg.sub! /Removed/, "remove"
|
|||
msg.sub! /Added/, "add"
|
|||
match = ""
|
|||
msg = "Failed to #{msg}"
|
|||
raise Proxy::DNS::Error.new(msg)
|
|||
else
|
|||
logger.info msg unless error_only
|
|||
end
|
|||
rescue Proxy::DNS::Error
|
|||
raise
|
|||
rescue
|
|||
logger.error "Dnscmd failed:\n" + (response.is_a?(Array) ? response.join("\n") : "Response was not an array! #{response}")
|
|||
raise Proxy::DNS::Error.new("Unknown error while processing '#{msg}'")
|
|||
end
|
|||
def dns_find key
|
|||
if match = key.match(/(\d{1,3})\.(\d{1,3})\.(\d{1,3})\.(\d{1,3})/)
|
|||
resolver.getname(match[1..4].reverse.join(".")).to_s
|
|||
else
|
|||
resolver.getaddress(key).to_s
|
|||
end
|
|||
rescue Resolv::ResolvError
|
|||
false
|
|||
end
|
|||
end
|
|||
end
|