GeniSliver.pm.in 24.5 KB
Newer Older
Leigh B. Stoller's avatar
Leigh B. Stoller committed
1
2
#!/usr/bin/perl -wT
#
3
# GENIPUBLIC-COPYRIGHT
4
# Copyright (c) 2008-2009 University of Utah and the Flux Group.
Leigh B. Stoller's avatar
Leigh B. Stoller committed
5
6
7
8
9
10
11
12
13
14
15
16
17
# All rights reserved.
#
package GeniSliver;

#
use strict;
use Exporter;
use vars qw(@ISA @EXPORT);

@ISA    = "Exporter";
@EXPORT = qw ( );

use GeniDB;
Leigh B. Stoller's avatar
Leigh B. Stoller committed
18
use GeniComponent;
Leigh B. Stoller's avatar
Leigh B. Stoller committed
19
use GeniSlice;
Leigh B. Stoller's avatar
Leigh B. Stoller committed
20
use GeniCredential;
21
use GeniCertificate;
Leigh B. Stoller's avatar
Leigh B. Stoller committed
22
use GeniAggregate;
23
use GeniUsage;
24
use GeniHRN;
Leigh B. Stoller's avatar
Leigh B. Stoller committed
25
# Hate to import all this crap; need a utility library.
26
use emutil;
Leigh B. Stoller's avatar
Leigh B. Stoller committed
27
use Experiment;
28
use OSinfo;
29
use Node;
Leigh B. Stoller's avatar
Leigh B. Stoller committed
30
use English;
Leigh B. Stoller's avatar
Leigh B. Stoller committed
31
use XML::Simple;
Leigh B. Stoller's avatar
Leigh B. Stoller committed
32
33
use Data::Dumper;
use File::Temp qw(tempfile);
Leigh B. Stoller's avatar
Leigh B. Stoller committed
34
use overload ('""' => 'Stringify');
Leigh B. Stoller's avatar
Leigh B. Stoller committed
35
36
37
38
39
40
41
42

# Configure variables
my $TB		   = "@prefix@";
my $TBOPS          = "@TBOPSEMAIL@";
my $TBAPPROVAL     = "@TBAPPROVALEMAIL@";
my $TBAUDIT   	   = "@TBAUDITEMAIL@";
my $BOSSNODE       = "@BOSSNODE@";
my $OURDOMAIN      = "@OURDOMAIN@";
43
my $PGENIDOMAIN    = "@PROTOGENI_DOMAIN@";
Leigh B. Stoller's avatar
Leigh B. Stoller committed
44
my $SIGNCRED	   = "$TB/sbin/signgenicred";
Leigh B. Stoller's avatar
Leigh B. Stoller committed
45
46
47
my $AVAIL	   = "$TB/sbin/avail";
my $NALLOC	   = "$TB/bin/nalloc";
my $NFREE	   = "$TB/bin/nfree";
Leigh B. Stoller's avatar
Leigh B. Stoller committed
48
my $NODEREBOOT	   = "$TB/bin/node_reboot";
49
my $NAMEDSETUP     = "$TB/sbin/named_setup";
Leigh B. Stoller's avatar
Leigh B. Stoller committed
50
51
my $PLABNODE       = "$TB/sbin/plabnodewrapper";
my $VNODESETUP     = "$TB/sbin/vnode_setup";
52
my $GENTOPOFILE    = "$TB/libexec/gentopofile";
Leigh B. Stoller's avatar
Leigh B. Stoller committed
53
54
55

# Cache of instances to avoid regenerating them.
my %slivers      = ();
56
BEGIN { use GeniUtil; GeniUtil::AddCache(\%slivers); }
Leigh B. Stoller's avatar
Leigh B. Stoller committed
57
58
59
60
61
62
63
64

#
# Lookup by idx, or uuid.
#
sub Lookup($$)
{
    my ($class, $token) = @_;
    my $query_result;
Leigh B. Stoller's avatar
Leigh B. Stoller committed
65
    my $idx;
Leigh B. Stoller's avatar
Leigh B. Stoller committed
66
67

    if ($token =~ /^\d+$/) {
Leigh B. Stoller's avatar
Leigh B. Stoller committed
68
	$idx = $token;
Leigh B. Stoller's avatar
Leigh B. Stoller committed
69
70
71
    }
    elsif ($token =~ /^\w+\-\w+\-\w+\-\w+\-\w+$/) {
	$query_result =
Leigh B. Stoller's avatar
Leigh B. Stoller committed
72
	    DBQueryWarn("select idx from geni_slivers ".
Leigh B. Stoller's avatar
Leigh B. Stoller committed
73
			"where uuid='$token'");
Leigh B. Stoller's avatar
Leigh B. Stoller committed
74
75
76
77
	    return undef
		if (! $query_result || !$query_result->numrows);

	    ($idx) = $query_result->fetchrow_array();
Leigh B. Stoller's avatar
Leigh B. Stoller committed
78
79
80
81
    }
    else {
	return undef;
    }
Leigh B. Stoller's avatar
Leigh B. Stoller committed
82
83
84
85
86
87
88
89

    # Look in cache first
    return $slivers{"$idx"}
        if (exists($slivers{"$idx"}));

    $query_result = DBQueryWarn("select * from geni_slivers ".
				"where idx='$idx'");

Leigh B. Stoller's avatar
Leigh B. Stoller committed
90
91
92
    return undef
	if (!$query_result || !$query_result->numrows);

Leigh B. Stoller's avatar
Leigh B. Stoller committed
93
94
    my $self              = {};
    $self->{'SLIVER'}     = $query_result->fetchrow_hashref();
95
    $self->{'SLICE'}      = undef;	# server
Leigh B. Stoller's avatar
Leigh B. Stoller committed
96
    $self->{'AGGREGATE'}  = undef;	# server
97
    $self->{'RSPEC'}      = undef;	# server
Leigh B. Stoller's avatar
Leigh B. Stoller committed
98
99
100
101
102
103

    my $rspec_string = $self->{'SLIVER'}->{'rspec_string'};
    if (defined($rspec_string) && $rspec_string ne "") {
	$self->{'RSPEC'} = XMLin($rspec_string,
				 ForceArray => ["node", "link"]);
    }
104
105
106
107
108
109
110
111
112
113
114

    #
    # Grab the certificate, since we will probably want it.
    #
    my $uuid = $self->{'SLIVER'}->{'uuid'};
    my $certificate = GeniCertificate->Lookup($uuid);
    if (!defined($certificate)) {
	print STDERR "Could not find certificate for sliver $idx ($uuid)\n";
	return undef;
    }
    $self->{'CERTIFICATE'} = $certificate;
115
116
117
118

    # Bless into sub package if called for.
    my $resource_type = $self->{'SLIVER'}->{'resource_type'};
    if (defined($resource_type) && $resource_type ne "") {
119
	bless($self, $class . "::" . $resource_type);
120
121
122
123
    }
    else {
	bless($self, $class);
    }
Leigh B. Stoller's avatar
Leigh B. Stoller committed
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
    
    # Add to cache. 
    $slivers{$self->{'SLIVER'}->{'idx'}} = $self;
    
    return $self;
}

