From c7a952b3dfc49150760a521388c1634ccc1ac88b Mon Sep 17 00:00:00 2001
From: Niller303 <2797716+Niller303@users.noreply.github.com>
Date: Tue, 20 Mar 2018 15:24:45 +0100
Subject: [PATCH] Added fix for bad drivers

---
 wifite/tools/airmon.py | 33 ++++++++++++++++++++++++++++++++-
 1 file changed, 32 insertions(+), 1 deletion(-)

diff --git a/wifite/tools/airmon.py b/wifite/tools/airmon.py
index 8853033..53283da 100755
--- a/wifite/tools/airmon.py
+++ b/wifite/tools/airmon.py
@@ -15,6 +15,10 @@ class Airmon(object):
     base_interface = None
     killed_network_manager = False
 
+    #see if_arp.h
+    ARPHRD_ETHER = 1 #managed
+    ARPHRD_IEEE80211_RADIOTAP = 803 #monitor
+
     def __init__(self):
         self.refresh()
 
@@ -56,6 +60,28 @@ class Airmon(object):
             interfaces.append(Interface(fields))
         return interfaces
 
+    @staticmethod
+    def start_baddriver(iface): #fix for bad drivers like the rtl8812AU
+    	os.system("ifconfig %s down; iwconfig %s mode monitor; ifconfig %s up" % (iface, iface, iface))
+	#You cannot trust the output of the rtl8812AU
+	#it says that interface 10 is the monitor interface, it isn't.
+	#its actually just the same name, but lets not trust it anyway
+	for x in os.listdir("/sys/class/net/."):
+	    with open("/sys/class/net/" + x + "/type", "r") as f:
+		if (int(f.read()) == Airmon.ARPHRD_IEEE80211_RADIOTAP):
+		    return x
+
+	return None
+
+    @staticmethod
+    def stop_baddriver(iface):
+    	os.system("ifconfig %s down; iwconfig %s mode managed; ifconfig %s up" % (iface, iface, iface))
+	with open("/sys/class/net/" + iface + "/type", "r") as f:
+	    if (int(f.read()) == Airmon.ARPHRD_ETHER): 
+	        return iface
+
+	return None
+
     @staticmethod
     def start(iface):
         '''
@@ -91,7 +117,9 @@ class Airmon(object):
 
         if mon_iface is None:
             # Airmon did not enable monitor mode on an interface
-            Color.pl("{R}failed{W}")
+	    mon_iface = Airmon.start_baddriver(iface)
+	    if mon_iface is None:
+                Color.pl("{R}failed{W}")
 
         mon_ifaces = Airmon.get_interfaces_in_monitor_mode()
 
@@ -134,6 +162,9 @@ class Airmon(object):
                 mon_iface = match.groups()[0]
                 break
 
+        if not mon_iface:
+	    mon_iface = Airmon.stop_baddriver(iface)
+
         if mon_iface:
             Color.pl('{R}disabled %s{W}' % mon_iface)
         else: