一个查看服务器实时流量的perl脚本 perl, 服务器, 脚本, 实时, 流量

jacky

1. #!/usr/bin/perl
   
2. 
   
3. BEGIN {
   
4.         eval { require Time::HiRes; };
   
5.         if ($@) {
   
6.                 warn "You don't have Time::HiRes installed. Fallback to perl's built-in time functions";
   
7.         } else {
   
8.                 import Time::HiRes qw(time);
   
9.         }
   
10. };
    
11. 
    
12. use strict;
    
13. use POSIX qw(strftime);
    
14. use Getopt::Std;
    
15. 
    
16. my $PROC_NET_DEV = '/proc/net/dev';        # proc device to get the network statistics
    
17. my $SLEEP_INTERVAL = 1;
    
18. 
    
19. my $def_show_fields = ('bytes/k,packets,errs/M,out.colls');
    
20. my $guess_field_length = 8;
    
21. my %field_length_taken =
    
22.         ( 'k' => 3, 'K' => 3,
    
23.                 'm' => 6, 'M' => 6,
    
24.                 'u' => 2,
    
25.                 1 => 0
    
26.         );
    
27. my %field_data_taken =
    
28.         ( 'k' => 1000, 'K' => 1024,
    
29.                 'm' => 1000000, 'M' => 1048576,
    
30.                 'u' => 512,
    
31.                 1 => 1
    
32.         );
    
33. 
    
34. 
    
35. my %dev_ignore_show =
    
36.         (
    
37.         'list' => 'lo',
    
38.         'type' => 0,        # 0 - ignore, 1 - show
    
39.         );
    
40. 
    
41. my $proc_net_dev_title_line;
    
42. my %in_fields;
    
43. my %out_fields;
    
44. my $dev_show;
    
45. 
    
46. my $start_time;
    
47. my $end_time;
    
48. my @total;
    
49. 
    
50. my @show_fields = ();
    
51. my @show_fields_unit = ();
    
52. my @show_fields_length = ();
    
53. 
    
54. my $max_run = 0;
    
55. my $inffinity_run = 1;
    
56. 
    
57. sub
    
58. init()
    
59. {
    
60.         if (open (DEV, "<", $PROC_NET_DEV)) {
    
61.                 $proc_net_dev_title_line = 0;
    
62.                
    
63.                 my $part1type = 'in';
    
64.                 my $part2type = 'out';
    
65.                 while (<DEV>) {
    
66.                         last if (/^\s*[\w\d]+:/);
    
67.                         ++$proc_net_dev_title_line;
    
68.                        
    
69.                         chomp;
    
70.                         my ($devname, $part1, $part2) = split(/\s*\|\s*/);
    
71. 
    
72.                         my @types1 = split(/\s+/, $part1);
    
73.                         my @types2 = split(/\s+/, $part2);
    
74. 
    
75.                         if (scalar(@types1) > 1) {
    
76.                                 if ($part1type eq 'in') {
    
77.                                         for (my $i = 0; $i < scalar(@types1); ++$i) {
    
78.                                                 $in_fields{$types1[$i]} = $i;
    
79.                                         }
    
80.                                 } elsif ($part1type eq 'out') {
    
81.                                         for (my $i = 0; $i < scalar(@types1); ++$i) {
    
82.                                                 $out_fields{$types1[$i]} = $i;
    
83.                                         }
    
84.                                 } else {
    
85.                                         die "unknown type $part1type";
    
86.                                 }
    
87.                                
    
88.                                 if ($part2type eq 'in') {
    
89.                                         for (my $i = 0; $i < scalar(@types2); ++$i) {
    
90.                                                 $in_fields{$types2[$i]} = $i + scalar(@types1);
    
91.                                         }
    
92.                                 } elsif ($part2type eq 'out') {
    
93.                                         for (my $i = 0; $i < scalar(@types2); ++$i) {
    
94.                                                 $out_fields{$types2[$i]} = $i + scalar(@types1);
    
95.                                         }
    
96.                                 } else {
    
97.                                         die "unknown type $part2type";
    
98.                                 }
    
99.                                
    
100.                         } else {
     
101.                                 if ($types1[0] =~ /^(receive|recv|in)$/i) {
     
102.                                         $part1type = 'in'; $part2type = 'out';
     
103.                                 } else {
     
104.                                         $part1type = 'out'; $part2type = 'in';
     
105.                                 }
     
106.                         }
     
107.                 }
     
108. 
     
109.                 close (DEV);
     
110.         } # endif (open (DEV, "<", "/proc/net/dev"))
     
111.         else
     
112.         {
     
113.                 die "Unsupported /proc implementation. We can show you nothing.";
     
114.         }
     
115. }
     
116. 
     
