#!/usr/local/cpanel/3rdparty/bin/perl
# cpanel - scripts/ensure_hostname_resolves Copyright 2022 cPanel, L.L.C.
# All rights reserved.
# copyright@cpanel.net http://cpanel.net
# This code is subject to the cPanel license. Unauthorized copying is prohibited
package scripts::ensure_hostname_resolves;
=encoding utf8
=head1 NAME
ensure_hostname_resolves
=head1 DESCRIPTION
Utility that checks to ensure that the hostname resolves to the server.
It can also update the hostname to use an 'autodomain' hostname which is guaranteed to resolve.
=head1 SYNOPSIS
ensure_hostname_resolves [OPTIONS]
--yes Skip the prompts, and assume 'yes'
--help This documentation.
=cut
use cPstrict;
use Try::Tiny;
use IO::Prompt ();
use Cpanel::Exception ();
use Cpanel::Hostname ();
use Whostmgr::Hostname ();
use Cpanel::Domain::ExternalResolver ();
use constant _OPTIONS => ('yes');
use parent qw( Cpanel::HelpfulScript );
exit __PACKAGE__->new(@ARGV)->run() unless caller();
sub run ($self) {
die Cpanel::Exception::create('RootRequired')->to_string_no_id() unless ( $> == 0 && $< == 0 );
my $hostname_resolves_local;
try {
if ( $hostname_resolves_local = $self->hostname_resolves() ) {
print "[+] Hostname resolves to the server.\n";
}
else {
print "[!] Hostname does NOT resolve to the server.\n";
}
}
catch {
my $str = Cpanel::Exception::get_string($_);
print "[!] Failed to determine if hostname resolves to the server: $str\n";
};
return 0 if $hostname_resolves_local;
if ( $self->getopt('yes') || $self->prompt_yn() ) {
print "[*] Configuring auto-resolving hostname...\n";
my $new_hostname = Whostmgr::Hostname::set_autodomain_hostname();
if ( $self->hostname_resolves() ) {
print "[+] Successfully updated hostname to auto-resolving hostname: $new_hostname\n";
}
else {
print "[!] Failed to update hostname to an auto-resolving hostname.\n";
return 1;
}
}
return 0;
}
sub hostname_resolves ($self) {
return Cpanel::Domain::ExternalResolver::domain_is_on_local_server( Cpanel::Hostname::gethostname(1) );
}
sub prompt_yn ($self) {
return IO::Prompt::prompt(
'-yes',
'-one_char',
'[*] Do you want the system to update the hostname to an auto-resolving hostname? [y/n] '
);
}
1;
Mini Shell