sima-cli 0.0.19__py3-none-any.whl → 0.0.20__py3-none-any.whl

sima_cli/__version__.py

@@ -1,2 +1,2 @@
 # sima_cli/__version__.py
-__version__ = "0.0.19"
+__version__ = "0.0.20"

sima_cli/cli.py

@@ -9,10 +9,11 @@ from sima_cli.__version__ import __version__
 from sima_cli.utils.config import CONFIG_PATH
 from sima_cli.install.optiview import install_optiview
 from sima_cli.install.hostdriver import install_hostdriver
+from sima_cli.serial.serial import connect_serial
 
 # Entry point for the CLI tool using Click's command group decorator
 @click.group()
-@click.option('-i', '--internal', is_flag=True, help="Use internal Artifactory resources.")
+@click.option('-i', '--internal', is_flag=True, help="Use internal Artifactory resources, Authorized Sima employees only")
 @click.pass_context
 def main(ctx, internal):
     """
@@ -226,18 +227,23 @@ def show_mla_memory_usage(ctx):
 @main.command(name="bootimg")
 @click.option("-v", "--version", required=True, help="Firmware version to download and write (e.g., 1.6.0)")
 @click.option("--boardtype", type=click.Choice(["modalix", "mlsoc"], case_sensitive=False), default="mlsoc", show_default=True, help="Target board type.")
+@click.option("--netboot", is_flag=True, default=False, show_default=True, help="Prepare image for network boot and launch TFTP server.")
+@click.option("--autoflash", is_flag=True, default=False, show_default=True, help="Net boot the DevKit and automatically flash the internal storage")
 @click.pass_context
-def bootimg_cmd(ctx, version, boardtype):
+def bootimg_cmd(ctx, version, boardtype, netboot, autoflash):
     """
-    Download and burn a bootable image onto removable media (e.g., SD card or USB stick).
+    Download and burn a removable media or setup TFTP boot.
 
     Examples:
       sima-cli bootimg -v 1.6.0
       sima-cli bootimg -v 1.6.0 --boardtype mlsoc
       sima-cli bootimg -v 1.6.0 --boardtype modalix
+      sima-cli bootimg -v 1.6.0 --boardtype modalix --netboot
+      sima-cli bootimg -v 1.6.0 --boardtype modalix --autoflash
     """
 
     from sima_cli.update.bootimg import write_image
+    from sima_cli.update.netboot import setup_netboot
 
     internal = ctx.obj.get("internal", False)
 
@@ -247,7 +253,12 @@ def bootimg_cmd(ctx, version, boardtype):
     
     try:
         boardtype = boardtype if boardtype != 'mlsoc' else 'davinci'
-        write_image(version, boardtype, 'yocto', internal)
+        if netboot or autoflash:
+            setup_netboot(version, boardtype, internal, autoflash)
+            click.echo("✅ Netboot image prepared and TFTP server is running.")
+        else:
+            write_image(version, boardtype, 'yocto', internal)
+            click.echo("✅ Boot image successfully written.")
         click.echo("✅ Boot image successfully written.")
     except Exception as e:
         click.echo(f"❌ Failed to write boot image: {e}", err=True)
@@ -269,10 +280,10 @@ def install_cmd(ctx, component, version):
     Install supported components such as SDKs or tools.
 
     Examples:
-      cli install palette -v 1.6.0
-      cli install hostdriver -v 1.6.0
 
-      cli install optiview
+        sima-cli install hostdriver -v 1.6.0
+
+        sima-cli install optiview
     """
     component = component.lower()
     internal = ctx.obj.get("internal", False)
@@ -295,9 +306,27 @@ def install_cmd(ctx, component, version):
         install_optiview()
 
     click.echo("✅ Installation complete.")
-    
 
 # ----------------------
+# Serial Subcommands
+# ----------------------
+@main.command(name="serial")
+@click.option("-b", "--baud", default=115200, show_default=True, help="Baud rate for the serial connection")
+@click.pass_context
+def serial_cmd(ctx, baud):
+    """
+    Connect to the UART serial console of the DevKit.
+
+    Automatically detects the serial port and launches a terminal emulator:
+
+    - macOS: uses 'picocom'
+    
+    - Linux: uses 'picocom'
+    
+    - Windows: shows PuTTY/Tera Term setup instructions
+    """
+    connect_serial(ctx, baud)
+# ----------------------
 # App Zoo Subcommands
 # ----------------------
 # @main.group()

sima_cli/install/optiview.py

@@ -1,9 +1,7 @@
 import sys
 import platform
 import click
-import urllib
 import json
-import tempfile
 import tarfile
 import os
 import subprocess
@@ -31,6 +29,12 @@ def install_optiview():
         version_url = f"{download_url_base}optiview/metadata.json"
         downloads_dir = Path.home() / "Downloads" / "optiview-installer"
         downloads_dir.mkdir(parents=True, exist_ok=True)
+
+        # Always redownload the metadata file to get the latest version
+        metadata_path = downloads_dir / "metadata.json"
+        if metadata_path.exists():
+            metadata_path.unlink()
+
         metadata_path = download_file_from_url(version_url, dest_folder=downloads_dir, internal=False)
 
         with open(metadata_path, "r") as f:
@@ -70,7 +74,8 @@ def install_optiview():
         else:
             subprocess.run(["bash", os.path.basename(script_path)], check=True, cwd=downloads_dir)
 
-        click.echo(f"✅ Optiview installed successfully. Run {downloads_dir}/run.sh to start optiview")
+        script_name = "run.bat" if platform.system() == "Windows" else "run.sh"
+        click.echo(f"✅ Optiview installed successfully. Run {downloads_dir}/{script_name} to start OptiView")
 
     except Exception as e:
         click.echo(f"❌ Installation failed: {e}")