117. sub
     
118. ntParseFields($)
     
119. {
     
120.         my $fieldDesc = shift;
     
121.         my @in_f;
     
122.         my @out_f;
     
123. 
     
124.         foreach (split(/\s*,\s*/, $fieldDesc)) {
     
125.                 if (lc(substr($_, 0, 3)) eq 'in.') {
     
126.                         push @in_f, $_;
     
127.                 } elsif (lc(substr($_, 0, 4)) eq 'out.') {
     
128.                         push @out_f, $_;
     
129.                 } else {
     
130.                         push @in_f, 'in.' . $_;
     
131.                         push @out_f, 'out.' . $_;
     
132.                 }
     
133.         }
     
134. 
     
135.         @show_fields = (@in_f, @out_f);
     
136.         foreach (@show_fields) {
     
137.                 if (/\/(\w)$/) {
     
138.                         if (lc($1) ne 'k' && lc($1) ne 'm' && lc($1) ne 'u') {
     
139.                                 die "unknown unit \"$1\"";
     
140.                         }
     
141.                         push @show_fields_unit, $1;
     
142.                         s/\/\w$//;
     
143.                 } else {
     
144.                         push @show_fields_unit, 1;
     
145.                 }
     
146.         }
     
147. }
     
148. 
     
149. sub
     
150. valueInArray($\@;$)
     
151. {
     
152.         my $value = shift;
     
153.         my $tmp = shift;
     
154.         my @array = @{$tmp};
     
155.         my $ignore_cases = shift;
     
156. 
     
157.         if ($ignore_cases) {
     
158.                 $value = lc($value);
     
159.                 @array = map { lc } @array;
     
160.         }
     
161.        
     
162.         foreach (@array) {
     
163.                 return 1 if ($value eq $_);
     
164.         }
     
165. 
     
166.         return 0;
     
167. }
     
168. 
     
169. sub
     
170. readProcDevNet()
     
171. {
     
172.         my %result = ();
     
173.        
     
174.         open (DEV, "<", $PROC_NET_DEV);
     
175.         for (my $i = 0; $i < $proc_net_dev_title_line; ++$i) {
     
176.                 <DEV>;
     
177.         }
     
178. 
     
179.         while (<DEV>) {
     
180.                 my ($devname, $data) = split(/:/);
     
181. 
     
182.                 $devname =~ s/^\s*//;
     
183.                 $devname =~ s/\s*$//;
     
184. 
     
185.                 my @devs = split(/\s*,\s*/, $dev_ignore_show{'list'});
     
186.                 if ( $dev_ignore_show{'type'} ) {
     
187.                         # show this devices only
     
188.                         next unless (valueInArray($devname, @devs, 1));
     
189.                 } else {
     
190.                         # ignore this device
     
191.                         next if (valueInArray($devname, @devs, 1));
     
192.                 }
     
193. 
     
194.                 chomp $data;
     
195.                 $data =~ s/^\s*//;
     
196.                 $data =~ s/\s*$//;
     
197.                 my @datas = split(/\s+/, $data);
     
198. 
     
199.                 my @tmp = ();
     
200.                
     
201.                 foreach (@show_fields) {
     
202.                         my ($type, $name) = split(/\./);
     
203.                         if ($type eq 'in') {
     
204.                                 if (exists $in_fields{$name}) {
     
205.                                         push @tmp, int($datas[$in_fields{$name}]);
     
206.                                 } else {
     
207.                                         warn "wrong field name in.$name";
     
208.                                         push @tmp, 0;
     
209.                                 }
     
210.                         } elsif ($type eq 'out') {
     
211.                                 if (exists $out_fields{$name}) {
     
212.                                         push @tmp, int($datas[$out_fields{$name}]);
     
213.                                 } else {
     
214.                                         warn "wrong field name out.$name";
     
215.                                         push @tmp, 0;
     
216.                                 }
     
217.                         } else {
     
218.                                 warn "wrong type $type.$name";
     
219.                                 push @tmp, 0;
     
220.                         }
     
221.                 }
     
222. 
     
223.                 $result{$devname} = \@tmp;
     
224.         }
     
225.         close (DEV);
     
226. 
     
227.         return \%result;
     
228. }
     
229. 
     
230. sub
     
231. max($)
     
232. {
     
233.         return $_[0] > $_[1] ? $_[0] : $_[1];
     
234. }
     
235. 
     
236. sub
     
237. ntSub(\@\@;$)
     
