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

        when(treeConnection.acquire()).thenReturn(treeConnection);
        when(treeConnection.getSession()).thenReturn(session);
        when(session.getTransport()).thenReturn(transport);
        handle = new SmbTreeHandleImpl(resourceLoc, treeConnection);
    }

    @Test
    @DisplayName("Constructor acquires tree connection; getConfig/isConnected delegate")

    }

    @Test
    void testConstructor() {
        assertEquals("test-channel", channelInfo.getChannelId());
        assertEquals(mockTransport, channelInfo.getTransport());
        assertEquals(localInterface, channelInfo.getLocalInterface());
        assertEquals(remoteInterface, channelInfo.getRemoteInterface());
        assertEquals(ChannelState.DISCONNECTED, channelInfo.getState());

        channel.setState(ChannelState.ESTABLISHED);

        // Verify channel properties
        assertEquals("test-channel", channel.getChannelId());
        assertEquals(mockTransport, channel.getTransport());
        assertEquals(localNic, channel.getLocalInterface());
        assertEquals(remoteNic, channel.getRemoteInterface());
        assertEquals(ChannelState.ESTABLISHED, channel.getState());

    void setUp() throws Exception {
        MockitoAnnotations.openMocks(this);
        when(session.getConfig()).thenReturn(config);
        when(session.getContext()).thenReturn(context);
        when(session.getTransport()).thenReturn(transport);
        when(session.acquire()).thenReturn(session);
        when(config.isTraceResourceUsage()).thenReturn(false);
        when(context.getConfig()).thenReturn(config);

        try (SmbTreeHandleImpl th = treeConn.connectHost(locator, locator.getServerWithDfs());
                SmbSessionImpl session = th.getSession();
                SmbTransportImpl transport = session.getTransport()) {
            try {
                entries = doMsrpcShareEnum(tc, locator, transport.getRemoteAddress());
            } catch (final IOException ioe) {
                if (th.isSMB2()) {

        when(tree.isConnected()).thenReturn(true);
        when(tree.getSession()).thenReturn(session);
        when(tree.acquire(false)).thenReturn(tree);
        when(session.getTransport()).thenReturn(transport);
        when(transport.isDisconnected()).thenReturn(false);
        when(transport.getRemoteHostName()).thenReturn("host");
        setTree(c, tree);

        }

        try (SmbSessionImpl sess = getSmbSession(ctx, targetHost, targetDomain);
                SmbTransportImpl transport = sess.getTransport();
                SmbTreeImpl ipc = sess.getSmbTree("IPC$", null)) {

            final DfsReferralRequestBuffer dfsReq = new DfsReferralRequestBuffer(path, 3);
            DfsReferralResponseBuffer dfsResp;