Fix other

.vscode/launch.json

@@ -0,0 +1,16 @@
+{
+    // Use IntelliSense to learn about possible attributes.
+    // Hover to view descriptions of existing attributes.
+    // For more information, visit: https://go.microsoft.com/fwlink/?linkid=830387
+    "version": "0.2.0",
+    "configurations": [
+
+        {
+            "name": "Launch Package",
+            "type": "go",
+            "request": "launch",
+            "mode": "auto",
+            "program": "${workspaceFolder}"
+        }
+    ]
+}

LICENSE

@@ -0,0 +1,7 @@
+Copyright 2025 Aurélie DELHAIE
+
+Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the “Software”), to deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED “AS IS”, WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

README.md

@@ -0,0 +1,9 @@
+# Win2Linux
+
+The software is still in alpha.
+
+Small script in Go that allows you to reboot Windows to another operating system registered in the UEFI database.
+
+![image](./img/main_screenshot.png)
+
+Inspiriation from "Restart To" gnome extension (https://github.com/tiagoporsch/restartto)

main.go

@@ -1,27 +1,21 @@
 package main
 
 import (
+	"Win2Linux/pkg/bcdedit"
+	"Win2Linux/pkg/windows"
 	_ "embed"
 	"fmt"
 	"os"
-	"os/exec"
-	"regexp"
-	"strings"
+
+	"Win2Linux/pkg/errors"
 
 	"github.com/getlantern/systray"
 )
 
 //go:generate go-winres make
 
-type (
-	Entry struct {
-		Key        string
-		Attributes map[string]string
-	}
-)
-
-var (
-	r *regexp.Regexp = regexp.MustCompile(`-+`)
+const (
+	GeneralFailure int32 = 0x01
 )
 
 //go:embed tray.ico
@@ -34,10 +28,17 @@ func main() {
 }
 
 func onReady() {
-	entries := list()
 	systray.SetTitle("Win2Linux")
 	systray.SetTooltip("Switch from Windows to Linux")
 	systray.SetIcon(icon)
+	entries, err := bcdedit.List()
+	if err != nil {
+		if wErr, ok := err.(*errors.WinError); ok {
+			windows.Fatal("Win2Linux", fmt.Sprintf("error: %s", err), int(wErr.ExitCode()))
+		} else {
+			windows.Fatal("Win2Linux", fmt.Sprintf("error: %s", err), 1)
+		}
+	}
 
 	for _, entry := range entries {
 		fmt.Println(entry.Attributes)
@@ -48,11 +49,29 @@ func onReady() {
 		}
 		mCustom := systray.AddMenuItem(key, "Switch to"+key)
 		go func() {
+			for {
 				<-mCustom.ClickedCh
-			fmt.Println(uuid)
-			reboot(uuid)
+				if uuid == "{fwbootmgr}" {
+					err := bcdedit.RebootToFirmware()
+					if err != nil {
+						if wErr, ok := err.(*errors.WinError); ok {
+							windows.Fatal("Win2Linux", fmt.Sprintf("error: %s", err), int(wErr.ExitCode()))
+						} else {
+							windows.Fatal("Win2Linux", fmt.Sprintf("error: %s", err), 1)
+						}
+					}
+					return
+				}
+				err := bcdedit.Reboot(uuid)
+				if err != nil {
+					if wErr, ok := err.(*errors.WinError); ok {
+						windows.Fatal("Win2Linux", fmt.Sprintf("error: %s", err), int(wErr.ExitCode()))
+					} else {
+						windows.Fatal("Win2Linux", fmt.Sprintf("error: %s", err), 1)
+					}
+				}
+			}
 		}()
-
 	}
 	systray.AddSeparator()
 	mQuit := systray.AddMenuItem("Quit", "Quit")
@@ -66,69 +85,3 @@ func onReady() {
 func onExit() {
 	// clean up here
 }
-
-func list() []Entry {
-	cmd := exec.Command("bcdedit", "/enum", "firmware")
-	out, err := cmd.Output()
-	if err != nil {
-		panic(err)
-	}
-	return parse(string(out))
-}
-
-func parse(out string) []Entry {
-	lines := strings.Split(out, "\r\n")
-	section := false
-	lastLine := ""
-	secName := ""
-	a := make(map[string][]string)
-	for _, l := range lines {
-		if !section {
-			if r.Match([]byte(l)) {
-				secName = lastLine
-				section = true
-			}
-		} else {
-			if len(l) != 0 {
-				a[secName] = append(a[secName], l)
-			} else {
-				section = false
-			}
-		}
-
-		lastLine = l
-	}
-
-	var entries []Entry
-	for k, sec := range a {
-		entry := Entry{
-			Key:        k,
-			Attributes: make(map[string]string),
-		}
-		for _, l := range sec {
-			l = strings.Join(strings.Fields(l), " ")
-			val := strings.Split(l, " ")
-			if len(val) != 2 {
-				if len(val) > 2 {
-					val[1] = strings.Join(val[1:], " ")
-				}
-				continue
-			}
-			entry.Attributes[val[0]] = val[1]
-		}
-		entries = append(entries, entry)
-	}
-	return entries
-}
-
-func reboot(uuid string) {
-	if err := exec.Command("bcdedit", "/Set", "{fwbootmgr}", "BootSequence", uuid, "/addFirst").Run(); err != nil {
-		panic(err)
-	}
-
-	if err := exec.Command("shutdown", "/r", "/t", "0").Run(); err != nil {
-		panic(err)
-	}
-
-	os.Exit(0)
-}

pkg/bcdedit/bcdedit.go

@@ -0,0 +1,119 @@
+package bcdedit
+
+import (
+	"Win2Linux/pkg/windows/cmd"
+	"regexp"
+	"strings"
+)
+
+type (
+	Entry struct {
+		Key        string
+		Attributes map[string]string
+	}
+)
+
+var (
+	r *regexp.Regexp = regexp.MustCompile(`-+`)
+)
+
+func List() ([]Entry, error) {
+	args := []string{
+		"bcdedit",
+		"/enum",
+		"firmware",
+	}
+	out, err := cmd.Execute(args)
+	if err != nil {
+		return nil, err
+	}
+	return parse(out), nil
+}
+
+func Reboot(uuid string) error {
+	args := []string{
+		"bcdedit",
+		"/bootsequence",
+		uuid,
+		"/addfirst",
+	}
+	_, err := cmd.Execute(args)
+	if err != nil {
+		return err
+	}
+
+	args = []string{
+		"shutdown",
+		"/r",
+		"/t",
+		"0",
+	}
+	_, err = cmd.Execute(args)
+	if err != nil {
+		return err
+	}
+
+	return nil
+}
+
+func RebootToFirmware() error {
+	args := []string{
+		"shutdown",
+		"/r",
+		"/fw",
+		"/t",
+		"0",
+	}
+
+	_, err := cmd.Execute(args)
+	if err != nil {
+		return err
+	}
+
+	return nil
+}
+
+func parse(out string) []Entry {
+	lines := strings.Split(out, "\r\n")
+	section := false
+	lastLine := ""
+	secName := ""
+	a := make(map[string][]string)
+	for _, l := range lines {
+		if !section {
+			if r.Match([]byte(l)) {
+				secName = lastLine
+				section = true
+			}
+		} else {
+			if len(l) != 0 {
+				a[secName] = append(a[secName], l)
+			} else {
+				section = false
+			}
+		}
+
+		lastLine = l
+	}
+
+	var entries []Entry
+	for k, sec := range a {
+		entry := Entry{
+			Key:        k,
+			Attributes: make(map[string]string),
+		}
+		for _, l := range sec {
+			l = strings.Join(strings.Fields(l), " ")
+			val := strings.Split(l, " ")
+			if len(val) != 2 {
+				if len(val) > 2 {
+					val[1] = strings.Join(val[1:], " ")
+				}
+				continue
+			}
+			entry.Attributes[val[0]] = val[1]
+		}
+		entries = append(entries, entry)
+	}
+	return entries
+}

pkg/errors/errors.go

@@ -0,0 +1,25 @@
+package errors
+
+import "fmt"
+
+type (
+	WinError struct {
+		err   string
+		_code uint32
+	}
+)
+
+func New(err any, exitCode uint32) *WinError {
+	return &WinError{
+		err:   fmt.Sprintf("%v", err),
+		_code: exitCode,
+	}
+}
+
+func (e *WinError) Error() string {
+	return e.err
+}
+
+func (e *WinError) ExitCode() uint32 {
+	return e._code
+}

pkg/windows/cmd/cmd.go

@@ -0,0 +1,18 @@
+package cmd
+
+import (
+	"Win2Linux/pkg/errors"
+	"fmt"
+	"os/exec"
+	"strings"
+)
+
+func Execute(c []string) (string, error) {
+	a := strings.Join(c, " ")
+	cmd := exec.Command("cmd", "/d", "/c", "chcp 65001>nul && "+a)
+	out, err := cmd.CombinedOutput()
+	if err != nil {
+		return "", errors.New(fmt.Sprintf("failed to run the command: %s: %s", err, out), uint32(cmd.ProcessState.ExitCode()))
+	}
+	return string(out), nil
+}

pkg/windows/windows.go

@@ -0,0 +1,36 @@
+package windows
+
+import (
+	"os"
+	"syscall"
+	"unsafe"
+)
+
+const (
+	NULL  = 0
+	MB_OK = 0
+)
+
+func Fatal(title, message string, exitCode int) {
+	messageBox(NULL, message, title, MB_OK)
+	os.Exit(exitCode)
+}
+
+// MessageBox of Win32 API.
+func messageBox(hwnd uintptr, caption, title string, flags uint) int {
+	_caption, err := syscall.UTF16PtrFromString(caption)
+	if err != nil {
+		panic(err)
+	}
+	_title, err := syscall.UTF16PtrFromString(title)
+	if err != nil {
+		panic(err)
+	}
+	ret, _, _ := syscall.NewLazyDLL("user32.dll").NewProc("MessageBoxW").Call(
+		uintptr(hwnd),
+		uintptr(unsafe.Pointer(_caption)),
+		uintptr(unsafe.Pointer(_title)),
+		uintptr(flags))
+
+	return int(ret)
+}

winres/winres.json

@@ -18,7 +18,7 @@
         "minimum-os": "win10",
         "execution-level": "requireAdministrator",
         "ui-access": false,
-        "auto-elevate": false,
+        "auto-elevate": true,
         "dpi-awareness": "system",
         "disable-theming": false,
         "disable-window-filtering": false,
@@ -36,15 +36,15 @@
     "#1": {
       "0000": {
         "fixed": {
-          "file_version": "0.0.1.0",
-          "product_version": "0.0.1.0"
+          "file_version": "0.0.3.0",
+          "product_version": "0.0.3.0"
         },
         "info": {
           "0409": {
             "Comments": "Switch from Windows to Linux",
             "CompanyName": "Aurelie Delhaie",
             "ProductName": "Win2Linux",
-            "ProductVersion": "0.0.1.0"
+            "ProductVersion": "0.0.3.0"
           }
         }
       }