Kharec/urupam
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: add DNS cache and subprocess resolution

Browse Source
This commit is contained in:
Sandro CURY CAZZANIGA
2025-12-29 16:11:24 +01:00
parent 39bead9da1
commit e6fc9c919f
1 changed files with 59 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

57
lib/Urupam/Validation.pm
View File

@@ -4,6 +4,7 @@ use Mojo::Base -base;
use Mojo::URL; use Mojo::URL;
use Mojo::UserAgent; use Mojo::UserAgent;
use Mojo::Promise; use Mojo::Promise;
use Mojo::IOLoop;
use Urupam::Utils qw(sanitize_url); use Urupam::Utils qw(sanitize_url);
use Socket use Socket
qw(getaddrinfo getnameinfo NI_NUMERICHOST NI_NUMERICSERV AF_INET AF_INET6 SOCK_STREAM); qw(getaddrinfo getnameinfo NI_NUMERICHOST NI_NUMERICSERV AF_INET AF_INET6 SOCK_STREAM);
@@ -25,6 +26,9 @@ my @BLOCKED_DOMAINS = qw(
localhost 127.0.0.1 0.0.0.0 ::1 localhost 127.0.0.1 0.0.0.0 ::1
); );
my $DNS_CACHE_TTL = 300;
my %dns_cache;
has ua => sub { has ua => sub {
my $self = shift; my $self = shift;
Mojo::UserAgent->new( Mojo::UserAgent->new(
@@ -146,25 +150,64 @@ sub _resolve_host {
[ { type => 'ipv6', ip => $ipv6_host } ] ); [ { type => 'ipv6', ip => $ipv6_host } ] );
} }
my $cache_key = lc($host);
my $now = time();
if ( exists $dns_cache{$cache_key} ) {
my $cached = $dns_cache{$cache_key};
if ( $now < $cached->{expires} ) {
return Mojo::Promise->resolve( $cached->{addresses} );
}
delete $dns_cache{$cache_key};
}
my $promise = Mojo::Promise->new;
Mojo::IOLoop->subprocess(
sub {
my ($hostname) = @_;
my ( $err, @results ) = my ( $err, @results ) =
getaddrinfo( $host, undef, { socktype => SOCK_STREAM } ); getaddrinfo( $hostname, undef, { socktype => SOCK_STREAM } );
return Mojo::Promise->resolve( [] ) if $err; return { error => $err, results => \@results };
},
sub {
my ( $subprocess, $err, $data ) = @_;
if ($err) {
$promise->resolve( [] );
return;
}
my $res = $data;
if ( $res->{error} ) {
$promise->resolve( [] );
return;
}
my @addresses; my @addresses;
for my $res (@results) { for my $result ( @{ $res->{results} } ) {
my ( $hostnum, undef ) = my ( $hostnum, undef ) =
getnameinfo( $res->{addr}, NI_NUMERICHOST | NI_NUMERICSERV ); getnameinfo( $result->{addr},
NI_NUMERICHOST | NI_NUMERICSERV );
next unless defined $hostnum && length $hostnum; next unless defined $hostnum && length $hostnum;
if ( $res->{family} == AF_INET ) { if ( $result->{family} == AF_INET ) {
push @addresses, { type => 'ipv4', ip => $hostnum }; push @addresses, { type => 'ipv4', ip => $hostnum };
} }
elsif ( $res->{family} == AF_INET6 ) { elsif ( $result->{family} == AF_INET6 ) {
push @addresses, { type => 'ipv6', ip => $hostnum }; push @addresses, { type => 'ipv6', ip => $hostnum };
} }
} }
return Mojo::Promise->resolve( \@addresses ); my $addresses_ref = \@addresses;
$dns_cache{$cache_key} = {
addresses => $addresses_ref,
expires => $now + $DNS_CACHE_TTL
};
$promise->resolve($addresses_ref);
},
$host
);
return $promise;
} }
sub is_blocked_url { sub is_blocked_url {