#
# Stringify for output.
#
sub Stringify($)
{
    my ($self) = @_;
    
    my $uuid = $self->uuid();
    my $idx  = $self->idx();

    return "[GeniSliver: $uuid, IDX: $idx]";
}

#
Leigh B. Stoller's avatar
Leigh B. Stoller committed
145
# Create a sliver record in the DB. On the client side we save the credential
Leigh B. Stoller's avatar
Leigh B. Stoller committed
146
# that allows control of it, for later operations.
Leigh B. Stoller's avatar
Leigh B. Stoller committed
147
#
148
sub Create($$$$$$$$$)
Leigh B. Stoller's avatar
Leigh B. Stoller committed
149
{
150
    my ($class, $slice, $owner, $uuid, $resource_uuid, $resource_type,
151
	$hrn, $nickname, $rspec) = @_;
Leigh B. Stoller's avatar
Leigh B. Stoller committed
152
    my @insert_data = ();
153
    my $certificate;
Leigh B. Stoller's avatar
Leigh B. Stoller committed
154
155
156

    # Every sliver gets a new unique index.
    my $idx = TBGetUniqueIndex('next_sliver', 1);
157
158
159
160
161
162
163
164
    
    # Create a cert pair, for this resource uuid.
    if (defined($uuid) && $resource_type eq "Node") {
	$certificate = GeniCertificate->Lookup($uuid);

	if (defined($certificate) && $certificate->hrn() ne $hrn) {
	    print STDERR "GeniSliver::Create: ".
		"Already have a certificate for $hrn/$uuid\n";
165
166
	    return undef;
	}
Leigh B. Stoller's avatar
Leigh B. Stoller committed
167
    }
168
169
170
    $hrn =~ /.*([^.]+)$/;
    my $urn = GeniHRN::Generate( "@OURDOMAIN@", "sliver", $1 );
    $certificate = GeniCertificate->Create("sliver", $urn, $hrn, $TBOPS, $uuid)
171
	if (!defined($certificate));
172
	
173
174
175
176
    if (!defined($certificate)) {
	print STDERR "GeniSliver::Create: ".
	    "Could not generate new certificate and UUID for $hrn/$uuid\n";
	return undef;
Leigh B. Stoller's avatar
Leigh B. Stoller committed
177
    }
Leigh B. Stoller's avatar
Leigh B. Stoller committed
178
    my $slice_uuid     = $slice->uuid();
179
    my $owner_uuid     = $owner->uuid();
Leigh B. Stoller's avatar
Leigh B. Stoller committed
180
181
182
183

    # Now tack on other stuff we need.
    push(@insert_data, "created=now()");
    push(@insert_data, "idx='$idx'");
184
185
186
    push(@insert_data, "hrn=" . DBQuoteSpecial($hrn));
    push(@insert_data, "nickname=" . DBQuoteSpecial($nickname))
	if (defined($nickname));
Leigh B. Stoller's avatar
Leigh B. Stoller committed
187
    push(@insert_data, "uuid='$uuid'");
188
189
    push(@insert_data, "resource_uuid='$resource_uuid'");
    push(@insert_data, "resource_type='$resource_type'");
Leigh B. Stoller's avatar
Leigh B. Stoller committed
190
191
192
    push(@insert_data, "creator_uuid='$owner_uuid'");
    push(@insert_data, "slice_uuid='$slice_uuid'");

Leigh B. Stoller's avatar
Leigh B. Stoller committed
193
194
195
196
197
198
199
    if (defined($rspec)) {
	my $rspec_string = XMLout($rspec, RootName => "rspec");
	my $safe_rspec   = DBQuoteSpecial($rspec_string);

	push(@insert_data, "rspec_string=$safe_rspec");
    }

Leigh B. Stoller's avatar
Leigh B. Stoller committed
200
    # Insert into DB.
201
202
203
    if (!DBQueryWarn("insert into geni_slivers set " .
		     join(",", @insert_data))) {
	$certificate->Delete();
Leigh B. Stoller's avatar
Leigh B. Stoller committed
204
205
206
	return undef;
    }

Leigh B. Stoller's avatar
Leigh B. Stoller committed
207
208
209
    my $sliver = GeniSliver->Lookup($idx);
    return undef
	if (!defined($sliver));
210
211
212
213
214

    if (GeniUsage->NewSliver($sliver, $slice, $owner)) {
	print STDERR
	    "GeniSliver::Create: GeniUsage->NewSliver($sliver) failed\n";
    }
Leigh B. Stoller's avatar
Leigh B. Stoller committed
215
    $sliver->{'AGGREGATE'} = undef;
Leigh B. Stoller's avatar
Leigh B. Stoller committed
216
    $sliver->{'SLICE'}     = undef;
Leigh B. Stoller's avatar
Leigh B. Stoller committed
217
218

    return $sliver;
Leigh B. Stoller's avatar
Leigh B. Stoller committed
219
220
221
222
223
}
# accessors
sub field($$) { return ((! ref($_[0])) ? -1 : $_[0]->{'SLIVER'}->{$_[1]}); }
sub idx($)		{ return field($_[0], "idx"); }
sub uuid($)		{ return field($_[0], "uuid"); }
224
225
sub hrn($)		{ return field($_[0], "hrn"); }
sub nickname($)		{ return field($_[0], "nickname"); }
Leigh B. Stoller's avatar
Leigh B. Stoller committed
226
227
228
sub slice_uuid($)	{ return field($_[0], "slice_uuid"); }
sub creator_uuid($)	{ return field($_[0], "creator_uuid"); }
sub created($)		{ return field($_[0], "created"); }
Leigh B. Stoller's avatar
Leigh B. Stoller committed
229
sub credential_idx($)	{ return field($_[0], "credential_idx"); }
Leigh B. Stoller's avatar
Leigh B. Stoller committed
230
sub resource_uuid($)	{ return field($_[0], "resource_uuid"); }
231
sub resource_type($)	{ return field($_[0], "resource_type"); }
Leigh B. Stoller's avatar
Leigh B. Stoller committed
232
233
234
sub component_uuid($)	{ return field($_[0], "component_uuid"); }
sub aggregate_uuid($)	{ return field($_[0], "aggregate_uuid"); }
sub rspec_string($)     { return field($_[0], "rspec_string"); }
Leigh B. Stoller's avatar
Leigh B. Stoller committed
235
sub status($)		{ return field($_[0], "status"); }
236
sub state($)		{ return field($_[0], "state"); }
237
238
sub cert($)		{ return $_[0]->{'CERTIFICATE'}->cert(); }
sub GetCertificate($)   { return $_[0]->{'CERTIFICATE'}; }
Leigh B. Stoller's avatar
Leigh B. Stoller committed
239
sub rspec($)            { return $_[0]->{'RSPEC'}; }
Leigh B. Stoller's avatar
Leigh B. Stoller committed
240
241
242
243

