Improved Security

git-svn-id: http://svn.igniterealtime.org/svn/repos/wildfire/trunk@6686 b35dd754-fafc-0310-a699-88a17e54d16e

src/java/org/jivesoftware/wildfire/mediaproxy/SmartSession.java

@@ -117,22 +117,21 @@ public class SmartSession extends MediaProxySession {
         public void run() {
             try {
                 long c = 0;
+                long band = System.currentTimeMillis();
                 while (true) {
                     // Block until a datagram appears:
                     packet = new DatagramPacket(buf, buf.length);
                     dataSocket.receive(packet);
 
-                    if (this.getPort() != packet.getPort())
-                        System.out.println(dataSocket.getLocalAddress().getHostAddress() + ":" + dataSocket.getLocalPort() + " relay to: " + packet.getAddress().getHostAddress() + ":" + packet.getPort());
-
-                    if (c++ < 5) {
-                        System.out.println("Received:" + dataSocket.getLocalAddress().getHostAddress() + ":" + dataSocket.getLocalPort());
-
-                        System.out.println("Addr: " + packet.getAddress().getHostName());
-                    }
-
+                    // Relay Destination
+                    if (c++ < 100) { // 100 packets are enough to discover relay address
                         this.setHost(packet.getAddress());
                         this.setPort(packet.getPort());
+                    } else {
+                        c = 1000; // Prevents long overflow
+                        // Check Source Address. If its different, discard packet. 
+                        if (!this.getHost().equals(packet.getAddress())) continue;
+                    }
 
                     boolean resend = true;