/*

 * 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.security.ssl;

/**

 * {@code OutputRecord} implementation for {@code SSLSocket}.

 */

final class SSLSocketOutputRecord extends OutputRecord implements SSLRecord {

    private OutputStream deliverStream = null;

        this.packetSize = SSLRecord.maxRecordSize;

    }

    @Override

    void encodeAlert(byte level, byte description) throws IOException {

        // use the buf of ByteArrayOutputStream

        int position = headerSize + writeCipher.getExplicitNonceSize();

        count = position;

        write(level);

        write(description);

                    ", length = " + (count - headerSize));

        }

        // Encrypt the fragment and wrap up a record.

        // deliver this message

        deliverStream.write(buf, 0, count);    // may throw IOException

        deliverStream.flush();                 // may throw IOException

        }

        // reset the internal buffer

        count = 0;

    }

    @Override

    void encodeHandshake(byte[] source,

            int offset, int length) throws IOException {

        if (firstMessage) {

            firstMessage = false;

            if ((helloVersion == ProtocolVersion.SSL20Hello) &&

                                            //  5: recode header size

                (source[offset + 4 + 2 + 32] == 0)) {

                                            // V3 session ID is empty

                                            //  4: handshake header size

                                            //  2: client_version in ClientHello

                                            // 32: random in ClientHello

                ByteBuffer v2ClientHello = encodeV2ClientHello(

                        source, (offset + 4), (length - 4));

                byte[] record = v2ClientHello.array();  // array offset is zero

                int limit = v2ClientHello.limit();

                }

                // deliver this message

                //

                // Version 2 ClientHello message should be plaintext.

                //

                // No max fragment length negotiation.

                deliverStream.write(record, 0, limit);

                deliverStream.flush();

                }

                return;

            }

        }

        byte handshakeType = source[0];

        }

        int fragLimit = getFragLimit();

        int position = headerSize + writeCipher.getExplicitNonceSize();

        if (count == 0) {

            count = position;

        }

        if ((count - position) < (fragLimit - length)) {

            write(source, offset, length);

            return;

        }

        for (int limit = (offset + length); offset < limit;) {

            int remains = (limit - offset) + (count - position);

            int fragLen = Math.min(fragLimit, remains);

            // use the buf of ByteArrayOutputStream

            write(source, offset, fragLen);

            if (remains < fragLimit) {

                return;

            }

                        ", length = " + (count - headerSize));

            }

            // Encrypt the fragment and wrap up a record.

            // deliver this message

            deliverStream.write(buf, 0, count);    // may throw IOException

            deliverStream.flush();                 // may throw IOException

            }

            // reset the offset

            offset += fragLen;

            // reset the internal buffer

            count = position;

        }

    }

    @Override

    void encodeChangeCipherSpec() throws IOException {

        // use the buf of ByteArrayOutputStream

        int position = headerSize + writeCipher.getExplicitNonceSize();

        count = position;

        write((byte)1);         // byte 1: change_cipher_spec(

        // Encrypt the fragment and wrap up a record.

        // deliver this message

        deliverStream.write(buf, 0, count);        // may throw IOException

        // deliverStream.flush();                  // flush in Finished

        }

        // reset the internal buffer

        count = 0;

    }

    @Override

    public void flush() throws IOException {

        int position = headerSize + writeCipher.getExplicitNonceSize();

        if (count <= position) {

            return;

        }

                    ", length = " + (count - headerSize));

        }

        // Encrypt the fragment and wrap up a record.

        // deliver this message

        deliverStream.write(buf, 0, count);    // may throw IOException

        deliverStream.flush();                 // may throw IOException

        }

        // reset the internal buffer

        count = 0;      // DON'T use position

    }

    @Override

    void deliver(byte[] source, int offset, int length) throws IOException {

                    "(2^64-1 packets). Closing connection.");

            }

            throw new SSLHandshakeException("sequence number overflow");

        }

        boolean isFirstRecordOfThePayload = true;

        for (int limit = (offset + length); offset < limit;) {

            int fragLen;

            if (packetSize > 0) {

                fragLen = Math.min(maxRecordSize, packetSize);

                fragLen = Math.min(fragLen, Record.maxDataSize);

            } else {

                fragLen = Record.maxDataSize;

            }

            if (fragmentSize > 0) {

                fragLen = Math.min(fragLen, fragmentSize);

            }

            if (isFirstRecordOfThePayload && needToSplitPayload()) {

                fragLen = 1;

                isFirstRecordOfThePayload = false;

            } else {

                fragLen = Math.min(fragLen, (limit - offset));

            }

            // use the buf of ByteArrayOutputStream

            int position = headerSize + writeCipher.getExplicitNonceSize();

            count = position;

            write(source, offset, fragLen);

            }

            // Encrypt the fragment and wrap up a record.

            // deliver this message

            deliverStream.write(buf, 0, count);    // may throw IOException

            deliverStream.flush();                 // may throw IOException

            }

            // reset the internal buffer

            count = 0;

            if (isFirstAppOutputRecord) {

                isFirstAppOutputRecord = false;

            }

            offset += fragLen;

        }

    }

    @Override

    void setDeliverStream(OutputStream outputStream) {

        this.deliverStream = outputStream;

    }

    /*

     * Need to split the payload except the following cases:

     *

     * 1. protocol version is TLS 1.1 or later;

     * 2. bulk cipher does not use CBC mode, including null bulk cipher suites.

     * 3. the payload is the first application record of a freshly

     *    negotiated TLS session.

     * 4. the CBC protection is disabled;

     *

     * By default, we counter chosen plaintext issues on CBC mode

     * ciphersuites in SSLv3/TLS1.0 by sending one byte of application

     * data in the first record of every payload, and the rest in

     * subsequent record(s). Note that the issues have been solved in

     * TLS 1.1 or later.

     *

     * It is not necessary to split the very first application record of

     * a freshly negotiated TLS session, as there is no previous

     * application data to guess.  To improve compatibility, we will not

     * split such records.

     *

     * This avoids issues in the outbound direction.  For a full fix,

     * the peer must have similar protections.

     */

    boolean needToSplitPayload() {

        return (!protocolVersion.useTLS11PlusSpec()) &&

                writeCipher.isCBCMode() && !isFirstAppOutputRecord &&

                Record.enableCBCProtection;

    }

    private int getFragLimit() {

        int fragLimit;

        if (packetSize > 0) {

            fragLimit = Math.min(maxRecordSize, packetSize);

            fragLimit = Math.min(fragLimit, Record.maxDataSize);

        } else {

            fragLimit = Record.maxDataSize;

        }

        if (fragmentSize > 0) {

            fragLimit = Math.min(fragLimit, fragmentSize);

        }

        return fragLimit;

    }

}