1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
|
#!/usr/bin/env python
import os
import flask
import hashlib
import functools
from boto.route53.record import ResourceRecordSets
from boto.route53.connection import Route53Connection
app = flask.Flask(__name__)
app.config.from_pyfile("/etc/ddns.cfg", silent=True)
app.config.from_pyfile("ddns.cfg", silent=True)
if "AMAZON_KEY_ID" not in app.config:
raise Exception("Not configured")
def returns_plain_text(f):
@functools.wraps(f)
def wrapper(*args, **kwargs):
return flask.Response(f(*args, **kwargs), content_type="text/plain")
return wrapper
def get_ip():
if "X-Forwarded-For" in flask.request.headers:
return flask.request.headers["X-Forwarded-For"]
else:
return flask.request.remote_addr
def update_record(zone, record, ip):
conn = Route53Connection(app.config["AMAZON_KEY_ID"],
app.config["AMAZON_SECRET_KEY"])
change_set = ResourceRecordSets(conn, conn.get_zone(zone).id)
change_set.add_change("UPSERT", record, type="A", ttl=60).add_value(ip)
change_set.commit()
@app.errorhandler(404)
@app.errorhandler(405)
@app.errorhandler(500)
def handle_error(ex):
response = flask.Response("Error", content_type="text/plain")
response.status_code = getattr(ex, "code", 500)
return response
@app.route("/new-secret", methods=["GET"])
@returns_plain_text
def new_secret():
return hashlib.sha256(os.urandom(100)).hexdigest()
@app.route("/update", methods=["POST"])
def update_ip():
key = flask.request.form.get("key")
config = app.config["CLIENTS"].get(key)
if not config:
flask.abort(404)
try:
update_record(config["zone"], config["resource"], get_ip())
return "OK"
except:
flask.abort(500)
@app.route("/", methods=["GET"])
@returns_plain_text
def handle_home():
return get_ip()
if __name__ == "__main__":
app.debug = True
app.run()
|
