JezK
Edit File: importmydnsdb
#!/usr/local/cpanel/3rdparty/bin/perl # cpanel - scripts/importmydnsdb 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 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'} }, ] ); } }