create_update, arp-ip.sh, sue-flash

Author: timon-sue

script/arp-compare/arp-compare.py

@@ -0,0 +1,64 @@
+import subprocess
+import curses
+import sys
+
+def run_arp_scan(interface):
+    cmd = ["arp-scan", "-l", "-I", interface]
+    output = subprocess.check_output(cmd, universal_newlines=True)
+    devices = parse_arp_scan_output(output)
+    return devices
+
+def parse_arp_scan_output(output):
+    devices = {}
+    lines = output.strip().split('\n')
+    for line in lines:
+        if line.startswith('10.1'):  # Modify this to match your network IP range
+            parts = line.split('\t')
+            ip = parts[0]
+            mac = parts[1]
+            name = parts[2] if len(parts) >= 3 else "Unknown"
+            devices[mac] = (ip, name)
+    return devices
+
+def compare_scans(scan1, scan2):
+    devices_only_in_scan1 = {mac: scan1[mac] for mac in scan1 if mac not in scan2}
+    devices_only_in_scan2 = {mac: scan2[mac] for mac in scan2 if mac not in scan1}
+    return devices_only_in_scan1, devices_only_in_scan2
+
+def wait_for_keypress():
+    stdscr = curses.initscr()
+    stdscr.addstr("Press any key to continue...")
+    stdscr.refresh()
+    stdscr.getch()
+    curses.endwin()
+
+def print_devices(devices):
+    for mac, (ip, name) in devices.items():
+        print(f"IP: {ip}\tMAC: {mac}\tName: {name}")
+
+def main():
+    if len(sys.argv) < 2:
+        print("Usage: python arp_scan.py <interface>")
+        return
+
+    interface = sys.argv[1]
+    print("Running first ARP scan...")
+    scan1 = run_arp_scan(interface)
+    print("First ARP scan completed.\n")
+
+    wait_for_keypress()
+
+    print("Running second ARP scan...")
+    scan2 = run_arp_scan(interface)
+    print("Second ARP scan completed.\n")
+
+    devices_only_in_scan1, devices_only_in_scan2 = compare_scans(scan1, scan2)
+
+    print("Devices only in first scan:")
+    print_devices(devices_only_in_scan1)
+    print("\nDevices only in second scan:")
+    print_devices(devices_only_in_scan2)
+
+if __name__ == "__main__":
+    main()
+

script/arp-ip.sh

@@ -0,0 +1,29 @@
+#!/bin/bash
+
+interface=$1
+
+if [[ -z "$interface" ]]; then
+  # Use the ifconfig command to get network interface information
+  ifconfig_output=$(ifconfig)
+
+  # Use grep and awk to extract the name of the active interface
+  interface=$(ip -o link show | awk '$2 != "lo:" && $9 == "UP" {print $2}' | cut -d ":" -f 1)
+
+  if [[ -z "$interface" ]]; then
+    echo "Error: failed to auto-detect active network interface."
+    exit 1
+  fi
+fi
+
+# Scan the local network interface for devices
+devices="`sudo arp-scan --interface=$interface --localnet | sed '1,2d'`"
+selected="`(echo "$devices" | fzf)`"
+
+if [[ -z "$selected" ]]; then
+  exit 1
+fi
+
+# Get the IP address of the selected device
+ip=$(sudo arp-scan --interface=$interface --localnet | grep "$selected" | awk '{print $1}')
+echo $ip
+

script/create_update_json.sh

