242 lines
8.4 KiB
Java
242 lines
8.4 KiB
Java
|
/*
|
||
|
|
* 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) { }
|
||
|
|
}
|
||
|
|
}
|