Commit ff1fcba4 authored by Leigh B Stoller's avatar Leigh B Stoller

Add script to create (and delete) image aliases:

	Usage: createimagealias [-r] <image> target1,target2,...
		-h   This message
	        -r   Delete alias
parent dda913b6
......@@ -1314,8 +1314,10 @@ use vars qw(@ISA @EXPORT $AUTOLOAD);
@ISA = "Exporter";
@EXPORT = qw ( );
use libtestbed;
use libdb;
use EmulabConstants;
use emutil;
use Image;
use OSinfo;
use overload ('""' => 'Stringify');
......@@ -1389,6 +1391,55 @@ sub isImageAlias($) { return 1; }
sub IsGeneric($) { return 0; }
sub path($) { return undef; }
#
# Create a new Image Alias.
#
sub Create($$$@)
{
my ($class, $name, $group, @targets) = @_;
#
# Grab unique ID. Table already locked.
#
my $imageid = TBGetUniqueIndex("next_osid");
my $uuid = NewUUID();
my $pid = $group->pid();
my $pid_idx = $group->pid_idx();
my $gid = $group->gid();
my $gid_idx = $group->gid_idx();
foreach my $target (@targets) {
my $target_name = $target->imagename();
my $target_imageid = $target->imageid();
DBQueryWarn("insert into image_aliases set ".
" imagename='$name',imageid='$imageid',uuid='$uuid', ".
" pid='$pid',pid_idx='$pid_idx',".
" gid='$gid',gid_idx='$gid_idx',".
" target_imagename='$target_name',".
" target_imageid='$target_imageid'")
or goto bad;
}
return OSImageAlias->Lookup($imageid);
bad:
DBQueryWarn("delete from image_aliases where imageid='$imageid'");
return -1;
}
#
# Delete.
#
sub Delete($)
{
my ($self) = @_;
my $imageid = $self->imageid();
return -1
if (!DBQueryWarn("delete from image_aliases ".
"where imageid='$imageid'"));
return 0;
}
#
# Lookup a "system" image, meaning one in the emulab-ops project.
#
......
......@@ -56,7 +56,7 @@ SBIN_SCRIPTS = vlandiff vlansync withadminprivs export_tables cvsupd.pl \
mktestbedtest fixrootcert addservers poolmonitor \
node_exclude managetaint shutdown-shared imagerelease \
runsonxen pxelinux_makeconf attend atten \
addrfdevice addrfpath reserve announce
addrfdevice addrfpath reserve announce createimagealias
WEB_SBIN_SCRIPTS= webnewnode webdeletenode webspewconlog webarchive_list \
webwanodecheckin webspewimage webdumpdescriptor webemulabfeature \
......
#!/usr/bin/perl -w
#
# Copyright (c) 2003-2016 University of Utah and the Flux Group.
#
# {{{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/>.
#
# }}}
#
use strict;
use English;
use Getopt::Std;
#
# Grant and revoke permission to use specific images.
#
sub usage()
{
print STDERR "Usage: createimagealias [-r] <image> target1,target2,...\n";
print STDERR " -h This message\n";
print STDERR " -r Delete alias\n";
exit(-1);
}
my $optlist = "r";
my $debug = 0;
my $remove = 0;
my @targets = ();
# Protos
sub fatal($);
#
# Configure variables
#
my $TB = "@prefix@";
#
# Testbed Support libraries
#
use lib "@prefix@/lib";
use emdb;
use EmulabConstants;
use libtestbed;
use Project;
use Group;
use User;
use OSImage;
#
# Turn off line buffering on output
#
$| = 1;
#
# Untaint the path
#
$ENV{'PATH'} = "/bin:/sbin:/usr/bin:";
#
# Parse command arguments. Once we return from getopts, all that should be
# left are the required arguments.
#
my %options = ();
if (! getopts($optlist, \%options)) {
usage();
}
if (defined($options{r})) {
$remove = 1;
}
usage()
if (@ARGV < 1);
my $imagename = shift(@ARGV);
#
# Verify user.
#
my $this_user = User->ThisUser();
if (! defined($this_user)) {
fatal("You ($UID) do not exist!");
}
# No aliases outside emulab-ops for now.
my $project = Project->Lookup(TBOPSPID());
fatal("Cannot lookup project " . TBOPSPID())
if (!defined($project));
my $pid = $project->pid();
if ($remove) {
my $image = OSImageAlias->Lookup($imagename);
fatal("No such image aliases exists")
if (!defined($image));
exit($image->Delete());
}
usage()
if (@ARGV < 1);
# Sanity check the images.
foreach my $imagename (@ARGV) {
my $image = OSImage->Lookup($imagename);
fatal("$pid,$imagename does not exist")
if (!defined($image));
fatal("Target image not in the same project as the alias")
if ($project->pid() ne $image->pid());
fatal("Target cannot be another alias!")
if ($image->isImageAlias());
my $tmp = $image->LookupMostRecent();
fatal("Target image $pid,$imagename is not the most recent")
if ($tmp->version() != $image->version());
push(@targets, $image);
}
my $image = OSImageAlias->Create($imagename,
$project->GetProjectGroup(), @targets);
fatal("Could not create image alias")
if (!defined($image));
exit(0);
sub fatal($)
{
my ($mesg) = $_[0];
die("*** $0:\n".
" $mesg\n");
}
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment