GeniCMV2.pm.in 44.6 KB
Newer Older
1
2
3
#!/usr/bin/perl -wT
#
# GENIPUBLIC-COPYRIGHT
Leigh B. Stoller's avatar
Leigh B. Stoller committed
4
# Copyright (c) 2008-2010 University of Utah and the Flux Group.
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
# All rights reserved.
#
package GeniCMV2;

#
# The server side of the CM interface on remote sites. Also communicates
# with the GMC interface at Geni Central as a client.
#
use strict;
use Exporter;
use vars qw(@ISA @EXPORT);

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

# Must come after package declaration!
use lib '@prefix@/lib';
use GeniDB;
use Genixmlrpc;
use GeniResponse;
use GeniTicket;
use GeniCredential;
use GeniCertificate;
use GeniSlice;
use GeniAggregate;
use GeniAuthority;
use GeniSliver;
use GeniUser;
use GeniRegistry;
use GeniUtil;
35
use GeniCM;
36
use GeniHRN;
37
use GeniXML;
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
use libtestbed qw(SENDMAIL);
use emutil;
# Hate to import all this crap; need a utility library.
use libdb qw(TBGetSiteVar EXPTSTATE_SWAPPED EXPTSTATE_ACTIVE TBOPSPID
	     TBDB_NODESTATE_TBFAILED);
use User;
use Node;
use OSinfo;
use Image;
use Interface;
use English;
use Data::Dumper;
use XML::Simple;
use Date::Parse;
use POSIX qw(strftime tmpnam);
use Time::Local;
use Experiment;
use VirtExperiment;
use Compress::Zlib;
use MIME::Base64;

# Configure variables
my $TB		   = "@prefix@";
my $TBOPS          = "@TBOPSEMAIL@";
my $TBAPPROVAL     = "@TBAPPROVALEMAIL@";
my $TBAUDIT   	   = "@TBAUDITEMAIL@";
my $BOSSNODE       = "@BOSSNODE@";
my $OURDOMAIN      = "@OURDOMAIN@";
my $PGENIDOMAIN    = "@PROTOGENI_DOMAIN@";
my $CREATEEXPT     = "$TB/bin/batchexp";
my $ENDEXPT        = "$TB/bin/endexp";
my $NALLOC	   = "$TB/bin/nalloc";
my $NFREE	   = "$TB/bin/nfree";
my $AVAIL	   = "$TB/sbin/avail";
my $PTOPGEN	   = "$TB/libexec/ptopgen";
my $TBSWAP	   = "$TB/bin/tbswap";
my $SWAPEXP	   = "$TB/bin/swapexp";
my $PLABSLICE	   = "$TB/sbin/plabslicewrapper";
my $NAMEDSETUP     = "$TB/sbin/named_setup";
my $VNODESETUP     = "$TB/sbin/vnode_setup";
my $GENTOPOFILE    = "$TB/libexec/gentopofile";
my $TARFILES_SETUP = "$TB/bin/tarfiles_setup";
my $MAPPER         = "$TB/bin/mapper";
my $VTOPGEN        = "$TB/bin/vtopgen";
my $SNMPIT         = "$TB/bin/snmpit";
my $PRERENDER      = "$TB/libexec/vis/prerender";
my $EMULAB_PEMFILE = "@prefix@/etc/genicm.pem";

my $API_VERSION = 2;

#
# Tell the client what API revision we support.  The correspondence
# between revision numbers and API features is to be specified elsewhere.
# No credentials are required.
#
sub GetVersion()
{
95
    my @input_rspec_versions = ( "0.1", "2" );
Gary Wong's avatar
Gary Wong committed
96
97
98
    my $blob = {
	"api" => $API_VERSION,
	"level" => 1,
99
	"input_rspec" => \@input_rspec_versions,
Gary Wong's avatar
Gary Wong committed
100
101
102
103
	"output_rspec" => "0.1"
    };

    return GeniResponse->Create( GENIRESPONSE_SUCCESS, $blob );
104
105
106
107
108
109
110
111
}

