Function TunnelMultiple

cli/pkg/tunnel/forwarding.go:285–343

转发指定SSH服务器的多个端口 到本地（Localhost）

(clientConn *ssh.Client, mapping map[string]string)

// 转发指定SSH服务器的多个端口 到本地（Localhost）
func TunnelMultiple(clientConn *ssh.Client, mapping map[string]string) error {

	if len(mapping) <= 0 {
		return nil
	}

	pipe := func(writer, reader net.Conn) {
		defer writer.Close()
		defer reader.Close()

		_, err := io.Copy(writer, reader)
		if err != nil {
			common.SmartIDELog.Warning(err.Error())
		}
	}

	//TODO: 这里可能会有问题
	for local, remote := range mapping {

		go func(local, remote string) {

			listener, err := net.Listen("tcp", local)
			if err != nil {
				common.SmartIDELog.Warning("failed to dial to remote: ", err.Error())
			}

			for {
				here, err := listener.Accept()
				if err != nil {
					common.SmartIDELog.Warning("failed to dial to remote: ", err.Error())
					time.Sleep(time.Second * 1)
					continue
				}
				if here == nil {
					common.SmartIDELog.Importance("本地连接失败" + local)
					time.Sleep(time.Second * 1)
					continue
				}
				go func(here net.Conn) {
					there, err := clientConn.Dial("tcp", remote)
					if there == nil {
						common.SmartIDELog.Importance("ssh 连接失败，请确保远程主机上相应端口已打开 " + remote)
						time.Sleep(time.Second * 1)
						return
					}
					if err != nil {
						common.SmartIDELog.Warning(err.Error())
					}
					go pipe(there, here)
					go pipe(here, there)
				}(here)
			}

		}(local, remote)

	}

	return nil