fixes

README.md

@@ -436,6 +436,7 @@ Usage: habu.crack.snmp [OPTIONS] IP
 
 Options:
   -p INTEGER  Port to use
+  -c TEXT     Community (default: list of most used)
   -s          Stop after first match
   -v          Verbose
   --help      Show this message and exit.

README.rst

@@ -436,6 +436,7 @@ habu.crack.snmp
 
    Options:
      -p INTEGER  Port to use
+     -c TEXT     Community (default: list of most used)
      -s          Stop after first match
      -v          Verbose
      --help      Show this message and exit.

habu/cli/cmd_crack_snmp.py

@@ -15,9 +15,10 @@
 @click.command()
 @click.argument('ip')
 @click.option('-p', 'port', default=161, help='Port to use')
+@click.option('-c', 'community', default=None, help='Community (default: list of most used)')
 @click.option('-s', 'stop', is_flag=True, default=False, help='Stop after first match')
 @click.option('-v', 'verbose', is_flag=True, default=False, help='Verbose')
-def cmd_crack_snmp(ip, port, stop, verbose):
+def cmd_crack_snmp(ip, community, port, stop, verbose):
     """Launches snmp-get queries against an IP, and tells you when
     finds a valid community string (is a simple SNMP cracker).
 
@@ -40,26 +41,32 @@ def cmd_crack_snmp(ip, port, stop, verbose):
     DATADIR = os.path.abspath(os.path.join(FILEDIR, '../data'))
     COMMFILE = Path(os.path.abspath(os.path.join(DATADIR, 'dict_snmp.txt')))
 
-    with COMMFILE.open() as cf:
-        communities = cf.read().split('\n')
+    if community:
+        communities = [community]
+    else:
+        with COMMFILE.open() as cf:
+            communities = cf.read().split('\n')
 
     conf.verb = False
 
-    pkt = IP(dst=ip)/UDP(sport=port, dport=port)/SNMP(community="public", PDU=SNMPget(varbindlist=[SNMPvarbind(oid=ASN1_OID("1.3.6.1"))]))
-
-    for community in communities:
+    for pkt in IP(dst=ip)/UDP(sport=port, dport=port)/SNMP(community="public", PDU=SNMPget(varbindlist=[SNMPvarbind(oid=ASN1_OID("1.3.6.1"))])):
 
         if verbose:
-            print('.', end='')
-            sys.stdout.flush()
+            print(pkt[IP].dst)
+
+        for community in communities:
+
+            if verbose:
+                print('.', end='')
+                sys.stdout.flush()
 
-        pkt[SNMP].community=community
-        ans = sr1(pkt, timeout=0.5, verbose=0)
+            pkt[SNMP].community=community
+            ans = sr1(pkt, timeout=0.5, verbose=0)
 
-        if ans and UDP in ans:
-            print('\nCommunity found:', community)
-            if stop:
-                break
+            if ans and UDP in ans:
+                print('\n{} - Community found: {}'.format(pkt[IP].dst, community))
+                if stop:
+                    break
 
     return True
 

setup.py

@@ -5,7 +5,7 @@
 
 setup(
     name='habu',
-    version='0.0.97',
+    version='0.0.98',
     description='Python Network Hacking Toolkit',
     long_description=readme,
     long_description_content_type='text/x-rst',