| /* |
| * Copyright (c) 2005, 2011, 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 java.io; |
| |
| import java.util.*; |
| import java.nio.charset.Charset; |
| import sun.nio.cs.StreamDecoder; |
| import sun.nio.cs.StreamEncoder; |
| |
| /** |
| * Methods to access the character-based console device, if any, associated |
| * with the current Java virtual machine. |
| * |
| * <p> Whether a virtual machine has a console is dependent upon the |
| * underlying platform and also upon the manner in which the virtual |
| * machine is invoked. If the virtual machine is started from an |
| * interactive command line without redirecting the standard input and |
| * output streams then its console will exist and will typically be |
| * connected to the keyboard and display from which the virtual machine |
| * was launched. If the virtual machine is started automatically, for |
| * example by a background job scheduler, then it will typically not |
| * have a console. |
| * <p> |
| * If this virtual machine has a console then it is represented by a |
| * unique instance of this class which can be obtained by invoking the |
| * {@link java.lang.System#console()} method. If no console device is |
| * available then an invocation of that method will return <tt>null</tt>. |
| * <p> |
| * Read and write operations are synchronized to guarantee the atomic |
| * completion of critical operations; therefore invoking methods |
| * {@link #readLine()}, {@link #readPassword()}, {@link #format format()}, |
| * {@link #printf printf()} as well as the read, format and write operations |
| * on the objects returned by {@link #reader()} and {@link #writer()} may |
| * block in multithreaded scenarios. |
| * <p> |
| * Invoking <tt>close()</tt> on the objects returned by the {@link #reader()} |
| * and the {@link #writer()} will not close the underlying stream of those |
| * objects. |
| * <p> |
| * The console-read methods return <tt>null</tt> when the end of the |
| * console input stream is reached, for example by typing control-D on |
| * Unix or control-Z on Windows. Subsequent read operations will succeed |
| * if additional characters are later entered on the console's input |
| * device. |
| * <p> |
| * Unless otherwise specified, passing a <tt>null</tt> argument to any method |
| * in this class will cause a {@link NullPointerException} to be thrown. |
| * <p> |
| * <b>Security note:</b> |
| * If an application needs to read a password or other secure data, it should |
| * use {@link #readPassword()} or {@link #readPassword(String, Object...)} and |
| * manually zero the returned character array after processing to minimize the |
| * lifetime of sensitive data in memory. |
| * |
| * <blockquote><pre>{@code |
| * Console cons; |
| * char[] passwd; |
| * if ((cons = System.console()) != null && |
| * (passwd = cons.readPassword("[%s]", "Password:")) != null) { |
| * ... |
| * java.util.Arrays.fill(passwd, ' '); |
| * } |
| * }</pre></blockquote> |
| * |
| * @author Xueming Shen |
| * @since 1.6 |
| */ |
| |
| public final class Console implements Flushable |
| { |
| /** |
| * Retrieves the unique {@link java.io.PrintWriter PrintWriter} object |
| * associated with this console. |
| * |
| * @return The printwriter associated with this console |
| */ |
| public PrintWriter writer() { |
| return pw; |
| } |
| |
| /** |
| * Retrieves the unique {@link java.io.Reader Reader} object associated |
| * with this console. |
| * <p> |
| * This method is intended to be used by sophisticated applications, for |
| * example, a {@link java.util.Scanner} object which utilizes the rich |
| * parsing/scanning functionality provided by the <tt>Scanner</tt>: |
| * <blockquote><pre> |
| * Console con = System.console(); |
| * if (con != null) { |
| * Scanner sc = new Scanner(con.reader()); |
| * ... |
| * } |
| * </pre></blockquote> |
| * <p> |
| * For simple applications requiring only line-oriented reading, use |
| * <tt>{@link #readLine}</tt>. |
| * <p> |
| * The bulk read operations {@link java.io.Reader#read(char[]) read(char[]) }, |
| * {@link java.io.Reader#read(char[], int, int) read(char[], int, int) } and |
| * {@link java.io.Reader#read(java.nio.CharBuffer) read(java.nio.CharBuffer)} |
| * on the returned object will not read in characters beyond the line |
| * bound for each invocation, even if the destination buffer has space for |
| * more characters. The {@code Reader}'s {@code read} methods may block if a |
| * line bound has not been entered or reached on the console's input device. |
| * A line bound is considered to be any one of a line feed (<tt>'\n'</tt>), |
| * a carriage return (<tt>'\r'</tt>), a carriage return followed immediately |
| * by a linefeed, or an end of stream. |
| * |
| * @return The reader associated with this console |
| */ |
| public Reader reader() { |
| return reader; |
| } |
| |
| /** |
| * Writes a formatted string to this console's output stream using |
| * the specified format string and arguments. |
| * |
| * @param fmt |
| * A format string as described in <a |
| * href="../util/Formatter.html#syntax">Format string syntax</a> |
| * |
| * @param args |
| * Arguments referenced by the format specifiers in the format |
| * string. If there are more arguments than format specifiers, the |
| * extra arguments are ignored. The number of arguments is |
| * variable and may be zero. The maximum number of arguments is |
| * limited by the maximum dimension of a Java array as defined by |
| * <cite>The Java™ Virtual Machine Specification</cite>. |
| * The behaviour on a |
| * <tt>null</tt> argument depends on the <a |
| * href="../util/Formatter.html#syntax">conversion</a>. |
| * |
| * @throws IllegalFormatException |
| * If a format string contains an illegal syntax, a format |
| * specifier that is incompatible with the given arguments, |
| * insufficient arguments given the format string, or other |
| * illegal conditions. For specification of all possible |
| * formatting errors, see the <a |
| * href="../util/Formatter.html#detail">Details</a> section |
| * of the formatter class specification. |
| * |
| * @return This console |
| */ |
| public Console format(String fmt, Object ...args) { |
| formatter.format(fmt, args).flush(); |
| return this; |
| } |
| |
| /** |
| * A convenience method to write a formatted string to this console's |
| * output stream using the specified format string and arguments. |
| * |
| * <p> An invocation of this method of the form <tt>con.printf(format, |
| * args)</tt> behaves in exactly the same way as the invocation of |
| * <pre>con.format(format, args)</pre>. |
| * |
| * @param format |
| * A format string as described in <a |
| * href="../util/Formatter.html#syntax">Format string syntax</a>. |
| * |
| * @param args |
| * Arguments referenced by the format specifiers in the format |
| * string. If there are more arguments than format specifiers, the |
| * extra arguments are ignored. The number of arguments is |
| * variable and may be zero. The maximum number of arguments is |
| * limited by the maximum dimension of a Java array as defined by |
| * <cite>The Java™ Virtual Machine Specification</cite>. |
| * The behaviour on a |
| * <tt>null</tt> argument depends on the <a |
| * href="../util/Formatter.html#syntax">conversion</a>. |
| * |
| * @throws IllegalFormatException |
| * If a format string contains an illegal syntax, a format |
| * specifier that is incompatible with the given arguments, |
| * insufficient arguments given the format string, or other |
| * illegal conditions. For specification of all possible |
| * formatting errors, see the <a |
| * href="../util/Formatter.html#detail">Details</a> section of the |
| * formatter class specification. |
| * |
| * @return This console |
| */ |
| public Console printf(String format, Object ... args) { |
| return format(format, args); |
| } |
| |
| /** |
| * Provides a formatted prompt, then reads a single line of text from the |
| * console. |
| * |
| * @param fmt |
| * A format string as described in <a |
| * href="../util/Formatter.html#syntax">Format string syntax</a>. |
| * |
| * @param args |
| * Arguments referenced by the format specifiers in the format |
| * string. If there are more arguments than format specifiers, the |
| * extra arguments are ignored. The maximum number of arguments is |
| * limited by the maximum dimension of a Java array as defined by |
| * <cite>The Java™ Virtual Machine Specification</cite>. |
| * |
| * @throws IllegalFormatException |
| * If a format string contains an illegal syntax, a format |
| * specifier that is incompatible with the given arguments, |
| * insufficient arguments given the format string, or other |
| * illegal conditions. For specification of all possible |
| * formatting errors, see the <a |
| * href="../util/Formatter.html#detail">Details</a> section |
| * of the formatter class specification. |
| * |
| * @throws IOError |
| * If an I/O error occurs. |
| * |
| * @return A string containing the line read from the console, not |
| * including any line-termination characters, or <tt>null</tt> |
| * if an end of stream has been reached. |
| */ |
| public String readLine(String fmt, Object ... args) { |
| String line = null; |
| synchronized (writeLock) { |
| synchronized(readLock) { |
| if (fmt.length() != 0) |
| pw.format(fmt, args); |
| try { |
| char[] ca = readline(false); |
| if (ca != null) |
| line = new String(ca); |
| } catch (IOException x) { |
| throw new IOError(x); |
| } |
| } |
| } |
| return line; |
| } |
| |
| /** |
| * Reads a single line of text from the console. |
| * |
| * @throws IOError |
| * If an I/O error occurs. |
| * |
| * @return A string containing the line read from the console, not |
| * including any line-termination characters, or <tt>null</tt> |
| * if an end of stream has been reached. |
| */ |
| public String readLine() { |
| return readLine(""); |
| } |
| |
| /** |
| * Provides a formatted prompt, then reads a password or passphrase from |
| * the console with echoing disabled. |
| * |
| * @param fmt |
| * A format string as described in <a |
| * href="../util/Formatter.html#syntax">Format string syntax</a> |
| * for the prompt text. |
| * |
| * @param args |
| * Arguments referenced by the format specifiers in the format |
| * string. If there are more arguments than format specifiers, the |
| * extra arguments are ignored. The maximum number of arguments is |
| * limited by the maximum dimension of a Java array as defined by |
| * <cite>The Java™ Virtual Machine Specification</cite>. |
| * |
| * @throws IllegalFormatException |
| * If a format string contains an illegal syntax, a format |
| * specifier that is incompatible with the given arguments, |
| * insufficient arguments given the format string, or other |
| * illegal conditions. For specification of all possible |
| * formatting errors, see the <a |
| * href="../util/Formatter.html#detail">Details</a> |
| * section of the formatter class specification. |
| * |
| * @throws IOError |
| * If an I/O error occurs. |
| * |
| * @return A character array containing the password or passphrase read |
| * from the console, not including any line-termination characters, |
| * or <tt>null</tt> if an end of stream has been reached. |
| */ |
| public char[] readPassword(String fmt, Object ... args) { |
| char[] passwd = null; |
| synchronized (writeLock) { |
| synchronized(readLock) { |
| try { |
| echoOff = echo(false); |
| } catch (IOException x) { |
| throw new IOError(x); |
| } |
| IOError ioe = null; |
| try { |
| if (fmt.length() != 0) |
| pw.format(fmt, args); |
| passwd = readline(true); |
| } catch (IOException x) { |
| ioe = new IOError(x); |
| } finally { |
| try { |
| echoOff = echo(true); |
| } catch (IOException x) { |
| if (ioe == null) |
| ioe = new IOError(x); |
| else |
| ioe.addSuppressed(x); |
| } |
| if (ioe != null) |
| throw ioe; |
| } |
| pw.println(); |
| } |
| } |
| return passwd; |
| } |
| |
| /** |
| * Reads a password or passphrase from the console with echoing disabled |
| * |
| * @throws IOError |
| * If an I/O error occurs. |
| * |
| * @return A character array containing the password or passphrase read |
| * from the console, not including any line-termination characters, |
| * or <tt>null</tt> if an end of stream has been reached. |
| */ |
| public char[] readPassword() { |
| return readPassword(""); |
| } |
| |
| /** |
| * Flushes the console and forces any buffered output to be written |
| * immediately . |
| */ |
| public void flush() { |
| pw.flush(); |
| } |
| |
| private Object readLock; |
| private Object writeLock; |
| private Reader reader; |
| private Writer out; |
| private PrintWriter pw; |
| private Formatter formatter; |
| private Charset cs; |
| private char[] rcb; |
| private static native String encoding(); |
| private static native boolean echo(boolean on) throws IOException; |
| private static boolean echoOff; |
| |
| private char[] readline(boolean zeroOut) throws IOException { |
| int len = reader.read(rcb, 0, rcb.length); |
| if (len < 0) |
| return null; //EOL |
| if (rcb[len-1] == '\r') |
| len--; //remove CR at end; |
| else if (rcb[len-1] == '\n') { |
| len--; //remove LF at end; |
| if (len > 0 && rcb[len-1] == '\r') |
| len--; //remove the CR, if there is one |
| } |
| char[] b = new char[len]; |
| if (len > 0) { |
| System.arraycopy(rcb, 0, b, 0, len); |
| if (zeroOut) { |
| Arrays.fill(rcb, 0, len, ' '); |
| } |
| } |
| return b; |
| } |
| |
| private char[] grow() { |
| assert Thread.holdsLock(readLock); |
| char[] t = new char[rcb.length * 2]; |
| System.arraycopy(rcb, 0, t, 0, rcb.length); |
| rcb = t; |
| return rcb; |
| } |
| |
| class LineReader extends Reader { |
| private Reader in; |
| private char[] cb; |
| private int nChars, nextChar; |
| boolean leftoverLF; |
| LineReader(Reader in) { |
| this.in = in; |
| cb = new char[1024]; |
| nextChar = nChars = 0; |
| leftoverLF = false; |
| } |
| public void close () {} |
| public boolean ready() throws IOException { |
| //in.ready synchronizes on readLock already |
| return in.ready(); |
| } |
| |
| public int read(char cbuf[], int offset, int length) |
| throws IOException |
| { |
| int off = offset; |
| int end = offset + length; |
| if (offset < 0 || offset > cbuf.length || length < 0 || |
| end < 0 || end > cbuf.length) { |
| throw new IndexOutOfBoundsException(); |
| } |
| synchronized(readLock) { |
| boolean eof = false; |
| char c = 0; |
| for (;;) { |
| if (nextChar >= nChars) { //fill |
| int n = 0; |
| do { |
| n = in.read(cb, 0, cb.length); |
| } while (n == 0); |
| if (n > 0) { |
| nChars = n; |
| nextChar = 0; |
| if (n < cb.length && |
| cb[n-1] != '\n' && cb[n-1] != '\r') { |
| /* |
| * we're in canonical mode so each "fill" should |
| * come back with an eol. if there no lf or nl at |
| * the end of returned bytes we reached an eof. |
| */ |
| eof = true; |
| } |
| } else { /*EOF*/ |
| if (off - offset == 0) |
| return -1; |
| return off - offset; |
| } |
| } |
| if (leftoverLF && cbuf == rcb && cb[nextChar] == '\n') { |
| /* |
| * if invoked by our readline, skip the leftover, otherwise |
| * return the LF. |
| */ |
| nextChar++; |
| } |
| leftoverLF = false; |
| while (nextChar < nChars) { |
| c = cbuf[off++] = cb[nextChar]; |
| cb[nextChar++] = 0; |
| if (c == '\n') { |
| return off - offset; |
| } else if (c == '\r') { |
| if (off == end) { |
| /* no space left even the next is LF, so return |
| * whatever we have if the invoker is not our |
| * readLine() |
| */ |
| if (cbuf == rcb) { |
| cbuf = grow(); |
| end = cbuf.length; |
| } else { |
| leftoverLF = true; |
| return off - offset; |
| } |
| } |
| if (nextChar == nChars && in.ready()) { |
| /* |
| * we have a CR and we reached the end of |
| * the read in buffer, fill to make sure we |
| * don't miss a LF, if there is one, it's possible |
| * that it got cut off during last round reading |
| * simply because the read in buffer was full. |
| */ |
| nChars = in.read(cb, 0, cb.length); |
| nextChar = 0; |
| } |
| if (nextChar < nChars && cb[nextChar] == '\n') { |
| cbuf[off++] = '\n'; |
| nextChar++; |
| } |
| return off - offset; |
| } else if (off == end) { |
| if (cbuf == rcb) { |
| cbuf = grow(); |
| end = cbuf.length; |
| } else { |
| return off - offset; |
| } |
| } |
| } |
| if (eof) |
| return off - offset; |
| } |
| } |
| } |
| } |
| |
| // Set up JavaIOAccess in SharedSecrets |
| static { |
| try { |
| // Add a shutdown hook to restore console's echo state should |
| // it be necessary. |
| sun.misc.SharedSecrets.getJavaLangAccess() |
| .registerShutdownHook(0 /* shutdown hook invocation order */, |
| false /* only register if shutdown is not in progress */, |
| new Runnable() { |
| public void run() { |
| try { |
| if (echoOff) { |
| echo(true); |
| } |
| } catch (IOException x) { } |
| } |
| }); |
| } catch (IllegalStateException e) { |
| // shutdown is already in progress and console is first used |
| // by a shutdown hook |
| } |
| |
| sun.misc.SharedSecrets.setJavaIOAccess(new sun.misc.JavaIOAccess() { |
| public Console console() { |
| if (istty()) { |
| if (cons == null) |
| cons = new Console(); |
| return cons; |
| } |
| return null; |
| } |
| |
| public Charset charset() { |
| // This method is called in sun.security.util.Password, |
| // cons already exists when this method is called |
| return cons.cs; |
| } |
| }); |
| } |
| private static Console cons; |
| private native static boolean istty(); |
| private Console() { |
| readLock = new Object(); |
| writeLock = new Object(); |
| String csname = encoding(); |
| if (csname != null) { |
| try { |
| cs = Charset.forName(csname); |
| } catch (Exception x) {} |
| } |
| if (cs == null) |
| cs = Charset.defaultCharset(); |
| out = StreamEncoder.forOutputStreamWriter( |
| new FileOutputStream(FileDescriptor.out), |
| writeLock, |
| cs); |
| pw = new PrintWriter(out, true) { public void close() {} }; |
| formatter = new Formatter(out); |
| reader = new LineReader(StreamDecoder.forInputStreamReader( |
| new FileInputStream(FileDescriptor.in), |
| readLock, |
| cs)); |
| rcb = new char[1024]; |
| } |
| } |