blob: 6ca63cb0f0f43380c5833f03718e70209ddfbb7b [file] [log] [blame]
/*
* Copyright (c) 2001, 2018, Oracle and/or its affiliates. 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 Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
* or visit www.oracle.com if you need additional information or have any
* questions.
*/
package nsk.share.jpda;
import java.io.*;
import java.net.*;
import nsk.share.*;
/**
* This class implements basic connection channel via TCP/IP sockets.
*/
class BasicSocketConnection {
protected static int TRACE_LEVEL_PACKETS = 10;
protected static int TRACE_LEVEL_THREADS = 20;
protected static int TRACE_LEVEL_ACTIONS = 30;
protected static int TRACE_LEVEL_SOCKETS = 40;
protected static int TRACE_LEVEL_IO = 50;
protected String name = null;
protected ServerSocket serverSocket = null;
protected Socket socket = null;
protected InputStream sin = null;
protected OutputStream sout = null;
protected Process connectingProcess = null;
protected volatile boolean connected = false;
protected volatile boolean closed = false;
protected volatile boolean connectionClosed = false;
protected volatile boolean shouldStop = false;
protected Log.Logger logger = null;
/**
* Make an empty connection with specified name.
*
* @param logger
* Logger object for printing log messages
* @param name
* connection name
*/
public BasicSocketConnection(Log.Logger logger, String name) {
this.logger = logger;
this.name = name;
}
/**
* Try to bind connection to the local port.
*
* @param port
* port number to bind to
*
* @throws IOException
* if error occured while binding
*/
protected void tryBind(int port) throws IOException {
logger.trace(TRACE_LEVEL_IO, "Binding for " + name + " connection to port: " + port);
serverSocket = new ServerSocket(port, 1);
logger.trace(TRACE_LEVEL_IO, "Bound for " + name + " connection to port: " + port);
}
/**
* Bind connection to the local port for specified timeout.
*
* @param port
* port number to bind to
* @param timeout
* binding timeout in milliseconds
*
* @throws Failure
* if error ocured while binding
*/
protected void bind(int port, long timeout) {
BindException bindException = null;
long timeToFinish = System.currentTimeMillis() + timeout;
for (long i = 0; !shouldStop && (timeout == 0 || System.currentTimeMillis() < timeToFinish); i++) {
try {
tryBind(port);
return;
} catch (BindException e) {
bindException = e;
logger.display("Attempt #" + i + " to bind to port " + port + " failed:\n\t" + e);
try {
Thread.sleep(DebugeeBinder.CONNECT_TRY_DELAY);
} catch (InterruptedException ie) {
ie.printStackTrace(logger.getOutStream());
throw new Failure("Thread interrupted while binding for " + name + " connection to port " + port + ":\n\t" + ie);
}
} catch (IOException e) {
e.printStackTrace(logger.getOutStream());
throw new Failure("Caught IOException while binding for " + name + " connection to port " + port + ":\n\t" + e);
}
}
throw new Failure("Unable to bind for " + name + " connection to port " + port + " for " + timeout + "ms timeout:\n\t" + bindException);
}
/**
* Accept connection at the bound port for specified timeout.
*
* @param timeout
* accepting timeout in milliseconds
*
* @throws Failure
* if error occured while accepting connection
*/
public void accept(long timeout) {
int port = serverSocket.getLocalPort();
logger.trace(TRACE_LEVEL_IO, "Listening for " + name + " connection at port: " + port);
socket = null;
try {
if (timeout > Integer.MAX_VALUE) {
throw new TestBug("Too large timeout long value: " + timeout + " (can't cast it to int)");
}
serverSocket.setSoTimeout((int)timeout);
long waitStartTime = System.currentTimeMillis();
/*
* We found that sometimes (very rarely) on Solaris ServerSocket.accept() throws InterruptedIOException
* even if connection timeout (specified through ServerSocket.setSoTimeout) didn't expire.
* Following code tries to catch such case and call ServerSocket.accept() while timeout didn't expire.
*/
do {
try {
socket = serverSocket.accept();
logger.trace(TRACE_LEVEL_IO, "Accepted " + name + " connection at port: " + port);
} catch (InterruptedIOException e) {
long interruptTime = System.currentTimeMillis();
long waitTime = interruptTime - waitStartTime;
logger.display("Caught InterruptedIOException. Wait start time: " + waitStartTime + ", exception was thrown at: "
+ interruptTime + ", wait time: " + (interruptTime - waitStartTime) + ", actual timeout: " + timeout);
// if waitTime was too small call ServerSocket.accept() one more time
if (!shouldStop && (waitTime < (timeout / 2))) {
logger.display("InterruptedIOException was thrown too early, trying to call ServerSocket.accept() one more time");
continue;
} else {
if (!shouldStop) {
logger.complain("Caught InterruptedIOException while listening for " + name + " connection at port " + port + ":\n\t" + e);
throw new Failure("Connection for " + name +
" at port " + port +
" wasn't accepted in " + timeout + "ms");
} else {
logger.display("Listening was interrupted (caught InterruptedIOException while listening for " + name + " connection at port " + port + ":\n\t" + e + ")");
break;
}
}
}
} while (socket == null);
} catch (IOException e) {
if (!shouldStop) {
e.printStackTrace(logger.getOutStream());
throw new Failure("Caught IOException while listening for " + name + " connection at port " + port + ":\n\t" + e);
} else {
logger.display("Listening was interrupted (caught InterruptedIOException while listening for " + name + " connection at port " + port + ":\n\t" + e + ")");
}
} finally {
closeServerConnection();
}
if (!shouldStop) {
if (socket == null) {
throw new Failure("No " + name + " connection accepted at port " + port + " for " + timeout + "ms timeout");
}
onConnected();
}
}
/**
* Attach connection to the remote host and port.
*
* @param host
* name of remote host to attach to
* @param port
* port number to attach to
*
* @throws Failure
* if error occured while attaching
*/
public void attach(String host, int port) {
try {
logger.trace(TRACE_LEVEL_IO, "Attaching for " + name + " connection to host: " + host + ":" + port);
socket = new Socket(host, port);
socket.setTcpNoDelay(true);
logger.trace(TRACE_LEVEL_IO, "Attached for " + name + " connection to host: " + host + ":" + port);
} catch (IOException e) {
e.printStackTrace(logger.getOutStream());
throw new Failure("Caught IOException while attaching for " + name + " connection to " + host + ":" + port + ":\n\t" + e);
}
if (!shouldStop) {
onConnected();
}
}
/**
* Continuously attach to the remote host for the specified timeout.
*
* @param host
* name of remote host to attach to
* @param port
* port number to attach to
* @param timeout
* attaching timeout in milliseconds
*
* @throws Failure
* if error occured while attaching
*/
public void continueAttach(String host, int port, long timeout) {
socket = null;
long timeToFinish = System.currentTimeMillis() + timeout;
ConnectException lastException = null;
logger.trace(TRACE_LEVEL_IO, "Attaching for " + name + " connection to host: " + host + ":" + port);
try {
for (long i = 0; !shouldStop && (timeout == 0 || System.currentTimeMillis() < timeToFinish); i++) {
try {
socket = new Socket(host, port);
logger.trace(TRACE_LEVEL_IO, "Attached for " + name + " connection to host: " + host + ":" + port);
break;
} catch (ConnectException e) {
logger.display("Attempt #" + i + " to attach for " + name + " connection failed:\n\t" + e);
lastException = e;
// check if listening process still alive
if (!checkConnectingProcess()) {
shouldStop = true;
throw new Failure("Break attaching to " + name + " connection: " + "listening process exited");
}
// sleep between attempts
try {
Thread.sleep(DebugeeBinder.CONNECT_TRY_DELAY);
} catch (InterruptedException ie) {
throw new Failure("Thread interrupted while attaching for " + name + " connection to " + host + ":" + port + ":\n\t" + ie);
}
}
}
} catch (IOException e) {
e.printStackTrace(logger.getOutStream());
throw new Failure("Caught IOException while attaching for " + name + " connection to " + host + ":" + port + ":\n\t" + e);
}
if (!shouldStop) {
if (socket == null) {
throw new Failure("Unable to attach for " + name + " connection to " + host + ":" + port + " for " + timeout + "ms timeout:\n\t"
+ lastException);
}
onConnected();
}
}
/**
* Set already bound serverSocket for further connection.
*/
public void setServerSocket(ServerSocket serverSocket) {
this.serverSocket = serverSocket;
}
/**
* Set already connected socket for connection.
*/
public void setSocket(Socket socket) {
this.socket = socket;
if (!shouldStop) {
onConnected();
}
}
/**
* Get socket of already established connection.
*/
public Socket getSocket() {
return socket;
}
/**
* Return true if another connecting process is still alive.
*/
public boolean checkConnectingProcess() {
if (connectingProcess == null) {
// no process to check
return true;
}
try {
int exitCode = connectingProcess.exitValue();
} catch (IllegalThreadStateException e) {
// process is still alive
return true;
}
// process exited
return false;
}
/**
* Set another connecting process to control if it is still alive.
*/
public void setConnectingProcess(Process process) {
connectingProcess = process;
}
/**
* Check if connection is established.
*/
public boolean isConnected() {
return connected;
}
/**
* Close socket and associated streams.
*/
public void close() {
if (!closed) {
shouldStop = true;
closeConnection();
closed = true;
}
}
/**
* Send the specified byte throw the connection.
*/
public void writeByte(byte b) throws IOException {
logger.trace(TRACE_LEVEL_IO, "Writing byte: " + b);
sout.write(b);
sout.flush();
logger.trace(TRACE_LEVEL_IO, "Wrote byte: " + b);
}
/**
* Read a byte and return it or -1.
*/
public int readByte() throws IOException {
logger.trace(TRACE_LEVEL_IO, "Reading byte");
int b = sin.read();
logger.trace(TRACE_LEVEL_IO, "Received byte: " + b);
return b;
}
/**
* Perform some actions after connection established.
*/
protected void onConnected() {
if (!shouldStop) {
setSocketOptions();
makeSocketStreams();
connected = true;
}
}
/**
* Set socket options after connection established.
*/
protected void setSocketOptions() {
}
/**
* Close server socket.
*/
protected void closeServerConnection() {
if (serverSocket != null) {
try {
serverSocket.close();
logger.trace(TRACE_LEVEL_IO, "ServerSocket closed: " + serverSocket);
} catch (IOException e) {
logger.display("# WARNING: " + "Caught IOException while closing ServerSocket of " + name + " connection:\n\t" + e);
}
}
}
/**
* Close socket of connection to remote host.
*/
protected void closeHostConnection() {
if (socket != null) {
try {
socket.close();
logger.trace(TRACE_LEVEL_IO, "Socket closed: " + socket);
} catch (IOException e) {
logger.display("# WARNING: " + "Caught IOException while closing socket of " + name + " connection:\n\t" + e);
}
}
}
/**
* Close socket streams.
*/
protected void closeSocketStreams() {
if (sout != null) {
try {
logger.trace(TRACE_LEVEL_IO, "Closing socket output stream: " + sout);
sout.close();
logger.trace(TRACE_LEVEL_IO, "Output stream closed: " + sout);
} catch (IOException e) {
logger.display("# WARNING: " + "Caught IOException while closing OutputStream of " + name + " connection:\n\t" + e);
}
}
if (sin != null) {
try {
logger.trace(TRACE_LEVEL_IO, "Closing socket input stream: " + sin);
sin.close();
logger.trace(TRACE_LEVEL_IO, "Input stream closed: " + sin);
} catch (IOException e) {
logger.display("# WARNING: " + "Caught IOException while closing InputStream of" + name + " connection:\n\t" + e);
}
}
}
/**
* Close sockets and associated streams.
*/
protected void closeConnection() {
if (connectionClosed)
return;
logger.trace(TRACE_LEVEL_IO, "Closing " + name + " connection");
closeSocketStreams();
closeHostConnection();
closeServerConnection();
connectionClosed = true;
}
/**
* Make up socket streams after connection established.
*/
protected void makeSocketStreams() {
try {
logger.trace(TRACE_LEVEL_IO, "Getting input/output socket streams for " + name + " connection");
sout = socket.getOutputStream();
logger.trace(TRACE_LEVEL_IO, "Got socket output stream: " + sout);
sin = socket.getInputStream();
logger.trace(TRACE_LEVEL_IO, "Got socket input stream: " + sin);
} catch (IOException e) {
e.printStackTrace(logger.getOutStream());
throw new Failure("Caught exception while making streams for " + name + " connection:\n\t" + e);
}
}
} // BasicSocketConnection
/**
* This class implements connection channel via TCP/IP sockets. After connection
* established special inner threads are started, which periodically test the
* connection by pinging each other. If ping timeout occurs connection is closed
* and any thread waiting for read from this connection gets exception.
*
* @see #setPingTimeout(long)
*/
public class SocketConnection extends BasicSocketConnection {
private static final long PING_INTERVAL = 1 * 1000; // milliseconds
private static byte DATA_BYTE = (byte) 0x03;
private static byte DISCONNECT_BYTE = (byte) 0x04;
private final Object inLock = new Object();
private ObjectInputStream in = null;
private final Object outLock = new Object();
private ObjectOutputStream out = null;
private volatile long pingTimeout = 0; // don't use ping
/**
* Make an empty connection with specified name.
*
* @param log
* Log object for printing log messages
* @param name
* connection name
*/
public SocketConnection(Log log, String name) {
this(new Log.Logger(log, name + " connection> "), name);
}
/**
* Make an empty connection with specified name.
*
* @param logger
* Logger object for printing log messages
* @param name
* connection name
*/
public SocketConnection(Log.Logger logger, String name) {
super(logger, name);
}
/**
* Set ping timeout in milliseconds (0 means don't use ping at all).
*/
public void setPingTimeout(long timeout) {
logger.display("# WARNING: Setting ping timeout for " + name + " connection ingnored: " + timeout + " ms");
pingTimeout = timeout;
}
/**
* Returns value of current ping timeout in milliseconds (0 means ping is
* not used).
*/
public long getPingTimeout() {
return pingTimeout;
}
/**
* Receive an object from remote host.
*/
public Object readObject() {
if (!isConnected()) {
throw new Failure("Unable to read object from not established " + name + " connection");
}
try {
return doReadObject();
} catch (EOFException e) {
return null;
} catch (Exception e) {
e.printStackTrace(logger.getOutStream());
throw new Failure("Caught Exception while reading an object from " + name + " connection:\n\t" + e);
}
}
/**
* Send an object to remote host.
*/
public void writeObject(Object object) {
if (!isConnected()) {
throw new Failure("Unable to send object throw not established " + name + " connection:\n\t" + object);
}
try {
doWriteObject(object);
} catch (IOException e) {
e.printStackTrace(logger.getOutStream());
throw new Failure("Caught IOException while writing an object to " + name + " connection:\n\t" + e);
}
}
/**
* Close socket and associated streams and finish all internal threads.
*/
public void close() {
if (!closed) {
// disconnect();
shouldStop = true;
super.close();
closed = true;
}
}
/**
* Perform some actions after connection has established.
*/
protected void onConnected() {
super.onConnected();
}
/**
* Do write an object to the connection channel.
*/
private void doWriteObject(Object object) throws IOException {
logger.trace(TRACE_LEVEL_IO, "writing object: " + object);
synchronized(outLock) {
out.writeObject(object);
out.flush();
}
logger.trace(TRACE_LEVEL_PACKETS, "* sent: " + object);
}
/**
* Do read an object from the connection channel.
*/
private Object doReadObject() throws IOException, ClassNotFoundException {
logger.trace(TRACE_LEVEL_IO, "Reading object");
Object object = null;
synchronized(inLock) {
object = in.readObject();
}
logger.trace(TRACE_LEVEL_PACKETS, "* recv: " + object);
return object;
}
/**
* Close socket streams.
*/
protected void closeSocketStreams() {
synchronized(outLock) {
if (out != null) {
try {
logger.trace(TRACE_LEVEL_IO, "Closing socket output stream: " + out);
out.close();
logger.trace(TRACE_LEVEL_IO, "Output stream closed: " + out);
} catch (IOException e) {
logger.display("# WARNING: " + "Caught IOException while closing ObjectOutputStream of " + name + " connection:\n\t" + e);
}
}
}
synchronized(inLock) {
if (in != null) {
try {
logger.trace(TRACE_LEVEL_IO, "Closing socket input stream: " + in);
in.close();
logger.trace(TRACE_LEVEL_IO, "Input stream closed: " + in);
} catch (IOException e) {
logger.display("# WARNING: " + "Caught IOException while closing ObjectInputStream of" + name + " connection:\n\t" + e);
}
}
}
super.closeSocketStreams();
}
/**
* Close sockets and associated streams.
*/
protected void closeConnection() {
if (connectionClosed)
return;
connected = false;
shouldStop = true;
super.closeConnection();
}
/**
* Make up object streams for socket.
*/
protected void makeSocketStreams() {
try {
logger.trace(TRACE_LEVEL_IO, "Making input/output object streams for " + name + " connection");
synchronized(outLock) {
out = new ObjectOutputStream(socket.getOutputStream());
out.flush();
}
logger.trace(TRACE_LEVEL_IO, "Output stream created: " + out);
synchronized(inLock) {
in = new ObjectInputStream(socket.getInputStream());
}
logger.trace(TRACE_LEVEL_IO, "Input stream created: " + in);
} catch (IOException e) {
e.printStackTrace(logger.getOutStream());
throw new Failure("Caught exception while making streams for " + name + " connection:\n\t" + e);
}
}
} // SocketConnection