Commit 5ab25d16 by devttys0

Updated binwalk.__init__.py

parent da583ef4
...@@ -8,6 +8,10 @@ import binwalk.config ...@@ -8,6 +8,10 @@ import binwalk.config
import binwalk.plugin import binwalk.plugin
from binwalk.compat import * from binwalk.compat import *
class Type(object):
Primary = 1
Support = 2
class ModuleOption(object): class ModuleOption(object):
''' '''
A container class that allows modules to declare command line options. A container class that allows modules to declare command line options.
...@@ -114,6 +118,9 @@ class Module(object): ...@@ -114,6 +118,9 @@ class Module(object):
# The module title, as displayed in help output # The module title, as displayed in help output
TITLE = "" TITLE = ""
# Default type is primary
TYPE = Types.Primary
# A list of binwalk.module.ModuleOption command line options # A list of binwalk.module.ModuleOption command line options
CLI = [] CLI = []
...@@ -417,6 +424,7 @@ class Modules(object): ...@@ -417,6 +424,7 @@ class Modules(object):
self._set_arguments(list(args), kwargs) self._set_arguments(list(args), kwargs)
for module in self.list(): for module in self.list():
if module.TYPE == Type.Primary:
obj = self.run(module) obj = self.run(module)
if obj.enabled: if obj.enabled:
run_modules.append(obj) run_modules.append(obj)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment