#!/usr/bin/perl
# $Id: perldig 1815 2020-10-14 21:55:18Z willem $

=head1 NAME

perldig - Perl script to perform DNS queries

=head1 SYNOPSIS

C<perldig> [ C<@>I<nameserver> ] I<name> [ I<type> [ I<class> ] ]

=head1 DESCRIPTION

Performs a DNS query on the given name.  The record type
and class can also be specified; if left blank they default
to A and IN.

=head1 AUTHOR

Michael Fuhr <mike@fuhr.org>

=head1 SEE ALSO

L<perl(1)>, L<axfr>, L<check_soa>, L<check_zone>, L<mresolv>, L<mx>,
L<Net::DNS>

=cut

use strict;
use warnings;

use File::Basename;
use Net::DNS;

my $res = Net::DNS::Resolver->new;

if (@ARGV && ($ARGV[0] =~ /^@/)) {
	my $nameserver = shift;
	$nameserver =~ s/^@//;
	$res->nameservers($nameserver);
}

die "Usage: ", basename($0), " [ \@nameserver ] name [ type [ class ] ]\n"
	unless (@ARGV >= 1) && (@ARGV <= 3);

my ($name, $type, $class) = @ARGV;
$type  ||= "A";
$class ||= "IN";

if (uc($type) eq "AXFR") {
	
	my @rrs = $res->axfr($name, $class);
	
	if (@rrs) {
		foreach my $rr (@rrs) {
			$rr->print;
		}		
	} else {
		die "zone transfer failed: ", $res->errorstring, "\n";
	}
	
} else {

	my $answer = $res->send($name, $type, $class);
	
	if ($answer) {
		$answer->print;
	} else {
		die "query failed: ", $res->errorstring, "\n";
	}
}