238. {
     
239.         my $array1 = shift;
     
240.         my $array2 = shift;
     
241.         my $maxuint = shift || 4294967295;
     
242. 
     
243.         my @result = ();
     
244.        
     
245.         my $maxi = max(scalar(@$array1), scalar(@$array2));
     
246.         for (my $i = 0; $i < $maxi; ++$i) {
     
247.                 my $num1 = $array1->[$i] || 0;
     
248.                 my $num2 = $array2->[$i] || 0;
     
249. 
     
250.                 $num1 -= $num2;
     
251.                 $num1 += $maxuint if ($num1 < 0);
     
252.                 push @result, $num1;
     
253.         }
     
254. 
     
255.         return \@result;
     
256. }
     
257. 
     
258. sub
     
259. ntCalcRate($\%\%)
     
260. {
     
261.         my $time1 = shift;
     
262.         my $time2 = shift;
     
263.         my $data1 = shift;
     
264.         my $data2 = shift;
     
265. 
     
266.         my $timeEpo = $time2 - $time1;
     
267.         my $subResult;
     
268.         my @result = ();
     
269.         foreach my $dev (keys %$data1) {
     
270.                 $subResult = ntSub(@{$data2->{$dev}}, @{$data1->{$dev}});
     
271.                 for (my $i = 0 ; $i < scalar(@$subResult); ++$i) {
     
272.                         $result[$i] += $subResult->[$i];
     
273.                 }
     
274.         }
     
275. 
     
276.         $end_time = $time2;
     
277.         for (my $i = 0; $i < scalar(@result); ++$i) {
     
278.                 $total[$i] += $result[$i];
     
279.                 $result[$i] /= $field_data_taken{$show_fields_unit[$i]} * $timeEpo;
     
280.         }
     
281.         wantarray() ? @result : \@result;
     
282. }
     
283. 
     
284. my $title;
     
285. sub
     
286. ntTitle()
     
