python scripts for wifi diag with pyshark

This commit is contained in:
SushantBawiskar
2020-09-30 11:14:10 +05:30
parent 35047536b8
commit e95704311b
3 changed files with 809 additions and 0 deletions

View File

@@ -0,0 +1,42 @@
import matplotlib.pyplot as plt
import base64
from io import BytesIO
import os
import datetime
time = (datetime.datetime.now().time())
name = "WIFI_Diag"
name += str(time)
name += "/"
name = name.replace(":","_")
if not os.path.exists(str(name)):
os.makedirs(str(name))
class Plot:
def __init__(self):
# print("In Plot")
pass
def bar(self, datax="", datay=" ", title="Temp", xaxis="xaxis", yaxis="yaxis",figname="temp"):
# fig = plt.figure()
self.tmpfile = BytesIO()
plt.xlabel(xaxis)
plt.ylabel(yaxis)
plt.title(title)
plt.xticks(rotation=90)
plt.rc('xtick', labelsize=8)
plt.rc('ytick', labelsize=8)
plt.bar(datax, datay)
path = plt.savefig(str(name)+str(figname)+".png",bbox_inches='tight')
# plt.savefig(str(title)+".png")
plt.savefig(self.tmpfile, format='png')
self.encoded = base64.b64encode(self.tmpfile.getvalue()).decode('utf-8')
# print("self.encoded",path)
plt.clf()
return str(name)+str(figname)+".png"

View File

