From dec275aa4dca6dade2ed2db79d5c61b69f782b1c Mon Sep 17 00:00:00 2001
From: vanitasvitae <vanitasvitae@fsfe.org>
Date: Fri, 30 Jun 2017 15:19:36 +0200
Subject: [PATCH] Add methods to enable/disable proxy types

---
 .../jingle_s5b/JingleS5BTransportManager.java | 18 ++++++++++++++++++
 .../jingle_s5b/JingleS5BTransportSession.java | 19 +++++++++++--------
 2 files changed, 29 insertions(+), 8 deletions(-)

diff --git a/smack-extensions/src/main/java/org/jivesoftware/smackx/jingle/transports/jingle_s5b/JingleS5BTransportManager.java b/smack-extensions/src/main/java/org/jivesoftware/smackx/jingle/transports/jingle_s5b/JingleS5BTransportManager.java
index 67e31f0b1..2274aaac7 100644
--- a/smack-extensions/src/main/java/org/jivesoftware/smackx/jingle/transports/jingle_s5b/JingleS5BTransportManager.java
+++ b/smack-extensions/src/main/java/org/jivesoftware/smackx/jingle/transports/jingle_s5b/JingleS5BTransportManager.java
@@ -55,6 +55,9 @@ public final class JingleS5BTransportManager extends JingleTransportManager<Jing
     private List<Bytestream.StreamHost> localStreamHosts = null;
     private List<Bytestream.StreamHost> availableStreamHosts = null;
 
+    private static boolean useLocalCandidates = true;
+    private static boolean useExternalCandidates = true;
+
     private JingleS5BTransportManager(XMPPConnection connection) {
         super(connection);
         JingleContentProviderManager.addJingleContentTransportProvider(getNamespace(), new JingleS5BTransportProvider());
@@ -213,4 +216,19 @@ public final class JingleS5BTransportManager extends JingleTransportManager<Jing
         return jingle;
     }
 
+    public static void setUseLocalCandidates(boolean localCandidates) {
+        JingleS5BTransportManager.useLocalCandidates = localCandidates;
+    }
+
+    public static void setUseExternalCandidates(boolean externalCandidates) {
+        JingleS5BTransportManager.useExternalCandidates = externalCandidates;
+    }
+
+    public static boolean isUseLocalCandidates() {
+        return useLocalCandidates;
+    }
+
+    public static boolean isUseExternalCandidates() {
+        return useExternalCandidates;
+    }
 }
diff --git a/smack-extensions/src/main/java/org/jivesoftware/smackx/jingle/transports/jingle_s5b/JingleS5BTransportSession.java b/smack-extensions/src/main/java/org/jivesoftware/smackx/jingle/transports/jingle_s5b/JingleS5BTransportSession.java
index c3f14aa85..ba03e1d58 100644
--- a/smack-extensions/src/main/java/org/jivesoftware/smackx/jingle/transports/jingle_s5b/JingleS5BTransportSession.java
+++ b/smack-extensions/src/main/java/org/jivesoftware/smackx/jingle/transports/jingle_s5b/JingleS5BTransportSession.java
@@ -78,16 +78,19 @@ public class JingleS5BTransportSession extends JingleTransportSession<JingleS5BT
                         Socks5Utils.createDigest(sid, jingleSession.getLocal(), jingleSession.getRemote()));
 
         //Local host
-        for (Bytestream.StreamHost host : transportManager().getLocalStreamHosts()) {
-            jb.addTransportCandidate(new JingleS5BTransportCandidate(host, 100, JingleS5BTransportCandidate.Type.direct));
+        if (JingleS5BTransportManager.isUseLocalCandidates()) {
+            for (Bytestream.StreamHost host : transportManager().getLocalStreamHosts()) {
+                jb.addTransportCandidate(new JingleS5BTransportCandidate(host, 100, JingleS5BTransportCandidate.Type.direct));
+            }
         }
 
-        List<Bytestream.StreamHost> remoteHosts;
-        try {
-            remoteHosts = transportManager().getAvailableStreamHosts();
-        } catch (InterruptedException | XMPPException.XMPPErrorException | SmackException.NotConnectedException | SmackException.NoResponseException e) {
-            LOGGER.log(Level.WARNING, "Could not determine available StreamHosts.", e);
-            remoteHosts = Collections.emptyList();
+        List<Bytestream.StreamHost> remoteHosts = Collections.emptyList();
+        if (JingleS5BTransportManager.isUseExternalCandidates()) {
+            try {
+                remoteHosts = transportManager().getAvailableStreamHosts();
+            } catch (InterruptedException | XMPPException.XMPPErrorException | SmackException.NotConnectedException | SmackException.NoResponseException e) {
+                LOGGER.log(Level.WARNING, "Could not determine available StreamHosts.", e);
+            }
         }
 
         for (Bytestream.StreamHost host : remoteHosts) {