T[] castBuffer = (T[]) buffer;
    sort(castBuffer, 0, bufferSize, comparator);
    if (bufferSize > k) {
      Arrays.fill(buffer, k, buffer.length, null);
      bufferSize = k;
      threshold = buffer[k - 1];
    }
    // Up to bufferSize, all elements of buffer are real Ts (not null unless T includes null)
    T[] topK = Arrays.copyOf(castBuffer, bufferSize);
    // we have to support null elements, so no ImmutableList for us

        // Given
        smbComTreeDisconnect = new SmbComTreeDisconnect(config);
        int[] bufferSizes = { 0, 10, 50, 100, 1024 };

        for (int bufferSize : bufferSizes) {
            byte[] dst = new byte[bufferSize];

            // When
            int result = smbComTreeDisconnect.writeParameterWordsWireFormat(dst, 0);

            // Then

        @DisplayName("Should handle various buffer configurations")
        void testVariousBufferConfigurations(int bufferSize, int offset, int length) {
            byte[] buffer = new byte[bufferSize];
            FileFsSizeInformation info = new FileFsSizeInformation();

            if (offset + length <= bufferSize && length >= 24) {
                assertDoesNotThrow(() -> {
                    info.decode(buffer, offset, length);

        String testPath = "\\server\\share";
        byte[] pathBytes = testPath.getBytes(java.nio.charset.StandardCharsets.UTF_16LE);
        int bufferSize = stringDataStart + pathBytes.length + 2; // +2 for null terminator
        byte[] buffer = new byte[bufferSize];

        // Path consumed (2 bytes) - value: 10 * 2 = 20
        SMBUtil.writeInt2(20, buffer, 0);

        // Number of referrals (2 bytes) - value: 1

      }
      OPCODE_CONTROL_CLOSE -> {
        var code = CLOSE_NO_STATUS_CODE
        var reason = ""
        val bufferSize = controlFrameBuffer.size
        if (bufferSize == 1L) {
          throw ProtocolException("Malformed close payload length of 1.")
        } else if (bufferSize != 0L) {
          code = controlFrameBuffer.readShort().toInt()
          reason = controlFrameBuffer.readUtf8()

     * @return The buffer size
     */
    public int getBufferSize() {
        return bufferSize;
    }

    /**
     * Sets the buffer size for data transfers.
     *
     * @param bufferSize The buffer size
     */
    public void setBufferSize(final int bufferSize) {
        this.bufferSize = bufferSize;
    }

    /**
     * Gets the local IP address for passive mode FTP.
     *

    void testWriteBytesWireFormatWithDifferentBufferSizes(int bufferSize) {
        request = new Smb2QueryInfoRequest(mockConfig, testFileId);

        if (bufferSize > 0) {
            Encodable mockInputBuffer = mock(Encodable.class);
            when(mockInputBuffer.encode(any(byte[].class), anyInt())).thenReturn(bufferSize);
            when(mockInputBuffer.size()).thenReturn(bufferSize);

            try {

            final int bufferSize, final Consumer<String> outputCallback) {
        final ProcessBuilder pb = new ProcessBuilder(cmdList);
        pbCall.accept(pb);
        destroyProcess(sessionId);
        JobProcess jobProcess;
        try {
            jobProcess = new JobProcess(pb.start(), bufferSize, outputCallback);

        @DisplayName("Should handle maximum buffer index")
        void testMaxBufferIndex() throws SMBProtocolDecodingException {
            // Given
            int bufferSize = 10000;
            byte[] buffer = new byte[bufferSize];
            int bufferIndex = bufferSize - 4;
            SMBUtil.writeInt2(4, buffer, bufferIndex);
            SMBUtil.writeInt2(0, buffer, bufferIndex + 2);

            // When

            this.fst = synonymMap.fst;
            this.fstReader = fst.getBytesReader();
            scratchArc = new FST.Arc<>();
        }

        ch = 0;
        readBuffer = new char[BUFFER_SIZE];
        readBufferIndex = BUFFER_SIZE;
        readBufferLen = 0;
        block = new StringBuilder();
        nextBlkStart = 0;
        queue = new PriorityQueue<>(100, new MyTokensComparator());