vendor/github.com/google/goterm/term/ssh.go - main - githubmirror/origin

vendor/github.com/google/goterm/term/ssh.go
203 строки6 KB
Enrique Llorente
virt, live-migration: add per role tests
04 ноя 2024, 12:21
04 ноя 2024, 12:217174ae9
// Copyright 2015 The Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.


package term

const (
	// Terminal attribute types.
	sshIflag = iota
	sshOflag
	sshCflag
	sshLflag
	sshCchar
	sshTspeed
	sshNOP

	// SSH terminal attributes.
	sshTTYOPEND    = 0
	sshVINTR       = 1
	sshVQUIT       = 2
	sshVERASE      = 3
	sshVKILL       = 4
	sshVEOF        = 5
	sshVEOL        = 6
	sshVEOL2       = 7
	sshVSTART      = 8
	sshVSTOP       = 9
	sshVSUSP       = 10
	sshVDSUSP      = 11
	sshVREPRINT    = 12
	sshVWERASE     = 13
	sshVLNEXT      = 14
	sshVFLUSH      = 15
	sshVSWTCH      = 16
	sshVSTATUS     = 17
	sshVDISCARD    = 18
	sshIGNPAR      = 30
	sshPARMRK      = 31
	sshINPCK       = 32
	sshISTRIP      = 33
	sshINLCR       = 34
	sshIGNCR       = 35
	sshICRNL       = 36
	sshIUCLC       = 37
	sshIXON        = 38
	sshIXANY       = 39
	sshIXOFF       = 40
	sshIMAXBEL     = 41
	sshISIG        = 50
	sshICANON      = 51
	sshXCASE       = 52
	sshECHO        = 53
	sshECHOE       = 54
	sshECHOK       = 55
	sshECHONL      = 56
	sshNOFLSH      = 57
	sshTOSTOP      = 58
	sshIEXTEN      = 59
	sshECHOCTL     = 60
	sshECHOKE      = 61
	sshPENDIN      = 62
	sshOPOST       = 70
	sshOLCUC       = 71
	sshONLCR       = 72
	sshOCRNL       = 73
	sshONOCR       = 74
	sshONLRET      = 75
	sshCS7         = 90
	sshCS8         = 91
	sshPARENB      = 92
	sshPARODD      = 93
	sshTTYOPISPEED = 128
	sshTTYOPOSPEED = 129
)

