Current Path : /scripts/ |
|
Current File : //scripts/importmydnsdb |
#!/usr/local/cpanel/3rdparty/bin/perl
# cpanel - scripts/importmydnsdb Copyright 2019 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
use strict;
use Cpanel::DNSLib::MydnsDb ();
use Cpanel::Validate::Domain::Tiny ();
use Cpanel::MysqlUtils::Connect ();
use Cpanel::NameServer::Conf::Mydns ();
use Cpanel::NameServer::Utils::MyDNS ();
use Cpanel::PID ();
use Cpanel::SafeRun::Errors ();
use Cpanel::Usage ();
use Cpanel::NameServer::Utils::MyDNS::DB ();
use Try::Tiny;
my ( $force, $debug );
Cpanel::Usage::wrap_options( \@ARGV, \&usage, { 'force' => \$force, 'debug' => \$debug } );
if ( $> != 0 ) {
print STDERR "$0: must run as root\n";
exit 1;
}
my @messages = ();
my $pid_obj = Cpanel::PID->new( { 'pid_file' => '/var/run/importmydnsdb.pid' } );
unless ( $pid_obj->create_pid_file() > 0 ) {
push @messages, "Importmydnsdb appears to be running already.";
push @messages, "Please wait for the import to finish before attempting another.";
send_notification(
{
action => "already_running",
result => "failure"
}
);
exit 1;
}
# perform a sanitfy check and try to repair if possible
my ( $mydns_is_sanitize, $message ) = Cpanel::NameServer::Utils::MyDNS::DB::sanity_check();
unless ($mydns_is_sanitize) {
push @messages, "Errors have been detected with the MyDNS configuration, trying to repair";
Cpanel::SafeRun::Errors::saferunallerrors('/usr/local/cpanel/bin/build_mydns_conf');
( $mydns_is_sanitize, $message ) = Cpanel::NameServer::Utils::MyDNS::DB::sanity_check();
unless ($mydns_is_sanitize) {
push @messages, "Cannot repair your MyDNS configuration, please reinstall it";
send_notification(
{
action => "cannot_repair",
result => "failure"
}
);
exit 1;
}
push @messages, "MyDNS configuration repaired successfully";
}
my $namedconf_obj = Cpanel::NameServer::Conf::Mydns->new();
$namedconf_obj->initialize();
$namedconf_obj->check_zonedir_cache();
my $bind_dir = $namedconf_obj->{'config'}->{'zonedir'};
my $dbh;
try {
$dbh = Cpanel::MysqlUtils::Connect::get_dbi_handle( 'extra_args' => { 'PrintError' => 0 } );
}
catch {
send_notification(
{
action => "cannot_connect_to_database",
result => "failure",
additional_parameters => {
error_details => $_,
}
}
);
exit 1;
};
my $settings_ref = $namedconf_obj->{'dns_settings'};
my $soa_table = $settings_ref->{'database'} . '.' . $settings_ref->{'soa-table'};
my $dns_db = $namedconf_obj->{'dns_db'};
my $get_zones_query = qq {
SELECT origin FROM $soa_table
};
my $sth = $dbh->prepare($get_zones_query);
unless ($sth) {
send_notification(
{
action => "cannot_get_zones",
result => "failure",
additional_parameters => { error_details => $dbh->errstr }
}
);
exit 1;
}
unless ( eval { $sth->execute() } ) {
send_notification(
{
action => "cannot_get_zones",
result => "failure",
additional_parameters => { error_details => $dbh->errstr }
}
);
exit 1;
}
my %existing_zones = ();
while ( my $data = $sth->fetchrow_hashref() ) {
$data->{'origin'} =~ s/\.$//;
$existing_zones{ $data->{'origin'} } = 1;
}
$sth->finish;
my $zone_dh;
unless ( opendir $zone_dh, $bind_dir ) {
send_notification(
{
action => "cannot_read_zone_file",
result => "failure",
additional_parameters => {
zone_file_dir => $bind_dir,
error_details => scalar $!
}
}
);
exit 1;
}
my @zonedir_contents = readdir($zone_dh);
closedir $zone_dh;
my $sth_put_soa = $dbh->prepare( $dns_db->{put_soa_query} );
unless ($sth_put_soa) {
send_notification(
{
action => "database_error",
result => "failure",
additional_parameters => {
error_details => $dbh->errstr,
}
}
);
exit 1;
}
my $sth_get_soa = $dbh->prepare( $dns_db->{get_soa_id} );
unless ($sth_get_soa) {
send_notification(
{
action => "database_error",
result => "failure",
additional_parameters => {
error_details => $dbh->errstr,
}
}
);
exit 1;
}
my $sth_put_rr = $dbh->prepare( $dns_db->{put_rr_query} );
unless ($sth_put_rr) {
send_notification(
{
action => "database_error",
result => "failure",
additional_parameters => {
error_details => $dbh->errstr,
}
}
);
exit 1;
}
if ($force) {
unless ( $dns_db->purge_rss() ) {
send_notification(
{
action => "cannot_purge_rss",
result => "failure"
}
);
exit 1;
}
}
my %needed_zone_files = ();
push @messages, "Checking zones in MyDNS database";
foreach my $zonefile (@zonedir_contents) {
if ( $zonefile =~ /^([\w\-.]+)\.db$/ && Cpanel::Validate::Domain::Tiny::validdomainname($1) ) {
my $zone = $1;
if ( $force || !$existing_zones{$zone} ) {
$needed_zone_files{$zonefile} = $zone;
next;
}
if ($debug) {
push @messages, "$zone already exists in MyDNS, not imported";
}
}
}
push @messages, "Done";
push @messages, "Importing zones into MyDNS database";
my $counter = 0;
foreach my $zonefile ( keys %needed_zone_files ) {
my ( $zone_fh, $zonedata );
unless ( open $zone_fh, '<', $bind_dir . '/' . $zonefile ) {
push @messages, "Unable to read $bind_dir" . '/' . $zonefile;
next;
}
{
local $/;
$zonedata = readline($zone_fh);
}
close $zone_fh;
my ( $soa_hr, $rr_ar ) = Cpanel::NameServer::Utils::MyDNS::parsebind( $zonedata, $needed_zone_files{$zonefile} );
unless ( scalar keys(%$soa_hr) ) {
push @messages, "Unable to parse and save $needed_zone_files{$zonefile}\n";
next;
}
my $ns = ( $soa_hr->{ns} =~ m/\.$/ ) ? $soa_hr->{ns} : $soa_hr->{ns} . '.';
my $origin = ( $soa_hr->{origin} =~ m/\.$/ ) ? $soa_hr->{origin} : $soa_hr->{origin} . '.';
my $serial = ( defined $soa_hr->{serial} ) ? $soa_hr->{serial} : Cpanel::DNSLib::MydnsDb::get_new_serial();
eval {
unless ( eval { $sth_put_soa->execute( $origin, $ns, $soa_hr->{mbox}, $serial, $soa_hr->{refresh}, $soa_hr->{retry}, $soa_hr->{expire}, $soa_hr->{minimum}, $soa_hr->{ttl}, 'Y' ) } ) {
push @messages, "Database error, unable to save soa records for $needed_zone_files{$zonefile}";
next;
}
$sth_put_soa->finish();
};
if ($@) {
push @messages, "Database error, saving soa record failed: $@";
next;
}
my $soa_id;
eval {
unless ( eval { $sth_get_soa->execute($origin) } ) {
push @messages, "Database error, unable to get soa id for $needed_zone_files{$zonefile}";
next;
}
my $result = $sth_get_soa->fetchrow_hashref();
$soa_id = $result->{'id'};
$sth_get_soa->finish();
};
if ($@) {
push @messages, "Database error, fetching soa record failed: $@";
next;
}
eval {
foreach my $rr_data ( @{$rr_ar} ) {
my $aux = ( exists $rr_data->{aux} && $rr_data->{aux} =~ m/^\d+$/ ) ? $rr_data->{aux} : 0;
unless ( eval { $sth_put_rr->execute( $soa_id, $rr_data->{name}, $rr_data->{type}, $aux, $rr_data->{data}, $rr_data->{ttl}, 'Y' ) } ) {
push @messages, "Database error, unable to save $rr_data->{name} for $needed_zone_files{$zonefile}";
}
$sth_put_rr->finish();
}
};
if ($@) {
push @messages, "Database error, inserting rr record failed: $@";
}
$counter++;
if ( $counter % 100000 == 0 ) {
send_notification(
{
action => "zones_imported_successfully",
result => "in_progress",
additional_parameters => { imported_count => $counter }
}
);
}
if ($debug) {
push @messages, "$needed_zone_files{$zonefile} is successfully saved";
}
}
push @messages, "$counter zones have been successfully imported";
unless ($force) {
}
my $mydnschkbin = 0;
if ( -x $namedconf_obj->{'mydnschkbin'} ) {
push @messages, Cpanel::SafeRun::Errors::saferunallerrors( $namedconf_obj->{'mydnschkbin'}, '--consistency-only' );
push @messages, 'Zone consistency check is done';
$mydnschkbin = 1;
}
else {
push @messages, 'There is no mydnscheck in your system, zone consistency check is skipped';
}
send_notification(
{
action => 'successful_completion',
result => 'success',
additional_parameters => {
mydnschkbin => $mydnschkbin,
force => $force,
imported_count => $counter,
log_ => \@messages,
}
}
);
$pid_obj->remove_pid_file();
exit 0;
sub usage {
print <<EO_USAGE;
Usage: importmydnsdb [options]
Options:
--help Brief help message
--force Overwrite existing MyDNS records. This will cause downtime for DNS while records are rebuilt.
--debug Prints extra processing information
EO_USAGE
exit 0;
}
sub send_notification {
my ($obj) = @_;
if ( $obj->{'result'} eq 'success' ) {
my $log_file = join( "\n", @{ $obj->{'additional_parameters'}->{'log_'} } );
require Cpanel::Notify;
return Cpanel::Notify::notification_class(
'class' => 'ImportMyDNSdb::Success',
'application' => 'ImportMyDNSdb::Success',
'constructor_args' => [
action => $obj->{'action'},
mydnschkbin => $mydnschkbin,
attach_files => [ { name => 'import.log.txt', content => \$log_file } ],
%{ $obj->{'additional_parameters'} },
]
);
}
elsif ( $obj->{'result'} eq 'in_progress' ) {
require Cpanel::Notify;
return Cpanel::Notify::notification_class(
'class' => 'ImportMyDNSdb::InProgress',
'application' => 'ImportMyDNSdb::InProgress',
'constructor_args' => [
action => $obj->{'action'},
%{ $obj->{'additional_parameters'} },
]
);
}
else {
require Cpanel::Notify;
return Cpanel::Notify::notification_class(
'class' => 'ImportMyDNSdb::Failure',
'application' => 'ImportMyDNSdb::Failure',
'constructor_args' => [
action => $obj->{'action'},
%{ $obj->{'additional_parameters'} },
]
);
}
}
Copyright 2K16 - 2K18 Indonesian Hacker Rulez