support for multiple hosts in one instance
[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 RRDs;
10 use Data::Dumper;
11
12 $UPSGRAPH::outdir = "";
13 $UPSGRAPH::step = 60;
14 $UPSGRAPH::keep = (370*24*60*60)/$UPSGRAPH::step;
15 $UPSGRAPH::hosts = ();
16
17 do $ARGV[0] or die "can't read config: $!";
18
19 my $outdir = $UPSGRAPH::outdir;
20 my $step = $UPSGRAPH::step;
21 my $keep = $UPSGRAPH::keep;
22 my $hosts = $UPSGRAPH::hosts;
23
24 sub rrdcreate(@) {
25 my $newrrd = shift;
26 my $field = shift;
27 my $vars = shift;
28 my $start = shift;
29
30 my @cmd = ("${newrrd}", "--step=${step}");
31
32 if (defined($start)) {
33 push @cmd, "--start=${start}";
34 }
35
36 push @cmd, "DS:${field}:GAUGE:600:" .
37 $vars->{$field}->{'min'} . ":" .
38 $vars->{$field}->{'max'} . " ";
39
40 push @cmd, "RRA:AVERAGE:0.5:1:${keep}";
41
42 RRDs::create(@cmd);
43 if (RRDs::error) {
44 print "Error while creating: " . RRDs::error . "\n";
45 exit 1;
46 }
47 }
48
49 sub fetch_snmp(@) {
50 my $address = shift;
51 my $community = shift;
52 my $oid = shift;
53
54 (my $session, my $error) = Net::SNMP->session(Hostname => $address,
55 Community => $community);
56
57 die "session error: $error" unless ($session);
58
59 $session->translate(0);
60
61 my $val = $session->get_request($oid);
62
63 $session->close;
64
65 $val;
66 }
67
68 if ($> == 0) {
69 if (@ARGV != 2) {
70 print STDERR "Running as root, please provide UID as 2th argument!\n";
71 exit(1);
72 }
73
74 print "Running as root, switching to ".$ARGV[1]."\n";
75 $< = $> = $ARGV[1];
76 }
77
78 foreach my $host (@$hosts) {
79 my $rrdfile = $host->{'rrdfile'};
80
81 if (-e "${rrdfile}") {
82 print "Reading old ${rrdfile} to preserve data...\n";
83
84 my $rrdinfo = RRDs::info("${rrdfile}");
85 if (RRDs::error) {
86 print "Error while getting info: " . RRDs::error . "\n";
87 exit 1;
88 }
89
90 (my $start, my $ostep, my $names, my $data) =
91 RRDs::fetch("${rrdfile}",
92 "-s " . (time() - ($rrdinfo->{'rra[0].rows'} * $rrdinfo->{'step'})),
93 "AVERAGE");
94
95 if (RRDs::error) {
96 print "Error while fetching data: " . RRDs::error . "\n";
97 exit 1;
98 }
99
100 foreach my $field (@$names) {
101 if (! -e "${rrdfile}.${field}") {
102 rrdcreate("${rrdfile}.${field}",
103 "${field}",
104 $host->{'vars'},
105 (${start}-${ostep}));
106 }
107 }
108
109 my $pos = $start;
110 foreach my $line (@$data) {
111 foreach my $field (@$names) {
112 my $val = shift (@$line);
113 $val = 'U' if (!defined($val));
114
115 RRDs::update("${rrdfile}.${field}", "${pos}:${val}");
116 if (RRDs::error) {
117 print "Can't insert data: " . RRDs::error . "\n";
118 exit 1;
119 }
120
121 }
122
123 $pos += $ostep;
124
125 if ((($pos-$start)/$ostep) == $#$data) {
126 last;
127 }
128 }
129
130 rename("${rrdfile}", "${rrdfile}.old") or die "Can't rename old file: $!\n";
131 }
132
133 foreach my $field (@{$host->{'fields'}}) {
134 if (! -e "${rrdfile}.${field}") {
135 print "Creating ${rrdfile}.${field}...\n";
136 rrdcreate("${rrdfile}.${field}",
137 "${field}",
138 $host->{'vars'});
139 }
140
141 my $rrdinfo = RRDs::info("${rrdfile}.${field}");
142 if (RRDs::error) {
143 print "Error while getting info: " . RRDs::error . "\n";
144 exit 1;
145 }
146
147 if ($rrdinfo->{'rra[0].rows'} != $keep) {
148 print "Resizing ${rrdfile}.${field} from " . $rrdinfo->{'rra[0].rows'} .
149 " to ${keep} samples.\n";
150
151 (my $start, my $ostep, my $names, my $data) =
152 RRDs::fetch("${rrdfile}.${field}",
153 "-s " . (time() - ($rrdinfo->{'rra[0].rows'} * $rrdinfo->{'step'})),
154 "AVERAGE");
155
156 if (RRDs::error) {
157 print "Error while fetching data: " . RRDs::error . "\n";
158 exit 1;
159 }
160
161 rrdcreate("${rrdfile}.${field}.new",
162 "${field}",
163 $host->{'vars'},
164 (${start}-${ostep}));
165
166 print "Preserving data since " . localtime($start) . "\n";
167
168 my $pos = $start;
169 foreach my $line (@$data) {
170 my $vline = "${pos}";
171
172 foreach my $val (@$line) {
173 $val = 'U' if (!defined($val));
174 $vline .= ":${val}";
175 }
176 RRDs::update("${rrdfile}.${field}.new", $vline) or die "Can't insert data\n";
177
178 if (RRDs::error) {
179 print "Error while updating: " . RRDs::error . "\n";
180 exit 1;
181 }
182 $pos += $ostep;
183
184 if ((($pos-$start)/$ostep) == $#$data) {
185 last;
186 }
187 }
188
189 rename("${rrdfile}.${field}", "${rrdfile}.${field}.old") or die "Can't rename old file: $!\n";
190 rename("${rrdfile}.${field}.new", "${rrdfile}.${field}") or die "Can't rename new file: $!\n";
191
192 $rrdinfo = RRDs::info("${rrdfile}.${field}");
193 if (RRDs::error) {
194 print "Error while getting info: " . RRDs::error . "\n";
195 exit 1;
196 }
197
198 if ($rrdinfo->{'rra[0].rows'} != $keep) {
199 print "Failed!\n";
200 exit 1;
201 }
202 }
203 }
204 }
205
206 my $child = fork();
207
208 die "fork failed!" if (!defined($child));
209
210 exit 0 if ($child != 0);
211
212 while(1) {
213 open(HTML, ">${outdir}/index.html.new");
214
215 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>USV status</title></head>';
216 print HTML '<body bgcolor="#ffffff">';
217
218 foreach my $host (@$hosts) {
219 my $vars = $host->{'vars'};
220 my $rrdfile = $host->{'rrdfile'};
221 my $hostname = $host->{'name'};
222
223 foreach my $var (@{$host->{'fields'}}) {
224 delete $vars->{$var}->{'value'};
225
226 my $result = fetch_snmp($host->{'address'}, $host->{'community'}, $vars->{$var}->{'oid'});
227 next unless (defined $result);
228
229 $vars->{$var}->{'value'} = $result->{$vars->{$var}->{'oid'}};
230 if (defined($vars->{$var}->{'factor'})) {
231 $vars->{$var}->{'value'} *= $vars->{$var}->{'factor'};
232 }
233 }
234
235 foreach my $var (@{$host->{'fields'}}) {
236 if (!(defined($vars->{$var}->{'value'}))) {
237 $vars->{$var}->{'value'} = 'U';
238 }
239 RRDs::update("${rrdfile}.${var}", "N:" . $vars->{$var}->{'value'});
240 }
241 if (RRDs::error) {
242 print "Error while updating: " . RRDs::error . "\n";
243 }
244
245 foreach my $var (@{$host->{'fields'}}) {
246 my @graphdef = ("-t", $hostname." - ".$vars->{$var}->{'name'}, "DEF:${var}=${rrdfile}.${var}:${var}:AVERAGE", "LINE1:${var}#FF0000");
247 (my $averages, my $width, my $height) =
248 RRDs::graph("${outdir}/${hostname}.${var}.png.new",
249 "-w", "720", @graphdef);
250
251 if (RRDs::error) {
252 print "Error while graphing: " . RRDs::error . "\n";
253 } else {
254 rename("${outdir}/${hostname}.${var}.png.new", "${outdir}/${hostname}.${var}.png");
255 }
256
257 print HTML "<a href=\"${hostname}.${var}.html\"><img src=\"${hostname}.${var}.png\" width=\"${width}\" height=\"${height}\" border=\"0\"></a>";
258
259 open (HTML2, ">${outdir}/${hostname}.${var}.html.new");
260 print HTML2 "<html><head><title>" . $vars->{$var}->{'name'} . "</title></head>";
261 print HTML2 '<body bgcolor="#ffffff">';
262
263
264 push @graphdef, "VDEF:min=${var},MINIMUM";
265 push @graphdef, "GPRINT:min:Minimum\\: %.2lf";
266
267 push @graphdef, "VDEF:avg=${var},AVERAGE";
268 push @graphdef, "GPRINT:avg:Average\\: %.2lf";
269
270 push @graphdef, "VDEF:max=${var},MAXIMUM";
271 push @graphdef, "GPRINT:max:Maximum\\: %.2lf";
272
273 push @graphdef, "VDEF:cur=${var},LAST";
274 push @graphdef, "GPRINT:cur:Current\\: %.2lf";
275
276 ($averages, $width, $height) =
277 RRDs::graph("${outdir}/${hostname}.${var}.long.png.new",
278 "-w", "1008", @graphdef);
279
280 if (RRDs::error) {
281 print "Error while graphing: " . RRDs::error . "\n";
282 } else {
283 rename("${outdir}/${hostname}.${var}.long.png.new", "${outdir}/${hostname}.${var}.long.png");
284 }
285
286 print HTML2 "<img src=\"${hostname}.${var}.long.png\" width=\"${width}\" height=\"${height}\"><br>";
287
288 ($averages, $width, $height) =
289 RRDs::graph("${outdir}/${hostname}.${var}.week.png.new",
290 "-w", "1008", "-e", "now", "-s", "end-1w", @graphdef);
291
292 if (RRDs::error) {
293 print "Error while graphing: " . RRDs::error . "\n";
294 } else {
295 rename("${outdir}/${hostname}.${var}.week.png.new", "${outdir}/${hostname}.${var}.week.png");
296 }
297
298 print HTML2 "<img src=\"${hostname}.${var}.week.png\" width=\"${width}\" height=\"${height}\"><br>";
299
300 ($averages, $width, $height) =
301 RRDs::graph("${outdir}/${hostname}.${var}.year.png.new",
302 "-w", "1008", "-e", "now", "-s", "end-1y", @graphdef);
303
304 if (RRDs::error) {
305 print "Error while graphing: " . RRDs::error . "\n";
306 } else {
307 rename("${outdir}/${hostname}.${var}.year.png.new", "${outdir}/${hostname}.${var}.year.png");
308 }
309
310 print HTML2 "<img src=\"${hostname}.${var}.year.png\" width=\"${width}\" height=\"${height}\"><br>";
311
312 print HTML2 "</body></html>\n";
313 close(HTML2);
314 rename("${outdir}/${hostname}.${var}.html.new", "${outdir}/${hostname}.${var}.html");
315 }
316 }
317
318 print HTML "</body></html>\n";
319 print HTML "<br>Generated on: " . localtime(time());
320 print HTML ' by <a href="http://git.zerfleddert.de/cgi-bin/gitweb.cgi/upsgraph">upsgraph</a>.';
321
322 close(HTML);
323
324 rename("${outdir}/index.html.new", "${outdir}/index.html");
325
326 sleep(${step}/2);
327 }
Impressum, Datenschutz