override fun write(
      source: Buffer,
      byteCount: Long,
    ) {
      check(!closed) { "closed" }
      if (contentLength != -1L && bytesReceived + byteCount > contentLength) {
        throw ProtocolException(
          "expected $contentLength bytes but received ${bytesReceived + byteCount}",
        )
      }
      try {
        super.write(source, byteCount)
        this.bytesReceived += byteCount

        }

        this.byteCount = SMBUtil.readInt2(buffer, bufferIndex);
        bufferIndex += 2;

        if (this.byteCount != 0) {
            int n = readBytesWireFormat(buffer, bufferIndex);
            if ((n != this.byteCount) && log.isTraceEnabled()) {
                log.trace("byteCount=" + this.byteCount + " but readBytesWireFormat returned " + n);
            }

        wordCount /= 2;
        dst[start] = (byte) (wordCount & 0xFF);

        byteCount = writeBytesWireFormat(dst, dstIndex + 2);
        dst[dstIndex] = (byte) (byteCount & 0xFF);
        dstIndex++;
        dst[dstIndex++] = (byte) (byteCount >> 8 & 0xFF);
        dstIndex += byteCount;

        /* Normally, without intervention everything would batch

        this.wordCount /= 2;
        dst[start] = (byte) (this.wordCount & 0xFF);

        this.byteCount = writeBytesWireFormat(dst, dstIndex + 2);
        dst[dstIndex] = (byte) (this.byteCount & 0xFF);
        dstIndex++;
        dst[dstIndex++] = (byte) (this.byteCount >> 8 & 0xFF);
        dstIndex += this.byteCount;

        /*
         * Normally, without intervention everything would batch

        assertTrue(result.contains("byteCount=8"));
        assertTrue(result.contains("oemDomainName=TEST_DOMAIN"));
    }

    // Helper method to set protected byteCount field using reflection
    private void setByteCount(SmbComNegotiateResponse response, int byteCount) {
        try {
            Field field = response.getClass().getSuperclass().getDeclaredField("byteCount");
            field.setAccessible(true);

      }
    }

    override fun read(
      sink: Buffer,
      byteCount: Long,
    ): Long {
      require(byteCount >= 0L) { "byteCount < 0: $byteCount" }
      check(!closed) { "closed" }
      if (bytesRemaining == 0L) return -1

      val read = super.read(sink, minOf(bytesRemaining, byteCount))
      if (read == -1L) {

        serverData.encryptionKeyLength = 8;
        response.byteCount = 15; // 8 bytes key + 6 bytes "DOMAIN" + 1 null terminator

        // Prepare byte array
        byte[] encryptionKey = "12345678".getBytes();
        byte[] domainNameBytes = "DOMAIN".getBytes(ServerMessageBlock.OEM_ENCODING);
        ByteBuffer buffer = ByteBuffer.allocate(response.byteCount);
        buffer.put(encryptionKey);
        buffer.put(domainNameBytes);

            }
            bufferIndex += wordCount * 2;
        }

        byteCount = readInt2(buffer, bufferIndex);
        bufferIndex += 2;

        if (byteCount != 0) {
            int n = readBytesWireFormat(buffer, bufferIndex);
            if ((n != byteCount) && (LogStream.level >= 5)) {
                log.println("byteCount=" + byteCount + " but readBytesWireFormat returned " + n);
            }

        private long totalBytesWritten = 0L;
        private boolean completed = false;

        @Override
        public void write(Buffer source, long byteCount) throws IOException {
          super.write(source, byteCount);
          totalBytesWritten += byteCount;
          progressListener.update(totalBytesWritten, contentLength(), completed);
        }

        @Override
        public void close() throws IOException {

  private val bytesPerPeriod: Long,
  private val periodDelayNanos: Long,
) : Sink by delegate {
  private var bytesWrittenSinceLastDelay = 0L

  override fun write(
    source: Buffer,
    byteCount: Long,
  ) {
    var bytesLeft = byteCount

    while (bytesLeft > 0) {
      if (bytesWrittenSinceLastDelay == bytesPerPeriod) {
        flush()
        socket.sleepWhileOpen(periodDelayNanos)
        bytesWrittenSinceLastDelay = 0