def generate_bitrate_plot_data(input_file, home_ip, website='youtube.com', is_incoming=True, protocols=None):
if protocols is None:
protocols = ['TCP']
start_timestamp = None
byte_rate_list = GaplessList(fill=0)
with open(input_file, 'rb') as csv_file:
data_reader = csv.DictReader(csv_file, delimiter=',')
for row in data_reader:
# set start_timestamp if not set
start_timestamp = float(row['timestamp']) if start_timestamp is None else start_timestamp
if website in row['website'] and (row['protocol'] in protocols):
if row['dst' if is_incoming else 'src'] == home_ip:
packet_time = int(float(row['timestamp']) - start_timestamp)
byte_rate_list.increment_element(packet_time, int(row['len']))
return byte_rate_list.get_list()
def generate_bitrate_plot_data(input_file,
home_ip,
website='youtube.com',
is_incoming=True,
protocols=None):
if protocols is None:
protocols = ['TCP']
start_timestamp = None
byte_rate_list = GaplessList(fill=0)
with open(input_file, 'rb') as csv_file:
data_reader = csv.DictReader(csv_file, delimiter=',')
for row in data_reader:
# set start_timestamp if not set
start_timestamp = float(
row['timestamp']
) if start_timestamp is None else start_timestamp
if website in row['website'] and (row['protocol'] in protocols):
if row['dst' if is_incoming else 'src'] == home_ip:
packet_time = int(
float(row['timestamp']) - start_timestamp)
byte_rate_list.increment_element(packet_time,
int(row['len']))
return byte_rate_list.get_list()
