tunasync/tunasync/clt_server.py

#!/usr/bin/env python2
# -*- coding:utf-8 -*-
import socket
import os
import json
import struct


class ControlServer(object):

    valid_commands = set((
        "start", "stop", "restart", "status", "log",
    ))

    def __init__(self, address, mgr_chan, cld_chan):
        self.address = address
        self.mgr_chan = mgr_chan
        self.cld_chan = cld_chan
        try:
            os.unlink(self.address)
        except OSError:
            if os.path.exists(self.address):
                raise Exception("file exists: {}".format(self.address))
        self.sock = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
        self.sock.bind(self.address)
        os.chmod(address, 0o700)

        print("Control Server listening on: {}".format(self.address))
        self.sock.listen(1)

    def serve_forever(self):
        while 1:
            conn, _ = self.sock.accept()

            try:
                length = struct.unpack('!H', conn.recv(2))[0]
                content = conn.recv(length)
                cmd = json.loads(content)
                if cmd['cmd'] not in self.valid_commands:
                    raise Exception("Invalid Command")
                self.mgr_chan.put(("CMD", (cmd['cmd'], cmd['target'], cmd["kwargs"])))
            except Exception as e:
                print(e)
                res = "Invalid Command"
            else:
                res = self.cld_chan.get()

            conn.sendall(struct.pack('!H', len(res)))
            conn.sendall(res)
            conn.close()


def run_control_server(address, mgr_chan, cld_chan):
    cs = ControlServer(address, mgr_chan, cld_chan)
    cs.serve_forever()

# vim: ts=4 sw=4 sts=4 expandtab