justin/sshrpc
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Added ssh Connection to callback so additional info is available about connection

Browse Source
This commit is contained in:
Justin 2015-04-07 13:11:27 -04:00
parent 91d8c1f8ab
commit b1cb51d887
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
server.go
View File

@ -16,7 +16,7 @@ const DefaultRPCChannel = "RPCChannel"
const RPCSubsystem = "RPCSubsystem" const RPCSubsystem = "RPCSubsystem"
// CallbackFunc to be called when reverse RPC client is created // CallbackFunc to be called when reverse RPC client is created
type CallbackFunc func(RPCClient *rpc.Client) type CallbackFunc func(RPCClient *rpc.Client, conn ssh.Conn)
// Server represents an SSH Server that spins up RPC servers when requested. // Server represents an SSH Server that spins up RPC servers when requested.
type Server struct { type Server struct {
@ -122,7 +122,7 @@ func (s *Server) handleChannels(chans <-chan ssh.NewChannel, sshConn ssh.Conn) {
} }
rpcClient := rpc.NewClient(clientChannel) rpcClient := rpc.NewClient(clientChannel)
if s.CallbackFunc != nil { if s.CallbackFunc != nil {
s.CallbackFunc(rpcClient) s.CallbackFunc(rpcClient, sshConn)
} }
log.Printf("Started SSH RPC client") log.Printf("Started SSH RPC client")
default: default: