All new accounts created on Gitlab now require administrator approval. If you invite any collaborators, please let Flux staff know so they can approve the accounts.

changepid.in 3.68 KB
Newer Older
1 2
#!/usr/bin/perl -w
#
Leigh B Stoller's avatar
Leigh B Stoller committed
3
# Copyright (c) 2006-2015 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 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 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135
#
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 B. Stoller's avatar
Leigh B. Stoller committed
136
	"    Project '$oldpid' must be an unapproved project!\n");
137 138 139 140 141 142
}

#
# Change a bunch of tables!
#
print "Updating group_membership table ...\n";
Leigh B. Stoller's avatar
Leigh B. Stoller committed
143
DBQueryFatal("update group_membership set pid='$newpid',gid='$newpid' ".
144 145 146
	     "where pid='$oldpid'");

print "Updating groups table ...\n";
Leigh B Stoller's avatar
Leigh B Stoller committed
147
DBQueryFatal("update groups set pid='$newpid',gid=pid,unix_name='$newpid' ".
148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163
	     "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'");