mirror of https://github.com/sileht/bird-lg.git
127 lines
3.7 KiB
Python
127 lines
3.7 KiB
Python
# -*- coding: utf-8 -*-
|
|
# vim: ts=4
|
|
###
|
|
#
|
|
# Copyright (c) 2006 Mehdi Abaakouk
|
|
#
|
|
# This program is free software; you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License version 3 as
|
|
# published by the Free Software Foundation
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program; if not, write to the Free Software
|
|
# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
|
|
#
|
|
###
|
|
|
|
|
|
import logging
|
|
from logging import handlers
|
|
import subprocess
|
|
import sys
|
|
import urllib
|
|
|
|
import flask
|
|
|
|
import bird
|
|
|
|
app = flask.Flask(__name__)
|
|
app.debug = app.config["DEBUG"]
|
|
app.config.from_pyfile('lgproxy.cfg')
|
|
|
|
file_handler = handlers.TimedRotatingFileHandler(
|
|
filename=app.config["LOG_FILE"], when="midnight")
|
|
app.logger.setLevel(getattr(logging, app.config["LOG_LEVEL"].upper()))
|
|
app.logger.addHandler(file_handler)
|
|
|
|
|
|
@app.before_request
|
|
def access_log_before(*args, **kwargs):
|
|
app.logger.info("[%s] flash.request %s, %s",
|
|
flask.request.remote_addr,
|
|
flask.request.url, "|".join(
|
|
["%s:%s" % (k, v) for k, v in
|
|
flask.request.headers.items()])
|
|
)
|
|
|
|
|
|
@app.after_request
|
|
def access_log_after(response, *args, **kwargs):
|
|
app.logger.info("[%s] reponse %s, %s", flask.request.remote_addr,
|
|
flask.request.url, response.status_code)
|
|
return response
|
|
|
|
|
|
def check_accesslist():
|
|
if (app.config["ACCESS_LIST"] and
|
|
flask.request.remote_addr not in app.config["ACCESS_LIST"]):
|
|
flask.abort(401)
|
|
|
|
|
|
@app.route("/traceroute")
|
|
@app.route("/traceroute6")
|
|
def traceroute():
|
|
check_accesslist()
|
|
|
|
if (sys.platform.startswith('freebsd') or sys.platform.startswith('netbsd')
|
|
or sys.platform.startswith('openbsd')):
|
|
traceroute4 = ['traceroute']
|
|
traceroute6 = ['traceroute6']
|
|
else: # For Linux
|
|
traceroute4 = ['traceroute', '-4']
|
|
traceroute6 = ['traceroute', '-6']
|
|
|
|
src = []
|
|
if flask.request.path == '/traceroute6':
|
|
traceroute = traceroute6
|
|
if app.config.get("IPV6_SOURCE", ""):
|
|
src = ["-s", app.config.get("IPV6_SOURCE")]
|
|
else:
|
|
traceroute = traceroute4
|
|
if app.config.get("IPV4_SOURCE", ""):
|
|
src = ["-s", app.config.get("IPV4_SOURCE")]
|
|
|
|
query = flask.request.args.get("q", "")
|
|
query = urllib.unquote(query)
|
|
|
|
if sys.platform.startswith('freebsd') or sys.platform.startswith('netbsd'):
|
|
options = ['-a', '-q1', '-w1', '-m15']
|
|
elif sys.platform.startswith('openbsd'):
|
|
options = ['-A', '-q1', '-w1', '-m15']
|
|
else: # For Linux
|
|
options = ['-A', '-q1', '-N32', '-w1', '-m15']
|
|
command = traceroute + src + options + [query]
|
|
p = subprocess.Popen(command, stdout=subprocess.PIPE)
|
|
return p.communicate()[0].decode('utf-8', 'ignore').replace("\n", "<br>")
|
|
|
|
|
|
@app.route("/bird")
|
|
@app.route("/bird6")
|
|
def bird_query():
|
|
check_accesslist()
|
|
|
|
if flask.request.path == "/bird":
|
|
b = bird.BirdSocket(file=app.config.get("BIRD_SOCKET"))
|
|
elif flask.request.path == "/bird6":
|
|
b = bird.BirdSocket(file=app.config.get("BIRD6_SOCKET"))
|
|
else:
|
|
return "No bird socket selected"
|
|
|
|
query = flask.request.args.get("q", "")
|
|
query = urllib.unquote(query)
|
|
|
|
status, result = b.cmd(query)
|
|
b.close()
|
|
# FIXME: use status
|
|
return result
|
|
|
|
|
|
if __name__ == "__main__":
|
|
app.logger.info("lgproxy start")
|
|
app.run("0.0.0.0")
|