summaryrefslogtreecommitdiffstats
path: root/lib/Select/Retry.pm
blob: 11a79e417c1a6cf1c3cec8d6fb194213a39ff760 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
package Select::Retry;
use strict;
use warnings;

use Sub::Exporter -setup => {
    exports => [ 'retry_select' ],
    groups => { default => [ 'retry_select' ] },
};

sub retry_select {
    my ($mode, $timeout, @handles) = @_;

    my $got_winch;
    my $old_winch = $SIG{WINCH};
    local $SIG{WINCH} = sub {
        $got_winch = 1;
        $old_winch->() if ref($old_winch) && ref($old_winch) eq 'CODE';
    };

    my ($out, $eout);
    my ($in, $ein) = (_build_select_vec(@handles)) x 2;
    my $res;
    if ($mode eq 'r') {
        $res = select($out = $in, undef, $eout = $ein, $timeout);
    }
    else {
        $res = select(undef, $out = $in, $eout = $ein, $timeout);
    }
    my $again = $!{EAGAIN} || $!{EINTR};

    if (($res == -1 && $again) || $got_winch) {
        return retry_select(@_);
    }
    elsif ($res == -1) {
        Carp::croak("select failed: $!");
    }

    return ($out, $eout);
}

sub _build_select_vec {
    my @handles = @_;

    my $vec = '';
    for my $handle (@handles) {
        vec($vec, fileno($handle), 1) = 1;
    }

    return $vec;
}

1;