GeniSliver.pm.in 23 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;
Leigh B. Stoller's avatar
Leigh B. Stoller committed
24
# Hate to import all this crap; need a utility library.
25
use emutil qw(TBGetUniqueIndex);
Leigh B. Stoller's avatar
Leigh B. Stoller committed
26
use Experiment;
27
use OSinfo;
Leigh B. Stoller's avatar
Leigh B. Stoller committed
28
use English;
Leigh B. Stoller's avatar
Leigh B. Stoller committed
29
use XML::Simple;
Leigh B. Stoller's avatar
Leigh B. Stoller committed
30
31
use Data::Dumper;
use File::Temp qw(tempfile);
Leigh B. Stoller's avatar
Leigh B. Stoller committed
32
use overload ('""' => 'Stringify');
Leigh B. Stoller's avatar
Leigh B. Stoller committed
33
34
35
36
37
38
39
40

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

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

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

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

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

    # 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
88
89
90
    return undef
	if (!$query_result || !$query_result->numrows);

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

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

    #
    # 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;
113
114
115
116

    # Bless into sub package if called for.
    my $resource_type = $self->{'SLIVER'}->{'resource_type'};
    if (defined($resource_type) && $resource_type ne "") {
117
	bless($self, $class . "::" . $resource_type);
118
119
120
121
    }
    else {
	bless($self, $class);
    }
Leigh B. Stoller's avatar
Leigh B. Stoller committed
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
    
    # 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
143
# Create a sliver record in the DB. On the client side we save the credential
Leigh B. Stoller's avatar
Leigh B. Stoller committed
144
# that allows control of it, for later operations.
Leigh B. Stoller's avatar
Leigh B. Stoller committed
145
#
146
sub Create($$$$$$$$$)
Leigh B. Stoller's avatar
Leigh B. Stoller committed
147
{
148
    my ($class, $slice, $owner, $uuid, $resource_uuid, $resource_type,
149
	$hrn, $nickname, $rspec) = @_;
Leigh B. Stoller's avatar
Leigh B. Stoller committed
150
    my @insert_data = ();
151
    my $certificate;
Leigh B. Stoller's avatar
Leigh B. Stoller committed
152
153
154

    # Every sliver gets a new unique index.
    my $idx = TBGetUniqueIndex('next_sliver', 1);
155
156
157
158
159
160
161
162
    
    # 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";
163
164
	    return undef;
	}
Leigh B. Stoller's avatar
Leigh B. Stoller committed
165
    }
166
167
    $certificate = GeniCertificate->Create("sliver", $hrn, $TBOPS, $uuid)
	if (!defined($certificate));
168
	
169
170
171
172
    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
173
    }
Leigh B. Stoller's avatar
Leigh B. Stoller committed
174
    my $slice_uuid     = $slice->uuid();
175
    my $owner_uuid     = $owner->uuid();
Leigh B. Stoller's avatar
Leigh B. Stoller committed
176
177
178
179

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

Leigh B. Stoller's avatar
Leigh B. Stoller committed
189
190
191
192
193
194
195
    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
196
    # Insert into DB.
197
198
199
    if (!DBQueryWarn("insert into geni_slivers set " .
		     join(",", @insert_data))) {
	$certificate->Delete();
Leigh B. Stoller's avatar
Leigh B. Stoller committed
200
201
202
	return undef;
    }

Leigh B. Stoller's avatar
Leigh B. Stoller committed
203
204
205
    my $sliver = GeniSliver->Lookup($idx);
    return undef
	if (!defined($sliver));
206
207
208
209
210

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

    return $sliver;