#
# Respond to a Resolve request. 
#
sub Resolve($)
{
    my ($argref) = @_;
112
113
    my $credentials = $argref->{'credentials'};
    my $urn         = $argref->{'urn'};
114
    my $admin       = 0;
115

116
117
118
119
120
121
122
123
124
125
    if (! (defined($credentials) && defined($urn))) {
	return GeniResponse->MalformedArgsResponse("Missing arguments");
    }
    if (! GeniHRN::IsValid($urn)) {
	return GeniResponse->MalformedArgsResponse("Invalid URN");
    }
    my $credential = CheckCredentials($credentials);
    return $credential
	if (GeniResponse::IsResponse($credential));

126
127
128
    my ($object, $type) = LookupURN($urn);
    return $object
	if (GeniResponse::IsResponse($object));
129
130
131
132
133
134

    #
    # This is a convenience for testing. If a local user and that
    # user is an admin person, then do whatever it says. This is
    # easier then trying to do this with credential privs.
    #
135
    my $user = GeniCM::CreateUserFromCertificate($credential->owner_cert());
136
137
138
    if (defined($user) && $user->IsLocal() && $user->admin()) {
	$admin = 1;
    }
139
140
    
    if ($type eq "node") {
141
	my $node  = $object;
142
	my $rspec = GeniCM::GetAdvertisement(0, $node->node_id(), "0.1");
143
144
	if (! defined($rspec)) {
	    return GeniResponse->Create(GENIRESPONSE_ERROR, undef,
145
					"Error getting advertisement");
146
147
148
149
150
	}
	# Return a blob.
	my $blob = { "hrn"          => "${PGENIDOMAIN}." . $node->node_id(),
		     "uuid"         => $node->uuid(),
		     "role"	    => $node->role(),
151
152
		     "hostname"     =>
			 GeniUtil::FindHostname($node->node_id()),
153
154
155
156
157
158
159
160
161
162
		     "physctrl"     => 
			 Interface->LookupControl($node->phys_nodeid())->IP(),
		     "urn"          => GeniHRN::Generate($OURDOMAIN,
							 "node",
							 $node->node_id()),
		     "rspec"        => $rspec
		   };

	return GeniResponse->Create(GENIRESPONSE_SUCCESS, $blob);
    }
Leigh B. Stoller's avatar
Leigh B. Stoller committed
163
    if ($type eq "slice") {
164
165
	my $slice = $object;

Leigh B. Stoller's avatar
Leigh B. Stoller committed
166
167
168
169
	#
	# In this implementation, the caller must hold a valid slice
	# credential for the slice being looked up. 
	#
170
	if (! ($admin || $slice->urn() eq $credential->target_urn())) {
Leigh B. Stoller's avatar
Leigh B. Stoller committed
171
172
173
174
175
176
177
	    return GeniResponse->Create(GENIRESPONSE_FORBIDDEN());
	}
	# Return a blob.
	my $blob = { "urn"          => $urn };

	my $aggregate = GeniAggregate->SliceAggregate($slice);
	if (defined($aggregate)) {
178
	    $blob->{'sliver_urn'} = $aggregate->urn();
Leigh B. Stoller's avatar
Leigh B. Stoller committed
179
180
181
	}
	my $ticket = GeniTicket->SliceTicket($slice);
	if (defined($ticket)) {
182
	    $blob->{'ticket_urn'} = $ticket->urn();
Leigh B. Stoller's avatar
Leigh B. Stoller committed
183
184
185
186
	}
	return GeniResponse->Create(GENIRESPONSE_SUCCESS, $blob);
    }
    if ($type eq "sliver") {
187
188
	my $sliver = $object;

Leigh B. Stoller's avatar
Leigh B. Stoller committed
189
190
191
192
	#
	# In this implementation, the caller must hold a valid slice
	# or sliver credential for the slice being looked up. 
	#
193
	if (! ($admin ||
194
	       $sliver->urn() eq $credential->target_urn() ||
195
	       $sliver->slice_uuid() eq $credential->target_uuid())) {
Leigh B. Stoller's avatar
Leigh B. Stoller committed
196
197
	    return GeniResponse->Create(GENIRESPONSE_FORBIDDEN);
	}
198
199
	my $manifest = $sliver->GetManifest(1);
	if (!defined($manifest)) {
Leigh B. Stoller's avatar
Leigh B. Stoller committed
200
201
202
203
204
205
206
207
208
	    return GeniResponse->Create(GENIRESPONSE_ERROR);
	}
	# Return a blob.
	my $blob = { "urn"          => $urn,
		     "manifest"     => $manifest,
		 };
	return GeniResponse->Create(GENIRESPONSE_SUCCESS, $blob);
    }
    if ($type eq "ticket") {
209
210
	my $ticket = $object;

Leigh B. Stoller's avatar
Leigh B. Stoller committed
211
212
213
214
	#
	# In this implementation, the caller must hold a valid slice
	# or sliver credential to get the ticket.
	#
215
	my $slice = GeniSlice->Lookup($ticket->slice_urn());
Leigh B. Stoller's avatar
Leigh B. Stoller committed
216
217
218
219
	if (!defined($slice)) {
	    print STDERR "Could not find slice for $ticket\n";
	    return GeniResponse->Create(GENIRESPONSE_ERROR);
	}
220
	if (! ($admin || $slice->urn() eq $credential->target_urn())) {
Leigh B. Stoller's avatar
Leigh B. Stoller committed
221
222
223
224
225
	    #
	    # See if its the sliver credential. 
	    #
	    my $aggregate = GeniAggregate->SliceAggregate($slice);
	    if (!defined($aggregate) ||
226
		$aggregate->urn() ne $credential->target_urn()) {
Leigh B. Stoller's avatar
Leigh B. Stoller committed
227
228
229
230
231
		return GeniResponse->Create(GENIRESPONSE_FORBIDDEN());
	    }
	}
	return GeniResponse->Create(GENIRESPONSE_SUCCESS, $ticket->asString());
    }
232
233
    return GeniResponse->Create(GENIRESPONSE_UNSUPPORTED, undef,
				"Cannot resolve $type at this authority");
234
235
236
237
238
239
240
241
}

#
# Discover resources on this component, returning a resource availablity spec
#
sub DiscoverResources($)
{
    my ($argref) = @_;
242
243
244
245
246
247
248
249
250
251
    my $credentials = $argref->{'credentials'};
    my $available   = $argref->{'available'} || 0;
    my $compress    = $argref->{'compress'} || 0;

    if (! (defined($credentials))) {
	return GeniResponse->MalformedArgsResponse("Missing arguments");
    }
    my $credential = CheckCredentials($credentials);
    return $credential
	if (GeniResponse::IsResponse($credential));
252

253
254
    my $credential_objects = [];
    foreach my $credstr (@$credentials) {
255
256
257
        my $cred = CheckCredential($credstr);
        push(@$credential_objects, $cred) 
            if(!GeniResponse::IsResponse($cred));
258
259
260
    }
    return GeniCM::DiscoverResourcesAux($available, $compress,
        $credential_objects);
261
262
263
264
265
266
267
268
}

#
# Create a Sliver.
#
sub CreateSliver($)
{
    my ($argref) = @_;
269
270
271
272
273
274
275
    my $slice_urn    = $argref->{'slice_urn'};
    my $rspecstr     = $argref->{'rspec'};
    my $credentials  = $argref->{'credentials'};
    my $keys         = $argref->{'keys'};
    my $impotent     = $argref->{'impotent'} || 0;
    
    # For now, I am not worrying about the slice_urn argument.
276
277
    if (! (defined($credentials) &&
	   defined($slice_urn) && defined($rspecstr))) {
278
279
	return GeniResponse->MalformedArgsResponse("Missing arguments");
    }
280
281
282
283
284
285
    if (! ($rspecstr =~ /^[\040-\176\012\015\011]+$/)) {
	return GeniResponse->MalformedArgsResponse("Bad characters in rspec");
    }
    if (! GeniHRN::IsValid($slice_urn)) {
	return GeniResponse->MalformedArgsResponse("Bad characters in URN");
    }
286
287
288
    my $credential = CheckCredentials($credentials);
    return $credential
	if (GeniResponse::IsResponse($credential));
289

290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
    #
    # In this implementation, the user must provide a slice credential,
    # so we ignore the slice_urn. For CreateSliver(), the slice must not
    # be instantiated.
    #
    my ($slice,$aggregate) = Credential2SliceAggregate($credential);
    if (defined($slice)) {
	if ($slice_urn ne $slice->urn()) {
	    return GeniResponse->Create(GENIRESPONSE_FORBIDDEN(), undef,
					"Credential does not match the URN");
	}
	if (defined($aggregate)) {
	    return GeniResponse->Create(GENIRESPONSE_REFUSED, undef,
					"Must delete existing slice first");
	}
    }
306
    my $rspec = GeniCM::GetTicketAux($credential,
307
				     $rspecstr, 0, $impotent, 1, 0, undef);
308
309
310
    return $rspec
	if (GeniResponse::IsResponse($rspec));

311
312
313
314
    # Make sure that the next phase sees all changes.
    Experiment->FlushAll();
    Node->FlushAll();

315
    my $response = GeniCM::SliverWorkAux($credential,
316
					 $rspec, $keys, 0, $impotent, 1, 0);
317

318
319
320
321
322
323
324
    if (GeniResponse::IsError($response)) {
	#
	# We have to make sure there is nothing left over since there
	# is no actual ticket, so the resources will not get cleaned
	# up by the daemon. This is mostly cause I am reaching into
	# the V1 code, and its messy.
	#
325
	my $slice = GeniSlice->Lookup($credential->target_urn());
326
327
328
329
330
331
332
333
	if ($slice->Lock() != 0) {
	    print STDERR "CreateSliver: Could not lock $slice before delete\n";
	    return $response;
	}
	if (defined($slice)) {
	    GeniCM::CleanupDeadSlice($slice, 1);
	}
    }
334
335
    my ($sliver_credential, $sliver_manifest) = @{ $response->{'value'} };
    
336
337
338
    #
    # Leave the slice intact on error, so we can go look at it. 
    #
339
    $slice = GeniSlice->Lookup($credential->target_urn());
340
341
342
343
344
345
346
347
348
349
    if (!defined($slice)) {
	print STDERR "CreateSliver: Could not find slice for $credential\n";
	return GeniResponse->Create(GENIRESPONSE_ERROR, undef,
				    "Internal Error");
    }
    if ($slice->Lock() != 0) {
	print STDERR "CreateSliver: Could not lock $slice before start\n";
	return GeniResponse->Create(GENIRESPONSE_ERROR, undef,
				    "Internal Error");
    }
350
    $aggregate = GeniAggregate->SliceAggregate($slice);
351
352
353
354
355
    if (!defined($aggregate)) {
	print STDERR "CreateSliver: Could not find aggregate for $slice\n";
	return GeniResponse->Create(GENIRESPONSE_ERROR, undef,
				    "Internal Error");
    }
356
357
358
359
    # Make sure that the next phase sees all changes.
    Experiment->FlushAll();
    Node->FlushAll();

Leigh B. Stoller's avatar
Leigh B. Stoller committed
360
    if ($aggregate->Start($API_VERSION, 0) != 0) {
361
362
363
	return GeniResponse->Create(GENIRESPONSE_ERROR, undef,
				    "Could not start sliver");
    }
364
365
366
367
368
369
370
371
    GeniCM::UpdateManifest($slice);
    $sliver_manifest = $aggregate->GetManifest(1);
    if (!defined($sliver_manifest)) {
	print STDERR "CreateSliver: Could not get manifest for $aggregate\n";
	return GeniResponse->Create(GENIRESPONSE_ERROR, undef,
				    "Internal Error");
    }
    
372
    $slice->UnLock();
373
374
    return GeniResponse->Create(GENIRESPONSE_SUCCESS,
				[$sliver_credential, $sliver_manifest]);
375
376
377
378
379
380
}

