Add argcomplete functionality

Author: AdrianVollmer

examples/DumpNTLMInfo.py

@@ -1,4 +1,5 @@
 #!/usr/bin/env python
+# PYTHON_ARGCOMPLETE_OK
 # Impacket - Collection of Python classes for working with network protocols.
 #
 # Copyright Fortra, LLC and its affiliated companies 
@@ -37,6 +38,7 @@
 import os
 import sys
 import argparse
+import argcomplete
 import logging
 import struct
 import socket
@@ -644,6 +646,8 @@ def __convert_size(self, size_bytes):
     parser.add_argument('-protocol', choices=['SMB', 'RPC'], nargs='?', metavar="protocol",
                         help='Protocol to use (SMB or RPC). Default is SMB, port 135 uses RPC normally.')
 
+    argcomplete.autocomplete(parser, always_complete_options=False)
+
     if len(sys.argv) == 1:
         parser.print_help()
         sys.exit(1)

examples/Get-GPPPassword.py

@@ -1,4 +1,5 @@
 #!/usr/bin/env python3
+# PYTHON_ARGCOMPLETE_OK
 # Impacket - Collection of Python classes for working with network protocols.
 #
 # Copyright Fortra, LLC and its affiliated companies 
@@ -20,6 +21,7 @@
 #
 
 import argparse
+import argcomplete
 import base64
 import xml
 import charset_normalizer
@@ -227,6 +229,8 @@ def parse_args():
     group.add_argument("-target-ip", action="store", metavar="ip address", help="IP Address of the target machine. If omitted it will use whatever was specified as target. This is useful when target is the NetBIOS name and you cannot resolve it")
     group.add_argument("-port", choices=["139", "445"], nargs="?", default="445", metavar="destination port", help="Destination port to connect to SMB Server")
 
+    argcomplete.autocomplete(parser, always_complete_options=False)
+
     if len(sys.argv) == 1:
         parser.print_help()
         sys.exit(1)

examples/GetADComputers.py

@@ -1,4 +1,5 @@
 #!/usr/bin/env python
+# PYTHON_ARGCOMPLETE_OK
 # Impacket - Collection of Python classes for working with network protocols.
 #
 # Copyright Fortra, LLC and its affiliated companies 
@@ -30,6 +31,7 @@
 from __future__ import print_function
 from __future__ import unicode_literals
 import argparse
+import argcomplete
 import logging
 import sys
 import dns.resolver
@@ -209,8 +211,7 @@ def run(self):
                                                                               'If ommited, the domain part (FQDN) '
                                                                               'specified in the account parameter will be used')
 
-    
-
+    argcomplete.autocomplete(parser, always_complete_options=False)
 
     if len(sys.argv)==1:
         parser.print_help()

examples/GetADUsers.py

@@ -1,4 +1,5 @@
 #!/usr/bin/env python
+# PYTHON_ARGCOMPLETE_OK
 # Impacket - Collection of Python classes for working with network protocols.
 #
 # Copyright Fortra, LLC and its affiliated companies 
@@ -28,6 +29,7 @@
 from __future__ import print_function
 from __future__ import unicode_literals
 import argparse
+import argcomplete
 import logging
 import sys
 from datetime import datetime
@@ -173,6 +175,8 @@ def run(self):
                                                                               'If ommited, the domain part (FQDN) '
                                                                               'specified in the account parameter will be used')
 
+    argcomplete.autocomplete(parser, always_complete_options=False)
+
     if len(sys.argv)==1:
         parser.print_help()
         sys.exit(1)

examples/GetLAPSPassword.py

@@ -1,4 +1,5 @@
 #!/usr/bin/env python
+# PYTHON_ARGCOMPLETE_OK
 # Impacket - Collection of Python classes for working with network protocols.
 #
 # Copyright Fortra, LLC and its affiliated companies 
@@ -37,6 +38,7 @@
 from pyasn1.codec.der import decoder
 from pyasn1_modules import rfc5652
 import argparse
+import argcomplete
 import json
 import logging
 import sys
@@ -277,6 +279,8 @@ def run(self):
                                                                                 'Required when querying a Windows Server 2025'
                                                                                 'domain controller with LDAPS enforced.')
 
+    argcomplete.autocomplete(parser, always_complete_options=False)
+
     if len(sys.argv)==1:
         parser.print_help()
         sys.exit(1)

examples/GetNPUsers.py

@@ -1,4 +1,5 @@
 #!/usr/bin/env python
+# PYTHON_ARGCOMPLETE_OK
 # Impacket - Collection of Python classes for working with network protocols.
 #
 # Copyright Fortra, LLC and its affiliated companies 