Leigh B. Stoller's avatar
Leigh B. Stoller committed
215
216
217
218
219
}
# accessors
sub field($$) { return ((! ref($_[0])) ? -1 : $_[0]->{'SLIVER'}->{$_[1]}); }
sub idx($)		{ return field($_[0], "idx"); }
sub uuid($)		{ return field($_[0], "uuid"); }
220
221
sub hrn($)		{ return field($_[0], "hrn"); }
sub nickname($)		{ return field($_[0], "nickname"); }
Leigh B. Stoller's avatar
Leigh B. Stoller committed
222
223
224
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
225
sub credential_idx($)	{ return field($_[0], "credential_idx"); }
Leigh B. Stoller's avatar
Leigh B. Stoller committed
226
sub resource_uuid($)	{ return field($_[0], "resource_uuid"); }
227
sub resource_type($)	{ return field($_[0], "resource_type"); }
Leigh B. Stoller's avatar
Leigh B. Stoller committed
228
229
230
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
231
sub status($)		{ return field($_[0], "status"); }
232
233
sub cert($)		{ return $_[0]->{'CERTIFICATE'}->cert(); }
sub GetCertificate($)   { return $_[0]->{'CERTIFICATE'}; }
Leigh B. Stoller's avatar
Leigh B. Stoller committed
234
sub rspec($)            { return $_[0]->{'RSPEC'}; }
Leigh B. Stoller's avatar
Leigh B. Stoller committed
235
236
237
238

#
# Delete the sliver. The sliver should not be provisioned when this done.
#
239
sub Delete($$)
Leigh B. Stoller's avatar
Leigh B. Stoller committed
240
{
241
    my ($self, $purge) = @_;
Leigh B. Stoller's avatar
Leigh B. Stoller committed
242
243
244
245

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

246
247
    my $idx  = $self->idx();
    my $uuid = $self->uuid();
Leigh B. Stoller's avatar
Leigh B. Stoller committed
248

249
250
251
252
    if (GeniUsage->DestroySliver($self, $purge)) {
	print STDERR
	    "GeniSliver::Delete: GeniUsage->DestroySliver($self) failed\n";
    }
253
254
    DBQueryWarn("delete from geni_credentials where this_uuid='$uuid'")
	or return -1;
255
256
257
258
    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
259
260
    DBQueryWarn("delete from geni_slivers where idx='$idx'")
	or return -1;
261

262
263
264
    # Delete from cache. 
    delete($slivers{$idx});
    
Leigh B. Stoller's avatar
Leigh B. Stoller committed
265
266
267
    return 0;
}

Leigh B. Stoller's avatar
Leigh B. Stoller committed
268
269
270
271
272
273
274
275
276
277
#
# 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
278
279
    my $idx      = $self->idx();
    my $agg_uuid = $aggregate->uuid();
Leigh B. Stoller's avatar
Leigh B. Stoller committed
280
281
282

    return -1
	if (!DBQueryWarn("update geni_slivers set ".
Leigh B. Stoller's avatar
Leigh B. Stoller committed
283
			 "  aggregate_uuid='$agg_uuid' ".
Leigh B. Stoller's avatar
Leigh B. Stoller committed
284
285
			 "where idx='$idx'"));
    
286
287
288
289
290
291
292
    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
293
    $self->{'SLIVER'}->{'aggregate_uuid'} = $agg_uuid;
Leigh B. Stoller's avatar
Leigh B. Stoller committed
294
295
296
297
    $self->{'AGGREGATE'} = $aggregate;
    return 0;
}

298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
#
# 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
320
321
322
323
324
325
326
327
328
329
330
331
#
# 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
332
	if (!defined($self->aggregate_uuid()));
Leigh B. Stoller's avatar
Leigh B. Stoller committed
333

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

343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
#
# 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;
}

Leigh B. Stoller's avatar
Leigh B. Stoller committed
364
365
366
367
368
369
370
371
372
373
374
375
376
#
# 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
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
#
# 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;
}

402
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
428
#
# 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;
}

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
455
456
457
458
459
460
461
#
# 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;
}

