mirror of
https://github.com/nfstream/nfstream.git
synced 2026-04-28 06:59:28 +00:00
43 lines
1.9 KiB
Python
43 lines
1.9 KiB
Python
"""
|
|
------------------------------------------------------------------------------------------------------------------------
|
|
flow_printer.py
|
|
Copyright (C) 2019-22 - NFStream Developers
|
|
This file is part of NFStream, a Flexible Network Data Analysis Framework (https://www.nfstream.org/).
|
|
NFStream is free software: you can redistribute it and/or modify it under the terms of the GNU Lesser General Public
|
|
License as published by the Free Software Foundation, either version 3 of the License, or (at your option) any later
|
|
version.
|
|
NFStream is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty
|
|
of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more details.
|
|
You should have received a copy of the GNU Lesser General Public License along with NFStream.
|
|
If not, see <http://www.gnu.org/licenses/>.
|
|
------------------------------------------------------------------------------------------------------------------------
|
|
"""
|
|
|
|
from nfstream import NFStreamer
|
|
import sys
|
|
|
|
|
|
if __name__ == "__main__": # Mandatory if you are running on Windows Platform
|
|
input_filepaths = []
|
|
for path in sys.argv[1:]:
|
|
input_filepaths.append(path)
|
|
if len(input_filepaths) == 1: # Single file / Interface
|
|
input_filepaths = input_filepaths[0]
|
|
|
|
flow_streamer = NFStreamer(
|
|
source=input_filepaths, statistical_analysis=False, idle_timeout=1
|
|
)
|
|
result = {}
|
|
try:
|
|
for flow in flow_streamer:
|
|
print(flow)
|
|
try:
|
|
result[flow.application_name] += flow.bidirectional_packets
|
|
except KeyError:
|
|
result[flow.application_name] = flow.bidirectional_packets
|
|
print("\nSummary (Application Name: Packets):")
|
|
print(result)
|
|
except KeyboardInterrupt:
|
|
print("\nSummary (Application Name: Packets):")
|
|
print(result)
|
|
print("Terminated.")
|