#
# Delete the sliver. The sliver should not be provisioned when this done.
#
244
sub Delete($$)
Leigh B. Stoller's avatar
Leigh B. Stoller committed
245
{
246
    my ($self, $purge) = @_;
Leigh B. Stoller's avatar
Leigh B. Stoller committed
247
248
249
250

    return -1
	if (! ref($self));

251
252
    my $idx  = $self->idx();
    my $uuid = $self->uuid();
Leigh B. Stoller's avatar
Leigh B. Stoller committed
253

254
255
256
257
    if (GeniUsage->DestroySliver($self, $purge)) {
	print STDERR
	    "GeniSliver::Delete: GeniUsage->DestroySliver($self) failed\n";
    }
258
259
    DBQueryWarn("delete from geni_credentials where this_uuid='$uuid'")
	or return -1;
260
261
262
263
    if ($self->resource_type() ne "Node") {
	DBQueryWarn("delete from geni_certificates where uuid='$uuid'")
	    or return -1;
    }
Leigh B. Stoller's avatar
Leigh B. Stoller committed
264
265
    DBQueryWarn("delete from geni_slivers where idx='$idx'")
	or return -1;
266

267
268
269
    # Delete from cache. 
    delete($slivers{$idx});
    
Leigh B. Stoller's avatar
Leigh B. Stoller committed
270
271
272
    return 0;
}

Leigh B. Stoller's avatar
Leigh B. Stoller committed
273
274
275
276
277
278
279
280
281
282
#
# Set the aggregate for a sliver.
#
sub SetAggregate($$)
{
    my ($self, $aggregate) = @_;

    return -1
	if (! (ref($self) && ref($aggregate)));

Leigh B. Stoller's avatar
Leigh B. Stoller committed
283
284
    my $idx      = $self->idx();
    my $agg_uuid = $aggregate->uuid();
Leigh B. Stoller's avatar
Leigh B. Stoller committed
285
286
287

    return -1
	if (!DBQueryWarn("update geni_slivers set ".
Leigh B. Stoller's avatar
Leigh B. Stoller committed
288
			 "  aggregate_uuid='$agg_uuid' ".
Leigh B. Stoller's avatar
Leigh B. Stoller committed
289
290
			 "where idx='$idx'"));
    
291
292
293
294
295
296
297
    if (!DBQueryWarn("update sliver_history set ".
		     "  aggregate_uuid='$agg_uuid' ".
		     "where idx='$idx'")) {
	print STDERR "GeniSliver::SetAggregate: ".
	    "Failed to update sliver_history for $self\n";
    }

Leigh B. Stoller's avatar
Leigh B. Stoller committed
298
    $self->{'SLIVER'}->{'aggregate_uuid'} = $agg_uuid;
Leigh B. Stoller's avatar
Leigh B. Stoller committed
299
300
301
302
    $self->{'AGGREGATE'} = $aggregate;
    return 0;
}

