|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
*/ |
|
|
|
package sun.nio.ch; |
|
|
|
import java.io.IOException; |
|
import java.io.InputStream; |
|
import java.io.OutputStream; |
|
import java.net.InetAddress; |
|
import java.net.InetSocketAddress; |
|
import java.net.Socket; |
|
import java.net.SocketAddress; |
|
import java.net.SocketException; |
|
import java.net.SocketImpl; |
|
import java.net.SocketOption; |
|
import java.net.SocketTimeoutException; |
|
import java.net.StandardSocketOptions; |
|
import java.nio.ByteBuffer; |
|
import java.nio.channels.Channels; |
|
import java.nio.channels.ClosedChannelException; |
|
import java.nio.channels.IllegalBlockingModeException; |
|
import java.nio.channels.SocketChannel; |
|
import java.security.AccessController; |
|
import java.security.PrivilegedExceptionAction; |
|
import java.util.*; |
|
|
|
|
|
// Make a socket channel look like a socket. |
|
// |
|
// The only aspects of java.net.Socket-hood that we don't attempt to emulate |
|
// here are the interrupted-I/O exceptions (which our Solaris implementations |
|
// attempt to support) and the sending of urgent data. Otherwise an adapted |
|
// socket should look enough like a real java.net.Socket to fool most of the |
|
// developers most of the time, right down to the exception message strings. |
|
// |
|
// The methods in this class are defined in exactly the same order as in |
|
// java.net.Socket so as to simplify tracking future changes to that class. |
|
// |
|
|
|
class SocketAdaptor |
|
extends Socket |
|
{ |
|
|
|
|
|
private final SocketChannelImpl sc; |
|
|
|
|
|
private volatile int timeout = 0; |
|
|
|
private SocketAdaptor(SocketChannelImpl sc) throws SocketException { |
|
super((SocketImpl) null); |
|
this.sc = sc; |
|
} |
|
|
|
public static Socket create(SocketChannelImpl sc) { |
|
try { |
|
return new SocketAdaptor(sc); |
|
} catch (SocketException e) { |
|
throw new InternalError("Should not reach here"); |
|
} |
|
} |
|
|
|
public SocketChannel getChannel() { |
|
return sc; |
|
} |
|
|
|
// Override this method just to protect against changes in the superclass |
|
|
|
public void connect(SocketAddress remote) throws IOException { |
|
connect(remote, 0); |
|
} |
|
|
|
public void connect(SocketAddress remote, int timeout) throws IOException { |
|
if (remote == null) |
|
throw new IllegalArgumentException("connect: The address can't be null"); |
|
if (timeout < 0) |
|
throw new IllegalArgumentException("connect: timeout can't be negative"); |
|
|
|
synchronized (sc.blockingLock()) { |
|
if (!sc.isBlocking()) |
|
throw new IllegalBlockingModeException(); |
|
|
|
try { |
|
if (timeout == 0) { |
|
sc.connect(remote); |
|
return; |
|
} |
|
|
|
sc.configureBlocking(false); |
|
try { |
|
if (sc.connect(remote)) |
|
return; |
|
long to = timeout; |
|
for (;;) { |
|
if (!sc.isOpen()) |
|
throw new ClosedChannelException(); |
|
long st = System.currentTimeMillis(); |
|
|
|
int result = sc.poll(Net.POLLCONN, to); |
|
if (result > 0 && sc.finishConnect()) |
|
break; |
|
to -= System.currentTimeMillis() - st; |
|
if (to <= 0) { |
|
try { |
|
sc.close(); |
|
} catch (IOException x) { } |
|
throw new SocketTimeoutException(); |
|
} |
|
} |
|
} finally { |
|
try { |
|
sc.configureBlocking(true); |
|
} catch (ClosedChannelException e) { } |
|
} |
|
|
|
} catch (Exception x) { |
|
Net.translateException(x, true); |
|
} |
|
} |
|
|
|
} |
|
|
|
public void bind(SocketAddress local) throws IOException { |
|
try { |
|
sc.bind(local); |
|
} catch (Exception x) { |
|
Net.translateException(x); |
|
} |
|
} |
|
|
|
public InetAddress getInetAddress() { |
|
SocketAddress remote = sc.remoteAddress(); |
|
if (remote == null) { |
|
return null; |
|
} else { |
|
return ((InetSocketAddress)remote).getAddress(); |
|
} |
|
} |
|
|
|
public InetAddress getLocalAddress() { |
|
if (sc.isOpen()) { |
|
InetSocketAddress local = sc.localAddress(); |
|
if (local != null) { |
|
return Net.getRevealedLocalAddress(local).getAddress(); |
|
} |
|
} |
|
return new InetSocketAddress(0).getAddress(); |
|
} |
|
|
|
public int getPort() { |
|
SocketAddress remote = sc.remoteAddress(); |
|
if (remote == null) { |
|
return 0; |
|
} else { |
|
return ((InetSocketAddress)remote).getPort(); |
|
} |
|
} |
|
|
|
public int getLocalPort() { |
|
SocketAddress local = sc.localAddress(); |
|
if (local == null) { |
|
return -1; |
|
} else { |
|
return ((InetSocketAddress)local).getPort(); |
|
} |
|
} |
|
|
|
private class SocketInputStream |
|
extends ChannelInputStream |
|
{ |
|
private SocketInputStream() { |
|
super(sc); |
|
} |
|
|
|
protected int read(ByteBuffer bb) |
|
throws IOException |
|
{ |
|
synchronized (sc.blockingLock()) { |
|
if (!sc.isOpen()) { |
|
throw new ClosedChannelException(); |
|
} |
|
if (!sc.isBlocking()) |
|
throw new IllegalBlockingModeException(); |
|
|
|
if (timeout == 0) |
|
return sc.read(bb); |
|
|
|
sc.configureBlocking(false); |
|
try { |
|
int n; |
|
if ((n = sc.read(bb)) != 0) |
|
return n; |
|
long to = timeout; |
|
for (;;) { |
|
if (!sc.isOpen()) |
|
throw new ClosedChannelException(); |
|
long st = System.currentTimeMillis(); |
|
int result = sc.poll(Net.POLLIN, to); |
|
if (result > 0) { |
|
if ((n = sc.read(bb)) != 0) |
|
return n; |
|
} |
|
to -= System.currentTimeMillis() - st; |
|
if (to <= 0) |
|
throw new SocketTimeoutException(); |
|
} |
|
} finally { |
|
try { |
|
sc.configureBlocking(true); |
|
} catch (ClosedChannelException e) { } |
|
} |
|
} |
|
} |
|
} |
|
|
|
private InputStream socketInputStream = null; |
|
|
|
public InputStream getInputStream() throws IOException { |
|
if (!sc.isOpen()) |
|
throw new SocketException("Socket is closed"); |
|
if (!sc.isConnected()) |
|
throw new SocketException("Socket is not connected"); |
|
if (!sc.isInputOpen()) |
|
throw new SocketException("Socket input is shutdown"); |
|
if (socketInputStream == null) { |
|
try { |
|
socketInputStream = AccessController.doPrivileged( |
|
new PrivilegedExceptionAction<InputStream>() { |
|
public InputStream run() throws IOException { |
|
return new SocketInputStream(); |
|
} |
|
}); |
|
} catch (java.security.PrivilegedActionException e) { |
|
throw (IOException)e.getException(); |
|
} |
|
} |
|
return socketInputStream; |
|
} |
|
|
|
public OutputStream getOutputStream() throws IOException { |
|
if (!sc.isOpen()) |
|
throw new SocketException("Socket is closed"); |
|
if (!sc.isConnected()) |
|
throw new SocketException("Socket is not connected"); |
|
if (!sc.isOutputOpen()) |
|
throw new SocketException("Socket output is shutdown"); |
|
OutputStream os = null; |
|
try { |
|
os = AccessController.doPrivileged( |
|
new PrivilegedExceptionAction<OutputStream>() { |
|
public OutputStream run() throws IOException { |
|
return Channels.newOutputStream(sc); |
|
} |
|
}); |
|
} catch (java.security.PrivilegedActionException e) { |
|
throw (IOException)e.getException(); |
|
} |
|
return os; |
|
} |
|
|
|
private void setBooleanOption(SocketOption<Boolean> name, boolean value) |
|
throws SocketException |
|
{ |
|
try { |
|
sc.setOption(name, value); |
|
} catch (IOException x) { |
|
Net.translateToSocketException(x); |
|
} |
|
} |
|
|
|
private void setIntOption(SocketOption<Integer> name, int value) |
|
throws SocketException |
|
{ |
|
try { |
|
sc.setOption(name, value); |
|
} catch (IOException x) { |
|
Net.translateToSocketException(x); |
|
} |
|
} |
|
|
|
private boolean getBooleanOption(SocketOption<Boolean> name) throws SocketException { |
|
try { |
|
return sc.getOption(name).booleanValue(); |
|
} catch (IOException x) { |
|
Net.translateToSocketException(x); |
|
return false; |
|
} |
|
} |
|
|
|
private int getIntOption(SocketOption<Integer> name) throws SocketException { |
|
try { |
|
return sc.getOption(name).intValue(); |
|
} catch (IOException x) { |
|
Net.translateToSocketException(x); |
|
return -1; |
|
} |
|
} |
|
|
|
public void setTcpNoDelay(boolean on) throws SocketException { |
|
setBooleanOption(StandardSocketOptions.TCP_NODELAY, on); |
|
} |
|
|
|
public boolean getTcpNoDelay() throws SocketException { |
|
return getBooleanOption(StandardSocketOptions.TCP_NODELAY); |
|
} |
|
|
|
public void setSoLinger(boolean on, int linger) throws SocketException { |
|
if (!on) |
|
linger = -1; |
|
setIntOption(StandardSocketOptions.SO_LINGER, linger); |
|
} |
|
|
|
public int getSoLinger() throws SocketException { |
|
return getIntOption(StandardSocketOptions.SO_LINGER); |
|
} |
|
|
|
public void sendUrgentData(int data) throws IOException { |
|
int n = sc.sendOutOfBandData((byte) data); |
|
if (n == 0) |
|
throw new IOException("Socket buffer full"); |
|
} |
|
|
|
public void setOOBInline(boolean on) throws SocketException { |
|
setBooleanOption(ExtendedSocketOption.SO_OOBINLINE, on); |
|
} |
|
|
|
public boolean getOOBInline() throws SocketException { |
|
return getBooleanOption(ExtendedSocketOption.SO_OOBINLINE); |
|
} |
|
|
|
public void setSoTimeout(int timeout) throws SocketException { |
|
if (timeout < 0) |
|
throw new IllegalArgumentException("timeout can't be negative"); |
|
this.timeout = timeout; |
|
} |
|
|
|
public int getSoTimeout() throws SocketException { |
|
return timeout; |
|
} |
|
|
|
public void setSendBufferSize(int size) throws SocketException { |
|
|
|
if (size <= 0) |
|
throw new IllegalArgumentException("Invalid send size"); |
|
setIntOption(StandardSocketOptions.SO_SNDBUF, size); |
|
} |
|
|
|
public int getSendBufferSize() throws SocketException { |
|
return getIntOption(StandardSocketOptions.SO_SNDBUF); |
|
} |
|
|
|
public void setReceiveBufferSize(int size) throws SocketException { |
|
|
|
if (size <= 0) |
|
throw new IllegalArgumentException("Invalid receive size"); |
|
setIntOption(StandardSocketOptions.SO_RCVBUF, size); |
|
} |
|
|
|
public int getReceiveBufferSize() throws SocketException { |
|
return getIntOption(StandardSocketOptions.SO_RCVBUF); |
|
} |
|
|
|
public void setKeepAlive(boolean on) throws SocketException { |
|
setBooleanOption(StandardSocketOptions.SO_KEEPALIVE, on); |
|
} |
|
|
|
public boolean getKeepAlive() throws SocketException { |
|
return getBooleanOption(StandardSocketOptions.SO_KEEPALIVE); |
|
} |
|
|
|
public void setTrafficClass(int tc) throws SocketException { |
|
setIntOption(StandardSocketOptions.IP_TOS, tc); |
|
} |
|
|
|
public int getTrafficClass() throws SocketException { |
|
return getIntOption(StandardSocketOptions.IP_TOS); |
|
} |
|
|
|
public void setReuseAddress(boolean on) throws SocketException { |
|
setBooleanOption(StandardSocketOptions.SO_REUSEADDR, on); |
|
} |
|
|
|
public boolean getReuseAddress() throws SocketException { |
|
return getBooleanOption(StandardSocketOptions.SO_REUSEADDR); |
|
} |
|
|
|
public void close() throws IOException { |
|
sc.close(); |
|
} |
|
|
|
public void shutdownInput() throws IOException { |
|
try { |
|
sc.shutdownInput(); |
|
} catch (Exception x) { |
|
Net.translateException(x); |
|
} |
|
} |
|
|
|
public void shutdownOutput() throws IOException { |
|
try { |
|
sc.shutdownOutput(); |
|
} catch (Exception x) { |
|
Net.translateException(x); |
|
} |
|
} |
|
|
|
public String toString() { |
|
if (sc.isConnected()) |
|
return "Socket[addr=" + getInetAddress() + |
|
",port=" + getPort() + |
|
",localport=" + getLocalPort() + "]"; |
|
return "Socket[unconnected]"; |
|
} |
|
|
|
public boolean isConnected() { |
|
return sc.isConnected(); |
|
} |
|
|
|
public boolean isBound() { |
|
return sc.localAddress() != null; |
|
} |
|
|
|
public boolean isClosed() { |
|
return !sc.isOpen(); |
|
} |
|
|
|
public boolean isInputShutdown() { |
|
return !sc.isInputOpen(); |
|
} |
|
|
|
public boolean isOutputShutdown() { |
|
return !sc.isOutputOpen(); |
|
} |
|
|
|
} |