* @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;
}
/**
* 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 <anything> -> CR NULL
+ output.write(master.C_CR);
+ output.write(master.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 <something> <IAC> <IAC>).
+ output.write(writeBuffer, 0, writeBufferI);
+ writeBufferI = 0;
+ }
+
+ // Pull the next byte
+ byte ch = b[i + off];
+
+ if (master.binaryMode == true) {
+
+ if (ch == master.TELNET_IAC) {
+ // IAC -> IAC IAC
+ writeBuffer[writeBufferI++] = (byte)master.TELNET_IAC;
+ writeBuffer[writeBufferI++] = (byte)master.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 == master.C_CR) {
+ if (writeCR == true) {
+ // Flush the previous CR to the stream.
+ // CR <anything> -> CR NULL
+ writeBuffer[writeBufferI++] = (byte)master.C_CR;
+ writeBuffer[writeBufferI++] = (byte)master.C_NUL;
+ }
+ writeCR = true;
+ } else if (ch == master.C_LF) {
+ if (writeCR == true) {
+ // CR LF -> CR LF
+ writeBuffer[writeBufferI++] = (byte)master.C_CR;
+ writeBuffer[writeBufferI++] = (byte)master.C_LF;
+ writeCR = false;
+ } else {
+ // Bare LF -> LF
+ writeBuffer[writeBufferI++] = ch;
+ }
+ } else if (ch == master.TELNET_IAC) {
+ if (writeCR == true) {
+ // CR <anything> -> CR NULL
+ writeBuffer[writeBufferI++] = (byte)master.C_CR;
+ writeBuffer[writeBufferI++] = (byte)master.C_NUL;
+ writeCR = false;
+ }
+ // IAC -> IAC IAC
+ writeBuffer[writeBufferI++] = (byte)master.TELNET_IAC;
+ writeBuffer[writeBufferI++] = (byte)master.TELNET_IAC;
+ } else {
+ if (writeCR == true) {
+ // CR <anything> -> CR NULL
+ writeBuffer[writeBufferI++] = (byte)master.C_CR;
+ writeBuffer[writeBufferI++] = (byte)master.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);
+ }
}
}
git://git.nikiroo.be / nikiroo-utils.git / blobdiff