minor fixes and removed the warning about ipv6 routes

habu/cli/cmd_arping.py

@@ -4,6 +4,8 @@
 
 import click
 
+logging.getLogger("scapy.runtime").setLevel(logging.ERROR)
+
 from scapy.all import ARP, IP, TCP, Ether, conf, srp
 
 

habu/cli/cmd_arpoison.py

@@ -1,7 +1,9 @@
 import time
-
+import logging
 import click
 
+logging.getLogger("scapy.runtime").setLevel(logging.ERROR)
+
 from scapy.all import ARP, Ether, conf, getmacbyip, sendp
 
 

habu/cli/cmd_arpsniff.py

@@ -1,6 +1,10 @@
 from time import time
 import click
 import habu.lib.manuf
+import logging
+
+logging.getLogger("scapy.runtime").setLevel(logging.ERROR)
+
 from scapy.all import ARP, IP, TCP, sniff, conf
 
 hosts = {}

habu/cli/cmd_b64.py

@@ -1,13 +1,16 @@
 import click
 import base64
+import sys
 #import codecs
 
 
 @click.command()
-@click.argument('f', type=click.File('r'))
+@click.argument('f', type=click.File('rb'))
 @click.option('-d', 'do_decode', is_flag=True, default=False, help='decode instead of encode')
 def cmd_b64(f, do_decode):
 
+    print("WARNING: This tool has problems decoding b64 data.", file=sys.stderr)
+
     data = f.read()
 
     if not data:
@@ -17,9 +20,9 @@ def cmd_b64(f, do_decode):
     #decoded = codecs.decode(data, "hex")
 
     if do_decode:
-        print(base64.b64decode(data).decode())
+        print(base64.b64decode(data), file=sys.stdout)
     else:
-        print(base64.b64encode(data.encode()).decode())
+        print(base64.b64encode(data).decode())
 
 
 if __name__ == '__main__':

habu/cli/cmd_dhcp_discover.py

@@ -1,4 +1,8 @@
 import click
+import logging
+
+logging.getLogger("scapy.runtime").setLevel(logging.ERROR)
+
 from scapy.all import ICMP, IP, conf, sr1, UDP, TCP, Ether, BOOTP, DHCP, get_if_raw_hwaddr, srp
 
 

habu/cli/cmd_ping.py

@@ -1,4 +1,8 @@
 import click
+import logging
+
+logging.getLogger("scapy.runtime").setLevel(logging.ERROR)
+
 from scapy.all import ICMP, IP, conf, sr1
 from time import sleep
 

habu/cli/cmd_portscan.py

@@ -1,4 +1,8 @@
 import click
+import logging
+
+logging.getLogger("scapy.runtime").setLevel(logging.ERROR)
+
 from scapy.all import ICMP, IP, conf, sr, sr1, TCP
 
 

habu/cli/cmd_protoscan.py

@@ -1,4 +1,8 @@
 import click
+import logging
+
+logging.getLogger("scapy.runtime").setLevel(logging.ERROR)
+
 from scapy.all import ICMP, IP, conf, sr, sr1, TCP
 
 

habu/cli/cmd_synflood.py

@@ -1,8 +1,11 @@
 import sys
+import logging
 from random import randint
 
 import click
 
+logging.getLogger("scapy.runtime").setLevel(logging.ERROR)
+
 from scapy.all import IP, TCP, Ether, RandMAC, conf, sendp
 
 
@@ -33,7 +36,7 @@ def cmd_synflood(ip, interface, count, port, forgemac, forgeip, verbose):
 
     counter = 0
 
-    print("please, remember to block your RST responses")
+    print("Please, remember to block your RST responses", file=sys.stderr)
 
     while True:
         if forgeip:

habu/cli/cmd_tcpflags.py

@@ -1,4 +1,8 @@
 import click
+import logging
+
+logging.getLogger("scapy.runtime").setLevel(logging.ERROR)
+
 from scapy.all import ICMP, IP, conf, sr, sr1, TCP
 
 

habu/cli/cmd_tcpscan.py

@@ -3,6 +3,9 @@
 
 import click
 import logging
+
+logging.getLogger("scapy.runtime").setLevel(logging.ERROR)
+
 from scapy.all import IP, TCP, conf, sr, sr1
 
 

habu/cli/cmd_traceroute.py

@@ -1,4 +1,8 @@
 import click
+import logging
+
+logging.getLogger("scapy.runtime").setLevel(logging.ERROR)
+
 from scapy.all import ICMP, IP, conf, sr1, UDP, TCP
 
 

setup.py

@@ -5,7 +5,7 @@
 
 setup(
     name='habu',
-    version='0.0.49',
+    version='0.0.50',
     description='Python Network Hacking Toolkit',
     long_description=readme,
     author='Fabian Martinez Portantier',