Client + Server + Exports: Implement optional WireGuard support

2 files changed, 16 insertions, 0 deletions

diff --git a/wrappers/python/eduvpn_common/loader.py b/wrappers/python/eduvpn_common/loader.py
index fc3d090..9463ab1 100644
--- a/wrappers/python/eduvpn_common/loader.py
+++ b/wrappers/python/eduvpn_common/loader.py
@@ -152,4 +152,8 @@ def initialize_functions(lib: CDLL) -> None:
         c_char_p,
         c_char_p,
     ], c_void_p
+    lib.SetSupportWireguard.argtypes, lib.SetSupportWireguard.restype = [
+        c_char_p,
+        c_int,
+    ], c_void_p
     lib.ShouldRenewButton.argtypes, lib.ShouldRenewButton.restype = [], int
diff --git a/wrappers/python/eduvpn_common/main.py b/wrappers/python/eduvpn_common/main.py
index 69d00db..1467adb 100644
--- a/wrappers/python/eduvpn_common/main.py
+++ b/wrappers/python/eduvpn_common/main.py
@@ -440,6 +440,18 @@ class EduVPN(object):
         if renew_err:
             raise renew_err
 
+    def set_support_wireguard(self, support: bool) -> None:
+        """Indicates whether or not the OS supports WireGuard connections.
+
+        :param support: bool: whether or not wireguard is supported
+
+        :raises WrappedError: An error by the Go library
+        """
+        support_err = self.go_function(self.lib.SetSupportWireguard, support)
+
+        if support_err:
+            raise support_err
+
     def should_renew_button(self) -> bool:
         """Whether or not the UI should show the renew button