1 files changed, 177 insertions, 0 deletions
diff --git a/Vland/ipc/ipc.py b/Vland/ipc/ipc.py
new file mode 100644
index 0000000..5961ed1
--- /dev/null
+++ b/Vland/ipc/ipc.py
@@ -0,0 +1,177 @@
+#  Copyright 2014-2015 Linaro Limited
+#
+#  This program is free software; you can redistribute it and/or modify
+#  it under the terms of the GNU General Public License as published by
+#  the Free Software Foundation; either version 2 of the License, or
+#  (at your option) any later version.
+#
+#  This program is distributed in the hope that it will be useful,
+#  but WITHOUT ANY WARRANTY; without even the implied warranty of
+#  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+#  GNU General Public License for more details.
+#
+#  You should have received a copy of the GNU General Public License
+#  along with this program; if not, write to the Free Software
+#  Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
+#  MA 02110-1301, USA.
+#
+#  Simple VLANd IPC module
+
+import socket
+import json
+import time
+import datetime
+import os
+import sys
+import logging
+
+from Vland.errors import CriticalError, InputError, ConfigError, SocketError
+
+class VlanIpc:
+    """VLANd IPC class"""
+
+    def __init__(self):
+        self.conn = None
+        self.socket = None
+
+    def server_init(self, host, port):
+        self.socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+        self.socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
+        self.conn = None
+
+        while True:
+            try:
+                self.socket.bind((host, port))
+                break
+            except socket.error as e:
+                print "Can't bind to port %d: %s" % (port, e)
+                time.sleep(1)
+
+    def server_listen(self):
+        if self.socket is None:
+            raise SocketError("Server can't receive data: no socket")
+        self.socket.listen(1)
+
+    def server_recv(self):
+        if self.socket is None:
+            raise SocketError("Server can't receive data: no socket")
+
+        self.conn, addr = self.socket.accept()
+        logging.debug("server: Connection from")
+        logging.debug(addr)
+        data = self.conn.recv(8)  # 32bit limit
+        count = int(data, 16)
+        c = 0
+        data = ''
+        while c < count:
+            data += self.conn.recv(1)
+            c += 1
+        try:
+            json_data = json.loads(data)
+        except ValueError:
+            self.conn.close()
+            self.conn = None
+            raise SocketError("Server unable to decode receieved data: corrupt?")
+
+        if 'client_name' not in json_data:
+            self.conn.close()
+            self.conn = None
+            raise SocketError("Server unable to detect client name: corrupt packet?")
+
+        return json_data
+
+    def server_reply(self, json_data):
+        if self.conn is None:
+            raise SocketError("Server can't send data: no connection")
+
+        data = self._format_message(json_data)
+        if not data:
+            self.conn.close()
+            self.conn = None
+            raise SocketError("Server unable to format reply data")
+
+        try:
+            # send the actual number of bytes to read.
+            self.conn.send(data[0])
+            # now send the bytes.
+            self.conn.send(data[1])
+        except socket.error as e:
+            logging.error("Can't send response to client: %s", e)
+            logging.error("Was trying to send data:")
+            logging.error(data)
+
+    def server_close(self):
+        if self.conn is not None:
+            self.conn.shutdown(socket.SHUT_RDWR)
+            self.conn.close()
+
+    def client_connect(self, host, port):
+        self.socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+        while True:
+            try:
+                ret = self.socket.connect_ex((host, port))
+                if ret:
+                    self.socket.close()
+                    self.socket = None
+                    raise SocketError("Client can't send connect: %s" % ret)
+                else:
+                    break
+            except socket.error:
+                time.sleep(1)
+        return True
+
+    def client_send(self, json_data):
+        if self.socket is None:
+            raise SocketError("Client can't send data: no socket")
+
+        data = self._format_message(json_data)
+        if not data:
+            self.socket.shutdown(socket.SHUT_RDWR)
+            self.socket.close()
+            self.socket = None
+            raise SocketError("Client unable to send data")
+
+        # send the actual number of bytes to read.
+        self.socket.send(data[0])
+        # now send the bytes.
+        self.socket.send(data[1])
+
+    def client_recv_and_close(self):
+        if self.socket is None:
+            raise SocketError("Client can't receieve data: no socket")
+
+        data = self.socket.recv(8)  # 32bit limit
+        count = int(data, 16)
+        c = 0
+        data = ''
+        while c < count:
+            data += self.socket.recv(1)
+            c += 1
+        try:
+            json_data = json.loads(data)
+        except ValueError:
+            self.socket.close()
+            self.socket = None
+            raise SocketError("Client unable to decode receieved data: corrupt?")
+
+        self.socket.shutdown(socket.SHUT_RDWR)
+        self.socket.close()
+
+        return json_data
+
+    # The default JSON serialiser code can't deal with datetime
+    # objects by default, so let's tell it how to.
+    def _json_serial(self, obj):
+        """JSON serializer for objects not serialisable by default json code"""
+        if isinstance(obj, datetime.datetime):
+            serial = obj.isoformat()
+            return serial
+
+    def _format_message(self, json_data):
+        try:
+            msgstr = json.dumps(json_data, default=self._json_serial)
+        except ValueError:
+            return None
+        # "header" calculation
+        msglen = "%08X" % len(msgstr)
+        return (msglen, msgstr)