X-Git-Url: http://git.nikiroo.be/?a=blobdiff_plain;f=src%2Fjexer%2Fnet%2FTelnetOutputStream.java;h=988d6ea3fa5c5dcb53a822edf63c62aa3994afdd;hb=daa4106c096cd4d2b92c3cbae6491edccd25fcc4;hp=b31cb03972a071cadda0b461b45ebf24a7a10f2b;hpb=ea91242c4516c212e07f06c1a7d44ab912018e36;p=fanfix.git diff --git a/src/jexer/net/TelnetOutputStream.java b/src/jexer/net/TelnetOutputStream.java index b31cb03..988d6ea 100644 --- a/src/jexer/net/TelnetOutputStream.java +++ b/src/jexer/net/TelnetOutputStream.java @@ -1,4 +1,4 @@ -/** +/* * Jexer - Java Text User Interface * * License: LGPLv3 or later @@ -33,6 +33,8 @@ package jexer.net; import java.io.OutputStream; import java.io.IOException; +import static jexer.net.TelnetSocket.*; + /** * TelnetOutputStream works with TelnetSocket to perform the telnet protocol. */ @@ -54,7 +56,7 @@ public final class TelnetOutputStream extends OutputStream { * @param master the master TelnetSocket * @param output the underlying socket's OutputStream */ - TelnetOutputStream(TelnetSocket master, OutputStream output) { + TelnetOutputStream(final TelnetSocket master, final OutputStream output) { this.master = master; this.output = output; } @@ -64,44 +66,185 @@ public final class TelnetOutputStream extends OutputStream { /** * Closes this output stream and releases any system resources associated * with this stream. + * + * @throws IOException if an I/O error occurs */ @Override public void close() throws IOException { - // TODO + if (output != null) { + output.close(); + output = null; + } } /** * Flushes this output stream and forces any buffered output bytes to be * written out. + * + * @throws IOException if an I/O error occurs */ @Override public void flush() throws IOException { + if ((master.binaryMode == false) && (writeCR == true)) { + // The last byte sent to this.write() was a CR, which was never + // actually sent. So send the CR in ascii mode, then flush. + // CR -> CR NULL + output.write(C_CR); + output.write(C_NUL); + writeCR = false; + } + output.flush(); } /** * Writes b.length bytes from the specified byte array to this output * stream. + * + * @param b the data. + * @throws IOException if an I/O error occurs */ @Override - public void write(byte[] b) throws IOException { - // TODO + public void write(final byte[] b) throws IOException { + writeImpl(b, 0, b.length); } /** * Writes len bytes from the specified byte array starting at offset off * to this output stream. + * + * @param b the data. + * @param off the start offset in the data. + * @param len the number of bytes to write. + * @throws IOException if an I/O error occurs */ @Override - public void write(byte[] b, int off, int len) throws IOException { - // TODO + public void write(final byte[] b, final int off, + final int len) throws IOException { + + writeImpl(b, off, len); } /** * Writes the specified byte to this output stream. + * + * @param b the byte to write. + * @throws IOException if an I/O error occurs */ @Override - public void write(int b) throws IOException { - // TODO + public void write(final int b) throws IOException { + byte [] bytes = new byte[1]; + bytes[0] = (byte)b; + writeImpl(bytes, 0, 1); + } + + /** + * Writes b.length bytes from the specified byte array to this output + * stream. Note package private access. + * + * @param b the data. + * @throws IOException if an I/O error occurs + */ + void rawWrite(final byte[] b) throws IOException { + output.write(b, 0, b.length); + } + + // Telnet protocol -------------------------------------------------------- + + /** + * When true, the last byte the caller passed to write() was a CR. + */ + private boolean writeCR = false; + + /** + * Writes len bytes from the specified byte array starting at offset off + * to this output stream. + * + * @param b the data. + * @param off the start offset in the data. + * @param len the number of bytes to write. + * @throws IOException if an I/O error occurs + */ + private void writeImpl(final byte[] b, final int off, + final int len) throws IOException { + + byte [] writeBuffer = new byte[Math.max(len, 4)]; + int writeBufferI = 0; + + for (int i = 0; i < len; i++) { + if (writeBufferI >= writeBuffer.length - 4) { + // Flush what we have generated so far and reset the buffer, + // because the next byte could generate up to 4 output bytes + // (CR ). + output.write(writeBuffer, 0, writeBufferI); + writeBufferI = 0; + } + + // Pull the next byte + byte ch = b[i + off]; + + if (master.binaryMode == true) { + + if (ch == TELNET_IAC) { + // IAC -> IAC IAC + writeBuffer[writeBufferI++] = (byte)TELNET_IAC; + writeBuffer[writeBufferI++] = (byte)TELNET_IAC; + } else { + // Anything else -> just send + writeBuffer[writeBufferI++] = ch; + } + continue; + } + + // Non-binary mode: more complicated. We use writeCR to handle + // the case that the last byte of b was a CR. + + // Bare carriage return -> CR NUL + if (ch == C_CR) { + if (writeCR == true) { + // Flush the previous CR to the stream. + // CR -> CR NULL + writeBuffer[writeBufferI++] = (byte)C_CR; + writeBuffer[writeBufferI++] = (byte)C_NUL; + } + writeCR = true; + } else if (ch == C_LF) { + if (writeCR == true) { + // CR LF -> CR LF + writeBuffer[writeBufferI++] = (byte)C_CR; + writeBuffer[writeBufferI++] = (byte)C_LF; + writeCR = false; + } else { + // Bare LF -> LF + writeBuffer[writeBufferI++] = ch; + } + } else if (ch == TELNET_IAC) { + if (writeCR == true) { + // CR -> CR NULL + writeBuffer[writeBufferI++] = (byte)C_CR; + writeBuffer[writeBufferI++] = (byte)C_NUL; + writeCR = false; + } + // IAC -> IAC IAC + writeBuffer[writeBufferI++] = (byte)TELNET_IAC; + writeBuffer[writeBufferI++] = (byte)TELNET_IAC; + } else { + if (writeCR == true) { + // CR -> CR NULL + writeBuffer[writeBufferI++] = (byte)C_CR; + writeBuffer[writeBufferI++] = (byte)C_NUL; + writeCR = false; + } else { + // Normal character */ + writeBuffer[writeBufferI++] = ch; + } + } + + } // while (i < userbuf.length) + + if (writeBufferI > 0) { + // Flush what we have generated so far and reset the buffer. + output.write(writeBuffer, 0, writeBufferI); + } } }