From f19bedac1f78c7c54a1cffe5ff53e3aaa4de8fc7 Mon Sep 17 00:00:00 2001
From: Thiago Camargo <thiago@b35dd754-fafc-0310-a699-88a17e54d16e>
Date: Thu, 22 Mar 2007 19:57:50 +0000
Subject: [PATCH] Multiple STUN Servers Support

git-svn-id: http://svn.igniterealtime.org/svn/repos/smack/trunk@7650 b35dd754-fafc-0310-a699-88a17e54d16e
---
 .../jivesoftware/smackx/jingle/nat/STUN.java  | 81 +++++++++++--------
 .../smackx/jingle/nat/STUNResolverTest.java   |  5 +-
 2 files changed, 49 insertions(+), 37 deletions(-)

diff --git a/jingle/extension/source/org/jivesoftware/smackx/jingle/nat/STUN.java b/jingle/extension/source/org/jivesoftware/smackx/jingle/nat/STUN.java
index c78e828fa..bab56abec 100644
--- a/jingle/extension/source/org/jivesoftware/smackx/jingle/nat/STUN.java
+++ b/jingle/extension/source/org/jivesoftware/smackx/jingle/nat/STUN.java
@@ -33,6 +33,8 @@ import org.jivesoftware.smackx.packet.DiscoverItems;
 import org.xmlpull.v1.XmlPullParser;
 
 import java.util.Iterator;
+import java.util.List;
+import java.util.ArrayList;
 
 /**
  * STUN IQ Packet used to request and retrieve a STUN server and port to make p2p connections easier. STUN is usually used by Jingle Media Transmission between two parties that are behind NAT.
@@ -45,8 +47,7 @@ import java.util.Iterator;
  */
 public class STUN extends IQ {
 
-    private int port = -1;
-    private String host;
+    private List<StunServerAddress> servers = new ArrayList<StunServerAddress>();
 
     /**
      * Element name of the packet extension.
@@ -74,39 +75,12 @@ public class STUN extends IQ {
     }
 
     /**
-     * Get the Host Address
+     * Get a list of STUN Servers recommended by the Server
      *
      * @return
      */
-    public String getHost() {
-        return host;
-    }
-
-    /**
-     * Set the Host Address
-     *
-     * @param host
-     */
-    public void setHost(String host) {
-        this.host = host;
-    }
-
-    /**
-     * Get STUN port
-     *
-     * @return
-     */
-    public int getPort() {
-        return port;
-    }
-
-    /**
-     * Set STUN port
-     *
-     * @param port
-     */
-    public void setPort(int port) {
-        this.port = port;
+    public List<StunServerAddress> getServers() {
+        return servers;
     }
 
     /**
@@ -153,14 +127,19 @@ public class STUN extends IQ {
 
                 if (eventType == XmlPullParser.START_TAG) {
                     if (elementName.equals("server")) {
+                        String host = null;
+                        String port = null;
                         for (int i = 0; i < parser.getAttributeCount(); i++) {
                             if (parser.getAttributeName(i).equals("host"))
-                                iq.setHost(parser.getAttributeValue(i));
+                                host = parser.getAttributeValue(i);
                             else if (parser.getAttributeName(i).equals("udp"))
-                                iq.setPort(Integer.parseInt(parser.getAttributeValue(i)));
+                                port = parser.getAttributeValue(i);
                         }
+                        if (host != null && port != null)
+                            iq.servers.add(new StunServerAddress(host, port));
                     }
-                } else if (eventType == XmlPullParser.END_TAG) {
+                }
+                else if (eventType == XmlPullParser.END_TAG) {
                     if (parser.getName().equals(ELEMENT_NAME)) {
                         done = true;
                     }
@@ -239,4 +218,36 @@ public class STUN extends IQ {
         }
         return false;
     }
+
+    /**
+     * Provides easy abstract to store STUN Server Addresses and Ports
+     */
+    public static class StunServerAddress {
+
+        private String server;
+        private String port;
+
+        public StunServerAddress(String server, String port) {
+            this.server = server;
+            this.port = port;
+        }
+
+        /**
+         * Get the Host Address
+         *
+         * @return
+         */
+        public String getServer() {
+            return server;
+        }
+
+        /**
+         * Get the Server Port
+         *
+         * @return
+         */
+        public String getPort() {
+            return port;
+        }
+    }
 }
diff --git a/jingle/extension/test/org/jivesoftware/smackx/jingle/nat/STUNResolverTest.java b/jingle/extension/test/org/jivesoftware/smackx/jingle/nat/STUNResolverTest.java
index 79ad92dfa..e996ea1c0 100644
--- a/jingle/extension/test/org/jivesoftware/smackx/jingle/nat/STUNResolverTest.java
+++ b/jingle/extension/test/org/jivesoftware/smackx/jingle/nat/STUNResolverTest.java
@@ -169,8 +169,9 @@ public class STUNResolverTest extends SmackTestCase {
 
         System.out.println(STUN.serviceAvailable(getConnection(0)));
         STUN stun = STUN.getSTUNServer(getConnection(0));
-        System.out.println(stun.getHost() + ":" + stun.getPort());
-
+        for (STUN.StunServerAddress stunServerAddress : stun.getServers())
+            System.out.println(stunServerAddress.getServer() + ":" + stunServerAddress.getPort());
+        
     }
 
     /**