/*
* Copyright 2006 Sun Microsystems, Inc. All Rights Reserved.
* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
*
* This code is free software; you can redistribute it and/or modify it
* under the terms of the GNU General Public License version 2 only, as
* published by the Free Software Foundation.
*
* This code is distributed in the hope that it will be useful, but WITHOUT
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
* version 2 for more details (a copy is included in the LICENSE file that
* accompanied this code).
*
* You should have received a copy of the GNU General Public License version
* 2 along with this work; if not, write to the Free Software Foundation,
* Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
*
* Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
* CA 95054 USA or visit www.sun.com if you need additional information or
* have any questions.
*/
/* @test
* @bug 6285901
* @summary Check no data is written to wrong socket channel during async closing.
* @author Xueming Shen
*/
import java.io.*;
import java.nio.*;
import java.nio.channels.*;
import java.net.*;
public class AsyncCloseChannel {
static boolean failed = false;
static boolean keepGoing = true;
static String host = "127.0.0.1";
static int sensorPort = 3010;
static int targetPort = 3020;
static int maxAcceptCount = 1000;
static int acceptCount = 0;
public static void main(String args[]) throws Exception {
if (System.getProperty("os.name").startsWith("Windows")) {
System.err.println("WARNING: Still does not work on Windows!");
return;
}
Thread ss = new SensorServer(); ss.start();
Thread ts = new TargetServer(); ts.start();
Thread sc = new SensorClient(); sc.start();
Thread tc = new TargetClient(); tc.start();
while(acceptCount < maxAcceptCount && !failed) {
Thread.yield();
}
keepGoing = false;
try {
ss.interrupt();
ts.interrupt();
sc.interrupt();
tc.interrupt();
} catch (Exception e) {}
if (failed)
throw new RuntimeException("AsyncCloseChannel2 failed after <"
+ acceptCount + "> times of accept!");
}
static class SensorServer extends ThreadEx {
public void runEx() throws Exception {
ServerSocket server;
server = new ServerSocket(sensorPort);
while(keepGoing) {
try {
final Socket s = server.accept();
new Thread() {
public void run() {
try {
int c = s.getInputStream().read();
if(c != -1) {
// No data is ever written to the peer's socket!
System.out.println("Oops: read a character: "
+ (char) c);
failed = true;
}
} catch (IOException ex) {
ex.printStackTrace();
} finally {
closeIt(s);
}
}
}.start();
} catch (IOException ex) {
//ex.printStackTrace();
}
}
}
}
static class TargetServer extends ThreadEx {
public void runEx() throws Exception {
ServerSocket server;
server = new ServerSocket(targetPort);
while (keepGoing) {
try {
final Socket s = server.accept();
acceptCount++;
new Thread() {
public void run() {
boolean empty = true;
try {
for(;;) {
int c = s.getInputStream().read();
if(c == -1) {
if(!empty)
break;
}
empty = false;
}
} catch (IOException ex) {
ex.printStackTrace();
} finally {
closeIt(s);
}
}
}.start();
} catch (IOException ex) {
//ex.printStackTrace();
}
}
}
}
static class SensorClient extends Thread {
private static boolean wake;
private static SensorClient theClient;
public void run() {
while (keepGoing) {
Socket s = null;
try {
s = new Socket();
synchronized(this) {
while(!wake) {
try {
wait();
} catch (InterruptedException ex) { }
}
}
wake = false;
s.connect(new InetSocketAddress(host, sensorPort));
try {
Thread.sleep(10);
} catch (InterruptedException ex) { }
} catch (IOException ex) {
System.out.println("Exception on sensor client " + ex.getMessage());
} finally {
if(s != null) {
try {
s.close();
} catch(IOException ex) { ex.printStackTrace();}
}
}
}
}
public SensorClient() {
theClient = this;
}
public static void wakeMe() {
synchronized(theClient) {
wake = true;
theClient.notify();
}
}
}
static class TargetClient extends Thread {
volatile boolean ready = false;
public void run() {
while(keepGoing) {
try {
final SocketChannel s = SocketChannel.open(
new InetSocketAddress(host, targetPort));
s.finishConnect();
s.socket().setSoLinger(false, 0);
ready = false;
Thread t = new Thread() {
public void run() {
ByteBuffer b = ByteBuffer.allocate(1);
try {
for(;;) {
b.clear();
b.put((byte) 'A');
b.flip();
s.write(b);
ready = true;
}
} catch (IOException ex) {
if(!(ex instanceof ClosedChannelException))
System.out.println("Exception in target client child "
+ ex.toString());
}
}
};
t.start();
while(!ready)
Thread.yield();
s.close();
SensorClient.wakeMe();
t.join();
} catch (IOException ex) {
System.out.println("Exception in target client parent "
+ ex.getMessage());
} catch (InterruptedException ex) {}
}
}
}
static abstract class ThreadEx extends Thread {
public void run() {
try {
runEx();
} catch (Exception ex) {
ex.printStackTrace();
}
}
abstract void runEx() throws Exception;
}
public static void closeIt(Socket s) {
try {
if(s != null)
s.close();
} catch (IOException ex) { }
}
}