Merge pull request #1530 from BishopFox/sgn-stage-listener

Encode Shellcode with SGN when generating via stage-listener
BishopFox · Dec 30, 2023 · 04e869c · 04e869c
2 parents b4cf848 + 32a6045
commit 04e869c
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/client/command/generate/helpers.go b/client/command/generate/helpers.go
@@ -28,6 +28,22 @@ func GetSliverBinary(profile *clientpb.ImplantProfile, con *console.SliverConsol
 		return data, err
 	}
 	data = generated.GetFile().GetData()
+
+	if profile.Config.Format == clientpb.OutputFormat_SHELLCODE && profile.Config.SGNEnabled {
+		encodeResp, err := con.Rpc.ShellcodeEncoder(context.Background(), &clientpb.ShellcodeEncodeReq{
+			Encoder:      clientpb.ShellcodeEncoder_SHIKATA_GA_NAI,
+			Architecture: profile.Config.GOARCH,
+			Iterations:   1,
+			BadChars:     []byte{},
+			Data:         data,
+		})
+		if err != nil {
+			con.PrintErrorf("Error encoding shellcode")
+			return nil, err
+		}
+		data = encodeResp.Data
+	}
+
 	_, err = con.Rpc.SaveImplantProfile(context.Background(), profile)
 	if err != nil {
 		con.PrintErrorf("Error updating implant profile\n")