@@ -0,0 +1,78 @@
+#!/bin/bash
+
+# Define the file name
+file="image.swu"
+url=""
+json_file="info.json"
+forced_update=false
+
+# Parse command line arguments using a loop
+while [[ $# -gt 0 ]]; do
+    case "$1" in
+        --file=*)
+            file="${1#*=}"
+            ;;
+        --url=*)
+            url="${1#*=}"
+            ;;
+        --force=*)
+            forced_update="${1#*=}"
+            if [[ "$forced_update" != "true" && "$forced_update" != "false" ]]; then
+                echo "Error: Invalid argument for --force. Please use 'true' or 'false'."
+                exit 1
+            fi
+            ;;
+        --mac=*)
+            mac="${1#*=}"
+            ;;
+        *)
+            echo "Error: Invalid argument: $1"
+            exit 1
+            ;;
+    esac
+    shift
+done
+
+# Check if the --url argument is provided and not empty
+if [ -z "$url" ]; then
+    echo "Error: Missing or empty --url argument. Please provide a valid URL."
+    exit 1
+fi
+
+# Use sed to extract the third line of the file
+third_line=$(sed -n '3p' "$file")
+
+# Use grep and regular expressions to extract the version variable
+if [[ $third_line =~ version\ =\ \"([^\"]+)\" ]]; then
+    version="${BASH_REMATCH[1]}"
+    echo "Found version in $file: $version"
+else
+    echo "Error: Unable to extract version from the third line of $file."
+    exit 1
+fi
+
+# Use stat to get the size of the file
+file_size=$(stat -c %s "$file")
+echo "File size of $file is $file_size bytes."
+
+# If a MAC address is provided, append it to the file name (after extracting the info)
+if [ -n "$mac" ]; then
+    json_file="info.json_mac_$mac"
+fi
+
+# Generate the JSON content
+json_content=$(cat <<EOF
+{
+    "version": "$version",
+    "url": "$url",
+    "imageSize": $file_size,
+    "forcedUpdate": $forced_update
+}
+EOF
+)
+
+# Write the JSON content to the JSON file
+echo "$json_content" > "$json_file"
+
+echo "JSON file '$json_file' created with the following content:"
+cat "$json_file"

script/sue-flash.sh

@@ -0,0 +1,67 @@
+#!/bin/bash
+
+# Function to check if a string is a valid IP address
+is_valid_ip() {
+  local ip="$1"
+  if [[ "$ip" =~ ^[0-9]+\.[0-9]+\.[0-9]+\.[0-9]+$ ]]; then
+    # Split the IP address into octets
+    IFS='.' read -ra octets <<< "$ip"
+    for octet in "${octets[@]}"; do
+      if ((octet < 0 || octet > 255)); then
+        return 1  # Invalid octet value
+      fi
+    done
+    return 0  # Valid IP address
+  else
+    return 1  # Invalid format
+  fi
+}
+
+# Check if fzf is installed
+if ! command -v fzf &>/dev/null; then
+    echo "Error: fzf is not installed. Please install fzf."
+    exit 1
+fi
+
+# Function to find .swu files
+find_swu_file() {
+    local selected_file
+    selected_file=$(find . -type l -name "image.swu" 2>/dev/null | fzf --prompt "Select .swu file: ")
+    
+    if [ -z "$selected_file" ]; then
+        return 1
+    fi
+    
+    echo "$selected_file"
+    return 0
+}
+
+# Attempt to find .swu file in the current directory or its subdirectories
+if selected_file=$(find_swu_file); then
+    echo "Selected file: $selected_file"
+fi
+
+# arg can be ip or interface for discovery
+if is_valid_ip "$1"; then
+    echo "valid ip given, using"
+    device_ip="$1"
+else
+    echo "no valid ip, assuming it's an interface"
+    device_ip=$(arp-ip $1) # can be empty for interface auto-try
+fi
+
+# SSH to the device and run fwup
+ssh root@$device_ip "nsdk_cli invoke firmwareupdate:startLocalUpdate"
+
+# Wait for the device to reboot (you might need to customize this)
+sleep 15
+
+# Use curl to simulate the file upload through the web interface
+curl -F "image=@$selected_file" http://$device_ip/handle_post_request
+
+# You might need to customize the URL and form field names depending on your device's web interface
+
+# Add additional logic to verify the update process if needed
+# For example, you can check the device's status page or log files
+
+echo "Firmware update completed."