sima_cli/serial/serial.py

@@ -0,0 +1,114 @@
+import platform
+import subprocess
+import shutil
+import click
+from sima_cli.utils.env import is_sima_board
+
+def connect_serial(ctx, baud):
+    """
+    Connect to the UART serial console of the DevKit.
+    Automatically installs required tools if missing.
+    """
+    if is_sima_board():
+        click.echo("🚫 This command is not supported on the DevKit. Please run it from your host machine.")
+        ctx.exit(1)
+
+    system = platform.system()
+    internal = ctx.obj.get("internal", False)
+
+    if system == "Darwin":
+        _connect_mac(baud)
+    elif system == "Linux":
+        _connect_linux(baud)
+    elif system == "Windows":
+        _print_windows_instructions()
+    else:
+        click.echo(f"⚠️ Unsupported OS: {system}. Only macOS, Linux, and Windows are supported.")
+        ctx.exit(1)
+
+    click.echo("✅ Serial session ended.")
+
+
+def _connect_mac(baud):
+    terminal = "picocom"
+    if not shutil.which(terminal):
+        click.echo("⚙️ 'picocom' is not installed. Attempting to install with Homebrew...")
+        if shutil.which("brew"):
+            subprocess.run(["brew", "install", "picocom"], check=True)
+        else:
+            click.echo("❌ Homebrew not found. Please install Homebrew first: https://brew.sh/")
+            raise SystemExit(1)
+
+    ports = sorted(
+        subprocess.getoutput("ls /dev/tty.usbserial-* /dev/cu.usbserial-* 2>/dev/null").splitlines()
+    )
+    if not ports:
+        click.echo("❌ No USB serial device found.")
+        raise SystemExit(1)
+
+    click.echo(f"Connecting to device with picocom ({baud} baud)...")
+    click.echo("🧷 To exit: Press Ctrl + A, then Ctrl + X")
+    click.echo("📜 Scrollback will work in your terminal as expected.\n")
+    
+    if not click.confirm("Proceed to connect?", default=True):
+        click.echo("❎ Connection aborted by user.")
+        return
+
+    port = ports[0]
+    click.echo(f"🔌 Connecting to {port} with picocom (115200 8N1)...")
+    try:
+        subprocess.run([
+            terminal,
+            "-b", f"{baud}",
+            "--databits", "8",
+            "--parity", "n",
+            "--stopbits", "1",
+            port
+        ])
+    except KeyboardInterrupt:
+        click.echo("\n❎ Serial connection interrupted by user.")
+
+
+def _connect_linux(baud):
+    terminal = "picocom"
+    if not shutil.which(terminal):
+        click.echo("⚙️ 'picocom' is not installed. Attempting to install via apt...")
+        if shutil.which("apt-get"):
+            subprocess.run(["sudo", "apt-get", "update"], check=True)
+            subprocess.run(["sudo", "apt-get", "install", "-y", "picocom"], check=True)
+        else:
+            click.echo("❌ 'apt-get' not found. Please install picocom manually.")
+            raise SystemExit(1)
+
+    ports = sorted(
+        subprocess.getoutput("ls /dev/ttyUSB* 2>/dev/null").splitlines()
+    )
+    if not ports:
+        click.echo("❌ No USB serial device found.")
+        raise SystemExit(1)
+
+    port = ports[0]
+    click.echo(f"🔌 Connecting to {port} with picocom ({baud} 8N1)...")
+    try:
+        subprocess.run(
+            ["sudo", terminal, "-b", f"{baud}", "--databits", "8", "--parity", "n", "--stopbits", "1", port]
+        )
+    except KeyboardInterrupt:
+        click.echo("\n❎ Serial connection interrupted by user.")
+
+
+def _print_windows_instructions():
+    click.echo("📘 To connect to the DevKit via a serial terminal on Windows, follow these steps:\n")
+
+    click.echo("1. Identify the COM Port:")
+    click.echo("   • Open **Device Manager** → Expand **Ports (COM & LPT)**.")
+    click.echo("   • Look for an entry like **USB Serial Port (COMx)**.\n")
+
+    click.echo("2. Install and Open a Serial Terminal:")
+    click.echo("   • Use **PuTTY** (Download from https://www.putty.org/) or **Tera Term**.")
+    click.echo("   • Set the **Connection Type** to **Serial**.")
+    click.echo("   • Enter the correct **COM Port** (e.g., COM3).")
+    click.echo("   • Set **Baud Rate** to **115200**.")
+    click.echo("   • Click **Open** to start the connection.\n")
+
+    click.echo("🔌 You are now ready to connect to the DevKit over serial.")

sima_cli/update/bootimg.py

@@ -4,13 +4,14 @@ import subprocess
 import sys
 import os
 import select
-import time
 import re
+import threading
+from tftpy import TftpServer
+
 try:
     from tqdm import tqdm
 except ImportError:
     tqdm = None
-from sima_cli.update.updater import download_image
 
 
 def list_removable_devices():
@@ -177,7 +178,7 @@ def copy_image_to_device(image_path, device_path):
 
         # Regex to parse dd progress (more robust to handle variations)
         pattern = re.compile(
-            r"(?:(?:dd: )?(\d+)\s+bytes(?:.*?)copied)|(?:^(\d+)\s+bytes\s+transferred)",
+            r"(?:dd:\s*)?(?P<bytes>\d+)\s+bytes(?:\s+\(.*?\))?\s+(?:transferred|copied),?\s+[\d\.]+\s*s?,?\s+[\d\.]+\s+MB/s",
             re.IGNORECASE
         )
 
