}

  override fun requestBodyStart(call: Call) {
    logWithTime("requestBodyStart")
  }

  override fun requestBodyEnd(
    call: Call,
    byteCount: Long,
  ) {
    logWithTime("requestBodyEnd: byteCount=$byteCount")
  }

  override fun requestFailed(
    call: Call,
    ioe: IOException,
  ) {
    logWithTime("requestFailed: $ioe")
  }

        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);

  }

  override fun requestBodyStart(call: Call) {
    logWithTime("requestBodyStart")
  }

  override fun requestBodyEnd(
    call: Call,
    byteCount: Long,
  ) {
    logWithTime("requestBodyEnd: byteCount=$byteCount")
  }

  override fun requestFailed(
    call: Call,
    ioe: IOException,
  ) {
    logWithTime("requestFailed: $ioe")
  }

) : Sink by delegate {
  private var bytesWritten = 0L

  override fun write(
    source: Buffer,
    byteCount: Long,
  ) {
    if (byteCount == 0L) return // Avoid double-triggering.

    if (bytesWritten == triggerByteCount) {
      source.skip(byteCount)
      return
    }

    val toWrite = minOf(byteCount, triggerByteCount - bytesWritten)
    bytesWritten += toWrite

    delegate.write(source, toWrite)

        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 {

    object : ForwardingSink(sink) {
      @Throws(IOException::class)
      override fun write(
        source: Buffer,
        byteCount: Long,
      ) {
        val bytes = source.readByteString(byteCount)
        delegate.write(
          Buffer()
            .write(bytes.toAsciiUppercase()),
          byteCount,
        )
      }
    }

  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

import okhttp3.Headers.Companion.toHeaders
import org.junit.jupiter.api.Test

class HeadersJvmTest {
  @Test fun byteCount() {
    assertThat(Headers.EMPTY.byteCount()).isEqualTo(0L)
    assertThat(
      Headers
        .Builder()
        .add("abc", "def")
        .build()
        .byteCount(),
    ).isEqualTo(10L)
    assertThat(
      Headers
        .Builder()
        .add("abc", "def")

      }

      override fun read(
        sink: Buffer,
        byteCount: Long,
      ): Long {
        require(byteCount >= 0L) { "byteCount < 0: $byteCount" }
        check(currentPart == this) { "closed" }

        return source.timeout().intersectWith(timeout) {
          when (val limit = currentPartBytesRemaining(maxByteCount = byteCount)) {
            0L -> -1L // No more bytes in this part.

  ) = onEvent(RequestHeadersEnd(System.nanoTime(), call, request.headers.byteCount()))

  override fun requestBodyStart(call: Call) =
    onEvent(
      RequestBodyStart(
        System.nanoTime(),
        call,
      ),
    )

  override fun requestBodyEnd(
    call: Call,
    byteCount: Long,
  ) = onEvent(RequestBodyEnd(System.nanoTime(), call, byteCount))

  override fun requestFailed(
    call: Call,