Add dynamic dns handler

5 files changed, 119 insertions, 0 deletions

diff --git a/ddns/Dockerfile b/ddns/Dockerfile
new file mode 100644
index 0000000..b112f56
--- /dev/null
+++ b/ddns/Dockerfile
@@ -0,0 +1,19 @@
+FROM ubuntu:14.04
+MAINTAINER Michael Crute <mike@crute.us>
+
+RUN export DEBIAN_FRONTEND=noninteractive && \
+    apt-get update && \
+    apt-get install -y curl python-pip uwsgi uwsgi-plugin-python && \
+    pip install flask boto && \
+    useradd -r -M -d /srv -s /bin/nologin ddns
+
+ADD uwsgi.ini /srv
+ADD bin/* /srv/bin/
+
+RUN \
+    apt-get clean && \
+    rm -rf /var/lib/apt/lists/* && \
+    rm -rf /tmp/*
+
+STOPSIGNAL SIGINT
+CMD ["/usr/bin/uwsgi", "--ini", "/srv/uwsgi.ini"]
diff --git a/ddns/Makefile b/ddns/Makefile
new file mode 100644
index 0000000..0d9df95
--- /dev/null
+++ b/ddns/Makefile
@@ -0,0 +1,11 @@
+all:
+        docker build -t ddns .
+
+all-no-cache:
+        docker build --no-cache -t ddns .
+
+run:
+        docker run -d \
+                -p 9000:9000 \
+                -v /srv/ddns/ddns.cfg:/etc/ddns.cfg \
+                ddns
diff --git a/ddns/bin/ddns.py b/ddns/bin/ddns.py
new file mode 100755
index 0000000..5ec92da
--- /dev/null
+++ b/ddns/bin/ddns.py
@@ -0,0 +1,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()
diff --git a/ddns/bin/su-exec b/ddns/bin/su-exec
new file mode 100755
index 0000000..940f452
--- /dev/null
+++ b/ddns/bin/su-exec
diff --git a/ddns/uwsgi.ini b/ddns/uwsgi.ini
new file mode 100644
index 0000000..8478fc0
--- /dev/null
+++ b/ddns/uwsgi.ini
@@ -0,0 +1,10 @@
+[uwsgi]
+master = true
+socket = :9000
+uid = ddns
+gid = ddns
+plugin = python
+chdir = /srv/bin
+workers = 2
+module = ddns:app
+harakiri = 300