Skip to content

K
koko
Commit 216880ee authored by Eric's avatar Eric
Browse files
Options

restruct code

parent 7ca4f893
Hide whitespace changes
Inline Side-by-side
Showing with 208 additions and 188 deletions
go.mod
View file @ 216880ee
...@@ -4,10 +4,9 @@ go 1.12 ...@@ -4,10 +4,9 @@ go 1.12
require ( require (
github.com/anmitsu/go-shlex v0.0.0-20161002113705-648efa622239 // indirect github.com/anmitsu/go-shlex v0.0.0-20161002113705-648efa622239 // indirect
github.com/gdamore/tcell v1.1.1 github.com/flynn/go-shlex v0.0.0-20150515145356-3f9db97f8568 // indirect
github.com/gliderlabs/ssh v0.1.3 github.com/gliderlabs/ssh v0.1.3
github.com/kr/pty v1.1.4 // indirect github.com/mattn/go-runewidth v0.0.4 // indirect
github.com/mattn/go-runewidth v0.0.4
github.com/olekukonko/tablewriter v0.0.1 github.com/olekukonko/tablewriter v0.0.1
github.com/satori/go.uuid v1.2.0 github.com/satori/go.uuid v1.2.0
github.com/sirupsen/logrus v1.4.0 github.com/sirupsen/logrus v1.4.0
......
pkg/core/init.go deleted 100644 → 0
View file @ 7ca4f893
package core
import (
"github.com/sirupsen/logrus"
)
var log *logrus.Logger
const maxBufferSize = 1024 * 4
func init() {
log = logrus.New()
}
pkg/core/proxychannel.go deleted 100644 → 0
View file @ 7ca4f893
package core
import (
"context"
uuid "github.com/satori/go.uuid"
)
type ProxyChannel interface {
UUID() string
ReceiveRequestChannel(ctx context.Context) chan<- []byte
SendResponseChannel(ctx context.Context) <-chan []byte
Wait() error
}
func NewMemoryChannel(nConn *NodeConn, useS Conn) *memoryChannel {
m := &memoryChannel{
uuid: nConn.UUID(),
conn: nConn,
}
return m
}
type memoryChannel struct {
uuid uuid.UUID
conn *NodeConn
}
func (m *memoryChannel) UUID() string {
return m.uuid.String()
}
func (m *memoryChannel) SendResponseChannel(ctx context.Context) <-chan []byte {
go m.conn.handleResponse(ctx)
return m.conn.outChan
}
func (m *memoryChannel) ReceiveRequestChannel(ctx context.Context) chan<- []byte {
go m.conn.handleRequest(ctx)
return m.conn.inChan
}
func (m *memoryChannel) Wait() error {
return m.conn.Wait()
}
pkg/core/specialrule.go → pkg/parser/specialrule.go
View file @ 216880ee
package core package parser
type SpecialRuler interface { type SpecialRuler interface {
......
pkg/core/terminal.go → pkg/parser/terminal.go
View file @ 216880ee
package core package parser
import ( import (
"bytes" "bytes"
......
pkg/core/whiteblackrule.go → pkg/parser/whiteblackrule.go
View file @ 216880ee
package core package parser
import ( import (
"regexp" "regexp"
......
pkg/core/switch.go → pkg/proxy/switch.go
View file @ 216880ee
package core package proxy
import ( import (
"cocogo/pkg/transport"
"cocogo/pkg/userhome"
"context" "context"
"sync" "sync"
"github.com/sirupsen/logrus"
) )
var log = logrus.New()
var Manager = &manager{ var Manager = &manager{
container: new(sync.Map), container: new(sync.Map),
} }
...@@ -13,7 +19,7 @@ type manager struct { ...@@ -13,7 +19,7 @@ type manager struct {
container *sync.Map container *sync.Map
} }
func (m *manager) add(uHome SessionHome) { func (m *manager) add(uHome userhome.SessionHome) {
m.container.Store(uHome.SessionID(), uHome) m.container.Store(uHome.SessionID(), uHome)
} }
...@@ -23,35 +29,35 @@ func (m *manager) delete(roomID string) { ...@@ -23,35 +29,35 @@ func (m *manager) delete(roomID string) {
} }
func (m *manager) search(roomID string) (SessionHome, bool) { func (m *manager) search(roomID string) (userhome.SessionHome, bool) {
if uHome, ok := m.container.Load(roomID); ok { if uHome, ok := m.container.Load(roomID); ok {
return uHome.(SessionHome), ok return uHome.(userhome.SessionHome), ok
} }
return nil, false return nil, false
} }
func (m *manager) JoinShareRoom(roomID string, uConn Conn) { func (m *manager) JoinShareRoom(roomID string, uConn userhome.Conn) {
if userHome, ok := m.search(roomID); ok { if userHome, ok := m.search(roomID); ok {
userHome.AddConnection(uConn) userHome.AddConnection(uConn)
} }
} }
func (m *manager) ExitShareRoom(roomID string, uConn Conn) { func (m *manager) ExitShareRoom(roomID string, uConn userhome.Conn) {
if userHome, ok := m.search(roomID); ok { if userHome, ok := m.search(roomID); ok {
userHome.RemoveConnection(uConn) userHome.RemoveConnection(uConn)
} }
} }
func (m *manager) Switch(ctx context.Context, userHome SessionHome, pChannel ProxyChannel) error { func (m *manager) Switch(ctx context.Context, uHome userhome.SessionHome, agent transport.Agent) error {
m.add(userHome) m.add(uHome)
defer m.delete(userHome.SessionID()) defer m.delete(uHome.SessionID())
subCtx, cancelFunc := context.WithCancel(ctx) subCtx, cancelFunc := context.WithCancel(ctx)
userSendRequestStream := userHome.SendRequestChannel(subCtx) userSendRequestStream := uHome.SendRequestChannel(subCtx)
userReceiveStream := userHome.ReceiveResponseChannel(subCtx) userReceiveStream := uHome.ReceiveResponseChannel(subCtx)
nodeRequestChan := pChannel.ReceiveRequestChannel(subCtx) nodeRequestChan := agent.ReceiveRequestChannel(subCtx)
nodeSendResponseStream := pChannel.SendResponseChannel(subCtx) nodeSendResponseStream := agent.SendResponseChannel(subCtx)
for userSendRequestStream != nil || nodeSendResponseStream != nil { for userSendRequestStream != nil || nodeSendResponseStream != nil {
select { select {
...@@ -59,6 +65,7 @@ func (m *manager) Switch(ctx context.Context, userHome SessionHome, pChannel Pro ...@@ -59,6 +65,7 @@ func (m *manager) Switch(ctx context.Context, userHome SessionHome, pChannel Pro
if !ok { if !ok {
log.Warn("userSendRequestStream close") log.Warn("userSendRequestStream close")
userSendRequestStream = nil userSendRequestStream = nil
close(nodeRequestChan)
continue continue
} }
nodeRequestChan <- buf1 nodeRequestChan <- buf1
...@@ -72,9 +79,11 @@ func (m *manager) Switch(ctx context.Context, userHome SessionHome, pChannel Pro ...@@ -72,9 +79,11 @@ func (m *manager) Switch(ctx context.Context, userHome SessionHome, pChannel Pro
} }
userReceiveStream <- buf2 userReceiveStream <- buf2
case <-ctx.Done(): case <-ctx.Done():
log.Info("proxy end by context done")
cancelFunc()
return nil return nil
} }
} }
log.Info("switch end") log.Info("proxy end")
return nil return nil
} }
pkg/sshd/interactive.go
View file @ 216880ee
package sshd package sshd
import ( import (
"cocogo/pkg/core"
"cocogo/pkg/model" "cocogo/pkg/model"
"cocogo/pkg/proxy"
"cocogo/pkg/transport"
"cocogo/pkg/userhome"
"context" "context"
"fmt" "fmt"
"io" "io"
...@@ -178,10 +180,7 @@ func (s *sshInteractive) changeLanguage() { ...@@ -178,10 +180,7 @@ func (s *sshInteractive) changeLanguage() {
} }
func (s *sshInteractive) JoinShareRoom(roomID string) { func (s *sshInteractive) JoinShareRoom(roomID string) {
sshConn := &SSHConn{ sshConn := userhome.NewSSHConn(s.sess)
conn: s.sess,
uuid: generateNewUUID(),
}
ctx, cancelFuc := context.WithCancel(s.sess.Context()) ctx, cancelFuc := context.WithCancel(s.sess.Context())
_, winCh, _ := s.sess.Pty() _, winCh, _ := s.sess.Pty()
...@@ -198,7 +197,7 @@ func (s *sshInteractive) JoinShareRoom(roomID string) { ...@@ -198,7 +197,7 @@ func (s *sshInteractive) JoinShareRoom(roomID string) {
} }
} }
}() }()
core.Manager.JoinShareRoom(roomID, sshConn) proxy.Manager.JoinShareRoom(roomID, sshConn)
log.Info("exit room id:", roomID) log.Info("exit room id:", roomID)
cancelFuc() cancelFuc()
...@@ -359,30 +358,28 @@ func (s *sshInteractive) Proxy(asset model.Asset, systemUser model.SystemUserAut ...@@ -359,30 +358,28 @@ func (s *sshInteractive) Proxy(asset model.Asset, systemUser model.SystemUserAut
3. 创建一个NodeConn，及相关的channel 可以是MemoryChannel 或者是redisChannel 3. 创建一个NodeConn，及相关的channel 可以是MemoryChannel 或者是redisChannel
4. session Home 与 proxy channel 交换数据 4. session Home 与 proxy channel 交换数据
*/ */
sshConn := &SSHConn{ ptyReq, winChan, _ := s.sess.Pty()
conn: s.sess, sshConn := userhome.NewSSHConn(s.sess)
uuid: generateNewUUID(), serverAuth := transport.ServerAuth{
}
serverAuth := core.ServerAuth{
IP: asset.Ip, IP: asset.Ip,
Port: asset.Port, Port: asset.Port,
UserName: systemUser.UserName, UserName: systemUser.UserName,
Password: systemUser.Password, Password: systemUser.Password,
PublicKey: parsePrivateKey(systemUser.PrivateKey)} PublicKey: parsePrivateKey(systemUser.PrivateKey)}
nodeConn, err := core.NewNodeConn(serverAuth, sshConn) nodeConn, err := transport.NewNodeConn(s.sess.Context(), serverAuth, ptyReq, winChan)
if err != nil { if err != nil {
log.Error(err) log.Error(err)
return err return err
} }
defer nodeConn.Close() defer nodeConn.Close()
memChan := core.NewMemoryChannel(nodeConn, sshConn) memChan := transport.NewMemoryAgent(nodeConn)
userHome := core.NewUserSessionHome(sshConn) Home := userhome.NewUserSessionHome(sshConn)
log.Info("session Home ID: ", userHome.SessionID()) log.Info("session Home ID: ", Home.SessionID())
err = core.Manager.Switch(context.TODO(), userHome, memChan) err = proxy.Manager.Switch(s.sess.Context(), Home, memChan)
if err != nil { if err != nil {
log.Error(err) log.Error(err)
} }
......
pkg/sshd/server.go
View file @ 216880ee
...@@ -5,6 +5,7 @@ import ( ...@@ -5,6 +5,7 @@ import (
"cocogo/pkg/config" "cocogo/pkg/config"
"cocogo/pkg/model" "cocogo/pkg/model"
"io" "io"
"runtime"
"strconv" "strconv"
"sync" "sync"
"text/template" "text/template"
...@@ -75,6 +76,8 @@ func connectHandler(sess ssh.Session) { ...@@ -75,6 +76,8 @@ func connectHandler(sess ssh.Session) {
log.Info("accept one session") log.Info("accept one session")
userInteractive.displayHelpInfo() userInteractive.displayHelpInfo()
userInteractive.StartDispatch() userInteractive.StartDispatch()
log.Info("finish one session")
runtime.GC()
} else { } else {
_, err := io.WriteString(sess, "No PTY requested.\n") _, err := io.WriteString(sess, "No PTY requested.\n")
......
pkg/sshd/sshutil.go
View file @ 216880ee
package sshd package sshd
import ( import (
uuid "github.com/satori/go.uuid"
gossh "golang.org/x/crypto/ssh" gossh "golang.org/x/crypto/ssh"
) )
func parsePrivateKey(privateKey string) gossh.Signer { func parsePrivateKey(privateKey string) gossh.Signer {
private, err := gossh.ParsePrivateKey([]byte(privateKey)) private, err := gossh.ParsePrivateKey([]byte(privateKey))
if err != nil { if err != nil {
log.Info("Failed to parse private key: ", err) log.Error("Failed to parse private key: ", err)
} }
return private return private
} }
func generateNewUUID() uuid.UUID {
return uuid.NewV4()
}
pkg/sshd/template.go
View file @ 216880ee
...@@ -7,7 +7,7 @@ const welcomeTemplate = ` ...@@ -7,7 +7,7 @@ const welcomeTemplate = `
{{.Tab}}3) Enter {{.ColorCode}}p{{.ColorEnd}} to display the host you have permission.{{.EndLine}} {{.Tab}}3) Enter {{.ColorCode}}p{{.ColorEnd}} to display the host you have permission.{{.EndLine}}
{{.Tab}}4) Enter {{.ColorCode}}g{{.ColorEnd}} to display the node that you have permission.{{.EndLine}} {{.Tab}}4) Enter {{.ColorCode}}g{{.ColorEnd}} to display the node that you have permission.{{.EndLine}}
{{.Tab}}5) Enter {{.ColorCode}}g{{.ColorEnd}} + {{.ColorCode}}NodeID{{.ColorEnd}} to display the host under the node, such as g1. {{.EndLine}} {{.Tab}}5) Enter {{.ColorCode}}g{{.ColorEnd}} + {{.ColorCode}}NodeID{{.ColorEnd}} to display the host under the node, such as g1. {{.EndLine}}
{{.Tab}}6) Enter {{.ColorCode}}s{{.ColorEnd}} Chinese-english switch.{{.EndLine}} {{.Tab}}6) Enter {{.ColorCode}}s{{.ColorEnd}} Chinese-english proxy.{{.EndLine}}
{{.Tab}}7) Enter {{.ColorCode}}h{{.ColorEnd}} help.{{.EndLine}} {{.Tab}}7) Enter {{.ColorCode}}h{{.ColorEnd}} help.{{.EndLine}}
{{.Tab}}8) Enter {{.ColorCode}}r{{.ColorEnd}} to refresh your assets and nodes.{{.EndLine}} {{.Tab}}8) Enter {{.ColorCode}}r{{.ColorEnd}} to refresh your assets and nodes.{{.EndLine}}
{{.Tab}}0) Enter {{.ColorCode}}q{{.ColorEnd}} exit.{{.EndLine}} {{.Tab}}0) Enter {{.ColorCode}}q{{.ColorEnd}} exit.{{.EndLine}}
......
pkg/transport/conn.go 0 → 100644
View file @ 216880ee
package transport
import "context"
type Conn interface {
ReceiveRequest(context.Context, <-chan []byte, chan<- []byte)
SendResponse(context.Context, chan<- []byte)
}
pkg/core/nodeconn.go → pkg/transport/sshconn.go
View file @ 216880ee
package core package transport
import ( import (
"cocogo/pkg/parser"
"context" "context"
"fmt" "fmt"
"io" "io"
"github.com/sirupsen/logrus"
"github.com/gliderlabs/ssh" "github.com/gliderlabs/ssh"
uuid "github.com/satori/go.uuid" uuid "github.com/satori/go.uuid"
gossh "golang.org/x/crypto/ssh" gossh "golang.org/x/crypto/ssh"
) )
type Conn interface { var log = logrus.New()
SessionID() string
User() string
UUID() uuid.UUID
Pty() (ssh.Pty, <-chan ssh.Window, bool)
Context() context.Context const maxBufferSize = 1024 * 4
io.Reader
io.WriteCloser
}
type ServerAuth struct { type ServerAuth struct {
IP string IP string
...@@ -57,13 +49,12 @@ func CreateNodeSession(authInfo ServerAuth) (c *gossh.Client, s *gossh.Session, ...@@ -57,13 +49,12 @@ func CreateNodeSession(authInfo ServerAuth) (c *gossh.Client, s *gossh.Session,
return client, s, nil return client, s, nil
} }
func NewNodeConn(authInfo ServerAuth, userS Conn) (*NodeConn, error) { func NewNodeConn(ctx context.Context, authInfo ServerAuth, ptyReq ssh.Pty, winCh <-chan ssh.Window) (*NodeConn, error) {
c, s, err := CreateNodeSession(authInfo) c, s, err := CreateNodeSession(authInfo)
if err != nil { if err != nil {
return nil, err return nil, err
} }
ptyReq, winCh, _ := userS.Pty()
err = s.RequestPty(ptyReq.Term, ptyReq.Window.Height, ptyReq.Window.Width, gossh.TerminalModes{}) err = s.RequestPty(ptyReq.Term, ptyReq.Window.Height, ptyReq.Window.Width, gossh.TerminalModes{})
if err != nil { if err != nil {
return nil, err return nil, err
...@@ -81,7 +72,7 @@ func NewNodeConn(authInfo ServerAuth, userS Conn) (*NodeConn, error) { ...@@ -81,7 +72,7 @@ func NewNodeConn(authInfo ServerAuth, userS Conn) (*NodeConn, error) {
if err != nil { if err != nil {
return nil, err return nil, err
} }
ctx, cancelFunc := context.WithCancel(userS.Context()) ctx, cancelFunc := context.WithCancel(ctx)
nConn := &NodeConn{ nConn := &NodeConn{
uuid: uuid.NewV4(), uuid: uuid.NewV4(),
...@@ -91,9 +82,7 @@ func NewNodeConn(authInfo ServerAuth, userS Conn) (*NodeConn, error) { ...@@ -91,9 +82,7 @@ func NewNodeConn(authInfo ServerAuth, userS Conn) (*NodeConn, error) {
ctxCancelFunc: cancelFunc, ctxCancelFunc: cancelFunc,
stdin: nodeStdin, stdin: nodeStdin,
stdout: nodeStdout, stdout: nodeStdout,
tParser: NewTerminalParser(), tParser: parser.NewTerminalParser(),
inChan: make(chan []byte),
outChan: make(chan []byte),
} }
go nConn.windowChangeHandler(winCh) go nConn.windowChangeHandler(winCh)
...@@ -107,16 +96,14 @@ type NodeConn struct { ...@@ -107,16 +96,14 @@ type NodeConn struct {
conn *gossh.Session conn *gossh.Session
stdin io.Writer stdin io.Writer
stdout io.Reader stdout io.Reader
tParser *TerminalParser tParser *parser.TerminalParser
currentCommandInput string currentCommandInput string
currentCommandResult string currentCommandResult string
rulerFilters []RuleFilter rulerFilters []parser.RuleFilter
specialCommands []SpecialRuler specialCommands []parser.SpecialRuler
inSpecialStatus bool inSpecialStatus bool
ctx context.Context ctx context.Context
ctxCancelFunc context.CancelFunc ctxCancelFunc context.CancelFunc
inChan chan []byte
outChan chan []byte
} }
func (n *NodeConn) UUID() uuid.UUID { func (n *NodeConn) UUID() uuid.UUID {
...@@ -171,13 +158,69 @@ func (n *NodeConn) windowChangeHandler(winCH <-chan ssh.Window) { ...@@ -171,13 +158,69 @@ func (n *NodeConn) windowChangeHandler(winCH <-chan ssh.Window) {
} }
func (n *NodeConn) handleRequest(ctx context.Context) { func (n *NodeConn) Close() {
select {
case <-n.ctx.Done():
return
default:
_ = n.conn.Close()
_ = n.client.Close()
n.ctxCancelFunc()
log.Info("Close conn")
}
}
func (n *NodeConn) SendResponse(ctx context.Context, outChan chan<- []byte) {
buf := make([]byte, maxBufferSize)
defer close(outChan)
for { for {
nr, err := n.stdout.Read(buf)
if err != nil {
log.Error("read conn err:", err)
return
}
if n.tParser.Started && nr > 0 {
n.FilterSpecialCommand(buf[:nr])
switch {
case n.inSpecialStatus:
// 进入特殊命令状态，
case n.tParser.InputStatus:
n.tParser.CmdInputBuf.Write(buf[:nr])
case n.tParser.OutputStatus:
n.tParser.CmdOutputBuf.Write(buf[:nr])
default:
}
}
select { select {
case <-ctx.Done(): case <-ctx.Done():
log.Info("SendResponse finish by context done")
return return
case buf, ok := <-n.inChan: default:
copyBuf := make([]byte, len(buf[:nr]))
copy(copyBuf, buf[:nr])
outChan <- copyBuf
}
}
}
func (n *NodeConn) ReceiveRequest(ctx context.Context, inChan <-chan []byte, outChan chan<- []byte) {
defer n.Close()
for {
select {
case <-ctx.Done():
log.Error("ReceiveRequest finish by context done")
return
case buf, ok := <-inChan:
if !ok { if !ok {
log.Error("ReceiveRequest finish by inChan close")
return return
} }
...@@ -194,7 +237,7 @@ func (n *NodeConn) handleRequest(ctx context.Context) { ...@@ -194,7 +237,7 @@ func (n *NodeConn) handleRequest(ctx context.Context) {
n.currentCommandInput = n.tParser.ParseCommandInput() n.currentCommandInput = n.tParser.ParseCommandInput()
if n.FilterWhiteBlackRule(n.currentCommandInput) { if n.FilterWhiteBlackRule(n.currentCommandInput) {
msg := fmt.Sprintf("\r\n cmd '%s' is forbidden \r\n", n.currentCommandInput) msg := fmt.Sprintf("\r\n cmd '%s' is forbidden \r\n", n.currentCommandInput)
n.outChan <- []byte(msg) outChan <- []byte(msg)
ctrU := []byte{21, 13} // 清除行并换行 ctrU := []byte{21, 13} // 清除行并换行
_, err := n.stdin.Write(ctrU) _, err := n.stdin.Write(ctrU)
if err != nil { if err != nil {
...@@ -219,56 +262,12 @@ func (n *NodeConn) handleRequest(ctx context.Context) { ...@@ -219,56 +262,12 @@ func (n *NodeConn) handleRequest(ctx context.Context) {
n.tParser.InputStatus = true n.tParser.InputStatus = true
} }
_, _ = n.stdin.Write(buf) _, err := n.stdin.Write(buf)
if err != nil {
} log.Error("write conn err:", err)
} return
}
func (n *NodeConn) handleResponse(ctx context.Context) {
buf := make([]byte, maxBufferSize)
defer close(n.outChan)
for {
nr, err := n.stdout.Read(buf)
if err != nil {
return
}
if n.tParser.Started && nr > 0 {
n.FilterSpecialCommand(buf[:nr])
switch {
case n.inSpecialStatus:
// 进入特殊命令状态，
case n.tParser.InputStatus:
n.tParser.CmdInputBuf.Write(buf[:nr])
case n.tParser.OutputStatus:
n.tParser.CmdOutputBuf.Write(buf[:nr])
default:
} }
} }
select {
case <-ctx.Done():
return
default:
copyBuf := make([]byte, len(buf[:nr]))
copy(copyBuf, buf[:nr])
n.outChan <- copyBuf
}
}
}
func (n *NodeConn) Close() {
select {
case <-n.ctx.Done():
return
default:
_ = n.conn.Close()
_ = n.client.Close()
n.ctxCancelFunc()
} }
} }
pkg/transport/transport.go 0 → 100644
View file @ 216880ee
package transport
import (
"context"
uuid "github.com/satori/go.uuid"
)
type Agent interface {
ReceiveRequestChannel(ctx context.Context) chan<- []byte
SendResponseChannel(ctx context.Context) <-chan []byte
}
func NewMemoryAgent(nConn Conn) *memoryAgent {
m := &memoryAgent{
conn: nConn,
inChan: make(chan []byte),
outChan: make(chan []byte),
}
return m
}
type memoryAgent struct {
uuid uuid.UUID
conn Conn
inChan chan []byte
outChan chan []byte
}
func (m *memoryAgent) SendResponseChannel(ctx context.Context) <-chan []byte {
go m.conn.SendResponse(ctx, m.outChan)
return m.outChan
}
func (m *memoryAgent) ReceiveRequestChannel(ctx context.Context) chan<- []byte {
go m.conn.ReceiveRequest(ctx, m.inChan, m.outChan)
return m.inChan
}
pkg/userhome/conn.go 0 → 100644
View file @ 216880ee
package userhome
import (
"context"
"io"
"github.com/sirupsen/logrus"
"github.com/gliderlabs/ssh"
uuid "github.com/satori/go.uuid"
)
var log = logrus.New()
const maxBufferSize = 1024 * 4
type Conn interface {
SessionID() string
User() string
UUID() uuid.UUID
Pty() (ssh.Pty, <-chan ssh.Window, bool)
Context() context.Context
io.Reader
io.WriteCloser
}
pkg/sshd/sshconn.go → pkg/userhome/sshconn.go
View file @ 216880ee
package sshd package userhome
import ( import (
"context" "context"
...@@ -7,7 +7,13 @@ import ( ...@@ -7,7 +7,13 @@ import (
uuid "github.com/satori/go.uuid" uuid "github.com/satori/go.uuid"
) )
// ssh方式连接coco的用户request 将实现Conn接口 func NewSSHConn(sess ssh.Session) *SSHConn {
return &SSHConn{
conn: sess,
uuid: uuid.NewV4(),
}
}
type SSHConn struct { type SSHConn struct {
conn ssh.Session conn ssh.Session
uuid uuid.UUID uuid uuid.UUID
...@@ -44,7 +50,3 @@ func (s *SSHConn) Write(b []byte) (n int, err error) { ...@@ -44,7 +50,3 @@ func (s *SSHConn) Write(b []byte) (n int, err error) {
func (s *SSHConn) Close() error { func (s *SSHConn) Close() error {
return s.conn.Close() return s.conn.Close()
} }
// ws方式连接coco的用户request 将实现Conn接口
type WSConn struct {
}
pkg/core/userhome.go → pkg/userhome/userhome.go
View file @ 216880ee
package core package userhome
import ( import (
"context" "context"
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment