mirror of
				https://github.com/xcat2/xcat-core.git
				synced 2025-10-31 03:12:30 +00:00 
			
		
		
		
	git-svn-id: https://svn.code.sf.net/p/xcat/code/xcat-core/trunk@14914 8638fb3e-16cb-4fca-ae20-7b5d299a9bcd
		
			
				
	
	
		
			133 lines
		
	
	
		
			3.6 KiB
		
	
	
	
		
			Perl
		
	
	
	
	
	
			
		
		
	
	
			133 lines
		
	
	
		
			3.6 KiB
		
	
	
	
		
			Perl
		
	
	
	
	
	
| # IBM(c) 2007 EPL license http://www.eclipse.org/legal/epl-v10.html
 | |
| #-------------------------------------------------------
 | |
| 
 | |
| =head1
 | |
|   xCAT plugin package to handle getpostscript command
 | |
| 
 | |
| =cut
 | |
| 
 | |
| #-------------------------------------------------------
 | |
| package xCAT_plugin::getpostscript;
 | |
| use xCAT::Utils;
 | |
| use xCAT::MsgUtils;
 | |
| use xCAT::NodeRange;
 | |
| 1;
 | |
| 
 | |
| #-------------------------------------------------------
 | |
| 
 | |
| =head3  handled_commands
 | |
| 
 | |
| Return list of commands handled by this plugin
 | |
| 
 | |
| =cut
 | |
| 
 | |
| #-------------------------------------------------------
 | |
| 
 | |
| sub handled_commands
 | |
| {
 | |
|     return { 
 | |
|                'getpostscript' => "getpostscript",
 | |
|                'postage' => "getpostscript"
 | |
|            };
 | |
| }
 | |
| 
 | |
| 
 | |
| #-------------------------------------------------------
 | |
| 
 | |
| =head3  process_request
 | |
| 
 | |
|   Process the command
 | |
| 
 | |
| =cut
 | |
| 
 | |
| #-------------------------------------------------------
 | |
| sub process_request
 | |
| {
 | |
|     my $request  = shift;
 | |
|     my $callback = shift;
 | |
|     my $nodes    = $request->{node};
 | |
|     my $command  = $request->{command}->[0];
 | |
|     my $args     = $request->{arg};
 | |
|     my $envs     = $request->{env};
 | |
|     my $rsp;
 | |
|     my $i = 1;
 | |
|     my @nodes=@$nodes; 
 | |
|     # do your processing here
 | |
|     # return info
 | |
|     
 | |
|     if($command eq 'postage'){
 | |
|         return postage($nodes, $callback);
 | |
|     }
 | |
| 
 | |
|     my $client;
 | |
|     if ($::XCATSITEVALS{nodeauthentication}) { #if requiring node authentication, this request will have a certificate associated with it, use it instead of name resolution
 | |
| 	unless (ref $request->{username}) { return; } #TODO: log an attempt without credentials? 
 | |
| 	$client = $request->{username}->[0];
 | |
|     } else {
 | |
| 	    unless ($request->{'_xcat_clienthost'}->[0]) {
 | |
| 	      #ERROR? malformed request
 | |
| 	      return; #nothing to do here...
 | |
| 	    }
 | |
| 	    $client = $request->{'_xcat_clienthost'}->[0];
 | |
|     }
 | |
| 
 | |
|     if ($client) { ($client) = noderange($client) };
 | |
|     unless ($client) { #Not able to do identify the host in question
 | |
|        xCAT::MsgUtils->message("S","Received getpostscript from $client, which couldn't be correlated to a node (domain mismatch?)");
 | |
|       return;
 | |
|     }
 | |
|     my $state;
 | |
|     if ($request->{scripttype}) { $state = $request->{scripttype}->[0];}
 | |
| 
 | |
|     require xCAT::Postage;
 | |
|     my $args = $request->{arg};
 | |
|     my @scriptcontents;
 | |
|     if( defined($args) && grep(/version2/, @$args)) {
 | |
|         @scriptcontents = xCAT::Postage::makescript([$client],$state,$callback);  # the new method, use the template
 | |
|     } else {
 | |
|         #print "client:$client\n";
 | |
|         @scriptcontents = xCAT::Postage::makescript($client,$state,$callback); # the original method
 | |
|     }
 | |
|     if (scalar(@scriptcontents)) {
 | |
|        $rsp->{data} = \@scriptcontents;
 | |
|     }
 | |
|     $callback->($rsp);
 | |
| }
 | |
| 
 | |
| sub postage {
 | |
|     my $nodes = shift;
 | |
|     my $callback = shift;
 | |
|     require xCAT::Postage;
 | |
|     foreach my $node (@$nodes){
 | |
|         my @scriptcontents = xCAT::Postage::makescript($node,'postscripts',$callback);
 | |
|         my $ps = 0;
 | |
|         my $pbs = 0;
 | |
|         foreach(@scriptcontents){
 | |
|             chomp($_);
 | |
|             if($_ =~ "postscripts-start-here"){
 | |
|                 $ps = 1;
 | |
|                 next;
 | |
|               
 | |
|             }
 | |
|             if($_ =~ "postscripts-end-here"){
 | |
|                 $ps = 0;
 | |
|                 next;
 | |
|             }
 | |
|             if($_ =~ "postbootscripts-start-here"){
 | |
|                 $pbs = 1;
 | |
|                 next;
 | |
|             }
 | |
|             if($_ =~ "postbootscripts-end-here"){
 | |
|                 $pbs = 0;
 | |
|                 next;
 | |
|             }
 | |
|             if($ps eq 1){ 
 | |
|                 $callback->({info => ["$node: postscript: $_"]});
 | |
|             }
 | |
|             if($pbs eq 1){
 | |
|                 $callback->({info => ["$node: postbootscript: $_"]});
 | |
|             }
 | |
|         }
 | |
|     }  
 | |
| }
 |