@@ -29,6 +30,7 @@
 from __future__ import division
 from __future__ import print_function
 import argparse
+import argcomplete
 import datetime
 import logging
 import random
@@ -369,6 +371,8 @@ def request_multiple_TGTs(self, usernames):
                                                                               'If ommited, the domain part (FQDN) '
                                                                               'specified in the account parameter will be used')
 
+    argcomplete.autocomplete(parser, always_complete_options=False)
+
     if len(sys.argv)==1:
         parser.print_help()
         print("\nThere are a few modes for using this script")
@@ -383,6 +387,7 @@ def request_multiple_TGTs(self, usernames):
         print("\n3. Request TGTs for all users")
         print("\n\tGetNPUsers.py contoso.com/emily:password -request or GetNPUsers.py contoso.com/emily")
         print("\n4. Request TGTs for users in a file")
+
         print("\n\tGetNPUsers.py -no-pass -usersfile users.txt contoso.com/")
         print("\nFor this operation you don\'t need credentials.")
         sys.exit(1)

examples/GetUserSPNs.py

@@ -1,4 +1,5 @@
 #!/usr/bin/env python
+# PYTHON_ARGCOMPLETE_OK
 # Impacket - Collection of Python classes for working with network protocols.
 #
 # Copyright Fortra, LLC and its affiliated companies 
@@ -33,6 +34,7 @@
 from __future__ import division
 from __future__ import print_function
 import argparse
+import argcomplete
 import logging
 import sys
 from datetime import datetime
@@ -497,6 +499,8 @@ def request_multiple_TGSs(self, usernames):
                                                                             'If ommited, the domain part (FQDN) '
                                                                             'specified in the account parameter will be used')
 
+    argcomplete.autocomplete(parser, always_complete_options=False)
+
     if len(sys.argv) == 1:
         parser.print_help()
         sys.exit(1)

examples/addcomputer.py

@@ -1,4 +1,5 @@
 #!/usr/bin/env python
+# PYTHON_ARGCOMPLETE_OK
 # Impacket - Collection of Python classes for working with network protocols.
 #
 # Copyright Fortra, LLC and its affiliated companies 
@@ -39,6 +40,7 @@
 
 import ldap3
 import argparse
+import argcomplete
 import logging
 import sys
 import string
@@ -416,6 +418,7 @@ def run(self):
                                                                       'Useful if you can\'t translate the FQDN.'
                                                                       'specified in the account parameter will be used')
 
+    argcomplete.autocomplete(parser, always_complete_options=False)
 
     if len(sys.argv)==1:
         parser.print_help()

examples/atexec.py

@@ -1,4 +1,5 @@
 #!/usr/bin/env python
+# PYTHON_ARGCOMPLETE_OK
 # Impacket - Collection of Python classes for working with network protocols.
 #
 # Copyright Fortra, LLC and its affiliated companies 
@@ -26,6 +27,7 @@
 import string
 import sys
 import argparse
+import argcomplete
 import time
 import random
 import logging
@@ -271,6 +273,8 @@ def cmd_split(cmdline):
                                          'If omitted it will use the domain part (FQDN) specified in the target parameter')
     group.add_argument('-keytab', action="store", help='Read keys for SPN from keytab file')
 
+    argcomplete.autocomplete(parser, always_complete_options=False)
+
     if len(sys.argv)==1:
         parser.print_help()
         sys.exit(1)

examples/attrib.py

@@ -1,4 +1,5 @@
 #!/usr/bin/env python
+# PYTHON_ARGCOMPLETE_OK
 # Impacket - Collection of Python classes for working with network protocols.
 #
 # Copyright Fortra, LLC and its affiliated companies 
@@ -21,6 +22,7 @@
 from __future__ import annotations
 import sys
 import argparse
+import argcomplete
 import logging
 import ntpath
 from dataclasses import dataclass
@@ -246,6 +248,8 @@ def main():
     set_parser.add_argument('-p', '--pinned', dest='pinned', action='store_true', help="This attribute indicates user intent that the file or directory should be kept fully present locally even when not being actively accessed. This attribute is for use with hierarchical storage management software.")
     set_parser.add_argument('-u', '--unpinned', dest='unpinned', action='store_true', help="This attribute indicates that the file or directory should not be kept fully present locally except when being actively accessed. This attribute is for use with hierarchical storage management software.")
 
+    argcomplete.autocomplete(parser, always_complete_options=False)
+
     if len(sys.argv)==1:
         parser.print_help()
         sys.exit(1)