NetworkStream: tag received NetworkMessage with sender's IP protocol, address:port...

diff --git a/src/uk/ac/ntu/n0521366/wsyd/libs/net/NetworkStream.java b/src/uk/ac/ntu/n0521366/wsyd/libs/net/NetworkStream.java
index 6125720..01d6789 100644 (file)
--- a/src/uk/ac/ntu/n0521366/wsyd/libs/net/NetworkStream.java
+++ b/src/uk/ac/ntu/n0521366/wsyd/libs/net/NetworkStream.java
@@ -210,6 +210,7 @@ public class NetworkStream extends SwingWorker<Boolean, NetworkMessage> {
 
             message = (NetworkMessage)_ois.readObject();
             message.setKey(_key);
+            message._receivedFrom = new WSYD_SocketAddress(this.getRemoteAddress(), WSYD_SocketAddress.Protocol.TCP);
             publish(message);
 
             result = true;