462
#
Leigh B. Stoller's avatar
Leigh B. Stoller committed
463
464
# Create a signed credential for this sliver, issued to the provided user.
# The credential will grant all permissions for now.
465
#
Leigh B. Stoller's avatar
Leigh B. Stoller committed
466
467
468
# Should we store these credentials in the DB, recording what we hand out?
#
sub NewCredential($$)
469
{
Leigh B. Stoller's avatar
Leigh B. Stoller committed
470
    my ($self, $owner) = @_;
471

Leigh B. Stoller's avatar
Leigh B. Stoller committed
472
473
    return undef
	if (! (ref($self) && ref($owner)));
474

Leigh B. Stoller's avatar
Leigh B. Stoller committed
475
476
477
478
479
    my $credential = GeniCredential->Create($self, $owner);
    if (!defined($credential)) {
	print STDERR "Could not create credential for $self, $owner\n";
	return undef;
    }
480
481
482
    if (defined($self->nickname())) {
	$credential->AddExtension("nickname", $self->nickname());
    }
Leigh B. Stoller's avatar
Leigh B. Stoller committed
483
484
485
486
487
488
    if ($credential->Sign($self->GetCertificate()) != 0) {
	print STDERR "Could not sign credential for $self, $owner\n";
	return undef;
    }
    return $credential;
}
489

490
491
492
493
494
495
496
497
498
499
500
501
502
############################################################################
#
# 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;
503
use GeniUtil;
Leigh B. Stoller's avatar
Leigh B. Stoller committed
504
use Experiment;
Leigh B. Stoller's avatar
Leigh B. Stoller committed
505
use XML::Simple;
506
507
use libdb qw(TBDB_ALLOCSTATE_RES_INIT_DIRTY TBDB_NODESTATE_SHUTDOWN
	     TBResolveNextOSID);
508

509
sub Create($$$$$$)
510
{
511
    my ($class, $slice, $user, $resource_uuid, $sliver_uuid, $rspec) = @_;
Leigh B. Stoller's avatar
Leigh B. Stoller committed
512
513
514
515
516
    my $virtualization_type = $rspec->{'virtualization_type'};

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

Leigh B. Stoller's avatar
Leigh B. Stoller committed
520
521
522
523
524
525
526
527
    #
    # 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";
528
	return undef;
Leigh B. Stoller's avatar
Leigh B. Stoller committed
529
    }
530
531
532
533
534
535
536

    #
    # 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()))) {
537
538
539
540
541
542
	my $reservation = $node->Reservation();
	if (!defined($reservation)) {
	    print STDERR "$node was already released from $slice\n";
	    return undef;
	}
	if (! $reservation->SameExperiment($experiment)) {
543
	    print STDERR "$node is reserved to another, not $experiment\n";
544
545
546
	    # Signal error so we can look at what happened.
	    return undef;
	}
Leigh B. Stoller's avatar
Leigh B. Stoller committed
547
    }
548
    my $hrn = "${PGENIDOMAIN}." . $node->node_id();
549
    my $nickname = $rspec->{'virtual_id'};
Leigh B. Stoller's avatar
Leigh B. Stoller committed
550
    $sliver_uuid = $node->uuid();
Leigh B. Stoller's avatar
Leigh B. Stoller committed
551
552
553
554
555
556
    
    #
    # 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
557
558
559
	my $vnode = $experiment->VnameToNode($nickname);
	if (!defined($vnode)) {
	    print STDERR "Could not lookup node $nickname in $experiment\n";
560
	    return undef;
Leigh B. Stoller's avatar
Leigh B. Stoller committed
561
	}
Leigh B. Stoller's avatar
Leigh B. Stoller committed
562
563
	$hrn = "${PGENIDOMAIN}." . $vnode->node_id();
	$sliver_uuid = $vnode->uuid();
Leigh B. Stoller's avatar
Leigh B. Stoller committed
564
    }
565
    return GeniSliver->Create($slice, $user, $sliver_uuid, $resource_uuid,
566
			      "Node", $hrn, $nickname, $rspec);
567
568
}