@@ -330,7 +331,6 @@ def write_image(version: str, board: str, swtype: str, internal: bool = False):
     except Exception as e:
         raise RuntimeError(f"❌ Failed to write image: {e}")
 
-
 if __name__ == "__main__":
     if len(sys.argv) != 2:
         click.echo("❌ Usage: python write_bootimg.py <image_file>")

sima_cli/update/netboot.py

@@ -0,0 +1,408 @@
+from sima_cli.update.updater import download_image
+from sima_cli.utils.net import get_local_ip_candidates
+from sima_cli.update.remote import get_remote_board_info, copy_file_to_remote_board, DEFAULT_PASSWORD, run_remote_command, init_ssh_session
+import os
+import platform
+import threading
+import socket
+import select
+import time
+import logging
+import click
+from errno import EINTR
+from tftpy import TftpServer, TftpException, TftpTimeout, TftpTimeoutExpectACK, DEF_TFTP_PORT, DEF_TIMEOUT_RETRIES
+from tftpy.TftpContexts import TftpContextServer
+from tftpy.TftpPacketFactory import TftpPacketFactory
+
+# Configuration constants
+MAX_BLKSIZE = 1468  # Block size for MTU compatibility
+SOCK_TIMEOUT = 2    # Timeout for faster retransmits
+
+log = logging.getLogger("tftpy.InteractiveTftpServer")
+emmc_image_paths = []
+
+def flash_emmc(client_manager, emmc_image_paths):
+    """Flash eMMC on a selected client device."""
+    clients = [
+        (ip, info) for ip, info in client_manager.get_client_info()
+        if info.get("state") == "Connected"
+    ]
+
+    if not clients:
+        click.echo("📭 No connected clients available to flash.")
+        return
+
+    if len(clients) == 1:
+        selected_ip = clients[0][0]
+    else:
+        click.echo("👥 Multiple connected clients found. Select one to flash:")
+        for idx, (ip, info) in enumerate(clients, 1):
+            board_info = info.get("board_info") or "Unknown"
+            click.echo(f"   {idx}. {ip} - {board_info}")
+        while True:
+            choice = input("Enter the number of the client to flash: ").strip()
+            if choice.isdigit() and 1 <= int(choice) <= len(clients):
+                selected_ip = clients[int(choice) - 1][0]
+                break
+            click.echo("❌ Invalid choice. Try again.")
+
+    click.echo(f"📡 Selected client: {selected_ip}")
+    remote_dir = "/tmp"
+
+    for path in emmc_image_paths:
+        click.echo(f"📤 Copying {path} to {selected_ip}:{remote_dir}")
+        success = copy_file_to_remote_board(selected_ip, path, remote_dir, passwd=DEFAULT_PASSWORD)
+        if not success:
+            click.echo(f"❌ Failed to copy {path} to {selected_ip}. Aborting.")
+            return
+
+    try:
+        ssh = init_ssh_session(selected_ip, password=DEFAULT_PASSWORD)
+
+        # Step a: Check if eMMC exists
+        check_cmd = "[ -e /dev/mmcblk0p1 ] || (echo '❌ /dev/mmcblk0p1 not found'; exit 1)"
+        run_remote_command(ssh, check_cmd)
+
+        # Step b: Flash with bmaptool
+        wic_path = next((p for p in emmc_image_paths if p.endswith(".wic.gz")), None)
+        if not wic_path:
+            click.echo("❌ No .wic.gz image found in emmc_image_paths.")
+            return
+
+        # Step c: umount the emmc
+        pre_unmount_cmd = (
+            "sudo mount | grep mmcblk0 | awk '{print $3}' | while read mnt; do "
+            "sudo umount \"$mnt\"; done"
+        )
+        run_remote_command(ssh, pre_unmount_cmd)
+
+        # Step d: bmaptool copy the image
+        filename = os.path.basename(wic_path)
+        remote_path = f"/tmp/{filename}"        
+        flash_cmd = f"sudo bmaptool copy {remote_path} /dev/mmcblk0"
+        run_remote_command(ssh, flash_cmd)
+
+        # Step d: Fix GPT
+        fix_cmd = 'sudo printf "fix\n" | sudo parted ---pretend-input-tty /dev/mmcblk0 print'
+        run_remote_command(ssh, fix_cmd)
+
+        click.echo("✅ Flash completed. Please reboot the board to boot from eMMC.")
+    except Exception as e:
+        click.echo(f"❌ Flashing failed: {e}")
+
+class ClientManager:
+    """Manages TFTP client state and monitoring."""
+    def __init__(self):
+        self.clients = {}
+        self.lock = threading.Lock()
+        self.shutdown_event = threading.Event()
+
+    def add_client(self, ip, filename):
+        """Add a new client with initial state."""
+        with self.lock:
+            if ip not in self.clients:
+                start_time = time.time()
+                self.clients[ip] = {
+                    'state': 'Booting',
+                    'filename': filename,
+                    'timestamp': start_time,
+                    'board_info': None
+                }
+                click.echo(f"📥 New client connected: {ip}")
+                if filename:
+                    click.echo(f"📄 Client {ip} requested file: {filename}")
+                # Start monitoring thread
+                threading.Thread(
+                    target=self.monitor_client,
+                    args=(ip, start_time),
+                    daemon=True
+                ).start()
+
+    def monitor_client(self, ip, start_time):
+        """Monitor client by calling get_remote_board_info after 1 minutes, retry every 10 seconds."""
+        try:
+            # Wait ~1 minute before first attempt
+            if self.shutdown_event.wait(timeout=max(0, 60 - (time.time() - start_time))):
+                return
+
+            while not self.shutdown_event.is_set():
+                try:
+                    board_type, build_version = get_remote_board_info(ip)
+
+                    if board_type and build_version:
+                        with self.lock:
+                            self.clients[ip]['state'] = 'Connected'
+                            self.clients[ip]['board_info'] = f"{board_type}|{build_version}"
+                        click.echo(f"✅ Board info retrieved for {ip}: {board_type} with {build_version}")
+                        break 
+                    else:
+                        log.debug(f"Incomplete board info for {ip}: {board_type}, {build_version}")
+
+                except Exception as e:
+                    log.info(f"get_remote_board_info failed for {ip}, retrying in 10s: {e}")
+
+                if self.shutdown_event.wait(timeout=10):
+                    break 
+
+        except Exception as e:
+            log.error(f"Unexpected error while monitoring {ip}: {e}")
+
+    def get_client_info(self):
+        """Return sorted client information for display."""
+        with self.lock:
+            return sorted(self.clients.items(), key=lambda x: x[0])
+
+    def shutdown(self):
+        """Signal monitoring threads to exit."""
+        self.shutdown_event.set()
+
+class InteractiveTftpServer(TftpServer):
+    """Custom TFTP server with client logging and monitoring."""
+    def __init__(self, tftproot, client_manager):
+        super().__init__(tftproot)
+        self.client_manager = client_manager
+
+    def listen(self, listenip="", listenport=DEF_TFTP_PORT, timeout=SOCK_TIMEOUT, retries=DEF_TIMEOUT_RETRIES):
+        """Override listen to log client IPs and filenames."""
+        tftp_factory = TftpPacketFactory()
+        if not listenip:
+            listenip = "0.0.0.0"
+        log.info(f"Server requested on ip {listenip}, port {listenport}")
+        try:
+            self.sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
+            self.sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
+            self.sock.bind((listenip, listenport))
+            self.sock.setblocking(0)
+            _, self.listenport = self.sock.getsockname()
+        except OSError as err:
+            raise err
+
+        self.is_running.set()
+        log.info("Starting receive loop...")
+        while True:
+            log.debug("shutdown_immediately is %s" % self.shutdown_immediately)
+            log.debug("shutdown_gracefully is %s" % self.shutdown_gracefully)
+            if self.shutdown_immediately:
+                log.info("Shutting down now. Session count: %d" % len(self.sessions))
+                self.sock.close()
+                for key in self.sessions:
+                    log.warning("Forcefully closed session with %s" % self.sessions[key].host)
+                    self.sessions[key].end()
+                self.sessions = []
+                self.is_running.clear()
+                self.shutdown_gracefully = self.shutdown_immediately = False
+                self.client_manager.shutdown()
+                break
+            elif self.shutdown_gracefully:
+                if not self.sessions:
+                    log.info("In graceful shutdown mode and all sessions complete.")
+                    self.sock.close()
+                    self.is_running.clear()
+                    self.shutdown_gracefully = self.shutdown_immediately = False
+                    self.client_manager.shutdown()
+                    break
+
+            inputlist = [self.sock]
+            for key in self.sessions:
+                inputlist.append(self.sessions[key].sock)
+
+            try:
+                readyinput, _, _ = select.select(inputlist, [], [], timeout)
+            except OSError as err:
+                if err.errno == EINTR:
+                    log.debug("Interrupted syscall, retrying")
+                    continue
+                else:
+                    raise
+
+            deletion_list = []
+            for readysock in readyinput:
+                if readysock == self.sock:
+                    log.debug("Data ready on our main socket")
+                    buffer, (raddress, rport) = self.sock.recvfrom(MAX_BLKSIZE)
+                    log.debug("Read %d bytes", len(buffer))
+
+                    if self.shutdown_gracefully:
+                        log.warning("Discarding data on main port, in graceful shutdown mode")
+                        continue
+
+                    key = f"{raddress}:{rport}"
+                    if key not in self.sessions:
+                        log.debug("Creating new server context for session key = %s" % key)
+                        filename = None
+                        if buffer[:2] == b'\x00\x01':  # RRQ packet
+                            filename = buffer[2:].split(b'\x00')[0].decode()
+                        self.client_manager.add_client(raddress, filename)
+                        self.sessions[key] = TftpContextServer(
+                            raddress,
+                            rport,
+                            timeout,
+                            self.root,
+                            self.dyn_file_func,
+                            self.upload_open,
+                            retries=retries
+                        )
+                        try:
+                            self.sessions[key].start(buffer)
+                        except TftpTimeoutExpectACK:
+                            self.sessions[key].timeout_expectACK = True
+                        except TftpException as err:
+                            deletion_list.append(key)
+                            log.error("Fatal exception thrown from session %s: %s" % (key, str(err)))
+                    else:
+                        log.warning("received traffic on main socket for existing session??")
+                    log.info("Currently handling these sessions:")
+                    for session_key, session in list(self.sessions.items()):
+                        log.info("    %s" % session)
+                else:
+                    for key in self.sessions:
+                        if readysock == self.sessions[key].sock:
+                            log.debug("Matched input to session key %s" % key)
+                            self.sessions[key].timeout_expectACK = False
+                            try:
+                                self.sessions[key].cycle()
+                                if self.sessions[key].state is None:
+                                    log.info("Successful transfer.")
+                                    deletion_list.append(key)
+                            except TftpTimeoutExpectACK:
+                                self.sessions[key].timeout_expectACK = True
+                            except TftpException as err:
+                                deletion_list.append(key)
+                                log.error("Fatal exception thrown from session %s: %s" % (key, str(err)))
+                            break
+                    else:
+                        log.error("Can't find the owner for this packet. Discarding.")
+
+            now = time.time()
+            for key in self.sessions:
+                try:
+                    self.sessions[key].checkTimeout(now)
+                except TftpTimeout as err:
+                    log.error(str(err))
+                    self.sessions[key].retry_count += 1
+                    if self.sessions[key].retry_count >= self.sessions[key].retries:
+                        log.debug("hit max retries on %s, giving up" % self.sessions[key])
+                        deletion_list.append(key)
+                    else:
+                        log.debug("resending on session %s" % self.sessions[key])
+                        self.sessions[key].state.resendLast()
+
+            for key in deletion_list:
+                log.info("Session %s complete" % key)
+                if key in self.sessions:
+                    log.debug("Gathering up metrics from session before deleting")
+                    self.sessions[key].end()
+                    metrics = self.sessions[key].metrics
+                    if metrics.duration == 0:
+                        log.info("Duration too short, rate undetermined")
+                    else:
+                        log.info("Transferred %d bytes in %.2f seconds" % (metrics.bytes, metrics.duration))
+                        log.info("Average rate: %.2f kbps" % metrics.kbps)
+                        click.echo(f"✅ Transfer to {self.sessions[key].host} complete: "
+                                   f"{metrics.bytes} bytes in {metrics.duration:.2f} s ({metrics.kbps:.2f} kbps)")
+                    log.info("%.2f bytes in resent data" % metrics.resent_bytes)
+                    log.info("%d duplicate packets" % metrics.dupcount)
+                    log.debug("Deleting session %s" % key)
+                    del self.sessions[key]
+                    log.debug("Session list is now %s" % self.sessions)
+                else:
+                    log.warning("Strange, session %s is not on the deletion list" % key)
+
+        self.is_running.clear()
+        self.shutdown_gracefully = self.shutdown_immediately = False
+        self.client_manager.shutdown()
+
+def run_cli(client_manager):
+    """Run the interactive CLI for netboot commands."""
+    click.echo("\n🛠  Type 'c' to see connected IPs and board info, 'f' to flash eMMC, or 'q' to quit.\n")
+    while True:
+        try:
+            user_input = input("netboot> ").strip().lower()
+            if user_input in {"q", "quit", "exit"}:
+                click.echo("🛑 Shutting down TFTP server.")
+                return True
+            elif user_input == "c":
+                client_info = client_manager.get_client_info()
+                if client_info:
+                    click.echo("🧾 TFTP client IPs and status:")
+                    for ip, info in client_info:
+                        state = info['state']
+                        filename = info['filename'] or "Unknown"
+                        timestamp = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime(info['timestamp']))
+                        board_info = info['board_info']
+                        click.echo(f"   • {ip}: {state}, Initial File: {filename}, First seen: {timestamp}")
+                        if board_info:
+                            click.echo(f"     Board Info: {board_info}")
+                else:
+                    click.echo("📭 No TFTP client requests received yet.")
+            elif user_input == "f":
+                click.echo("🔧 Initiating eMMC flash (implementation pending).")
+                flash_emmc(client_manager, emmc_image_paths)
+            elif user_input == "":
+                continue
+            else:
+                click.echo("❓ Unknown command. Try 'c' to print client list, 'f' to flash emmc, or 'q'.")
+        except (KeyboardInterrupt, EOFError):
+            click.echo("\n🛑 Exiting netboot session.")
+            return True
+
+def setup_netboot(version: str, board: str, internal: bool = False, autoflash: bool = False):
+    """
+    Download and serve a bootable image for network boot over TFTP with client monitoring.
+
+    Parameters:
+        version (str): Firmware version to download (e.g., "1.6.0").
+        board (str): Target board type, e.g., "modalix" or "davinci".
+        internal (bool): Whether to use internal download sources. Defaults to False.
+        autoflash (bool): Whether to automatically flash the devkit when networked booted. Defaults to False.
+
+    Raises:
+        RuntimeError: If the download or TFTP setup fails.
+    """
+    global emmc_image_paths
+
+    if platform.system() == "Windows":
+        click.echo("❌ Netboot with built-in TFTP is not supported on Windows. Use macOS or Linux.")
+        exit(1)
+
+    try:
+        click.echo(f"⬇️ Downloading netboot image for version: {version}, board: {board}")
+        file_list = download_image(version, board, swtype="yocto", internal=internal, update_type='netboot')
+        if not isinstance(file_list, list):
+            raise ValueError("Expected list of extracted files, got something else.")
+        extract_dir = os.path.dirname(file_list[0])
+        click.echo(f"📁 Image extracted to: {extract_dir}")
+        # Extract specific image paths
+        wic_gz_file = next((f for f in file_list if f.endswith(".wic.gz")), None)
+        bmap_file = next((f for f in file_list if f.endswith(".wic.bmap")), None)
+        emmc_image_paths = [p for p in [wic_gz_file, bmap_file] if p]
+        click.echo(f"📁 eMMC image paths are: {emmc_image_paths}")
+
+    except Exception as e:
+        raise RuntimeError(f"❌ Failed to download and extract netboot image: {e}")
+
+    try:
+        click.echo(f"🚀 Starting TFTP server in: {extract_dir}")
+        ip_candidates = get_local_ip_candidates()
+        if not ip_candidates:
+            click.echo("❌ No suitable local IP addresses found.")
+            exit(1)
+
+        click.echo("🌐 TFTP server is listening on these interfaces (UDP port 69):")
+        for iface, ip in ip_candidates:
+            click.echo(f"   🔹 {iface}: {ip}")
+
+        client_manager = ClientManager()
+
+        server = InteractiveTftpServer(tftproot=extract_dir, client_manager=client_manager)
+        server_thread = threading.Thread(target=server.listen, args=('0.0.0.0', 69), daemon=True)
+        server_thread.start()
+
+        if run_cli(client_manager):
+            server.stop(now=True)
+            client_manager.shutdown()
+
+    except PermissionError:
+        raise RuntimeError("❌ Permission denied. You must run this command with sudo to bind to port 69.")
+    except OSError as e:
+        raise RuntimeError(f"❌ Failed to start TFTP server: {e}")

