dumperrorlog.proxy.in 2.16 KB
Newer Older
1 2
#!/usr/bin/perl -wT
#
3
# Copyright (c) 2005-2011 University of Utah and the Flux Group.
4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22
# 
# {{{EMULAB-LICENSE
# 
# This file is part of the Emulab network testbed software.
# 
# This file is free software: you can redistribute it and/or modify it
# under the terms of the GNU Affero General Public License as published by
# the Free Software Foundation, either version 3 of the License, or (at
# your option) any later version.
# 
# This file is distributed in the hope that it will be useful, but WITHOUT
# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
# FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Affero General Public
# License for more details.
# 
# You should have received a copy of the GNU Affero General Public License
# along with this file.  If not, see <http://www.gnu.org/licenses/>.
# 
# }}}
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
#
use English;

#
# usage: dumperrorlog.proxy
#
sub usage() {
    print STDOUT "Usage: dumperrorlog.proxy tablename\n";
    exit(-1);
}
my $tablename;

#
# Configure variables
#
my $PWDFILE   = "/usr/testbed/etc/errorlog.pwd";
my $tempfile  = "testbed_dumperrorlog_tempfile";
my $DBNAME    = "errorlog";
my $DBUSER    = "errorlog";

#
# We don't want to run this script unless its the real version.
#
if ($UID != 0) {
    die("*** $0:\n".
	"    Must be root!\n");
}

# un-taint path
$ENV{'PATH'} = '/bin:/sbin:/usr/bin:/usr/sbin:/usr/local/bin';
delete @ENV{'IFS', 'CDPATH', 'ENV', 'BASH_ENV'};

#
# Turn off line buffering on output
#
$| = 1;

# Load the Testbed support stuff.
use lib "@prefix@/lib";
use libtestbed;
63
use libtbdb;
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

sub fatal($) {
    my ($msg) = @_;

    die("*** $0:\n".
	"    $msg\n");
}

if (@ARGV != 1) {
    usage();
}
$tablename = $ARGV[0];

#
# Untaint the arguments.
#
if ($tablename =~ /^([-\w]+)$/) {
    $tablename = $1;
}
else {
    die("Tainted argument $tablename!\n");
}

#
# Grab the password we need.
#
my $pwd = `cat $PWDFILE`;
if ($pwd =~ /^([\w]*)$/) {
    $pwd = $1;
}
else {
    fatal("Bad characters in password!");
}

98 99
if (TBDBConnect($DBNAME, $DBUSER, $pwd) < 0) {
    fatal("Could not connect to errorlog database!");
100
}
101 102
DBQueryFatal("load data infile '${tempfile}.${tablename}' ".
	     "replace into table $tablename");
103 104
exit(0);