Exemplo n.º 1
0
 def __init__(self):
     Cmd.__init__(self)
     self.prompt = "hephaestus> "
     self.intro = (
         "\n"
         + ruler("*")
         + "WELCOME to hephaestus\n"
         + ruler("*")
         + "hephaestus is the stateless firewall configuration parser for JunOS.\nGet help with 'help (<cmd>)' or '?', or issue 'userguide' to get started.\n"
         + ruler("*")
         + "\n"
     )
     self.myconfig = Baseconfig()
     self.configinterpreter = Interpreter_config()
     self.configinterpreter.myconfig = self.myconfig
     self.scaninterpreter = Interpreter_scan()
     self.scaninterpreter.myconfig = self.myconfig
     self.updateinterpreter = Interpreter_update()
     self.updateinterpreter.myconfig = self.myconfig
     self.queryinterpreter = Interpreter_query()
     self.queryinterpreter.myconfig = self.myconfig
     self.resultsinterpreter = Interpreter_results()
     self.resultsinterpreter.myconfig = self.myconfig
Exemplo n.º 2
0
class Interpreter(Cmd):
    """Interpreter class based on the 'cmd' library."""

    def __init__(self):
        Cmd.__init__(self)
        self.prompt = "hephaestus> "
        self.intro = (
            "\n"
            + ruler("*")
            + "WELCOME to hephaestus\n"
            + ruler("*")
            + "hephaestus is the stateless firewall configuration parser for JunOS.\nGet help with 'help (<cmd>)' or '?', or issue 'userguide' to get started.\n"
            + ruler("*")
            + "\n"
        )
        self.myconfig = Baseconfig()
        self.configinterpreter = Interpreter_config()
        self.configinterpreter.myconfig = self.myconfig
        self.scaninterpreter = Interpreter_scan()
        self.scaninterpreter.myconfig = self.myconfig
        self.updateinterpreter = Interpreter_update()
        self.updateinterpreter.myconfig = self.myconfig
        self.queryinterpreter = Interpreter_query()
        self.queryinterpreter.myconfig = self.myconfig
        self.resultsinterpreter = Interpreter_results()
        self.resultsinterpreter.myconfig = self.myconfig

    def setconfig(self, cfg):
        self.myconfig = cfg

    def emptyline(self):
        pass

    def do_shell(self, line):
        "Run a shell command"
        print("\nRunning shell command: " + line + "\n")
        output = os.popen(line).read()
        print(output)
        self.last_output = output

    def do_exit(self, line):
        return True

    def help_exit(self):
        print("\nExit this program.\n")

    def help_help(self):
        print("\nOverride worked...\n")

    def help_shell(self):
        print("\nOverride worked...\n")

    # interesting functions
    def do_userguide(self, line):
        """

    A punchy, pithy user guide will go here soon. For now:

    Basics:

      1. Enter Configuration mode, load a file, show contents, exit
           config > load > show > exit
      2. Enter Scan mode, autoscan the network
           scan > autoscan
      3. Enter Update mode, update local cache of router configs
           update > update
      4. Enter Query mode, run query
           query > query
      5. Enter Results mode, view results
           results > summary
           results > diagram
           results > detail
    """
        self.help_userguide()

    def help_userguide(self):
        """Print help on do_userguide() method"""
        print
        print(inspect.getdoc(self.do_userguide))
        print

    def do_config(self, line):
        self.configinterpreter.cmdloop()

    def help_config(self):
        print(
            "\nBefore you can scan a network or run any queries, you must first enter configuration to enter the necessary AS/network information.\n"
        )

    def do_scan(self, line):
        self.scaninterpreter.cmdloop()

    def help_scan(self):
        print("\nScan the configured network and identify any routers present.\n")

    def do_update(self, line):
        self.updateinterpreter.cmdloop()

    def help_update(self):
        print("\nFetch configuration from the identified routers.\n")

    def do_query(self, line):
        self.queryinterpreter.cmdloop()

    def help_query(self):
        print(
            "\nRun a query against the network state to determine if a given flow will be accepted, rejected or discarded.\n"
        )

    def do_results(self, line):
        self.resultsinterpreter.cmdloop()

    def help_results(self):
        print("\nView results in more detail.\n")