more fixes, so many API endpoints
[kismet-logviewer.git] / logviewer / dbview / views.py
index 1344acb..4efde62 100644 (file)
@@ -11,7 +11,6 @@ from django.views.decorators.csrf import csrf_exempt
 def load_db(query):
     dir_list = os.listdir("logs/")
     connection = sqlite3.connect("logs/"+dir_list[0])
-    #connection.row_factory = lambda cursor, row: row[0]
     cursor = connection.cursor()
     rows = cursor.execute(query).fetchall()
     return(rows)
@@ -21,9 +20,59 @@ def index(request):
     #if request.method == 'POST':
     #    print("GOT A POST")
     if request.path == "/devices/views/all_views.json":
-        all_views = open('dbview/all_views.json')
-        return HttpResponse(all_views, content_type='text/json')
+        uuid_members="["
+        dev_count=list(load_db("select count(device) from devices where type='Wi-Fi AP'"))
+        (devcount,) = dev_count[0]
+        uuid_members = uuid_members + "{ \"kismet.devices.view.description\": \"IEEE802.11 Access Points\", \"kismet.devices.view.id\": \"phydot11_accesspoints\", \"kismet.devices.view.size\": "+str(devcount)+" },"
+        dev_count=list(load_db("select count(device) from devices where phyname='IEEE802.11'"))
+        (devcount,) = dev_count[0]
+        uuid_members = uuid_members + "{ \"kismet.devices.view.description\": \"IEEE802.11 devices\", \"kismet.devices.view.id\": \"phy-IEEE802.11\", \"kismet.devices.view.size\": "+str(devcount)+" },"
+        dev_count=list(load_db("select count(device) from devices where type='RTL433'"))
+        (devcount,) = dev_count[0]
+        uuid_members = uuid_members + "{ \"kismet.devices.view.description\": \"RTL433 devices\", \"kismet.devices.view.id\": \"phy-RTL433\", \"kismet.devices.view.size\": "+str(devcount)+" },"
+        dev_count=list(load_db("select count(device) from devices where type='Z-wave'"))
+        (devcount,) = dev_count[0]
+        uuid_members = uuid_members + "{ \"kismet.devices.view.description\": \"Z-Wave devices\", \"kismet.devices.view.id\": \"phy-Z-Wave\", \"kismet.devices.view.size\": "+str(devcount)+" },"
+        dev_count=list(load_db("select count(device) from devices where type='BR/EDR'"))
+        (devcount,) = dev_count[0]
+        uuid_members = uuid_members + "{ \"kismet.devices.view.description\": \"Bluetooth devices\", \"kismet.devices.view.id\": \"phy-Bluetooth\", \"kismet.devices.view.size\": "+str(devcount)+" },"
+        dev_count=list(load_db("select count(device) from devices where type='UAV'"))
+        (devcount,) = dev_count[0]
+        uuid_members = uuid_members + "{ \"kismet.devices.view.description\": \"UAV devices\", \"kismet.devices.view.id\": \"phy-UAV\", \"kismet.devices.view.size\": "+str(devcount)+" },"
+        dev_count=list(load_db("select count(device) from devices where type='NrfMousejack'"))
+        (devcount,) = dev_count[0]
+        uuid_members = uuid_members + "{ \"kismet.devices.view.description\": \"NrfMousejack devices\", \"kismet.devices.view.id\": \"phy-NrfMousejack\", \"kismet.devices.view.size\": "+str(devcount)+"},"
+        dev_count=list(load_db("select count(device) from devices where type='BTLE'"))
+        (devcount,) = dev_count[0]
+        uuid_members = uuid_members + "{ \"kismet.devices.view.description\": \"BTLE devices\", \"kismet.devices.view.id\": \"phy-BTLE\", \"kismet.devices.view.size\": "+str(devcount)+" },"
+        dev_count=list(load_db("select count(device) from devices where phyname='AMR'"))
+        (devcount,) = dev_count[0]
+        uuid_members = uuid_members + "{ \"kismet.devices.view.description\": \"RTLAMR devices\", \"kismet.devices.view.id\": \"phy-RTLAMR\", \"kismet.devices.view.size\": "+str(devcount)+" },"
+        dev_count=list(load_db("select count(device) from devices where phyname='ADSB'"))
+        (devcount,) = dev_count[0]
+        uuid_members = uuid_members + "{ \"kismet.devices.view.description\": \"RTLADSB devices\", \"kismet.devices.view.id\": \"phy-RTLADSB\", \"kismet.devices.view.size\": "+str(devcount)+" },"
+        dev_count=list(load_db("select count(device) from devices where phyname='802.15.4'"))
+        (devcount,) = dev_count[0]
+        uuid_members = uuid_members + "{ \"kismet.devices.view.description\": \"802.15.4 devices\", \"kismet.devices.view.id\": \"phy-802.15.4\", \"kismet.devices.view.size\": "+str(devcount)+" },"
+        dev_count=list(load_db("select count(device) from devices where phyname='RADIATION'"))
+        (devcount,) = dev_count[0]
+        uuid_members = uuid_members + "{ \"kismet.devices.view.description\": \"RADIATION devices\", \"kismet.devices.view.id\": \"phy-RADIATION\", \"kismet.devices.view.size\": "+str(devcount)+" },"
+        total_dev=list(load_db("select count(device) from devices"))
+        (devcount,) = total_dev[0]
+        uuid_members=uuid_members+"{ \"kismet.devices.view.description\": \"All devices\", \"kismet.devices.view.id\": \"all\", \"kismet.devices.view.size\": "+str(devcount)+" },"
+        uuid_list = list(load_db("select uuid from datasources"))
+        for uuid in uuid_list:
+            (single_uuid,) = uuid
+            uuid_count = list(load_db("select count(*) from data where datasource='"+str(single_uuid)+"'"))
+            (single_uuid_count,) = uuid_count[0]
+            uuid_members = uuid_members + "{\"kismet.devices.view.description\": \"Devices seen by datasource "+single_uuid+"\","
+            uuid_members = uuid_members + "\"kismet.devices.view.id\": \"seenby-"+single_uuid+"\","
+            uuid_members = uuid_members + "\"kismet.devices.view.size\": "+str(single_uuid_count)+"},"
+        uuid_members=uuid_members[:-1]
+        uuid_members=uuid_members+"]"
+        return HttpResponse(uuid_members, content_type='text/json')
     elif request.path == "/system/user_status.json":
