Merge pull request #198 from SYANiDE-/main

Add shell command history feature (readline)

Author: aniqfakhrul

powerview/__init__.py

@@ -9,6 +9,7 @@
 from powerview.utils.parsers import powerview_arg_parse, arg_parse
 from powerview.utils.shell import get_prompt
 from powerview.utils.colors import bcolors, Gradient
+from powerview.utils.history import *
 
 import ldap3
 import random
@@ -758,13 +759,21 @@ def main():
                                         powerview.remove_gplink(guid=pv_args.guid, targetidentity=pv_args.targetidentity, args=pv_args)
                                 else:
                                     logging.error("-GUID and -TargetIdentity flags are required")
+                            elif pv_args.module.casefold() == 'history':
+                                hist = get_shell_history(pv_args.last, pv_args.unique)
+                                for index, item in list(enumerate(hist,1))[::-1]:
+                                    if pv_args.noNumber:
+                                        bol = f""
+                                    else:
+                                        bol = f"[{index}] "
+                                    print(f"{bol}{item}")
+                            elif pv_args.module.casefold() == 'clear':
+                                clear_screen()
                             elif pv_args.module.casefold() == 'exit':
                                 if mcp_running:
                                     powerview.mcp_server.stop()
                                 log_handler.save_history()
                                 sys.exit(0)
-                            elif pv_args.module.casefold() == 'clear':
-                                clear_screen()
 
                             if entries:
                                 if pv_args.outfile:

powerview/utils/completer.py

@@ -129,6 +129,7 @@
     'Stop-NetService':['-Computer','-Name'],
     'Add-GPLink':['-GUID','-TargetIdentity','-LinkEnabled','-Enforced','-SearchBase','-Server','-OutFile'],
     'Remove-GPLink':['-GUID','-TargetIdentity','-SearchBase','-Server','-OutFile'],
+    'history':['-Last','-Unique','-NoNumber'],
     'clear':'',
     'exit':'',
 }

powerview/utils/history.py

@@ -0,0 +1,22 @@
+from sys import platform
+if platform == "linux" or platform == "linux2":
+    import gnureadline as readline
+else:
+    import readline
+
+def get_shell_history(last,unique):
+    '''return the last x number of {,unique} shell history'''
+    hist_len = readline.get_current_history_length()-1
+    if last > hist_len:
+        last = hist_len
+    returnable_hist_items = []
+    cursor = hist_len
+    while (not len(returnable_hist_items) == last) and (not cursor == 1):
+        analyze_this = readline.get_history_item(cursor)
+        if unique == True:
+            if not analyze_this in returnable_hist_items:
+                returnable_hist_items.append(analyze_this)
+        else:
+            returnable_hist_items.append(analyze_this)
+        cursor = cursor - 1
+    return returnable_hist_items

powerview/utils/parsers.py

@@ -1101,8 +1101,14 @@ def powerview_arg_parse(cmd):
 	remove_gplink_parser.add_argument('-Server', action='store', dest='server')
 	remove_gplink_parser.add_argument('-OutFile', action='store', dest='outfile')
 
-	subparsers.add_parser('exit', exit_on_error=False)
+	# shell history
+	history_parser = subparsers.add_parser('history', exit_on_error=False)
+	history_parser.add_argument('-Last',action='store',type=int, default=10, dest='last')
+	history_parser.add_argument('-Unique',action='store_true', dest='unique')
+	history_parser.add_argument('-NoNumber',action='store_true', dest='noNumber')
+
 	subparsers.add_parser('clear', exit_on_error=False)
+	subparsers.add_parser('exit', exit_on_error=False)
 
 	try:
 		args, unknown = parser.parse_known_args(cmd)