-
Notifications
You must be signed in to change notification settings - Fork 0
/
socks.go
86 lines (71 loc) · 2.03 KB
/
socks.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
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
package core
/*
Sliver Implant Framework
Copyright (C) 2021 Bishop Fox
This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program. If not, see <https://www.gnu.org/licenses/>.
*/
import (
"sync"
"github.com/starkzarn/glod/protobuf/rpcpb"
"github.com/starkzarn/glod/protobuf/glodpb"
)
var (
// TunSocksTunnels - Interacting with duplex SocksTunnels
SocksTunnels = tcpTunnel{
tunnels: map[uint64]*TcpTunnel{},
mutex: &sync.Mutex{},
}
)
type TcpTunnel struct {
ID uint64
SessionID string
ToImplantSequence uint64
ToImplantMux sync.Mutex
FromImplant chan *glodpb.SocksData
FromImplantSequence uint64
Client rpcpb.SliverRPC_SocksProxyServer
}
type tcpTunnel struct {
tunnels map[uint64]*TcpTunnel
mutex *sync.Mutex
}
func (t *tcpTunnel) Create(sessionID string) *TcpTunnel {
tunnelID := NewTunnelID()
session := Sessions.Get(sessionID)
tunnel := &TcpTunnel{
ID: tunnelID,
SessionID: session.ID,
//ToImplant: make(chan []byte),
FromImplant: make(chan *glodpb.SocksData),
}
t.mutex.Lock()
defer t.mutex.Unlock()
t.tunnels[tunnel.ID] = tunnel
return tunnel
}
func (t *tcpTunnel) Close(tunnelID uint64) error {
t.mutex.Lock()
defer t.mutex.Unlock()
tunnel := t.tunnels[tunnelID]
if tunnel == nil {
return ErrInvalidTunnelID
}
delete(t.tunnels, tunnelID)
//close(tunnel.ToImplant)
close(tunnel.FromImplant)
return nil
}
func (t *tcpTunnel) Get(tunnelID uint64) *TcpTunnel {
t.mutex.Lock()
defer t.mutex.Unlock()
return t.tunnels[tunnelID]
}