mirror of
https://github.com/lxc/incus
synced 2026-08-02 05:26:46 +00:00
65 lines
1.4 KiB
Go
65 lines
1.4 KiB
Go
//go:build windows
|
|
|
|
package main
|
|
|
|
import (
|
|
"io"
|
|
"os"
|
|
"os/signal"
|
|
|
|
"github.com/gorilla/websocket"
|
|
"golang.org/x/sys/windows"
|
|
|
|
"github.com/lxc/incus/v7/shared/api"
|
|
"github.com/lxc/incus/v7/shared/logger"
|
|
)
|
|
|
|
// Windows doesn't process ANSI sequences natively, so we wrap
|
|
// os.Stdout for improved user experience for Windows client
|
|
type WrappedWriteCloser struct {
|
|
io.Closer
|
|
wrapper io.Writer
|
|
}
|
|
|
|
func (wwc *WrappedWriteCloser) Write(p []byte) (int, error) {
|
|
return wwc.wrapper.Write(p)
|
|
}
|
|
|
|
func (c *cmdExec) getTERM() (string, bool) {
|
|
return "dumb", true
|
|
}
|
|
|
|
func (c *cmdExec) controlSocketHandler(control *websocket.Conn) {
|
|
ch := make(chan os.Signal, 10)
|
|
signal.Notify(ch, os.Interrupt)
|
|
|
|
closeMsg := websocket.FormatCloseMessage(websocket.CloseNormalClosure, "")
|
|
defer control.WriteMessage(websocket.CloseMessage, closeMsg)
|
|
|
|
for {
|
|
sig := <-ch
|
|
switch sig {
|
|
case os.Interrupt:
|
|
logger.Debugf("Received '%s signal', forwarding to executing program.", sig)
|
|
err := c.forwardSignal(control, windows.SIGINT)
|
|
if err != nil {
|
|
logger.Debugf("Failed to forward signal '%s'.", windows.SIGINT)
|
|
return
|
|
}
|
|
|
|
default:
|
|
break
|
|
}
|
|
}
|
|
}
|
|
|
|
func (c *cmdExec) forwardSignal(control *websocket.Conn, sig windows.Signal) error {
|
|
logger.Debugf("Forwarding signal: %s", sig)
|
|
|
|
msg := api.InstanceExecControl{}
|
|
msg.Command = "signal"
|
|
msg.Signal = int(sig)
|
|
|
|
return control.WriteJSON(msg)
|
|
}
|
