* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 */

package jcifs.smb1.dcerpc.msrpc;

import jcifs.smb1.smb1.FileEntry;
import jcifs.smb1.smb1.SmbShareInfo;

/**
 * MS-RPC share enumeration operation.
 *
 * This class implements the Server Service (SRVSVC) ShareEnumAll operation
 * for enumerating network shares on a remote server.
 */
public class MsrpcShareEnum extends srvsvc.ShareEnumAll {

        when(tree.acquire(false)).thenReturn(tree);
        when(tree.getTreeType()).thenReturn(7);
        when(tree.getShare()).thenReturn("SHARE");
        setTree(c, tree);

        assertEquals(7, c.getTreeType());
        assertEquals("SHARE", c.getConnectedShare());
        // try-with-resources closes the acquired tree
        verify(tree, atLeastOnce()).close();
    }

    @Test

                    for (int i = 0; i < opsPerThread; i++) {
                        String share = "share" + (threadId % 3); // Use 3 different shares

                        // Get or create tree
                        SmbTreeImpl tree = session.getSmbTree(share, null);
                        assertNotNull(tree, "Tree should not be null");

                        // Verify we can acquire the tree

 * on the server, such as a file share or printer.
 *
 * @author mbechler
 */
public class Smb2TreeConnectRequest extends ServerMessageBlock2Request<Smb2TreeConnectResponse> {

    private int treeFlags;
    private final String path;

    /**
     * Creates a new SMB2 tree connect request to establish a connection to a network share.
     *
     * @param config the CIFS configuration

        return this.sessionKey;
    }

    @Override
    public SmbTreeImpl getSmbTree(String share, String service) {
        if (share == null) {
            share = "IPC$";
        }

        synchronized (this.trees) {
            for (SmbTreeImpl t : this.trees) {
                if (t.matches(share, service)) {
                    return t.acquire();
                }
            }

                        .withCopyFileToContainer(MountableFile.forHostPath(tempDir.resolve("shared")), "/share/shared")
                        .withCommand("-u", USERNAME + ";" + PASSWORD, "-s", "public;/share/public;yes;no;yes;all;;all;all", "-s",
                                "shared;/share/shared;no;no;no;all;" + USERNAME + ";all;all", "-g", "log level = 1", "-g",

        @CsvSource({ "'smb://user:password@server/share', 'smb://user:******@server/share'",
                "'smb://domain\\user:secret@host/path', 'smb://domain\\user:******@host/path'",
                "'smb2://admin:admin123@192.168.1.1/folder', 'smb2://admin:******@192.168.1.1/folder'",
                "'smbs://test:p@******@****.***/share', 'smbs://test:******@example.com/share'" })

    void happyPath_enumerates_and_closes(String wildcard) throws Exception {
        // Setup required for this test
        when(parent.getLocator()).thenReturn(locator);
        when(locator.getUNCPath()).thenReturn("\\\\server\\share\\dir\\");
        when(tree.getConfig()).thenReturn(config);
        when(config.getMaximumBufferSize()).thenReturn(65535);
        when(config.getListSize()).thenReturn(65535);

        when(mockDir.getPath()).thenReturn("/share/folder/");

        // Implementation uses dir.getPath() to check if file should be in that path
        SmbFilenameFilter filter = (dir, name) -> {
            if (dir == null)
                return false;
            String path = dir.getPath();
            // Accept files in /share/folder/ that are text files

    class HappyPath {
        @Test
        void acceptWithMockedFile() throws Exception {
            SmbFile mockFile = mock(SmbFile.class);
            when(mockFile.getPath()).thenReturn("/share/file.txt");
            assertTrue(ALWAYSACTIVE.accept(mockFile));
        }
    }

    @Nested
    @DisplayName("Invalid input – null file handling")
    class InvalidPath {
        @Test