303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
#
# And clear the aggregate.
#
sub ClearAggregate($$)
{
    my ($self) = @_;

    return -1
	if (! ref($self));

    my $idx      = $self->idx();

    return -1
	if (!DBQueryWarn("update geni_slivers set ".
			 "  aggregate_uuid=NULL ".
			 "where idx='$idx'"));
    
    $self->{'SLIVER'}->{'aggregate_uuid'} = undef;
    $self->{'AGGREGATE'} = undef;
    return 0;
}

Leigh B. Stoller's avatar
Leigh B. Stoller committed
325
326
327
328
329
330
331
332
333
334
335
336
#
# Get the aggregate for a sliver.
#
sub GetAggregate($)
{
    my ($self) = @_;

    return undef
	if (! ref($self));

    return $self->{'AGGREGATE'} if (defined($self->{'AGGREGATE'}));
    return undef
Leigh B. Stoller's avatar
Leigh B. Stoller committed
337
	if (!defined($self->aggregate_uuid()));
Leigh B. Stoller's avatar
Leigh B. Stoller committed
338

Leigh B. Stoller's avatar
Leigh B. Stoller committed
339
    my $aggregate = GeniAggregate->Lookup($self->aggregate_uuid());
Leigh B. Stoller's avatar
Leigh B. Stoller committed
340
341
342
343
344
345
346
347
    if (!defined($aggregate)) {
	print STDERR "Could not get aggregate object associated with $self\n";
	return undef;
    }
    $self->{'AGGREGATE'} = $aggregate;
    return $aggregate;
}

348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
#
# Set the status for the sliver.
#
sub SetStatus($$)
{
    my ($self, $status) = @_;

    return undef
	if (! ref($self));

    my $idx = $self->idx();
    
    return -1
	if (!DBQueryWarn("update geni_slivers set ".
			 "  status='$status' ".
			 "where idx='$idx'"));
    
    $self->{'SLIVER'}->{'status'} = $status;
    return 0;
}

369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
#
# Set the state for the sliver.
#
sub SetState($$)
{
    my ($self, $state) = @_;

    return undef
	if (! ref($self));

    my $idx = $self->idx();
    
    return -1
	if (!DBQueryWarn("update geni_slivers set ".
			 "  state='$state' ".
			 "where idx='$idx'"));
    
    $self->{'SLIVER'}->{'state'} = $state;
    return 0;
}

Leigh B. Stoller's avatar
Leigh B. Stoller committed
390
391
392
393
394
395
396
397
398
399
400
401
402
#
# Get the experiment for the slice this sliver belongs to.
#
sub GetExperiment($)
{
    my ($self) = @_;

    return undef
	if (! ref($self));

    return Experiment->Lookup($self->slice_uuid());
}

Leigh B. Stoller's avatar
Leigh B. Stoller committed
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
#
# Get the slice for the sliver.
#
sub GetSlice($)
{
    my ($self) = @_;

    return undef
	if (! ref($self));

    return $self->{'SLICE'} if (defined($self->{'SLICE'}));

    if (!defined($self->slice_uuid())) {
	print STDERR "No slice associated with $self\n";
	return undef;
    }
    my $slice = GeniSlice->Lookup($self->slice_uuid());
    if (!defined($slice)) {
	print STDERR "Could not get slice object associated with $self\n";
	return undef;
    }
    $self->{'SLICE'} = $slice;
    return $slice;
}

428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
#
# Look up a list of slivers for a locally instantiated slice. 
# Used by the CM.
#
sub SliceSlivers($$$)
{
    my ($class, $slice, $pref) = @_;

    my $slice_uuid = $slice->uuid();
    my @result = ();

    my $query_result =
	DBQueryWarn("select idx from geni_slivers ".
		    "where slice_uuid='$slice_uuid'");
    return -1
	if (!$query_result);

    while (my ($idx) = $query_result->fetchrow_array()) {
	my $sliver = GeniSliver->Lookup($idx);
	return -1
	    if (!defined($sliver));
	push(@result, $sliver);
    }
    @$pref = @result;
    return 0;
}

455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
#
# Find slivers "dependent" on this sliver, as for interfaces on nodes.
#
sub DependentSlivers($$)
{
    my ($self, $pref) = @_;

    return -1
	if (! (ref($self) && ref($pref)));
    @$pref = ();

    my $idx = $self->idx();
    my $resource_uuid = $self->resource_uuid();

    my $query_result =
	DBQueryWarn("select idx from geni_slivers ".
		    "where idx!='$idx' and resource_uuid='$resource_uuid'");
    return -1
	if (!$query_result);

    my @result = ();
    while (my ($idx) = $query_result->fetchrow_array()) {
	my $sliver = GeniSliver->Lookup($idx);
	if (!defined($sliver)) {
	    print STDERR "Could not get sliver object for $idx\n";
	    return -1;
	}
	push(@result, $sliver);
    }
    @$pref = @result;
    return 0;
}

488
#
Leigh B. Stoller's avatar
Leigh B. Stoller committed
489
490
# Create a signed credential for this sliver, issued to the provided user.
# The credential will grant all permissions for now.
491
#
Leigh B. Stoller's avatar
Leigh B. Stoller committed
492
493
494
# Should we store these credentials in the DB, recording what we hand out?
#
sub NewCredential($$)
495
{
Leigh B. Stoller's avatar
Leigh B. Stoller committed
496
    my ($self, $owner) = @_;
497

Leigh B. Stoller's avatar
Leigh B. Stoller committed
498
499
    return undef
	if (! (ref($self) && ref($owner)));
500

Leigh B. Stoller's avatar
Leigh B. Stoller committed
501
502
503
504
505
    my $credential = GeniCredential->Create($self, $owner);
    if (!defined($credential)) {
	print STDERR "Could not create credential for $self, $owner\n";
	return undef;
    }
506
507
508
    if (defined($self->nickname())) {
	$credential->AddExtension("nickname", $self->nickname());
    }
Leigh B. Stoller's avatar
Leigh B. Stoller committed
509
510
511
512
513
514
    if ($credential->Sign($self->GetCertificate()) != 0) {
	print STDERR "Could not sign credential for $self, $owner\n";
	return undef;
    }
    return $credential;
}
515

516
517
518
519
520
521
522
523
524
525
526
527
528
############################################################################
#
# The server side methods are in packages which inherit from above.
#
package GeniSliver::Node;
use vars qw(@ISA);
@ISA = "GeniSliver";

use GeniDB;
use GeniComponent;
use GeniSlice;
use GeniCredential;
use GeniCertificate;
529
use GeniUtil;
Leigh B. Stoller's avatar
Leigh B. Stoller committed
530
use Experiment;
531
use emutil;
Leigh B. Stoller's avatar
Leigh B. Stoller committed
532
use XML::Simple;
533
534
use libdb qw(TBDB_ALLOCSTATE_RES_INIT_DIRTY TBDB_NODESTATE_SHUTDOWN
	     TBResolveNextOSID);
535

536
sub Create($$$$$$)
537
{
538
    my ($class, $slice, $user, $resource_uuid, $sliver_uuid, $rspec) = @_;
Leigh B. Stoller's avatar
Leigh B. Stoller committed
539
540
541
542
543
    my $virtualization_type = $rspec->{'virtualization_type'};

    my $experiment = $slice->GetExperiment();
    if (!defined($experiment)) {
	print STDERR "Could not map $slice to its experiment\n";
544
	return undef;
Leigh B. Stoller's avatar
Leigh B. Stoller committed
545
    }
546

Leigh B. Stoller's avatar
Leigh B. Stoller committed
547
548
549
550
551
552
553
554
    #
    # the node is already allocated to the sliver, but still need to enter
    # a virt_nodes entry, and possibly more virt table entries, so that the
    # node will boot properly, and is otherwise controllable.
    #
    my $node = Node->Lookup($resource_uuid);
    if (!defined($node)) {
	print STDERR "Could not map node $resource_uuid to its object\n";
555
	return undef;
Leigh B. Stoller's avatar
Leigh B. Stoller committed
556
    }
557
558
559
560
561
562
563

    #
    # An artifact of Emulab is that for shared/remote nodes, the physical
    # node is already allocated, but not to the current experiment. An
    #
    if (! ($node->sharing_mode() ||
	   ($node->isremotenode() && $node->isvirtnode()))) {
564
565
566
567
568
569
	my $reservation = $node->Reservation();
	if (!defined($reservation)) {
	    print STDERR "$node was already released from $slice\n";
	    return undef;
	}
	if (! $reservation->SameExperiment($experiment)) {
570
	    print STDERR "$node is reserved to another, not $experiment\n";
571
572
573
	    # Signal error so we can look at what happened.
	    return undef;
	}
Leigh B. Stoller's avatar
Leigh B. Stoller committed
574
    }
575
    my $hrn = "${PGENIDOMAIN}." . $node->node_id();
576
    my $nickname = $rspec->{'virtual_id'};
Leigh B. Stoller's avatar
Leigh B. Stoller committed
577
    $sliver_uuid = $node->uuid();
Leigh B. Stoller's avatar
Leigh B. Stoller committed
578
579
580
581
582
583
    
    #
    # The resource UUID refers to the physical node, but the virtualization
    # type might require a vnode.
    # 
    if ($virtualization_type eq "emulab-vnode") {
Leigh B. Stoller's avatar
Leigh B. Stoller committed
584
585
586
	my $vnode = $experiment->VnameToNode($nickname);
	if (!defined($vnode)) {
	    print STDERR "Could not lookup node $nickname in $experiment\n";
587
	    return undef;
Leigh B. Stoller's avatar
Leigh B. Stoller committed
588
	}
Leigh B. Stoller's avatar
Leigh B. Stoller committed
589
590
	$hrn = "${PGENIDOMAIN}." . $vnode->node_id();
	$sliver_uuid = $vnode->uuid();
Leigh B. Stoller's avatar
Leigh B. Stoller committed
591
    }
592
    return GeniSliver->Create($slice, $user, $sliver_uuid, $resource_uuid,
593
			      "Node", $hrn, $nickname, $rspec);
594
595
}