#
# Delete a Sliver.
#
sub DeleteSliver($)
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
{
    my ($argref) = @_;
    my $sliver_urn   = $argref->{'sliver_urn'};
    my $credentials  = $argref->{'credentials'};
    my $impotent     = $argref->{'impotent'} || 0;

    if (! (defined($credentials) && defined($sliver_urn))) {
	return GeniResponse->MalformedArgsResponse("Missing arguments");
    }
    if (! GeniHRN::IsValid($sliver_urn)) {
	return GeniResponse->MalformedArgsResponse("Bad characters in URN");
    }
    my $credential = CheckCredentials($credentials);
    return $credential
	if (GeniResponse::IsResponse($credential));

    #
    # In this implementation, the user must provide a slice or sliver
    # credential
    #
    my ($slice, $aggregate) = Credential2SliceAggregate($credential);
    if (! (defined($slice) && defined($aggregate))) {
	return GeniResponse->Create(GENIRESPONSE_SEARCHFAILED, undef,
				    "Sliver does not exist");
    }
    if ($sliver_urn ne $aggregate->urn()) {
	return GeniResponse->Create(GENIRESPONSE_FORBIDDEN(), undef,
				    "Credential does not match the URN");
    }

    #
    # We need this below to sign the ticket.
    #
    my $authority = GeniCertificate->LoadFromFile($EMULAB_PEMFILE);
    if (!defined($authority)) {
416
	print STDERR " Could not load $EMULAB_PEMFILE\n";
417
418
419
420
421
422
	return GeniResponse->Create(GENIRESPONSE_ERROR);
	
    }
    #
    # We need the user to sign the new ticket to. 
    #
423
    my $user = GeniCM::CreateUserFromCertificate($credential->owner_cert());
424
    if (!defined($user)) {
425
	return GeniResponse->Create(GENIRESPONSE_ERROR);
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
    }
    
    my $response = GeniCM::DeleteSliverAux($credential, $impotent, 1);
    return $response
	if (GeniResponse::IsResponse($response));

    #
    # In the v2 API, return a new ticket for the resources
    # (which were not released). As with all tickets, it will
    # expire very quickly. 
    #
    #
    # Create a new ticket from the manifest.
    #
    my $manifest = $aggregate->GetManifest(0);
    if (!defined($manifest)) {
	print STDERR "No manifest found for $aggregate\n";
	$response = GeniResponse->Create(GENIRESPONSE_ERROR);
	goto bad;
    }
446
447
    my $ticket = GeniTicket->Create($authority, $user,
				    GeniXML::Serialize($manifest));
448
449
450
451
452
    if (!defined($ticket)) {
	print STDERR "Could not create new ticket for $slice\n";
	$response = GeniResponse->Create(GENIRESPONSE_ERROR);
	goto bad;
    }
453
    $ticket->SetSlice($slice);
454
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
    
    if ($ticket->Sign()) {
	$ticket->Delete();
	print STDERR "Could not sign new ticket $ticket\n";
	$response = GeniResponse->Create(GENIRESPONSE_ERROR);
	goto bad;
    }
    if ($ticket->Store()) {
	$ticket->Delete();
	print STDERR "Could not store new ticket $ticket\n";
	$response = GeniResponse->Create(GENIRESPONSE_ERROR);
	goto bad;
    }
    my $slice_uuid = $slice->uuid();
    DBQueryWarn("delete from geni_manifests ".
		"where slice_uuid='$slice_uuid'");
    $slice->UnLock();
    return GeniResponse->Create(GENIRESPONSE_SUCCESS, $ticket->asString());

  bad:
    if (GeniCM::CleanupDeadSlice($slice) != 0) {
	print STDERR "Could not cleanup slice\n";
    }
    return $response;
}

#
# Delete a Slice
#
sub DeleteSlice($)
484
485
{
    my ($argref) = @_;
486
487
488
489
    my $slice_urn    = $argref->{'slice_urn'};
    my $credentials  = $argref->{'credentials'};
    my $impotent     = $argref->{'impotent'} || 0;

490
    if (! (defined($credentials) && defined($slice_urn))) {
491
492
	return GeniResponse->MalformedArgsResponse("Missing arguments");
    }
493
494
495
    if (! GeniHRN::IsValid($slice_urn)) {
	return GeniResponse->MalformedArgsResponse("Bad characters in URN");
    }
496
497
498
    my $credential = CheckCredentials($credentials);
    return $credential
	if (GeniResponse::IsResponse($credential));
499

500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
    #
    # In this implementation, the user must provide a slice credential.
    #
    my ($slice, $aggregate) = Credential2SliceAggregate($credential);
    if (! defined($slice)) {
	return GeniResponse->Create(GENIRESPONSE_SEARCHFAILED, undef,
				    "No such slice here");
    }
    if ($slice_urn ne $slice->urn()) {
	return GeniResponse->Create(GENIRESPONSE_FORBIDDEN(), undef,
				    "Credential does not match the URN");
    }
    if ($slice->Lock() != 0) {
	return GeniResponse->BusyResponse();
    }
Leigh B. Stoller's avatar
Leigh B. Stoller committed
515
    if (GeniCM::CleanupDeadSlice($slice, 1) != 0) {
516
517
518
519
	return GeniResponse->Create(GENIRESPONSE_ERROR, undef,
				    "Could not cleanup slice");
    }
    return GeniResponse->Create(GENIRESPONSE_SUCCESS);
520
521
522
523
524
525
526
527
}

