Function main

dshell/decode.py:166–372

(plugin_args=None, **kwargs)

def main(plugin_args=None, **kwargs):
    global plugin_chain

    if not plugin_args:
        plugin_args = {}

    # dictionary of all available plugins: {name: module path}
    plugin_map = get_plugins()

    # Attempt to catch segfaults caused when certain linktypes (e.g. 204) are
    # given to pcapy
    faulthandler.enable()

    if not plugin_chain:
        logger.error("No plugin selected")
        sys.exit(1)

    plugin_chain[0].defrag_ip = kwargs.get("defrag", False)

    # Setup logging
    log_format = "%(levelname)s (%(name)s) - %(message)s"
    if kwargs.get("verbose", False):
        log_level = logging.INFO
    elif kwargs.get("debug", False):
        log_level = logging.DEBUG
    elif kwargs.get("quiet", False):
        log_level = logging.CRITICAL
    else:
        log_level = logging.WARNING
    logging.basicConfig(format=log_format, level=log_level)

    # since pypacker handles its own exceptions (loudly), this attempts to keep
    # it quiet
    logging.getLogger("pypacker").setLevel(logging.CRITICAL)

    if kwargs.get("allcc", False):
        # Activate all country code (allcc) mode to display all 3 GeoIP2 country
        # codes
        dshell.core.geoip.acc = True

    dshell.core.geoip.check_file_dates()

    # If alternate output module is selected, tell each plugin to use that
    # instead
    if kwargs.get("omodule", None):
        try:
            # TODO: Create a factory classmethod in the base Output class (e.g. "from_name()") instead.
            omodule = import_module("dshell.output."+kwargs["omodule"])
            omodule = omodule.obj
            for plugin in plugin_chain:
                # TODO: Should we have a single instance of the Output module used by all plugins?
                oomodule = omodule()
                plugin.out = oomodule
        except ImportError as e:
            logger.error("Could not import module named '{}'. Use --list-output flag to see available modules".format(kwargs["omodule"]))
            sys.exit(1)

    # Check if any user-defined output arguments are provided