import java.io.OutputStream;

/**
 * An OutputStream that counts the number of bytes written.
 *
 * @author Chris Nokleberg
 * @since 1.0
 */
@J2ktIncompatible
@GwtIncompatible
public final class CountingOutputStream extends FilterOutputStream {

  private long count;

  /**
   * Wraps another output stream, counting the number of bytes written.
   *
   * @param out the output stream to be wrapped
   */

    int written = super.writePayload(dst, dstIndex);
    
    if (rdmaChannelInfo != null) {
        // Add RDMA read channel info
        writeInt4(dst, dstIndex + written, rdmaChannelInfo.getRemoteKey());
        written += 4;
        writeInt8(dst, dstIndex + written, rdmaChannelInfo.getAddress());
        written += 8;
        writeInt4(dst, dstIndex + written, rdmaChannelInfo.getLength());

     *
     * @param s the short value to encode
     * @param dst the destination byte array
     * @param di the starting index in the destination array
     * @return the number of bytes written (2)
     */
    public static int enc_uint16be(final short s, final byte[] dst, int di) {
        dst[di] = (byte) (s >> 8 & 0xFF);
        di++;
        dst[di] = (byte) (s & 0xFF);
        return 2;
    }

    private int totalBytesWritten;

    /**
     * Gets the number of bytes written in the last chunk
     * @return the chunkBytesWritten
     */
    public int getChunkBytesWritten() {
        return this.chunkBytesWritten;
    }

    /**
     * Gets the number of chunks successfully written
     * @return the chunksWritten
     */
    public int getChunksWritten() {

        packet.questionCount = 1;
        packet.answerCount = 2;
        packet.authorityCount = 3;
        packet.additionalCount = 4;

        int written = packet.writeHeaderWireFormat(dst, 0);
        assertEquals(NameServicePacket.HEADER_LENGTH, written);

        assertEquals((byte) 0x12, dst[0]); // nameTrnId
        assertEquals((byte) 0x34, dst[1]);

        byte[] dst = new byte[10];

        // writeSetupWireFormat writes 4 bytes:
        // - subCommand byte
        // - 3 zero bytes (padding and no FID)
        int written = pipe.writeSetupWireFormat(dst, 0);

        assertEquals(4, written, "Setup should write 4 bytes");
        assertEquals(SmbComTransaction.TRANS_WAIT_NAMED_PIPE, dst[0], "First byte should be subCommand");
        assertEquals(0, dst[1], "Second byte should be 0");

     */
    public SmbComWriteResponse(final Configuration config) {
        super(config);
    }

    /**
     * Gets the number of bytes written.
     *
     * @return the count of bytes written
     */
    public long getCount() {
        return this.count;
    }

    @Override
    protected int writeParameterWordsWireFormat(final byte[] dst, final int dstIndex) {
        return 0;
    }

            byte[] buffer = new byte[8];
            byte[] original = buffer.clone();

            // Act
            int written = resp.writeBytesWireFormat(buffer, index);

            // Assert
            assertEquals(0, written, "Should report 0 bytes written");
            assertArrayEquals(original, buffer, "Buffer must remain unchanged");
        }

        @Test

        // Total bytes written: 8+8+8+8+2+6=40
        // But writeInt8 at position 34 needs to access position 34+7=41
        // So we need at least 42 bytes in the array
        byte[] dst = new byte[42];

        // When
        int result = trans2SetFileInformation.writeDataWireFormat(dst, 0);

        // Then
        assertEquals(40, result); // Returns 40 bytes written
    }

    @Test

        super(config);
    }

    /**
     * Returns the number of bytes written.
     *
     * @return the count
     */
    public final int getCount() {
        return this.count;
    }

    /**
     * Returns the number of bytes remaining to be written.
     *
     * @return the remaining
     */
    public final int getRemaining() {
        return this.remaining;