System Version: {{.Version}}
+{{.Message}}
+ + + +`)) + +func runHttp() { + gin.SetMode(gin.ReleaseMode) + + g := gin.Default() + + r := multitemplate.NewRenderer() + r.Add("index.html", indexPage) + r.Add("reboot.html", rebootPage) + + g.HTMLRender = r + + g.GET("/", renderIndex) + + g.POST("/address", handleAddress) + g.POST("/system", handleSystem) + g.POST("/application", handleApplication) + + g.GET("/reboot", handleReboot) + g.GET("/restore", handleRestore) + + g.Run(":8988") +} + +func handleAddress(c *gin.Context) { + mu.Lock() + defer mu.Unlock() + + addr := c.PostForm("address") + addr2 := c.PostForm("address2") + + if err := setIpAddress(addr, addr2); err != nil { + c.String(http.StatusBadRequest, "address: setIpAddress: %s", err.Error()) + return + } + + c.HTML(http.StatusOK, "reboot.html", gin.H{"Message": "IP address is updated."}) +} + +func handleSystem(c *gin.Context) { + mu.Lock() + defer mu.Unlock() + + if err := saveFormFile(c, "image", "/tmp/system.zip"); err != nil { + c.String(http.StatusBadRequest, "system: saveFormFile: %s", err.Error()) + return + } + + if err := updateSystem(); err != nil { + c.String(http.StatusBadRequest, "system: updateSystem: %s", err.Error()) + return + } + + c.HTML(http.StatusOK, "reboot.html", gin.H{"Message": "System is updated."}) +} + +func handleApplication(c *gin.Context) { + mu.Lock() + defer mu.Unlock() + + if err := saveFormFile(c, "image", "/tmp/application.zip"); err != nil { + c.String(http.StatusBadRequest, "application: saveFormFile: %s", err.Error()) + return + } + + if err := updateApp(); err != nil { + c.String(http.StatusBadRequest, "application: updateApplication: %s", err.Error()) + return + } + + c.HTML(http.StatusOK, "reboot.html", gin.H{"Message": "Application is updated."}) +} + +func handleReboot(c *gin.Context) { + reboot() + + c.String(http.StatusOK, "OK") +} + +func handleRestore(c *gin.Context) { + mu.Lock() + defer mu.Unlock() + + restore() + + c.HTML(http.StatusOK, "reboot.html", gin.H{"Message": "Restoration is done."}) +} + +func renderIndex(c *gin.Context) { + device := "System" + if len(os.Args) > 1 { + device = os.Args[1] + } + + c.HTML(http.StatusOK, "index.html", gin.H{ + "Device": device, + "IP": getIpAddress(), + "IP2": getIpAddress2(), + "Version": getVersion(), + }) +} + +func saveFormFile(c *gin.Context, name string, dst string) error { + file, err := c.FormFile(name) + if err != nil { + return err + } + + return c.SaveUploadedFile(file, dst) +} diff --git a/go/src/themaru/ip.go b/go/src/themaru/ip.go new file mode 100644 index 0000000..a3f09bd --- /dev/null +++ b/go/src/themaru/ip.go @@ -0,0 +1,77 @@ +package main + +import ( + "errors" + "regexp" + "strconv" +) + +func getIpAddress() string { + env, err := getEnv(0) + if err != nil { + return "" + } + + return env["ipaddr"] +} + +func getIpAddress2() string { + env, err := getEnv(0) + if err != nil { + return "" + } + + return env["ipaddr2"] +} + +func setIpAddress(address, address2 string) error { + address = extractIpAddress(address) + if address == "" { + return errors.New("invalid IP address") + } + + address2 = extractIpAddress(address2) + if address2 == "" { + return errors.New("invalid IP address2") + } + + env, err := getEnv(0) + if err != nil { + return err + } + + env["ipaddr"] = address + env["ipaddr2"] = address2 + + err = setEnv(0, env) + if err != nil { + return err + } + + return nil +} + +func extractIpAddress(address string) string { + r := regexp.MustCompile(`([0-9]+)\.([0-9]+)\.([0-9]+)\.([0-9]+)`) + s := r.FindStringSubmatch(address) + if len(s) != 5 { + return "" + } + + for i, v := range s { + if i == 0 { + continue + } + + u, err := strconv.ParseUint(v, 10, 8) + if err != nil { + return "" + } + + if u > 255 { + return "" + } + } + + return s[0] +} diff --git a/go/src/themaru/main.go b/go/src/themaru/main.go new file mode 100644 index 0000000..241ff1a --- /dev/null +++ b/go/src/themaru/main.go @@ -0,0 +1,5 @@ +package main + +func main() { + runHttp() +} diff --git a/go/src/themaru/system.go b/go/src/themaru/system.go new file mode 100644 index 0000000..abdc75c --- /dev/null +++ b/go/src/themaru/system.go @@ -0,0 +1,114 @@ +package main + +import ( + "io/ioutil" + "os/exec" +) + +func getVersion() string { + env, err := getEnv(1) + if err != nil { + return "" + } + + version, exist := env["falinux_version"] + if !exist { + return "" + } + + return version +} + +func updateSystem() error { + env, err := getEnv(1) + if err != nil { + return err + } + + bootpart := env["bootpart"] + + var dir string + var partname string + switch bootpart { + case "1": + dir = "/boot1" + partname = "/dev/mmcblk3p2" + env["bootpart"] = "2" + case "2": + dir = "/boot0" + partname = "/dev/mmcblk3p1" + env["bootpart"] = "1" + default: + dir = "/boot0" + partname = "/dev/mmcblk3p1" + env["bootpart"] = "1" + } + + if !partitionIsExists() { + makePartition() + } else { + formatPartition(partname) + } + + if err := exec.Command("mount", "-o", "rw", dir).Run(); err != nil { + return err + } + defer exec.Command("umount", dir).Run() + + if err := exec.Command("unzip", "-o", "/tmp/system.zip", "-d", dir).Run(); err != nil { + return err + } + + exec.Command("sync").Run() + + if ver, err := ioutil.ReadFile(dir + "/version"); err != nil { + env["falinux_version"] = "" + } else { + env["falinux_version"] = string(ver) + } + + return setEnv(1, env) +} + +func updateApp() error { + const cmdline = `\ + rm -rf /app/bin.old /app/bin.new; \ + mkdir /app/bin.new && \ + unzip -o /tmp/application.zip -d /app/bin.new && \ + sync && \ + mv /app/bin /app/bin.old; \ + mv /app/bin.new /app/bin; \ + sync` + + return exec.Command("/bin/sh", "-c", cmdline).Run() +} + +func partitionIsExists() bool { + err := exec.Command("test", "-e", "/dev/mmcblk3p1", "-a", "-e", "/dev/mmcblk3p2").Run() + return err == nil +} + +func makePartition() { + exec.Command("umount", "/app").Run() + exec.Command("/bin/sh", "-c", "/root/.falinux/mk-mmc-part.sh").Run() +} + +func formatPartition(partname string) { + exec.Command("mkfs.fat", partname).Run() +} + +func reboot() { + exec.Command("killall", "-19", "watchdog").Run() + exec.Command("reboot").Run() +} + +func restore() { + m := make(map[string]string) + + setEnv(0, m) + setEnv(1, m) + setEnv(2, m) + setEnv(3, m) + + exec.Command("/bin/sh", "-c", "/root/.falinux/mk-mmc-part.sh").Run() +} diff --git a/go/src/themaru/upx b/go/src/themaru/upx new file mode 100755 index 0000000000000000000000000000000000000000..1c53e841ff1ca756edab1220104ecf98c65682b3 GIT binary patch literal 422160 zcmZ^~Q>-vRvn{x7+qP}nwr$(C@on3-ZQHhOYyN{fx$`icR93HARjGdJPAXmgqB3GY zfPeu1X8{550sI$3&=vb{m%#rK{x1{&-~a#y_#frJVnP2e4d8$M{|EK#!TwwQKL`Lo z@ZbEu^8QEpZwdh5@IUqbU;JM;LmkKe>=6D36i-}C{y%yD8~6VQ|C