mirror of
https://github.com/netbirdio/netbird.git
synced 2026-04-21 17:56:39 +00:00
128 lines
2.5 KiB
Go
128 lines
2.5 KiB
Go
//go:build !windows
|
|
|
|
package client
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"os"
|
|
"os/signal"
|
|
"syscall"
|
|
|
|
log "github.com/sirupsen/logrus"
|
|
"golang.org/x/crypto/ssh"
|
|
"golang.org/x/term"
|
|
)
|
|
|
|
func (c *Client) setupTerminalMode(ctx context.Context, session *ssh.Session) error {
|
|
stdinFd := int(os.Stdin.Fd())
|
|
|
|
if !term.IsTerminal(stdinFd) {
|
|
return c.setupNonTerminalMode(ctx, session)
|
|
}
|
|
|
|
fd := int(os.Stdin.Fd())
|
|
|
|
state, err := term.MakeRaw(fd)
|
|
if err != nil {
|
|
return c.setupNonTerminalMode(ctx, session)
|
|
}
|
|
|
|
if err := c.setupTerminal(session, fd); err != nil {
|
|
if restoreErr := term.Restore(fd, state); restoreErr != nil {
|
|
log.Debugf("restore terminal state: %v", restoreErr)
|
|
}
|
|
return err
|
|
}
|
|
|
|
c.terminalState = state
|
|
c.terminalFd = fd
|
|
|
|
sigChan := make(chan os.Signal, 1)
|
|
signal.Notify(sigChan, syscall.SIGINT, syscall.SIGTERM, syscall.SIGQUIT)
|
|
|
|
go func() {
|
|
defer signal.Stop(sigChan)
|
|
select {
|
|
case <-ctx.Done():
|
|
if err := term.Restore(fd, state); err != nil {
|
|
log.Debugf("restore terminal state: %v", err)
|
|
}
|
|
case sig := <-sigChan:
|
|
if err := term.Restore(fd, state); err != nil {
|
|
log.Debugf("restore terminal state: %v", err)
|
|
}
|
|
signal.Reset(sig)
|
|
s, ok := sig.(syscall.Signal)
|
|
if !ok {
|
|
log.Debugf("signal %v is not a syscall.Signal: %T", sig, sig)
|
|
return
|
|
}
|
|
if err := syscall.Kill(syscall.Getpid(), s); err != nil {
|
|
log.Debugf("kill process with signal %v: %v", s, err)
|
|
}
|
|
}
|
|
}()
|
|
|
|
return nil
|
|
}
|
|
|
|
func (c *Client) setupNonTerminalMode(_ context.Context, session *ssh.Session) error {
|
|
return nil
|
|
}
|
|
|
|
// restoreWindowsConsoleState is a no-op on Unix systems
|
|
func (c *Client) restoreWindowsConsoleState() error {
|
|
return nil
|
|
}
|
|
|
|
func (c *Client) setupTerminal(session *ssh.Session, fd int) error {
|
|
w, h, err := term.GetSize(fd)
|
|
if err != nil {
|
|
return fmt.Errorf("get terminal size: %w", err)
|
|
}
|
|
|
|
modes := ssh.TerminalModes{
|
|
ssh.ECHO: 1,
|
|
ssh.TTY_OP_ISPEED: 14400,
|
|
ssh.TTY_OP_OSPEED: 14400,
|
|
// Ctrl+C
|
|
ssh.VINTR: 3,
|
|
// Ctrl+\
|
|
ssh.VQUIT: 28,
|
|
// Backspace
|
|
ssh.VERASE: 127,
|
|
// Ctrl+U
|
|
ssh.VKILL: 21,
|
|
// Ctrl+D
|
|
ssh.VEOF: 4,
|
|
ssh.VEOL: 0,
|
|
ssh.VEOL2: 0,
|
|
// Ctrl+Q
|
|
ssh.VSTART: 17,
|
|
// Ctrl+S
|
|
ssh.VSTOP: 19,
|
|
// Ctrl+Z
|
|
ssh.VSUSP: 26,
|
|
// Ctrl+O
|
|
ssh.VDISCARD: 15,
|
|
// Ctrl+R
|
|
ssh.VREPRINT: 18,
|
|
// Ctrl+W
|
|
ssh.VWERASE: 23,
|
|
// Ctrl+V
|
|
ssh.VLNEXT: 22,
|
|
}
|
|
|
|
terminal := os.Getenv("TERM")
|
|
if terminal == "" {
|
|
terminal = "xterm-256color"
|
|
}
|
|
|
|
if err := session.RequestPty(terminal, h, w, modes); err != nil {
|
|
return fmt.Errorf("request pty: %w", err)
|
|
}
|
|
|
|
return nil
|
|
}
|