]> git.zerfleddert.de Git - upsgraph/blob - upsgraph.pl
e846a23b42eeecca4219fbcc561ea5ed9e225207
[upsgraph] / upsgraph.pl
1 #!/usr/bin/perl -w
2
3 if ((@ARGV != 1) && (@ARGV != 2)) {
4 print STDERR "Syntax: ${0} configfile [uid]\n";
5 exit(1);
6 }
7
8 use Net::SNMP;
9 use IO::Socket::INET;
10 use RRDs;
11 use File::Copy;
12 use Data::Dumper;
13
14 $UPSGRAPH::outdir = "";
15 $UPSGRAPH::step = 60;
16 $UPSGRAPH::keep = (370*24*60*60)/$UPSGRAPH::step;
17 $UPSGRAPH::hosts = ();
18
19 do $ARGV[0] or die "can't read config: $!";
20
21 my $outdir = $UPSGRAPH::outdir;
22 my $step = $UPSGRAPH::step;
23 my $keep = $UPSGRAPH::keep;
24 my $hosts = $UPSGRAPH::hosts;
25
26 sub rrdcreate(@) {
27 my $newrrd = shift;
28 my $field = shift;
29 my $vars = shift;
30 my $start = shift;
31
32 my @cmd = ("${newrrd}", "--step=${step}");
33
34 if (defined($start)) {
35 push @cmd, "--start=${start}";
36 }
37
38 push @cmd, "DS:${field}:GAUGE:600:" .
39 $vars->{$field}->{'min'} . ":" .
40 $vars->{$field}->{'max'} . " ";
41
42 push @cmd, "RRA:AVERAGE:0.5:1:${keep}";
43
44 RRDs::create(@cmd);
45 if (RRDs::error) {
46 print "Error while creating: " . RRDs::error . "\n";
47 exit 1;
48 }
49 }
50
51 sub fetch_snmp(@) {
52 my $address = shift;
53 my $community = shift;
54 my $oid = shift;
55
56 (my $session, my $error) = Net::SNMP->session(Hostname => $address,
57 Community => $community);
58
59 if (!$session) {
60 print STDERR "session error: $error";
61 }
62
63 $session->translate(0);
64
65 my $result = $session->get_request($oid);
66
67 $session->close;
68
69 return undef if (!defined($result));
70
71 $result->{$oid};
72 }
73
74 sub fetch_tcp(@) {
75 my $address = shift;
76 my $port = shift;
77
78 my $sock = IO::Socket::INET->new(PeerAddr => $address,
79 PeerPort => $port,
80 Proto => 'tcp',
81 Timeout => 1);
82
83 return undef if (!$sock);
84
85 chomp(my $value = <$sock>);
86
87 close($sock);
88
89 if (!$value) {
90 return undef;
91 }
92
93 $value=~ s/\s//g;
94
95 $value;
96 }
97
98 if ($> == 0) {
99 if (@ARGV != 2) {
100 print STDERR "Running as root, please provide UID as 2th argument!\n";
101 exit(1);
102 }
103
104 print "Running as root, switching to ".$ARGV[1]."\n";
105 $< = $> = $ARGV[1];
106 }
107
108 foreach my $host (@$hosts) {
109 my $rrdfile = $host->{'rrdfile'};
110
111 foreach my $var (keys(%{$host->{'vars'}})) {
112 $host->{'vars'}->{$var}->{'min'} = 'U' if (!defined($host->{'vars'}->{$var}->{'min'}));
113 $host->{'vars'}->{$var}->{'max'} = 'U' if (!defined($host->{'vars'}->{$var}->{'max'}));
114 }
115
116 if (-e "${rrdfile}") {
117 print "Reading old ${rrdfile} to preserve data...\n";
118
119 my $rrdinfo = RRDs::info("${rrdfile}");
120 if (RRDs::error) {
121 print "Error while getting info: " . RRDs::error . "\n";
122 exit 1;
123 }
124
125 (my $start, my $ostep, my $names, my $data) =
126 RRDs::fetch("${rrdfile}",
127 "-s " . (time() - ($rrdinfo->{'rra[0].rows'} * $rrdinfo->{'step'})),
128 "AVERAGE");
129
130 if (RRDs::error) {
131 print "Error while fetching data: " . RRDs::error . "\n";
132 exit 1;
133 }
134
135 foreach my $field (@$names) {
136 if (! -e "${rrdfile}.${field}") {
137 rrdcreate("${rrdfile}.${field}",
138 "${field}",
139 $host->{'vars'},
140 (${start}-${ostep}));
141 }
142 }
143
144 my $pos = $start;
145 foreach my $line (@$data) {
146 foreach my $field (@$names) {
147 my $val = shift (@$line);
148 $val = 'U' if (!defined($val));
149
150 RRDs::update("${rrdfile}.${field}", "${pos}:${val}");
151 if (RRDs::error) {
152 print "Can't insert data: " . RRDs::error . "\n";
153 exit 1;
154 }
155
156 }
157
158 $pos += $ostep;
159
160 if ((($pos-$start)/$ostep) == $#$data) {
161 last;
162 }
163 }
164
165 rename("${rrdfile}", "${rrdfile}.old") or die "Can't rename old file: $!\n";
166 }
167
168 foreach my $field (@{$host->{'fields'}}) {
169 if (! -e "${rrdfile}.${field}") {
170 print "Creating ${rrdfile}.${field}...\n";
171 rrdcreate("${rrdfile}.${field}",
172 "${field}",
173 $host->{'vars'});
174 }
175
176 my $rrdinfo = RRDs::info("${rrdfile}.${field}");
177 if (RRDs::error) {
178 print "Error while getting info: " . RRDs::error . "\n";
179 exit 1;
180 }
181
182 if (defined($rrdinfo->{"ds[${field}].min"})) {
183 if ($rrdinfo->{"ds[${field}].min"} ne $host->{'vars'}->{$field}->{'min'}) {
184 RRDs::tune("${rrdfile}.${field}","-i",$field.":".$host->{'vars'}->{$field}->{'min'});
185 }
186 } else {
187 if ($host->{'vars'}->{$field}->{'min'} ne 'U') {
188 RRDs::tune("${rrdfile}.${field}","-i",$field.":".$host->{'vars'}->{$field}->{'min'});
189 }
190 }
191
192 if (RRDs::error) {
193 print "Error while setting min: " . RRDs::error . "\n";
194 exit 1;
195 }
196
197 if (defined($rrdinfo->{"ds[${field}].max"})) {
198 if ($rrdinfo->{"ds[${field}].max"} ne $host->{'vars'}->{$field}->{'max'}) {
199 RRDs::tune("${rrdfile}.${field}","-a",$field.":".$host->{'vars'}->{$field}->{'max'});
200 }
201 } else {
202 if ($host->{'vars'}->{$field}->{'max'} ne 'U') {
203 RRDs::tune("${rrdfile}.${field}","-a",$field.":".$host->{'vars'}->{$field}->{'max'});
204 }
205 }
206
207 if (RRDs::error) {
208 print "Error while setting max: " . RRDs::error . "\n";
209 exit 1;
210 }
211
212 if ($rrdinfo->{'rra[0].rows'} != $keep) {
213 print "Resizing ${rrdfile}.${field} from " . $rrdinfo->{'rra[0].rows'} .
214 " to ${keep} samples.\n";
215
216 (my $start, my $ostep, my $names, my $data) =
217 RRDs::fetch("${rrdfile}.${field}",
218 "-s " . (time() - ($rrdinfo->{'rra[0].rows'} * $rrdinfo->{'step'})),
219 "AVERAGE");
220
221 if (RRDs::error) {
222 print "Error while fetching data: " . RRDs::error . "\n";
223 exit 1;
224 }
225
226 rrdcreate("${rrdfile}.${field}.new",
227 "${field}",
228 $host->{'vars'},
229 (${start}-${ostep}));
230
231 print "Preserving data since " . localtime($start) . "\n";
232
233 my $pos = $start;
234 foreach my $line (@$data) {
235 my $vline = "${pos}";
236
237 foreach my $val (@$line) {
238 $val = 'U' if (!defined($val));
239 $vline .= ":${val}";
240 }
241 RRDs::update("${rrdfile}.${field}.new", $vline) or die "Can't insert data\n";
242
243 if (RRDs::error) {
244 print "Error while updating: " . RRDs::error . "\n";
245 exit 1;
246 }
247 $pos += $ostep;
248
249 if ((($pos-$start)/$ostep) == $#$data) {
250 last;
251 }
252 }
253
254 rename("${rrdfile}.${field}", "${rrdfile}.${field}.old") or die "Can't rename old file: $!\n";
255 rename("${rrdfile}.${field}.new", "${rrdfile}.${field}") or die "Can't rename new file: $!\n";
256
257 $rrdinfo = RRDs::info("${rrdfile}.${field}");
258 if (RRDs::error) {
259 print "Error while getting info: " . RRDs::error . "\n";
260 exit 1;
261 }
262
263 if ($rrdinfo->{'rra[0].rows'} != $keep) {
264 print "Failed!\n";
265 exit 1;
266 }
267 }
268 }
269 }
270
271 my $child = fork();
272
273 die "fork failed!" if (!defined($child));
274
275 exit 0 if ($child != 0);
276
277 while(1) {
278 open(HTML, ">${outdir}/index.html.new");
279
280 print HTML '<html><head><meta http-equiv="refresh" content="60"/><meta http-equiv="cache-control" content="no-cache"/><meta http-equiv="pragma" content="no-cache"/><meta http_equiv="expires" content="Sat, 26 Jul 1997 05:00:00 GMT"/><title>Status</title></head>';
281 print HTML '<body bgcolor="#ffffff">';
282
283 foreach my $host (@$hosts) {
284 print HTML "[<a href=\"#".${host}->{'name'}."\">".${host}->{'name'}."</a>]&nbsp;";
285 }
286 print HTML "<br>\n";
287
288 foreach my $host (@$hosts) {
289 print HTML "<br>\n";
290 print HTML "<a name=\"".${host}->{'name'}."\"></a>\n";
291 my $vars = $host->{'vars'};
292 my $rrdfile = $host->{'rrdfile'};
293 my $hostname = $host->{'name'};
294
295 foreach my $var (@{$host->{'fields'}}) {
296 delete $vars->{$var}->{'value'};
297
298 my $result;
299
300 if ((!defined($vars->{$var}->{'proto'})) ||
301 ($vars->{$var}->{'proto'} eq '') ||
302 ($vars->{$var}->{'proto'} eq 'snmp')) {
303 $result = fetch_snmp($host->{'address'}, $host->{'community'}, $vars->{$var}->{'oid'});
304 } elsif ($vars->{$var}->{'proto'} eq 'tcp') {
305 $result = fetch_tcp($host->{'address'}, $vars->{$var}->{'port'});
306 }
307
308 next unless (defined $result);
309
310 $vars->{$var}->{'value'} = $result;
311 if (defined($vars->{$var}->{'factor'})) {
312 $vars->{$var}->{'value'} *= $vars->{$var}->{'factor'};
313 }
314 }
315
316 foreach my $var (@{$host->{'fields'}}) {
317 if (!(defined($vars->{$var}->{'value'}))) {
318 $vars->{$var}->{'value'} = 'U';
319 }
320 RRDs::update("${rrdfile}.${var}", "N:" . $vars->{$var}->{'value'});
321 }
322 if (RRDs::error) {
323 print "Error while updating: " . RRDs::error . "\n";
324 }
325
326 foreach my $var (@{$host->{'fields'}}) {
327 my @graphdef = ("--lazy", "-t", $hostname." - ".$vars->{$var}->{'name'}, "DEF:${var}=${rrdfile}.${var}:${var}:AVERAGE", "LINE1:${var}#FF0000");
328 (my $averages, my $width, my $height) =
329 RRDs::graph("${outdir}/${hostname}.${var}.png.work",
330 "-w", "720", @graphdef);
331
332 if (RRDs::error) {
333 print "Error while graphing: " . RRDs::error . "\n";
334 } else {
335 copy("${outdir}/${hostname}.${var}.png.work", "${outdir}/${hostname}.${var}.png.new");
336 rename("${outdir}/${hostname}.${var}.png.new", "${outdir}/${hostname}.${var}.png");
337 }
338
339 print HTML "<a href=\"${hostname}.${var}.html\"><img src=\"${hostname}.${var}.png\" width=\"${width}\" height=\"${height}\" border=\"0\"></a><br>\n";
340
341 open (HTML2, ">${outdir}/${hostname}.${var}.html.new");
342 print HTML2 "<html><head><title>" . $vars->{$var}->{'name'} . "</title></head>";
343 print HTML2 '<body bgcolor="#ffffff">';
344
345
346 push @graphdef, "VDEF:min=${var},MINIMUM";
347 push @graphdef, "GPRINT:min:Minimum\\: %.2lf";
348
349 push @graphdef, "VDEF:avg=${var},AVERAGE";
350 push @graphdef, "GPRINT:avg:Average\\: %.2lf";
351
352 push @graphdef, "VDEF:max=${var},MAXIMUM";
353 push @graphdef, "GPRINT:max:Maximum\\: %.2lf";
354
355 push @graphdef, "VDEF:cur=${var},LAST";
356 push @graphdef, "GPRINT:cur:Current\\: %.2lf";
357
358 ($averages, $width, $height) =
359 RRDs::graph("${outdir}/${hostname}.${var}.long.png.work",
360 "-w", "1008", @graphdef);
361
362 if (RRDs::error) {
363 print "Error while graphing: " . RRDs::error . "\n";
364 } else {
365 copy("${outdir}/${hostname}.${var}.long.png.work", "${outdir}/${hostname}.${var}.long.png.new");
366 rename("${outdir}/${hostname}.${var}.long.png.new", "${outdir}/${hostname}.${var}.long.png");
367 }
368
369 print HTML2 "<img src=\"${hostname}.${var}.long.png\" width=\"${width}\" height=\"${height}\"><br>";
370
371 ($averages, $width, $height) =
372 RRDs::graph("${outdir}/${hostname}.${var}.week.png.work",
373 "-w", "1008", "-e", "now", "-s", "end-1w", @graphdef);
374
375 if (RRDs::error) {
376 print "Error while graphing: " . RRDs::error . "\n";
377 } else {
378 copy("${outdir}/${hostname}.${var}.week.png.work", "${outdir}/${hostname}.${var}.week.png.new");
379 rename("${outdir}/${hostname}.${var}.week.png.new", "${outdir}/${hostname}.${var}.week.png");
380 }
381
382 print HTML2 "<img src=\"${hostname}.${var}.week.png\" width=\"${width}\" height=\"${height}\"><br>";
383
384 ($averages, $width, $height) =
385 RRDs::graph("${outdir}/${hostname}.${var}.year.png.work",
386 "-w", "1008", "-e", "now", "-s", "end-1y", @graphdef);
387
388 if (RRDs::error) {
389 print "Error while graphing: " . RRDs::error . "\n";
390 } else {
391 copy("${outdir}/${hostname}.${var}.year.png.work", "${outdir}/${hostname}.${var}.year.png.new");
392 rename("${outdir}/${hostname}.${var}.year.png.new", "${outdir}/${hostname}.${var}.year.png");
393 }
394
395 print HTML2 "<img src=\"${hostname}.${var}.year.png\" width=\"${width}\" height=\"${height}\"><br>";
396
397 print HTML2 "</body></html>\n";
398 close(HTML2);
399 rename("${outdir}/${hostname}.${var}.html.new", "${outdir}/${hostname}.${var}.html");
400 }
401 }
402
403 print HTML "</body></html>\n";
404 print HTML "<br>Generated on: " . localtime(time());
405 print HTML ' by <a href="http://git.zerfleddert.de/cgi-bin/gitweb.cgi/upsgraph">upsgraph</a>.';
406
407 close(HTML);
408
409 rename("${outdir}/index.html.new", "${outdir}/index.html");
410
411 sleep(${step}/2);
412 }
Impressum, Datenschutz