287. {
     
288.         if (defined $title) {
     
289.                 print $title;
     
290.                 return;
     
291.         }
     
292.        
     
293.         my @format1;
     
294.         my @name1;
     
295.         my @format2;
     
296.         my @name2;
     
297.         my $nflag;
     
298.         my $last_2nd;
     
299.         my $dev = '(' . $dev_show . ')';
     
300.        
     
301.         push @format1, '@' . '|' x max(length($dev) - 1, 7);        # time
     
302.         push @name1, $dev;
     
303.         push @format2, '@' . '>' x (length($format1[0]) - 1);
     
304.         push @name2, 'time';
     
305.         $last_2nd = 1;
     
306.        
     
307.         for (my $i = 0; $i < scalar(@show_fields); ++$i) {
     
308.                 my ($type, $name) = split(/\./, $show_fields[$i]);
     
309.                 $name = $show_fields_unit[$i] . $name if ($show_fields_unit[$i] != 1);
     
310.                 push @name2, $name;
     
311.                 $show_fields_length[$i] = max($guess_field_length - $field_length_taken{$show_fields_unit[$i]}, length($name) + 1);
     
312.                 push @format2, '@' . '>' x ($show_fields_length[$i] - 1);
     
313. 
     
314.                 if (!defined $nflag) {
     
315.                         if (lc($type) eq 'in') {
     
316.                                 $nflag = 1;
     
317.                         } elsif (lc($type) eq 'out') {
     
318.                                 $nflag = 2;
     
319.                         } else {
     
320.                                 die "...";
     
321.                         }
     
322.                 } elsif ($nflag == 1) {
     
323.                         if (lc($type) eq 'in') {
     
324.                         } elsif (lc($type) eq 'out') {
     
325.                                 $nflag = 2;
     
326.                                 push @name1, 'input';
     
327.                                 push @format1, '@' . '|' x (length( join(' ', @format2[$last_2nd .. $#format2-1]) ) - 1);
     
328.                                 $last_2nd = $#format2;
     
329.                         } else {
     
330.                                 die '...';
     
331.                         }
     
332.                 } elsif ($nflag == 2) {
     
333.                         if (lc($type) eq 'in') {
     
334.                                 $nflag = 1;
     
335.                                 push @name1, 'output';
     
336.                                 push @format1, '@' . '|' x (length( join(' ', @format2[$last_2nd .. $#format2-1]) ) - 1);
     
337.                                 $last_2nd = $#format2;
     
338.                         } elsif (lc($type) eq 'out') {
     
339.                         } else {
     
340.                                 die '...';
     
341.                         }
     
342.                 } else {
     
343.                         die "...?";
     
344.                 }
     
345.         }
     
346.        
     
347.         if ($nflag == 1) {
     
348.                 push @name1, 'input';
     
349.                 push @format1, '@' . '|' x (length( join(' ', @format2[$last_2nd .. $#format2]) ) - 1);
     
350.         } elsif ($nflag == 2) {
     
351.                 push @name1, 'output';
     
352.                 push @format1, '@' . '|' x (length( join(' ', @format2[$last_2nd .. $#format2]) ) - 1);
     
353.         } else {
     
354.                 die "...";
     
355.         }
     
356. 
     
357.         formline join('*', @format1) . "\n", @name1;
     
358.         formline join(' ', @format2) . "\n", @name2;
     
359.         $title = $^A;
     
360.         $^A = '';
     
361.        
     
362.         print $title;
     
363.         return;
     
364. }
     
365. 
     
366. my $lineformat;
     
367. sub
     
368. ntLine($@)
     
369. {
     
370.         my $the_time = strftime("%H:%M:%S", localtime(shift));
     
371.         if (!defined $lineformat) {
     
372.                
     
373.                 my @format = ();
     
374.                 my $dev = '(' . $dev_show . ')';
     
375.        
     
376.                 push @format, '@' . '>' x max(length($dev) - 1, 7);
     
377.                 for (my $i = 0; $i < scalar(@show_fields_length); ++$i) {
     
378.                         push @format, '@' . '#' x ($show_fields_length[$i] - 1);
     
379.                 }
     
380. 
     
381.                 $lineformat = join(' ', @format);
     
382.         }
     
383.        
     
384.         formline $lineformat, $the_time, @_;
     
385.         print $^A, "\n";
     
386.         $^A = '';
     
387. }
     
388. 
     
389. sub
     
390. ntStatics()
     
391. {
     
392.         my $timeEpo = $end_time - $start_time;
     
393.        
     
394.         printf("\n\nStatics: begin at %s\tend at %s\n",
     
395.                 strftime("%H:%M:%S", localtime($start_time)),
     
396.                 strftime("%H:%M:%S", localtime($end_time)) );
     
397.         ntTitle();
     
398.        
     
399.         for (my $i = 0; $i < scalar(@total); ++$i) {
     
400.                 $total[$i] /= $timeEpo * $field_data_taken{$show_fields_unit[$i]};
     
401.         }
     
402.         ntLine($end_time, @total);
     
403.         exit(0);
     
404. }
     
405. 
     
406. sub
     
407. ntParseArgument(\%)
     
408. {
     
409.         my $arg = shift;
     
410. 
     
411.         foreach my $keyname (keys %{$arg}) {
     
412.                 if ($keyname eq 'i') {
     
413.                         # interface to show
     
414.                         $dev_ignore_show{'list'} = $arg->{'i'};
     
415.                         $dev_ignore_show{'type'} = 1;
     
416.                 } elsif ($keyname eq 'I') {
     
417.                         # interface to ignore
     
418.                         $dev_ignore_show{'list'} = $arg->{'I'};
     
419.                         $dev_ignore_show{'type'} = 0;
     
420.                 } elsif ($keyname eq 'c') {
     
421.             $max_run = $arg->{'c'};
     
422.             $inffinity_run = 0;
     
423.         }
     
424.         }
     
425. 
     
426.         return;
     
427. }
     
428. 
     
429. sub
     
430. ntShowHelp($)
     
431. {
     
432.         print <<"EOF";
     
433.     nettraf - Show the network traffic in a linux box.
     
434.         nettraf will get the statistics from the proc file system ( NORMALLY, /proc/net/dev)
     
435. 
     
436.         i                        Interface list to show.
     
437.         I                        Ignore interface list.
     
438.         c           Max show lines.
     
439. EOF
     
440.         exit($_[0]);
     
441. }
     
442. 
     
443. ###########################
     
444. # main
     
445. ###########################
     
446. {
     
447.         my %arguments;
     
448.         unless (getopts('i:I:c:', \%arguments)) {
     
449.                 ntShowHelp(-1);
     
450.         }
     
451. 
     
452.         ntParseArgument(%arguments);
     
453.         init();
     
454.         $SIG{INT} = \&ntStatics;
     
455.         ntParseFields($def_show_fields);
     
456.         my $this_time = time;
     
457.         my $this_data = readProcDevNet();
     
458.         $start_time = $this_time;
     
459. 
     
460.         my $line_show_count = 0;       
     
461.         $dev_show = join('+', keys %$this_data);
     
462.         while ($inffinity_run || $line_show_count < $max_run) {
     
463.                 ntTitle() if (($line_show_count++ % 25) == 0);
     
464. 
     
465.                 sleep($SLEEP_INTERVAL);
     
466.                 my $last_data = $this_data;
     
467.                 my $last_time = $this_time;
     
468.                 $this_time = time;
     
469.                 $this_data = readProcDevNet();
     
470. 
     
471.                 my $calcResult = ntCalcRate($last_time, $this_time, %$last_data, %$this_data);
     
472.                 ntLine($this_time, @$calcResult);
     
473.         }
     
474.     ntStatics();
     
475. }

复制代码