Leigh B. Stoller's avatar
Leigh B. Stoller committed
569
570
#
# Provision a slice. We actually did this when the ticket was requested.
Leigh B. Stoller's avatar
Leigh B. Stoller committed
571
# We fill in some virt table stuff so that tbswap will work.
Leigh B. Stoller's avatar
Leigh B. Stoller committed
572
#
Leigh B. Stoller's avatar
Leigh B. Stoller committed
573
sub Provision($;$)
Leigh B. Stoller's avatar
Leigh B. Stoller committed
574
{
Leigh B. Stoller's avatar
Leigh B. Stoller committed
575
    my ($self, $extraargs) = @_;
Leigh B. Stoller's avatar
Leigh B. Stoller committed
576
577
578
579
580
581
582

    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
583
    # node will boot properly, and is otherwise controllable.
Leigh B. Stoller's avatar
Leigh B. Stoller committed
584
585
586
587
588
589
    #
    my $experiment = Experiment->Lookup($self->slice_uuid());
    if (!defined($experiment)) {
	print STDERR "Could not map $self to its experiment\n";
	return -1;
    }
590
    my $uuid = $self->uuid();
Leigh B. Stoller's avatar
Leigh B. Stoller committed
591
    return 0
592
593
	if (!defined($uuid));
    my $node       = Node->Lookup($uuid);
Leigh B. Stoller's avatar
Leigh B. Stoller committed
594
    if (!defined($node)) {
595
	print STDERR "Could not map node $uuid to its object\n";
Leigh B. Stoller's avatar
Leigh B. Stoller committed
596
597
	return -1;
    }
598
    my $node_id     = $node->node_id();
Leigh B. Stoller's avatar
Leigh B. Stoller committed
599
600
    my $reservation = $node->Reservation();
    if (!defined($reservation)) {
601
	print STDERR "$node was already released from slice\n";
Leigh B. Stoller's avatar
Leigh B. Stoller committed
602
603
	return -1;
    }
604
    if (! $reservation->SameExperiment($experiment)) {
605
	print STDERR "$node is reserved to another, not $experiment\n";
606
607
608
609
610
	# Signal error so we can look at what happened.
	return -1;
    }
    my $pid = $experiment->pid();
    my $eid = $experiment->eid();
611

612
    my $redirected = 0;
613
614
615
616
617
    if (exists($self->rspec()->{'tmcd_server'}) &&
	exists($self->rspec()->{'tmcd_nodeid'})) {
	my $tmcd_redirect =
	    $self->rspec()->{'tmcd_server'} . ":" .
	    $self->rspec()->{'tmcd_nodeid'};
Leigh B. Stoller's avatar
Leigh B. Stoller committed
618

619
620
	if ($node->ModifyReservation({"tmcd_redirect" => $tmcd_redirect})){
	    return -1;
Leigh B. Stoller's avatar
Leigh B. Stoller committed
621
	}
622
	$redirected = 1;
623
    }
Leigh B. Stoller's avatar
Leigh B. Stoller committed
624

625
626
627
628
629
630
631
632
633
634
635
636
637
    #
    # An emulab cluster node uses a vnode on the physnode, even for a
    # dedicated physical nodes. We need to tell tmcd about the pnode,
    # but not the vnode since it is going to redirect to tmcd on the
    # remote Emulab controlling the experiment.
    #
    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
638
	}
639

640
641
642
643
	# Not redirected. Use local tmcd anyway.
	$node->ModifyReservation({"genisliver_idx" => $self->idx()})
	    if (!$redirected);

Leigh B. Stoller's avatar
Leigh B. Stoller committed
644
645
646
647
648
649
	if (exists($self->rspec()->{'virtualization_subtype'})) {
	    my $subtype = $self->rspec()->{'virtualization_subtype'};
	    
	    if (!$pnode->sharing_mode()) {
		$pnode->ModifyReservation({"genisliver_idx" => $self->idx()});
	    }
650
	}
Leigh B. Stoller's avatar
Leigh B. Stoller committed
651
652
    }
    else {
653
654
655
656
657
658
659
	#
	# 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
660
661
662
663
664
665
666
667

	#
	# 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;
	}