@@ -0,0 +1,243 @@
import datetime
import matplotlib.pyplot as plt
import base64
from io import BytesIO
html = open("wifi_diag.html", 'w')
def htmlstart():
start = """<!DOCTYPE html>
<!-- This is for Banner -->
<body>
<title> WIFI Diag Report </title> <link rel='shortcut icon' href='canvil.ico' type='image/x-icon' />
<link rel='stylesheet' href='report.css' />
<link rel='stylesheet' href='custom.css' />
<style>
pre {
overflow: auto;
}
img {
width: 100%;
max-width: 8in;
}
</style>
</head>
<div class='Section report_banner-1000x205' style='background-image:url("report_banner-1000x205.jpg");background-repeat:no-repeat;padding:0;margin:0;min-width:1000px; min-height:205px;width:1000px; height:205px;max-width:1000px; max-height:205px;'>
<img align='right' style='padding:0;margin:5;width:200px;' src='CandelaLogo2-90dpi-200x90-trans.png' border='0' />
<div class='HeaderStyle'>
<h1 class='TitleFontPrint'>WIFI Diag Report</h1><br/><h3 class='TitleFontPrintSub'></h3><br><br/>
<h4 class='TitleFontPrintSub'><br><br></br></br>"""
html.write(start)
html.write(str(datetime.datetime.now()))
html.write("</h4></div></div></body>")
def downloadBtn():
text = """<tr><p align="right">
<td><input class="btn" type="submit" value="Print" name="Submit" id="printbtn" onclick="window.print()" /></td>
</tr></p>"""
html.write(text)
def htmlobj(text):
html.write("<h2> Objecitve </h2>")
html.write("<p>")
html.write(str(text))
html.write("</p>")
def htmlpointview():
pointview = """
<!--For Point view -->
<meta name="viewport" content="width=device-width, initial-scale=1">
<style>
ul, #myUL {
list-style-type: none;
}
#myUL {
margin: 0;
padding: 0;
}
.box {
cursor: pointer;
-webkit-user-select: none; /* Safari 3.1+ */
-moz-user-select: none; /* Firefox 2+ */
-ms-user-select: none; /* IE 10+ */
user-select: none;
}
.box::before {
content: "\\2610";
color: black;
display: inline-block;
margin-right: 6px;
}
.check-box::before {
content: "\\2611";
color: dodgerblue;
}
.nested {
display: none;
}
.active {
display: block;
}
* {
box-sizing: border-box;
}
.column {
float: left;
width: 100%;
}
.row::after {
content: "";
clear: both;
display: table;
}
.borderexample {
border-style:solid;
border-color: hsl(0, 0%, 73%);
}
.btn {
background-color: DodgerBlue;
border: none;
color: white;
padding: 12px 30px;
cursor: pointer;
font-size: 20px;
}
/* Darker background on mouse-over
.btn:hover {
background-color: RoyalBlue;
}
</style>
</head>
"""
html.write(str(pointview))
def htmlTableSummary(Summary):
tableSummary = "<body><!-- This is heading and Summary --><h2> About: </h2> \
<p>"+str(Summary)+"</p> \n"
html.write(str(tableSummary))
def myUL():
html.write("<ul id='myUL'>")
def htmlSpanBox(Table):
tablespan = "<li><span class= 'box borderexample'>"+str(Table)+"</span> <ul class='nested'> "
html.write(tablespan)
def htmlSpanBox1():
tablespan = "<li><span class='box'> Table </span> <ul class='nested'> "
html.write(tablespan)
def htmltable(Heading,data,image1,image2,image3):
# htmlBreak(1)
htmlSpanBox(Heading)
htmlTableSummary("This is table summary below Objective \n")
htmlSpanBox1()
html.write("<li><span> ")
html.write(str(data))
html.write("</span></li> ")
htmltableimage(image1,image2,image3)
def htmltableimage(image1, image2, image3):
if image1 != "0":
imagePre = "<div class='column'> \
<img style='padding:0;margin:3;width:450px;' src="
html.write(imagePre)
html.write(image1)
imagepost = " border='0' /> \
</div>"
html.write(imagepost)
if image2 != "0":
imagePre = "<div class='column'> \
<img style='padding:0;margin:3;width:450px;' src="
html.write(imagePre)
html.write(image2)
imagepost = " border='0' /> \
</div>\n"
html.write(imagepost)
if image3 != "0":
imagePre = "<div class='column'> \
<img style='padding:0;margin:3;width:450px;' src="
html.write(imagePre)
html.write(image2)
imagepost = " border='0' /> \
</div>"
html.write(imagepost)
html.write("</ul></ul></li><br>")
def closemyUl():
html.write("</ul>\n")
def htmlclose():
data = """
<script>\n
var toggler = document.getElementsByClassName("box");\n
var i;\n
for (i = 0; i < toggler.length; i++) {\n
toggler[i].addEventListener("click", function() {\n
this.parentElement.querySelector(".nested").classList.toggle("active");\n
this.classList.toggle("check-box");\n
});\n
}\n
</script>
</body>
"""
html.write(data)
html.write("</html>")
def htmlBreak(lines):
for i in range(lines):
html.write("<br></br>")
def htmlSpace(space):
Space = """<style>
.tab1 {
tab-size:""" + str(space) + """;
}
</style>"""
html.write(Space)
def htmlText(text, title):
title = "<h2" + str(title) + "</h2>"
data = "<p>" + str(text) + "<p/>"
html.write(title)
html.write(data)
#htmlstart()
#htmlobjective()
#htmlpointview()
#htmlsummary()
#myUl
#htmlspanbox()
#htmltable
#closemyul()
#htmlclose()

View File

