/* | 
|
 * Copyright (c) 1994, 2013, 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.  Oracle designates this | 
|
 * particular file as subject to the "Classpath" exception as provided | 
|
 * by Oracle in the LICENSE file that accompanied this code. | 
|
 * | 
|
 * 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 sun.net;  | 
|
import java.io.*;  | 
|
import java.net.Socket;  | 
|
import java.net.InetAddress;  | 
|
import java.net.InetSocketAddress;  | 
|
import java.net.UnknownHostException;  | 
|
import java.net.Proxy;  | 
|
import java.util.Arrays;  | 
|
import java.security.AccessController;  | 
|
import java.security.PrivilegedAction;  | 
|
/** | 
|
 * This is the base class for network clients. | 
|
 * | 
|
 * @author      Jonathan Payne | 
|
*/  | 
|
public class NetworkClient { | 
|
    /* Default value of read timeout, if not specified (infinity) */ | 
|
public static final int DEFAULT_READ_TIMEOUT = -1;  | 
|
    /* Default value of connect timeout, if not specified (infinity) */ | 
|
public static final int DEFAULT_CONNECT_TIMEOUT = -1;  | 
|
protected Proxy proxy = Proxy.NO_PROXY;  | 
|
    /** Socket for communicating with server. */ | 
|
protected Socket serverSocket = null;  | 
|
    /** Stream for printing to the server. */ | 
|
public PrintStream serverOutput;  | 
|
    /** Buffered stream for reading replies from server. */ | 
|
public InputStream serverInput;  | 
|
protected static int defaultSoTimeout;  | 
|
protected static int defaultConnectTimeout;  | 
|
protected int readTimeout = DEFAULT_READ_TIMEOUT;  | 
|
protected int connectTimeout = DEFAULT_CONNECT_TIMEOUT;  | 
|
    /* Name of encoding to use for output */ | 
|
protected static String encoding;  | 
|
    static { | 
|
        final int vals[] = {0, 0}; | 
|
final String encs[] = { null };  | 
|
AccessController.doPrivileged(  | 
|
new PrivilegedAction<Void>() {  | 
|
public Void run() {  | 
|
vals[0] = Integer.getInteger("sun.net.client.defaultReadTimeout", 0).intValue();  | 
|
vals[1] = Integer.getInteger("sun.net.client.defaultConnectTimeout", 0).intValue();  | 
|
encs[0] = System.getProperty("file.encoding", "ISO8859_1");  | 
|
return null;  | 
|
}  | 
|
});  | 
|
if (vals[0] != 0) {  | 
|
defaultSoTimeout = vals[0];  | 
|
}  | 
|
if (vals[1] != 0) {  | 
|
defaultConnectTimeout = vals[1];  | 
|
}  | 
|
encoding = encs[0];  | 
|
        try { | 
|
if (!isASCIISuperset (encoding)) {  | 
|
encoding = "ISO8859_1";  | 
|
}  | 
|
} catch (Exception e) {  | 
|
encoding = "ISO8859_1";  | 
|
}  | 
|
}  | 
|
    /** | 
|
     * Test the named character encoding to verify that it converts ASCII | 
|
     * characters correctly. We have to use an ASCII based encoding, or else | 
|
     * the NetworkClients will not work correctly in EBCDIC based systems. | 
|
     * However, we cannot just use ASCII or ISO8859_1 universally, because in | 
|
     * Asian locales, non-ASCII characters may be embedded in otherwise | 
|
     * ASCII based protocols (eg. HTTP). The specifications (RFC2616, 2398) | 
|
     * are a little ambiguous in this matter. For instance, RFC2398 [part 2.1] | 
|
     * says that the HTTP request URI should be escaped using a defined | 
|
     * mechanism, but there is no way to specify in the escaped string what | 
|
     * the original character set is. It is not correct to assume that | 
|
     * UTF-8 is always used (as in URLs in HTML 4.0).  For this reason, | 
|
     * until the specifications are updated to deal with this issue more | 
|
     * comprehensively, and more importantly, HTTP servers are known to | 
|
     * support these mechanisms, we will maintain the current behavior | 
|
     * where it is possible to send non-ASCII characters in their original | 
|
     * unescaped form. | 
|
*/  | 
|
private static boolean isASCIISuperset (String encoding) throws Exception {  | 
|
String chkS = "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZ"+  | 
|
                        "abcdefghijklmnopqrstuvwxyz-_.!~*'();/?:@&=+$,"; | 
|
        // Expected byte sequence for string above | 
|
        byte[] chkB = { 48,49,50,51,52,53,54,55,56,57,65,66,67,68,69,70,71,72, | 
|
73,74,75,76,77,78,79,80,81,82,83,84,85,86,87,88,89,90,97,98,99,  | 
|
100,101,102,103,104,105,106,107,108,109,110,111,112,113,114,  | 
|
115,116,117,118,119,120,121,122,45,95,46,33,126,42,39,40,41,59,  | 
|
47,63,58,64,38,61,43,36,44};  | 
|
byte[] b = chkS.getBytes (encoding);  | 
|
return Arrays.equals (b, chkB);  | 
|
}  | 
|
    /** Open a connection to the server. */ | 
|
public void openServer(String server, int port)  | 
|
throws IOException, UnknownHostException {  | 
|
if (serverSocket != null)  | 
|
closeServer();  | 
|
serverSocket = doConnect (server, port);  | 
|
        try { | 
|
serverOutput = new PrintStream(new BufferedOutputStream(  | 
|
serverSocket.getOutputStream()),  | 
|
true, encoding);  | 
|
} catch (UnsupportedEncodingException e) {  | 
|
throw new InternalError(encoding +"encoding not found", e);  | 
|
}  | 
|
serverInput = new BufferedInputStream(serverSocket.getInputStream());  | 
|
}  | 
|
    /** | 
|
     * Return a socket connected to the server, with any | 
|
     * appropriate options pre-established | 
|
*/  | 
|
protected Socket doConnect (String server, int port)  | 
|
throws IOException, UnknownHostException {  | 
|
Socket s;  | 
|
if (proxy != null) {  | 
|
if (proxy.type() == Proxy.Type.SOCKS) {  | 
|
s = AccessController.doPrivileged(  | 
|
new PrivilegedAction<Socket>() {  | 
|
public Socket run() {  | 
|
return new Socket(proxy);  | 
|
}});  | 
|
} else if (proxy.type() == Proxy.Type.DIRECT) {  | 
|
s = createSocket();  | 
|
            } else { | 
|
// Still connecting through a proxy  | 
|
                // server & port will be the proxy address and port | 
|
s = new Socket(Proxy.NO_PROXY);  | 
|
}  | 
|
} else  | 
|
s = createSocket();  | 
|
// Instance specific timeouts do have priority, that means  | 
|
// connectTimeout & readTimeout (-1 means not set)  | 
|
// Then global default timeouts  | 
|
        // Then no timeout. | 
|
if (connectTimeout >= 0) {  | 
|
s.connect(new InetSocketAddress(server, port), connectTimeout);  | 
|
        } else { | 
|
if (defaultConnectTimeout > 0) {  | 
|
s.connect(new InetSocketAddress(server, port), defaultConnectTimeout);  | 
|
            } else { | 
|
s.connect(new InetSocketAddress(server, port));  | 
|
}  | 
|
}  | 
|
if (readTimeout >= 0)  | 
|
s.setSoTimeout(readTimeout);  | 
|
else if (defaultSoTimeout > 0) {  | 
|
s.setSoTimeout(defaultSoTimeout);  | 
|
}  | 
|
return s;  | 
|
}  | 
|
    /** | 
|
     * The following method, createSocket, is provided to allow the | 
|
     * https client to override it so that it may use its socket factory | 
|
     * to create the socket. | 
|
*/  | 
|
protected Socket createSocket() throws IOException {  | 
|
return new java.net.Socket();  | 
|
}  | 
|
protected InetAddress getLocalAddress() throws IOException {  | 
|
if (serverSocket == null)  | 
|
throw new IOException("not connected");  | 
|
return AccessController.doPrivileged(  | 
|
new PrivilegedAction<InetAddress>() {  | 
|
public InetAddress run() {  | 
|
return serverSocket.getLocalAddress();  | 
|
}  | 
|
});  | 
|
}  | 
|
    /** Close an open connection to the server. */ | 
|
public void closeServer() throws IOException {  | 
|
if (! serverIsOpen()) {  | 
|
return;  | 
|
}  | 
|
serverSocket.close();  | 
|
serverSocket = null;  | 
|
serverInput = null;  | 
|
serverOutput = null;  | 
|
}  | 
|
    /** Return server connection status */ | 
|
    public boolean serverIsOpen() { | 
|
return serverSocket != null;  | 
|
}  | 
|
    /** Create connection with host <i>host</i> on port <i>port</i> */ | 
|
public NetworkClient(String host, int port) throws IOException {  | 
|
openServer(host, port);  | 
|
}  | 
|
    public NetworkClient() {} | 
|
    public void setConnectTimeout(int timeout) { | 
|
connectTimeout = timeout;  | 
|
}  | 
|
    public int getConnectTimeout() { | 
|
return connectTimeout;  | 
|
}  | 
|
    /** | 
|
     * Sets the read timeout. | 
|
     * | 
|
     * Note: Public URLConnection (and protocol specific implementations) | 
|
     * protect against negative timeout values being set. This implementation, | 
|
     * and protocol specific implementations, use -1 to represent the default | 
|
     * read timeout. | 
|
     * | 
|
     * This method may be invoked with the default timeout value when the | 
|
     * protocol handler is trying to reset the timeout after doing a | 
|
     * potentially blocking internal operation, e.g. cleaning up unread | 
|
     * response data, buffering error stream response data, etc | 
|
*/  | 
|
    public void setReadTimeout(int timeout) { | 
|
if (timeout == DEFAULT_READ_TIMEOUT)  | 
|
timeout = defaultSoTimeout;  | 
|
if (serverSocket != null && timeout >= 0) {  | 
|
            try { | 
|
serverSocket.setSoTimeout(timeout);  | 
|
} catch(IOException e) {  | 
|
// We tried...  | 
|
}  | 
|
}  | 
|
readTimeout = timeout;  | 
|
}  | 
|
    public int getReadTimeout() { | 
|
return readTimeout;  | 
|
}  | 
|
}  |