mirror of
https://github.com/MrAle98/sliver
synced 2026-06-08 11:54:46 +00:00
moloch
121 lines
2.7 KiB
Go
121 lines
2.7 KiB
Go
package command
|
|
|
|
import (
|
|
clientpb "sliver/protobuf/client"
|
|
sliverpb "sliver/protobuf/sliver"
|
|
"strconv"
|
|
"time"
|
|
|
|
"github.com/AlecAivazis/survey"
|
|
"github.com/golang/protobuf/proto"
|
|
)
|
|
|
|
const (
|
|
// ANSI Colors
|
|
normal = "\033[0m"
|
|
black = "\033[30m"
|
|
red = "\033[31m"
|
|
green = "\033[32m"
|
|
orange = "\033[33m"
|
|
blue = "\033[34m"
|
|
purple = "\033[35m"
|
|
cyan = "\033[36m"
|
|
gray = "\033[37m"
|
|
bold = "\033[1m"
|
|
clearln = "\r\x1b[2K"
|
|
upN = "\033[%dA"
|
|
downN = "\033[%dB"
|
|
underline = "\033[4m"
|
|
|
|
// Info - Display colorful information
|
|
Info = bold + cyan + "[*] " + normal
|
|
// Warn - Warn a user
|
|
Warn = bold + red + "[!] " + normal
|
|
// Debug - Display debug information
|
|
Debug = bold + purple + "[-] " + normal
|
|
// Woot - Display success
|
|
Woot = bold + green + "[$] " + normal
|
|
)
|
|
|
|
var (
|
|
|
|
// ActiveSliver - The current sliver we're interacting with
|
|
ActiveSliver = &activeSliver{
|
|
observers: []observer{},
|
|
}
|
|
|
|
defaultTimeout = 30 * time.Second
|
|
stdinReadTimeout = 10 * time.Millisecond
|
|
)
|
|
|
|
// RPCServer - Function used to send/recv envelopes
|
|
type RPCServer func(*sliverpb.Envelope, time.Duration) chan *sliverpb.Envelope
|
|
|
|
type observer func()
|
|
|
|
type activeSliver struct {
|
|
Sliver *clientpb.Sliver
|
|
observers []observer
|
|
}
|
|
|
|
func (s *activeSliver) AddObserver(fn observer) {
|
|
s.observers = append(s.observers, fn)
|
|
}
|
|
|
|
func (s *activeSliver) SetActiveSliver(sliver *clientpb.Sliver) {
|
|
s.Sliver = sliver
|
|
for _, fn := range s.observers {
|
|
fn()
|
|
}
|
|
}
|
|
|
|
func (s *activeSliver) DisableActiveSliver() {
|
|
s.Sliver = nil
|
|
for _, fn := range s.observers {
|
|
fn()
|
|
}
|
|
}
|
|
|
|
// Get Sliver by session ID or name
|
|
func getSliver(arg string, rpc RPCServer) *clientpb.Sliver {
|
|
resp := <-rpc(&sliverpb.Envelope{
|
|
Type: clientpb.MsgSessions,
|
|
Data: []byte{},
|
|
}, defaultTimeout)
|
|
sessions := &clientpb.Sessions{}
|
|
proto.Unmarshal((resp).Data, sessions)
|
|
|
|
for _, sliver := range sessions.Slivers {
|
|
if strconv.Itoa(int(sliver.ID)) == arg || sliver.Name == arg {
|
|
return sliver
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// SliverSessionsByName - Return all sessions for a Sliver by name
|
|
func SliverSessionsByName(name string, rpc RPCServer) []*clientpb.Sliver {
|
|
resp := <-rpc(&sliverpb.Envelope{
|
|
Type: clientpb.MsgSessions,
|
|
Data: []byte{},
|
|
}, defaultTimeout)
|
|
allSessions := &clientpb.Sessions{}
|
|
proto.Unmarshal((resp).Data, allSessions)
|
|
|
|
sessions := []*clientpb.Sliver{}
|
|
for _, sliver := range allSessions.Slivers {
|
|
if sliver.Name == name {
|
|
sessions = append(sessions, sliver)
|
|
}
|
|
}
|
|
return sessions
|
|
}
|
|
|
|
// This should be called for any dangerous (OPSEC-wise) functions
|
|
func isUserAnAdult() bool {
|
|
confirm := false
|
|
prompt := &survey.Confirm{Message: "This action is bad OPSEC, are you an adult?"}
|
|
survey.AskOne(prompt, &confirm, nil)
|
|
return confirm
|
|
}
|