Merge pull request #18 from doronz88/feature/commands

Feature/commands

commands.md

@@ -283,7 +283,7 @@ optional arguments:
 ```
 ## function-lines
 ```
-usage: function-lines [-h]
+usage: function-lines [-h] [--after]
 
 get all function's lines
 
@@ -301,6 +301,7 @@ EXAMPLE:
 
 optional arguments:
   -h, --help  show this help message and exit
+  --after     include only function lines which occur after currentresultset
 ```
 ## function-start
 ```

fa/commands/function_lines.py

@@ -26,16 +26,23 @@ def get_parser():
     p = utils.ArgumentParserNoExit('function-lines',
                                    description=DESCRIPTION,
                                    formatter_class=RawTextHelpFormatter)
+    p.add_argument('--after', action='store_true',
+                   help='include only function lines which occur after current'
+                        'resultset')
     return p
 
 
 @context.ida_context
 @utils.yield_unique
-def function_lines(addresses):
+def function_lines(addresses, after=False):
     for address in addresses:
         for item in idautils.FuncItems(address):
-            yield item
+            if not after:
+                yield item
+            else:
+                if item > address:
+                    yield item
 
 
 def run(segments, args, addresses, interpreter=None, **kwargs):
-    return list(function_lines(addresses))
+    return list(function_lines(addresses, args.after))

scripts/git/pre-commit

@@ -17,7 +17,10 @@ def main():
                            'stops parsing current SIG if current ' \
                            'resultset is empty\n'
 
-    for filename in os.listdir(COMMANDS_ROOT):
+    commands = os.listdir(COMMANDS_ROOT)
+    commands.sort()
+
+    for filename in commands:
         if filename.endswith('.py') and \
                 filename not in ('__init__.py',):
             command = os.path.splitext(filename)[0]