diff options
Diffstat (limited to 'code')
| -rw-r--r-- | code/perpacket/parser.py | 50 |
1 files changed, 34 insertions, 16 deletions
diff --git a/code/perpacket/parser.py b/code/perpacket/parser.py index 422b699..37e7a8a 100644 --- a/code/perpacket/parser.py +++ b/code/perpacket/parser.py @@ -1,4 +1,6 @@ import csv +import numpy as np +import matplotlib.pyplot as plt with open('packetdata.csv', mode='r') as csv_file: csv_reader = csv.DictReader(csv_file) @@ -12,24 +14,40 @@ with open('packetdata.csv', mode='r') as csv_file: dest = row["Destination"] srcPort = row["Info"].split(' > ')[0].split(' ')[0] destPort = row["Info"].split(' > ')[1].split(' ')[1] - startTime = float(row["Time"]) + timeElapsed = float(row["Time delta from previous captured frame"]) for flow in flowsTCP: if src in flow and dest in flow and srcPort in flow and destPort in flow: flowExists = True + flow[4] += timeElapsed + flow[5] += 1 break if not flowExists: - flowsTCP.append([src, dest, srcPort, destPort, startTime, startTime]) - if row["Protocol"] == 'UDP': - flowExists = False - src = row["Source"] - dest = row["Destination"] - srcPort = row["Info"].split(' > ')[0].split(' ')[0] - destPort = row["Info"].split(' > ')[1].split(' ')[1] - for flow in flowsUDP: - if src in flow and dest in flow and srcPort in flow and destPort in flow: - flowExists = True - break - if not flowExists: - flowsUDP.append([src, dest, srcPort, destPort]) - print('TCP Length ' + str(len(flowsTCP))) - print('UDP Length ' + str(len(flowsUDP))) + flowsTCP.append([src, dest, srcPort, destPort, timeElapsed, 1]) + # if row["Protocol"] == 'UDP': + # flowExists = False + # src = row["Source"] + # dest = row["Destination"] + # srcPort = row["Info"].split(' > ')[0].split(' ')[0] + # destPort = row["Info"].split(' > ')[1].split(' ')[1] + # for flow in flowsUDP: + # if src in flow and dest in flow and srcPort in flow and destPort in flow: + # flowExists = True + # break + # if not flowExists: + # flowsUDP.append([src, dest, srcPort, destPort]) + flowsTCPTimeAverages = [] + for flow in flowsTCP: + flowsTCPTimeAverages.append(flow[4] / flow[5]) + with open('tcp_arrivel_time.csv', mode='w', newline='') as tcpArrival: + fieldnames = ['time'] + writer = csv.DictWriter(tcpArrival, fieldnames=fieldnames) + writer.writeheader() + for flow in flowsTCPTimeAverages: + writer.writerow({'time': flow}) + # num_bins = 20 + # counts, bin_edges = np.histogram (flowsTCPTimeAverages, bins=num_bins, normed=True) + # cdf = np.cumsum (counts) + # plt.plot (bin_edges[1:], cdf/cdf[-1]) + # plt.show() + # print(flowsTCPTimeAverages) + # print('UDP Length ' + str(len(flowsUDP))) |