Small script that stops denyhosts, delete an ip from all the blacklist files and starts denyhosts

unban-ip.sh

#!/usr/bin/env ruby

base_path="/var/lib/denyhosts"
hosts_deny="/etc/hosts.deny"
denyhosts_files=%w(hosts hosts-restricted hosts-root hosts-valid users-hosts)

files=denyhosts_files.map{|file| "#{base_path}/#{file}"}.push hosts_deny
denyhosts_stop="/etc/init.d/denyhosts stop"
denyhosts_start="/etc/init.d/denyhosts start"

if ARGV.length!= 1
  puts "Wrong number of arguments"
  puts "Usage: unban-ip"
else

  `#{denyhosts_stop}`
  ip=ARGV[0]
  files.each do |file|
    text=File.read(file)
    if file == hosts_deny
      buffer=text.gsub(/sshd: #{ip}\n/,"")
    else
      buffer=text.gsub(/#{ip}:(.*)\n/,"")
    end
    File.open(file,"w") {|fw| fw.write buffer }
  end
  `#{denyhosts_start}`
end