#
# Get a Sliver (credential)
#
sub GetSliver($)
{
    my ($argref) = @_;
528
529
    my $slice_urn    = $argref->{'slice_urn'};
    my $credentials  = $argref->{'credentials'};
530

531
    if (! (defined($credentials) && defined($slice_urn))) {
532
533
	return GeniResponse->MalformedArgsResponse("Missing arguments");
    }
534
535
536
    if (! GeniHRN::IsValid($slice_urn)) {
	return GeniResponse->MalformedArgsResponse("Bad characters in URN");
    }
537
538
539
540
    my $credential = CheckCredentials($credentials);
    return $credential
	if (GeniResponse::IsResponse($credential));

541
542
543
544
545
546
547
548
549
550
551
552
    #
    # In this implementation, the user must provide a slice credential.
    #
    my ($slice, $aggregate) = Credential2SliceAggregate($credential);
    if (! (defined($slice) && defined($aggregate))) {
	return GeniResponse->Create(GENIRESPONSE_BADARGS, undef,
				    "No slice or aggregate here");
    }
    if ($slice_urn ne $slice->urn()) {
	return GeniResponse->Create(GENIRESPONSE_FORBIDDEN(), undef,
				    "Credential does not match the URN");
    }
553
    return GeniCM::GetSliverAux($credential);
554
555
556
}

#
557
# Start a sliver (not sure what this means yet, so reboot for now).
558
#
559
sub StartSliver($)
560
561
{
    my ($argref) = @_;
562
    my $slice_urn    = $argref->{'slice_urn'};
563
    my $sliver_urns  = $argref->{'component_urns'};
564
    my $credentials  = $argref->{'credentials'};
565
    my $manifest     = $argref->{'manifest'};
Leigh B. Stoller's avatar
Leigh B. Stoller committed
566
    
567
568
    return SliverAction("start",
			$slice_urn, $sliver_urns, $credentials, $manifest);
Leigh B. Stoller's avatar
Leigh B. Stoller committed
569
570
571
572
573
574
575
576
577
}

sub StopSliver($)
{
    my ($argref) = @_;
    my $slice_urn    = $argref->{'slice_urn'};
    my $sliver_urns  = $argref->{'component_urns'};
    my $credentials  = $argref->{'credentials'};

578
579
    return SliverAction("stop",
			$slice_urn, $sliver_urns, $credentials, undef);
Leigh B. Stoller's avatar
Leigh B. Stoller committed
580
581
582
583
584
585
}

sub RestartSliver($)
{
    my ($argref) = @_;
    my $slice_urn    = $argref->{'slice_urn'};
586
    my $sliver_urns  = $argref->{'sliver_urns'};
Leigh B. Stoller's avatar
Leigh B. Stoller committed
587
    my $credentials  = $argref->{'credentials'};
588
    my $manifest     = $argref->{'manifest'};
Leigh B. Stoller's avatar
Leigh B. Stoller committed
589

590
591
    return SliverAction("restart",
			$slice_urn, $sliver_urns, $credentials, $manifest);
Leigh B. Stoller's avatar
Leigh B. Stoller committed
592
}
593

