Current Path : /proc/3/cwd/scripts/ |
|
Current File : //proc/3/cwd/scripts/fixetchosts |
#!/usr/local/cpanel/3rdparty/bin/perl
package scripts::fixetchosts;
# cpanel - fixetchosts Copyright 2010 cPanel, Inc.
# All rights Reserved.
# copyright@cpanel.net http://cpanel.net
# This code is subject to the cPanel license. Unauthorized copying is prohibited
use strict;
use Cpanel::SafeFile ();
use Cpanel::Validate::Hostname ();
use Cpanel::IP::Loopback ();
use Cpanel::DIp::MainIP ();
use Cpanel::Sys::Hostname::FQDN ();
use Cpanel::FileUtils::TouchFile ();
use IO::Handle ();
exit script(@ARGV) unless caller;
sub script {
return run('/etc/hosts');
}
sub run {
my ($file) = @_;
my $mainip = Cpanel::DIp::MainIP::getmainserverip();
my $hostname = Cpanel::Sys::Hostname::FQDN::get_fqdn_hostname();
my $shorthostname = Cpanel::Sys::Hostname::FQDN::get_short_hostname();
Cpanel::FileUtils::TouchFile::touchfile($file);
die "The hostname $hostname is not valid.\n" unless Cpanel::Validate::Hostname::is_valid($hostname);
update( $file, $mainip, $hostname, $shorthostname );
return 0;
}
sub update {
my ( $file, $mainip, $hostname, $shorthostname ) = @_;
my $hosts_fh = IO::Handle->new();
my $hosts_lock = Cpanel::SafeFile::safeopen( $hosts_fh, '+<', $file );
if ($hosts_lock) {
my %SEENIP;
my @ETCHOSTS;
while ( my $line = readline $hosts_fh ) {
chomp $line;
if ( $line =~ m/^\s*#/ || $line =~ m/^\s*$/ ) {
push @ETCHOSTS, { 'raw' => $line };
}
else {
my ( $ip, $hosts ) = split( /\s+/, $line, 2 );
my @line_hosts;
foreach my $host ( split( /\s+/, $hosts ) ) {
next if ( $ip ne $mainip && $host eq $hostname ); # Drop mismatch hostnames for the main IP
next if ( $ip eq $mainip && ( $host eq $hostname || $host eq $shorthostname ) ); # These will be re-added
push @line_hosts, $host if ( !grep { $_ eq $host } @line_hosts ); # Drop duplicates while preserving order
}
if ( !@line_hosts ) { ## Drop empty lines
next if $ip eq $mainip; # Prevent duplicate commented lines for the main IP
push @ETCHOSTS, { 'raw' => '# ' . $line }; # Preserve lines that were commented out in full
}
else { ## Save valid entries
push @ETCHOSTS, { 'ip' => $ip, 'hosts' => \@line_hosts };
$SEENIP{$ip} = 1;
}
}
}
seek( $hosts_fh, 0, 0 );
my $added_hostname = 0;
foreach my $line_ref (@ETCHOSTS) {
if ( exists $line_ref->{'raw'} ) {
print {$hosts_fh} $line_ref->{'raw'} . "\n";
}
else {
my $ip = $line_ref->{'ip'};
my $hosts_ref = $line_ref->{'hosts'};
if ( $ip eq $mainip && !$added_hostname ) {
$added_hostname = 1;
unshift @{$hosts_ref}, $hostname, $shorthostname;
}
elsif ( Cpanel::IP::Loopback::is_loopback($ip) ) {
if ( !grep { $_ eq 'localhost' } @{$hosts_ref} ) {
unshift @{$hosts_ref}, 'localhost';
}
}
print {$hosts_fh} $ip . "\t\t" . join( ' ', @{$hosts_ref} ) . "\n";
}
}
if ( !exists $SEENIP{'127.0.0.1'} ) {
print {$hosts_fh} "127.0.0.1\t\tlocalhost\n";
}
if ( !exists $SEENIP{$mainip} ) {
print {$hosts_fh} $mainip . "\t\t" . $hostname . ' ' . $shorthostname . "\n";
}
truncate( $hosts_fh, tell($hosts_fh) );
Cpanel::SafeFile::safeclose( $hosts_fh, $hosts_lock );
}
return;
}
Copyright 2K16 - 2K18 Indonesian Hacker Rulez