+        #Hardcoded for now
         user_status = open('dbview/user_status.json')
         return HttpResponse(user_status, content_type='text/json')
     elif request.path == "/session/check_setup_ok":
@@ -31,10 +80,12 @@ def index(request):
     elif request.path == "/session/check_login":
         return HttpResponse('Login valid')
     elif request.path == "/dynamic.js":
+        #INCOMPLETE - read the devices and create a dynamic.js output
         devices = load_db("select distinct(typestring) from datasources")
         load_file = open('static/dynamic.js')
         return HttpResponse(load_file, content_type='application/javascript')
     elif request.path == "/gps/location.json":
+        #hardcoded cus it doesnt matter
         user_status = open('dbview/gps_status.json')
         return HttpResponse(user_status, content_type='text/json')
     elif request.path == "/alerts/wrapped/last-time/0/alerts.json":
@@ -47,20 +98,44 @@ def index(request):
         alert_string = alert_string + "] ,\"kismet.alert.timestamp\": "+str(time.time())+"}"
         return HttpResponse(alert_string, content_type='text/json')
     elif request.path == "/phy/phy80211/ssids/views/ssids.json":
-        user_status = open('dbview/ssids.json')
-        return HttpResponse(user_status, content_type='text/json')
+        ssid_count = list(load_db("select count(device) from devices where type='Wi-Fi AP'"))
+        ssid_list = "{ \"recordsTotal\": "+str(ssid_count[0][0])+", \"data\": ["
+        ssids = list(load_db("select cast(device as text) from devices where type='Wi-Fi AP'"))
+        for ssid in ssids:
+            (single_ssid,) = ssid
+            ssid_json = json.loads(single_ssid)
+            try:
+                ssid_list = ssid_list + "{"
+                ssid_list = ssid_list + "\"dot11.ssidgroup.first_time\": \"" + str(ssid_json['dot11.device']['dot11.device.last_beaconed_ssid_record']['dot11.advertisedssid.first_time']) +"\","
+                ssid_list = ssid_list + "\"dot11.ssidgroup.ssid_len\": \"" + str(ssid_json['dot11.device']['dot11.device.last_beaconed_ssid_record']['dot11.advertisedssid.ssidlen'])+"\","
+                ssid_list = ssid_list + "\"dot11.ssidgroup.crypt_set\": \"" + str(ssid_json['dot11.device']['dot11.device.last_beaconed_ssid_record']['dot11.advertisedssid.crypt_set'])+"\","
+                ssid_list = ssid_list + "\"dot11.ssidgroup.hash\": \"" + str(ssid_json['dot11.device']['dot11.device.last_beaconed_ssid_record']['dot11.advertisedssid.ssid_hash'])+"\","
+                ssid_list = ssid_list + "\"dot11.ssidgroup.advertising_devices_len\": \"" + str(ssid_json['dot11.device']['dot11.device.num_advertised_ssids'])+"\","
+                ssid_list = ssid_list + "\"dot11.ssidgroup.probing_devices_len\": \"" + str(ssid_json['dot11.device']['dot11.device.num_probed_ssids'])+"\","
+                ssid_list = ssid_list + "\"dot11.ssidgroup.ssid\": \"" + str(ssid_json['dot11.device']['dot11.device.last_beaconed_ssid_record']['dot11.advertisedssid.ssid'])+"\","
+                ssid_list = ssid_list + "\"dot11.ssidgroup.responding_devices_len\": \"" + str(ssid_json['dot11.device']['dot11.device.last_beaconed_ssid_record']['dot11.advertisedssid.probe_response'])+"\","
+                ssid_list = ssid_list + "\"dot11.ssidgroup.last_time\": \"" + str(ssid_json['dot11.device']['dot11.device.last_beaconed_ssid_record']['dot11.advertisedssid.last_time'])+"\""
+                ssid_list = ssid_list + "},"
+            except:
+                ssid_list = ssid_list[:-1]
+                print("Skipping")
+        ssid_list = ssid_list[:-2]+ "}], \"draw\": 3, \"recordsFiltered\": "+str(ssid_count[0][0])+" }"
+        return HttpResponse(ssid_list, content_type='text/json')
     elif request.path == "/system/status.json":
