fix(webconsole,climc): cloud shell within docker compose environment

pkg/webconsole/command/climc_ssh_command.go

@@ -0,0 +1,116 @@
+// Copyright 2019 Yunion
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package command
+
+import (
+	"fmt"
+	"io/ioutil"
+	"os"
+	"os/exec"
+
+	"yunion.io/x/pkg/errors"
+
+	"yunion.io/x/onecloud/pkg/mcclient"
+	"yunion.io/x/onecloud/pkg/webconsole/helper"
+)
+
+type ClimcSshInfo struct {
+	IpAddr   string `json:"ip_addr"`
+	Username string `json:"username"`
+}
+
+type ClimcSshCommand struct {
+	*BaseCommand
+	Info    *ClimcSshInfo
+	s       *mcclient.ClientSession
+	keyFile string
+	buffer  []byte
+}
+
+func NewClimcSshCommand(info *ClimcSshInfo, s *mcclient.ClientSession) (*ClimcSshCommand, error) {
+	if info.IpAddr == "" {
+		return nil, fmt.Errorf("Empty host ip address")
+	}
+	if info.Username == "" {
+		return nil, fmt.Errorf("Empty username")
+	}
+	privateKey, err := helper.GetValidPrivateKey(info.IpAddr, 22, info.Username, "")
+	if err != nil {
+		return nil, errors.Wrap(err, "get cloud admin private key")
+	}
+	file, err := ioutil.TempFile("", fmt.Sprintf("id_rsa.%s.", info.IpAddr))
+	if err != nil {
+		return nil, err
+	}
+	defer file.Close()
+	filename := file.Name()
+	{
+		err = os.Chmod(filename, 0600)
+		if err != nil {
+			return nil, err
+		}
+		_, err = file.Write([]byte(privateKey))
+		if err != nil {
+			return nil, err
+		}
+	}
+	name := "bash"
+	args := []string{
+		"-c",
+		fmt.Sprintf("ssh -o StrictHostKeyChecking=no -i %s %s@%s", filename, info.Username, info.IpAddr),
+	}
+	bCmd := NewBaseCommand(s, name, args...)
+	cmd := &ClimcSshCommand{
+		BaseCommand: bCmd,
+		Info:        info,
+		s:           s,
+		keyFile:     filename,
+		buffer:      []byte{},
+	}
+	return cmd, nil
+}
+
+func (c ClimcSshCommand) GetCommand() *exec.Cmd {
+	cmd := c.BaseCommand.GetCommand()
+	cmd.Env = append(cmd.Env, "TERM=xterm-256color")
+	return cmd
+}
+
+func (c ClimcSshCommand) GetProtocol() string {
+	return PROTOCOL_TTY
+}
+
+func (c ClimcSshCommand) Cleanup() error {
+	if len(c.keyFile) > 0 {
+		os.Remove(c.keyFile)
+		c.keyFile = ""
+	}
+	return nil
+}
+
+func (c *ClimcSshCommand) Scan(d byte, send func(msg string)) {
+	switch d {
+	case '\r': // 换行
+		send("\r\n")
+		c.buffer = []byte{}
+	case '\u007f': // 退格
+		if len(c.buffer) > 0 {
+			c.buffer = c.buffer[:len(c.buffer)-1]
+			send("\b \b")
+		}
+	default:
+		c.buffer = append(c.buffer, d)
+	}
+}

pkg/webconsole/command/socat_command.go

@@ -0,0 +1,63 @@
+// Copyright 2019 Yunion
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package command
+
+import (
+	"fmt"
+	"os/exec"
+
+	"yunion.io/x/onecloud/pkg/mcclient"
+)
+
+type SocatInfo struct {
+	IP   string `json:"ip"`
+	Port int    `json:"port"`
+}
+
+type SocatCmd struct {
+	*BaseCommand
+	Info *SocatInfo
+	s    *mcclient.ClientSession
+}
+
+func NewSocatCommand(info *SocatInfo, s *mcclient.ClientSession) (*SocatCmd, error) {
+	if info.IP == "" {
+		return nil, fmt.Errorf("Empty remote ip address")
+	}
+	if info.Port <= 0 {
+		return nil, fmt.Errorf("Invalid port %d", info.Port)
+	}
+
+	name := "bash"
+	args := []string{
+		"-c",
+		fmt.Sprintf("socat FILE:`tty`,raw,echo=0 TCP:%s:%d", info.IP, info.Port),
+	}
+	cmd := NewBaseCommand(s, name, args...)
+	scmd := &SocatCmd{
+		BaseCommand: cmd,
+		Info:        info,
+	}
+
+	return scmd, nil
+}
+
+func (c *SocatCmd) GetCommand() *exec.Cmd {
+	return c.BaseCommand.GetCommand()
+}
+
+func (c *SocatCmd) GetProtocol() string {
+	return PROTOCOL_TTY
+}