Leigh B. Stoller's avatar
Leigh B. Stoller committed
596
597
#
# Provision a slice. We actually did this when the ticket was requested.
Leigh B. Stoller's avatar
Leigh B. Stoller committed
598
# We fill in some virt table stuff so that tbswap will work.
Leigh B. Stoller's avatar
Leigh B. Stoller committed
599
#
Leigh B. Stoller's avatar
Leigh B. Stoller committed
600
sub Provision($;$)
Leigh B. Stoller's avatar
Leigh B. Stoller committed
601
{
Leigh B. Stoller's avatar
Leigh B. Stoller committed
602
    my ($self, $extraargs) = @_;
Leigh B. Stoller's avatar
Leigh B. Stoller committed
603
604
605
606
607
608
609

    return -1
	if (! ref($self));

    #
    # the node is already allocated to the sliver, but still need to enter
    # a virt_nodes entry, and possibly more virt table entries, so that the
Leigh B. Stoller's avatar
Leigh B. Stoller committed
610
    # node will boot properly, and is otherwise controllable.
Leigh B. Stoller's avatar
Leigh B. Stoller committed
611
612
613
614
615
616
    #
    my $experiment = Experiment->Lookup($self->slice_uuid());
    if (!defined($experiment)) {
	print STDERR "Could not map $self to its experiment\n";
	return -1;
    }
617
    my $uuid = $self->uuid();
Leigh B. Stoller's avatar
Leigh B. Stoller committed
618
    return 0
619
620
	if (!defined($uuid));
    my $node       = Node->Lookup($uuid);
Leigh B. Stoller's avatar
Leigh B. Stoller committed
621
    if (!defined($node)) {
622
	print STDERR "Could not map node $uuid to its object\n";
Leigh B. Stoller's avatar
Leigh B. Stoller committed
623
624
	return -1;
    }
625
    my $node_id     = $node->node_id();
Leigh B. Stoller's avatar
Leigh B. Stoller committed
626
627
    my $reservation = $node->Reservation();
    if (!defined($reservation)) {
628
	print STDERR "$node was already released from slice\n";
Leigh B. Stoller's avatar
Leigh B. Stoller committed
629
630
	return -1;
    }
631
    if (! $reservation->SameExperiment($experiment)) {
632
	print STDERR "$node is reserved to another, not $experiment\n";
633
634
635
636
637
	# Signal error so we can look at what happened.
	return -1;
    }
    my $pid = $experiment->pid();
    my $eid = $experiment->eid();
638

639
640
641
642
643
644
645
    if (!$node->isremotenode() &&
	exists($self->rspec()->{'virtualization_type'}) &&
	$self->rspec()->{'virtualization_type'} eq "emulab-vnode") {
	my $pnode = Node->Lookup($node->phys_nodeid());
	if (!defined($pnode)) {
	    print STDERR "Could not get pnode object for $node\n";
	    return -1;
Leigh B. Stoller's avatar
Leigh B. Stoller committed
646
	}
647

648
649
	# Mark
	$node->ModifyReservation({"genisliver_idx" => $self->idx()});
650

Leigh B. Stoller's avatar
Leigh B. Stoller committed
651
652
653
654
655
656
	if (exists($self->rspec()->{'virtualization_subtype'})) {
	    my $subtype = $self->rspec()->{'virtualization_subtype'};
	    
	    if (!$pnode->sharing_mode()) {
		$pnode->ModifyReservation({"genisliver_idx" => $self->idx()});
	    }
657
	}
Leigh B. Stoller's avatar
Leigh B. Stoller committed
658
659
    }
    else {
660
661
662
663
664
665
666
	#
	# For a "raw" node, there is no vnode, so this is the pnode
	# we need to mark for tmcd.
	#
	if ($node->ModifyReservation({"genisliver_idx" => $self->idx()})) {
	    return -1;
	}
Leigh B. Stoller's avatar
Leigh B. Stoller committed
667
668
669
670
671
672
673
674

	#
	# Terrible place to do this. 
	#
	if ($node->OSSelect("<DEFAULT>", "def_boot_osid", 0) != 0) {
	    print STDERR "Could not os_select $node to default\n";
	    return -1;
	}
675
    }
676
    $self->SetStatus("ready");
Leigh B. Stoller's avatar
Leigh B. Stoller committed
677
678
679
680
681
682
    return 0;
}

