/**
     * Get the transport for this channel
     *
     * @return SMB transport
     */
    public SmbTransport getTransport() {
        return transport;
    }

    /**
     * Set the transport for this channel
     *
     * @param transport SMB transport
     */
    public void setTransport(SmbTransport transport) {
        this.transport = transport;
    }

    /**
     * Get local network interface

        verify(enc, times(1)).decryptMessage(eq(new byte[] { 5 }));
    }

    @Test
    @DisplayName("getTransport returns acquired transport instance")
    void testGetTransportAcquire() {
        SmbSessionImpl session = newSession();
        clearInvocations(transport);
        SmbTransportImpl t = session.getTransport();
        assertSame(transport, t);
        verify(transport, times(1)).acquire();
    }

    @Test

        if (channelGroup != null) {
            for (ChannelInfo channel : channelGroup.getChannels()) {
                try {
                    if (channel.getTransport() != null) {
                        channel.getTransport().disconnect(true);
                    }
                } catch (Exception e) {
                    log.warn("Error closing channel {}: {}", channel.getChannelId(), e.getMessage());

            return transport.getNegotiateResponse().getSendBufferSize();
        }
    }

    @Override
    public int getReceiveBufferSize() throws SmbException {
        try (SmbSessionImpl session = this.treeConnection.getSession(); SmbTransportImpl transport = session.getTransport()) {

    
    // Write some data
    try (OutputStream os = file.getOutputStream()) {
        os.write("test data".getBytes());
    }
    
    // Simulate network disconnection
    file.getTree().getSession().getTransport().disconnect();
    
    // Try to read - should trigger reconnection
    try (InputStream is = file.getInputStream()) {
        byte[] buffer = new byte[9];
        is.read(buffer);

        try (SmbSessionImpl session = getSession()) {
            if (session == null) {
                return;
            }
            try (SmbTransportImpl transport = session.getTransport()) {
                synchronized (transport) {
                    final SmbTreeImpl t = getTreeInternal();
                    if (t != null) {
                        try {

        if (this.connectionState.get() == 2) {
            // we are connected, so we know
            return isDfs();
        }
        try (SmbTransportImpl transport = this.session.getTransport()) {
            return transport.getNegotiateResponse().isDFSSupported();
        }
    }

    /**
     * @return the session this tree is connected in
     */
    public SmbSessionImpl getSession() {

        request.setFlags(SMB2_SESSION_FLAG_BINDING);
        request.setSecurityBuffer(bindingHash);
        
        Smb2SessionSetupResponse response = (Smb2SessionSetupResponse) 
            channel.getTransport().send(request);
        
        if (!response.isSuccess()) {
            throw new IOException("Channel binding failed: " + response.getStatus());
        }
    }
    

        try {
            log.info("Attempting to recover channel {}", channel.getChannelId());

            // Disconnect existing transport
            SmbTransport oldTransport = channel.getTransport();
            if (oldTransport != null) {
                try {
                    oldTransport.close();
                } catch (Exception e) {
                    log.debug("Error disconnecting old transport", e);

        return channels.values().stream().filter(c -> c.getTransport() == transport).findFirst().orElse(null);
    }

    /**
     * Remove a channel
     *
     * @param channel channel to remove
     */
    public void removeChannel(ChannelInfo channel) {
        channels.remove(channel.getChannelId());
        try {
            if (channel.getTransport() != null) {
                channel.getTransport().close();