@@ -28,6 +28,7 @@ import (
28
28
"github.com/AlecAivazis/survey/v2"
29
29
"github.com/bishopfox/sliver/client/command/environment"
30
30
"github.com/bishopfox/sliver/client/command/exec"
31
+ "github.com/bishopfox/sliver/client/command/extensions"
31
32
"github.com/bishopfox/sliver/client/command/filesystem"
32
33
"github.com/bishopfox/sliver/client/command/network"
33
34
"github.com/bishopfox/sliver/client/command/privilege"
@@ -212,9 +213,23 @@ func renderTaskResponse(task *clientpb.BeaconTask, con *console.SliverConsoleCli
212
213
}
213
214
environment .PrintUnsetEnvInfo (unsetEnvReq .Name , unsetEnv , con )
214
215
216
+ // ---------------------
217
+ // Call extension commands
218
+ // ---------------------
219
+ case sliverpb .MsgCallExtensionReq :
220
+ callExtension := & sliverpb.CallExtension {}
221
+ err := proto .Unmarshal (task .Response , callExtension )
222
+ if err != nil {
223
+ con .PrintErrorf ("Failed to decode task response: %s\n " , err )
224
+ return
225
+ }
226
+ extensions .PrintExtOutput ("" , "" , callExtension , con )
227
+
215
228
// ---------------------
216
229
// Exec commands
217
230
// ---------------------
231
+ case sliverpb .MsgInvokeExecuteAssemblyReq :
232
+ fallthrough
218
233
case sliverpb .MsgInvokeInProcExecuteAssemblyReq :
219
234
fallthrough
220
235
case sliverpb .MsgExecuteAssemblyReq :
@@ -235,6 +250,7 @@ func renderTaskResponse(task *clientpb.BeaconTask, con *console.SliverConsoleCli
235
250
Flags : grumble.FlagMap {
236
251
"save" : & grumble.FlagMapItem {Value : false , IsDefault : true },
237
252
"loot" : & grumble.FlagMapItem {Value : false , IsDefault : true },
253
+ "name" : & grumble.FlagMapItem {Value : "" , IsDefault : true },
238
254
},
239
255
}
240
256
exec .HandleExecuteAssemblyResponse (execAssembly , assemblyPath , hostname , ctx , con )
0 commit comments