gnh1201/caterpillar
1
0
Fork 0
mirror of https://github.com/gnh1201/caterpillar.git synced 2025-09-09 19:29:40 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update worker.pl

Browse Source
This commit is contained in:
Namhyeon Go 2024-03-07 16:00:07 +09:00 committed by GitHub
parent 118746e2e4
commit e7137b0079
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 12 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
assets/perl/worker.pl
View File

@ -1,5 +1,9 @@
use JSON; use JSON;
use IO::Socket::INET; use IO::Socket::INET;
use Time::HiRes qw(time);
use constant DEFAULT_SOCKET_TIMEOUT => 1;
use constant STATEFUL_SOCKET_TIMEOUT => 30;
sub jsonrpc2_encode { sub jsonrpc2_encode {
my ($method, $params, $id) = @_; my ($method, $params, $id) = @_;
@ -59,7 +63,8 @@ sub read_from_remote_server {
my $sock = IO::Socket::INET->new( my $sock = IO::Socket::INET->new(
PeerAddr => $remote_address, PeerAddr => $remote_address,
PeerPort => $remote_port, PeerPort => $remote_port,
Proto => 'tcp' Proto => 'tcp',
Timeout => DEFAULT_SOCKET_TIMEOUT
); );
if (!$sock) { if (!$sock) {
@ -148,11 +153,12 @@ sub relay_connect {
my $scheme = $params->{'scheme'}; my $scheme = $params->{'scheme'};
my $datetime = $params->{'datetime'}; my $datetime = $params->{'datetime'};
my $starttime = Time::HiRes::time(); my $starttime = time();
my $conn = IO::Socket::INET->new( my $conn = IO::Socket::INET->new(
PeerAddr => $client_address, PeerAddr => $client_address,
PeerPort => $client_port, PeerPort => $client_port,
Proto => 'tcp' Proto => 'tcp',
Timeout => STATEFUL_SOCKET_TIMEOUT
); );
if (!$conn) { if (!$conn) {
my $error = { my $error = {
@ -162,7 +168,7 @@ sub relay_connect {
}; };
print jsonrpc2_error_encode($error, $id); print jsonrpc2_error_encode($error, $id);
} else { } else {
my $stoptime = Time::HiRes::time(); my $stoptime = time();
my $connection_speed = int(($stoptime - $starttime) * 1000); my $connection_speed = int(($stoptime - $starttime) * 1000);
my $data = jsonrpc2_encode("relay_accept", { my $data = jsonrpc2_encode("relay_accept", {
success => 1, success => 1,
@ -184,10 +190,10 @@ if ($context->{'jsonrpc'} eq "2.0") {
my $method = $context->{'method'}; my $method = $context->{'method'};
given ($method) { given ($method) {
when ("relay_request") { when ("relay_request") {
relay_request($context->{'params'}, $context->{'id'}); # Stateless mode relay_request($context->{'params'}, $context->{'id'}); # stateless mode
} }
when ("relay_connect") { when ("relay_connect") {
relay_connect($context->{'params'}, $context->{'id'}); # Stateful mode relay_connect($context->{'params'}, $context->{'id'}); # stateful mode
} }
} }
} }