Adding attribute supported_os to remaining commands

Payload_Type/cazalla/cazalla/agent_functions/download.py

@@ -35,6 +35,9 @@ class DownloadCommand(CommandBase):
     supported_ui_features = ["file_browser:download"]
     author = "@KaseyaOFSTeam"
     argument_class = DownloadArguments
+    attributes = CommandAttributes(
+        supported_os=[SupportedOS.Windows]
+    )
 
     async def opsec_pre(self, taskData: PTTaskMessageAllData) -> PTTTaskOPSECPreTaskMessageResponse:
         """

Payload_Type/cazalla/cazalla/agent_functions/exit.py

@@ -25,6 +25,9 @@ class ExitCommand(CommandBase):
     author = "Kaseya OFSTeam"
     argument_class = ExitArguments
     attackmapping = []
+    attributes = CommandAttributes(
+        supported_os=[SupportedOS.Windows]
+    )
 
     async def create_go_tasking(self, taskData: PTTaskMessageAllData) -> PTTaskCreateTaskingMessageResponse:
         response = PTTaskCreateTaskingMessageResponse(

Payload_Type/cazalla/cazalla/agent_functions/inline_execute_assembly.py

@@ -266,7 +266,8 @@ class InlineExecuteAssemblyCommand(CommandBase):
     completion_functions = {"coff_completion_callback": coff_completion_callback}
     attributes = CommandAttributes(
         dependencies=["inline_execute"],
-        alias=True
+        alias=True,
+        supported_os=[SupportedOS.Windows]
     )
 
     async def opsec_pre(self, taskData: PTTaskMessageAllData) -> PTTTaskOPSECPreTaskMessageResponse:

Payload_Type/cazalla/cazalla/agent_functions/kill.py

@@ -73,6 +73,9 @@ class KillCommand(CommandBase):
     attackmapping = ["T1489"]
     supported_ui_features = ["process_browser:kill"]
     argument_class = KillArguments
+    attributes = CommandAttributes(
+        supported_os=[SupportedOS.Windows]
+    )
 
     async def opsec_pre(self, taskData: PTTaskMessageAllData) -> PTTTaskOPSECPreTaskMessageResponse:
         """

Payload_Type/cazalla/cazalla/agent_functions/list_tokens.py

@@ -52,6 +52,9 @@ class ListTokensCommand(CommandBase):
     author = "@KaseyaOFSTeam"
     supported_ui_features = ["process_browser:list_tokens"]
     argument_class = ListTokensArguments
+    attributes = CommandAttributes(
+        supported_os=[SupportedOS.Windows]
+    )
 
     async def opsec_pre(self, taskData: PTTaskMessageAllData) -> PTTTaskOPSECPreTaskMessageResponse:
         """

Payload_Type/cazalla/cazalla/agent_functions/make_token.py

@@ -75,6 +75,9 @@ class MakeTokenCommand(CommandBase):
     author = "@KaseyaOFSTeam"
     attackmapping = ["T1134.003"]
     argument_class = MakeTokenArguments
+    attributes = CommandAttributes(
+        supported_os=[SupportedOS.Windows]
+    )
 
     async def opsec_pre(self, taskData: PTTaskMessageAllData) -> PTTTaskOPSECPreTaskMessageResponse:
         """

Payload_Type/cazalla/cazalla/agent_functions/rev2self.py

@@ -23,6 +23,9 @@ class Rev2SelfCommand(CommandBase):
     author = "@KaseyaOFSTeam"
     attackmapping = ["T1134.001"]
     argument_class = Rev2SelfArguments
+    attributes = CommandAttributes(
+        supported_os=[SupportedOS.Windows]
+    )
 
     async def opsec_pre(self, taskData: PTTaskMessageAllData) -> PTTTaskOPSECPreTaskMessageResponse:
         """

Payload_Type/cazalla/cazalla/agent_functions/screenshot.py

@@ -25,6 +25,9 @@ class ScreenshotCommand(CommandBase):
     author = "@KaseyaOFSTeam"
     argument_class = ScreenshotArguments
     attackmapping = ["T1113"]
+    attributes = CommandAttributes(
+        supported_os=[SupportedOS.Windows]
+    )
 
     async def opsec_pre(self, taskData: PTTaskMessageAllData) -> PTTTaskOPSECPreTaskMessageResponse:
         """

Payload_Type/cazalla/cazalla/agent_functions/shell.py

@@ -36,7 +36,7 @@ class ShellCommand(CommandBase):
     attackmapping = ["T1059"]
     argument_class = ShellArguments
     attributes = CommandAttributes(
-        supported_os=[SupportedOS.MacOS, SupportedOS.Linux, SupportedOS.Windows]
+        supported_os=[SupportedOS.Windows]
     )
 
     async def opsec_pre(self, taskData: PTTaskMessageAllData) -> PTTTaskOPSECPreTaskMessageResponse:

Payload_Type/cazalla/cazalla/agent_functions/steal_token.py

@@ -73,6 +73,9 @@ class StealTokenCommand(CommandBase):
     attackmapping = ["T1134.001"]
     supported_ui_features = ["process_browser:steal_token"]
     argument_class = StealTokenArguments
+    attributes = CommandAttributes(
+        supported_os=[SupportedOS.Windows]
+    )
 
     async def opsec_pre(self, taskData: PTTaskMessageAllData) -> PTTTaskOPSECPreTaskMessageResponse:
         """

Payload_Type/cazalla/cazalla/agent_functions/upload.py

@@ -66,6 +66,9 @@ class UploadCommand(CommandBase):
     supported_ui_features = ["file_browser:upload"]
     author = "@KaseyaOFSTeam"
     argument_class = UploadArguments
+    attributes = CommandAttributes(
+        supported_os=[SupportedOS.Windows]
+    )
 
     async def opsec_pre(self, taskData: PTTaskMessageAllData) -> PTTTaskOPSECPreTaskMessageResponse:
         """

Payload_Type/cazalla/cazalla/agent_functions/whoami.py

@@ -23,6 +23,9 @@ class WhoamiCommand(CommandBase):
     author = "@KaseyaOFSTeam"
     attackmapping = ["T1083"]
     argument_class = WhoamiArguments
+    attributes = CommandAttributes(
+        supported_os=[SupportedOS.Windows]
+    )
 
     async def create_go_tasking(self, taskData: PTTaskMessageAllData) -> PTTaskCreateTaskingMessageResponse:
         response = PTTaskCreateTaskingMessageResponse(