From 3fd4df2a8ff280bfa31a6d952d5f213b0ce63747 Mon Sep 17 00:00:00 2001
From: Thiago Camargo <thiago@b35dd754-fafc-0310-a699-88a17e54d16e>
Date: Thu, 15 Mar 2007 22:09:21 +0000
Subject: [PATCH] Public IP Discovering

git-svn-id: http://svn.igniterealtime.org/svn/repos/smack/trunk@7562 b35dd754-fafc-0310-a699-88a17e54d16e
---
 .../smackx/jingle/nat/RTPBridge.java          | 87 ++++++++++++++++++-
 1 file changed, 83 insertions(+), 4 deletions(-)

diff --git a/jingle/extension/source/org/jivesoftware/smackx/jingle/nat/RTPBridge.java b/jingle/extension/source/org/jivesoftware/smackx/jingle/nat/RTPBridge.java
index 22a9a0b4d..7adc40cf7 100644
--- a/jingle/extension/source/org/jivesoftware/smackx/jingle/nat/RTPBridge.java
+++ b/jingle/extension/source/org/jivesoftware/smackx/jingle/nat/RTPBridge.java
@@ -33,6 +33,10 @@ import org.jivesoftware.smackx.packet.DiscoverItems;
 import org.xmlpull.v1.XmlPullParser;
 
 import java.util.Iterator;
+import java.util.Enumeration;
+import java.net.NetworkInterface;
+import java.net.SocketException;
+import java.net.InetAddress;
 
 /**
  * RTPBridge IQ Packet used to request and retrieve a RTPBridge Candidates that can be used for a Jingle Media Transmission between two parties that are behind NAT.
@@ -58,7 +62,8 @@ public class RTPBridge extends IQ {
     private BridgeAction bridgeAction = BridgeAction.create;
 
     private enum BridgeAction {
-        create, change
+
+        create, change, publicip
     }
 
     /**
@@ -89,6 +94,15 @@ public class RTPBridge extends IQ {
         this.sid = sid;
     }
 
+    /**
+     * Creates a RTPBridge Instance with defined Session ID
+     *
+     * @param action
+     */
+    public RTPBridge(BridgeAction action) {
+        this.bridgeAction = action;
+    }
+
     /**
      * Creates a RTPBridge Instance with defined Session ID
      *
@@ -288,8 +302,10 @@ public class RTPBridge extends IQ {
 
         if (bridgeAction.equals(BridgeAction.create))
             str.append("<candidate/>");
-        else
+        else if (bridgeAction.equals(BridgeAction.change))
             str.append("<relay ").append(getAttributes()).append(" />");
+        else
+            str.append("<publicip ").append(getAttributes()).append(" />");
 
         str.append("</" + ELEMENT_NAME + ">");
         return str.toString();
@@ -346,8 +362,16 @@ public class RTPBridge extends IQ {
                                 iq.setPortB(Integer.parseInt(parser.getAttributeValue(i)));
                         }
                     }
-
-                } else if (eventType == XmlPullParser.END_TAG) {
+                    else if (elementName.equals("publicip")) {
+                        //String p = parser.getAttributeName(0);
+                        int x = parser.getAttributeCount();
+                        for (int i = 0; i < parser.getAttributeCount(); i++) {
+                            if (parser.getAttributeName(i).equals("ip"))
+                                iq.setIp(parser.getAttributeValue(i));
+                        }
+                    }
+                }
+                else if (eventType == XmlPullParser.END_TAG) {
                     if (parser.getName().equals(RTPBridge.ELEMENT_NAME)) {
                         done = true;
                     }
@@ -459,4 +483,59 @@ public class RTPBridge extends IQ {
         return response;
     }
 
+    /**
+     * Get Public Address from the Server.
+     *
+     * @param xmppConnection
+     * @return public IP String or null if not found
+     */
+    public static String getPublicIP(XMPPConnection xmppConnection) {
+
+        if (!xmppConnection.isConnected()) {
+            return null;
+        }
+
+        RTPBridge rtpPacket = new RTPBridge(RTPBridge.BridgeAction.publicip);
+        rtpPacket.setTo(RTPBridge.NAME + "." + xmppConnection.getServiceName());
+        rtpPacket.setType(Type.SET);
+
+        // System.out.println("Relayed to: " + candidate.getIp() + ":" + candidate.getPort());
+
+        PacketCollector collector = xmppConnection
+                .createPacketCollector(new PacketIDFilter(rtpPacket.getPacketID()));
+
+        xmppConnection.sendPacket(rtpPacket);
+
+        RTPBridge response = (RTPBridge) collector
+                .nextResult(SmackConfiguration.getPacketReplyTimeout());
+
+        // Cancel the collector.
+        collector.cancel();
+
+        if (response.getIp() == null || response.getIp().equals("")) return null;
+
+        Enumeration ifaces = null;
+        try {
+            ifaces = NetworkInterface.getNetworkInterfaces();
+        }
+        catch (SocketException e) {
+            e.printStackTrace();
+        }
+        while (ifaces!=null&&ifaces.hasMoreElements()) {
+
+            NetworkInterface iface = (NetworkInterface) ifaces.nextElement();
+            Enumeration iaddresses = iface.getInetAddresses();
+
+            while (iaddresses.hasMoreElements()) {
+                InetAddress iaddress = (InetAddress) iaddresses.nextElement();
+                if (!iaddress.isLoopbackAddress())
+                    if (iaddress.getHostAddress().indexOf(response.getIp()) >= 0)
+                        return null;
+
+            }
+        }
+
+        return response.getIp();
+    }
+
 }