mirror of
https://github.com/cmur2/dyndnsd.git
synced 2024-12-22 20:54:24 +01:00
225 lines
6.4 KiB
Ruby
225 lines
6.4 KiB
Ruby
#!/usr/bin/env ruby
|
|
|
|
require 'etc'
|
|
require 'logger'
|
|
require 'ipaddr'
|
|
require 'json'
|
|
require 'yaml'
|
|
require 'rack'
|
|
require 'metriks'
|
|
require 'metriks/reporter/graphite'
|
|
|
|
require 'dyndnsd/generator/bind'
|
|
require 'dyndnsd/updater/command_with_bind_zone'
|
|
require 'dyndnsd/responder/dyndns_style'
|
|
require 'dyndnsd/responder/rest_style'
|
|
require 'dyndnsd/database'
|
|
require 'dyndnsd/version'
|
|
|
|
module Dyndnsd
|
|
def self.logger
|
|
@logger
|
|
end
|
|
|
|
def self.logger=(logger)
|
|
@logger = logger
|
|
end
|
|
|
|
class LogFormatter
|
|
def call(lvl, time, progname, msg)
|
|
"[%s] %-5s %s\n" % [Time.now.strftime('%Y-%m-%d %H:%M:%S'), lvl, msg.to_s]
|
|
end
|
|
end
|
|
|
|
class Daemon
|
|
def initialize(config, db, updater, responder)
|
|
@users = config['users']
|
|
@domain = config['domain']
|
|
@db = db
|
|
@updater = updater
|
|
@responder = responder
|
|
|
|
@db.load
|
|
@db['serial'] ||= 1
|
|
@db['hosts'] ||= {}
|
|
(@db.save; update) if @db.changed?
|
|
end
|
|
|
|
def update
|
|
@updater.update(@db)
|
|
end
|
|
|
|
def is_fqdn_valid?(hostname)
|
|
return false if hostname.length < @domain.length + 2
|
|
return false if not hostname.end_with?(@domain)
|
|
name = hostname.chomp(@domain)
|
|
return false if not name.match(/^[a-zA-Z0-9_-]+\.$/)
|
|
true
|
|
end
|
|
|
|
def call(env)
|
|
return @responder.response_for_error(:method_forbidden) if env["REQUEST_METHOD"] != "GET"
|
|
return @responder.response_for_error(:not_found) if env["PATH_INFO"] != "/nic/update"
|
|
|
|
params = Rack::Utils.parse_query(env["QUERY_STRING"])
|
|
|
|
return @responder.response_for_error(:hostname_missing) if not params["hostname"]
|
|
|
|
hostnames = params["hostname"].split(',')
|
|
|
|
# Check if hostname match rules
|
|
hostnames.each do |hostname|
|
|
return @responder.response_for_error(:hostname_malformed) if not is_fqdn_valid?(hostname)
|
|
end
|
|
|
|
user = env["REMOTE_USER"]
|
|
|
|
hostnames.each do |hostname|
|
|
return @responder.response_for_error(:host_forbidden) if not @users[user]['hosts'].include? hostname
|
|
end
|
|
|
|
myip = nil
|
|
|
|
if params.has_key?("myip6")
|
|
# require presence of myip parameter as valid IPAddr (v4) and valid myip6
|
|
return @responder.response_for_error(:host_forbidden) if not params["myip"]
|
|
begin
|
|
IPAddr.new(params["myip"], Socket::AF_INET)
|
|
IPAddr.new(params["myip6"], Socket::AF_INET6)
|
|
|
|
# myip will be an array
|
|
myip = [params["myip"], params["myip6"]]
|
|
rescue ArgumentError
|
|
return @responder.response_for_error(:host_forbidden)
|
|
end
|
|
else
|
|
# fallback value, always present
|
|
myip = env["REMOTE_ADDR"]
|
|
|
|
# check whether X-Real-IP header has valid IPAddr
|
|
if env.has_key?("HTTP_X_REAL_IP")
|
|
begin
|
|
IPAddr.new(env["HTTP_X_REAL_IP"])
|
|
myip = env["HTTP_X_REAL_IP"]
|
|
rescue ArgumentError
|
|
end
|
|
end
|
|
|
|
# check whether myip parameter has valid IPAddr
|
|
if params.has_key?("myip")
|
|
begin
|
|
IPAddr.new(params["myip"])
|
|
myip = params["myip"]
|
|
rescue ArgumentError
|
|
end
|
|
end
|
|
end
|
|
|
|
Metriks.meter('requests.valid').mark
|
|
Dyndnsd.logger.info "Request to update #{hostnames} to #{myip} for user #{user}"
|
|
|
|
changes = []
|
|
hostnames.each do |hostname|
|
|
if (not @db['hosts'].include? hostname) or (@db['hosts'][hostname] != myip)
|
|
changes << :good
|
|
@db['hosts'][hostname] = myip
|
|
Metriks.meter('requests.good').mark
|
|
else
|
|
changes << :nochg
|
|
Metriks.meter('requests.nochg').mark
|
|
end
|
|
end
|
|
|
|
if @db.changed?
|
|
@db['serial'] += 1
|
|
Dyndnsd.logger.info "Committing update ##{@db['serial']}"
|
|
@db.save
|
|
update
|
|
Metriks.meter('updates.committed').mark
|
|
end
|
|
|
|
@responder.response_for_changes(changes, myip)
|
|
end
|
|
|
|
def self.run!
|
|
if ARGV.length != 1
|
|
puts "Usage: dyndnsd config_file"
|
|
exit 1
|
|
end
|
|
|
|
config_file = ARGV[0]
|
|
|
|
if not File.file?(config_file)
|
|
puts "Config file not found!"
|
|
exit 1
|
|
end
|
|
|
|
puts "DynDNSd version #{Dyndnsd::VERSION}"
|
|
puts "Using config file #{config_file}"
|
|
|
|
config = YAML::load(File.open(config_file, 'r') { |f| f.read })
|
|
|
|
if config['logfile']
|
|
Dyndnsd.logger = Logger.new(config['logfile'])
|
|
else
|
|
Dyndnsd.logger = Logger.new(STDOUT)
|
|
end
|
|
|
|
Dyndnsd.logger.progname = "dyndnsd"
|
|
Dyndnsd.logger.formatter = LogFormatter.new
|
|
|
|
Dyndnsd.logger.info "Starting..."
|
|
|
|
# drop privs (first change group than user)
|
|
Process::Sys.setgid(Etc.getgrnam(config['group']).gid) if config['group']
|
|
Process::Sys.setuid(Etc.getpwnam(config['user']).uid) if config['user']
|
|
|
|
# configure metriks
|
|
if config['graphite']
|
|
host = config['graphite']['host'] || 'localhost'
|
|
port = config['graphite']['port'] || 2003
|
|
options = {}
|
|
options[:prefix] = config['graphite']['prefix'] if config['graphite']['prefix']
|
|
reporter = Metriks::Reporter::Graphite.new(host, port, options)
|
|
reporter.start
|
|
else
|
|
reporter = Metriks::Reporter::ProcTitle.new
|
|
reporter.add 'good', 'sec' do
|
|
Metriks.meter('requests.good').mean_rate
|
|
end
|
|
reporter.add 'nochg', 'sec' do
|
|
Metriks.meter('requests.nochg').mean_rate
|
|
end
|
|
reporter.start
|
|
end
|
|
|
|
# configure daemon
|
|
db = Database.new(config['db'])
|
|
updater = Updater::CommandWithBindZone.new(config['domain'], config['updater']['params']) if config['updater']['name'] == 'command_with_bind_zone'
|
|
responder = Responder::DynDNSStyle.new
|
|
|
|
# configure rack
|
|
app = Daemon.new(config, db, updater, responder)
|
|
app = Rack::Auth::Basic.new(app, "DynDNS") do |user,pass|
|
|
allow = ((config['users'].has_key? user) and (config['users'][user]['password'] == pass))
|
|
if not allow
|
|
Dyndnsd.logger.warn "Login failed for #{user}"
|
|
Metriks.meter('requests.auth_failed').mark
|
|
end
|
|
allow
|
|
end
|
|
|
|
Signal.trap('INT') do
|
|
Dyndnsd.logger.info "Quitting..."
|
|
Rack::Handler::WEBrick.shutdown
|
|
end
|
|
Signal.trap('TERM') do
|
|
Dyndnsd.logger.info "Quitting..."
|
|
Rack::Handler::WEBrick.shutdown
|
|
end
|
|
|
|
Rack::Handler::WEBrick.run app, :Host => config['host'], :Port => config['port']
|
|
end
|
|
end
|
|
end
|