#  Créé par AppleStorm { Hwc-crew (http://www.orgasm.re)
#  le 26 - 09 - 2012
#  Gestion de ban plage ip pour serveur nginx

#!/usr/bin/env ruby
# CONSTANTES PARAMETRABLES 
@line = 0
@fichier = "/etc/nginx/banips.conf" # Votre fichier de listing banip
@exe = "/etc/init.d/nginx restart"
# ------------------------- #

# FONCTIONS Personalisables #
def outputUsage()
        puts "============== BanIps :: You Fail bitch ! =============="
        puts " usage : <dir>/<script> [command]   [ip]"
        puts " Commandes :"
        puts "                add [ip]        :        Ajoute le deny à [ip]"
        puts "                del [ip]        :        Delette le ban à [ip]"
        puts "                view                :        Afficher les ips ban."
        puts "                view [ip]        :         Retourne Oui ou Non. "
        puts ""
        puts " example :"
        puts "                <dir>/<script> add 192.168.0.0"
        puts "                <dir>/<script> del 192.168.0.0"
        puts "========================================================"
end
def find_ip(masque)
        tableau = File.open(@fichier, "r")
        i = 1
        a = false
        tableau.each_line { |ligne|
                if not ligne.match( masque ).nil?
                        puts "Trouvé : #{i} - #{ligne}"
                        @line = i-1
                        a = true
                        i += 1
                else
                        i += 1
                end
        }
        if a != true
                puts "---- #{masque} non trouvé -----"
        end        
end
def delete_ip(line)
        line_arr = File.readlines(@fichier)
        line_arr.each do |index|
                line_arr.delete_at(line)
        end
        File.open(@fichier, "w") do |f|
                line_arr.each{ |ligne| f.puts(ligne) }
        end
end
if not ARGV[0].nil?
        if not ARGV[1].nil?
                # Commandes
                ip = ARGV[1].split('.')
                masque = ip[0]+'.'+ip[1]
                pip = "deny "+ip[0]+'.'+ip[1]+'.0.0/24;'
                case ARGV[0]
                        when "add"
                                #lignedecode
                                puts "======================================================="
                                puts "BanIps : Vous avez ajouté un ban : " + pip
                                puts "======================================================="
                                fichier = File.open(@fichier, "a+")
                                fichier.write pip+"\n"
                                fichier.close
                                exec(@exe)
                        when "del"
                                #lignedelete
                                find_ip(pip)
                                delete_ip(@line)
                                exec(@exe)
                        when "view"
                                #ligneview
                                puts "Vous souhaitez voir le ban : " + pip
                                find_ip(pip)
                        else
                                outputUsage()
                end
        else
                if ARGV[0] == "view"
                        file = File.open(@fichier, "r")
                        i = 1
                        file.each_line { |ligne|
                                puts "#{i} - #{ligne}"
                                i += 1
                        }
                        file.close
                end
        end
else
        outputUsage()
end