+        #Hardcoded - Setup for other users
         user_status = open('dbview/status.json')
         return HttpResponse(user_status, content_type='text/json')
     elif request.path == "/alerts/alerts_view.json":
-        #MAY NOT BE COMPLETE
+        #INCOMPLETE - Check device Mappings
+        total_alerts=list(load_db("select count(json) from alerts"))
+        (alert_count,) = total_alerts[0]
         alerts = list(load_db("select cast(json as text) from alerts"))
-        alert_string="["
+        alert_string="{\"recordsTotal\": "+str(alert_count)+",\"data\": ["
         for alert in alerts:
             (single_alert,) = alert
             alert_string = alert_string + single_alert + ","
         alert_string = alert_string[:-1]
-        alert_string = alert_string + "]"
+        alert_string = alert_string + "], \"draw\": 6,\"recordsFiltered\": "+str(alert_count)+"}"
         return HttpResponse(alert_string, content_type='text/json')
     elif request.path == "/messagebus/last-time/0/messages.json":
         messages = list(load_db("select * from messages DESC limit 30"))
@@ -81,7 +156,7 @@ def index(request):
                 flag = 20
             if message[3] == "ERROR":
                 flag = 20
-            message_string = message_string + "\"kismet.messagebus.message_flags\": \"" + flag + "\","
+            message_string = message_string + "\"kismet.messagebus.message_flags\": \"" + str(flag) + "\","
             message_string = message_string + "\"kismet.messagebus.message_time\": \"" + str(message[0]) + "\""
             message_string = message_string + "},"
         message_string = message_string[:-1]
@@ -92,10 +167,11 @@ def index(request):
         return HttpResponse(user_status, content_type='text/json')
     elif request.path == "/devices/views/all/devices.json":
         #gotta figure out paging
+
         total_dev=list(load_db("select count(device) from devices"))
         (dev_count,) = total_dev[0]
         dev_string = "{ \"recordsTotal\": "+str(dev_count)+", \"data\": ["
-        dev_list = list(load_db("select cast(device as text) from devices limit 50"))
+        dev_list = list(load_db("select cast(device as text) from devices limit 126"))
         for device in dev_list:
             (dev,) = device
             dev_string = dev_string + dev + ","
@@ -105,10 +181,22 @@ def index(request):
     elif request.path == "/eventbus/events.ws":
         return HttpResponse("[]", content_type='text/json')
     elif request.path == "/devices/multikey/as-object/devices.json":
-        print("here")
+        #ClientMap incomplete.... figure out where the rest of the JSON comes from
+        search_json = ""
+        multikey = "{"
         for key, value in request.POST.items():
-            print("-----")
-            print(key)
-            print(value)
-            print("-----")
-        return HttpResponse("[]", content_type='text/json')
+            search_json=json.loads(value)
+            for device in search_json['devices']:
+                device_json = list(load_db("select cast(device as text) from devices where devkey='"+str(device)+"'"))
+                (tmp,) = device_json
+                device_json_x = json.loads(str(tmp[0]))
+                multikey = multikey + "\""+str(device)+"\": {"
+                for field in search_json['fields']:
+                    if (field[0:6] == "kismet"):
+                        multikey = multikey + "\""+field+"\": \""+device_json_x[field]+"\","
+                multikey = multikey + "\"dot11.device.client_map\": {},"
+                multikey = multikey[:-1]
+                multikey = multikey + "},"
+        multikey = multikey[:-1]
+        multikey = multikey + "}"
+        return HttpResponse(multikey, content_type='text/json')