594
sub SliverAction($$$$$)
Leigh B. Stoller's avatar
Leigh B. Stoller committed
595
{
596
    my ($action, $slice_urn, $sliver_urns, $credentials, $manifest) = @_;
597
    my $response;
Leigh B. Stoller's avatar
Leigh B. Stoller committed
598
    
599
600
    if (! (defined($credentials) &&
	   (defined($slice_urn) || defined($sliver_urns)))) {
601
602
603
604
605
606
607
608
609
610
611
	return GeniResponse->MalformedArgsResponse("Missing arguments");
    }
    my $credential = CheckCredentials($credentials);
    return $credential
	if (GeniResponse::IsResponse($credential));

    $credential->HasPrivilege( "pi" ) or
	$credential->HasPrivilege( "info" ) or
	return GeniResponse->Create(GENIRESPONSE_FORBIDDEN, undef,
				    "Insufficient privilege");

612
613
614
615
616
617
618
619
620
    if (defined($manifest)) {
	$manifest = GeniXML::Parse($manifest);
	if (!defined($manifest)) {
	    print STDERR "Error reading manifest\n";
	    return GeniResponse->Create(GENIRESPONSE_BADARGS, undef,
					"Bad manifest");
	}
    }
    
621
622
623
    #
    # For now, only allow top level aggregate or the slice
    #
624
    my ($slice, $aggregate) = Credential2SliceAggregate($credential);
Srikanth's avatar
Srikanth committed
625

626
627
628
629
630
    if ( (!defined($slice)) && 
          ($credential->target_urn() =~ /\+authority\+cm$/)) {
          # administrative credentials are presented.
          my $cm_urn = GeniHRN::Generate($OURDOMAIN, "authority", "cm");
          if ($cm_urn != $credential->target_urn()) {
Srikanth's avatar
Srikanth committed
631
            return GeniResponse->Create(GENIRESPONSE_FORBIDDEN(), undef,
632
633
634
635
                      "Credential target does not match CM URN");
          }

      if (!defined($slice_urn)) {
Srikanth's avatar
Srikanth committed
636
637
          return GeniResponse->MalformedArgsResponse("Missing arguments");
      }       
638
639
640
641
642
643
644
645
      $slice = GeniSlice->Lookup($slice_urn);
      return GeniResponse->Create(GENIRESPONSE_ERROR, undef, 
                "No Slice with urn $slice_urn here")
          if (!defined($slice));
      $aggregate = GeniAggregate->SliceAggregate($slice);
      return GeniResponse->Create(GENIRESPONSE_ERROR, undef, 
                      "No Aggregate here")
          if (!defined($aggregate));
Srikanth's avatar
Srikanth committed
646
    } 
647

648
649
650
651
652
653
    if (! (defined($slice) && defined($aggregate))) {
	return GeniResponse->Create(GENIRESPONSE_BADARGS, undef,
				    "No slice or aggregate here");
    }
    if (defined($slice_urn)) {
	if (! GeniHRN::IsValid($slice_urn)) {
Leigh B. Stoller's avatar
Leigh B. Stoller committed
654
655
	    return
		GeniResponse->MalformedArgsResponse("Bad characters in URN");
656
	}
657
658
659
	if ($slice_urn ne $slice->urn()) {
	    return GeniResponse->Create(GENIRESPONSE_FORBIDDEN(), undef,
					"Credential does not match the URN");
660
	}
661
662
663
664
665
666
667
668
669
670
671
672
673
674
675
676
677
678
    }
    if ($slice->Lock() != 0) {
	return GeniResponse->BusyResponse();
    }
    # Shutdown slices get nothing.
    if ($slice->shutdown()) {
	$slice->UnLock();
	return GeniResponse->Create(GENIRESPONSE_FORBIDDEN, undef,
				    "Slice has been shutdown");
    }
    if ($aggregate->ComputeState()) {
	$slice->UnLock();
	print STDERR "Could not determine current state\n";
	return GeniResponse->Create(GENIRESPONSE_ERROR);
    }
    my $CheckState = sub {
	my ($object, $action) = @_;

Leigh B. Stoller's avatar
Leigh B. Stoller committed
679
	if ($action eq "start") {
680
	    if ($object->state() ne "stopped" && $object->state() ne "mixed") {
Leigh B. Stoller's avatar
Leigh B. Stoller committed
681
682
683
684
685
		return GeniResponse->Create(GENIRESPONSE_REFUSED, undef,
					    "Sliver is not stopped (yet)");
	    }
	}
	elsif ($action eq "stop") {
686
	    if ($object->state() ne "started" && $object->state() ne "mixed") {
Leigh B. Stoller's avatar
Leigh B. Stoller committed
687
688
689
690
691
		return GeniResponse->Create(GENIRESPONSE_REFUSED, undef,
					    "Sliver is not started (yet)");
	    }
	}
	elsif ($action eq "restart") {
692
	    if ($object->state() ne "started" && $object->state() ne "mixed") {
Leigh B. Stoller's avatar
Leigh B. Stoller committed
693
694
695
		return GeniResponse->Create(GENIRESPONSE_REFUSED, undef,
					    "Sliver is not started (yet)");
	    }
696
697
698
699
700
701
702
703
	}
	return 0;
    };
    my $PerformAction = sub {
	my ($object, $action) = @_;

	if ($action eq "start") {
	    if ($object->Start($API_VERSION, 0) != 0) {
Leigh B. Stoller's avatar
Leigh B. Stoller committed
704
		return GeniResponse->Create(GENIRESPONSE_ERROR, undef,
705
					    "Could not start sliver");
Leigh B. Stoller's avatar
Leigh B. Stoller committed
706
	    }
707
	}
708
709
710
711
712
713
714
715
716
717
718
719
720
721
722
723
724
725
726
727
	elsif ($action eq "stop") {
	    if ($object->Stop($API_VERSION) != 0) {
		return GeniResponse->Create(GENIRESPONSE_ERROR, undef,
					    "Could not stop sliver");
	    }
	}
	elsif ($action eq "restart") {
	    if ($object->Start($API_VERSION, 1) != 0) {
		return GeniResponse->Create(GENIRESPONSE_ERROR, undef,
					    "Could not restart sliver");
	    }	    
	}
	return 0;
    };

    if (defined($slice_urn)) {
	$response = &$CheckState($aggregate, $action);
	goto bad
	    if (GeniResponse::IsResponse($response));
	    
728
729
730
731
732
733
734
735
736
	if ($action eq "start" && defined($manifest)) {
	    if ($aggregate->ProcessManifest($manifest)) {
		$response = GeniResponse->Create(GENIRESPONSE_ERROR,
						 undef,
						 "Error processing manifest");
		goto bad;
	    }
	}
	
737
738
739
740
	$response = &$PerformAction($aggregate, $action);
	goto bad
	    if (GeniResponse::IsResponse($response));

741
	if ($action eq "start" || $action eq "restart") {
742
743
	    GeniCM::UpdateManifest($slice);
	}
744
745
	$slice->UnLock();
	return GeniResponse->Create(GENIRESPONSE_SUCCESS);
746
    }
Leigh B. Stoller's avatar
Leigh B. Stoller committed
747
    else {
748
	my @slivers = ();
Leigh B. Stoller's avatar
Leigh B. Stoller committed
749

750
751
752
753
754
	#
	# Sanity check all arguments before doing anything.
	#
	foreach my $urn (@{ $sliver_urns }) {
	    my $sliver = GeniSliver->Lookup($urn);
Leigh B. Stoller's avatar
Leigh B. Stoller committed
755
756
757
758
759
760
	    if (!defined($sliver)) {
		$response = GeniResponse->Create(GENIRESPONSE_SEARCHFAILED,
						 undef,
						 "Nothing here by that name");
		goto bad;
	    }
761
762
763
764
765
766
767
768
	    
	    $response = &$CheckState($sliver, $action);
	    goto bad
		if (GeniResponse::IsResponse($response));

	    push(@slivers, $sliver);
	}
	foreach my $sliver (@slivers) {
769
770
771
772
773
774
775
776
	    if ($action eq "start" && defined($manifest)) {
		if ($sliver->ProcessManifest($manifest)) {
		    $response = GeniResponse->Create(GENIRESPONSE_ERROR,
				     undef,
				     "Error processing manifest for $sliver");
		    goto bad;
		}
	    }
777
778
779
780
781
782
	    $response = &$PerformAction($sliver, $action);
	    goto bad
		if (GeniResponse::IsResponse($response));
	}
	$slice->UnLock();
	return GeniResponse->Create(GENIRESPONSE_SUCCESS);
Leigh B. Stoller's avatar
Leigh B. Stoller committed
783
    }
784
785
786
  bad:
    $slice->UnLock();
    return $response;
787
788
789
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
}