668
    }
669
    $self->SetStatus("ready");
Leigh B. Stoller's avatar
Leigh B. Stoller committed
670
671
672
673
674
675
    return 0;
}

#
# Unprovision a sliver. 
#
676
sub UnProvision($;$)
Leigh B. Stoller's avatar
Leigh B. Stoller committed
677
{
678
    my ($self, $nophysfree) = @_;
Leigh B. Stoller's avatar
Leigh B. Stoller committed
679
680
681

    return -1
	if (! ref($self));
682
683
    $nophysfree = 0
	if (!defined($nophysfree));
Leigh B. Stoller's avatar
Leigh B. Stoller committed
684
685

    my $experiment = Experiment->Lookup($self->slice_uuid());
Leigh B. Stoller's avatar
Leigh B. Stoller committed
686
687
688
689
    if (!defined($experiment)) {
	print STDERR "Could not map $self to its experiment\n";
	return -1;
    }
690
    my $uuid = $self->uuid();
Leigh B. Stoller's avatar
Leigh B. Stoller committed
691
    return 0
692
693
	if (!defined($uuid));
    my $node       = Node->Lookup($uuid);
Leigh B. Stoller's avatar
Leigh B. Stoller committed
694
    if (!defined($node)) {
Leigh B. Stoller's avatar
Leigh B. Stoller committed
695
	# Lets call this nonfatal since it might be a virtnode that 
696
	print STDERR "Could not map node $uuid to its object\n";
Leigh B. Stoller's avatar
Leigh B. Stoller committed
697
	return 0;
Leigh B. Stoller's avatar
Leigh B. Stoller committed
698
699
700
701
702
703
704
    }
    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
705
	my $node_id = $node->node_id();
Leigh B. Stoller's avatar
Leigh B. Stoller committed
706
707
	my $pid = $experiment->pid();
	my $eid = $experiment->eid();
Leigh B. Stoller's avatar
Leigh B. Stoller committed
708

709
	if ($node->isremotenode() && $node->isvirtnode()) {
Leigh B. Stoller's avatar
Leigh B. Stoller committed
710
	    system("$VNODESETUP -p -q -m -k $pid $eid $node_id");
711
	    if ($?) {
Leigh B. Stoller's avatar
Leigh B. Stoller committed
712
		print STDERR "$VNODESETUP -k failed on $node_id\n";
713
714
715
		return -1;
	    }
	}
Leigh B. Stoller's avatar
Leigh B. Stoller committed
716
717
718
719
720
721
	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;
	    }
722
	}
Leigh B. Stoller's avatar
Leigh B. Stoller committed
723

724
725
	if (!$node->isremotenode() &&
	    $self->rspec()->{'virtualization_type'} eq "emulab-vnode") {
Leigh B. Stoller's avatar
Leigh B. Stoller committed
726
727
728
729
730
731
	    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;
	    }
732
733
734

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

748
	    if (scalar(@vnodes) > 1 || $nophysfree) {
749
750
751
752
753
754
		system("$NFREE -q $pid $eid $node_id");
	    }
	    else {
		system("$NFREE -x -q $pid $eid $pnode_id");
		$pnode->Refresh();
	    }
Leigh B. Stoller's avatar
Leigh B. Stoller committed
755
756
	}
	else {
757
	    goto skip
Leigh B. Stoller's avatar
Leigh B. Stoller committed
758
		if ($nophysfree);
759
	    system("$NFREE -q $pid $eid $node_id");
Leigh B. Stoller's avatar
Leigh B. Stoller committed
760
	}
Leigh B. Stoller's avatar
Leigh B. Stoller committed
761
762
	if ($?) {
	    print STDERR "Could not deallocate $node from $self\n";
Leigh B. Stoller's avatar
Leigh B. Stoller committed
763
	    $node->Refresh();
Leigh B. Stoller's avatar
Leigh B. Stoller committed
764
765
	    return -1;
	}