#
# Unprovision a sliver. 
#
683
sub UnProvision($;$)
Leigh B. Stoller's avatar
Leigh B. Stoller committed
684
{
685
    my ($self, $nophysfree) = @_;
Leigh B. Stoller's avatar
Leigh B. Stoller committed
686
687
688

    return -1
	if (! ref($self));
689
690
    $nophysfree = 0
	if (!defined($nophysfree));
Leigh B. Stoller's avatar
Leigh B. Stoller committed
691
692

    my $experiment = Experiment->Lookup($self->slice_uuid());
Leigh B. Stoller's avatar
Leigh B. Stoller committed
693
694
695
696
    if (!defined($experiment)) {
	print STDERR "Could not map $self to its experiment\n";
	return -1;
    }
697
    my $uuid = $self->uuid();
Leigh B. Stoller's avatar
Leigh B. Stoller committed
698
    return 0
699
700
	if (!defined($uuid));
    my $node       = Node->Lookup($uuid);
Leigh B. Stoller's avatar
Leigh B. Stoller committed
701
    if (!defined($node)) {
Leigh B. Stoller's avatar
Leigh B. Stoller committed
702
	# Lets call this nonfatal since it might be a virtnode that 
703
	print STDERR "Could not map node $uuid to its object\n";
Leigh B. Stoller's avatar
Leigh B. Stoller committed
704
	return 0;
Leigh B. Stoller's avatar
Leigh B. Stoller committed
705
706
707
708
709
710
711
    }
    my $reservation = $node->Reservation();
    if (!defined($reservation)) {
	print STDERR "$node was already released from $self\n";
	return 0;
    }
    if ($reservation->SameExperiment($experiment)) {
Leigh B. Stoller's avatar
Leigh B. Stoller committed
712
	my $node_id = $node->node_id();
Leigh B. Stoller's avatar
Leigh B. Stoller committed
713
714
	my $pid = $experiment->pid();
	my $eid = $experiment->eid();
Leigh B. Stoller's avatar
Leigh B. Stoller committed
715

716
	if ($node->isremotenode() && $node->isvirtnode()) {
Leigh B. Stoller's avatar
Leigh B. Stoller committed
717
	    system("$VNODESETUP -p -q -m -k $pid $eid $node_id");
718
	    if ($?) {
Leigh B. Stoller's avatar
Leigh B. Stoller committed
719
		print STDERR "$VNODESETUP -k failed on $node_id\n";
720
721
722
		return -1;
	    }
	}
Leigh B. Stoller's avatar
Leigh B. Stoller committed
723
724
725
726
727
728
	elsif ($node->sharing_mode()) {
	    system("$VNODESETUP -j -q -m -k $pid $eid $node_id");
	    if ($?) {
		print STDERR "$VNODESETUP -k failed on $node_id\n";
		return -1;
	    }
729
	}
Leigh B. Stoller's avatar
Leigh B. Stoller committed
730

731
732
	if (!$node->isremotenode() &&
	    $self->rspec()->{'virtualization_type'} eq "emulab-vnode") {
Leigh B. Stoller's avatar
Leigh B. Stoller committed
733
734
735
736
737
738
	    my $pnode_id = $node->phys_nodeid();
	    my $pnode = Node->Lookup($pnode_id);
	    if (!defined($pnode)) {
		print STDERR "Could not get pnode object for $pnode_id\n";
		return -1;
	    }
739
740
741

	    #
	    # If this is the last virtnode on the physnode, release the
Leigh B. Stoller's avatar
Leigh B. Stoller committed
742
743
	    # physnode too. Unless its a shared host, in which case just
	    # deallocate the virtnode.
744
	    #
Leigh B. Stoller's avatar
Leigh B. Stoller committed
745
746
747
748
749
750
	    my @vnodes = ();
	    
	    if ($pnode->sharing_mode()) {
		$nophysfree = 1;
	    }
	    elsif ($pnode->VirtualNodes(\@vnodes) != 0) {
751
		print STDERR "Could not get vnode list for $pnode\n";
Leigh B. Stoller's avatar
Leigh B. Stoller committed
752
753
		return -1;
	    }
754

755
	    if (scalar(@vnodes) > 1 || $nophysfree) {
756
		system("$NFREE -q $pid $eid $node_id");
757
758
759
		# Unincorporate the node. Bogus, needs fixing.
		$pnode->ModifyReservation({"genisliver_idx" => 0})
		    if (! ($pnode->sharing_mode() || scalar(@vnodes) > 1));
760
761
762
763
764
	    }
	    else {
		system("$NFREE -x -q $pid $eid $pnode_id");
		$pnode->Refresh();
	    }
Leigh B. Stoller's avatar
Leigh B. Stoller committed
765
766
	}
	else {
767
768
769
770
771
	    if ($nophysfree) {
		# Unincorporate the node. Bogus, needs fixing.
		$node->ModifyReservation({"genisliver_idx" => 0});
		goto skip;
	    }
772
	    system("$NFREE -q $pid $eid $node_id");
Leigh B. Stoller's avatar
Leigh B. Stoller committed
773
	}
Leigh B. Stoller's avatar
Leigh B. Stoller committed
774
775
	if ($?) {
	    print STDERR "Could not deallocate $node from $self\n";
Leigh B. Stoller's avatar
Leigh B. Stoller committed
776
	    $node->Refresh();
Leigh B. Stoller's avatar
Leigh B. Stoller committed
777
778
	    return -1;
	}
779
      skip:
Leigh B. Stoller's avatar
Leigh B. Stoller committed
780
	$node->Refresh();
Leigh B. Stoller's avatar
Leigh B. Stoller committed
781
782
783
784
785
786
787
788
789
    }
    else {
	print STDERR "$node is reserved to another, not $self\n";
	# Signal error so we can look at what happened.
	return -1;
    }
    return 0;
}

790
791
792
793
794
795
796
797
798
799
800
801
802
803
804
805
806
807
808
809
810
811
812
813
814
815
816
817
818
819
820
821
822
823
824
825
826
827
828
829
830
831
832
833
834
835
836
837
838
839
840
841
842
843
844
845
846
847
848
849
850
851
#
# Process a manifest.
#
sub ProcessManifest($$)
{
    my ($self, $manifest) = @_;

    return -1
	if (! ref($self));

    my $experiment = Experiment->Lookup($self->slice_uuid());
    if (!defined($experiment)) {
	print STDERR "Could not map $self to its experiment\n";
	return -1;
    }
    my $pid = $experiment->pid();
    my $eid = $experiment->eid();
    
    my $uuid = $self->uuid();
    return 0
	if (!defined($uuid));
    my $node       = Node->Lookup($uuid);
    if (!defined($node)) {
	print STDERR "Could not map node $uuid to its object\n";
	return -1;
    }
    my $reservation = $node->Reservation();
    if (!defined($reservation)) {
	print STDERR "$node was already released from $self\n";
	return -1;
    }
    if ($reservation->SameExperiment($experiment)) {
	#
	# Find the corresponding node in the manifest.
	#
	foreach my $ref (@{$manifest->{'node'}}) {
	    if ($ref->{'sliver_uuid'} eq $uuid) {
		# startup command.
		if (exists($ref->{'startup_command'})) {
		    my $startupcmd = $ref->{'startup_command'};
	    
		    if (! TBcheck_dbslot($startupcmd, "virt_nodes",
					 "startupcmd",
					 TBDB_CHECKDBSLOT_WARN|
					 TBDB_CHECKDBSLOT_ERROR)) {
			print STDERR "Invalid startup command '$startupcmd'\n";
			return -1;
		    }
		    $node->Update({"startupcmd" => $startupcmd});
		}
		return 0;
	    }
	}
    }
    else {
	print STDERR "$node is reserved to another, not $self\n";
	# Signal error so we can look at what happened.
	return -1;
    }
    return 0;
}