@@ -0,0 +1,524 @@
"""
-----------------------------------------------------------------------------
Name : WIFI Diag
Author : Sushant Bawiskar
Date : 20 September 2020
------------------------------------------------------------------------------
"""
"""
Example: python PcaplibFiles.py --input "11ax.pcapng","sta1.pcap"
"""
import datetime
import pyshark
import pandas as pd
from bokeh.plotting import figure, output_file, show, save
from bokeh.io.export import get_screenshot_as_png, export_png, export_svgs, export_svg
import matplotlib.pyplot as plt
from plotly.offline import iplot, init_notebook_mode
import plotly.express as px
import plotly.graph_objects as go
from plotly.subplots import make_subplots
import matplotlib.pyplot as plt
import base64
from io import BytesIO
from htmlText import *
from Dataplot import Plot
import shutil
import argparse
import logging
import numpy as np
import os
def PacketHistogram(subtype_list, Managementls, Controlls, Data_framels, count):
# Created a Dictonary of Management Frame : {Subtype} , Control Frame : {Subtype} , Data Frame : {Subtype}
Type_Subtype = {"Management Frame": [Managementls], "Control Frame": [Controlls], "Data Frame": [Data_framels]}
Type_list = []
Sub_list = []
pack_list = []
per_list = []
# To calculate Total number of Subtype of packets in Type
# Ex. To calculate how many packets have a subtype which are in Management/Control/Data Frame Type
for Type, Subtype in Type_Subtype.items():
liskeys = []
for key in subtype_list.values():
if (key in liskeys):
continue
val = Subtype[0].count(key)
liskeys.append(key)
# liskeys.append(key)
if (val != 0):
# Type_list = [Type,key,val,(val*100)/count]
Type_list.append(str(Type))
Sub_list.append(key)
pack_list.append(val)
per_list.append((round((val * 100) / count, 2)))
# Type_list.append("")
Sub = Sub_list
NewSubList = Sub_list
# NewSubList.append("Sum: ")
#
# pack_list.append(sum(pack_list))
NewPerList = per_list
# NewPerList.append(sum(NewPerList))
print(len(subtype_list),len(NewSubList),len(pack_list),print(NewPerList))
df_Type = pd.DataFrame(({"Type": Type_list, "Subtype": NewSubList, "Packet to packet": pack_list, "Percentage": NewPerList}))
# print("df_Type",df_Type)
df_Type = df_Type.to_html()
# NewPerList.pop()
# NewSubList.pop()
plot = Plot()
path = plot.bar(datax=NewSubList,datay=NewPerList,title="Type/SubType plot",xaxis="Subtype",yaxis="Percentage",figname="Type")
htmltable(" Packet Type histogram", df_Type, str(path), "0", "0")
def RateHistogram(DataRate, PhyType, SignalStrength, count):
countUniqueData = []
perUniqueData = []
countUniquePhy = []
perUniquePhy = []
countUniqueSignal = []
perUniqueSignal = []
# This is for Data Table Histogram
uniqueData = np.unique(DataRate)
for i in uniqueData:
countUniqueData.append(DataRate.count(i))
uniqueData = [i for i in uniqueData]
# uniqueData.append("Sum: ")
# countUniqueData.append(sum(countUniqueData))
dictRate = (dict(zip(uniqueData, countUniqueData, )))
for c in countUniqueData:
perUniqueData.append(round((c * 100) / count, 2))
df_Rate = pd.DataFrame({"Rate MBPS": [i for i in dictRate.keys()], "Packet to packet": [j for j in dictRate.values()], "Percentage": [k for k in perUniqueData]})
df_Rate = df_Rate.T
df_Rate.columns = df_Rate.iloc[0]
df_Rate = df_Rate.drop(df_Rate.iloc[0].index.name)
df_Rate = df_Rate.to_html()
# uniqueData.pop()
# perUniqueData.pop()
plot1 = Plot()
path = plot1.bar(datax=uniqueData, datay=perUniqueData, title="Rate plot", xaxis="Rate MBPS", yaxis="Percentage",
figname="rate")
htmltable(" Encoding rate histogram.", df_Rate, str(path), "0", "0")
# This is for Phy Histogram
uniquePhy = np.unique(PhyType)
for j in uniquePhy:
countUniquePhy.append(PhyType.count(j))
uniquePhy = [i for i in uniquePhy]
# uniquePhy.append("Sum: ")
# countUniquePhy.append(sum(countUniquePhy))
dictPhy = (dict(zip(uniquePhy, countUniquePhy)))
for d in countUniquePhy:
perUniquePhy.append(round((d * 100) / count, 2))
df_Phy = pd.DataFrame({"Phy": [i for i in dictPhy.keys()], "Packet to Packet": [j for j in dictPhy.values()],"Percentage": [k for k in perUniquePhy]})
# print("df_Phy",df_Phy)
# df_Phy = df_Phy.to_html()
df_Phy = df_Phy.T
df_Phy.columns = df_Phy.iloc[0]
df_Phy = df_Phy.drop(df_Phy.iloc[0].index.name)
df_Phy = df_Phy.to_html()
dictphys = [i for i in dictPhy.keys()]
# dictphys.pop()
# perUniquePhy.pop()
plot2 = Plot()
path = plot2.bar(datax=dictphys, datay=perUniquePhy, title="Phy plot", xaxis="Subtype", yaxis="Percentage",
figname="Phy")
htmltable(" Phy Histogram.",df_Phy,str(path),"0","0")
# This is for Signal Histogram
uniqueSignal = np.unique(SignalStrength)
for k in uniqueSignal:
countUniqueSignal.append(SignalStrength.count(k))
uniqueSignal = [i for i in uniqueSignal]
# uniqueSignal.append("Sum: ")
# countUniqueSignal.append(sum(countUniqueSignal))
dictSig = (dict(zip(uniqueSignal, countUniqueSignal)))
for e in countUniqueSignal:
perUniqueSignal.append(round((e * 100) / count, 2))
# perUniqueSignal.append(sum(perUniqueSignal))
# pd.DataFrame.reset_index(drop=True,inplace=True)
# df_Sig = pd.DataFrame({"Signal": [i for i in dictSig.keys()], "Packet to Packet": [j for j in dictSig.values()],
# "Percentage": [k for k in perUniqueSignal]})
# pd.DataFrame.reset_index(drop=True,inplace=True)
print([k for k in dictSig.keys()])
print([i for i in dictSig.values()])
print("perUniqueSignal",perUniqueSignal)
# pd.DataFrame
df_Sig = pd.DataFrame({"Signal":[k for k in dictSig.keys()],"Packet":[i for i in dictSig.values()],"Percentage":[j for j in perUniqueSignal]})
df_Sig = df_Sig.T
df_Sig.columns = df_Sig.iloc[0]
df_Sig = df_Sig.drop(df_Sig.iloc[0].index.name)
# df_Sig.columns.name = None
# df_Sig.index.name = "Signal"
print("df_Sig",df_Sig)
# print("df_Sig",df_Sig)
# df_Sig = df_Sig.to_html()
# df_Sig = df_Sig.transpose()
df_Sig = df_Sig.to_html()
# perUniqueSignal.pop()
dictSigs = [i for i in dictSig.keys()]
# dictSigs.pop()
plot3 = Plot()
path = plot2.bar(datax=dictSigs, datay=perUniqueSignal, title="Signal plot", xaxis="Signal", yaxis="Percentage",
figname="Signal")
htmltable(" Signal Histogram.", df_Sig, str(path), "0", "0")
# print(dictSigs,perUniqueSignal)
class shark:
def __init__(self):
# FilePath having pcap file
# self.FilePath = "wifi_diag.pcap"
# self.FilePath = "C:\Candela\Scripts\Lanforge scripts\lanforge-scripts-master\wifi_diag\wifi_diag.pcapng"
# self.FilePath = "wifi_diag.pcap"
# self.FilePath = "C:\candela\pcap\wifi.pcapng"
self.FilePath = "C:\candela\pcap\\ac1_28Sept.pcapng"
# self.FilePath = "C:\Candela\My_Scripts\WIFI_diag11ax\wifi.pcapng"
# self.FilePath = "wifi_diag.pcap"
self.cap = pyshark.FileCapture(self.FilePath)
print("Strt time stamp :",datetime.datetime.now())
def Extract(self):
type_list = {"0": "Management frame", "1": "Control Frame", "2": "Data frame"}
subtype_list = {"80": "Beacon frame", "d0": "Action", "b4": "Request-to-send", "d4": "Acknowledgement", \
"88": "QoS Data", "84": "Block Ack Req", "94": "Block Ack Req", "08": "Data", \
"40": "Probe Request", "50": "Probe Response", "b0": "Authentication",
"a2": "Disassociate", "a8": "QoS Data + CF-Poll", "c8":"QoS Null function", \
"10": "Association Response", "00": "Association Request", "c4": "Clear-to-send", \
"98": "QoS Data + CF-Acknowledgment", "24": "Trigger", "28": "Data + CF-Poll" ,\
"d8": "Unknown", "54": "VHT/HE NDP Announcement", "e8": "QoS CF-Poll", \
"b8" : "QoS Data + CF-Ack + CF-Poll", "18": "Data + CF-Ack", "48" : "Null function", \
"69" : "CF-Poll"
}
Managementls = []
Controlls = []
Data_framels = []
PhyType = []
DataRate = []
SignalStrength = []
MCSIndex = []
uniqueMCSIndex = []
countUniqueMCSIndex = []
MCSIndex
Bandwidth = []
countUniqueBandwidth = []
PHY = []
countUniquePHY = []
Spatial_Stream = []
countUniqueSpatial_stream = []
count = 0
vWLAN_RADIO = 0
vsignalstrength = 0
vPhy = 0
vdatarate = 0
vWLAN = 0
""" Comment
"""
# print(wlan_radio_Fields_keys,wlan_radio_Fields_values)
# if "wlan_radio.phy" and "wlan_radio.11ac.bandwidth" and "wlan_radio.11ac.mcs" and "wlan_radio.11ac.nss" in wlan_radio_Fields_keys:
# # print(dir(packet.wlan_radio._all_fields["wlan_radio.phy"]))
# # print(packet.wlan_radio._all_fields["wlan_radio.phy"].showname_value) #for PHY
# # print(packet.wlan_radio._all_fields["wlan_radio.11ac.bandwidth"].showname_value) #for BW
# # print(packet.wlan_radio._all_fields["wlan_radio.11ac.mcs"].showname_value) #for MCS
# # print(packet.wlan_radio._all_fields["wlan_radio.11ac.nss"].showname_value) #for Spatial streams
# PHY.append(packet.wlan_radio._all_fields["wlan_radio.phy"].showname_value)
# Bandwidth.append(packet.wlan_radio._all_fields["wlan_radio.11ac.bandwidth"].showname_value)
# MCSIndex.append(packet.wlan_radio._all_fields["wlan_radio.11ac.mcs"].showname_value)
# Spatial_Stream.append(packet.wlan_radio._all_fields["wlan_radio.11ac.nss"].showname_value)
# _data = (packet.wlan_radio._all_fields)
# print(type(_data))
# phy, _11ac_short_gi, _11ac_bandwidth, _11ac_user, _11ac_mcs, _11ac_nss, _11ac_fec, _data_rate = (packet.wlan_radio._all_fields)
# print(phy.get_field_value)
"""
comment
"""
for packet in self.cap:
count += 1
# print(count)
try:
WLAN_RADIO = packet.wlan_radio
vWLAN_RADIO = 1
try:
signalstrength = WLAN_RADIO.signal_dbm
vsignalstrength = 1
except:
pass
try:
phy = WLAN_RADIO.phy.showname_value
vPhy = 1
except:
pass
try:
datarate = WLAN_RADIO.data_rate
vdatarate = 1
except:
pass
except:
pass
# print("WLAN RADIO NOT FOUND")
try:
RADIOTAP = packet.radiotap
except:
# print("RADIOTAP NOT FOUND")
pass
try:
WLAN = packet.wlan
vWLAN = 1
except:
pass
# print("WLAN NOT FOUND")
PacketCount = (packet.number)
# print("PacketCount: ", PacketCount)
if vWLAN == 1:
# print("WLAN found")
# Type/Subtype raw value
type_raw = (str(packet.wlan.fc_type.raw_value))
subtype_raw = ((packet.wlan.fc_type_subtype.raw_value))
# Name of values Types/Subtype
type = (str(packet.wlan.fc_type.showname_value))
subtype = (str(packet.wlan.fc_type_subtype.showname_value))
# Sorting Subtypes by Types as a refrence
if (type_raw == "0"):
try:
Managementls.append(subtype_list[subtype_raw])
except:
pass
elif (type_raw == "1"):
try:
Controlls.append(subtype_list[subtype_raw])
except:
print("Control Frame", subtype, subtype_raw)
elif (type_raw == "2"):
try:
Data_framels.append(subtype_list[subtype_raw])
except:
print("Data frame", subtype, subtype_raw)
else:
print("\nMissing Type in table : ", type_raw, subtype, subtype_raw, "\n")
else:
print("Packet Number :",count,":"," WLAN NOT FOUND")
vWLAN = 0
if vWLAN_RADIO == 1:
# print("vWLAN Found")
wlan_radio_Fields_keys = []
wlan_radio_Fields_values = []
for keys, values in packet.wlan_radio._all_fields.items():
# print(keys, ":", values)
wlan_radio_Fields_keys.append(keys)
wlan_radio_Fields_values.append(values)
if vdatarate == 1:
DataRate.append(datarate)
if vPhy == 1:
PhyType.append(phy)
if vsignalstrength == 1:
SignalStrength.append(signalstrength)
vWLAN_RADIO = 0
else:
print("Packet Number :",count," WLAN RADIO NOT FOUND")
try:
vMCS = 0
{"MCSIndex":['0 (BPSK 1/2)','1 (QPSK 1/2)','2 (QPSK 3/4)','4 (16-QAM 3/4)','5 (64-QAM 2/3)']}
{"Spatial_stream":["2"]}
{}
if "wlan_radio.11ac.bandwidth" and "wlan_radio.11ac.bandwidth" and "wlan_radio.phy" and \
"wlan_radio.11ac.mcs" and "wlan_radio.11ac.nss" in wlan_radio_Fields_keys:
# print(dir(packet.wlan_radio._all_fields["wlan_radio.phy"]))
# print(packet.wlan_radio._all_fields["wlan_radio.phy"].showname_value) #for PHY
# print(packet.wlan_radio._all_fields["wlan_radio.11ac.bandwidth"].showname_value) #for BW
# print(packet.wlan_radio._all_fields["wlan_radio.11ac.mcs"].showname_value) #for MCS
# print(packet.wlan_radio._all_fields["wlan_radio.11ac.nss"].showname_value) #for Spatial streams
PHY.append(packet.wlan_radio._all_fields["wlan_radio.phy"].showname_value)
Bandwidth.append(packet.wlan_radio._all_fields["wlan_radio.11ac.bandwidth"].showname_value)
MCSIndex.append(packet.wlan_radio._all_fields["wlan_radio.11ac.mcs"].showname_value)
Spatial_Stream.append(packet.wlan_radio._all_fields["wlan_radio.11ac.nss"].showname_value)
vMCS = 1
except:
pass
# print("MCSIndex",len(MCSIndex),MCSIndex)
# print("Bandwidth",len(Bandwidth),Bandwidth)
# print("PHY",len(PHY),PHY)
# print("Spatial_Stream",len(Spatial_Stream),Spatial_Stream)
uniqueMCSIndex = np.unique(MCSIndex)
uniqueBandwidth = ((np.unique(Bandwidth)))
uniquePHY = ((np.unique(PHY)))
uniqueSpatial_stream = ((np.unique(Spatial_Stream)))
exit()
print("After appending time stamp :", datetime.datetime.now())
# print(count)
RateHistogram(DataRate, PhyType, SignalStrength, count)
print("After RateHist time stamp :", datetime.datetime.now())
PacketHistogram(subtype_list, Managementls, Controlls, Data_framels, count)
print("After PacketHist time stamp :", datetime.datetime.now())
if __name__ == "__main__":
# parser = argparse.ArgumentParser(description="To create a single pcap file combining multiple pcap files")
# # parser.add_argument("-o", "--output", type=str, help="Enter the output pcap file name")
# parser.add_argument("-i", "--input", type=str,
# help="Enter the Name of the pcap files which needs to be combined")
#
# args = None
#
# try:
# args = parser.parse_args()
# output = "wifi_diag.pcap"
#
# if (args.input is not None):
# input = args.input # 'C:\candela\pcap\\11ax.pcapng','C:\candela\pcap\sta1.pcap'
# input = input.split(",")
# print(input)
#
# # if (args.input is None):
# # input = "11ax.pcapng", "sta1.pcap"
#
# except Exception as e:
# logging.exception(e)
# exit(2)
#
# with open(output, 'wb') as wfd:
# print("input", input)
# for f in (input):
# with open(f, 'rb') as fd:
# shutil.copyfileobj(fd, wfd)
htmlstart()
downloadBtn()
htmlobj("This is HTML objective")
htmlpointview()
htmlTableSummary("This is html table summary")
myUL()
Extract = shark()
Extract.Extract()
# htmltable()
closemyUl()
htmlclose()