From 6981666c6d8f639a1ff9c09a3bc08769e19928af Mon Sep 17 00:00:00 2001
From: jwijenbergh <jeroenwijenbergh@protonmail.com>
Date: Tue, 20 Dec 2022 15:35:44 +0100
Subject: Failover: Initial implementation

---
 wrappers/python/eduvpn_common/loader.py |  3 +++
 wrappers/python/eduvpn_common/main.py   | 16 +++++++++++++++-
 wrappers/python/eduvpn_common/types.py  |  1 +
 3 files changed, 19 insertions(+), 1 deletion(-)

(limited to 'wrappers/python')

diff --git a/wrappers/python/eduvpn_common/loader.py b/wrappers/python/eduvpn_common/loader.py
index 3de3de5..1090619 100644
--- a/wrappers/python/eduvpn_common/loader.py
+++ b/wrappers/python/eduvpn_common/loader.py
@@ -7,6 +7,7 @@ from eduvpn_common import __version__
 from eduvpn_common.types import (
     ConfigError,
     DataError,
+    ReadRxBytes,
     VPNStateChange,
 )
 
@@ -151,3 +152,5 @@ def initialize_functions(lib: CDLL) -> None:
         c_int,
     ], c_void_p
     lib.ShouldRenewButton.argtypes, lib.ShouldRenewButton.restype = [], int
+    lib.StartFailover.argtypes, lib.StartFailover.restype = [c_char_p, c_char_p, c_int, ReadRxBytes], DataError
+    lib.CancelFailover.argtypes, lib.CancelFailover.restype = [c_char_p], c_void_p
diff --git a/wrappers/python/eduvpn_common/main.py b/wrappers/python/eduvpn_common/main.py
index 20d646f..3cb45e1 100644
--- a/wrappers/python/eduvpn_common/main.py
+++ b/wrappers/python/eduvpn_common/main.py
@@ -7,7 +7,7 @@ from eduvpn_common.event import EventHandler
 from eduvpn_common.loader import initialize_functions, load_lib
 from eduvpn_common.server import Profiles, Server, get_transition_server, get_servers
 from eduvpn_common.state import State, StateType
-from eduvpn_common.types import VPNStateChange, decode_res, encode_args, get_data_error
+from eduvpn_common.types import ReadRxBytes, VPNStateChange, decode_res, encode_args, get_data_error, get_bool
 
 
 class EduVPN(object):
@@ -502,6 +502,20 @@ class EduVPN(object):
 
         return servers
 
+    def start_failover(self, gateway: str, wg_mtu: int, readrxbytes: ReadRxBytes) -> bool:
+        dropped, dropped_err = self.go_function(
+            self.lib.StartFailover, gateway, wg_mtu, readrxbytes,
+            decode_func=lambda lib, x: get_data_error(lib, x, get_bool),
+        )
+        if dropped_err:
+            raise dropped_err
+        return dropped
+
+    def cancel_failover(self):
+        cancel_err = self.go_function(self.lib.CancelFailover)
+        if cancel_err:
+            raise cancel_err
+
 
 eduvpn_objects: Dict[str, EduVPN] = {}
 
diff --git a/wrappers/python/eduvpn_common/types.py b/wrappers/python/eduvpn_common/types.py
index 07a02d3..7e3ce9a 100644
--- a/wrappers/python/eduvpn_common/types.py
+++ b/wrappers/python/eduvpn_common/types.py
@@ -166,6 +166,7 @@ class ConfigError(Structure):
 
 # The type for a Go state change callback
 VPNStateChange = CFUNCTYPE(c_int, c_char_p, c_int, c_int, c_void_p)
+ReadRxBytes = CFUNCTYPE(c_ulonglong)
 
 
 def encode_args(args: List[Any], types: List[Any]) -> Iterator[Any]:
-- 
cgit v1.2.3