var convertSSH = map[uint8]struct {
	tType  uint
	native uint32
}{
	sshTTYOPEND:    {tType: sshNOP},
	sshVINTR:       {tType: sshCchar, native: VINTR},
	sshVQUIT:       {tType: sshCchar, native: VQUIT},
	sshVERASE:      {tType: sshCchar, native: VERASE},
	sshVKILL:       {tType: sshCchar, native: VKILL},
	sshVEOF:        {tType: sshCchar, native: VEOF},
	sshVEOL:        {tType: sshCchar, native: VEOL},
	sshVEOL2:       {tType: sshCchar, native: VEOL2},
	sshVSTART:      {tType: sshCchar, native: VSTART},
	sshVSTOP:       {tType: sshCchar, native: VSTOP},
	sshVSUSP:       {tType: sshCchar, native: VSUSP},
	sshVDSUSP:      {tType: sshCchar, native: sshNOP},
	sshVREPRINT:    {tType: sshCchar, native: VREPRINT},
	sshVWERASE:     {tType: sshCchar, native: VWERASE},
	sshVLNEXT:      {tType: sshCchar, native: VLNEXT},
	sshVFLUSH:      {tType: sshNOP},
	sshVSWTCH:      {tType: sshCchar, native: VSWTC},
	sshVSTATUS:     {tType: sshNOP},
	sshVDISCARD:    {tType: sshCchar, native: VDISCARD},
	sshIGNPAR:      {tType: sshIflag, native: IGNPAR},
	sshPARMRK:      {tType: sshIflag, native: PARMRK},
	sshINPCK:       {tType: sshIflag, native: INPCK},
	sshISTRIP:      {tType: sshIflag, native: ISTRIP},
	sshINLCR:       {tType: sshIflag, native: INLCR},
	sshIGNCR:       {tType: sshIflag, native: IGNCR},
	sshICRNL:       {tType: sshIflag, native: ICRNL},
	sshIUCLC:       {tType: sshIflag, native: IUCLC},
	sshIXON:        {tType: sshIflag, native: IXON},
	sshIXANY:       {tType: sshIflag, native: IXANY},
	sshIXOFF:       {tType: sshIflag, native: IXOFF},
	sshIMAXBEL:     {tType: sshIflag, native: IMAXBEL},
	sshISIG:        {tType: sshLflag, native: ISIG},
	sshICANON:      {tType: sshLflag, native: ICANON},
	sshXCASE:       {tType: sshLflag, native: XCASE},
	sshECHO:        {tType: sshLflag, native: ECHO},
	sshECHOE:       {tType: sshLflag, native: ECHOE},
	sshECHOK:       {tType: sshLflag, native: ECHOK},
	sshECHONL:      {tType: sshLflag, native: ECHONL},
	sshNOFLSH:      {tType: sshLflag, native: NOFLSH},
	sshTOSTOP:      {tType: sshLflag, native: TOSTOP},
	sshIEXTEN:      {tType: sshLflag, native: IEXTEN},
	sshECHOCTL:     {tType: sshLflag, native: ECHOCTL},
	sshECHOKE:      {tType: sshLflag, native: ECHOKE},
	sshPENDIN:      {tType: sshNOP},
	sshOPOST:       {tType: sshOflag, native: OPOST},
	sshOLCUC:       {tType: sshOflag, native: OLCUC},
	sshONLCR:       {tType: sshOflag, native: ONLCR},
	sshOCRNL:       {tType: sshOflag, native: OCRNL},
	sshONOCR:       {tType: sshOflag, native: ONOCR},
	sshONLRET:      {tType: sshOflag, native: ONLRET},
	sshCS7:         {tType: sshCflag, native: CS7},
	sshCS8:         {tType: sshCflag, native: CS8},
	sshPARENB:      {tType: sshCflag, native: PARENB},
	sshPARODD:      {tType: sshCflag, native: PARODD},
	sshTTYOPISPEED: {tType: sshTspeed},
	sshTTYOPOSPEED: {tType: sshTspeed},
}

// ToSSH converts the Termios attributes to SSH attributes usable as ssh.TerminalModes.
func (t *Termios) ToSSH() map[uint8]uint32 {
	sshModes := make(map[uint8]uint32, len(convertSSH))
	var flags uint32
	for sshID, tios := range convertSSH {
		switch tios.tType {
		case sshIflag:
			flags = t.Iflag
		case sshOflag:
			flags = t.Oflag
		case sshLflag:
			flags = t.Lflag
		case sshCflag:
			flags = t.Cflag
		case sshCchar:
			sshModes[sshID] = uint32(t.Cc[tios.native])
			continue
		case sshTspeed:
			sshModes[sshTTYOPISPEED], sshModes[sshTTYOPOSPEED] = t.Ispeed, t.Ospeed
			continue
		default:
			continue
		}
		var onOff uint32
		if tios.native&flags > 0 {
			onOff = 1
		}
		sshModes[sshID] = onOff
	}
	return sshModes
}

// FromSSH converts SSH attributes to Termios attributes.
func (t *Termios) FromSSH(termModes map[uint8]uint32) {
	var flags *uint32
	for sshID, val := range termModes {
		switch convertSSH[sshID].tType {
		case sshIflag:
			flags = &t.Iflag
		case sshOflag:
			flags = &t.Oflag
		case sshLflag:
			flags = &t.Lflag
		case sshCflag:
			flags = &t.Cflag
		case sshCchar:
			t.Cc[convertSSH[sshID].native] = byte(val)
			continue
		case sshTspeed:
			if sshID == sshTTYOPISPEED {
				t.Ispeed = val
			} else {
				t.Ospeed = val
			}
			continue
		default:
			continue
		}
		if val > 0 {
			*flags |= convertSSH[sshID].native
		} else {
			*flags &^= convertSSH[sshID].native
		}
	}
}