sima_cli/update/remote.py

@@ -84,7 +84,7 @@ def get_remote_board_info(ip: str, passwd: str = DEFAULT_PASSWORD) -> Tuple[str,
         return board_type, build_version
 
     except Exception as e:
-        click.echo(f"Unable to retrieve board info {e}")
+        click.echo(f"Unable to retrieve board info with error: {e}, board may be still booting.")
         return "", ""
 
 
@@ -95,7 +95,7 @@ def _scp_file(sftp, local_path: str, remote_path: str):
     sftp.put(local_path, remote_path)
     click.echo("✅ Upload complete")
 
-def _run_remote_command(ssh, command: str, password: str = DEFAULT_PASSWORD):
+def run_remote_command(ssh, command: str, password: str = DEFAULT_PASSWORD):
     """
     Run a remote command over SSH and stream its output live to the console.
     If the command starts with 'sudo', pipe in the password.
@@ -140,6 +140,13 @@ def _run_remote_command(ssh, command: str, password: str = DEFAULT_PASSWORD):
     for line in remaining_err.splitlines():
         click.echo(f"⚠️ {line}")
 
+def init_ssh_session(ip: str, password: str = DEFAULT_PASSWORD):
+    ssh = paramiko.SSHClient()
+    ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
+
+    ssh.connect(ip, username=DEFAULT_USER, password=password, timeout=10)
+    return ssh
+
 def reboot_remote_board(ip: str, passwd: str):
     """
     Reboot remote board by sending SSH command
@@ -150,12 +157,36 @@ def reboot_remote_board(ip: str, passwd: str):
 
         ssh.connect(ip, username=DEFAULT_USER, password=passwd, timeout=10)
 
-        _run_remote_command(ssh, "sudo systemctl stop watchdog", password=passwd)
-        _run_remote_command(ssh, "sudo bash -c 'echo b > /proc/sysrq-trigger'", password=passwd)
+        run_remote_command(ssh, "sudo systemctl stop watchdog", password=passwd)
+        run_remote_command(ssh, "sudo bash -c 'echo b > /proc/sysrq-trigger'", password=passwd)
 
     except Exception as reboot_err:
         click.echo(f"⚠️  Unable to connect to the remote board")
 
+
+def copy_file_to_remote_board(ip: str, file_path: str, remote_dir: str, passwd: str):
+    """
+    Copy a file to the remote board over SSH.
+    Assumes default credentials: sima / edgeai.
+    """
+    ssh = paramiko.SSHClient()
+    ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
+
+    try:
+        ssh.connect(ip, username=DEFAULT_USER, password=passwd, timeout=10)
+        sftp = ssh.open_sftp()
+
+        # Upload the file
+        base_file_path = os.path.basename(file_path)
+        click.echo(f"📤 Uploading {file_path} → {remote_dir}")
+        sftp.put(file_path, os.path.join(remote_dir, base_file_path))
+        return True
+
+    except Exception as e:
+        click.echo(f"❌ Remote file copy failed: {e}")
+
+    return False
+
 def push_and_update_remote_board(ip: str, troot_path: str, palette_path: str, passwd: str, reboot_and_wait: bool):
     """
     Upload and install firmware images to remote board over SSH.
@@ -177,7 +208,7 @@ def push_and_update_remote_board(ip: str, troot_path: str, palette_path: str, pa
         click.echo("🚀 Uploaded tRoot image.")
 
         # Run tRoot update
-        _run_remote_command(
+        run_remote_command(
             ssh,
             f"sudo swupdate -H simaai-image-troot:1.0 -i /tmp/{troot_name}", password=passwd
         )
@@ -192,7 +223,7 @@ def push_and_update_remote_board(ip: str, troot_path: str, palette_path: str, pa
         click.echo("🚀 Uploaded system image.")
 
         # Run Palette update
-        _run_remote_command(
+        run_remote_command(
             ssh,
             f"sudo swupdate -H simaai-image-palette:1.0 -i /tmp/{palette_name}",
             password=passwd
@@ -206,11 +237,12 @@ def push_and_update_remote_board(ip: str, troot_path: str, palette_path: str, pa
             click.echo("🔁 Rebooting board after update. Waiting for reconnection...")
 
             try:
-                _run_remote_command(ssh, "sudo systemctl stop watchdog", password=passwd)
-                _run_remote_command(ssh, "sudo bash -c 'echo b > /proc/sysrq-trigger'", password=passwd)
+                run_remote_command(ssh, "sudo systemctl stop watchdog", password=passwd)
+                run_remote_command(ssh, "sudo bash -c 'echo b > /proc/sysrq-trigger'", password=passwd)
 
             except Exception as reboot_err:
                 click.echo(f"⚠️  SSH connection lost due to reboot (expected): {reboot_err}, please powercycle the board...")
+                click.confirm("⚠️  Have you powercycled the board?", default=True, abort=True)
 
             try:
                 ssh.close()
@@ -229,7 +261,7 @@ def push_and_update_remote_board(ip: str, troot_path: str, palette_path: str, pa
                 ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
                 ssh.connect(ip, username=DEFAULT_USER, password=passwd, timeout=10)
 
-                _run_remote_command(ssh, "grep SIMA_BUILD_VERSION /etc/build 2>/dev/null || grep SIMA_BUILD_VERSION /etc/buildinfo 2>/dev/null", password=passwd)
+                run_remote_command(ssh, "grep SIMA_BUILD_VERSION /etc/build 2>/dev/null || grep SIMA_BUILD_VERSION /etc/buildinfo 2>/dev/null", password=passwd)
                 ssh.close()
             except Exception as e:
                 click.echo(f"❌ Unable to validate the version: {e}")

sima_cli/update/updater.py

@@ -144,7 +144,18 @@ def _extract_required_files(tar_path: str, board: str, update_type: str = 'stand
             f"simaai-image-palette-{board}.wic.gz",
             f"simaai-image-palette-{board}.wic.bmap"
         }
-
+    elif update_type == 'netboot':
+        target_filenames = {
+            "Image",
+            "netboot.scr.uimg",
+            f"{board}-hhhl.dtb",
+            f"{board}-som.dtb",
+            f"{board}-dvt.dtb",
+            f"{board}-hhhl_x16.dtb",
+            f"simaai-image-palette-{board}.wic.gz",
+            f"simaai-image-palette-{board}.wic.bmap",
+            f"simaai-image-palette-{board}.cpio.gz"
+        }
     extracted_paths = []
 
     try:

sima_cli/utils/net.py

@@ -0,0 +1,29 @@
+import psutil
+import socket
+
+def get_local_ip_candidates():
+    """
+    Return a list of IPv4 addresses on physical interfaces,
+    excluding VPN, loopback, and link-local interfaces.
+    """
+    vpn_prefixes = ("tun", "tap", "utun", "tailscale", "wg", "docker")  # WireGuard, Tailscale, etc.
+    ip_list = []
+
+    for iface_name, iface_addrs in psutil.net_if_addrs().items():
+        # Exclude VPN or tunnel interfaces
+        if iface_name.startswith(vpn_prefixes):
+            continue
+
+        for addr in iface_addrs:
+            if addr.family == socket.AF_INET:
+                ip = addr.address
+
+                # Skip loopback and link-local
+                if ip.startswith("127.") or ip.startswith("169.254."):
+                    continue
+
+                ip_list.append((iface_name, ip))
+
+    # Prioritize physical interfaces: eth0, en0, wlan0, etc.
+    ip_list.sort(key=lambda x: (not x[0].startswith(("eth", "en", "wlan", "wl")), x[0]))
+    return ip_list

{sima_cli-0.0.19.dist-info → sima_cli-0.0.20.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sima-cli
-Version: 0.0.19
+Version: 0.0.20
 Summary: CLI tool for SiMa Developer Portal to download models, firmware, and apps.
 Home-page: https://developer.sima.ai/
 Author: SiMa.ai
@@ -24,6 +24,8 @@ Requires-Dist: paramiko
 Requires-Dist: plotext
 Requires-Dist: rich
 Requires-Dist: InquirerPy
+Requires-Dist: tftpy
+Requires-Dist: psutil
 Dynamic: author
 Dynamic: license-file
 Dynamic: requires-python

{sima_cli-0.0.19.dist-info → sima_cli-0.0.20.dist-info}/RECORD

@@ -1,7 +1,7 @@
 sima_cli/__init__.py,sha256=Nb2jSg9-CX1XvSc1c21U9qQ3atINxphuNkNfmR-9P3o,332
 sima_cli/__main__.py,sha256=ehzD6AZ7zGytC2gLSvaJatxeD0jJdaEvNJvwYeGsWOg,69
-sima_cli/__version__.py,sha256=2xNp40YbbA9TjpFNNqUbe-Cjo-Xe8LvBbe2706ROs3k,49
-sima_cli/cli.py,sha256=k4w4NB6yfqjTrNBEv338yQTBWIMZ-ul7W-0vo7y01dc,11326
+sima_cli/__version__.py,sha256=U-M6HF-1PsXkez7yhqMRRsr6FObzLVRQ6Spmo4bkl30,49
+sima_cli/cli.py,sha256=b4ScjEa5Ltikry4ftZdJaNxBlSMYtmjYlodqzbRyWRk,12658
 sima_cli/app_zoo/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sima_cli/app_zoo/app.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sima_cli/auth/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -13,27 +13,31 @@ sima_cli/download/__init__.py,sha256=6y4O2FOCYFR2jdnQoVi3hRtEoZ0Gw6rydlTy1SGJ5FE
 sima_cli/download/downloader.py,sha256=pHfqcg_ujBQjds_EkcRV85M2mRYGrysoZaiR-FIrpf4,5161
 sima_cli/install/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sima_cli/install/hostdriver.py,sha256=kAWDLebs60mbWIyTbUxmNrChcKW1uD5r7FtWNSUVUE4,5852
-sima_cli/install/optiview.py,sha256=VT8C1_EX-DW-n7DuxI9d0vPb87JBX-Rg9Bw9JHh1aeo,3108
+sima_cli/install/optiview.py,sha256=i5eWVor-9MScEfrQm3Ty9OP4VpSsCgWvNh7AvYdZu7s,3365
 sima_cli/install/palette.py,sha256=uRznoHa4Mv9ZXHp6AoqknfC3RxpYNKi9Ins756Cyifk,3930
 sima_cli/mla/meminfo.py,sha256=ndc8kQJmWGEIdvNh6iIhATGdrkqM2pbddr_eHxaPNfg,1466
 sima_cli/model_zoo/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sima_cli/model_zoo/model.py,sha256=q91Nrg62j1TqwPO8HiX4nlEFCCmzNEFcyFTBVMbJm8w,9836
 sima_cli/sdk/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sima_cli/sdk/syscheck.py,sha256=h9zCULW67y4i2hqiGc-hc1ucBDShA5FAe9NxwBGq-fM,4575
+sima_cli/serial/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+sima_cli/serial/serial.py,sha256=6xRta_PzE_DmmooYq35lbK76TYpAny5SEJAdYC_3fH0,4141
 sima_cli/update/__init__.py,sha256=0P-z-rSaev40IhfJXytK3AFWv2_sdQU4Ry6ei2sEus0,66
 sima_cli/update/bmaptool.py,sha256=KrhUGShBwY4Wzz50QiuMYAxxPgEy1nz5C68G-0a4qF4,4988
-sima_cli/update/bootimg.py,sha256=iMe8b42JK0KF-udSB6Kl044XUJtNgMUmBNxyuWneAkU,13387
+sima_cli/update/bootimg.py,sha256=AOZI9bXrY8x-1gwlbIINdOUNO_naw9qteNJqeQ9_o-Y,13407
 sima_cli/update/local.py,sha256=CyUFLs5Lz5w4VyM6ip4wndKBBLz3_KZ-3scEvSiOrcg,3299
+sima_cli/update/netboot.py,sha256=xTtRf8LMuqC_Ye-m6tlv5kbwkZwexc623kRymwiLTf4,18528
 sima_cli/update/query.py,sha256=cVkUMLZkONJ2XMEwqEC-JqLVB38hOqfWM2hB2ehBK6Y,3272
-sima_cli/update/remote.py,sha256=DvcvUX1ydpyF4DwUsVXagb7My6H2MNoGiCQfGfO06iI,8939
-sima_cli/update/updater.py,sha256=SprOcvSzRSwViGd2dYTtyTQfiwdLQcqZoS3KBQpxz4s,18466
+sima_cli/update/remote.py,sha256=RXQbNCDr7d8wLJ7pdGwA6G3gzgwrfZ9l_7YNYUfGHDU,10067
+sima_cli/update/updater.py,sha256=1K87YDj_isuMtL38ZI-hwBTIRmk3WqbkE6s-J0IVp-Q,18884
 sima_cli/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sima_cli/utils/artifactory.py,sha256=6YyVpzVm8ATy7NEwT9nkWx-wptkXrvG7Wl_zDT6jmLs,2390
 sima_cli/utils/config.py,sha256=wE-cPQqY_gOqaP8t01xsRHD9tBUGk9MgBUm2GYYxI3E,1616
 sima_cli/utils/config_loader.py,sha256=7I5we1yiCai18j9R9jvhfUzAmT3OjAqVK35XSLuUw8c,2005
 sima_cli/utils/env.py,sha256=Jrb062EnpMBr1jGMotBlI2j9LEH6W1Z5Tgt6LHY7yYQ,5753
+sima_cli/utils/net.py,sha256=WVntA4CqipkNrrkA4tBVRadJft_pMcGYh4Re5xk3rqo,971
 sima_cli/utils/network.py,sha256=UvqxbqbWUczGFyO-t1SybG7Q-x9kjUVRNIn_D6APzy8,1252
-sima_cli-0.0.19.dist-info/licenses/LICENSE,sha256=a260OFuV4SsMZ6sQCkoYbtws_4o2deFtbnT9kg7Rfd4,1082
+sima_cli-0.0.20.dist-info/licenses/LICENSE,sha256=a260OFuV4SsMZ6sQCkoYbtws_4o2deFtbnT9kg7Rfd4,1082
 tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tests/test_app_zoo.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tests/test_auth.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -42,8 +46,8 @@ tests/test_download.py,sha256=t87DwxlHs26_ws9rpcHGwr_OrcRPd3hz6Zmm0vRee2U,4465
 tests/test_firmware.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tests/test_model_zoo.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tests/test_utils.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-sima_cli-0.0.19.dist-info/METADATA,sha256=7_B7RJl70GRcQOHvMQRrxhjvQhuQD9pqX3FUQvKRr1M,3631
-sima_cli-0.0.19.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-sima_cli-0.0.19.dist-info/entry_points.txt,sha256=xRYrDq1nCs6R8wEdB3c1kKuimxEjWJkHuCzArQPT0Xk,47
-sima_cli-0.0.19.dist-info/top_level.txt,sha256=FtrbAUdHNohtEPteOblArxQNwoX9_t8qJQd59fagDlc,15
-sima_cli-0.0.19.dist-info/RECORD,,
+sima_cli-0.0.20.dist-info/METADATA,sha256=0KwVrHOfciP7OQWLcpOhnarYPVPkTcffT4QSKNXqzwo,3674
+sima_cli-0.0.20.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+sima_cli-0.0.20.dist-info/entry_points.txt,sha256=xRYrDq1nCs6R8wEdB3c1kKuimxEjWJkHuCzArQPT0Xk,47
+sima_cli-0.0.20.dist-info/top_level.txt,sha256=FtrbAUdHNohtEPteOblArxQNwoX9_t8qJQd59fagDlc,15
+sima_cli-0.0.20.dist-info/RECORD,,