Merge pull request #13 from s0lst1c3/add-feature-module-maker

Add feature module maker

Author: s0lst1c3

Changelog

@@ -1,2 +1,5 @@
 0.0.1 - s0lst1c3 <gabriel@solstice.sh>
 Initial release.
+
+0.1.0 - s0lst1c3 <gabriel@solstice.sh>
+Added module maker.

README.md

@@ -2,7 +2,7 @@
 
 by [@s0lst1c3](https://twitter.com/s0lst1c3)
 
-Current release: [v0.0.1-alpha](https://github.com/s0lst1c3/dropengine/releases/tag/v0.0.1-alpha)
+Current release: [v0.0.1-alpha](https://github.com/s0lst1c3/dropengine/releases/tag/v0.1.1-alpha)
 
 Supports _Python 3.7+_.
 

__version__.py

@@ -1,4 +1,4 @@
-__version__ = '0.0.1'
+__version__ = '0.1.1'
 __codename__ = 'lambsauce'
 __author__ = '@s0lst1c3'
 __contact__ = 'gabriel<<<at>>solstice(dot))sh'

core/dispatcher_cli.py

@@ -633,8 +633,6 @@ def add_arguments(parser):
                                 default=None,
                                 choices=Dispatcher.get_choices('./modules/input/crypters', 'MCrypter'),
                                 help='Select crypter')
-
-
 
         modules_group.add_argument('--interface',
                                 dest='interface',

csharp_ini_maker.py

@@ -0,0 +1,109 @@
+import sys
+import argparse
+
+from antlr4 import *
+from module_maker.CSharpLexer import CSharpLexer
+from module_maker.CSharpParser import CSharpParser
+from module_maker.CSharpParserListener import CSharpParserListener
+
+
+from module_maker.symbol_ini import SymbolINI
+
+
+OMODULE_TYPES = [
+
+    'decrypter',
+    'dkey',
+    'executor',
+    'imports',
+    'inner_shell',
+    'output_modules',
+    'payload_main',
+    'postmodule',
+    'premodule',
+    'runner',
+]
+
+IMODULE_TYPES = [
+
+]
+
+MODULE_DIRECTIONS = [
+
+    'output',
+    'input',
+]
+
+def cli():
+
+    parser = argparse.ArgumentParser()
+
+    parser.add_argument('--input-file', '-i',
+                        dest='input_file',
+                        type=str,
+                        required=True,
+                        help='Specify source code file to input')
+
+    parser.add_argument('--output-file', '-o',
+                        dest='output_file',
+                        type=str,
+                        required=True,
+                        help='Specify name for output INI file')
+
+    args = parser.parse_args()
+
+    return args
+
+def gather(args):
+
+    input_handle = FileStream(args.input_file)
+    lexer = CSharpLexer(input_handle)
+    tokens = CommonTokenStream(lexer)
+    parser = CSharpParser(tokens)
+    tree = parser.compilation_unit()
+
+    listener = CSharpParserListener()
+    walker = ParseTreeWalker()
+    walker.walk(listener, tree)
+
+    with open(args.output_file, 'w') as output_handle:
+
+        output_handle.write('[vars]\n')
+        for v in listener.vars:
+            output_handle.write(v+'\n')
+        output_handle.write('\n')
+
+        output_handle.write('[methods]\n')
+        for v in listener.methods:
+            output_handle.write(v+'\n')
+        output_handle.write('\n')
+
+        output_handle.write('[class_decls]\n')
+        for v in listener.class_decls:
+            output_handle.write(v+'\n')
+        output_handle.write('\n')
+
+        output_handle.write('[params]\n')
+        for v in listener.params:
+            output_handle.write(v+'\n')
+        output_handle.write('\n')
+
+        output_handle.write('[delegates]\n')
+        for v in listener.delegates:
+            output_handle.write(v+'\n')
+        output_handle.write('\n')
+
+        output_handle.write('[imports]\n')
+        for v in listener.imports:
+            output_handle.write(v+'\n')
+        output_handle.write('\n')
+
+def main():
+
+    args = cli()
+
+    gather(args)
+
+if __name__ == '__main__':
+    cli()
+    main()