From da19f40ab42a7570d5a288239cc4dade56139082 Mon Sep 17 00:00:00 2001
From: "Kali Kaneko (leap communications)" <kali@leap.se>
Date: Thu, 15 Jun 2017 01:16:24 +0200
Subject: [bug] reset traffic when stopping vpn

also, refactor a bit VPNManagement so that the `connect_to_management`
method does not receive connection details.

I plan to refactor management so that it is a separate object from the
control object (but we need to stablish better its relation with the
status parsing class).

- Resolves: #8834
---
 src/leap/bitmask/vpn/_management.py | 74 ++++++++++++++-----------------------
 1 file changed, 27 insertions(+), 47 deletions(-)

(limited to 'src/leap/bitmask/vpn/_management.py')

diff --git a/src/leap/bitmask/vpn/_management.py b/src/leap/bitmask/vpn/_management.py
index 6e24f478..18e52ed7 100644
--- a/src/leap/bitmask/vpn/_management.py
+++ b/src/leap/bitmask/vpn/_management.py
@@ -15,7 +15,7 @@ except ImportError:
     from psutil import AccessDenied as psutil_AccessDenied
     PSUTIL_2 = True
 
-from ._telnet import UDSTelnet
+from leap.bitmask.vpn._telnet import UDSTelnet
 
 
 class OpenVPNAlreadyRunning(Exception):
@@ -47,6 +47,17 @@ class VPNManagement(object):
         self._tn = None
         self.aborted = False
 
+    def set_connection(self, host, port):
+        """
+        :param host: either socket path (unix) or socket IP
+        :type host: str
+
+        :param port: either string "unix" if it's a unix socket, or port
+                     otherwise
+        """
+        self._host = host
+        self._port = port
+
     def _seek_to_eof(self):
         """
         Read as much as available. Position seek pointer to end of stream
@@ -71,7 +82,6 @@ class VPNManagement(object):
         :return: response read
         :rtype: list
         """
-        # leap_assert(self._tn, "We need a tn connection!")
 
         try:
             self._tn.write("%s\n" % (command,))
@@ -108,23 +118,17 @@ class VPNManagement(object):
         self._tn.get_socket().close()
         self._tn = None
 
-    def _connect_management(self, socket_host, socket_port):
+    def connect_to_management(self):
         """
-        Connects to the management interface on the specified
-        socket_host socket_port.
-
-        :param socket_host: either socket path (unix) or socket IP
-        :type socket_host: str
+        Connects to the management interface.
 
-        :param socket_port: either string "unix" if it's a unix
-                            socket, or port otherwise
         :type socket_port: str
         """
         if self.is_connected():
             self._close_management_socket()
 
         try:
-            self._tn = UDSTelnet(socket_host, socket_port)
+            self._tn = UDSTelnet(self._host, self._port)
 
             # XXX make password optional
             # specially for win. we should generate
@@ -138,42 +142,16 @@ class VPNManagement(object):
                 self._tn.read_eager()
 
         except Exception as e:
+            print "ERROR", e
             self.log.warn('Could not connect to OpenVPN yet: %r' % (e,))
             self._tn = None
 
-    def _connectCb(self, *args):
-        """
-        Callback for connection.
-
-        :param args: not used
-        """
-        if not self._tn:
-            self.log.warn('Cannot connect to management...')
-
-    def _connectErr(self, failure):
-        """
-        Errorback for connection.
-
-        :param failure: Failure
-        """
-        self.log.failure('Error while connecting to management!')
-
-    def connect_to_management(self, host, port):
-        """
-        Connect to a management interface.
-
-        :param host: the host of the management interface
-        :type host: str
-
-        :param port: the port of the management interface
-        :type port: str
-
-        :returns: a deferred
-        """
-        self.connectd = defer.maybeDeferred(
-            self._connect_management, host, port)
-        self.connectd.addCallbacks(self._connectCb, self._connectErr)
-        return self.connectd
+        if self._tn:
+            return True
+        else:
+            print "ERROR!"
+            #self.log.failure('Error while connecting to management!')
+            return False
 
     def is_connected(self):
         """
@@ -205,7 +183,7 @@ class VPNManagement(object):
                            'not alive.')
             return
         if not self.aborted and not self.is_connected():
-            self.connect_to_management(self._socket_host, self._socket_port)
+            self.connect_to_management()
             reactor.callLater(
                 self.CONNECTION_RETRY_TIME,
                 self.try_to_connect_to_management, retry + 1)
@@ -235,6 +213,8 @@ class VPNManagement(object):
             if state != self._last_state:
                 # XXX this status object is the vpn status observer
                 if self._status:
+                    # XXX DEBUG -----------------------
+                    print "SETTING STATUS", state
                     self._status.set_status(state, None)
                 self._last_state = state
 
@@ -316,7 +296,7 @@ class VPNManagement(object):
         """
         if self._socket_port == "unix":
             self.log.debug('Cleaning socket file temp folder')
-            tempfolder = _first(os.path.split(self._socket_host))
+            tempfolder = _first(os.path.split(self._host))
             if tempfolder and os.path.isdir(tempfolder):
                 try:
                     shutil.rmtree(tempfolder)
@@ -397,7 +377,7 @@ class VPNManagement(object):
                 port = cmdline[index + 2]
                 self.log.debug("Trying to connect to %s:%s"
                                % (host, port))
-                self.connect_to_management(host, port)
+                self.connect_to_management()
 
                 # XXX this has a problem with connections to different
                 # remotes. So the reconnection will only work when we are
-- 
cgit v1.2.3