thelilfrog/Win2Linux
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 3 Wiki Activity

Fix other

Browse Source
This commit is contained in:
Aurélie DELHAIE
2026-04-19 12:56:16 +02:00
parent 669c5b3011
commit 42effa04a5
7 changed files with 223 additions and 103 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
.vscode/launch.json vendored Normal file
View File

@@ -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}"
}
]
}

132
main.go
View File

@@ -1,28 +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
@@ -35,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)
@@ -49,14 +49,29 @@ func onReady() {
}
mCustom := systray.AddMenuItem(key, "Switch to"+key)
go func() {
<-mCustom.ClickedCh
if uuid == "{fwbootmgr}" {
rebootToFirmware()
return
for {
<-mCustom.ClickedCh
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)
}
}
}
reboot(uuid)
}()
}
systray.AddSeparator()
mQuit := systray.AddMenuItem("Quit", "Quit")
@@ -70,82 +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 {
fatal(fmt.Sprintf("failed to run bcdedit: %s\n\nRun this program in administrator mode", err), 1)
}
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) {
out, err := exec.Command("bcdedit", "/Set", "{fwbootmgr}", "BootSequence", uuid, "/addFirst").Output()
if err != nil {
windows.MessageBox(windows.NULL, fmt.Sprintf("failed to execute bcdedit: %s", out), "Win2Linux", windows.MB_OK)
return
}
out, err = exec.Command("shutdown", "/r", "/t", "0").Output()
if err != nil {
windows.MessageBox(windows.NULL, fmt.Sprintf("failed to restart the computer: %s", out), "Win2Linux", windows.MB_OK)
}
}
func rebootToFirmware() {
out, err := exec.Command("shutdown", "/r", "/fw", "/t", "0").Output()
if err != nil {
windows.MessageBox(windows.NULL, fmt.Sprintf("failed to restart the computer: %s", out), "Win2Linux", windows.MB_OK)
}
}
func fatal(message string, exitCode int) {
windows.MessageBox(windows.NULL, message, "Win2Linux", windows.MB_OK)
os.Exit(exitCode)
}

119
pkg/bcdedit/bcdedit.go Normal file
View File

@@ -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
}

25
pkg/errors/errors.go Normal file
View File

@@ -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
}

18
pkg/windows/cmd/cmd.go Normal file
View File

@@ -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
}

8
pkg/windows/windows.go
View File

@@ -1,6 +1,7 @@
package windows
import (
"os"
"syscall"
"unsafe"
)
@@ -10,8 +11,13 @@ const (
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 {
func messageBox(hwnd uintptr, caption, title string, flags uint) int {
_caption, err := syscall.UTF16PtrFromString(caption)
if err != nil {
panic(err)

8
winres/winres.json
View File

@@ -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"
}
}
}