this.dataCount = dataCount;
    }

    /**
     * Sets the transaction buffer
     * @param buffer
     *            the transaction buffer to use
     */
    public void setBuffer(final byte[] buffer) {
        this.txn_buf = buffer;
    }

    /**
     * Releases and returns the transaction buffer
     * @return the txn_buf
     */
    public byte[] releaseBuffer() {

        this.maxDataCount = maxDataCount;
    }

    /**
     * Sets the transaction buffer
     * @param buffer
     *            the transaction buffer to use
     */
    public void setBuffer(final byte[] buffer) {
        this.txn_buf = buffer;
    }

    /**
     * Releases and returns the transaction buffer
     * @return the txn_buf
     */
    public byte[] releaseBuffer() {

    }

    @Test
    @DisplayName("Test buffer management")
    void testBufferManagement() {
        byte[] buffer = new byte[1024];
        buffer[0] = 0x42;

        transaction.setBuffer(buffer);

        byte[] released = transaction.releaseBuffer();
        assertSame(buffer, released);
        assertEquals(0x42, released[0]);

        // After release, getting buffer again should return null

    @Test
    @DisplayName("Test buffer management")
    void testBufferManagement() {
        byte[] buffer = new byte[1024];
        buffer[0] = 0x42;
        buffer[1] = 0x43;

        response.setBuffer(buffer);

        byte[] released = response.releaseBuffer();
        assertSame(buffer, released);
        assertEquals(0x42, released[0]);
        assertEquals(0x43, released[1]);

    }

    @Test
    @DisplayName("Test buffer management")
    void testBufferManagement() {
        // Test buffer setting and releasing
        byte[] testBuffer = new byte[1024];
        response.setBuffer(testBuffer);

        byte[] releasedBuffer = response.releaseBuffer();
        assertSame(testBuffer, releasedBuffer);

        // After release, the buffer should be null

        RdmaMemoryRegion sendRegion = bufferManager.getSendRegion(requestBuffer.remaining());
        sendRegion.getBuffer().put(requestBuffer);
        sendRegion.getBuffer().flip();
        
        try {
            // Send via RDMA
            rdmaConnection.send(sendRegion.getBuffer(), sendRegion);
            rdmaConnection.consumeSendCredit();
            
            // Receive response

            // Test data preparation
            byte[] testData = "Test RDMA data transfer".getBytes();
            sendBuffer.getBuffer().put(testData);
            sendBuffer.getBuffer().flip();
            assertEquals(testData.length, sendBuffer.getBuffer().remaining(), "Buffer should contain test data");

            // Clean up
            bufferManager.releaseSendRegion(sendBuffer);

  private final FbosByteSource byteSource;
  private final State state;

  /** ByteArrayOutputStream that exposes its internals. */
  private static final class MemoryOutput extends ByteArrayOutputStream {
    byte[] getBuffer() {
      return buf;
    }

    int getCount() {
      return count;
    }
  }

  /** Returns the file holding the data (possibly null). */
  @VisibleForTesting
  @Nullable File getFile() {

        if (this.state == 0) {
            bind();
        }
        final byte[] inB = this.transportContext.getBufferCache().getBuffer();
        final byte[] out = this.transportContext.getBufferCache().getBuffer();
        try {
            final NdrBuffer buf = encodeMessage(msg, out);
            final int off = sendFragments(msg, out, buf);

        boolean isDirect;
        DcerpcException de;

        if (state == 0) {
            bind();
        }

        isDirect = true;

        stub = jcifs.smb1.smb1.BufferCache.getBuffer();
        try {
            int off, tot, n;

            buf = new NdrBuffer(stub, 0);

            msg.flags = DCERPC_FIRST_FRAG | DCERPC_LAST_FRAG;
            msg.call_id = call_id++;