82 lines
2.1 KiB
Python
82 lines
2.1 KiB
Python
import threading
|
|
import time
|
|
import Event
|
|
from logger import Logger
|
|
import BaseHTTPServer
|
|
import SocketServer
|
|
import cgi
|
|
import urlparse
|
|
|
|
import Entry
|
|
|
|
class LocalException(Exception):
|
|
def __init__(self, msg):
|
|
Logger.log("Exception in HttpCmdReceiver: %s" % msg)
|
|
self.msg = msg
|
|
|
|
class HttpCmdHandler(BaseHTTPServer.BaseHTTPRequestHandler):
|
|
server_version = "YadynCMDHTTP/1.0"
|
|
|
|
@classmethod
|
|
def setClassParams(cls, entries, adminPwd):
|
|
cls.entries = entries
|
|
cls.adminPwd = adminPwd
|
|
|
|
def do_GET(self):
|
|
Logger.log("Request: %s" % self.path)
|
|
try:
|
|
if self.command == "GET":
|
|
q = urlparse.urlparse(self.path).query
|
|
c = cgi.parse_qs(q)
|
|
if not c.has_key('c'):
|
|
raise LocalException('no cmd given')
|
|
cmd = c['c'][0]
|
|
Logger.log("Cmd: " + cmd)
|
|
if cmd == 'register':
|
|
self.register(c)
|
|
else:
|
|
raise LocalException('unknown cmd')
|
|
|
|
self.send_response(200, "OK")
|
|
self.end_headers()
|
|
except LocalException, e:
|
|
self.send_response(400, e.msg)
|
|
self.end_headers()
|
|
|
|
def register(self, c):
|
|
try:
|
|
subdomain = c['subdomain'][0]
|
|
sharedSecret = c['sharedsecret'][0]
|
|
except KeyError, e:
|
|
raise LocalException('Parameter missing: ' + str(e))
|
|
|
|
dynid = subdomain
|
|
zone = 'yadynns.net'
|
|
|
|
if HttpCmdHandler.entries.has_key(dynid):
|
|
raise LocalException("duplicate dynid")
|
|
for entry in HttpCmdHandler.entries.values():
|
|
if entry.name == subdomain and entry.zone == zone:
|
|
raise LocalException("duplicate full name")
|
|
newEntry = Entry.Entry(dynid, sharedSecret, subdomain, zone)
|
|
HttpCmdHandler.entries[dynid] = newEntry
|
|
|
|
|
|
class HttpCmdServer(SocketServer.ThreadingTCPServer):
|
|
def __init__(self, serverAddr):
|
|
self.allow_reuse_address = True
|
|
SocketServer.ThreadingTCPServer.__init__(self, serverAddr, HttpCmdHandler)
|
|
|
|
|
|
class HttpCmdReceiver(threading.Thread):
|
|
def __init__(self, httpCmdRecvAddr, entries, adminPwd):
|
|
threading.Thread.__init__(self)
|
|
self.httpCmdRecvAddr = httpCmdRecvAddr
|
|
HttpCmdHandler.setClassParams(entries, adminPwd)
|
|
self.setDaemon(True)
|
|
|
|
def run(self):
|
|
server = HttpCmdServer(self.httpCmdRecvAddr)
|
|
server.serve_forever()
|
|
|