125 lines
3.4 KiB
Perl
125 lines
3.4 KiB
Perl
package StartXDummy;
|
||
# vim:ts=4:sw=4:expandtab
|
||
|
||
use strict;
|
||
use warnings;
|
||
use Exporter 'import';
|
||
use Time::HiRes qw(sleep);
|
||
use v5.10;
|
||
|
||
our @EXPORT = qw(start_xdummy);
|
||
|
||
my @pids;
|
||
my $x_socketpath = '/tmp/.X11-unix/X';
|
||
|
||
# reads in a whole file
|
||
sub slurp {
|
||
open(my $fh, '<', shift) or return '';
|
||
local $/;
|
||
<$fh>;
|
||
}
|
||
|
||
# forks an Xdummy or Xdmx process
|
||
sub fork_xserver {
|
||
my $keep_xdummy_output = shift;
|
||
my $displaynum = shift;
|
||
my $pid = fork();
|
||
die "Could not fork: $!" unless defined($pid);
|
||
if ($pid == 0) {
|
||
# Child, close stdout/stderr, then start Xdummy.
|
||
if (!$keep_xdummy_output) {
|
||
close STDOUT;
|
||
close STDERR;
|
||
}
|
||
|
||
exec @_;
|
||
exit 1;
|
||
}
|
||
push(@complete_run::CLEANUP, sub {
|
||
kill(15, $pid);
|
||
# Unlink the X11 socket, Xdmx seems to leave it there.
|
||
unlink($x_socketpath . $displaynum);
|
||
});
|
||
|
||
push @pids, $pid;
|
||
|
||
return $x_socketpath . $displaynum;
|
||
}
|
||
|
||
# Blocks until the socket paths specified in the given array reference actually
|
||
# exist.
|
||
sub wait_for_x {
|
||
my ($sockets_waiting) = @_;
|
||
|
||
# Wait until Xdmx actually runs. Pretty ugly solution, but as long as we
|
||
# can’t socket-activate X11…
|
||
while (1) {
|
||
@$sockets_waiting = grep { ! -S $_ } @$sockets_waiting;
|
||
last unless @$sockets_waiting;
|
||
sleep 0.1;
|
||
}
|
||
}
|
||
|
||
=head2 start_xdummy($parallel)
|
||
|
||
Starts C<$parallel> (or number of cores * 2 if undef) Xdummy processes (see
|
||
the file ./Xdummy) and returns two arrayrefs: a list of X11 display numbers to
|
||
the Xdummy processes and a list of PIDs of the processes.
|
||
|
||
=cut
|
||
|
||
sub start_xdummy {
|
||
my ($parallel, $numtests, $keep_xdummy_output) = @_;
|
||
|
||
my @displays = ();
|
||
my @childpids = ();
|
||
|
||
$SIG{CHLD} = sub {
|
||
my $child = waitpid -1, POSIX::WNOHANG;
|
||
@pids = grep { $_ != $child } @pids;
|
||
return unless @pids == 0;
|
||
print STDERR "All Xdummy processes died.\n";
|
||
print STDERR "Use ./complete-run.pl --parallel 1 --keep-xdummy-output\n";
|
||
exit 1;
|
||
};
|
||
|
||
# Yeah, I know it’s non-standard, but Perl’s POSIX module doesn’t have
|
||
# _SC_NPROCESSORS_CONF.
|
||
my $cpuinfo = slurp('/proc/cpuinfo');
|
||
my $num_cores = scalar grep { /model name/ } split("\n", $cpuinfo);
|
||
# If /proc/cpuinfo does not exist, we fall back to 2 cores.
|
||
$num_cores ||= 2;
|
||
|
||
# If unset, we use num_cores * 2.
|
||
$parallel ||= ($num_cores * 2);
|
||
|
||
# If we are running a small number of tests, don’t over-parallelize.
|
||
$parallel = $numtests if $numtests < $parallel;
|
||
|
||
# First get the last used display number, then increment it by one.
|
||
# Effectively falls back to 1 if no X server is running.
|
||
my ($displaynum) = map { /(\d+)$/ } reverse sort glob($x_socketpath . '*');
|
||
$displaynum++;
|
||
|
||
say "Starting $parallel Xdummy instances, starting at :$displaynum...";
|
||
|
||
my @sockets_waiting;
|
||
for (1 .. $parallel) {
|
||
# We use -config /dev/null to prevent Xdummy from using the system
|
||
# Xorg configuration. The tests should be independant from the
|
||
# actual system X configuration.
|
||
my $socket = fork_xserver($keep_xdummy_output, $displaynum,
|
||
'./Xdummy', ":$displaynum", '-config', '/dev/null',
|
||
'-nolisten', 'tcp');
|
||
push(@displays, ":$displaynum");
|
||
push(@sockets_waiting, $socket);
|
||
$displaynum++;
|
||
}
|
||
|
||
wait_for_x(\@sockets_waiting);
|
||
|
||
return @displays;
|
||
}
|
||
|
||
1
|