Dump BLE image and verify firmware version (Linux32)

Signed-off-by: Brian Baltz <brian.a.baltz@intel.com>

Author: Brian Baltz

arduino101load/arduino101load

@@ -3,8 +3,9 @@ package main
 import (
 	"bufio"
 	"fmt"
-	"github.com/codeskyblue/go-sh"
+	"github.com/mattn/go-shellwords"
 	"io"
+	"io/ioutil"
 	"os"
 	"os/exec"
 	"path/filepath"
@@ -21,16 +22,18 @@ func PrintlnVerbose(a ...interface{}) {
 	}
 }
 
-func main() {
-	fmt.Println("Starting download script...")
+func main_load(args []string) {
 
 	// 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 /
 
-	args := os.Args[1:]
+	if len(args) < 4 {
+		fmt.Println("Not enough arguments")
+		os.Exit(1)
+	}
 
 	bin_path := args[0]
 	dfu := bin_path + "/dfu-util"
@@ -42,6 +45,14 @@ func main() {
 	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 verbosity == "quiet" {
 		verbose = false
 	} else {
@@ -62,54 +73,81 @@ func main() {
 		}
 	}
 
-	for counter < 10 && board_found == false {
-		PrintlnVerbose("Waiting for device...")
-		out, err := sh.Command(dfu, dfu_flags, "-l").Output()
+	dfu_search_command := []string{dfu, dfu_flags, "-l"}
+
+	for counter < 100 && board_found == false {
+		if counter%10 == 0 {
+			PrintlnVerbose("Waiting for device...")
+		}
+		err, found, _ := launchCommandAndWaitForOutput(dfu_search_command, "sensor_core", false)
 		if err != nil {
 			fmt.Println(err)
 			os.Exit(1)
 		}
-		if counter == 4 {
+		if counter == 40 {
 			fmt.Println("Flashing is taking longer than expected")
 			fmt.Println("Try pressing MASTER_RESET button")
 		}
-		if strings.Contains(string(out), "sensor_core") {
+		if found == true {
 			board_found = true
 			PrintlnVerbose("Device found!")
 			break
 		}
-		time.Sleep(1000 * time.Millisecond)
+		time.Sleep(100 * time.Millisecond)
 		counter++
 	}
 
 	if board_found == false {
-		fmt.Println("ERROR: Device is not responding.")
+		fmt.Println("ERROR: Timed out waiting for Arduino 101 on " + com_port)
 		os.Exit(1)
 	}
 
-	dfu_download := []string{dfu, dfu_flags, "-D", bin_file_name, "-v", "--alt", "7", "-R"}
+	if ble_compliance_string != "" {
 
-	oscmd := exec.Command(dfu_download[0], dfu_download[1:]...)
+		// obtain a temporary filename
+		tmpfile, _ := ioutil.TempFile(os.TempDir(), "dfu")
+		tmpfile.Close()
+		os.Remove(tmpfile.Name())
 
-	tellCommandNotToSpawnShell(oscmd)
+		// reset DFU interface counter
+		dfu_reset_command := []string{dfu, dfu_flags, "-U", tmpfile.Name(), "--alt", "8", "-K", "1"}
 
-	stdout, _ := oscmd.StdoutPipe()
+		err, _, _ := launchCommandAndWaitForOutput(dfu_reset_command, "", false)
+		if err != nil {
+			fmt.Println(err)
+			os.Exit(1)
+		}
 
-	stderr, _ := oscmd.StderrPipe()
+		os.Remove(tmpfile.Name())
 
-	multi := io.MultiReader(stderr, stdout)
+		// download a piece of BLE firmware
+		dfu_ble_dump_command := []string{dfu, dfu_flags, "-U", tmpfile.Name(), "--alt", "8", "-K", ble_compliance_offset}
 
-	err := oscmd.Start()
+		err, _, _ = launchCommandAndWaitForOutput(dfu_ble_dump_command, "", false)
+		if err != nil {
+			fmt.Println(err)
+			os.Exit(1)
+		}
 
-	in := bufio.NewScanner(multi)
+		// check for BLE library compliance
+		PrintlnVerbose("Verifying BLE version:", ble_compliance_string)
+		found := searchBLEversionInDFU(tmpfile.Name(), ble_compliance_string)
 
-	in.Split(bufio.ScanLines)
+		// 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")
+			os.Exit(1)
+		}
+		PrintlnVerbose("BLE version: verified")
 
-	for in.Scan() {
-		PrintlnVerbose(in.Text())
 	}
 
-	err = oscmd.Wait()
+	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.")
@@ -119,3 +157,106 @@ func main() {
 		os.Exit(1)
 	}
 }
+
+func main_debug(args []string) {
+
+	if len(args) < 1 {
+		fmt.Println("Not enough arguments")
+		os.Exit(1)
+	}
+
+	verbose = true
+
+	type Command struct {
+		command    string
+		background bool
+	}
+
+	var commands []Command
+
+	fullcmdline := strings.Join(args[:], "")
+	temp_commands := strings.Split(fullcmdline, ";")
+	for _, command := range temp_commands {
+		background_commands := strings.Split(command, "&")
+		for i, command := range background_commands {
+			var cmd Command
+			cmd.background = (i < len(background_commands)-1)
+			cmd.command = command
+			commands = append(commands, cmd)
+		}
+	}
+
+	var err error
+
+	for _, command := range commands {
+		fmt.Println("command: " + command.command)
+		cmd, _ := shellwords.Parse(command.command)
+		fmt.Println(cmd)
+		if command.background == false {
+			err, _, _ = launchCommandAndWaitForOutput(cmd, "", true)
+		} else {
+			err, _ = launchCommandBackground(cmd, "", true)
+		}
+		if err != nil {
+			fmt.Println("ERROR: Command \" " + command.command + " \" failed")
+			os.Exit(1)
+		}
+	}
+	os.Exit(0)
+}
+
+func main() {
+	name := os.Args[0]
+	args := os.Args[1:]
+
+	if strings.Contains(name, "load") {
+		fmt.Println("Starting download script...")
+		main_load(args)
+	}
+
+	if strings.Contains(name, "debug") {
+		fmt.Println("Starting debug script...")
+		main_debug(args)
+	}
+
+	fmt.Println("Wrong executable name")
+	os.Exit(1)
+}
+
+func searchBLEversionInDFU(file string, string_to_search string) bool {
+	read, _ := ioutil.ReadFile(file)
+	return strings.Contains(string(read), string_to_search)
+}
+
+func launchCommandAndWaitForOutput(command []string, stringToSearch string, print_output bool) (error, bool, string) {
+	oscmd := exec.Command(command[0], command[1:]...)
+	tellCommandNotToSpawnShell(oscmd)
+	stdout, _ := oscmd.StdoutPipe()
+	stderr, _ := oscmd.StderrPipe()
+	multi := io.MultiReader(stderr, stdout)
+	err := oscmd.Start()
+	in := bufio.NewScanner(multi)
+	in.Split(bufio.ScanLines)
+	found := false
+	out := ""
+	for in.Scan() {
+		if print_output {
+			PrintlnVerbose(in.Text())
+		}
+		out += in.Text() + "\n"
+		if stringToSearch != "" {
+			if strings.Contains(in.Text(), stringToSearch) {
+				found = true
+			}
+		}
+	}
+	err = oscmd.Wait()
+	return err, found, out
+}
+
+func launchCommandBackground(command []string, stringToSearch string, print_output bool) (error, bool) {
+	oscmd := exec.Command(command[0], command[1:]...)
+	tellCommandNotToSpawnShell(oscmd)
+	err := oscmd.Start()
+	return err, false
+}