Swiftdiff options
| -rw-r--r-- | src/com/isode/stroke/network/JavaConnection.java | 36 |
1 files changed, 21 insertions, 15 deletions
diff --git a/src/com/isode/stroke/network/JavaConnection.java b/src/com/isode/stroke/network/JavaConnection.java index ade52f0..e5369bf 100644 --- a/src/com/isode/stroke/network/JavaConnection.java +++ b/src/com/isode/stroke/network/JavaConnection.java @@ -10,6 +10,7 @@ package com.isode.stroke.network; import java.io.IOException; import java.net.InetSocketAddress; +import java.net.Socket; import java.nio.ByteBuffer; import java.nio.channels.SocketChannel; import java.util.ArrayList; @@ -34,8 +35,8 @@ public class JavaConnection extends Connection implements EventOwner { public void run() { try { - InetSocketAddress isa = new InetSocketAddress(address_.getAddress().getInetAddress(),address_.getPort()); - socketChannel_ = SocketChannel.open(isa); + socketChannel_ = SocketChannel.open( + new InetSocketAddress(address_.getAddress().getInetAddress(),address_.getPort())); /* By default, SocketChannels start off in blocking mode, which * isn't what we want @@ -49,7 +50,7 @@ public class JavaConnection extends Connection implements EventOwner { while (!disconnecting_) { while (!writeBuffer_.isEmpty()) { ByteArray data = writeBuffer_.get(0); - ByteBuffer bb = ByteBuffer.wrap(data.getData()); + ByteBuffer byteBuffer = ByteBuffer.wrap(data.getData()); try { /* Because the SocketChannel is non-blocking, we have to * be prepared to cope with the write operation not @@ -57,8 +58,8 @@ public class JavaConnection extends Connection implements EventOwner { */ boolean finishedWriting = false; while (!finishedWriting && !disconnecting_) { - socketChannel_.write(bb); - finishedWriting = (bb.remaining() == 0); + socketChannel_.write(byteBuffer); + finishedWriting = (byteBuffer.remaining() == 0); if (!finishedWriting) { try { /* Give the output buffer a chance to empty */ @@ -78,18 +79,18 @@ public class JavaConnection extends Connection implements EventOwner { ByteArray data = new ByteArray(); try { - ByteBuffer bb = ByteBuffer.allocate(1024); + ByteBuffer byteBuffer = ByteBuffer.allocate(1024); - int count = socketChannel_.read(bb); + int count = socketChannel_.read(byteBuffer); while (count > 0) { - bb.flip(); - byte[] result = new byte[bb.remaining()]; - bb.get(result); - bb.compact(); + byteBuffer.flip(); + byte[] result = new byte[byteBuffer.remaining()]; + byteBuffer.get(result); + byteBuffer.compact(); for (int i=0; i<result.length; i++) { data.append(result[i]); } - count = socketChannel_.read(bb); + count = socketChannel_.read(byteBuffer); } if (count == -1) { /* socketChannel input has reached "end-of-stream", which @@ -164,7 +165,6 @@ public class JavaConnection extends Connection implements EventOwner { @Override public void connect(HostAddressPort address) { - hostAddressPort_ = address; worker_ = new Worker(address); Thread workerThread = new Thread(worker_); workerThread.setDaemon(true); @@ -183,7 +183,14 @@ public class JavaConnection extends Connection implements EventOwner { @Override public HostAddressPort getLocalAddress() { - return hostAddressPort_; + if (socketChannel_ == null) { + return null; + } + Socket socket = socketChannel_.socket(); + if (socket == null) { + return null; + } + return new HostAddressPort(new HostAddress(socket.getLocalAddress()), socket.getLocalPort()); } @Override @@ -198,6 +205,5 @@ public class JavaConnection extends Connection implements EventOwner { private boolean disconnecting_ = false; private SocketChannel socketChannel_; private Worker worker_; - private HostAddressPort hostAddressPort_; } |