X-Git-Url: http://cvs.zerfleddert.de/cgi-bin/gitweb.cgi/upsgraph/blobdiff_plain/6aa53fc5257183707fa0e2f56fb210b1fadad7a6..af4abf85a0f06568e1bed0cd2e3b36363b0b37ed:/upsgraph.pl

diff --git a/upsgraph.pl b/upsgraph.pl
index 073a439..bef10a5 100755
--- a/upsgraph.pl
+++ b/upsgraph.pl
@@ -1,7 +1,7 @@
 #!/usr/bin/perl -w
 
-if (@ARGV != 3) {
-	print STDERR "Syntax: ${0} host /path/to/file.rrd /output/directory/\n";
+if ((@ARGV != 3) && (@ARGV != 4)) {
+	print STDERR "Syntax: ${0} host /path/to/file.rrd /output/directory/ [uid]\n";
 	exit(1);
 }
 
@@ -97,6 +97,16 @@ my $vars = {
 
 use Net::SNMP;
 
+if ($> == 0) {
+	if (@ARGV != 4) {
+		print STDERR "Running as root, please provide UID as 4th argument!\n";
+		exit(1);
+	}
+
+	print "Running as root, switching to ".$ARGV[3]."\n";
+	$< = $> = $ARGV[3];
+}
+
 if (! -e "${rrdfile}") {
 	my $cmd = "rrdtool create \"${rrdfile}\" ";
 	foreach my $var (@fields) {
@@ -169,6 +179,7 @@ while(1) {
 
 	print HTML "</body></html>\n";
 	print HTML "<br>Generated on: " . localtime(time());
+	print HTML ' by <a href="http://git.zerfleddert.de/cgi-bin/gitweb.cgi/upsgraph">upsgraph</a>.';
 
 	close(HTML);