deleteimagedata.in 5.23 KB
Newer Older
1 2
#!/usr/bin/perl -w
#
3
# Copyright (c) 2008-2017 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 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40
# 
# {{{GENIPUBLIC-LICENSE
# 
# GENI Public License
# 
# Permission is hereby granted, free of charge, to any person obtaining
# a copy of this software and/or hardware specification (the "Work") to
# deal in the Work without restriction, including without limitation the
# rights to use, copy, modify, merge, publish, distribute, sublicense,
# and/or sell copies of the Work, and to permit persons to whom the Work
# is furnished to do so, subject to the following conditions:
# 
# The above copyright notice and this permission notice shall be
# included in all copies or substantial portions of the Work.
# 
# THE WORK IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
# OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
# MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
# NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
# HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
# WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
# OUT OF OR IN CONNECTION WITH THE WORK OR THE USE OR OTHER DEALINGS
# IN THE WORK.
# 
# }}}
#
use strict;
use English;
use Getopt::Std;
use Date::Parse;
use Data::Dumper;

#
# Delete image data from the image server.
# 
sub usage()
{
41 42 43
    print "Usage: deleteimagedata [-dn] <image urn> <image uuid>\n";
    print "Usage: deleteimagedata [-dn] <imageid>\n";
    print "Usage: deleteimagedata [-dn] prune\n";
44 45
    exit(1);
}
46
my $optlist  = "dn";
47
my $debug    = 0;
48
my $impotent = 0;
49
my $errormsg;
50
my @images = ();
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

#
# Configure variables
#
my $TB		  = "@prefix@";
my $TBOPS         = "@TBOPSEMAIL@";
my $TBLOGS        = "@TBLOGSEMAIL@";
my $PGENIDOMAIN   = "@PROTOGENI_DOMAIN@";
my $PGENISUPPORT  = @PROTOGENI_SUPPORT@;
my $PGENILOCALUSER= @PROTOGENI_LOCALUSER@;
my $CMCERT	  = "$TB/etc/genicm.pem";
my $OURDOMAIN     = "@OURDOMAIN@";
my $CLIENTVERS    = 1;

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

# Protos
sub fatal($);

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

# Now we can load the libraries after setting the proper DB.
use lib '@prefix@/lib';
79
use emdb;
80 81
use libaudit;
use emutil;
82
use GeniDB qw(GENICM_DBNAME DBConnect);
83 84 85
use GeniCertificate;
use GeniAuthority;
use GeniImage;
86
use GeniResponse;
87 88
use GeniHRN;
use libEmulab;
89
use OSImage;
90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109

if (!$PGENISUPPORT) {
    print STDERR "You are not a protogeni site\n";
    exit(0);
}
if (!GetSiteVar("protogeni/use_imagetracker")) {
    print STDERR "Image tracking is not enabled for this site\n";
    exit(0);
}

#
# Check args.
#
my %options = ();
if (! getopts($optlist, \%options)) {
    usage();
}
if (defined($options{"d"})) {
    $debug = 1;
}
110 111 112
if (defined($options{"n"})) {
    $impotent = 1;
}
113 114 115 116 117
if (@ARGV == 2) {
    usage()
	if (! GeniHRN::IsValid($ARGV[0]));
    usage()
	if (! ValidUUID($ARGV[1]));
118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143
    my ($imageurn,$imageuuid) = @ARGV;
    push(@images, [$imageurn, $imageuuid]);
}
elsif (@ARGV == 1 && $ARGV[0] eq "prune") {
    my $query_result =
	emdb::DBQueryFatal("select pid,gid,imagename,isdataset,uuid ".
			   "   from image_versions ".
			   "where deleted is not null and ".
			   "      ims_reported is not null");

    while (my ($pid,$gid,$name,$isdataset,$uuid) =
	   $query_result->fetchrow_array()) {
	my $domain = $OURDOMAIN;
	my $urn;

	if ($isdataset) {
	    $domain .= ":${pid}";
	    $domain .= ":${gid}" if ($pid ne $gid);
	    
	    $urn = GeniHRN::Generate($domain, "imdataset", $name);
	}
	else {
	    $urn = GeniHRN::Generate($domain, "image", $pid . "//" . $name);
	}
	push(@images, [$urn, $uuid]);
    }
144 145 146 147 148 149
}
elsif (@ARGV == 1) {
    my $image = OSImage->Lookup($ARGV[0]);
    if (!defined($image)) {
	fatal("No such image!");
    }
150
    my $imageurn = $image->urn();
151 152 153 154 155 156
    
    #
    # If no version on the command line, we want to delete the entire
    # descriptor at the server.
    #
    if ($ARGV[0] =~ /:\d+$/) {
157
	push(@images, [$imageurn, $image->uuid()]);
158 159
    }
    else {
160
	push(@images, [$imageurn, $image->image_uuid()]);
161 162 163 164 165
    }
}
else {
    usage();
}
166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184

# Connect to the proper DB.
DBConnect(GENICM_DBNAME());

#
# Load the CM cert to act as caller context.
#
my $certificate = GeniCertificate->LoadFromFile($CMCERT);
if (!defined($certificate)) {
    fatal("Could not load certificate from $CMCERT\n");
}
my $context = Genixmlrpc->Context($certificate);
if (!defined($context)) {
    fatal("Could not create context to talk to image server");
}
Genixmlrpc->SetContext($context);
# Shorten default timeout.
Genixmlrpc->SetTimeout(15);

185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209
foreach my $ref (@images) {
    my ($imageurn, $imageuuid) = @{ $ref };

    if ($impotent) {
	print "Would ask IMS to delete $imageurn\n";
	next;
    }
    # Retry a few times if we get an RPC error, usually transient.
    my $count = 3;
    while ($count) {
	my $rval = GeniImage::DeleteImageData($imageurn,
					      $imageuuid, \$errormsg);
	last
	    if ($rval == 0);

	fatal($errormsg)
	    if ($rval != GENIRESPONSE_RPCERROR);

	$count--;
	if ($count == 0) {
	    fatal("Too many RPC errors for $imageurn,$imageuuid");
	}
	print "Transient RPC error, will retry in five seconds\n";
	sleep(5);
    }
210 211 212 213 214 215 216 217 218 219 220 221
}
exit(0);

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

    print STDERR "*** $0:\n".
	         "    $msg\n";
    # exit value important.
    exit(-1);
}