Convert to flags

Author: facchinm

main.go

@@ -2,78 +2,76 @@ package main
 
 import (
 	"bufio"
+	"flag"
 	"fmt"
+	"github.com/kardianos/osext"
 	"github.com/mattn/go-shellwords"
 	"io"
 	"io/ioutil"
 	"os"
 	"os/exec"
 	"path/filepath"
 	"runtime"
+	"strconv"
 	"strings"
 	"time"
 )
 
-var verbose bool
+var (
+	verbose                = flag.Bool("v", false, "Show verbose logging")
+	quiet                  = flag.Bool("q", true, "Show quiet logging")
+	force                  = flag.Bool("f", false, "Force firmware update")
+	dfu_path               = flag.String("dfu", "", "Location of dfu-util binaries")
+	bin_file_name          = flag.String("bin", "", "Location of sketch binary")
+	com_port               = flag.String("port", "", "Upload serial port")
+	ble_compliance_string  = flag.String("ble_fw_str", "", "BLE FW ID string")
+	ble_compliance_offset  = flag.Int("ble_fw_pos", 0, "BLE FW ID offset")
+	rtos_compliance_string = flag.String("rots_fw_str", "", "RTOS FW ID string")
+	rtos_compliance_offset = flag.Int("rtos_fw_pos", 0, "RTOS FW ID offset")
+)
+
+const Version = "1.8.0"
+
+const dfu_flags = "-d,8087:0ABA"
+const rtos_firmware = "quark.bin"
+const ble_firmware = "ble_core.bin"
 
 func PrintlnVerbose(a ...interface{}) {
-	if verbose {
+	if *verbose {
 		fmt.Println(a...)
 	}
 }
 
-func main_load(args []string) {
+func main_load() {
 
-	// ARG 1: Path to binaries
-	// ARG 2: BIN File to download
-	// ARG 3: TTY port to use.
-	// ARG 4: quiet/verbose
-	// path may contain \ need to change all to /
-
-	if len(args) < 4 {
-		fmt.Println("Not enough arguments")
+	if *dfu_path == "" {
+		fmt.Println("Need to specify dfu-util location")
 		os.Exit(1)
 	}
 
-	bin_path := args[0]
-	dfu := bin_path + "/dfu-util"
-	dfu = filepath.ToSlash(dfu)
-	dfu_flags := "-d,8087:0ABA"
-
-	bin_file_name := args[1]
-
-	com_port := args[2]
-	verbosity := args[3]
-
-	ble_compliance_string := ""
-	ble_compliance_offset := ""
-	if len(args) >= 5 {
-		// Called by post 1.0.6 platform.txt
-		ble_compliance_string = args[4]
-		ble_compliance_offset = args[5]
+	if *bin_file_name == "" && *force == false {
+		fmt.Println("Need to specify a binary location or force FW update")
+		os.Exit(1)
 	}
 
-	if verbosity == "quiet" {
-		verbose = false
-	} else {
-		verbose = true
-	}
+	dfu := *dfu_path + "/dfu-util"
+	dfu = filepath.ToSlash(dfu)
 
-	PrintlnVerbose("Args to shell:", args)
-	PrintlnVerbose("Serial Port: " + com_port)
-	PrintlnVerbose("BIN FILE " + bin_file_name)
+	PrintlnVerbose("Serial Port: " + *com_port)
+	PrintlnVerbose("BIN FILE " + *bin_file_name)
 
 	counter := 0
 	board_found := false
 
 	if runtime.GOOS == "darwin" {
 		library_path := os.Getenv("DYLD_LIBRARY_PATH")
-		if !strings.Contains(library_path, bin_path) {
-			os.Setenv("DYLD_LIBRARY_PATH", bin_path+":"+library_path)
+		if !strings.Contains(library_path, *dfu_path) {
+			os.Setenv("DYLD_LIBRARY_PATH", *dfu_path+":"+library_path)
 		}
 	}
 
 	dfu_search_command := []string{dfu, dfu_flags, "-l"}
+	var err error
 
 	for counter < 100 && board_found == false {
 		if counter%10 == 0 {
@@ -98,11 +96,14 @@ func main_load(args []string) {
 	}
 
 	if board_found == false {
-		fmt.Println("ERROR: Timed out waiting for Arduino 101 on " + com_port)
+		fmt.Println("ERROR: Timed out waiting for Arduino 101 on " + *com_port)
 		os.Exit(1)
 	}
 
-	if ble_compliance_string != "" {
+	needUpdateRTOS := false
+	needUpdateBLE := false
+
+	if *ble_compliance_string != "" {
 
 		// obtain a temporary filename
 		tmpfile, _ := ioutil.TempFile(os.TempDir(), "dfu")
@@ -121,7 +122,7 @@ func main_load(args []string) {
 		os.Remove(tmpfile.Name())
 
 		// download a piece of BLE firmware
-		dfu_ble_dump_command := []string{dfu, dfu_flags, "-U", tmpfile.Name(), "--alt", "8", "-K", ble_compliance_offset}
+		dfu_ble_dump_command := []string{dfu, dfu_flags, "-U", tmpfile.Name(), "--alt", "8", "-K", strconv.Itoa(*ble_compliance_offset)}
 
 		err, _, _ = launchCommandAndWaitForOutput(dfu_ble_dump_command, "", false)
 		if err != nil {
@@ -130,29 +131,115 @@ func main_load(args []string) {
 		}
 
 		// check for BLE library compliance
-		PrintlnVerbose("Verifying BLE version:", ble_compliance_string)
-		found := searchBLEversionInDFU(tmpfile.Name(), ble_compliance_string)
+		PrintlnVerbose("Verifying BLE version:", *ble_compliance_string)
+		found := searchVersionInDFU(tmpfile.Name(), *ble_compliance_string)
+
+		// remove the temporary file
+		os.Remove(tmpfile.Name())
+
+		if !found {
+			needUpdateBLE = true
+		} else {
+			PrintlnVerbose("BLE version: verified")
+		}
+	}
+
+	if *rtos_compliance_string != "" {
+
+		// obtain a temporary filename
+		tmpfile, _ := ioutil.TempFile(os.TempDir(), "dfu")
+		tmpfile.Close()
+		os.Remove(tmpfile.Name())
+
+		// reset DFU interface counter
+		dfu_reset_command := []string{dfu, dfu_flags, "-U", tmpfile.Name(), "--alt", "2", "-K", "1"}
+
+		err, _, _ := launchCommandAndWaitForOutput(dfu_reset_command, "", false)
+		if err != nil {
+			fmt.Println(err)
+			os.Exit(1)
+		}
+
+		os.Remove(tmpfile.Name())
+
+		// download a piece of RTOS firmware
+		dfu_rtos_dump_command := []string{dfu, dfu_flags, "-U", tmpfile.Name(), "--alt", "2", "-K", strconv.Itoa(*rtos_compliance_offset)}
+
+		err, _, _ = launchCommandAndWaitForOutput(dfu_rtos_dump_command, "", false)
+		if err != nil {
+			fmt.Println(err)
+			os.Exit(1)
+		}
+
+		// check for BLE library compliance
+		PrintlnVerbose("Verifying RTOS version:", *rtos_compliance_string)
+		found := searchVersionInDFU(tmpfile.Name(), *rtos_compliance_string)
 
 		// remove the temporary file
 		os.Remove(tmpfile.Name())
 
 		if !found {
-			fmt.Println("!! BLE firmware version is not in sync with CurieBLE library !!")
-			fmt.Println("* Set Programmer to \"Arduino/Genuino 101 Firmware Updater\"")
-			fmt.Println("* Update it using \"Burn Bootloader\" menu")
+			needUpdateRTOS = true
+		} else {
+			PrintlnVerbose("RTOS version: verified")
+		}
+	}
+
+	executablePath, _ := osext.ExecutableFolder()
+	firmwarePath := executablePath + "/firmwares/"
+
+	// Save verbose flag
+	verbose_user := *verbose
+
+	if needUpdateBLE || *force == true {
+
+		*verbose = true
+		// flash current BLE firmware to partition 8
+		dfu_ble_flash_command := []string{dfu, dfu_flags, "-D", firmwarePath + ble_firmware, "--alt", "8"}
+
+		fmt.Println("ATTENTION: BLE firmware is being flashed")
+		fmt.Println("DO NOT DISCONNECT THE BOARD")
+
+		err, _, _ = launchCommandAndWaitForOutput(dfu_ble_flash_command, "", true)
+		if err != nil {
+			fmt.Println(err)
+			os.Exit(1)
+		}
+	}
+
+	if needUpdateRTOS || *force == true {
+
+		*verbose = true
+		// flash current RTOS firmware to partition 2
+		dfu_rtos_flash_command := []string{dfu, dfu_flags, "-D", firmwarePath + rtos_firmware, "--alt", "2"}
+
+		fmt.Println("ATTENTION: RTOS firmware is being flashed")
+		fmt.Println("DO NOT DISCONNECT THE BOARD")
+
+		err, _, _ = launchCommandAndWaitForOutput(dfu_rtos_flash_command, "", true)
+		if err != nil {
+			fmt.Println(err)
 			os.Exit(1)
 		}
-		PrintlnVerbose("BLE version: verified")
 	}
 
-	dfu_download := []string{dfu, dfu_flags, "-D", bin_file_name, "-v", "--alt", "7", "-R"}
-	err, _, _ := launchCommandAndWaitForOutput(dfu_download, "", true)
+	// Restore verbose flag
+	*verbose = verbose_user
+
+	// Finally flash the sketch
+
+	if *bin_file_name == "" {
+		os.Exit(0)
+	}
+
+	dfu_download := []string{dfu, dfu_flags, "-D", *bin_file_name, "-v", "--alt", "7", "-R"}
+	err, _, _ = launchCommandAndWaitForOutput(dfu_download, "", true)
 
 	if err == nil {
 		fmt.Println("SUCCESS: Sketch will execute in about 5 seconds.")
 		os.Exit(0)
 	} else {
-		fmt.Println("ERROR: Upload failed on " + com_port)
+		fmt.Println("ERROR: Upload failed on " + *com_port)
 		os.Exit(1)
 	}
 }
@@ -164,7 +251,7 @@ func main_debug(args []string) {
 		os.Exit(1)
 	}
 
-	verbose = true
+	*verbose = true
 
 	type Command struct {
 		command    string
@@ -205,24 +292,27 @@ func main_debug(args []string) {
 }
 
 func main() {
-	name := os.Args[0]
-	args := os.Args[1:]
+	name := filepath.Base(os.Args[0])
+
+	flag.Parse()
+
+	PrintlnVerbose(name + " " + Version + " - compiled with " + runtime.Version())
 
 	if strings.Contains(name, "load") {
 		fmt.Println("Starting download script...")
-		main_load(args)
+		main_load()
 	}
 
 	if strings.Contains(name, "debug") {
 		fmt.Println("Starting debug script...")
-		main_debug(args)
+		main_debug(os.Args[1:])
 	}
 
 	fmt.Println("Wrong executable name")
 	os.Exit(1)
 }
 
-func searchBLEversionInDFU(file string, string_to_search string) bool {
+func searchVersionInDFU(file string, string_to_search string) bool {
 	read, _ := ioutil.ReadFile(file)
 	return strings.Contains(string(read), string_to_search)
 }