65 lines
1.7 KiB
Perl
65 lines
1.7 KiB
Perl
#!/usr/bin/perl
|
|
|
|
# Copyright (C) Internet Systems Consortium, Inc. ("ISC")
|
|
#
|
|
# SPDX-License-Identifier: MPL-2.0
|
|
#
|
|
# This Source Code Form is subject to the terms of the Mozilla Public
|
|
# License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
# file, you can obtain one at https://mozilla.org/MPL/2.0/.
|
|
#
|
|
# See the COPYRIGHT file distributed with this work for additional
|
|
# information regarding copyright ownership.
|
|
|
|
use IO::Socket;
|
|
use IO::File;
|
|
use strict;
|
|
|
|
# Ignore SIGPIPE so we won't fail if peer closes a TCP socket early
|
|
local $SIG{PIPE} = 'IGNORE';
|
|
|
|
# Flush logged output after every line
|
|
local $| = 1;
|
|
|
|
my $server_addr = "10.53.0.4";
|
|
if (@ARGV > 0) {
|
|
$server_addr = @ARGV[0];
|
|
}
|
|
|
|
my $localport = int($ENV{'PORT'});
|
|
if (!$localport) { $localport = 5300; }
|
|
|
|
my $udpsock = IO::Socket::INET->new(LocalAddr => "$server_addr",
|
|
LocalPort => $localport, Proto => "udp", Reuse => 1) or die "$!";
|
|
my $tcpsock = IO::Socket::INET->new(LocalAddr => "$server_addr",
|
|
LocalPort => $localport, Proto => "tcp", Listen => 5, Reuse => 1) or die "$!";
|
|
|
|
print "listening on $server_addr:$localport.\n";
|
|
|
|
my $pidf = new IO::File "ans.pid", "w" or die "cannot open pid file: $!";
|
|
print $pidf "$$\n" or die "cannot write pid file: $!";
|
|
$pidf->close or die "cannot close pid file: $!";;
|
|
sub rmpid { unlink "ans.pid"; exit 1; };
|
|
|
|
$SIG{INT} = \&rmpid;
|
|
$SIG{TERM} = \&rmpid;
|
|
|
|
# Main
|
|
for (;;) {
|
|
my $rin;
|
|
my $rout;
|
|
|
|
$rin = '';
|
|
vec($rin, fileno($udpsock), 1) = 1;
|
|
vec($rin, fileno($tcpsock), 1) = 1;
|
|
|
|
select($rout = $rin, undef, undef, undef);
|
|
|
|
if (vec($rout, fileno($udpsock), 1)) {
|
|
printf "UDP request\n";
|
|
my $buf;
|
|
$udpsock->recv($buf, 512);
|
|
} elsif (vec($rout, fileno($tcpsock), 1)) {
|
|
printf "TCP request\n";
|
|
}
|
|
}
|