Fix migrate too

BishopFox · Dec 2, 2020 · 4524e44 · 4524e44
1 parent c70ea89
commit 4524e44
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/server/rpc/rpc-tasks.go b/server/rpc/rpc-tasks.go
@@ -60,7 +60,13 @@ func (rpc *Server) Migrate(ctx context.Context, req *clientpb.MigrateReq) (*sliv
 	name := path.Base(req.Config.GetName())
 	shellcode, err := getSliverShellcode(name)
 	if err != nil {
-		_, config := generate.ImplantConfigFromProtobuf(req.Config)
+		name, config := generate.ImplantConfigFromProtobuf(req.Config)
+		if name == "" {
+			name, err = generate.GetCodename()
+			if err != nil {
+				return nil, err
+			}
+		}
 		config.Format = clientpb.ImplantConfig_SHELLCODE
 		config.ObfuscateSymbols = false
 		shellcodePath, err := generate.SliverShellcode(name, config)