From e608c0dc10ed5ad912682bf84040cd09f26c4d62 Mon Sep 17 00:00:00 2001 From: Jordan Rodgers Date: Wed, 21 Mar 2018 22:20:41 -0400 Subject: [PATCH] properly validate hostname properly (hopefully for the last time) --- proxstar/starrs.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/proxstar/starrs.py b/proxstar/starrs.py index b54ca32..8a95155 100644 --- a/proxstar/starrs.py +++ b/proxstar/starrs.py @@ -1,3 +1,6 @@ +import psycopg2 + + def get_next_ip(starrs, range_name): c = starrs.cursor() try: @@ -42,6 +45,10 @@ def renew_ip(starrs, addr): def check_hostname(starrs, hostname): c = starrs.cursor() try: + c.execute("BEGIN") + c.callproc("api.initialize", ('root', )) + c.callproc("api.validate_name", (hostname, )) + c.execute("COMMIT") c.execute("BEGIN") c.callproc("api.initialize", ('root', )) c.callproc("api.validate_domain", (hostname, 'csh.rit.edu'))