#
# Get sliver status
#
sub SliverStatus($)
{
    my ($argref) = @_;
    my $slice_urn    = $argref->{'slice_urn'};
    my $credentials  = $argref->{'credentials'};

    if (! (defined($credentials) && defined($slice_urn))) {
	return GeniResponse->MalformedArgsResponse("Missing arguments");
    }
    if (! GeniHRN::IsValid($slice_urn)) {
	return GeniResponse->MalformedArgsResponse("Bad characters in URN");
    }
    my $credential = CheckCredentials($credentials);
    return $credential
	if (GeniResponse::IsResponse($credential));

    $credential->HasPrivilege( "pi" ) or
	$credential->HasPrivilege( "info" ) or
	return GeniResponse->Create(GENIRESPONSE_FORBIDDEN, undef,
				    "Insufficient privilege");

    #
    # For now, only allow top level aggregate or the slice
    #
    my ($slice, $aggregate) = Credential2SliceAggregate($credential);
    if (! (defined($slice) && defined($aggregate))) {
	return GeniResponse->Create(GENIRESPONSE_BADARGS, undef,
				    "No slice or aggregate here");
    }
    if ($slice_urn ne $slice->urn()) {
	return GeniResponse->Create(GENIRESPONSE_FORBIDDEN(), undef,
				    "Credential does not match the URN");
824
825
826
827
    }
    if ($slice->Lock() != 0) {
	return GeniResponse->BusyResponse();
    }
Leigh B. Stoller's avatar
Leigh B. Stoller committed
828
829
830
831
832
    if ($aggregate->ComputeState()) {
	print STDERR "SliverStatus: Could not compute state for $aggregate\n";
	$slice->UnLock();
	return GeniResponse->Create(GENIRESPONSE_ERROR);
    }
833
834
835
836
837
838
839
840
841
842
843

    #
    # Grab all the slivers for this slice, and then
    # look for just the nodes.
    #
    my @slivers    = ();
    if ($aggregate->SliverList(\@slivers) != 0) {
	print STDERR "SliverStatus: Could not get slivers for $aggregate\n";
	$slice->UnLock();
	return GeniResponse->Create(GENIRESPONSE_ERROR);
    }
Leigh B. Stoller's avatar
Leigh B. Stoller committed
844
845
846

    my $blob = {
	"state"   => $aggregate->state(),
Leigh B. Stoller's avatar
Leigh B. Stoller committed
847
	"status"  => $aggregate->status(),
Leigh B. Stoller's avatar
Leigh B. Stoller committed
848
849
850
	"details" => {},
    };
    
851
852
853
854
855
    foreach my $sliver (@slivers) {
	next
	    if ($sliver->isa("GeniAggregate"));
	next
	    if ($sliver->resource_type() ne "Node");
856

857
858
	my $sliver_urn    = $sliver->sliver_urn();
	my $component_urn = $sliver->component_urn();
Leigh B. Stoller's avatar
Leigh B. Stoller committed
859
860
861
	my $state         = $sliver->state();
	my $status        = $sliver->status();
	my $error         = "";
862

863
864
865
866
	# New is the same as stopped. Separate state is handy.
	$state = "stopped"
	    if ($state eq "new");

Leigh B. Stoller's avatar
Leigh B. Stoller committed
867
868
	if ($status eq "failed") {
	    $error = $sliver->ErrorLog();
869
	}
870
871
	$blob->{'details'}->{$sliver_urn} = {
	    "component_urn" => $component_urn,
872
873
874
875
876
877
878
	    "state"  => $state,
	    "status" => $status,
	    "error"  => $error,
	};
    }
    $slice->UnLock();
    return GeniResponse->Create(GENIRESPONSE_SUCCESS, $blob);
879
880
881
882
883
884
885
886
}

