No WS server running in stage2, lets handle this with redirecting

ws.SendMsg to log.Println

defines/defines.go

@@ -30,6 +30,9 @@ var TrainsUrl string
 // Default kernel pkg name
 var KernelPkg string
 
+// Are we running without WS backend
+var DisableWSMsg = false
+
 // Set our default bootstrap options
 var Bootstrap = false
 var BootstrapFatal = false

main.go

@@ -183,6 +183,12 @@ func main() {
 	// Update any variable locations
 	defines.SetLocs()
 
+	// Bypass normal startup and go straight to updating
+	if defines.Stage2Flag {
+		update.StartStage2()
+		os.Exit(0)
+	}
+
 	// Capture any sigint
 	interrupt := make(chan os.Signal, 1)
 	signal.Notify(interrupt, os.Interrupt)
@@ -236,11 +242,6 @@ func main() {
 		os.Exit(0)
 	}
 
-	if defines.Stage2Flag {
-		update.StartStage2()
-		os.Exit(0)
-	}
-
 	if defines.WebsocketFlag {
 		go startws(done)
 		log.Println("Listening on", defines.WebsocketAddr)

update/update.go

@@ -937,7 +937,6 @@ func startupgrade(kernelupdate bool) {
 
 func preparestage2() {
 	log.Println("Preparing to start update...")
-	logger.LogToFile("Preparing to start stage2 update....")
 
 	// Need to ensure ZFS is all mounted and ready
 	cmd := exec.Command("mount", "-u", "rw", "/")
@@ -990,6 +989,9 @@ func preparestage2() {
 
 func StartStage2() {
 
+	// No WS server to talk to
+	defines.DisableWSMsg = true
+
 	preparestage2()
 
 	doupdatefilemnt("")

ws/ws.go

@@ -14,6 +14,10 @@ type JSONReply struct {
 }
 
 func SendMsg(msg string, msg_type ...string) {
+	if defines.DisableWSMsg {
+		log.Println(msg)
+		return
+	}
 	m_type := "info"
 
 	if len(msg_type) > 0 {