diff -r 000000000000 -r 367d791aeb57 gmyth-stream/server/0.1/plugins/comm/tcp.py --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/gmyth-stream/server/0.1/plugins/comm/tcp.py Thu May 17 14:07:27 2007 +0100 @@ -0,0 +1,79 @@ +import lib +import time +import socket +import logging as log + +class Server(object): + + def __init__(self, config): + self.host = '0.0.0.0' + self.port = int(config.get("Comm", "port")) + self.finish = 0 + + addr = (self.host, self.port) + log.debug("Setup TCP server at %s:%s" % addr) + self.tcp = socket.socket(socket.AF_INET, socket.SOCK_STREAM) + self.tcp.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1) + self.tcp.bind(addr) + self.tcp.listen(1) + log.info("TCP server listening at %s:%s (sock=%d)" % + (self.host, self.port, self.tcp.fileno())) + + def getMsg(self): + bytes = [] + try: + while 1: + c = self.con.recv(1) + bytes.append(c) + if not c or c == "\n": + break + except Exception, e: + log.error("Error reading message from client: %s" % e) + return None + + if not bytes or bytes[-1] != "\n": + msg = "".join(bytes) + log.error("Invalid message from client: %r" % msg) + return None + + # remove \n and \r + bytes.pop() + if bytes[-1] == "\r": + bytes.pop() + + msg = "".join(bytes) + log.debug("RECV: %r" % msg) + return msg + + def sendMsg(self, msg): + log.debug("SEND: %r" % msg) + self.con.send(msg + "\n") + + def sendOk(self, payload=None): + self.sendMsg("OK %d" % bool(payload is not None)) + if payload is not None: + if not isinstance(payload, (tuple, list)): + payload = (payload,) + for e in payload: + self.sendMsg("+%s" % e) + self.sendMsg(".") + + def sendNotOk(self, reason=""): + self.sendMsg("NOTOK %r" % reason) + + def getRequest(self): + log.debug("Wait for client request at %s:%s (sock=%d)" % + (self.host, self.port, self.tcp.fileno())) + self.con, self.client = self.tcp.accept() + log.info("Incoming request from %s (con=%s)" % + (self.client, self.con.fileno())) + return (self.con, self.client, self.port) + + def disconnect_client(self, connection): + log.info("Closed request from %s (con=%s)" % + (self.client, self.con.fileno())) + connection.close() + + def stop(self): + log.debug("Stop") + self.tcp.close()