#
# Shutdown sliver
#
sub Shutdown($)
{
    my ($argref) = @_;
887
888
889
890
891
892
893
894
895
896
897
    my $slice_urn    = $argref->{'slice_urn'};
    my $clear        = $argref->{'clear'} || 0;
    my $credentials  = $argref->{'credentials'};

    # For now, I am not worrying about the slice_urn argument.
    if (! (defined($credentials))) {
	return GeniResponse->MalformedArgsResponse("Missing arguments");
    }
    my $credential = CheckCredentials($credentials);
    return $credential
	if (GeniResponse::IsResponse($credential));
898

899
    return GeniCM::ShutdownAux($credential, $clear);
900
901
902
}

#
903
# Renew a slice
904
#
905
sub RenewSlice($)
906
907
{
    my ($argref) = @_;
908
    my $slice_urn    = $argref->{'slice_urn'};
909
    my $valid_until  = $argref->{'valid_until'} || $argref->{'expiration'};
910
911
    my $credentials  = $argref->{'credentials'};

Leigh B. Stoller's avatar
Leigh B. Stoller committed
912
    if (! (defined($credentials) && defined($slice_urn))) {
913
914
	return GeniResponse->MalformedArgsResponse("Missing arguments");
    }
915
916
917
    if (! GeniHRN::IsValid($slice_urn)) {
	return GeniResponse->MalformedArgsResponse("Bad characters in URN");
    }
918
919
920
921
    my $credential = CheckCredentials($credentials);
    return $credential
	if (GeniResponse::IsResponse($credential));

922
923
924
925
926
927
928
929
930
931
932
933
    #
    # In this implementation, the user must provide a slice credential.
    #
    my ($slice, $aggregate) = Credential2SliceAggregate($credential);
    if (! (defined($slice) && defined($aggregate))) {
	return GeniResponse->Create(GENIRESPONSE_BADARGS, undef,
				    "No slice or aggregate here");
    }
    if ($slice_urn ne $slice->urn()) {
	return GeniResponse->Create(GENIRESPONSE_FORBIDDEN(), undef,
				    "Credential does not match the URN");
    }
934
935
    my $credential_objects = [];
    foreach my $credstr (@$credentials) {
936
937
938
        my $cred = CheckCredential($credstr);
        push(@$credential_objects, $cred) 
            if(!GeniResponse::IsResponse($cred));
939
940
    }
    return GeniCM::RenewSliverAux($credential_objects, $valid_until);
941
942
}

943
944
945
946
947
948
949
950
951
952
953
954
955
956
957
958
959
960
961
962
963
964
965
966
967
968
969
970
971
972
973
974
975
976
977
978
979
980
981
982
983
984
985
986
987
988
989
990
991
#
# Get a Ticket.
#
sub GetTicket($)
{
    my ($argref) = @_;
    my $slice_urn    = $argref->{'slice_urn'};
    my $rspecstr     = $argref->{'rspec'};
    my $credentials  = $argref->{'credentials'};
    my $impotent     = $argref->{'impotent'} || 0;

    if (! (defined($credentials) &&
	   defined($slice_urn) && defined($rspecstr))) {
	return GeniResponse->MalformedArgsResponse("Missing arguments");
    }
    if (! ($rspecstr =~ /^[\040-\176\012\015\011]+$/)) {
	return GeniResponse->MalformedArgsResponse("Bad characters in rspec");
    }
    my $credential = CheckCredentials($credentials);
    return $credential
	if (GeniResponse::IsResponse($credential));

    #
    # This implementation requires a slice credential, and it must
    # match the slice_urn.
    #
    my ($slice, $aggregate) = Credential2SliceAggregate($credential);
    if (defined($slice)) {
	if ($slice_urn ne $slice->urn()) {
	    return GeniResponse->Create(GENIRESPONSE_FORBIDDEN(), undef,
					"Credential does not match the URN");
	}
	#
	# GetTicket applies only to slices that are not active. Must
	# use UpdateSliver() for an active sliver.
	#
	if (defined($aggregate)) {
	    return GeniResponse->Create(GENIRESPONSE_REFUSED(), undef,
				"Cannot get a ticket for active sliver");
	}
	#
	# It is an error if there is an outstanding ticket. That ticket
	# must be released first.
	#
	my $ticket = GeniTicket->SliceTicket($slice);
	if (defined($ticket)) {
	    return GeniResponse->Create(GENIRESPONSE_REFUSED, undef,
				    "Must release unredeemed ticket first");
	}
992
993
994
995
	if ($slice->IsExpired()) {
	    return GeniResponse->Create(GENIRESPONSE_REFUSED, undef,
					"Slice has expired");
	}
996
997
998
999
1000
    }
    else {
	# Slice does not exist yet. 
    }
    return GeniCM::GetTicketAux($credential,