andreacavalli/netty5
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

System call needs to be made at least once so that it can fail when a user attempts to write on a closed channel

Browse Source
This commit is contained in:
Trustin Lee 2010-06-16 03:33:29 +00:00
parent 752e60a723
commit de90cd6a3c
4 changed files with 32 additions and 27 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
src/main/java/org/jboss/netty/channel/socket/nio/NioDatagramWorker.java
View File

@ -530,20 +530,26 @@ class NioDatagramWorker implements Runnable {
long localWrittenBytes = 0; long localWrittenBytes = 0;
SocketAddress raddr = evt.getRemoteAddress(); SocketAddress raddr = evt.getRemoteAddress();
if (raddr == null) { if (raddr == null) {
for (int i = writeSpinCount; i > 0 && !buf.finished(); i --) { for (int i = writeSpinCount; i > 0; i --) {
localWrittenBytes = buf.transferTo(ch); localWrittenBytes = buf.transferTo(ch);
if (localWrittenBytes != 0) { if (localWrittenBytes != 0) {
writtenBytes += localWrittenBytes; writtenBytes += localWrittenBytes;
break; break;
} }
if (buf.finished()) {
break;
}
} }
} else { } else {
for (int i = writeSpinCount; i > 0 && !buf.finished(); i --) { for (int i = writeSpinCount; i > 0; i --) {
localWrittenBytes = buf.transferTo(ch, raddr); localWrittenBytes = buf.transferTo(ch, raddr);
if (localWrittenBytes != 0) { if (localWrittenBytes != 0) {
writtenBytes += localWrittenBytes; writtenBytes += localWrittenBytes;
break; break;
} }
if (buf.finished()) {
break;
}
} }
} }

5
src/main/java/org/jboss/netty/channel/socket/nio/NioWorker.java
View File

@ -465,12 +465,15 @@ class NioWorker implements Runnable {
ChannelFuture future = evt.getFuture(); ChannelFuture future = evt.getFuture();
try { try {
long localWrittenBytes = 0; long localWrittenBytes = 0;
for (int i = writeSpinCount; i > 0 && !buf.finished(); i --) { for (int i = writeSpinCount; i > 0; i --) {
localWrittenBytes = buf.transferTo(ch); localWrittenBytes = buf.transferTo(ch);
if (localWrittenBytes != 0) { if (localWrittenBytes != 0) {
writtenBytes += localWrittenBytes; writtenBytes += localWrittenBytes;
break; break;
} }
if (buf.finished()) {
break;
}
} }
if (buf.finished()) { if (buf.finished()) {

2
src/main/java/org/jboss/netty/channel/socket/oio/OioDatagramWorker.java
View File

@ -101,7 +101,6 @@ class OioDatagramWorker implements Runnable {
try { try {
ChannelBuffer buf = (ChannelBuffer) message; ChannelBuffer buf = (ChannelBuffer) message;
int length = buf.readableBytes(); int length = buf.readableBytes();
if (length > 0) {
ByteBuffer nioBuf = buf.toByteBuffer(); ByteBuffer nioBuf = buf.toByteBuffer();
DatagramPacket packet; DatagramPacket packet;
if (nioBuf.hasArray()) { if (nioBuf.hasArray()) {
@ -120,7 +119,6 @@ class OioDatagramWorker implements Runnable {
} }
channel.socket.send(packet); channel.socket.send(packet);
fireWriteComplete(channel, length); fireWriteComplete(channel, length);
}
future.setSuccess(); future.setSuccess();
} catch (Throwable t) { } catch (Throwable t) {
future.setFailure(t); future.setFailure(t);

2
src/main/java/org/jboss/netty/channel/socket/oio/OioWorker.java
View File

@ -115,12 +115,10 @@ class OioWorker implements Runnable {
try { try {
ChannelBuffer a = (ChannelBuffer) message; ChannelBuffer a = (ChannelBuffer) message;
int length = a.readableBytes(); int length = a.readableBytes();
if (length > 0) {
synchronized (out) { synchronized (out) {
a.getBytes(a.readerIndex(), out, length); a.getBytes(a.readerIndex(), out, length);
} }
fireWriteComplete(channel, length); fireWriteComplete(channel, length);
}
future.setSuccess(); future.setSuccess();
} catch (Throwable t) { } catch (Throwable t) {
// Convert 'SocketException: Socket closed' to // Convert 'SocketException: Socket closed' to