mirror of
https://github.com/BishopFox/sliver
synced 2026-06-08 10:29:05 +00:00
64 lines
1.9 KiB
Go
64 lines
1.9 KiB
Go
package operator
|
|
|
|
/*
|
|
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 (
|
|
"context"
|
|
|
|
"github.com/bishopfox/sliver/client/console"
|
|
"github.com/bishopfox/sliver/client/prelude"
|
|
"github.com/bishopfox/sliver/protobuf/commonpb"
|
|
"github.com/desertbit/grumble"
|
|
)
|
|
|
|
func ConnectCmd(ctx *grumble.Context, con *console.SliverConsoleClient) {
|
|
url := ctx.Args.String("connection-string")
|
|
aesKey := ctx.Flags.String("aes-key")
|
|
agentRange := ctx.Flags.String("range")
|
|
skipExisting := ctx.Flags.Bool("skip-existing")
|
|
|
|
config := &prelude.OperatorConfig{
|
|
Range: agentRange,
|
|
OperatorURL: url,
|
|
RPC: con.Rpc,
|
|
AESKey: aesKey,
|
|
}
|
|
|
|
sessionMapper := prelude.InitSessionMapper(config)
|
|
|
|
con.PrintInfof("Connected to Operator at %s\n", url)
|
|
if !skipExisting {
|
|
sessions, err := con.Rpc.GetSessions(context.Background(), &commonpb.Empty{})
|
|
if err != nil {
|
|
con.PrintErrorf("Could not get session list: %s", err)
|
|
return
|
|
}
|
|
if len(sessions.Sessions) > 0 {
|
|
con.PrintInfof("Adding existing sessions ...\n")
|
|
for _, session := range sessions.Sessions {
|
|
err = sessionMapper.AddSession(session)
|
|
if err != nil {
|
|
con.PrintErrorf("Could not add session %s to session mapper: %s", session.Name, err)
|
|
}
|
|
}
|
|
con.PrintInfof("Done !\n")
|
|
}
|
|
}
|
|
}
|