changepid.in 2.98 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 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 63 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 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118
#!/usr/bin/perl -w
#
# EMULAB-COPYRIGHT
# Copyright (c) 2006 University of Utah and the Flux Group.
# All rights reserved.
#
use English;
use Getopt::Std;

#
# Change the pid of an UNAPPROVED project.
#
sub usage()
{
    print(STDERR "Usage: changepid <oldpid> <newpid>\n");
    exit(-1);
}
my $optlist = "df";
my $debug   = 0;
my $force   = 0;

#
# Configure variables
#
my $TB		= "@prefix@";

#
# Testbed Support libraries
#
use lib "@prefix@/lib";
use libaudit;
use libdb;
use libtestbed;

# Be careful not to exit on transient error
$libdb::DBQUERY_MAXTRIES = 30;

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

#
# Untaint the path
# 
$ENV{'PATH'} = "/bin:/sbin:/usr/bin:/usr/sbin";
delete @ENV{'IFS', 'CDPATH', 'ENV', 'BASH_ENV'};

#
# This script is setuid, so please do not run it as root. Hard to track
# what has happened.
#
if ($UID == 0) {
    die("*** $0:\n".
	"    Please do not run this as root! Its already setuid!\n");
}

# Only admin types!
if (!TBAdmin($UID)) {
    die("*** $0:\n".
	"    Only TB administrators can run this script!\n");
}

#
# Parse command arguments. Once we return from getopts, all that should be
# left are the required arguments.
#
%options = ();
if (! getopts($optlist, \%options)) {
    usage();
}
if (defined($options{"d"})) {
    $debug = 1;
}
if (defined($options{"f"})) {
    $force = 1;
}
if (@ARGV != 2) {
    usage();
}
my $oldpid = shift(@ARGV);
my $newpid = shift(@ARGV);

#
# This script is always audited. Mail is sent automatically upon exit.
#
if (AuditStart(0)) {
    #
    # Parent exits normally
    #
    exit(0);
}

#
# Make sure new project is not already taken.
# 
$query_result =
    DBQueryFatal("select approved from projects where pid='$newpid'");

if ($query_result->numrows) {
    die("*** $0:\n".
	"    Project '$newpid' already exists in the DB!\n");
}

#
# Get the project status. Only unapproved; too hard otherwise.
#
$query_result =
    DBQueryFatal("select approved from projects where pid='$oldpid'");

if ($query_result->numrows == 0) {
    die("*** $0:\n".
	"    No such project '$oldpid' in the DB!\n");
}
my ($approved) = $query_result->fetchrow_array();

if ($approved) {
    die("*** $0:\n".
Leigh Stoller's avatar
Leigh Stoller committed
119
	"    Project '$oldpid' must be an unapproved project!\n");
120 121 122 123 124 125
}

#
# Change a bunch of tables!
#
print "Updating group_membership table ...\n";
Leigh Stoller's avatar
Leigh Stoller committed
126
DBQueryFatal("update group_membership set pid='$newpid',gid='$newpid' ".
127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146
	     "where pid='$oldpid'");

print "Updating groups table ...\n";
DBQueryFatal("update groups set pid='$newpid',unix_name='$newpid' ".
	     "where pid='$oldpid'");

print "Updating group_stats table ...\n";
DBQueryFatal("update group_stats set pid='$newpid',gid='$newpid' ".
	     "where pid='$oldpid'");

print "Updating project_stats table ...\n";
DBQueryFatal("update project_stats set pid='$newpid' ".
	     "where pid='$oldpid'");

#
# Lastly change projects table; if anything breaks above we can still run
# this script to finish the job.
#
print "Updating projects table ...\n";
DBQueryFatal("update projects set pid='$newpid' where pid='$oldpid'");