smbprotocol

import os from smbprotocol.connection import Connection from smbprotocol.session import Session from smbprotocol.tree import TreeConnect from smbprotocol.open import Open, CreateDisposition # Replace with your SMB server details SERVER_IP = os.environ.get('SMB_SERVER_IP', '127.0.0.1') USERNAME = os.environ.get('SMB_USERNAME', 'guest') PASSWORD = os.environ.get('SMB_PASSWORD', '') SHARE_NAME = os.environ.get('SMB_SHARE_NAME', 'share') # e.g., 'myshare' def list_share_contents(): connection = None session = None tree_connect = None try: print(f"Connecting to {SERVER_IP}...") connection = Connection(SERVER_IP, port=445) connection.connect() print("Connection established.") print(f"Authenticating as {USERNAME}...") session = Session(connection, USERNAME, PASSWORD) session.connect() print("Authentication successful.") print(f"Connecting to share \\{SERVER_IP}\{SHARE_NAME}...") # SMB paths use backslashes, but smbprotocol handles forward slashes too tree_connect = TreeConnect(connection, session, f'\\\\{SERVER_IP}\\{SHARE_NAME}') tree_connect.connect() print(f"Connected to share '{SHARE_NAME}'.") print(f"Listing contents of '{SHARE_NAME}':") # Open the directory itself to enumerate its contents dir_open = Open(tree_connect, '/*', CreateDisposition.FILE_OPEN, access_mask=0x80000000) # FILE_LIST_DIRECTORY dir_open.create() for entry in dir_open.query_directory(): print(f" - {entry.file_name}") dir_open.close() except Exception as e: print(f"An error occurred: {e}") finally: if tree_connect: print("Disconnecting from tree connect...") tree_connect.disconnect() if session: print("Logging off session...") session.logoff() if connection: print("Disconnecting connection...") connection.disconnect() print("Cleanup complete.") if __name__ == '__main__': list_share_contents()