-
Notifications
You must be signed in to change notification settings - Fork 2
/
Copy pathProgram.cs
executable file
·290 lines (251 loc) · 18.1 KB
/
Program.cs
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
using System.IO;
using static System.Console;
using static NeTraf.TrafficDataPrinter;
using static NeTraf.HelperMethods;
using static NeTraf.FileManager;
using static NeTraf.Plotter;
using static System.IO.Path;
using System.Collections.Generic;
using System.Threading;
using System;
namespace NeTraf
{
public class Program
{
#region Fields
#region SettingFiles
private static string _ratePlotSettingsFilePath ;
private static string _dataPlotSettingsFilePath ;
#endregion
#region NetstatIptrafFiles
private static string _netstatOutputFilePath;
private static string _iptrafOutputFilePath;
#endregion
#region RawOuputFiles
private static string _totalTrafficPacketsRawOutputFilePath ;
private static string _totalTrafficBytesRawOutputFilePath ;
private static string _totalTrafficRateRawOutputFilePath ;
private static string _incomingTrafficPacketsRawOutputFilePath ;
private static string _outgoingTrafficPacketsRawOutputFilePath ;
private static string _incomingTrafficBytesRawOutputFilePath ;
private static string _outgoingTrafficBytesRawOutputFilePath ;
private static string _incomingTrafficRateRawOutputFilePath ;
private static string _outgoingTrafficRateRawOutputFilePath ;
#endregion
#region GraphicalOutputFiles
private static string _totalTrafficPacketsGraphicalOutputFilePath ;
private static string _totalTrafficBytesGraphicalOutputFilePath ;
private static string _totalTrafficRateGraphicalOutputFilePath ;
private static string _incomingTrafficPacketsGraphicalOutputFilePath ;
private static string _outgoingTrafficPacketsGraphicalOutputFilePath ;
private static string _incomingTrafficBytesGraphicalOutputFilePath ;
private static string _outgoingTrafficBytesGraphicalOutputFilePath ;
private static string _incomingTrafficRateGraphicalOutputFilePath ;
private static string _outgoingTrafficRateGraphicalOutputFilePath ;
#endregion
#region CommandsName
private const string _netstatCommandName = "netstat";
private const string _iptrafCommandName = "iptraf";
#endregion
#region AppParameters
private static string _interfaceNetwork;
private static string _processName;
private static string _rootOutputFolderPath;
private static uint _profilingTime;
#endregion
#region Parsers
private static NetstatParser _netstatParser;
private static IptrafParser _iptrafParser;
#endregion
#endregion
public static void Main(string[] args)
{
SetApplicationParameters(args);
var autoEvent = new AutoResetEvent(false);
var profilingTimer = new Timer(new TimerCallback(StartParsing),autoEvent,TimeSpan.FromMinutes(_profilingTime),Timeout.InfiniteTimeSpan);
PrepareApplicationFiles();
StartNetstat();
StartIptraf();
autoEvent.WaitOne();
profilingTimer.Dispose();
CleanIptraf();
}
#region Methods
public static void StartParsing(object stateInfo)
{
var autoEvent = (AutoResetEvent)stateInfo;
ShellCommand.Stop(_netstatCommandName);
ShellCommand.StopUsingBash(_iptrafCommandName);
List<uint> netstatPorts = _netstatParser.GetPortsFromNetstatOutput();
var trafficDataRowSets = _iptrafParser.GetIptrafTrafficDataRowSets(netstatPorts);
PrintRawOutputFiles(trafficDataRowSets);
PrintGraphicalOutputFiles(trafficDataRowSets);
autoEvent.Set();
}
public static void StartNetstat()
{
_netstatParser = new NetstatParser(_netstatOutputFilePath);
var netstatCommand = new ShellCommand("/bin/bash",'c', $"\"netstat -c --all --tcp --udp --program | grep {_processName} > {_netstatOutputFilePath}\"", "");
netstatCommand.Execute();
}
public static void StartIptraf()
{
_iptrafParser = new IptrafParser(_iptrafOutputFilePath);
var iptrafCommand = new ShellCommand("/bin/bash",'c', $"\"iptraf -B -u -s {_interfaceNetwork} -L {_iptrafOutputFilePath}\"", "");
iptrafCommand.Execute();
}
public static void SetApplicationParameters(string[] args)
{
if (args.Length == 0)
{
Write("Interface network : "); _interfaceNetwork = ReadLine();
Write("Process Name : "); _processName = ReadLine();
Write("Profilinf Time : "); _profilingTime = uint.Parse(ReadLine());
Write("Output Folder : "); _rootOutputFolderPath = ReadLine();
}
else
{
_interfaceNetwork = args[0];
_processName = args[1];
_profilingTime = uint.Parse(args[2]);
_rootOutputFolderPath = args[3];
}
}
public static void PrepareApplicationFiles()
{
_ratePlotSettingsFilePath = GetPlotSettingsFilePaths(RatePlotSettingsFileName);
_dataPlotSettingsFilePath = GetPlotSettingsFilePaths(DataPlotSettingsFileName);
CreateFolderIfInexists(_rootOutputFolderPath);
CreateFolderIfInexists(Path.Combine(_rootOutputFolderPath,RawOutputFilesFolderName));
CreateFolderIfInexists(Path.Combine(_rootOutputFolderPath,GraphicalOutputFilesFolderName));
CreateFolderIfInexists(Path.Combine(_rootOutputFolderPath,GraphicalOutputFilesFolderName));
SetRawOutputFilePaths(_rootOutputFolderPath);
SetGraphicalOutputFilePaths(_rootOutputFolderPath);
}
private static void PrintRawOutputFiles(List<TrafficDataRowSet> trafficDataRowSets)
{
PrintToOutputFile(trafficDataRowSets , _totalTrafficPacketsRawOutputFilePath , TrafficDataType.TotalData , TrafficMeasurementType.Packets);
PrintToOutputFile(trafficDataRowSets , _totalTrafficBytesRawOutputFilePath , TrafficDataType.TotalData , TrafficMeasurementType.Bytes);
PrintToOutputFile(trafficDataRowSets , _totalTrafficRateRawOutputFilePath , TrafficDataType.TotalData , TrafficMeasurementType.Rate);
PrintToOutputFile(trafficDataRowSets , _incomingTrafficPacketsRawOutputFilePath , TrafficDataType.IncomingData , TrafficMeasurementType.Packets);
PrintToOutputFile(trafficDataRowSets , _outgoingTrafficPacketsRawOutputFilePath , TrafficDataType.OutgoingData , TrafficMeasurementType.Packets);
PrintToOutputFile(trafficDataRowSets , _incomingTrafficBytesRawOutputFilePath , TrafficDataType.IncomingData , TrafficMeasurementType.Bytes);
PrintToOutputFile(trafficDataRowSets , _outgoingTrafficBytesRawOutputFilePath , TrafficDataType.OutgoingData , TrafficMeasurementType.Bytes);
PrintToOutputFile(trafficDataRowSets , _incomingTrafficRateRawOutputFilePath , TrafficDataType.IncomingData , TrafficMeasurementType.Rate);
PrintToOutputFile(trafficDataRowSets , _outgoingTrafficRateRawOutputFilePath , TrafficDataType.OutgoingData , TrafficMeasurementType.Rate);
}
private static void SetRawOutputFilePaths(string rootOutputFolderPath)
{
_netstatOutputFilePath = Combine(rootOutputFolderPath , RawOutputFilesFolderName , NetstatOuputFileName);
_iptrafOutputFilePath = Combine(rootOutputFolderPath , RawOutputFilesFolderName , IptrafOutputFileName);
CreateFileIfInexists(_netstatOutputFilePath);
CreateFileIfInexists(_iptrafOutputFilePath);
_totalTrafficPacketsRawOutputFilePath = Combine(rootOutputFolderPath , RawOutputFilesFolderName , TotalTrafficPacketsOutputFileName);
_totalTrafficBytesRawOutputFilePath = Combine(rootOutputFolderPath , RawOutputFilesFolderName , TotalTrafficBytesOutputFileName);
_totalTrafficRateRawOutputFilePath = Combine(rootOutputFolderPath , RawOutputFilesFolderName , TotalTrafficRateOutputFileName);
_incomingTrafficPacketsRawOutputFilePath = Combine(rootOutputFolderPath , RawOutputFilesFolderName , IncomingTrafficPacketsOutputFileName);
_outgoingTrafficPacketsRawOutputFilePath = Combine(rootOutputFolderPath , RawOutputFilesFolderName , OutgoingTrafficPacketsOutputFileName);
_incomingTrafficBytesRawOutputFilePath = Combine(rootOutputFolderPath , RawOutputFilesFolderName , IncomingTrafficBytesOutputFileName);
_outgoingTrafficBytesRawOutputFilePath = Combine(rootOutputFolderPath , RawOutputFilesFolderName , OutgoingTrafficBytesOutputFileName);
_incomingTrafficRateRawOutputFilePath = Combine(rootOutputFolderPath , RawOutputFilesFolderName , IncomingTrafficRateOutputFileName);
_outgoingTrafficRateRawOutputFilePath = Combine(rootOutputFolderPath , RawOutputFilesFolderName , OutgoingTrafficRateOutputFileName);
}
private static void SetGraphicalOutputFilePaths(string rootOutputFolderPath)
{
_totalTrafficPacketsGraphicalOutputFilePath = Combine(rootOutputFolderPath , GraphicalOutputFilesFolderName , TotalTrafficPacketsOutputFileName + ".png");
_totalTrafficBytesGraphicalOutputFilePath = Combine(rootOutputFolderPath , GraphicalOutputFilesFolderName , TotalTrafficBytesOutputFileName + ".png");
_totalTrafficRateGraphicalOutputFilePath = Combine(rootOutputFolderPath , GraphicalOutputFilesFolderName , TotalTrafficRateOutputFileName + ".png");
_incomingTrafficPacketsGraphicalOutputFilePath = Combine(rootOutputFolderPath , GraphicalOutputFilesFolderName , IncomingTrafficPacketsOutputFileName + ".png");
_outgoingTrafficPacketsGraphicalOutputFilePath = Combine(rootOutputFolderPath , GraphicalOutputFilesFolderName , OutgoingTrafficPacketsOutputFileName + ".png");
_incomingTrafficBytesGraphicalOutputFilePath = Combine(rootOutputFolderPath , GraphicalOutputFilesFolderName , IncomingTrafficBytesOutputFileName + ".png");
_outgoingTrafficBytesGraphicalOutputFilePath = Combine(rootOutputFolderPath , GraphicalOutputFilesFolderName , OutgoingTrafficBytesOutputFileName + ".png");
_incomingTrafficRateGraphicalOutputFilePath = Combine(rootOutputFolderPath , GraphicalOutputFilesFolderName , IncomingTrafficRateOutputFileName + ".png");
_outgoingTrafficRateGraphicalOutputFilePath = Combine(rootOutputFolderPath , GraphicalOutputFilesFolderName , OutgoingTrafficRateOutputFileName + ".png");
}
private static void PrintGraphicalOutputFiles(List<TrafficDataRowSet> trafficDataRowSets)
{
var totalRatePlotter = new Plotter(_totalTrafficRateRawOutputFilePath,
_totalTrafficRatePlotTitle,
_totalTrafficRateGraphicalOutputFilePath,
_ratePlotSettingsFilePath,
_totalTrafficRatePlotTitle,
_xLabel,
$"Speed({GetBytesConvertionTargetUnit(trafficDataRowSets,TrafficDataType.TotalData,TrafficMeasurementType.Rate).ToSimpleString()}/s)",
ConvertTime(IptrafParser.GetAverageRunningTime(trafficDataRowSets)));
var totalBytesPlotter = new Plotter(_totalTrafficBytesRawOutputFilePath,
_totalTrafficBytesPlotTitle,
_totalTrafficBytesGraphicalOutputFilePath,
_dataPlotSettingsFilePath,
_totalTrafficBytesPlotTitle,
_xLabel,
$"{GetBytesConvertionTargetUnit(trafficDataRowSets,TrafficDataType.TotalData,TrafficMeasurementType.Bytes).ToSimpleString()}",
ConvertTime(IptrafParser.GetAverageRunningTime(trafficDataRowSets)));
var totalPacketsPlotter = new Plotter(_totalTrafficPacketsRawOutputFilePath,
_totalTrafficPacketsPlotTitle,
_totalTrafficPacketsGraphicalOutputFilePath,
_dataPlotSettingsFilePath,
_totalTrafficPacketsPlotTitle,
_xLabel,
"Packets",
ConvertTime(IptrafParser.GetAverageRunningTime(trafficDataRowSets)));
var incomingRatePlotter = new Plotter(_incomingTrafficRateRawOutputFilePath,
_incomingTrafficRatePlotTitle,
_incomingTrafficRateGraphicalOutputFilePath,
_ratePlotSettingsFilePath,
_incomingTrafficRatePlotTitle,
_xLabel,
$"Speed({GetBytesConvertionTargetUnit(trafficDataRowSets,TrafficDataType.IncomingData,TrafficMeasurementType.Rate).ToSimpleString()}/s)",
ConvertTime(IptrafParser.GetAverageRunningTime(trafficDataRowSets)));
var outgoingRatePlotter = new Plotter(_outgoingTrafficRateRawOutputFilePath,
_outgoingTrafficRatePlotTitle,
_outgoingTrafficRateGraphicalOutputFilePath,
_ratePlotSettingsFilePath,
_outgoingTrafficRatePlotTitle,
_xLabel,
$"Speed({GetBytesConvertionTargetUnit(trafficDataRowSets,TrafficDataType.OutgoingData,TrafficMeasurementType.Rate).ToSimpleString()}/s)",
ConvertTime(IptrafParser.GetAverageRunningTime(trafficDataRowSets)));
var incomingBytesPlotter = new Plotter(_incomingTrafficBytesRawOutputFilePath,
_incomingTrafficBytesPlotTitle,
_incomingTrafficBytesGraphicalOutputFilePath,
_dataPlotSettingsFilePath,
_incomingTrafficBytesPlotTitle,
_xLabel,
$"{GetBytesConvertionTargetUnit(trafficDataRowSets,TrafficDataType.IncomingData,TrafficMeasurementType.Bytes).ToSimpleString()}",
ConvertTime(IptrafParser.GetAverageRunningTime(trafficDataRowSets)));
var outgoingBytesPlotter = new Plotter(_outgoingTrafficBytesRawOutputFilePath,
_outgoingTrafficBytesPlotTitle,
_outgoingTrafficBytesGraphicalOutputFilePath,
_dataPlotSettingsFilePath,
_outgoingTrafficBytesPlotTitle,
_xLabel,
$"{GetBytesConvertionTargetUnit(trafficDataRowSets,TrafficDataType.OutgoingData,TrafficMeasurementType.Bytes).ToSimpleString()}",
ConvertTime(IptrafParser.GetAverageRunningTime(trafficDataRowSets)));
var incomingPacketsPlotter = new Plotter(_incomingTrafficPacketsRawOutputFilePath,
_incomingTrafficPacketsPlotTitle,
_incomingTrafficPacketsGraphicalOutputFilePath,
_dataPlotSettingsFilePath,
_incomingTrafficPacketsPlotTitle,
_xLabel,
"Packets",
ConvertTime(IptrafParser.GetAverageRunningTime(trafficDataRowSets)));
var outgoingPacketsPlotter = new Plotter(_outgoingTrafficPacketsRawOutputFilePath,
_outgoingTrafficPacketsPlotTitle,
_outgoingTrafficPacketsGraphicalOutputFilePath,
_dataPlotSettingsFilePath,
_outgoingTrafficPacketsPlotTitle,
_xLabel,
"Packets",
ConvertTime(IptrafParser.GetAverageRunningTime(trafficDataRowSets)));
totalRatePlotter.Plot();
totalBytesPlotter.Plot();
totalPacketsPlotter.Plot();
incomingRatePlotter.Plot();
outgoingRatePlotter.Plot();
incomingBytesPlotter.Plot();
outgoingBytesPlotter.Plot();
incomingPacketsPlotter.Plot();
outgoingPacketsPlotter.Plot();
}
#endregion
}
}