| #!/usr/bin/perl -w |
| ############################################################################### |
| # $Id$ |
| ############################################################################### |
| # Licensed to the Apache Software Foundation (ASF) under one or more |
| # contributor license agreements. See the NOTICE file distributed with |
| # this work for additional information regarding copyright ownership. |
| # The ASF licenses this file to You under the Apache License, Version 2.0 |
| # (the "License"); you may not use this file except in compliance with |
| # the License. You may obtain a copy of the License at |
| # |
| # http://www.apache.org/licenses/LICENSE-2.0 |
| # |
| # Unless required by applicable law or agreed to in writing, software |
| # distributed under the License is distributed on an "AS IS" BASIS, |
| # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| # See the License for the specific language governing permissions and |
| # limitations under the License. |
| ############################################################################### |
| |
| =head1 NAME |
| |
| VCL::Module::OS.pm - VCL base operating system module |
| |
| =head1 SYNOPSIS |
| |
| Needs to be written |
| |
| =head1 DESCRIPTION |
| |
| This module provides VCL support operating systems. |
| |
| =cut |
| |
| ############################################################################### |
| package VCL::Module::OS; |
| |
| # Specify the lib path using FindBin |
| use FindBin; |
| use lib "$FindBin::Bin/../.."; |
| |
| # Configure inheritance |
| use base qw(VCL::Module); |
| |
| # Specify the version of this module |
| our $VERSION = '2.5.1'; |
| |
| # Specify the version of Perl to use |
| use 5.008000; |
| |
| use strict; |
| use warnings; |
| use diagnostics; |
| use English '-no_match_vars'; |
| use File::Temp qw(tempdir); |
| #use POSIX qw(tmpnam); |
| use File::Temp qw/ :POSIX /; |
| use Net::SSH::Expect; |
| use List::Util qw(min max); |
| |
| use VCL::utils; |
| |
| ############################################################################### |
| |
| =head1 OBJECT METHODS |
| |
| =cut |
| |
| #////////////////////////////////////////////////////////////////////////////// |
| |
| =head2 pre_capture |
| |
| Parameters : $arguments->{end_state} |
| Returns : boolean |
| Description : Performs the tasks common to all OS's that must be done to the |
| computer prior to capturing an image: |
| * Check if the computer is responding to SSH |
| * If not responding, check if computer is powered on |
| * Power on computer if powered off and wait for SSH to respond |
| * Create currentimage.txt file |
| |
| =cut |
| |
| sub pre_capture { |
| my $self = shift; |
| my $args = shift; |
| if (ref($self) !~ /VCL::Module/i) { |
| notify($ERRORS{'CRITICAL'}, 0, "subroutine was called as a function, it must be called as a class method"); |
| return; |
| } |
| |
| my $computer_node_name = $self->data->get_computer_node_name(); |
| notify($ERRORS{'OK'}, 0, "beginning common image capture preparation tasks"); |
| |
| # Make sure the computer is responding to SSH |
| # If it is not, check if it is powered on |
| if (!$self->is_ssh_responding()) { |
| notify($ERRORS{'OK'}, 0, "$computer_node_name is not responding to SSH, checking if it is powered on"); |
| my $power_status = $self->provisioner->power_status(); |
| if (!$power_status) { |
| notify($ERRORS{'WARNING'}, 0, "unable to complete capture preparation tasks, $computer_node_name is not responding to SSH and the power status could not be determined"); |
| return; |
| } |
| elsif ($power_status =~ /on/i) { |
| notify($ERRORS{'WARNING'}, 0, "unable to complete capture preparation tasks, $computer_node_name is powered on but not responding to SSH"); |
| return; |
| } |
| else { |
| notify($ERRORS{'DEBUG'}, 0, "$computer_node_name is powered off, attempting to power it on"); |
| if (!$self->provisioner->power_on()) { |
| notify($ERRORS{'WARNING'}, 0, "unable to complete capture preparation tasks, $computer_node_name could not be powered on"); |
| return; |
| } |
| |
| # Wait for computer to respond to SSH |
| if (!$self->wait_for_response(30, 300, 10)) { |
| notify($ERRORS{'WARNING'}, 0, "unable to complete capture preparation tasks, $computer_node_name never responded to SSH after it was powered on"); |
| return; |
| } |
| } |
| } |
| |
| # Delete an existing node configuration directory to clear out any scripts and log files from a previous image revision |
| my $node_configuration_directory = $self->get_node_configuration_directory(); |
| if ($node_configuration_directory) { |
| $self->delete_file($node_configuration_directory); |
| } |
| |
| # Create the currentimage.txt file |
| if (!$self->create_currentimage_txt()) { |
| notify($ERRORS{'WARNING'}, 0, "failed to create currentimage.txt on $computer_node_name"); |
| return 0; |
| } |
| |
| # Run custom pre_capture scripts |
| $self->run_stage_scripts('pre_capture'); |
| |
| # Delete reservation_info.json |
| my $reservation_info_json_file_path = $self->get_reservation_info_json_file_path(); |
| if ($reservation_info_json_file_path) { |
| $self->delete_file($reservation_info_json_file_path); |
| } |
| |
| notify($ERRORS{'OK'}, 0, "completed common image capture preparation tasks"); |
| return 1; |
| } |
| |
| #////////////////////////////////////////////////////////////////////////////// |
| |
| =head2 post_capture |
| |
| Parameters : none |
| Returns : boolean |
| Description : Performs the tasks common to all OS's that must be done to the |
| computer after capturing an image: |
| * Runs post_capture stage scripts on the management node if any |
| exist |
| |
| =cut |
| |
| sub post_capture { |
| my $self = shift; |
| if (ref($self) !~ /VCL::Module/i) { |
| notify($ERRORS{'CRITICAL'}, 0, "subroutine was called as a function, it must be called as a class method"); |
| return; |
| } |
| |
| # Run custom post_capture scripts |
| $self->run_stage_scripts('post_capture'); |
| |
| return 1; |
| } |
| |
| #////////////////////////////////////////////////////////////////////////////// |
| |
| =head2 post_load |
| |
| Parameters : none |
| Returns : boolean |
| Description : Performs common OS steps after an image is loaded. |
| |
| =cut |
| |
| sub post_load { |
| my $self = shift; |
| if (ref($self) !~ /VCL::Module/) { |
| notify($ERRORS{'CRITICAL'}, 0, "subroutine was called as a function, it must be called as a class method"); |
| return; |
| } |
| |
| # Run custom post_load scripts |
| $self->run_stage_scripts('post_load'); |
| |
| return 1; |
| } |
| |
| #////////////////////////////////////////////////////////////////////////////// |
| |
| =head2 reserve |
| |
| Parameters : none |
| Returns : boolean |
| Description : Performs common OS steps to reserve the computer for a user: |
| * The public IP address is updated if necessary |
| * If the computer is mapped to a NAT host: |
| ** General NAT host configuration is performed if not previously |
| done. |
| ** The NAT host is prepared for the reservation but specific port |
| forwardings are not added. |
| * User accounts are added |
| |
| Note: The 'reserve' subroutine should never open the firewall for |
| a connection. This is done by the 'grant_access' subroutine. |
| |
| =cut |
| |
| sub reserve { |
| my $self = shift; |
| if (ref($self) !~ /VCL::Module/) { |
| notify($ERRORS{'CRITICAL'}, 0, "subroutine was called as a function, it must be called as a class method"); |
| return; |
| } |
| |
| my $computer_node_name = $self->data->get_computer_node_name(); |
| my $nathost_hostname = $self->data->get_nathost_hostname(0); |
| |
| # Make sure the public IP address assigned to the computer matches the database |
| if (!$self->update_public_ip_address()) { |
| notify($ERRORS{'WARNING'}, 0, "unable to reserve computer, failed to update IP address"); |
| return; |
| } |
| |
| |
| # Check if the computer is mapped to a NAT host |
| if ($nathost_hostname) { |
| # Perform general NAT host configuration - this only needs to be done once, nat_configure_host checks if it was already done |
| if (!$self->nathost_os->firewall->nat_configure_host()) { |
| notify($ERRORS{'WARNING'}, 0, "unable to reserve $computer_node_name, failed to configure NAT on $nathost_hostname"); |
| return; |
| } |
| |
| # Perform reservation-specific NAT configuration |
| if (!$self->nathost_os->firewall->nat_configure_reservation()) { |
| notify($ERRORS{'WARNING'}, 0, "unable to reserve $computer_node_name, failed to configure NAT on $nathost_hostname for this reservation"); |
| return; |
| } |
| } |
| |
| |
| # Add user accounts to the computer |
| if (!$self->add_user_accounts()) { |
| notify($ERRORS{'WARNING'}, 0, "unable to reserve computer, failed add user accounts"); |
| return; |
| } |
| |
| return 1; |
| } |
| |
| #////////////////////////////////////////////////////////////////////////////// |
| |
| =head2 post_reserve |
| |
| Parameters : none |
| Returns : boolean |
| Description : Performs common OS steps after an image is loaded. |
| |
| =cut |
| |
| sub post_reserve { |
| my $self = shift; |
| if (ref($self) !~ /VCL::Module/) { |
| notify($ERRORS{'CRITICAL'}, 0, "subroutine was called as a function, it must be called as a class method"); |
| return 0; |
| } |
| |
| # Run custom post_reserve scripts |
| $self->run_stage_scripts('post_reserve'); |
| |
| return 1; |
| } |
| |
| #////////////////////////////////////////////////////////////////////////////// |
| |
| =head2 post_initial_connection |
| |
| Parameters : none |
| Returns : boolean |
| Description : Performs common OS steps after a user makes an initial |
| connection. |
| |
| =cut |
| |
| sub post_initial_connection { |
| my $self = shift; |
| if (ref($self) !~ /VCL::Module/) { |
| notify($ERRORS{'CRITICAL'}, 0, "subroutine was called as a function, it must be called as a class method"); |
| return; |
| } |
| |
| # Run custom post_initial_connection scripts |
| $self->run_stage_scripts('post_initial_connection'); |
| |
| return 1; |
| } |
| |
| #////////////////////////////////////////////////////////////////////////////// |
| |
| =head2 post_reservation |
| |
| Parameters : none |
| Returns : boolean |
| Description : Performs common OS steps after a user's reservation is over. |
| |
| =cut |
| |
| sub post_reservation { |
| my $self = shift; |
| if (ref($self) !~ /VCL::Module/) { |
| notify($ERRORS{'CRITICAL'}, 0, "subroutine was called as a function, it must be called as a class method"); |
| return 0; |
| } |
| |
| # Run custom post_reservation scripts |
| $self->run_stage_scripts('post_reservation'); |
| |
| return 1; |
| } |
| |
| #////////////////////////////////////////////////////////////////////////////// |
| |
| =head2 pre_reload |
| |
| Parameters : none |
| Returns : boolean |
| Description : Performs common OS steps prior to a computer being reloaded. |
| |
| =cut |
| |
| sub pre_reload { |
| my $self = shift; |
| if (ref($self) !~ /VCL::Module/) { |
| notify($ERRORS{'CRITICAL'}, 0, "subroutine was called as a function, it must be called as a class method"); |
| return 0; |
| } |
| |
| # Run custom pre_reload scripts |
| $self->run_stage_scripts('pre_reload'); |
| |
| return 1; |
| } |
| |
| #////////////////////////////////////////////////////////////////////////////// |
| |
| =head2 add_user_accounts |
| |
| Parameters : none |
| Returns : boolean |
| Description : Adds all user accounts to the computer for a reservation. The |
| reservationaccounts table is checked. If the user already exists |
| in the table, it is assumed the user was previously created and |
| nothing is done. If the user doesn't exist in the table it is |
| added. If an entry for a user exists in the reservationaccounts |
| table but the user is not assigned to the reservation, the user |
| is deleted from the computer. |
| |
| =cut |
| |
| sub add_user_accounts { |
| my $self = shift; |
| if (ref($self) !~ /VCL::Module/i) { |
| notify($ERRORS{'CRITICAL'}, 0, "subroutine was called as a function, it must be called as a class method"); |
| return; |
| } |
| |
| my $request_user_id = $self->data->get_user_id(); |
| my $request_state_name = $self->data->get_request_state_name(); |
| my $reservation_id = $self->data->get_reservation_id(); |
| my $reservation_users = $self->data->get_reservation_users(); |
| my $reservation_password = $self->data->get_reservation_password(0); |
| my $computer_node_name = $self->data->get_computer_node_name(); |
| |
| # Collect users in reservationaccounts table |
| my $reservation_accounts = get_reservation_accounts($reservation_id); |
| |
| # Add users |
| RESERVATION_USER: foreach my $user_id (sort keys %$reservation_users) { |
| my $username = $reservation_users->{$user_id}{unityid}; |
| my $uid = $reservation_users->{$user_id}{uid}; |
| my $root_access = $reservation_users->{$user_id}{ROOTACCESS}; |
| my $use_public_keys = $reservation_users->{$user_id}{usepublickeys}; |
| |
| # If the $use_public_keys flag is set, retrieve the keys |
| my $ssh_public_keys; |
| if ($use_public_keys) { |
| $ssh_public_keys = $reservation_users->{$user_id}{sshpublickeys}; |
| } |
| |
| my $password; |
| |
| # Check if entry needs to be added to the useraccounts table |
| if (defined($reservation_accounts->{$user_id}) && ($request_state_name =~ /servermodified/)) { |
| # Entry already exists in useraccounts table and is servermodified, assume everything is correct skip to next user |
| notify($ERRORS{'DEBUG'}, 0, "entry already exists in useraccounts table for $username (ID: $user_id) and request_state_name = $request_state_name"); |
| |
| # Make sure user's root access is correct - may have been moved from admin to access group, and vice versa |
| if ($root_access) { |
| $self->grant_administrative_access($username) if ($self->can('grant_administrative_access')); |
| } |
| else { |
| $self->revoke_administrative_access($username) if ($self->can('revoke_administrative_access')); |
| } |
| |
| next RESERVATION_USER; |
| } |
| else { |
| notify($ERRORS{'DEBUG'}, 0, "entry does not already exist in useraccounts table for $username (ID: $user_id)"); |
| |
| # Determine whether or not the user account's password should be set |
| my $should_set_user_password = 1; |
| if ($self->can('should_set_user_password')) { |
| $should_set_user_password = $self->should_set_user_password($user_id); |
| if (!defined($should_set_user_password)) { |
| notify($ERRORS{'CRITICAL'}, 0, "failed to determine if user account password should be set, user ID $user_id, assuming password should be set"); |
| $should_set_user_password = 1; |
| } |
| } |
| |
| if ($should_set_user_password) { |
| # Check if this is the request owner user ID and the reservation password has already been set |
| if ($user_id eq $request_user_id) { |
| if ($reservation_password) { |
| $password = $reservation_password; |
| notify($ERRORS{'DEBUG'}, 0, "user $username (ID: $user_id) is request owner, using existing reservation password: $password"); |
| } |
| else { |
| # Generate a new random password |
| $password = getpw(); |
| $self->data->set_reservation_password($password); |
| notify($ERRORS{'DEBUG'}, 0, "user $username (ID: $user_id) is request owner, generated new password: $password"); |
| |
| # Update the password in the reservation table |
| if (!update_reservation_password($reservation_id, $password)) { |
| $self->reservation_failed("failed to update password in the reservation table"); |
| return; |
| } |
| } |
| } |
| else { |
| # Generate a new random password |
| $password = getpw(); |
| notify($ERRORS{'DEBUG'}, 0, "user $username (ID: $user_id) is not the request owner, generated new password: $password"); |
| } |
| } |
| |
| # Add an entry to the useraccounts table |
| if (!add_reservation_account($reservation_id, $user_id, $password)) { |
| notify($ERRORS{'CRITICAL'}, 0, "failed to add entry to reservationaccounts table for $username (ID: $user_id)"); |
| return; |
| } |
| |
| # Create user on the OS |
| if (!$self->create_user({ |
| username => $username, |
| password => $password, |
| root_access => $root_access, |
| uid => $uid, |
| ssh_public_keys => $ssh_public_keys, |
| })) { |
| notify($ERRORS{'WARNING'}, 0, "failed to create user on $computer_node_name, removing entry added to reservationaccounts table"); |
| |
| # Delete entry to the useraccounts table |
| if (!delete_reservation_account($reservation_id, $user_id)) { |
| notify($ERRORS{'CRITICAL'}, 0, "failed to delete entry from reservationaccounts table for $username (ID: $user_id)"); |
| } |
| |
| return; |
| } |
| } |
| } |
| |
| # Remove anyone listed in reservationaccounts that is not a reservation user |
| foreach my $user_id (sort keys %$reservation_accounts) { |
| if (defined($reservation_users->{$user_id})) { |
| next; |
| } |
| |
| my $username = $reservation_accounts->{$user_id}{username}; |
| |
| notify($ERRORS{'OK'}, 0, "user $username (ID: $user_id) exists in reservationsaccounts table but is not assigned to this reservation, attempting to delete user"); |
| |
| # Delete the user from OS |
| if (!$self->delete_user($username)) { |
| notify($ERRORS{'WARNING'}, 0, "failed to delete user $username (ID: $user_id) from $computer_node_name"); |
| next; |
| } |
| |
| # Delete entry from reservationaccounts |
| if (!delete_reservation_account($reservation_id, $user_id)) { |
| notify($ERRORS{'CRITICAL'}, 0, "failed to delete entry from reservationaccounts table for user $username (ID: $user_id)"); |
| } |
| } |
| |
| return 1; |
| } |
| |
| #////////////////////////////////////////////////////////////////////////////// |
| |
| =head2 delete_user_accounts |
| |
| Parameters : none |
| Returns : boolean |
| Description : Deletes all user accounts from the computer which are assigned to |
| the reservation or an entry exists in the reservationaccounts |
| table. |
| |
| =cut |
| |
| sub delete_user_accounts { |
| my $self = shift; |
| if (ref($self) !~ /VCL::Module/i) { |
| notify($ERRORS{'CRITICAL'}, 0, "subroutine was called as a function, it must be called as a class method"); |
| return; |
| } |
| |
| my $reservation_id = $self->data->get_reservation_id(); |
| my $computer_node_name = $self->data->get_computer_node_name(); |
| |
| my %username_hash; |
| |
| my $reservation_users = $self->data->get_reservation_users(); |
| foreach my $user_id (sort keys %$reservation_users) { |
| my $username = $reservation_users->{$user_id}{unityid}; |
| $username_hash{$username} = $user_id; |
| } |
| |
| # Collect users in reservationaccounts table |
| my $reservation_accounts = get_reservation_accounts($reservation_id); |
| foreach my $user_id (sort keys %$reservation_accounts) { |
| my $username = $reservation_accounts->{$user_id}{username}; |
| $username_hash{$username} = $user_id; |
| } |
| |
| my $error_encountered = 0; |
| |
| # Delete users |
| foreach my $username (sort keys %username_hash) { |
| my $user_id = $username_hash{$username}; |
| |
| # Delete user on the OS |
| if (!$self->delete_user($username)) { |
| $error_encountered = 1; |
| notify($ERRORS{'WARNING'}, 0, "failed to delete user on $computer_node_name"); |
| } |
| |
| # Delete entry to the useraccounts table |
| delete_reservation_account($reservation_id, $user_id); |
| } |
| |
| return !$error_encountered; |
| } |
| |
| #////////////////////////////////////////////////////////////////////////////// |
| |
| =head2 get_source_configuration_directories |
| |
| Parameters : None |
| Returns : Array containing filesystem path strings |
| Description : Retrieves the $SOURCE_CONFIGURATION_DIRECTORY variable value for |
| the classes which the OS object is a member of and returns an |
| array containing these values. |
| |
| The first element of the array contains the value from the |
| top-most class where the $SOURCE_CONFIGURATION_DIRECTORY variable |
| was defined. The last element contains the value from the |
| bottom-most class, which is probably the class which was |
| instantiated. |
| |
| Example: An Windows XP OS object is instantiated from the XP |
| class, which is a subclass of the Version_5 class, which is a |
| subclass of the Windows class: |
| |
| VCL::Module::OS::Windows |
| ^ |
| VCL::Module::OS::Windows::Version_5 |
| ^ |
| VCL::Module::OS::Windows::Version_5::XP |
| |
| The XP and Windows classes each |
| have a $SOURCE_CONFIGURATION_DIRECTORY variable defined but the |
| Version_5 class does not. The array returned will be: |
| |
| [0] = '/usr/local/vcldev/current/bin/../tools/Windows' |
| [1] = '/usr/local/vcldev/current/bin/../tools/Windows_XP' |
| |
| =cut |
| |
| sub get_source_configuration_directories { |
| my $self = shift; |
| unless (ref($self) && $self->isa('VCL::Module')) { |
| notify($ERRORS{'CRITICAL'}, 0, "subroutine can only be called as a VCL module object method"); |
| return; |
| } |
| |
| # Get an array containing the names of the Perl packages the OS object is a class of |
| my @package_hierarchy = $self->get_package_hierarchy(); |
| |
| # Loop through each classes, retrieve any which have a $SOURCE_CONFIGURATION_DIRECTORY variable defined |
| my @directories = (); |
| for my $package_name (@package_hierarchy) { |
| my $source_configuration_directory = eval '$' . $package_name . '::SOURCE_CONFIGURATION_DIRECTORY'; |
| if ($EVAL_ERROR) { |
| notify($ERRORS{'WARNING'}, 0, "unable to determine source configuration directory for $package_name, error:\n$EVAL_ERROR"); |
| next; |
| } |
| elsif (!$source_configuration_directory) { |
| notify($ERRORS{'DEBUG'}, 0, "source configuration directory is not defined for $package_name"); |
| next; |
| } |
| |
| notify($ERRORS{'DEBUG'}, 0, "package source configuration directory: $source_configuration_directory"); |
| |
| # Add the directory path to the return array |
| # Use unshift to add to the beginning to the array |
| unshift @directories, $source_configuration_directory; |
| } |
| |
| return @directories; |
| } |
| |
| #////////////////////////////////////////////////////////////////////////////// |
| |
| =head2 create_currentimage_txt |
| |
| Parameters : None |
| Returns : boolean |
| Description : Creates the currentimage.txt file on the computer. |
| |
| =cut |
| |
| sub create_currentimage_txt { |
| my $self = shift; |
| if (ref($self) !~ /VCL::Module/i) { |
| notify($ERRORS{'CRITICAL'}, 0, "subroutine was called as a function, it must be called as a class method"); |
| return; |
| } |
| |
| my $image_id = $self->data->get_image_id(); |
| my $image_name = $self->data->get_image_name(); |
| my $image_prettyname = $self->data->get_image_prettyname(); |
| my $imagerevision_id = $self->data->get_imagerevision_id(); |
| my $imagerevision_date_created = $self->data->get_imagerevision_date_created(); |
| my $computer_id = $self->data->get_computer_id(); |
| my $computer_host_name = $self->data->get_computer_host_name(); |
| |
| my $file_contents = <<EOF; |
| $image_name |
| id=$image_id |
| prettyname=$image_prettyname |
| imagerevision_id=$imagerevision_id |
| imagerevision_datecreated=$imagerevision_date_created |
| computer_id=$computer_id |
| computer_hostname=$computer_host_name |
| EOF |
| |
| # Create the file |
| if ($self->create_text_file('~/currentimage.txt', $file_contents)) { |
| return 1; |
| } |
| else { |
| notify($ERRORS{'WARNING'}, 0, "failed to create currentimage.txt file on $computer_host_name"); |
| return; |
| } |
| } |
| |
| #////////////////////////////////////////////////////////////////////////////// |
| |
| =head2 get_currentimage_txt_contents |
| |
| Parameters : None |
| Returns : If successful: array |
| If failed: false |
| Description : Reads the currentimage.txt file on a computer and returns its |
| contents as an array. Each array element represents a line in |
| the file. |
| |
| =cut |
| |
| sub get_currentimage_txt_contents { |
| my $self = shift; |
| if (ref($self) !~ /VCL::Module/i) { |
| notify($ERRORS{'CRITICAL'}, 0, "subroutine was called as a function, it must be called as a class method"); |
| return; |
| } |
| |
| my $computer_node_name = $self->data->get_computer_node_name(); |
| |
| # Attempt to retrieve the contents of currentimage.txt |
| my $cat_command = "cat ~/currentimage.txt"; |
| my ($cat_exit_status, $cat_output) = $self->execute($cat_command,1); |
| if (!defined($cat_output)) { |
| notify($ERRORS{'WARNING'}, 0, "failed to execute command to failed to retrieve currentimage.txt from $computer_node_name"); |
| return; |
| } |
| elsif ($cat_exit_status ne '0') { |
| notify($ERRORS{'WARNING'}, 0, "failed to retrieve currentimage.txt from $computer_node_name, exit status: $cat_exit_status, output:\n@{$cat_output}"); |
| return; |
| } |
| else { |
| notify($ERRORS{'DEBUG'}, 0, "retrieved currentimage.txt contents from $computer_node_name:\n" . join("\n", @$cat_output)); |
| } |
| |
| my %output; |
| my @current_image_txt_contents = @{$cat_output}; |
| |
| my $current_image_name; |
| if (defined $current_image_txt_contents[0]) { |
| $output{"current_image_name"} = $current_image_txt_contents[0]; |
| } |
| |
| foreach my $l (@current_image_txt_contents) { |
| #remove any line break characters |
| $l =~ s/[\r\n]*//g; |
| my ($a, $b) = split(/=/, $l); |
| if (defined $b) { |
| $output{$a} = $b; |
| } |
| } |
| |
| return %output; |
| } ## end sub get_currentimage_txt_contents |
| |
| #////////////////////////////////////////////////////////////////////////////// |
| |
| =head2 get_current_imagerevision_id |
| |
| Parameters : none |
| Returns : integer |
| Description : Retrieves the imagerevision ID value from currentimage.txt. |
| |
| =cut |
| |
| sub get_current_imagerevision_id { |
| my $self = shift; |
| if (ref($self) !~ /VCL::Module/i) { |
| notify($ERRORS{'CRITICAL'}, 0, "subroutine was called as a function, it must be called as a class method"); |
| return; |
| } |
| |
| return $self->get_current_image_tag('imagerevision_id'); |
| } |
| |
| #////////////////////////////////////////////////////////////////////////////// |
| |
| =head2 wait_for_reboot |
| |
| Parameters : $total_wait_seconds, $attempt_delay_seconds, $attempt_limit |
| Returns : boolean |
| Description : Waits for the computer to become unresponsive, respond to ping, |
| then respond to SSH. |
| |
| =cut |
| |
| sub wait_for_reboot { |
| my $self = shift; |
| if (ref($self) !~ /VCL::Module/i) { |
| notify($ERRORS{'CRITICAL'}, 0, "subroutine was called as a function, it must be called as a class method"); |
| return; |
| } |
| |
| my $computer_node_name = $self->data->get_computer_node_name(); |
| |
| # Attempt to get the total number of seconds to wait from the arguments |
| my $total_wait_seconds_argument = shift; |
| if (!defined($total_wait_seconds_argument) || $total_wait_seconds_argument !~ /^\d+$/) { |
| $total_wait_seconds_argument = 300; |
| } |
| |
| # Seconds to wait in between loop attempts |
| my $attempt_delay_seconds_argument = shift; |
| if (!defined($attempt_delay_seconds_argument) || $attempt_delay_seconds_argument !~ /^\d+$/) { |
| $attempt_delay_seconds_argument = 15; |
| } |
| |
| # Number of power reset attempts to make if reboot fails |
| my $attempt_limit = shift; |
| if (!defined($attempt_limit) || $attempt_limit !~ /^\d+$/) { |
| $attempt_limit = 2; |
| } |
| elsif (!$attempt_limit) { |
| $attempt_limit = 1; |
| } |
| |
| ATTEMPT: |
| for (my $attempt = 1; $attempt <= $attempt_limit; $attempt++) { |
| my $total_wait_seconds = $total_wait_seconds_argument; |
| my $attempt_delay_seconds = $attempt_delay_seconds_argument; |
| |
| if ($attempt > 1) { |
| # Computer did not become responsive on previous attempt |
| notify($ERRORS{'OK'}, 0, "$computer_node_name reboot failed to complete on previous attempt, attempting hard power reset"); |
| |
| # Call provisioning module's power_reset() subroutine |
| if ($self->provisioner->power_reset()) { |
| notify($ERRORS{'OK'}, 0, "reboot attempt $attempt/$attempt_limit: initiated power reset on $computer_node_name"); |
| } |
| else { |
| notify($ERRORS{'WARNING'}, 0, "reboot failed, failed to initiate power reset on $computer_node_name"); |
| return 0; |
| } |
| |
| # Add 2 minutes for each attempt to $total_wait_seconds in case argument supplied wasn't long enough |
| $total_wait_seconds += (120 * $attempt); |
| } |
| |
| my $start_time = time; |
| |
| notify($ERRORS{'DEBUG'}, 0, "waiting for $computer_node_name to reboot: |
| attempt: $attempt/$attempt_limit |
| maximum wait time: $total_wait_seconds seconds |
| wait delay: $attempt_delay_seconds"); |
| |
| # Wait for the computer to become unresponsive to ping |
| if (!$self->wait_for_no_ping($total_wait_seconds, 5)) { |
| # Computer never stopped responding to ping |
| notify($ERRORS{'WARNING'}, 0, "$computer_node_name never became unresponsive to ping"); |
| next ATTEMPT; |
| } |
| |
| # Decrease $total_wait_seconds by the amount of time elapsed so far |
| my $no_ping_elapsed_seconds = (time - $start_time); |
| $total_wait_seconds -= $no_ping_elapsed_seconds; |
| |
| # Computer is unresponsive, reboot has begun |
| # Wait 5 seconds before beginning to check if computer is back online |
| notify($ERRORS{'DEBUG'}, 0, "$computer_node_name reboot has begun, sleeping for 5 seconds"); |
| sleep 5; |
| |
| # Wait for the computer to respond to ping |
| if (!$self->wait_for_ping($total_wait_seconds, $attempt_delay_seconds)) { |
| # Check if the computer was ever offline, it should have been or else reboot never happened |
| notify($ERRORS{'WARNING'}, 0, "$computer_node_name never responded to ping"); |
| next ATTEMPT; |
| } |
| |
| # Decrease $total_wait_seconds by the amount of time elapsed so far |
| my $ping_elapsed_seconds = (time - $start_time); |
| my $ping_actual_seconds = ($ping_elapsed_seconds - $no_ping_elapsed_seconds); |
| $total_wait_seconds -= $ping_elapsed_seconds; |
| |
| notify($ERRORS{'DEBUG'}, 0, "$computer_node_name is pingable, waiting for SSH to respond"); |
| |
| # Wait maximum of 3 minutes for ssh to respond |
| if (!$self->wait_for_ssh($total_wait_seconds, $attempt_delay_seconds)) { |
| notify($ERRORS{'WARNING'}, 0, "$computer_node_name never responded to SSH"); |
| next ATTEMPT; |
| } |
| |
| # Decrease $total_wait_seconds by the amount of time elapsed so far |
| my $ssh_elapsed_seconds = (time - $start_time); |
| my $ssh_actual_seconds = ($ssh_elapsed_seconds - $ping_elapsed_seconds); |
| |
| notify($ERRORS{'OK'}, 0, "$computer_node_name responded to SSH: |
| unresponsive: $no_ping_elapsed_seconds seconds |
| respond to ping: $ping_elapsed_seconds seconds ($ping_actual_seconds seconds after unresponsive) |
| respond to SSH $ssh_elapsed_seconds seconds ($ssh_actual_seconds seconds after ping)" |
| ); |
| return 1; |
| } |
| |
| # If loop completed, maximum number of reboot attempts was reached |
| notify($ERRORS{'WARNING'}, 0, "$computer_node_name reboot failed"); |
| return 0; |
| } |
| |
| #////////////////////////////////////////////////////////////////////////////// |
| |
| =head2 wait_for_ping |
| |
| Parameters : Maximum number of seconds to wait (optional), delay between attempts (optional) |
| Returns : If computer is pingable before the maximum amount of time has elapsed: 1 |
| If computer never responds to ping before the maximum amount of time has elapsed: 0 |
| Description : Attempts to ping the computer specified in the DataStructure |
| for the current reservation. It will wait up to a maximum number |
| of seconds. This can be specified by passing the subroutine an |
| integer value or the default value of 300 seconds will be used. The |
| delay between attempts can be specified as the 2nd argument in |
| seconds. The default value is 15 seconds. |
| |
| =cut |
| |
| sub wait_for_ping { |
| my $self = shift; |
| if (ref($self) !~ /VCL::Module/i) { |
| notify($ERRORS{'CRITICAL'}, 0, "subroutine was called as a function, it must be called as a class method"); |
| return; |
| } |
| |
| # Attempt to get the total number of seconds to wait from the arguments |
| my $total_wait_seconds = shift; |
| if (!defined($total_wait_seconds) || $total_wait_seconds !~ /^\d+$/) { |
| $total_wait_seconds = 300; |
| } |
| |
| # Seconds to wait in between loop attempts |
| my $attempt_delay_seconds = shift; |
| if (!defined($attempt_delay_seconds) || $attempt_delay_seconds !~ /^\d+$/) { |
| $attempt_delay_seconds = 15; |
| } |
| |
| my $computer_node_name = $self->data->get_computer_node_name(); |
| |
| my $message = "waiting for $computer_node_name to respond to ping"; |
| |
| # Call code_loop_timeout, specifify that it should call _pingnode with the computer name as the argument |
| return $self->code_loop_timeout(\&_pingnode, [$computer_node_name], $message, $total_wait_seconds, $attempt_delay_seconds); |
| } ## end sub wait_for_ping |
| |
| #////////////////////////////////////////////////////////////////////////////// |
| |
| =head2 wait_for_no_ping |
| |
| Parameters : Maximum number of seconds to wait (optional), seconds to delay between attempts (optional) |
| Returns : 1 if computer is not pingable, 0 otherwise |
| Description : Attempts to ping the computer specified in the DataStructure |
| for the current reservation. It will wait up to a maximum number |
| of seconds for ping to fail. The delay between attempts can be |
| specified as the 2nd argument in seconds. The default value is 15 |
| seconds. |
| |
| =cut |
| |
| sub wait_for_no_ping { |
| my $self = shift; |
| if (ref($self) !~ /VCL::Module/i) { |
| notify($ERRORS{'CRITICAL'}, 0, "subroutine was called as a function, it must be called as a class method"); |
| return; |
| } |
| |
| # Attempt to get the total number of seconds to wait from the arguments |
| my $total_wait_seconds = shift; |
| if (!defined($total_wait_seconds) || $total_wait_seconds !~ /^\d+$/) { |
| $total_wait_seconds = 300; |
| } |
| |
| # Seconds to wait in between loop attempts |
| my $attempt_delay_seconds = shift; |
| if (!defined($attempt_delay_seconds) || $attempt_delay_seconds !~ /^\d+$/) { |
| $attempt_delay_seconds = 15; |
| } |
| |
| my $computer_node_name = $self->data->get_computer_node_name(); |
| |
| my $message = "waiting for $computer_node_name to NOT respond to ping"; |
| |
| # Call code_loop_timeout and invert the result, specifify that it should call _pingnode with the computer name as the argument |
| return $self->code_loop_timeout(sub{return !_pingnode(@_)}, [$computer_node_name], $message, $total_wait_seconds, $attempt_delay_seconds); |
| } ## end sub wait_for_no_ping |
| |
| #////////////////////////////////////////////////////////////////////////////// |
| |
| =head2 wait_for_ssh |
| |
| Parameters : Seconds to wait (optional), seconds to delay between attempts (optional) |
| Returns : |
| Description : Attempts to communicate to the reservation computer via SSH. |
| SSH attempts are made until the maximum number of seconds has |
| elapsed. The maximum number of seconds can be specified as the |
| first argument. If an argument isn't supplied, a default value of |
| 300 seconds will be used. |
| |
| A delay occurs between attempts. This can be specified by passing |
| a 2nd argument. If a 2nd argument isn't supplied, a default value |
| of 15 seconds will be used. |
| |
| =cut |
| |
| sub wait_for_ssh { |
| my $self = shift; |
| if (ref($self) !~ /VCL::Module/i) { |
| notify($ERRORS{'CRITICAL'}, 0, "subroutine was called as a function, it must be called as a class method"); |
| return; |
| } |
| |
| # Attempt to get the total number of seconds to wait from the arguments |
| my $total_wait_seconds = shift; |
| if (!defined($total_wait_seconds) || $total_wait_seconds !~ /^\d+$/) { |
| $total_wait_seconds = 300; |
| } |
| |
| # Seconds to wait in between loop attempts |
| my $attempt_delay_seconds = shift; |
| if (!defined($attempt_delay_seconds) || $attempt_delay_seconds !~ /^\d+$/) { |
| $attempt_delay_seconds = 15; |
| } |
| |
| my $computer_node_name = $self->data->get_computer_node_name(); |
| |
| # Call the "can" function, it returns a code reference to the subroutine specified |
| # This is passed to code_loop_timeout which will then execute the code until it returns true |
| my $sub_ref = $self->can("is_ssh_responding"); |
| |
| my $message = "waiting for $computer_node_name to respond to SSH"; |
| |
| return $self->code_loop_timeout($sub_ref, [$self], $message, $total_wait_seconds, $attempt_delay_seconds); |
| } |
| |
| #////////////////////////////////////////////////////////////////////////////// |
| |
| =head2 is_ssh_responding |
| |
| Parameters : $computer_name (optional), $max_attempts (optional) |
| Returns : boolean |
| Description : Checks if the computer is responding to SSH. The SSH port is |
| first checked. If not open, 0 is returned. If the port is open a |
| test SSH command which simply echo's a string is attempted. The |
| default is to only attempt to run this command once. This can be |
| changed by supplying the $max_attempts argument. If the |
| $max_attempts is supplied but set to 0, only the port checks are |
| done. |
| |
| =cut |
| |
| sub is_ssh_responding { |
| my $self = shift; |
| if (ref($self) !~ /VCL::Module/i) { |
| notify($ERRORS{'CRITICAL'}, 0, "subroutine was called as a function, it must be called as a class method"); |
| return; |
| } |
| |
| my $computer_node_name; |
| my $max_attempts = 1; |
| |
| my $argument_1 = shift; |
| my $argument_2 = shift; |
| if ($argument_1) { |
| # Check if the argument is an integer |
| if ($argument_1 =~ /^\d+$/) { |
| $max_attempts = $argument_1; |
| } |
| else { |
| $computer_node_name = $argument_1; |
| if ($argument_2 && $argument_2 =~ /^\d+$/) { |
| $max_attempts = $argument_2; |
| } |
| } |
| } |
| |
| if (!$computer_node_name) { |
| $computer_node_name = $self->data->get_computer_node_name(); |
| } |
| |
| # If 'ssh_port' key is set in this object use it |
| my $port = $self->{ssh_port} || 22; |
| |
| # Try nmap to see if any of the ssh ports are open before attempting to run a test command |
| my $nmap_status = nmap_port($computer_node_name, $port) ? "open" : "closed"; |
| if ($nmap_status ne 'open') { |
| notify($ERRORS{'DEBUG'}, 0, "$computer_node_name is NOT responding to SSH, port $port is closed"); |
| return 0; |
| } |
| |
| if ($max_attempts) { |
| # Run a test SSH command |
| #my ($exit_status, $output) = $self->execute({ |
| # node => $computer_node_name, |
| # command => "echo \"testing ssh on $computer_node_name\"", |
| # max_attempts => $max_attempts, |
| # display_output => 0, |
| # timeout_seconds => 30, |
| # ignore_error => 1, |
| #}); |
| |
| my ($exit_status, $output) = $self->execute({ |
| node => $computer_node_name, |
| command => "echo \"testing ssh on $computer_node_name\"", |
| max_attempts => $max_attempts, |
| display_output => 0, |
| timeout_seconds => 15, |
| }); |
| |
| # The exit status will be 0 if the command succeeded |
| if (defined($output) && grep(/testing/, @$output)) { |
| notify($ERRORS{'DEBUG'}, 0, "$computer_node_name is responding to SSH, port $port: $nmap_status"); |
| return 1; |
| } |
| else { |
| notify($ERRORS{'DEBUG'}, 0, "$computer_node_name is NOT responding to SSH, SSH command failed, port $port: $nmap_status"); |
| return 0; |
| } |
| } |
| else { |
| return 1; |
| } |
| } |
| |
| #////////////////////////////////////////////////////////////////////////////// |
| |
| =head2 wait_for_response |
| |
| Parameters : Initial delay seconds (optional), SSH response timeout seconds (optional), SSH attempt delay seconds (optional) |
| Returns : If successful: true |
| If failed: false |
| Description : Waits for the reservation computer to respond to SSH after it |
| has been loaded. |
| |
| =cut |
| |
| sub wait_for_response { |
| my $self = shift; |
| if (ref($self) !~ /VCL::Module/i) { |
| notify($ERRORS{'CRITICAL'}, 0, "subroutine was called as a function, it must be called as a class method"); |
| return; |
| } |
| |
| my $start_time = time(); |
| |
| my $reservation_id = $self->data->get_reservation_id(); |
| my $computer_id = $self->data->get_computer_id(); |
| my $computer_node_name = $self->data->get_computer_node_name(); |
| |
| my $initial_delay_seconds = shift; |
| if (!defined $initial_delay_seconds) { |
| $initial_delay_seconds = 120; |
| } |
| |
| my $ssh_response_timeout_seconds = shift; |
| if (!defined $ssh_response_timeout_seconds) { |
| $ssh_response_timeout_seconds = 600; |
| } |
| |
| my $ssh_attempt_delay_seconds = shift; |
| if (!defined $ssh_attempt_delay_seconds) { |
| $ssh_attempt_delay_seconds = 15; |
| } |
| |
| # Sleep for the initial delay value if it has been set |
| # Check SSH once to bypass the initial delay if SSH is already responding |
| if ($initial_delay_seconds && !$self->is_ssh_responding()) { |
| notify($ERRORS{'OK'}, 0, "waiting $initial_delay_seconds seconds for $computer_node_name to boot"); |
| sleep $initial_delay_seconds; |
| notify($ERRORS{'OK'}, 0, "waited $initial_delay_seconds seconds for $computer_node_name to boot"); |
| } |
| |
| # Wait for SSH to respond, loop until timeout is reached |
| if (!$self->wait_for_ssh($ssh_response_timeout_seconds, $ssh_attempt_delay_seconds)) { |
| notify($ERRORS{'WARNING'}, 0, "failed to connect to $computer_node_name via SSH after $ssh_response_timeout_seconds seconds"); |
| return; |
| } |
| |
| my $end_time = time(); |
| my $duration = ($end_time - $start_time); |
| |
| insertloadlog($reservation_id, $computer_id, "machinebooted", "$computer_node_name is responding to SSH after $duration seconds"); |
| notify($ERRORS{'OK'}, 0, "$computer_node_name is responding to SSH after $duration seconds"); |
| return 1; |
| } |
| |
| #////////////////////////////////////////////////////////////////////////////// |
| |
| =head2 wait_for_port_open |
| |
| Parameters : $port_number, $connection_target (optional), $total_wait_seconds (optional), $attempt_delay_seconds (optional) |
| Returns : boolean |
| Description : Uses nmap to check if the port specified is open. Loops until the |
| port is open or $total_wait_seconds is reached. |
| |
| =cut |
| |
| sub wait_for_port_open { |
| my $self = shift; |
| if (ref($self) !~ /VCL::Module/i) { |
| notify($ERRORS{'CRITICAL'}, 0, "subroutine was called as a function, it must be called as a class method"); |
| return; |
| } |
| |
| my $computer_node_name = $self->data->get_computer_node_name(); |
| my $calling_subroutine = get_calling_subroutine(); |
| |
| my $port_number = shift; |
| if (!defined($port_number)) { |
| notify($ERRORS{'WARNING'}, 0, "port number argument was not supplied"); |
| return; |
| } |
| |
| my $connection_target = shift || $computer_node_name; |
| my $total_wait_seconds = shift || 60; |
| my $attempt_delay_seconds = shift || 5; |
| |
| my $mode = ($calling_subroutine =~ /wait_for_port_closed/ ? 'closed' : 'open'); |
| |
| my $message = "waiting for port $port_number to be $mode on $connection_target"; |
| $message .= " ($computer_node_name)" if ($connection_target ne $computer_node_name); |
| |
| # Essentially perform xnor on nmap_port result and $mode eq open |
| # Both either need to be true or false |
| # $mode eq open:true, nmap_port:true, result:true |
| # $mode eq open:false, nmap_port:true, result:false |
| # $mode eq open:true, nmap_port:false, result:false |
| # $mode eq open:false, nmap_port:false, result:true |
| my $sub_ref = sub{ |
| my $nmap_result = nmap_port(@_) || 0; |
| return $mode =~ /open/ == $nmap_result; |
| }; |
| return $self->code_loop_timeout($sub_ref, [$connection_target, $port_number], $message, $total_wait_seconds, $attempt_delay_seconds); |
| } |
| |
| #////////////////////////////////////////////////////////////////////////////// |
| |
| =head2 wait_for_port_closed |
| |
| Parameters : $port_number, $connection_target (optional), $total_wait_seconds (optional), $attempt_delay_seconds (optional) |
| Returns : boolean |
| Description : Uses nmap to check if the port specified is closed. Loops until |
| the port is open or $total_wait_seconds is reached. |
| |
| =cut |
| |
| sub wait_for_port_closed { |
| my $self = shift; |
| if (ref($self) !~ /VCL::Module/i) { |
| notify($ERRORS{'CRITICAL'}, 0, "subroutine was called as a function, it must be called as a class method"); |
| return; |
| } |
| return wait_for_port_open($self, @_); |
| } |
| |
| #////////////////////////////////////////////////////////////////////////////// |
| |
| =head2 update_ssh_known_hosts |
| |
| Parameters : $known_hosts_path (optional) |
| Returns : boolean |
| Description : Removes lines from the known_hosts file matching the computer |
| name or private IP address, then runs ssh-keyscan to add the |
| current keys to the known_hosts file. |
| |
| =cut |
| |
| sub update_ssh_known_hosts { |
| my $self = shift; |
| if (ref($self) !~ /VCL::Module/i) { |
| notify($ERRORS{'CRITICAL'}, 0, "subroutine was called as a function, it must be called as a class method"); |
| return; |
| } |
| |
| my $known_hosts_path = shift || "/root/.ssh/known_hosts"; |
| |
| my $computer_short_name = $self->data->get_computer_short_name(); |
| |
| # Get the computer private IP address |
| my $computer_private_ip_address; |
| if ($self->can("get_private_ip_address") && ($computer_private_ip_address = $self->get_private_ip_address())) { |
| notify($ERRORS{'DEBUG'}, 0, "retrieved private IP address for $computer_short_name using OS module: $computer_private_ip_address"); |
| } |
| elsif ($computer_private_ip_address = $self->data->get_computer_private_ip_address()) { |
| notify($ERRORS{'DEBUG'}, 0, "retrieved private IP address for $computer_short_name from database: $computer_private_ip_address"); |
| } |
| else { |
| notify($ERRORS{'WARNING'}, 0, "unable to retrieve private IP address for $computer_short_name using OS module or from database"); |
| } |
| |
| # Open the file, read the contents into an array, then close it |
| my @known_hosts_lines_original; |
| if (open FILE, "<", $known_hosts_path) { |
| @known_hosts_lines_original = <FILE>; |
| close FILE; |
| } |
| else { |
| notify($ERRORS{'WARNING'}, 0, "unable to open file for reading: $known_hosts_path"); |
| return; |
| } |
| |
| |
| # Loop through the lines |
| my @known_hosts_lines_modified; |
| for my $line (@known_hosts_lines_original) { |
| chomp $line; |
| next if (!$line); |
| |
| # Check if line matches the computer name or private IP address |
| if ($line =~ /(^|[\s,])$computer_short_name[\s,]/i) { |
| # Don't add the line to the array which will be added back to the file |
| notify($ERRORS{'DEBUG'}, 0, "removing line from $known_hosts_path matching computer name: $computer_short_name\n$line"); |
| next; |
| } |
| elsif ($line =~ /(^|[\s,])$computer_private_ip_address[\s,]/i) { |
| notify($ERRORS{'DEBUG'}, 0, "removing line from $known_hosts_path matching computer private IP address:$computer_private_ip_address\n$line"); |
| next; |
| } |
| |
| # Line doesn't match, add it to the array of lines for the new file |
| push @known_hosts_lines_modified, "$line\n"; |
| } |
| |
| |
| # Write the modified contents to the file |
| if (open FILE, ">", "$known_hosts_path") { |
| print FILE @known_hosts_lines_modified; |
| close FILE; |
| notify($ERRORS{'DEBUG'}, 0, "removed lines from $known_hosts_path matching $computer_short_name or $computer_private_ip_address"); |
| } |
| else { |
| notify($ERRORS{'WARNING'}, 0, "unable to open file for writing: $known_hosts_path"); |
| return; |
| } |
| |
| # Run ssh-keyscan |
| run_command("ssh-keyscan -t rsa '$computer_short_name' '$computer_private_ip_address' 2>&1 | grep -v '^#' >> $known_hosts_path"); |
| |
| return 1; |
| } |
| |
| |
| #////////////////////////////////////////////////////////////////////////////// |
| |
| =head2 server_request_set_fixed_ip |
| |
| Parameters : none |
| Returns : If successful: true |
| If failed: false |
| Description : |
| |
| =cut |
| |
| sub server_request_set_fixed_ip { |
| my $self = shift; |
| if (ref($self) !~ /VCL::Module/i) { |
| notify($ERRORS{'CRITICAL'}, 0, "subroutine was called as a function, it must be called as a class method"); |
| return; |
| } |
| |
| my $reservation_id = $self->data->get_reservation_id() || return; |
| my $computer_id = $self->data->get_computer_id() || return; |
| my $computer_node_name = $self->data->get_computer_node_name() || return; |
| my $image_os_name = $self->data->get_image_os_name() || return; |
| my $image_os_type = $self->data->get_image_os_type() || return; |
| my $computer_public_ip_address = $self->data->get_computer_public_ip_address(); |
| my $public_ip_configuration = $self->data->get_management_node_public_ip_configuration() || return; |
| my $server_request_id = $self->data->get_server_request_id(); |
| my $server_request_fixed_ip = $self->data->get_server_request_fixed_ip(); |
| |
| if ($server_request_id) { |
| if ($server_request_fixed_ip) { |
| #Update the info related to fixedIP |
| if (!$self->update_fixed_ip_info()) { |
| notify($ERRORS{'WARNING'}, 0, "Unable to update information related fixedIP for server_request $server_request_id"); |
| } |
| |
| #Confirm requested IP is not being used |
| if (!$self->confirm_fixed_ip_is_available()) { |
| #failed, insert into loadlog, fail reservation |
| insertloadlog($reservation_id, $computer_id, "failed","$server_request_fixed_ip is NOT available"); |
| return 0; |
| } |
| |
| #if set for static IPs, save the old address to restore |
| if ($public_ip_configuration =~ /static/i) { |
| notify($ERRORS{'DEBUG'}, 0, "saving original IP for restore on post reseration"); |
| my $original_IPvalue = "originalIPaddr_" . $server_request_id; |
| set_variable($original_IPvalue, $computer_public_ip_address); |
| } |
| |
| # Try to set the static public IP address using the OS module |
| if ($self->can("set_static_public_address")) { |
| if ($self->set_static_public_address()) { |
| notify($ERRORS{'DEBUG'}, 0, "set static public IP address on $computer_node_name using OS module's set_static_public_address() method"); |
| $self->data->set_computer_public_ip_address($server_request_fixed_ip); |
| |
| # Delete cached network configuration information so it is retrieved next time it is needed |
| delete $self->{network_configuration}; |
| |
| if (update_computer_public_ip_address($computer_id, $server_request_fixed_ip)) { |
| notify($ERRORS{'OK'}, 0, "updated public IP address in computer table for $computer_node_name, $server_request_fixed_ip"); |
| } |
| |
| #Update Hostname to match Public assigned name |
| if ($self->can("update_public_hostname")) { |
| if ($self->update_public_hostname()) { |
| notify($ERRORS{'OK'}, 0, "Updated hostname based on fixedIP $server_request_fixed_ip"); |
| } |
| } |
| } |
| else { |
| notify($ERRORS{'WARNING'}, 0, "failed to set static public IP address on $computer_node_name"); |
| insertloadlog($reservation_id, $computer_id, "failed"," Not able to assigne IPaddress $server_request_fixed_ip"); |
| return 0; |
| } |
| } |
| else { |
| notify($ERRORS{'WARNING'}, 0, "unable to set static public IP address on $computer_node_name, " . ref($self) . " module does not implement a set_static_public_address subroutine"); |
| } |
| } |
| } |
| |
| return 1; |
| |
| } |
| |
| |
| #////////////////////////////////////////////////////////////////////////////// |
| |
| =head2 confirm_fixed_ip_is_available |
| |
| Parameters : none |
| Returns : If successful: true |
| If failed: 0 |
| Description : Preforms checks to confirm the requested IP is not being used |
| -- Check VCL database computer table for IP |
| -- try to ping the IP |
| -- future; good to check with upstream network switch or control |
| |
| =cut |
| |
| sub confirm_fixed_ip_is_available { |
| my $self = shift; |
| if (ref($self) !~ /VCL::Module/i) { |
| notify($ERRORS{'CRITICAL'}, 0, "subroutine was called as a function, it must be called as a class method"); |
| return; |
| } |
| |
| my $reservation_id = $self->data->get_reservation_id() || return; |
| my $computer_id = $self->data->get_computer_id() || return; |
| my $computer_node_name = $self->data->get_computer_node_name() || return; |
| my $server_request_id = $self->data->get_server_request_id(); |
| my $server_request_fixed_ip = $self->data->get_server_request_fixed_ip(); |
| |
| #check VCL computer table |
| if (is_ip_assigned_query($server_request_fixed_ip, $computer_id)) { |
| notify($ERRORS{'WARNING'}, 0, "$server_request_fixed_ip is already assigned"); |
| insertloadlog($reservation_id, $computer_id, "failed","$server_request_fixed_ip is already assigned"); |
| return 0; |
| } |
| |
| #check if the assigned computer has the specified address |
| my $retrieved_public_ip_address = $self->get_public_ip_address(); |
| if ($retrieved_public_ip_address eq $server_request_fixed_ip) { |
| notify($ERRORS{'DEBUG'}, 0, "$server_request_fixed_ip is assigned to reserved computer, skipping ping test"); |
| return 1; |
| } |
| |
| #Is IP pingable |
| if (_pingnode($server_request_fixed_ip)) { |
| notify($ERRORS{'WARNING'}, 0, "$server_request_fixed_ip is answering ping test"); |
| insertloadlog($reservation_id, $computer_id, "failed","$server_request_fixed_ip is answering ping test, but is not assigned in VCL database"); |
| return 0; |
| } |
| |
| return 1; |
| } |
| |
| #////////////////////////////////////////////////////////////////////////////// |
| |
| =head2 update_public_ip_address |
| |
| Parameters : none |
| Returns : If successful: true |
| If failed: false |
| Description : Checks the IP configuration mode for the management node - |
| dynamic DHCP, manual DHCP, or static. If DHCP is used, the |
| public IP address is retrieved from the computer and the IP |
| address in the computer table is updated if necessary. If |
| static public IP addresses are used, the computer is configured |
| to use the public IP address stored in the computer table. |
| |
| =cut |
| |
| sub update_public_ip_address { |
| my $self = shift; |
| if (ref($self) !~ /VCL::Module/i) { |
| notify($ERRORS{'CRITICAL'}, 0, "subroutine was called as a function, it must be called as a class method"); |
| return; |
| } |
| |
| my $reservation_id = $self->data->get_reservation_id() || return; |
| my $computer_id = $self->data->get_computer_id() || return; |
| my $computer_node_name = $self->data->get_computer_node_name() || return; |
| my $image_os_name = $self->data->get_image_os_name() || return; |
| my $image_os_type = $self->data->get_image_os_type() || return; |
| my $computer_public_ip_address = $self->data->get_computer_public_ip_address(); |
| my $public_ip_configuration = $self->data->get_management_node_public_ip_configuration() || return; |
| my $nathost_hostname = $self->data->get_nathost_hostname(0); |
| my $nathost_internal_ip_address = $self->data->get_nathost_internal_ip_address(0); |
| |
| if ($public_ip_configuration =~ /dhcp/i) { |
| notify($ERRORS{'DEBUG'}, 0, "IP configuration is set to $public_ip_configuration, attempting to retrieve dynamic public IP address from $computer_node_name"); |
| |
| # Wait for the computer to be assigned a public IP address |
| # There is sometimes a delay |
| if (!$self->wait_for_public_ip_address()) { |
| notify($ERRORS{'WARNING'}, 0, "unable to update public IP address, $computer_node_name did not receive a public IP address via DHCP"); |
| return; |
| } |
| |
| # Retrieve the public IP address from the OS module |
| my $retrieved_public_ip_address = $self->get_public_ip_address(); |
| if ($retrieved_public_ip_address) { |
| notify($ERRORS{'DEBUG'}, 0, "retrieved public IP address from $computer_node_name using the OS module: $retrieved_public_ip_address"); |
| } |
| else { |
| notify($ERRORS{'WARNING'}, 0, "failed to retrieve dynamic public IP address from $computer_node_name"); |
| insertloadlog($reservation_id, $computer_id, "dynamicDHCPaddress", "failed to retrieve dynamic public IP address from $computer_node_name"); |
| return; |
| } |
| |
| # Update the Datastructure and computer table if the retrieved IP address does not match what is in the database |
| if ($computer_public_ip_address ne $retrieved_public_ip_address) { |
| $self->data->set_computer_public_ip_address($retrieved_public_ip_address); |
| |
| if (update_computer_public_ip_address($computer_id, $retrieved_public_ip_address)) { |
| notify($ERRORS{'OK'}, 0, "updated dynamic public IP address in computer table for $computer_node_name, $retrieved_public_ip_address"); |
| insertloadlog($reservation_id, $computer_id, "dynamicDHCPaddress", "updated dynamic public IP address in computer table for $computer_node_name, $retrieved_public_ip_address"); |
| } |
| else { |
| notify($ERRORS{'WARNING'}, 0, "failed to update dynamic public IP address in computer table for $computer_node_name, $retrieved_public_ip_address"); |
| insertloadlog($reservation_id, $computer_id, "dynamicDHCPaddress", "failed to update dynamic public IP address in computer table for $computer_node_name, $retrieved_public_ip_address"); |
| return; |
| } |
| } |
| else { |
| notify($ERRORS{'DEBUG'}, 0, "public IP address in computer table is already correct for $computer_node_name: $computer_public_ip_address"); |
| } |
| |
| # If the computer is assigned to a NAT host, make sure default gateway is correct |
| if ($nathost_hostname && $nathost_internal_ip_address) { |
| my $current_default_gateway = $self->get_default_gateway(); |
| if ($current_default_gateway) { |
| if ($current_default_gateway eq $nathost_internal_ip_address) { |
| notify($ERRORS{'DEBUG'}, 0, "static default gateway does NOT need to be set on $computer_node_name, default gateway assigned by DHCP matches NAT host internal IP address: $current_default_gateway"); |
| } |
| else { |
| notify($ERRORS{'OK'}, 0, "static default gateway needs to be set on $computer_node_name, default gateway assigned by DHCP ($current_default_gateway) does NOT match NAT host internal IP address: $nathost_internal_ip_address"); |
| $self->set_static_default_gateway(); |
| } |
| } |
| } |
| } |
| elsif ($public_ip_configuration =~ /static/i) { |
| notify($ERRORS{'DEBUG'}, 0, "IP configuration is set to $public_ip_configuration, attempting to set public IP address"); |
| |
| # Try to set the static public IP address using the OS module |
| if ($self->can("set_static_public_address")) { |
| if ($self->set_static_public_address()) { |
| notify($ERRORS{'DEBUG'}, 0, "set static public IP address on $computer_node_name using OS module's set_static_public_address() method"); |
| insertloadlog($reservation_id, $computer_id, "staticIPaddress", "set static public IP address on $computer_node_name"); |
| } |
| else { |
| notify($ERRORS{'WARNING'}, 0, "failed to set static public IP address on $computer_node_name"); |
| insertloadlog($reservation_id, $computer_id, "staticIPaddress", "failed to set static public IP address on $computer_node_name"); |
| return; |
| } |
| } |
| else { |
| notify($ERRORS{'WARNING'}, 0, "unable to set static public IP address on $computer_node_name, " . ref($self) . " module does not implement a set_static_public_address subroutine"); |
| } |
| } |
| else { |
| notify($ERRORS{'DEBUG'}, 0, "IP configuration is set to $public_ip_configuration, no public IP address updates necessary"); |
| } |
| |
| return 1; |
| } |
| |
| #////////////////////////////////////////////////////////////////////////////// |
| |
| =head2 get_correct_default_gateway |
| |
| Parameters : none |
| Returns : boolean |
| Description : Determines which IP address should be used for the computer's |
| default gateway: |
| * If the computer is configured to use a NAT host, the computer's |
| default gateway should be set to the NAT host's internal IP |
| address |
| * If NAT is not used and the management node profile is |
| configured to use static public IP addresses, the computer's |
| default gateway should be set to the management node profile's |
| "Public Gateway" setting |
| * If NAT is not used and the management node profile is |
| configured to use DHCP-assigned public IP addresses, the |
| computer's current default gateway should continue to be used |
| |
| =cut |
| |
| sub get_correct_default_gateway { |
| my $self = shift; |
| if (ref($self) !~ /VCL::Module::OS/i) { |
| notify($ERRORS{'CRITICAL'}, 0, "subroutine was called as a function, it must be called as a class method"); |
| return 0; |
| } |
| |
| my $computer_name = $self->data->get_computer_short_name(); |
| my $nathost_hostname = $self->data->get_nathost_hostname(0); |
| my $nathost_internal_ip_address = $self->data->get_nathost_internal_ip_address(0); |
| my $management_node_ip_configuration = $self->data->get_management_node_public_ip_configuration(); |
| |
| if ($nathost_internal_ip_address) { |
| notify($ERRORS{'DEBUG'}, 0, "$computer_name is configured to use NAT host $nathost_hostname, default gateway should be set to NAT host's internal IP address: $nathost_internal_ip_address"); |
| return $nathost_internal_ip_address; |
| } |
| elsif ($management_node_ip_configuration =~ /static/i) { |
| my $management_node_public_default_gateway = $self->data->get_management_node_public_default_gateway(); |
| if ($management_node_public_default_gateway) { |
| notify($ERRORS{'DEBUG'}, 0, "management node public IP mode is set to $management_node_ip_configuration, default gateway should be set to management node profile's default gateway setting: $management_node_public_default_gateway"); |
| return $management_node_public_default_gateway; |
| } |
| else { |
| notify($ERRORS{'WARNING'}, 0, "unable to determine correct default gateway to use on $computer_name, management node public IP mode is set to $management_node_ip_configuration but management node profile's default gateway setting could not be determined"); |
| return; |
| } |
| } |
| else { |
| # Management node configured to use DHCP for public IP addresses |
| # Get default gateway address assigned to computer |
| my $current_default_gateway = $self->get_public_default_gateway(); |
| if ($current_default_gateway) { |
| notify($ERRORS{'DEBUG'}, 0, "management node public IP mode is set to $management_node_ip_configuration, default gateway currently configured on $computer_name should be used: $current_default_gateway"); |
| return $current_default_gateway; |
| } |
| else { |
| notify($ERRORS{'WARNING'}, 0, "unable to determine correct default gateway to use on $computer_name, management node public IP mode is set to $management_node_ip_configuration but default gateway currently configured on $computer_name could not be determined"); |
| return; |
| } |
| } |
| } |
| |
| #////////////////////////////////////////////////////////////////////////////// |
| |
| =head2 get_current_image_tag |
| |
| Parameters : $tag_name |
| Returns : string |
| Description : Reads currentimage.txt and attempts to locate a line beginning |
| with the tag name specified by the argument. |
| If found, the tag value following the = sign is returned. |
| Null is returned if a line with the tag name doesn't exist. |
| |
| =cut |
| |
| sub get_current_image_tag { |
| my $self = shift; |
| if (ref($self) !~ /VCL::Module/i) { |
| notify($ERRORS{'CRITICAL'}, 0, "subroutine was called as a function, it must be called as a class method"); |
| return; |
| } |
| |
| my ($tag_name) = @_; |
| if (!defined($tag_name)) { |
| notify($ERRORS{'WARNING'}, 0, "property argument was not specified"); |
| return; |
| } |
| |
| my $computer_name = $self->data->get_computer_short_name(); |
| |
| my $current_image_file_path = 'currentimage.txt'; |
| |
| my @lines = $self->get_file_contents($current_image_file_path); |
| for my $line (@lines) { |
| my ($tag_value) = $line =~ /^$tag_name=(.*)$/gx; |
| if (defined($tag_value)) { |
| $tag_value = '' unless length($tag_value); |
| notify($ERRORS{'DEBUG'}, 0, "found '$tag_name' tag line in $current_image_file_path: '$line', returning tag value: '$tag_value'"); |
| return $tag_value; |
| } |
| } |
| |
| notify($ERRORS{'DEBUG'}, 0, "'$tag_name' tag is not set in $current_image_file_path, returning null"); |
| return; |
| } |
| |
| #////////////////////////////////////////////////////////////////////////////// |
| |
| =head2 set_current_image_tag |
| |
| Parameters : $tag_name, $tag_value |
| Returns : boolean |
| Description : Adds a line to currentimage.txt in the format: |
| <tag name>=<tag value> (timestamp) |
| |
| The tag value must be a non-empty string and may be the integer |
| 0. Example: |
| set_current_image_tag('mytag'); |
| |
| Line added: |
| mytag=0 (Wed Jun 29 17:47:36 2016) |
| |
| Any lines which already exist beginning with an identical tag |
| name are removed. |
| |
| indicating a loaded computer is |
| tainted and must be reloaded for any subsequent reservations. The |
| format of the line added is: |
| vcld_tainted=true (<time>) |
| |
| This line is added as a safety measure to prevent a computer |
| which was used for one reservation to ever be reserved for |
| another reservation without being reloaded. |
| |
| This line should be added whenever a user has been given the |
| connection information and had a chance to connect. It's assumed |
| a user connected and the computer is tainted whether or not an |
| actual logged in connection was detected. This is done for |
| safety. The connection/logged in checking mechanisms of different |
| OS's may not be perfect. |
| |
| This line is checked when a computer is reserved to make sure the |
| post_load tasks have run. A computer may be loaded but the |
| post_load tasks may not run if it is loaded manually or by some |
| other means not controlled by vcld. |
| |
| =cut |
| |
| sub set_current_image_tag { |
| my $self = shift; |
| if (ref($self) !~ /VCL::Module/i) { |
| notify($ERRORS{'CRITICAL'}, 0, "subroutine was called as a function, it must be called as a class method"); |
| return; |
| } |
| |
| my ($tag_name, $tag_value) = @_; |
| if (!$tag_name) { |
| notify($ERRORS{'WARNING'}, 0, "tag name argument was not specified"); |
| return; |
| } |
| elsif (!defined($tag_value)) { |
| notify($ERRORS{'WARNING'}, 0, "tag value argument was not specified"); |
| return; |
| } |
| elsif (!length($tag_value)) { |
| notify($ERRORS{'WARNING'}, 0, "tag value argument specified is an empty string"); |
| return; |
| } |
| |
| my $computer_name = $self->data->get_computer_short_name(); |
| |
| my $current_image_file_path = 'currentimage.txt'; |
| my $timestamp = makedatestring(); |
| my $tag_line = "$tag_name=$tag_value ($timestamp)"; |
| my $updated_contents = ''; |
| |
| my @existing_lines = $self->get_file_contents($current_image_file_path); |
| for my $existing_line (@existing_lines) { |
| # Skip blank lines and lines matching the tag name |
| if ($existing_line !~ /\w/ || $existing_line =~ /^$tag_name=/) { |
| next; |
| } |
| $updated_contents .= "$existing_line\n"; |
| } |
| $updated_contents .= $tag_line; |
| |
| if ($self->create_text_file($current_image_file_path, $updated_contents)) { |
| notify($ERRORS{'DEBUG'}, 0, "set '$tag_name' tag in $current_image_file_path on $computer_name:\n$updated_contents"); |
| return 1; |
| } |
| else { |
| notify($ERRORS{'WARNING'}, 0, "failed to set '$tag_name' tag in $current_image_file_path on $computer_name"); |
| return 0; |
| } |
| } |
| |
| #////////////////////////////////////////////////////////////////////////////// |
| |
| =head2 set_post_load_status |
| |
| Parameters : none |
| Returns : boolean |
| Description : Adds a line to currentimage.txt indicating the post-load tasks |
| have successfully been completed on a loaded computer. The |
| format of the line is: |
| tainted=true (Wed Jun 29 18:00:55 2016) |
| |
| =cut |
| |
| sub set_post_load_status { |
| my $self = shift; |
| if (ref($self) !~ /VCL::Module/i) { |
| notify($ERRORS{'CRITICAL'}, 0, "subroutine was called as a function, it must be called as a class method"); |
| return; |
| } |
| |
| return $self->set_current_image_tag('vcld_post_load', 'success'); |
| } |
| |
| #////////////////////////////////////////////////////////////////////////////// |
| |
| =head2 get_post_load_status |
| |
| Parameters : none |
| Returns : boolean |
| Description : Checks if a 'vcld_post_load' line exists in currentimage.txt. |
| |
| =cut |
| |
| sub get_post_load_status { |
| my $self = shift; |
| if (ref($self) !~ /VCL::Module/i) { |
| notify($ERRORS{'CRITICAL'}, 0, "subroutine was called as a function, it must be called as a class method"); |
| return; |
| } |
| |
| my $computer_name = $self->data->get_computer_short_name(); |
| |
| my ($post_load_status) = $self->get_current_image_tag('vcld_post_load'); |
| if (defined($post_load_status)) { |
| notify($ERRORS{'DEBUG'}, 0, "post-load tasks have been completed on $computer_name: $post_load_status"); |
| return 1; |
| } |
| else { |
| notify($ERRORS{'DEBUG'}, 0, "post-load tasks have NOT been completed on $computer_name"); |
| return 0; |
| } |
| } |
| |
| #////////////////////////////////////////////////////////////////////////////// |
| |
| =head2 set_tainted_status |
| |
| Parameters : $reason (optional) |
| Returns : boolean |
| Description : Adds a line to currentimage.txt indicating a loaded computer is |
| tainted and must be reloaded for any subsequent reservations. |
| |
| This line is added as a safety measure to prevent a computer |
| which was used for one reservation to ever be reserved for |
| another reservation without being reloaded. |
| |
| This line should be added whenever a user has been given the |
| connection information and had a chance to connect. It's assumed |
| a user connected and the computer is tainted whether or not an |
| actual logged in connection was detected. This is done for |
| safety. The connection/logged in checking mechanisms of different |
| OS's may not be perfect. |
| |
| This line is checked when a computer is reserved to make sure the |
| post_load tasks have run. A computer may be loaded but the |
| post_load tasks may not run if it is loaded manually or by some |
| other means not controlled by vcld. |
| |
| =cut |
| |
| sub set_tainted_status { |
| my $self = shift; |
| if (ref($self) !~ /VCL::Module/i) { |
| notify($ERRORS{'CRITICAL'}, 0, "subroutine was called as a function, it must be called as a class method"); |
| return; |
| } |
| |
| my $reason = shift || 'no reason given'; |
| |
| # This may be called multiple times for a reservation |
| # It's useful to append concatenate the status rather than overwriting it so you know all of the reasons a computer may be tainted |
| my $previous_tained_status = $self->get_tainted_status(); |
| if ($previous_tained_status) { |
| $reason = "$previous_tained_status, $reason"; |
| } |
| |
| return $self->set_current_image_tag('vcld_tainted', $reason); |
| } |
| |
| #////////////////////////////////////////////////////////////////////////////// |
| |
| =head2 get_tainted_status |
| |
| Parameters : none |
| Returns : string |
| Description : Checks if a line exists in currentimage.txt indicated a user may |
| have logged in. |
| |
| =cut |
| |
| sub get_tainted_status { |
| my $self = shift; |
| if (ref($self) !~ /VCL::Module/i) { |
| notify($ERRORS{'CRITICAL'}, 0, "subroutine was called as a function, it must be called as a class method"); |
| return; |
| } |
| |
| my $computer_name = $self->data->get_computer_short_name(); |
| |
| my $tainted_status = $self->get_current_image_tag('vcld_tainted'); |
| if (defined($tainted_status)) { |
| notify($ERRORS{'DEBUG'}, 0, "image currently loaded on $computer_name has been tainted: $tainted_status"); |
| return $tainted_status; |
| } |
| else { |
| notify($ERRORS{'DEBUG'}, 0, "image currently loaded on $computer_name has NOT been tainted"); |
| return 0; |
| } |
| } |
| |
| #////////////////////////////////////////////////////////////////////////////// |
| |
| =head2 get_private_interface_name |
| |
| Parameters : none |
| Returns : string |
| Description : Determines the private interface name based on the information in |
| the network configuration hash returned by |
| get_network_configuration. The interface which is assigned the |
| private IP address for the reservation computer is returned. |
| |
| =cut |
| |
| sub get_private_interface_name { |
| my $self = shift; |
| if (ref($self) !~ /VCL::Module/i) { |
| notify($ERRORS{'CRITICAL'}, 0, "subroutine was called as a function, it must be called as a class method"); |
| return; |
| } |
| |
| return $self->{private_interface_name} if defined $self->{private_interface_name}; |
| |
| # Get the network configuration hash reference |
| my $network_configuration = $self->get_network_configuration(); |
| if (!$network_configuration) { |
| notify($ERRORS{'WARNING'}, 0, "unable to determine private interface name, failed to retrieve network configuration"); |
| return; |
| } |
| |
| # Get the computer private IP address |
| my $computer_private_ip_address = $self->data->get_computer_private_ip_address(); |
| if (!$computer_private_ip_address) { |
| notify($ERRORS{'DEBUG'}, 0, "unable to retrieve computer private IP address from reservation data"); |
| return; |
| } |
| |
| # Loop through all of the network interfaces found |
| foreach my $interface_name (sort keys %$network_configuration) { |
| # Get the interface IP addresses and make sure an IP address was found |
| my @ip_addresses = keys %{$network_configuration->{$interface_name}{ip_address}}; |
| if (!@ip_addresses) { |
| notify($ERRORS{'DEBUG'}, 0, "interface is not assigned an IP address: $interface_name"); |
| next; |
| } |
| |
| # Check if interface has the private IP address assigned to it |
| if (grep { $_ eq $computer_private_ip_address } @ip_addresses) { |
| $self->{private_interface_name} = $interface_name; |
| notify($ERRORS{'DEBUG'}, 0, "determined private interface name: $self->{private_interface_name} (" . join (", ", @ip_addresses) . ")"); |
| return $self->{private_interface_name}; |
| } |
| } |
| |
| notify($ERRORS{'WARNING'}, 0, "failed to determine private interface name, no interface is assigned the private IP address for the reservation: $computer_private_ip_address\n" . format_data($network_configuration)); |
| return; |
| } |
| |
| #////////////////////////////////////////////////////////////////////////////// |
| |
| =head2 get_public_interface_name |
| |
| Parameters : $no_cache (optional), $ignore_error (optional) |
| Returns : string |
| Description : Determines the public interface name based on the information in |
| the network configuration hash returned by |
| get_network_configuration. |
| |
| =cut |
| |
| sub get_public_interface_name { |
| my $self = shift; |
| if (ref($self) !~ /VCL::Module/i) { |
| notify($ERRORS{'CRITICAL'}, 0, "subroutine was called as a function, it must be called as a class method"); |
| return; |
| } |
| |
| my $no_cache = shift || 0; |
| my $ignore_error = shift || 0; |
| |
| notify($ERRORS{'DEBUG'}, 0, "attempting to determine public interface name, no cache: $no_cache, ignore error: $ignore_error"); |
| |
| if ($no_cache) { |
| delete $self->{public_interface_name}; |
| } |
| elsif ($self->{public_interface_name}) { |
| #notify($ERRORS{'DEBUG'}, 0, "returning public interface name previously retrieved: $self->{public_interface_name}"); |
| return $self->{public_interface_name}; |
| } |
| |
| # Get the network configuration hash reference |
| my $network_configuration = $self->get_network_configuration($no_cache); |
| if (!$network_configuration) { |
| notify($ERRORS{'WARNING'}, 0, "unable to determine public interface name, failed to retrieve network configuration"); |
| return; |
| } |
| |
| # Get the computer private IP address |
| my $computer_private_ip_address = $self->data->get_computer_private_ip_address(); |
| if (!$computer_private_ip_address) { |
| notify($ERRORS{'DEBUG'}, 0, "unable to retrieve computer private IP address from reservation data"); |
| return; |
| } |
| |
| my $public_interface_name; |
| |
| # Loop through all of the network interfaces found |
| INTERFACE: for my $check_interface_name (sort keys %$network_configuration) { |
| |
| my $description = $network_configuration->{$check_interface_name}{description} || ''; |
| my $master = $network_configuration->{$check_interface_name}{master}; |
| |
| # Check if the interface should be ignored based on the name or description |
| if ($check_interface_name =~ /^(lo|sit\d)$/i) { |
| notify($ERRORS{'DEBUG'}, 0, "interface '$check_interface_name' ignored because its name is '$1'"); |
| next INTERFACE; |
| } |
| elsif ($check_interface_name =~ /(loopback|vmnet|afs|tunnel|6to4|isatap|teredo)/i) { |
| notify($ERRORS{'DEBUG'}, 0, "interface '$check_interface_name' ignored because its name contains '$1'"); |
| next INTERFACE; |
| } |
| elsif ($description =~ /(loopback|afs|tunnel|pseudo|6to4|isatap)/i) { |
| notify($ERRORS{'DEBUG'}, 0, "interface '$check_interface_name' ignored because its description contains '$1'"); |
| next INTERFACE; |
| } |
| elsif ($master) { |
| notify($ERRORS{'DEBUG'}, 0, "interface '$check_interface_name' ignored because it is a slave to $master"); |
| next INTERFACE; |
| } |
| |
| # If $public_interface_name hasn't been set yet, set it and continue checking the next interface |
| if (!$public_interface_name) { |
| my @check_ip_addresses = keys %{$network_configuration->{$check_interface_name}{ip_address}}; |
| my $matches_private = (grep { $_ eq $computer_private_ip_address } @check_ip_addresses) ? 1 : 0; |
| |
| if ($matches_private) { |
| if (scalar(@check_ip_addresses) == 1) { |
| notify($ERRORS{'DEBUG'}, 0, "'$check_interface_name' could not be the public interface, it is only assigned the private IP address"); |
| next INTERFACE; |
| } |
| |
| notify($ERRORS{'DEBUG'}, 0, "'$check_interface_name' is assigned private IP address, checking if other assigned IP addresses could potentially be public"); |
| CHECK_IP_ADDRESS: for my $check_ip_address (@check_ip_addresses) { |
| |
| if ($check_ip_address eq $computer_private_ip_address) { |
| notify($ERRORS{'DEBUG'}, 0, "ignoring private IP address ($check_ip_address) assigned to interface '$check_interface_name'"); |
| next CHECK_IP_ADDRESS; |
| } |
| elsif ($check_ip_address =~ /^(169\.254|0\.0\.0\.0)/) { |
| notify($ERRORS{'DEBUG'}, 0, "ignoring invalid IP address ($check_ip_address) assigned to interface '$check_interface_name'"); |
| next CHECK_IP_ADDRESS; |
| } |
| else { |
| notify($ERRORS{'DEBUG'}, 0, "'$check_interface_name' could potententially be public interface, assigned IP address: $check_ip_address"); |
| $public_interface_name = $check_interface_name; |
| last CHECK_IP_ADDRESS; |
| } |
| } |
| } |
| else { |
| # Does not match private IP address |
| notify($ERRORS{'DEBUG'}, 0, "'$check_interface_name' could potentially be public interface, not assigned private IP address"); |
| $public_interface_name = $check_interface_name; |
| } |
| |
| next INTERFACE; |
| } |
| |
| # Call the helper subroutine |
| # It uses recursion to avoid large/duplicated if-else blocks |
| $public_interface_name = $self->_get_public_interface_name_helper($check_interface_name, $public_interface_name); |
| if (!$public_interface_name) { |
| notify($ERRORS{'WARNING'}, 0, "failed to determine if '$check_interface_name' or '$public_interface_name' is more likely the public interface"); |
| next INTERFACE; |
| } |
| } |
| |
| if ($public_interface_name) { |
| $self->{public_interface_name} = $public_interface_name; |
| notify($ERRORS{'OK'}, 0, "determined the public interface name: '$self->{public_interface_name}'"); |
| return $self->{public_interface_name}; |
| } |
| else { |
| notify($ERRORS{'WARNING'}, 0, "failed to determine the public interface name:\n" . format_data($network_configuration)) unless $ignore_error; |
| return; |
| } |
| } |
| |
| #////////////////////////////////////////////////////////////////////////////// |
| |
| =head2 _get_public_interface_name_helper |
| |
| Parameters : $interface_name_1, $interface_name_2 |
| Returns : string |
| Description : Compares the network configuration of the interfaces passed as |
| the arguments. Returns the name of the interface more likely to |
| be the public interface. It checks the following: |
| 1. Is either interface assigned a public IP address? |
| - If only 1 interface is assigned a public IP address then that interface name is returned. |
| - If neither or both are assigned a public IP address: |
| 2. Is either interface assigned a default gateway? |
| - If only 1 interface is assigned a default gateway then that interface name is returned. |
| - If neither or both are assigned a default gateway: |
| 3. Is either interface assigned the private IP address? |
| - If only 1 interface is assigned the private IP address, then the other interface name is returned. |
| - If neither or both are assigned the private IP address, the first interface argument is returned |
| |
| =cut |
| |
| sub _get_public_interface_name_helper { |
| my $self = shift; |
| if (ref($self) !~ /VCL::Module/i) { |
| notify($ERRORS{'CRITICAL'}, 0, "subroutine was called as a function, it must be called as a class method"); |
| return; |
| } |
| |
| my ($interface_name_1, $interface_name_2, $condition) = @_; |
| |
| if (!$interface_name_1 || !$interface_name_2) { |
| notify($ERRORS{'WARNING'}, 0, "\$network_configuration, \$interface_name_1, and \$interface_name_2 arguments were not specified"); |
| return; |
| } |
| |
| my $network_configuration = $self->get_network_configuration(); |
| my @ip_addresses_1 = keys %{$network_configuration->{$interface_name_1}{ip_address}}; |
| my @ip_addresses_2 = keys %{$network_configuration->{$interface_name_2}{ip_address}}; |
| |
| if (!$condition || $condition eq 'matches_private') { |
| # Get the computer private IP address |
| my $computer_private_ip_address = $self->data->get_computer_private_ip_address(); |
| if (!$computer_private_ip_address) { |
| notify($ERRORS{'DEBUG'}, 0, "unable to retrieve computer private IP address from reservation data"); |
| return; |
| } |
| |
| my $matches_private_1 = (grep { $_ eq $computer_private_ip_address } @ip_addresses_1) ? 1 : 0; |
| my $matches_private_2 = (grep { $_ eq $computer_private_ip_address } @ip_addresses_2) ? 1 : 0; |
| |
| if ($matches_private_1 eq $matches_private_2) { |
| notify($ERRORS{'DEBUG'}, 0, "tie: both interfaces are/are not assigned the private IP address: $computer_private_ip_address, proceeding to check if either interface is assigned a public IP address"); |
| return $self->_get_public_interface_name_helper($interface_name_1, $interface_name_2, 'assigned_public'); |
| } |
| elsif ($matches_private_1) { |
| notify($ERRORS{'DEBUG'}, 0, "'$interface_name_2' is more likely the public interface, it is NOT assigned the private IP address: $computer_private_ip_address"); |
| return $interface_name_2; |
| } |
| else { |
| notify($ERRORS{'DEBUG'}, 0, "'$interface_name_1' is more likely the public interface, it is NOT assigned the private IP address: $computer_private_ip_address"); |
| return $interface_name_1; |
| } |
| } |
| elsif ($condition eq 'assigned_public') { |
| my $assigned_public_1 = (grep { is_public_ip_address($_) } @ip_addresses_1) ? 1 : 0; |
| my $assigned_public_2 = (grep { is_public_ip_address($_) } @ip_addresses_2) ? 1 : 0; |
| |
| if ($assigned_public_1 eq $assigned_public_2) { |
| notify($ERRORS{'DEBUG'}, 0, "tie: both interfaces are/are not assigned public IP addresses, proceeding to check default gateways"); |
| return $self->_get_public_interface_name_helper($interface_name_1, $interface_name_2, 'assigned_gateway'); |
| } |
| elsif ($assigned_public_1) { |
| notify($ERRORS{'DEBUG'}, 0, "'$interface_name_1' is more likely the public interface, it is assigned a public IP address, '$interface_name_2' is not"); |
| return $interface_name_1; |
| } |
| else { |
| notify($ERRORS{'DEBUG'}, 0, "'$interface_name_2' is more likely the public interface, it is assigned a public IP address, '$interface_name_1' is not"); |
| return $interface_name_2; |
| } |
| } |
| elsif ($condition eq 'assigned_gateway') { |
| my $assigned_default_gateway_1 = defined($network_configuration->{$interface_name_1}{default_gateway}) ? 1 : 0; |
| my $assigned_default_gateway_2 = defined($network_configuration->{$interface_name_2}{default_gateway}) ? 1 : 0; |
| |
| if ($assigned_default_gateway_1 eq $assigned_default_gateway_2) { |
| notify($ERRORS{'DEBUG'}, 0, "tie: both interfaces are/are not assigned a default gateway, returning '$interface_name_2'"); |
| return $interface_name_2; |
| } |
| elsif ($assigned_default_gateway_1) { |
| notify($ERRORS{'DEBUG'}, 0, "'$interface_name_1' is more likely the public interface, it is assigned a default gateway, '$interface_name_2' is not"); |
| return $interface_name_1; |
| } |
| else { |
| notify($ERRORS{'DEBUG'}, 0, "'$interface_name_2' is more likely the public interface, it is assigned a default gateway, '$interface_name_1' is not"); |
| return $interface_name_2; |
| } |
| } |
| else { |
| notify($ERRORS{'WARNING'}, 0, "unable to determine which interface is more likely the public interface, invalid \$condition argument: '$condition'"); |
| return; |
| } |
| } |
| |
| #////////////////////////////////////////////////////////////////////////////// |
| |
| =head2 wait_for_public_ip_address |
| |
| Parameters : $total_wait_seconds (optional), $attempt_delay_seconds (optional) |
| Returns : boolean |
| Description : Loops until the computer's public interface name can be retrieved |
| or the timeout is reached. The public interface name can only be |
| retrieved if an IP address is assigned to it. The default maximum |
| wait time is 60 seconds. |
| |
| =cut |
| |
| sub wait_for_public_ip_address { |
| my $self = shift; |
| if (ref($self) !~ /VCL::Module/i) { |
| notify($ERRORS{'CRITICAL'}, 0, "subroutine was called as a function, it must be called as a class method"); |
| return; |
| } |
| |
| # Attempt to get the total number of seconds to wait from the arguments |
| my $total_wait_seconds = shift; |
| if (!defined($total_wait_seconds) || $total_wait_seconds !~ /^\d+$/) { |
| $total_wait_seconds = 60; |
| } |
| |
| # Seconds to wait in between loop attempts |
| my $attempt_delay_seconds = shift; |
| if (!defined($attempt_delay_seconds) || $attempt_delay_seconds !~ /^\d+$/) { |
| $attempt_delay_seconds = 2; |
| } |
| |
| my $computer_node_name = $self->data->get_computer_node_name(); |
| |
| # Check if the public IP address was already retrieved |
| # Use cached data if available (0), ignore errors (1) |
| my $public_ip_address = $self->get_public_ip_address(0, 1); |
| if ($public_ip_address) { |
| notify($ERRORS{'DEBUG'}, 0, "$computer_node_name is already assigned a public IP address: $public_ip_address"); |
| return 1; |
| } |
| |
| my $sub_ref = $self->can("get_public_ip_address"); |
| notify($ERRORS{'DEBUG'}, 0, ""); |
| my $message = "waiting for $computer_node_name to get public IP address"; |
| |
| return $self->code_loop_timeout( |
| sub { |
| my $public_ip_address = $self->get_public_ip_address(1, 1); |
| if (!defined($public_ip_address)) { |
| return; |
| } |
| else { |
| notify($ERRORS{'DEBUG'}, 0, "$computer_node_name was assigned a public IP address: $public_ip_address"); |
| return 1; |
| } |
| }, |
| [$self, 1], |
| $message, |
| $total_wait_seconds, |
| $attempt_delay_seconds |
| ); |
| } |
| |
| #////////////////////////////////////////////////////////////////////////////// |
| |
| =head2 get_private_network_configuration |
| |
| Parameters : none |
| Returns : |
| Description : |
| |
| =cut |
| |
| sub get_private_network_configuration { |
| my $self = shift; |
| if (ref($self) !~ /VCL::Module/i) { |
| notify($ERRORS{'CRITICAL'}, 0, "subroutine was called as a function, it must be called as a class method"); |
| return; |
| } |
| |
| my $private_interface_name = $self->get_private_interface_name(); |
| if (!$private_interface_name) { |
| notify($ERRORS{'WARNING'}, 0, "unable to retrieve private network configuration, private interface name could not be determined"); |
| return; |
| } |
| |
| return $self->get_network_configuration()->{$private_interface_name}; |
| } |
| |
| #////////////////////////////////////////////////////////////////////////////// |
| |
| =head2 get_public_network_configuration |
| |
| Parameters : $no_cache (optional), $ignore_error (optional) |
| Returns : hash reference |
| Description : |
| |
| =cut |
| |
| sub get_public_network_configuration { |
| my $self = shift; |
| if (ref($self) !~ /VCL::Module/i) { |
| notify($ERRORS{'CRITICAL'}, 0, "subroutine was called as a function, it must be called as a class method"); |
| return; |
| } |
| |
| my $no_cache = shift || 0; |
| my $ignore_error = shift || 0; |
| |
| notify($ERRORS{'DEBUG'}, 0, "attempting to retrieve public network configuration, no cache: $no_cache, ignore error: $ignore_error"); |
| |
| my $public_interface_name = $self->get_public_interface_name($no_cache, $ignore_error); |
| if (!$public_interface_name) { |
| notify($ERRORS{'WARNING'}, 0, "unable to retrieve public network configuration, public interface name could not be determined") unless $ignore_error; |
| return; |
| } |
| |
| return $self->get_network_configuration()->{$public_interface_name}; |
| } |
| |
| #////////////////////////////////////////////////////////////////////////////// |
| |
| =head2 get_mac_address |
| |
| Parameters : |
| Returns : |
| Description : |
| |
| =cut |
| |
| sub get_mac_address { |
| my $self = shift; |
| if (ref($self) !~ /VCL::Module/i) { |
| notify($ERRORS{'CRITICAL'}, 0, "subroutine was called as a function, it must be called as a class method"); |
| return; |
| } |
| |
| # Check if a 'public' or 'private' network type argument was specified |
| # Assume 'public' if not specified |
| my $network_type = shift || 'public'; |
| $network_type = lc($network_type) if $network_type; |
| if ($network_type && $network_type !~ /(public|private)/i) { |
| notify($ERRORS{'WARNING'}, 0, "network type argument can only be 'public' or 'private'"); |
| return; |
| } |
| |
| # Get the public or private network configuration |
| # Use 'eval' to construct the appropriate subroutine name |
| my $network_configuration = eval "\$self->get_$network_type\_network_configuration()"; |
| if ($EVAL_ERROR || !$network_configuration) { |
| notify($ERRORS{'WARNING'}, 0, "unable to retrieve $network_type network configuration"); |
| return; |
| } |
| |
| my $mac_address = $network_configuration->{physical_address}; |
| if ($mac_address) { |
| notify($ERRORS{'DEBUG'}, 0, "returning $network_type MAC address: $mac_address"); |
| return $mac_address; |
| } |
| else { |
| notify($ERRORS{'WARNING'}, 0, "unable to determine $network_type MAC address, 'physical_address' key does not exist in the network configuration info: \n" . format_data($network_configuration)); |
| return; |
| } |
| } |
| |
| #////////////////////////////////////////////////////////////////////////////// |
| |
| =head2 get_private_mac_address |
| |
| Parameters : |
| Returns : |
| Description : |
| |
| =cut |
| |
| sub get_private_mac_address { |
| my $self = shift; |
| if (ref($self) !~ /VCL::Module/i) { |
| notify($ERRORS{'CRITICAL'}, 0, "subroutine was called as a function, it must be called as a class method"); |
| return; |
| } |
| |
| return $self->get_mac_address('private'); |
| } |
| |
| |
| #////////////////////////////////////////////////////////////////////////////// |
| |
| =head2 get_public_mac_address |
| |
| Parameters : |
| Returns : |
| Description : |
| |
| =cut |
| |
| sub get_public_mac_address { |
| my $self = shift; |
| if (ref($self) !~ /VCL::Module/i) { |
| notify($ERRORS{'CRITICAL'}, 0, "subroutine was called as a function, it must be called as a class method"); |
| return; |
| } |
| |
| return $self->get_mac_address('public'); |
| } |
| |
| #////////////////////////////////////////////////////////////////////////////// |
| |
| =head2 get_ip_address |
| |
| Parameters : $network_type (optional), $no_cache (optional), $ignore_error (optional) |
| Returns : string |
| Description : Returns the IP address of the computer. The $network_type |
| argument may either be 'public' or 'private'. If not supplied, |
| the default is to return the public IP address. |
| |
| =cut |
| |
| sub get_ip_address { |
| my $self = shift; |
| if (ref($self) !~ /VCL::Module/i) { |
| notify($ERRORS{'CRITICAL'}, 0, "subroutine was called as a function, it must be called as a class method"); |
| return; |
| } |
| |
| # Check if a 'public' or 'private' network type argument was specified |
| # Assume 'public' if not specified |
| my $network_type = shift || 'public'; |
| $network_type = lc($network_type) if $network_type; |
| if ($network_type && $network_type !~ /(public|private)/i) { |
| notify($ERRORS{'WARNING'}, 0, "network type argument can only be 'public' or 'private'"); |
| return; |
| } |
| |
| my $no_cache = shift || 0; |
| my $ignore_error = shift || 0; |
| |
| notify($ERRORS{'DEBUG'}, 0, "attempting to retrieve IP address, type: $network_type, no cache: $no_cache, ignore error: $ignore_error"); |
| |
| # Get the public or private network configuration |
| # Use 'eval' to construct the appropriate subroutine name |
| my $network_configuration = eval "\$self->get_$network_type\_network_configuration($no_cache, $ignore_error)"; |
| if ($EVAL_ERROR || !$network_configuration) { |
| notify($ERRORS{'WARNING'}, 0, "unable to retrieve $network_type network configuration") unless $ignore_error; |
| return; |
| } |
| |
| my $ip_address_info = $network_configuration->{ip_address}; |
| if (!defined($ip_address_info)) { |
| notify($ERRORS{'WARNING'}, 0, "$network_type network configuration info does not contain an 'ip_address' key") unless $ignore_error; |
| return; |
| } |
| |
| # Return the first valid IP address found |
| my $ip_address; |
| my @ip_addresses = keys %$ip_address_info; |
| if (!@ip_addresses) { |
| if (!$ignore_error) { |
| notify($ERRORS{'WARNING'}, 0, "unable to determine $network_type IP address, 'ip_address' value is not set in the network configuration info: \n" . format_data($network_configuration)); |
| } |
| return; |
| } |
| |
| # Interface has multiple IP addresses, try to find a valid one |
| for $ip_address (@ip_addresses) { |
| if ($ip_address !~ /(0\.0\.0\.0|169\.254\.)/) { |
| #notify($ERRORS{'DEBUG'}, 0, "returning $network_type IP address: $ip_address"); |
| return $ip_address; |
| } |
| else { |
| notify($ERRORS{'DEBUG'}, 0, "skipping invalid IP address assigned to $network_type interface: $ip_address, checking if another valid IP address is assigned"); |
| } |
| } |
| |
| notify($ERRORS{'WARNING'}, 0, "$network_type interface not assigned a valid IP address: " . join(", ", @ip_addresses)); |
| return; |
| } |
| |
| #////////////////////////////////////////////////////////////////////////////// |
| |
| =head2 get_ip_addresses |
| |
| Parameters : $no_cache (optional) |
| Returns : array |
| Description : Returns all of the IP addresses of the computer. |
| |
| =cut |
| |
| sub get_ip_addresses { |
| my ($self, $no_cache) = @_; |
| if (ref($self) !~ /VCL::Module/i) { |
| notify($ERRORS{'CRITICAL'}, 0, "subroutine was called as a function, it must be called as a class method"); |
| return; |
| } |
| |
| my $computer_name = $self->data->get_computer_short_name(); |
| |
| # Get the network configuration hash reference |
| my $network_configuration = $self->get_network_configuration($no_cache); |
| if (!$network_configuration) { |
| notify($ERRORS{'WARNING'}, 0, "unable to retrieve IP addresses from $computer_name, failed to retrieve network configuration"); |
| return; |
| } |
| |
| # Loop through all of the network interfaces found |
| my @ip_addresses; |
| for my $interface_name (sort keys %$network_configuration) { |
| if ($network_configuration->{$interface_name}{ip_address}) { |
| push @ip_addresses, keys %{$network_configuration->{$interface_name}{ip_address}}; |
| } |
| } |
| |
| notify($ERRORS{'DEBUG'}, 0, "retrieved IP addresses bound on $computer_name:\n" . join("\n", @ip_addresses)); |
| return @ip_addresses; |
| } |
| |
| #////////////////////////////////////////////////////////////////////////////// |
| |
| =head2 get_private_ip_address |
| |
| Parameters : $no_cache (optional), $ignore_error (optional) |
| Returns : string |
| Description : Returns the computer's private IP address. |
| |
| =cut |
| |
| sub get_private_ip_address { |
| my $self = shift; |
| if (ref($self) !~ /VCL::Module/i) { |
| notify($ERRORS{'CRITICAL'}, 0, "subroutine was called as a function, it must be called as a class method"); |
| return; |
| } |
| |
| my $no_cache = shift; |
| my $ignore_error = shift; |
| |
| return $self->get_ip_address('private', $no_cache, $ignore_error); |
| } |
| |
| #////////////////////////////////////////////////////////////////////////////// |
| |
| =head2 get_public_ip_address |
| |
| Parameters : $no_cache (optional), $ignore_error (optional) |
| Returns : string |
| Description : Returns the computer's public IP address. |
| |
| =cut |
| |
| sub get_public_ip_address { |
| my $self = shift; |
| if (ref($self) !~ /VCL::Module/i) { |
| notify($ERRORS{'CRITICAL'}, 0, "subroutine was called as a function, it must be called as a class method"); |
| return; |
| } |
| my $no_cache = shift || 0; |
| my $ignore_error = shift || 0; |
| |
| notify($ERRORS{'DEBUG'}, 0, "attempting to retrieve public IP address, no cache: $no_cache, ignore error: $ignore_error"); |
| return $self->get_ip_address('public', $no_cache, $ignore_error); |
| } |
| |
| #////////////////////////////////////////////////////////////////////////////// |
| |
| =head2 get_subnet_mask |
| |
| Parameters : |
| Returns : $ip_address |
| Description : |
| |
| =cut |
| |
| sub get_subnet_mask { |
| my $self = shift; |
| if (ref($self) !~ /VCL::Module/i) { |
| notify($ERRORS{'CRITICAL'}, 0, "subroutine was called as a function, it must be called as a class method"); |
| return; |
| } |
| |
| # Get the IP address argument |
| my $ip_address = shift; |
| if (!$ip_address) { |
| notify($ERRORS{'WARNING'}, 0, "unable to determine subnet mask, IP address argument was not specified"); |
| return; |
| } |
| |
| # Make sure network configuration was retrieved |
| my $network_configuration = $self->get_network_configuration(); |
| if (!$network_configuration) { |
| notify($ERRORS{'WARNING'}, 0, "unable to retrieve network configuration"); |
| return; |
| } |
| |
| for my $interface_name (keys(%$network_configuration)) { |
| my $ip_address_info = $network_configuration->{$interface_name}{ip_address}; |
| |
| if (!defined($ip_address_info->{$ip_address})) { |
| next; |
| } |
| |
| my $subnet_mask = $ip_address_info->{$ip_address}; |
| if ($subnet_mask) { |
| return $subnet_mask; |
| } |
| else { |
| notify($ERRORS{'WARNING'}, 0, "subnet mask is not set for interface '$interface_name' IP address $ip_address in network configuration:\n" . format_data($network_configuration)); |
| return; |
| } |
| } |
| |
| notify($ERRORS{'WARNING'}, 0, "interface with IP address $ip_address does not exist in the network configuration:\n" . format_data($network_configuration)); |
| return; |
| } |
| |
| #////////////////////////////////////////////////////////////////////////////// |
| |
| =head2 get_private_subnet_mask |
| |
| Parameters : |
| Returns : |
| Description : |
| |
| =cut |
| |
| sub get_private_subnet_mask { |
| my $self = shift; |
| if (ref($self) !~ /VCL::Module/i) { |
| notify($ERRORS{'CRITICAL'}, 0, "subroutine was called as a function, it must be called as a class method"); |
| return; |
| } |
| |
| return $self->get_subnet_mask($self->get_private_ip_address()); |
| } |
| |
| #////////////////////////////////////////////////////////////////////////////// |
| |
| =head2 get_public_subnet_mask |
| |
| Parameters : |
| Returns : |
| Description : |
| |
| =cut |
| |
| sub get_public_subnet_mask { |
| my $self = shift; |
| if (ref($self) !~ /VCL::Module/i) { |
| notify($ERRORS{'CRITICAL'}, 0, "subroutine was called as a function, it must be called as a class method"); |
| return; |
| } |
| |
| return $self->get_subnet_mask($self->get_public_ip_address()); |
| } |
| |
| #////////////////////////////////////////////////////////////////////////////// |
| |
| =head2 get_default_gateway |
| |
| Parameters : |
| Returns : |
| Description : |
| |
| =cut |
| |
| sub get_default_gateway { |
| my $self = shift; |
| if (ref($self) !~ /VCL::Module/i) { |
| notify($ERRORS{'CRITICAL'}, 0, "subroutine was called as a function, it must be called as a class method"); |
| return; |
| } |
| |
| # Check if a 'public' or 'private' network type argument was specified |
| # Assume 'public' if not specified |
| my $network_type = shift || 'public'; |
| $network_type = lc($network_type) if $network_type; |
| if ($network_type && $network_type !~ /(public|private)/i) { |
| notify($ERRORS{'WARNING'}, 0, "network type argument can only be 'public' or 'private'"); |
| return; |
| } |
| |
| # Get the public or private network configuration |
| # Use 'eval' to construct the appropriate subroutine name |
| my $network_configuration = eval "\$self->get_$network_type\_network_configuration()"; |
| if ($EVAL_ERROR || !$network_configuration) { |
| notify($ERRORS{'WARNING'}, 0, "unable to retrieve $network_type network configuration"); |
| return; |
| } |
| |
| my $default_gateway = $network_configuration->{default_gateway}; |
| if ($default_gateway) { |
| notify($ERRORS{'DEBUG'}, 0, "returning $network_type default gateway: $default_gateway"); |
| return $default_gateway; |
| } |
| else { |
| notify($ERRORS{'WARNING'}, 0, "unable to determine $network_type default gateway, 'default_gateway' key does not exist in the network configuration info: \n" . format_data($network_configuration)); |
| return; |
| } |
| } |
| |
| #////////////////////////////////////////////////////////////////////////////// |
| |
| =head2 get_private_default_gateway |
| |
| Parameters : |
| Returns : |
| Description : |
| |
| =cut |
| |
| sub get_private_default_gateway { |
| my $self = shift; |
| if (ref($self) !~ /VCL::Module/i) { |
| notify($ERRORS{'CRITICAL'}, 0, "subroutine was called as a function, it must be called as a class method"); |
| return; |
| } |
| |
| return $self->get_default_gateway('private'); |
| } |
| |
| #////////////////////////////////////////////////////////////////////////////// |
| |
| =head2 get_public_default_gateway |
| |
| Parameters : |
| Returns : |
| Description : |
| |
| =cut |
| |
| sub get_public_default_gateway { |
| my $self = shift; |
| if (ref($self) !~ /VCL::Module/i) { |
| notify($ERRORS{'CRITICAL'}, 0, "subroutine was called as a function, it must be called as a class method"); |
| return; |
| } |
| |
| return $self->get_default_gateway('public'); |
| } |
| |
| #////////////////////////////////////////////////////////////////////////////// |
| |
| =head2 get_dns_servers |
| |
| Parameters : $network_type |
| Returns : array |
| Description : Retrieves a list of DNS servers currently configured on the |
| computer. The $network_type argument may either be 'private' or |
| 'public'. The default is to retrieve the DNS servers configured |
| for the public interface. |
| |
| =cut |
| |
| sub get_dns_servers { |
| my $self = shift; |
| if (ref($self) !~ /VCL::Module/i) { |
| notify($ERRORS{'CRITICAL'}, 0, "subroutine was called as a function, it must be called as a class method"); |
| return; |
| } |
| |
| # Check if a 'public' or 'private' network type argument was specified |
| # Assume 'public' if not specified |
| my $network_type = shift || 'public'; |
| $network_type = lc($network_type) if $network_type; |
| if ($network_type && $network_type !~ /(public|private)/i) { |
| notify($ERRORS{'WARNING'}, 0, "network type argument can only be 'public' or 'private'"); |
| return; |
| } |
| |
| # Get the public or private network configuration |
| # Use 'eval' to construct the appropriate subroutine name |
| my $network_configuration = eval "\$self->get_$network_type\_network_configuration()"; |
| if ($EVAL_ERROR || !$network_configuration) { |
| notify($ERRORS{'WARNING'}, 0, "unable to retrieve $network_type network configuration"); |
| return; |
| } |
| |
| if (!defined($network_configuration->{dns_servers})) { |
| notify($ERRORS{'WARNING'}, 0, "unable to determine $network_type DNS servers, 'dns_servers' key does not exist in the network configuration info: \n" . format_data($network_configuration)); |
| return; |
| } |
| elsif (!ref($network_configuration->{dns_servers}) || ref($network_configuration->{dns_servers}) ne 'ARRAY') { |
| notify($ERRORS{'WARNING'}, 0, "unable to determine $network_type DNS servers, 'dns_servers' key is not an array reference in the network configuration info: \n" . format_data($network_configuration)); |
| return; |
| } |
| |
| my @dns_servers = @{$network_configuration->{dns_servers}}; |
| notify($ERRORS{'DEBUG'}, 0, "returning $network_type DNS servers: " . join(", ", @dns_servers)); |
| return @dns_servers; |
| } |
| |
| #////////////////////////////////////////////////////////////////////////////// |
| |
| =head2 get_private_dns_servers |
| |
| Parameters : none |
| Returns : array |
| Description : Retrieves a list of DNS servers currently configured for the |
| private interface on the computer. |
| |
| =cut |
| |
| sub get_private_dns_servers { |
| my $self = shift; |
| if (ref($self) !~ /VCL::Module/i) { |
| notify($ERRORS{'CRITICAL'}, 0, "subroutine was called as a function, it must be called as a class method"); |
| return; |
| } |
| |
| return $self->get_dns_servers('private'); |
| } |
| |
| #////////////////////////////////////////////////////////////////////////////// |
| |
| =head2 get_public_dns_servers |
| |
| Parameters : none |
| Returns : array |
| Description : Retrieves a list of DNS servers currently configured for the |
| public interface on the computer. |
| |
| =cut |
| |
| sub get_public_dns_servers { |
| my $self = shift; |
| if (ref($self) !~ /VCL::Module/i) { |
| notify($ERRORS{'CRITICAL'}, 0, "subroutine was called as a function, it must be called as a class method"); |
| return; |
| } |
| |
| return $self->get_dns_servers('public'); |
| } |
| |
| #////////////////////////////////////////////////////////////////////////////// |
| |
| =head2 create_text_file |
| |
| Parameters : $file_path, $file_contents, $append |
| Returns : boolean |
| Description : Creates a text file on the computer or appends to an existing |
| file. |
| |
| A trailing newline character is added to the end of the |
| file if one is not present in the $file_contents argument. |
| |
| It is assumed that when appending to an existing file, the value |
| of $file_contents is intended to be added on a new line at the |
| end of the file. The contents of the existing file are first |
| retrieved. If the existing file does not contain a trailing |
| newline, one is added before appending to the file. |
| |
| =cut |
| |
| sub create_text_file { |
| my $self = shift; |
| if (ref($self) !~ /VCL::Module/i) { |
| notify($ERRORS{'CRITICAL'}, 0, "subroutine was called as a function, it must be called as a class method"); |
| return; |
| } |
| |
| my ($file_path, $file_contents_string, $append) = @_; |
| if (!defined($file_path)) { |
| notify($ERRORS{'WARNING'}, 0, "file path argument was not supplied"); |
| return; |
| } |
| elsif (!defined($file_contents_string)) { |
| notify($ERRORS{'WARNING'}, 0, "file contents argument was not supplied"); |
| return; |
| } |
| |
| my $computer_node_name = $self->data->get_computer_node_name(); |
| |
| my $image_os_type = $self->data->get_image_os_type(); |
| my $newline; |
| if ($image_os_type =~ /win/i) { |
| $newline = "\r\n"; |
| } |
| else { |
| $newline = "\n"; |
| } |
| |
| # Used only to format notify messages |
| my $mode_string; |
| if ($append) { |
| $mode_string = 'append'; |
| |
| # Retrieve the contents of the existing file if necessary |
| if ($self->file_exists($file_path)) { |
| my $existing_file_contents = $self->get_file_contents($file_path); |
| if (!defined($existing_file_contents)) { |
| # Do not proceed if any problem occurred retrieving the existing file contents |
| # Otherwise, it would be overwritten and data would be lost |
| notify($ERRORS{'WARNING'}, 0, "failed to $mode_string text file on $computer_node_name, append argument was specified but contents of the existing file could not be retrieved: $file_path"); |
| return; |
| } |
| elsif ($existing_file_contents && $existing_file_contents !~ /\n$/) { |
| # Add a newline to the end of the existing contents if one isn't present |
| $existing_file_contents .= $newline; |
| $file_contents_string = $existing_file_contents . $file_contents_string; |
| } |
| |
| } |
| } |
| else { |
| $mode_string = 'create'; |
| } |
| |
| # Make sure the file contents ends with a newline |
| # This is helpful to prevent problems with files such as sshd_config and sudoers where a line might be echo'd to the end of it |
| # Without the newline, the last line could wind up being a merged line if a simple echo is used to add a line |
| if (length($file_contents_string) && $file_contents_string !~ /\n$/) { |
| $file_contents_string .= $newline; |
| } |
| |
| # Make line endings consistent |
| $file_contents_string =~ s/\r*\n/$newline/g; |
| |
| # Attempt to create the parent directory if it does not exist |
| if ($file_path =~ /[\\\/]/ && $self->can('create_directory')) { |
| my $parent_directory_path = parent_directory_path($file_path); |
| $self->create_directory($parent_directory_path) if $parent_directory_path; |
| } |
| |
| # The echo method will fail of the file contents are too large |
| # An "Argument list too long" error will be displayed |
| my $file_contents_length = length($file_contents_string); |
| if ($file_contents_length < 32000) { |
| # Create a command to echo the string and another to echo the hex string to the file |
| # The hex string command is preferred because it will handle special characters including single quotes |
| # Use -e to enable interpretation of backslash escapes |
| # Use -n to prevent trailing newline from being added |
| # However, the command may become very long and fail |
| # Convert the string to a string containing the hex value of each character |
| # This is done to avoid problems with special characters in the file contents |
| # Split the string up into an array if integers representing each character's ASCII decimal value |
| my @decimal_values = unpack("C*", $file_contents_string); |
| |
| # Convert the ASCII decimal values into hex values and add '\x' before each hex value |
| my @hex_values = map { '\x' . sprintf("%x", $_) } @decimal_values; |
| |
| # Join the hex values together into a string |
| my $hex_string = join('', @hex_values); |
| |
| # Enclose the file path in quotes if it contains a space |
| if ($file_path =~ /[\s]/) { |
| $file_path = "\"$file_path\""; |
| } |
| |
| # Attempt to create the file using the hex string |
| my $command = "echo -n -e \"$hex_string\" > $file_path"; |
| my ($exit_status, $output) = $self->execute($command, 0, 15, 1); |
| if (!defined($output)) { |
| notify($ERRORS{'DEBUG'}, 0, "failed to execute command to $mode_string file on $computer_node_name, attempting to create file on management node and copy it to $computer_node_name"); |
| } |
| elsif ($exit_status != 0 || grep(/^\w+:/i, @$output)) { |
| notify($ERRORS{'WARNING'}, 0, "failed to execute command to $mode_string a file on $computer_node_name, command: '$command', exit status: $exit_status, output:\n" . join("\n", @$output) . "\nattempting to create file on management node and copy it to $computer_node_name"); |
| } |
| elsif ($append) { |
| notify($ERRORS{'DEBUG'}, 0, $mode_string . ($append ? 'ed' : 'd') . " text file on $computer_node_name: $file_path"); |
| return 1; |
| } |
| else { |
| notify($ERRORS{'DEBUG'}, 0, $mode_string . ($append ? 'ed' : 'd') . " text file on $computer_node_name: $file_path"); |
| return 1; |
| } |
| } |
| else { |
| notify($ERRORS{'DEBUG'}, 0, "skipping attempt to $mode_string on $computer_node_name using echo, file content string length: $file_contents_length"); |
| } |
| |
| # File was not created using the quicker echo method |
| # Create a temporary file on the management node, copy it to the computer, then delete temporary file from management node |
| my $mn_temp_file_path = tmpnam(); |
| if (!$self->mn_os->create_text_file($mn_temp_file_path, $file_contents_string, $append)) { |
| notify($ERRORS{'WARNING'}, 0, "failed to create text file on $computer_node_name, temporary file could not be created on the management node"); |
| return; |
| } |
| if (!$self->copy_file_to($mn_temp_file_path, $file_path)) { |
| notify($ERRORS{'WARNING'}, 0, "failed to create text file, temporary file could not be dopied from management node to $computer_node_name: $mn_temp_file_path --> $file_path"); |
| return; |
| } |
| $self->mn_os->delete_file($mn_temp_file_path); |
| notify($ERRORS{'DEBUG'}, 0, "created text file on $computer_node_name by copying a file created on management node"); |
| return 1; |
| } |
| |
| #////////////////////////////////////////////////////////////////////////////// |
| |
| =head2 append_text_file |
| |
| Parameters : $file_path, $file_contents |
| Returns : boolean |
| Description : Appends to a text file on the computer. |
| |
| =cut |
| |
| sub append_text_file { |
| my $self = shift; |
| if (ref($self) !~ /VCL::Module/i) { |
| notify($ERRORS{'CRITICAL'}, 0, "subroutine was called as a function, it must be called as a class method"); |
| return; |
| } |
| |
| my ($file_path, $file_contents_string) = @_; |
| if (!$file_path || !defined($file_contents_string)) { |
| notify($ERRORS{'WARNING'}, 0, "file path and contents arguments were not supplied"); |
| return; |
| } |
| |
| return $self->create_text_file($file_path, $file_contents_string, 1); |
| } |
| |
| #////////////////////////////////////////////////////////////////////////////// |
| |
| =head2 set_text_file_line_endings |
| |
| Parameters : $file_path, $line_ending (optional) |
| Returns : boolean |
| Description : Changes the line endings of a text file. This is equivalent to |
| running unix2dos or dos2unix. The default line ending type is |
| unix. Windows-style line endings will be applied if the |
| $line_ending argument is supplied and contains 'win' or 'r'. |
| |
| =cut |
| |
| sub set_text_file_line_endings { |
| my $self = shift; |
| if (ref($self) !~ /VCL::Module/i) { |
| notify($ERRORS{'CRITICAL'}, 0, "subroutine was called as a function, it must be called as a class method"); |
| return; |
| } |
| |
| my ($file_path, $line_ending) = @_; |
| if (!$file_path) { |
| notify($ERRORS{'WARNING'}, 0, "file path argument was not supplied"); |
| return; |
| } |
| |
| my $computer_name = $self->data->get_computer_short_name(); |
| |
| my $file_contents_original = $self->get_file_contents($file_path); |
| if (!defined($file_contents_original)) { |
| notify($ERRORS{'WARNING'}, 0, "unable to set line endings for $file_path on $computer_name, failed to retrieve file contents"); |
| return; |
| } |
| my $file_contents_updated = $file_contents_original; |
| |
| my $line_ending_type; |
| if ($line_ending && $line_ending =~ /(r|win)/i) { |
| $file_contents_updated =~ s/\r?\n/\r\n/g; |
| $line_ending_type = 'Windows'; |
| } |
| else { |
| $file_contents_updated =~ s/\r//g; |
| $line_ending_type = 'Unix'; |
| } |
| |
| if ($file_contents_updated eq $file_contents_original) { |
| notify($ERRORS{'DEBUG'}, 0, "$line_ending_type-style line endings already set for $file_path on $computer_name"); |
| return 1; |
| } |
| elsif ($self->create_text_file($file_path, $file_contents_updated)) { |
| notify($ERRORS{'DEBUG'}, 0, "set $line_ending_type-style line endings for $file_path on $computer_name"); |
| return 1; |
| } |
| else { |
| notify($ERRORS{'WARNING'}, 0, "failed to set $line_ending_type-style line endings for $file_path on $computer_name, unable to overwrite file"); |
| return 0; |
| } |
| } |
| |
| #////////////////////////////////////////////////////////////////////////////// |
| |
| =head2 get_file_contents |
| |
| Parameters : $file_path |
| Returns : array or string |
| Description : Returns the contents of the file specified by the file path |
| argument. |
| |
| If the expected return value is an array, each array element |
| contains a string for each line from the file. Newlines and |
| carriage returns are not included. |
| |
| If the expected return value is a scalar, a string is returned. |
| |
| =cut |
| |
| sub get_file_contents { |
| my $self = shift; |
| if (ref($self) !~ /module/i) { |
| notify($ERRORS{'CRITICAL'}, 0, "subroutine was called as a function, it must be called as a class method"); |
| return; |
| } |
| |
| # Get the file path argument |
| my $file_path = shift; |
| if (!$file_path) { |
| notify($ERRORS{'WARNING'}, 0, "path argument was not specified"); |
| return; |
| } |
| |
| my $computer_short_name = $self->data->get_computer_short_name(); |
| |
| # Run cat to retrieve the contents of the file |
| my $command = "cat \"$file_path\""; |
| my ($exit_status, $output) = $self->execute($command, 0); |
| if (!defined($output)) { |
| notify($ERRORS{'WARNING'}, 0, "failed to execute command to read file on $computer_short_name: '$file_path'\ncommand: '$command'"); |
| return; |
| } |
| elsif (grep(/^cat: /, @$output)) { |
| notify($ERRORS{'WARNING'}, 0, "failed to read contents of file on $computer_short_name: '$file_path', exit status: $exit_status, output:\n" . join("\n", @$output)); |
| return; |
| } |
| else { |
| notify($ERRORS{'DEBUG'}, 0, "retrieved " . scalar(@$output) . " lines from file on $computer_short_name: '$file_path'"); |
| if (wantarray) { |
| return @$output; |
| } |
| else { |
| return join("\n", @$output); |
| } |
| } |
| } |
| |
| #////////////////////////////////////////////////////////////////////////////// |
| |
| =head2 remove_lines_from_file |
| |
| Parameters : $file_path, $pattern |
| Returns : integer or undefined |
| Description : Removes all lines containing the pattern from the file. The |
| pattern must be a regular expression. Returns the number of lines |
| removed from the file which may be 0. Returns undefined if an |
| error occurred. |
| |
| =cut |
| |
| sub remove_lines_from_file { |
| my $self = shift; |
| if (ref($self) !~ /module/i) { |
| notify($ERRORS{'CRITICAL'}, 0, "subroutine was called as a function, it must be called as a class method"); |
| return; |
| } |
| |
| my ($file_path, $pattern) = @_; |
| if (!$file_path || !$pattern) { |
| notify($ERRORS{'WARNING'}, 0, "file path and pattern arguments were not specified"); |
| return; |
| } |
| |
| my $computer_short_name = $self->data->get_computer_short_name(); |
| |
| my @lines_removed; |
| my @lines_retained; |
| |
| if (!$self->file_exists($file_path)) { |
| notify($ERRORS{'DEBUG'}, 0, "lines containing '$pattern' not removed because file does NOT exist: $file_path"); |
| return 0; |
| } |
| |
| my @lines = $self->get_file_contents($file_path); |
| for my $line (@lines) { |
| if ($line =~ /$pattern/) { |
| push @lines_removed, $line; |
| } |
| else { |
| push @lines_retained, $line; |
| } |
| } |
| |
| if (@lines_removed) { |
| my $lines_removed_count = scalar(@lines_removed); |
| my $new_file_contents = join("\n", @lines_retained) || ''; |
| notify($ERRORS{'DEBUG'}, 0, "removed $lines_removed_count line" . ($lines_removed_count > 1 ? 's' : '') . " from $file_path matching pattern: '$pattern'\n" . join("\n", @lines_removed)); |
| $self->create_text_file($file_path, $new_file_contents) || return; |
| } |
| else { |
| notify($ERRORS{'DEBUG'}, 0, "$file_path does NOT contain any lines matching pattern: '$pattern'"); |
| } |
| return scalar(@lines_removed); |
| } |
| |
| #////////////////////////////////////////////////////////////////////////////// |
| |
| =head2 execute |
| |
| Parameters : $command, $display_output (optional) |
| Returns : array ($exit_status, $output) |
| Description : Executes a command on the computer via SSH. |
| |
| =cut |
| |
| sub execute { |
| my @original_arguments = @_; |
| my ($argument) = @_; |
| my ($computer_name, $command, $display_output, $timeout_seconds, $max_attempts, $port, $user, $password, $identity_key, $ignore_error); |
| |
| my $self; |
| |
| # Check if this subroutine was called as an object method |
| if (ref($argument) && ref($argument) =~ /VCL::Module/) { |
| # Subroutine was called as an object method ($self->execute) |
| $self = shift; |
| ($argument) = @_; |
| |
| #notify($ERRORS{'DEBUG'}, 0, "called as an object method: " . ref($self)); |
| |
| # Get the computer name from the reservation data |
| $computer_name = $self->data->get_computer_node_name(); |
| if (!$computer_name) { |
| notify($ERRORS{'WARNING'}, 0, "called as an object method, failed to retrieve computer name from reservation data"); |
| return; |
| } |
| #notify($ERRORS{'DEBUG'}, 0, "retrieved computer name from reservation data: $computer_name"); |
| } |
| |
| my $no_persistent_connection = 0; |
| |
| # Check the argument type |
| if (ref($argument)) { |
| if (ref($argument) eq 'HASH') { |
| #notify($ERRORS{'DEBUG'}, 0, "first argument is a hash reference:\n" . format_data($argument)); |
| |
| $computer_name = $argument->{node} if (!$computer_name); |
| $command = $argument->{command}; |
| $display_output = $argument->{display_output}; |
| $timeout_seconds = $argument->{timeout_seconds}; |
| $max_attempts = $argument->{max_attempts}; |
| $port = $argument->{port}; |
| $user = $argument->{user}; |
| $password = $argument->{password}; |
| $identity_key = $argument->{identity_key}; |
| $ignore_error = $argument->{ignore_error}; |
| $no_persistent_connection = $argument->{no_persistent_connection}; |
| } |
| else { |
| notify($ERRORS{'WARNING'}, 0, "invalid argument reference type passed: " . ref($argument) . ", if a reference is passed as the argument it may only be a hash or VCL::Module reference"); |
| return; |
| } |
| } |
| else { |
| # Argument is not a reference, computer name must be the first argument unless this subroutine was called as an object method |
| # If called as an object method, $computer_name will already be populated |
| if (!$computer_name) { |
| $computer_name = shift; |
| #notify($ERRORS{'DEBUG'}, 0, "first argument is a scalar, should be the computer name: $computer_name, remaining arguments:\n" . format_data(\@_)); |
| } |
| else { |
| #notify($ERRORS{'DEBUG'}, 0, "first argument should be the command:\n" . format_data(\@_)); |
| } |
| |
| # Get the remaining arguments |
| ($command, $display_output, $timeout_seconds, $max_attempts, $port, $user, $password, $identity_key, $ignore_error) = @_; |
| } |
| |
| if (!$computer_name) { |
| notify($ERRORS{'WARNING'}, 0, "computer name could not be determined"); |
| return; |
| } |
| if (!$command) { |
| notify($ERRORS{'WARNING'}, 0, "command argument was not specified"); |
| return; |
| } |
| |
| # TESTING: use the new subroutine if $ENV{execute_new} is set and the command isn't one that's known to fail with the new subroutine |
| if ($ENV->{execute_new} && !$no_persistent_connection) { |
| my @excluded_commands = $command =~ /(vmkfstools|qemu-img|Convert-VHD|scp|shutdown|reboot)/i; |
| if (@excluded_commands) { |
| notify($ERRORS{'DEBUG'}, 0, "not using execute_new, command: $command\nexcluded commands matched:\n" . join("\n", @excluded_commands)); |
| } |
| else { |
| return execute_new(@original_arguments); |
| } |
| } |
| |
| # If 'ssh_user' key is set in this object, use it |
| # This allows OS modules to specify the username to use |
| if ($self && $self->{ssh_user}) { |
| #notify($ERRORS{'DEBUG'}, 0, "\$self->{ssh_user} is defined: $self->{ssh_user}"); |
| $user = $self->{ssh_user}; |
| } |
| elsif (!$port) { |
| $user = 'root'; |
| } |
| |
| # If 'ssh_port' key is set in this object, use it |
| # This allows OS modules to specify the port to use |
| if ($self && $self->{ssh_port}) { |
| #notify($ERRORS{'DEBUG'}, 0, "\$self->{ssh_port} is defined: $self->{ssh_port}"); |
| $port = $self->{ssh_port}; |
| } |
| elsif (!$port) { |
| $port = 22; |
| } |
| |
| my $arguments = { |
| node => $computer_name, |
| command => $command, |
| identity_paths => $identity_key, |
| user => $user, |
| port => $port, |
| output_level => $display_output, |
| max_attempts => $max_attempts, |
| timeout_seconds => $timeout_seconds, |
| }; |
| |
| # Run the command via SSH |
| my ($exit_status, $output) = run_ssh_command($arguments); |
| if (defined($exit_status) && defined($output)) { |
| if ($display_output) { |
| notify($ERRORS{'DEBUG'}, 0, "executed command: '$command', exit status: $exit_status, output:\n" . join("\n", @$output)); |
| } |
| return ($exit_status, $output); |
| } |
| else { |
| notify($ERRORS{'WARNING'}, 0, "failed to run command on $computer_name: $command") if $display_output; |
| return; |
| } |
| } |
| |
| #////////////////////////////////////////////////////////////////////////////// |
| |
| =head2 execute_new |
| |
| Parameters : $computer_name (conditional), $command, $display_output, $timeout_seconds, $max_attempts, $port, $user, $password |
| Returns : array ($exit_status, $output) |
| Description : Executes a command on the computer via SSH. |
| |
| =cut |
| |
| sub execute_new { |
| my ($argument) = @_; |
| my ($computer_name, $command, $display_output, $timeout_seconds, $max_attempts, $port, $user, $password, $identity_key, $ignore_error); |
| |
| my $self; |
| |
| # Check if this subroutine was called as an object method |
| if (ref($argument) && ref($argument) =~ /VCL::Module/) { |
| # Subroutine was called as an object method ($self->execute) |
| $self = shift; |
| ($argument) = @_; |
| |
| #notify($ERRORS{'DEBUG'}, 0, "called as an object method: " . ref($self)); |
| |
| # Get the computer name from the reservation data |
| $computer_name = $self->data->get_computer_node_name(); |
| if (!$computer_name) { |
| notify($ERRORS{'WARNING'}, 0, "called as an object method, failed to retrieve computer name from reservation data"); |
| return; |
| } |
| #notify($ERRORS{'DEBUG'}, 0, "retrieved computer name from reservation data: $computer_name"); |
| } |
| |
| # Check the argument type |
| if (ref($argument)) { |
| if (ref($argument) eq 'HASH') { |
| #notify($ERRORS{'DEBUG'}, 0, "first argument is a hash reference:\n" . format_data($argument)); |
| |
| $computer_name = $argument->{node} if (!$computer_name); |
| |