766
      skip:
Leigh B. Stoller's avatar
Leigh B. Stoller committed
767
	$node->Refresh();
Leigh B. Stoller's avatar
Leigh B. Stoller committed
768
769
770
771
772
773
774
775
776
777
    }
    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
778
# Start a sliver, which means what?
Leigh B. Stoller's avatar
Leigh B. Stoller committed
779
#
780
sub Start($)
Leigh B. Stoller's avatar
Leigh B. Stoller committed
781
782
783
784
785
786
787
788
789
790
791
{
    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
792
793
794
    my $pid = $experiment->pid();
    my $eid = $experiment->eid();
    
795
    my $uuid = $self->uuid();
Leigh B. Stoller's avatar
Leigh B. Stoller committed
796
    return 0
797
798
	if (!defined($uuid));
    my $node       = Node->Lookup($uuid);
Leigh B. Stoller's avatar
Leigh B. Stoller committed
799
    if (!defined($node)) {
800
	print STDERR "Could not map node $uuid to its object\n";
Leigh B. Stoller's avatar
Leigh B. Stoller committed
801
802
803
804
805
806
807
808
	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
809
	my $node_id = $node->node_id();
Leigh B. Stoller's avatar
Leigh B. Stoller committed
810

Leigh B. Stoller's avatar
Leigh B. Stoller committed
811
812
813
	#
	# Reboot and wait?
	#
Leigh B. Stoller's avatar
Leigh B. Stoller committed
814
815
816
817
818
819
820
821
	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
822
823
	return -1
	    if ($?);
Leigh B. Stoller's avatar
Leigh B. Stoller committed
824
825
826
827
828
    }
    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
829
    }
Leigh B. Stoller's avatar
Leigh B. Stoller committed
830
  done:
Leigh B. Stoller's avatar
Leigh B. Stoller committed
831
832
    return 0;
}
Leigh B. Stoller's avatar
Leigh B. Stoller committed
833

834
##########################################################################
835
#
836
837
838
839
840
841
842
843
844
package GeniSliver::Interface;
use vars qw(@ISA);
@ISA = "GeniSliver";

use GeniDB;
use GeniComponent;
use GeniSlice;
use GeniCredential;
use GeniCertificate;
845
use GeniUtil;
846
847
848

sub Create()
{
849
    my ($class, $slice, $user,
850
	$interface_uuid, $node, $linkname, $rspec) = @_;
851

852
853
854
    my $nickname = $node->node_id() . ".$linkname." .
	$rspec->{'virtual_interface_id'};
    my $hrn = "${PGENIDOMAIN}.$nickname";
855

856
857
858
859
    # Ignore the interface uuid and create a new one.
    # Temporary fix cause of shared interfaces on shared nodes.
    $interface_uuid = GeniUtil::NewUUID();

860
    return GeniSliver->Create($slice, $user, $interface_uuid,
861
862
			      $node->uuid(), "Interface",
			      $hrn, $nickname, $rspec);
863
864
865
866
867
868
}

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

869
870
871
872
    #
    # This is actually implemented in GeniAggregate since currently "link"
    # is the smallest entity; you cannot operate on an individual interface.
    #
873
874
875
876
877
878
879
    return -1
	if (! ref($self));

    return 0;
}

#
880
# Unprovision a single interface from a link/lan. 
881
882
883
884
885
886
887
888
#
sub UnProvision($)
{
    my ($self) = @_;

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

889
890
891
892
893
894
895
896
897
898
899
900
901
902
903
904
905
906
907
    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;
    }
908
909
910
911
    return 0;
}

#
912
# Start a sliver.
913
#
914
sub Start($)
915
916
917
{
    my ($self) = @_;

918
919
920
921
    #
    # This is actually implemented in GeniAggregate since currently "link"
    # is the smallest entity; you cannot operate on an individual interface.
    #
922
923
924
925
926
    return -1
	if (! ref($self));

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

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