leont / anyevent-handle-udp Goto Github PK

#! perl

use strict;
use warnings FATAL   => 'all';
use Test::More tests => 2;
use AnyEvent::Handle::UDP;
use Socket qw/unpack_sockaddr_in/;
use IO::Socket::INET;

alarm 6;

{

    my $addr = inet_aton('localhost');
    socket my $socket, PF_INET, SOCK_DGRAM, 0
      or die "Unable to create socket: $!";
    connect $socket, pack_sockaddr_in(0, $addr) or die "connect: $!";
    my $sockaddr = getsockname $socket;

    my $client;
    my $cb = sub {
        is shift, "TIMEOUT", 'first client timed out';
        $client->destroy;
    };

    $client = AnyEvent::Handle::UDP->new(
        connect    => $sockaddr,
        on_recv    => $cb,
        on_error   => $cb,
        timeout    => 1,
        on_timeout => sub { $cb->("TIMEOUT"); return; },
    );

    my $cb2    = AE::cv;
    my $client2 = AnyEvent::Handle::UDP->new(
        connect    => $sockaddr,
        on_recv    => $cb2,
        on_error   => $cb2,
        timeout    => 4,
        on_timeout => sub { $cb2->("TIMEOUT"); return; },
    );

    $client->push_send('test');
    $client2->push_send('test');

    is(
        $cb2->recv,
        "TIMEOUT",
        'second client timed out'
    );
}

#! perl

use strict;
use warnings FATAL   => 'all';
use Test::More tests => 1;
use AnyEvent::Handle::UDP;
use Socket qw/unpack_sockaddr_in/;
use IO::Socket::INET;

alarm 3;

{

    my $addr = inet_aton('localhost');
    socket my $socket, PF_INET, SOCK_DGRAM, 0
      or die "Unable to create socket: $!";

    connect $socket, pack_sockaddr_in(0, $addr) or die "connect: $!";

    my $sockaddr_closed = getsockname $socket;
    close $socket or die "close: $!";

    my $cb     = AE::cv;
    my $client = AnyEvent::Handle::UDP->new(
        connect  => $sockaddr_closed,
        on_recv  => $cb,
        on_error => sub { $cb->send($_[2]) },
        timeout => 2,
        on_timeout => $cb,
    );
    $client->push_send("test");
    is($cb->recv, "Couldn't recv: Connection refused", 'Got connection refused error');
}

sub push_send {
    my ($self, $message, $to) = @_;
    my $ret = $self->_send();
    $self->_push_writer($message, $to) if not defined $ret and ($! == EAGAIN or $! == EWOULDBLOCK);
    return;
}

sub push_send {
    my ($self, $message, $to) = @_;
    my $ret = $self->_send($message, $to);
    $self->_push_writer($message, $to) if not defined $ret and ($! == EAGAIN or $! == EWOULDBLOCK);
    return;
}