yadyn/server/HttpCmdReceiver.py

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()