alarig/bird-lg
1
0
Fork 0
mirror of https://github.com/sileht/bird-lg.git synced 2024-11-25 00:04:42 +01:00
Code Issues Releases Wiki Activity

Handle DNS for prefix

Browse source
This commit is contained in:
Mehdi ABAAKOUK 2011-12-16 13:19:48 +01:00
parent f6738ea4dc
commit 8519d6f8a0
1 changed files with 28 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

32
lg.py
View file

@ -1,12 +1,15 @@
#!/usr/bin/python #!/usr/bin/python
import sys import sys
from dns import resolver,reversename
from flask import Flask, render_template, jsonify from flask import Flask, render_template, jsonify
app = Flask(__name__) app = Flask(__name__)
import socket import socket
def get_ip(n, q):
return str(resolver.query(n,q)[0])
def check_mask(n): def check_mask(n):
if not n: if not n:
return True return True
@ -37,12 +40,32 @@ def hello():
@app.route("/<host>/<proto>/prefix/<prefix>") @app.route("/<host>/<proto>/prefix/<prefix>")
@app.route("/<host>/<proto>/prefix/<prefix>/<mask>") @app.route("/<host>/<proto>/prefix/<prefix>/<mask>")
def prefix(host, proto, prefix, mask=""): def prefix(host, proto, prefix, mask=""):
output = '<h3>' + host + '(' + proto + ') show route for ' + prefix + (mask and '/' + mask or '' ) + '</h3>' qprefix = prefix
# security check # security check
if ( (proto == "ipv6" and check_ipv6(prefix)) or (proto == "ipv4" and check_ipv4(prefix)) ) and check_mask(mask): allowed = True
if not check_mask(mask):
allowed = False
elif proto == "ipv6":
if not check_ipv6(prefix):
try:
qprefix = get_ip(prefix, "AAAA")
except:
allowed = False
elif proto == "ipv4":
if not check_ipv4(prefix):
try:
qprefix = get_ip(prefix, "A")
except:
allowed = False
else:
allowed = False
output = '<h3>' + host + '(' + proto + ') show route for ' + prefix + (prefix != qprefix and " (%s)"%qprefix or "") + (mask and '/' + mask or '' ) + '</h3>'
if allowed:
if mask: qprefix = qprefix +"/"+mask
if mask: prefix = prefix +"/"+mask if mask: prefix = prefix +"/"+mask
ok, string = get_cmd_result(host , proto, "show route for " + prefix) ok, string = get_cmd_result(host , proto, "show route for " + qprefix)
if ok: if ok:
string = "\n".join([ s.replace("1007-"," ") for s in string.split("\n") if not s.startswith("0000") ]) string = "\n".join([ s.replace("1007-"," ") for s in string.split("\n") if not s.startswith("0000") ])
output +='<pre>' + string + '</pre>' output +='<pre>' + string + '</pre>'
@ -50,6 +73,7 @@ def prefix(host, proto, prefix, mask=""):
output += string output += string
else: else:
output += prefix + ' not allowed' output += prefix + ' not allowed'
return render_template('index.html', output=output, typ="prefix", host=host+"/"+proto, prefix=prefix) return render_template('index.html', output=output, typ="prefix", host=host+"/"+proto, prefix=prefix)
@app.route("/<host>/<proto>/detail/<name>") @app.route("/<host>/<proto>/detail/<name>")