Leigh B. Stoller's avatar
Leigh B. Stoller committed
852
#
Leigh B. Stoller's avatar
Leigh B. Stoller committed
853
# Start a sliver, which means what?
Leigh B. Stoller's avatar
Leigh B. Stoller committed
854
#
855
sub Start($)
Leigh B. Stoller's avatar
Leigh B. Stoller committed
856
857
858
859
860
861
862
863
864
865
866
{
    my ($self) = @_;

    return -1
	if (! ref($self));

    my $experiment = Experiment->Lookup($self->slice_uuid());
    if (!defined($experiment)) {
	print STDERR "Could not map $self to its experiment\n";
	return -1;
    }
Leigh B. Stoller's avatar
Leigh B. Stoller committed
867
868
869
    my $pid = $experiment->pid();
    my $eid = $experiment->eid();
    
870
    my $uuid = $self->uuid();
Leigh B. Stoller's avatar
Leigh B. Stoller committed
871
    return 0
872
873
	if (!defined($uuid));
    my $node       = Node->Lookup($uuid);
Leigh B. Stoller's avatar
Leigh B. Stoller committed
874
    if (!defined($node)) {
875
	print STDERR "Could not map node $uuid to its object\n";
Leigh B. Stoller's avatar
Leigh B. Stoller committed
876
877
878
879
880
881
882
883
	return -1;
    }
    my $reservation = $node->Reservation();
    if (!defined($reservation)) {
	print STDERR "$node was already released from $self\n";
	return -1;
    }
    if ($reservation->SameExperiment($experiment)) {
Leigh B. Stoller's avatar
Leigh B. Stoller committed
884
	my $node_id = $node->node_id();
Leigh B. Stoller's avatar
Leigh B. Stoller committed
885

Leigh B. Stoller's avatar
Leigh B. Stoller committed
886
887
888
	#
	# Reboot and wait?
	#
Leigh B. Stoller's avatar
Leigh B. Stoller committed
889
890
891
892
893
894
895
896
	if ($node->isvirtnode() && $node->sharing_mode()) {
	    if ($node->eventstate() eq TBDB_NODESTATE_SHUTDOWN()) {
		system("$VNODESETUP -j -q -m $pid $eid $node_id");
	    }
	}
	else {
	    system("$NODEREBOOT -s $node_id");
	}
Leigh B. Stoller's avatar
Leigh B. Stoller committed
897
898
	return -1
	    if ($?);
Leigh B. Stoller's avatar
Leigh B. Stoller committed
899
900
901
902
903
    }
    else {
	print STDERR "$node is reserved to another, not $self\n";
	# Signal error so we can look at what happened.
	return -1;
Leigh B. Stoller's avatar
Leigh B. Stoller committed
904
    }
Leigh B. Stoller's avatar
Leigh B. Stoller committed
905
  done:
Leigh B. Stoller's avatar
Leigh B. Stoller committed
906
907
    return 0;
}
Leigh B. Stoller's avatar
Leigh B. Stoller committed
908

909
##########################################################################
910
#
911
912
913
914
915
916
917
918
919
package GeniSliver::Interface;
use vars qw(@ISA);
@ISA = "GeniSliver";

use GeniDB;
use GeniComponent;
use GeniSlice;
use GeniCredential;
use GeniCertificate;
920
use GeniUtil;
921
922
923

sub Create()
{
924
    my ($class, $slice, $user,
925
	$interface_uuid, $node, $linkname, $rspec) = @_;
926

927
928
929
    my $nickname = $node->node_id() . ".$linkname." .
	$rspec->{'virtual_interface_id'};
    my $hrn = "${PGENIDOMAIN}.$nickname";
930

931
932
933
934
    # Ignore the interface uuid and create a new one.
    # Temporary fix cause of shared interfaces on shared nodes.
    $interface_uuid = GeniUtil::NewUUID();

935
    return GeniSliver->Create($slice, $user, $interface_uuid,
936
937
			      $node->uuid(), "Interface",
			      $hrn, $nickname, $rspec);
938
939
940
941
942
943
}

sub Provision($)
{
    my ($self) = @_;

944
945
946
947
    #
    # This is actually implemented in GeniAggregate since currently "link"
    # is the smallest entity; you cannot operate on an individual interface.
    #
948
949
950
951
952
953
954
    return -1
	if (! ref($self));

    return 0;
}

#
955
# Unprovision a single interface from a link/lan. 
956
957
958
959
960
961
962
963
#
sub UnProvision($)
{
    my ($self) = @_;

    return -1
	if (! ref($self));

964
965
966
967
968
969
970
971
972
973
974
975
976
977
978
979
980
981
982
    my $aggregate = $self->GetAggregate();
    if (!defined($aggregate)) {
	print STDERR "Could not find aggregate for $self\n";
	return -1;
    }

    #
    # This is terrible; we need an interface to remove ports form vlans,
    # so that we do not have to tear down the entire vlan and recreate.
    #
    if ($aggregate->UnProvision() != 0) {
	print STDERR "Could not unprovision $aggregate\n";
	return -1;
    }
    $self->ClearAggregate();
    if ($aggregate->Provision() != 0) {
	print STDERR "Could not provision $aggregate\n";
	return -1;
    }
983
984
985
986
    return 0;
}

#
987
# Start a sliver.
988
#
989
sub Start($)
990
991
992
{
    my ($self) = @_;

993
994
995
996
    #
    # This is actually implemented in GeniAggregate since currently "link"
    # is the smallest entity; you cannot operate on an individual interface.
    #
997
998
999
1000
1001
    return -1
	if (! ref($self));

    return 0;
}
Leigh B. Stoller's avatar
Leigh B. Stoller committed
1002

Leigh B. Stoller's avatar
Leigh B. Stoller committed
1003
1004
# _Always_ make sure that this 1 is at the end of the file...
1;