user/dev discussion of public-inbox itself
 help / color / mirror / code / Atom feed
From: Eric Wong <e@yhbt.net>
To: meta@public-inbox.org
Subject: [PATCH 1/2] daemon: fix SIGUSR2 upgrade with -W0 (no workers)
Date: Sun, 22 Mar 2020 08:58:48 +0000	[thread overview]
Message-ID: <20200322085849.14360-2-e@yhbt.net> (raw)
In-Reply-To: <20200322085849.14360-1-e@yhbt.net>

Disabling workers via `-W0' blesses the contents of the
@listeners array, so we need to ensure we call fcntl on
the GLOB ref in ->{sock}.

Add tests to ensure USR2 works regardless of whether workers
are enabled or not.
---
 lib/PublicInbox/Daemon.pm |  3 ++
 t/httpd-unix.t            | 99 +++++++++++++++++++++++++++++++++++----
 2 files changed, 92 insertions(+), 10 deletions(-)

diff --git a/lib/PublicInbox/Daemon.pm b/lib/PublicInbox/Daemon.pm
index 43ef2691..3d582e35 100644
--- a/lib/PublicInbox/Daemon.pm
+++ b/lib/PublicInbox/Daemon.pm
@@ -403,6 +403,9 @@ sub upgrade { # $_[0] = signal name or number (unused)
 		$ENV{LISTEN_FDS} = scalar @listeners;
 		$ENV{LISTEN_PID} = $$;
 		foreach my $s (@listeners) {
+			# @listeners are globs with workers, PI::L w/o workers
+			$s = $s->{sock} if ref($s) eq 'PublicInbox::Listener';
+
 			my $fl = fcntl($s, F_GETFD, 0);
 			fcntl($s, F_SETFD, $fl &= ~FD_CLOEXEC);
 		}
diff --git a/t/httpd-unix.t b/t/httpd-unix.t
index b321c789..425ecfff 100644
--- a/t/httpd-unix.t
+++ b/t/httpd-unix.t
@@ -6,6 +6,8 @@ use warnings;
 use Test::More;
 use PublicInbox::TestCommon;
 use Errno qw(EADDRINUSE);
+use Cwd qw(abs_path);
+use Carp qw(croak);
 require_mods(qw(Plack::Util Plack::Builder HTTP::Date HTTP::Status));
 use IO::Socket::UNIX;
 my ($tmpdir, $for_destroy) = tmpdir();
@@ -79,9 +81,26 @@ check_sock($unix);
 	ok(-S $unix, 'unix socket still exists');
 }
 
+sub delay_until {
+	my $cond = shift;
+	for (1..1000) {
+		return if $cond->();
+		select undef, undef, undef, 0.012;
+	}
+	Carp::croak('condition failed');
+}
+
 SKIP: {
 	require_mods('Net::Server::Daemonize', 20);
 	my $pid_file = "$tmpdir/pid";
+	my $read_pid = sub {
+		my $f = shift;
+		open my $fh, '<', $f or die "open $f failed: $!";
+		my $pid = do { local $/; <$fh> };
+		chomp $pid;
+		$pid || 0;
+	};
+
 	for my $w (qw(-W0 -W1)) {
 		# wait for daemonization
 		$spawn_httpd->("-l$unix", '-D', '-P', $pid_file, $w);
@@ -90,18 +109,78 @@ SKIP: {
 		check_sock($unix);
 
 		ok(-f $pid_file, "$w pid file written");
-		open my $fh, '<', "$tmpdir/pid" or die "open failed: $!";
-		my $rpid = do { local $/; <$fh> };
-		chomp $rpid;
-		like($rpid, qr/\A\d+\z/s, "$w pid file looks like a pid");
-		is(kill('TERM', $rpid), 1, "signaled daemonized $w process");
-		for (1..100) {
-			kill(0, $rpid) or last;
-			select undef, undef, undef, 0.02;
-		}
-		is(kill(0, $rpid), 0, "daemonized $w process exited");
+		my $pid = $read_pid->($pid_file);
+		is(kill('TERM', $pid), 1, "signaled daemonized $w process");
+		delay_until(sub { !kill(0, $pid) });
+		is(kill(0, $pid), 0, "daemonized $w process exited");
 		ok(!-e $pid_file, "$w pid file unlinked at exit");
 	}
+
+	# try a USR2 upgrade with workers:
+	my $httpd = abs_path('blib/script/public-inbox-httpd');
+	$psgi = abs_path($psgi);
+	my $opt = { run_mode => 0 };
+
+	my @args = ("-l$unix", '-D', '-P', $pid_file, -1, $out, -2, $err);
+	$td = start_script([$httpd, @args, $psgi], undef, $opt);
+	$td->join;
+	is($?, 0, "daemonized process again");
+	check_sock($unix);
+	my $pid = $read_pid->($pid_file);
+	kill('USR2', $pid) or die "USR2 failed: $!";
+	delay_until(sub {
+		$pid != (eval { $read_pid->($pid_file) } // $pid)
+	});
+	my $new_pid = $read_pid->($pid_file);
+	isnt($new_pid, $pid, 'new child started');
+	my $old_pid = $read_pid->("$pid_file.oldbin");
+	is($old_pid, $pid, '.oldbin pid file written');
+
+	# first, back out of the upgrade
+	kill('QUIT', $new_pid) or die "kill new PID failed: $!";
+	delay_until(sub {
+		$pid == (eval { $read_pid->($pid_file) } // 0)
+	});
+	is($read_pid->($pid_file), $pid, 'old PID file restored');
+	ok(!-f "$pid_file.oldbin", '.oldbin PID file gone');
+
+	# retry USR2 upgrade
+	kill('USR2', $pid) or die "USR2 failed: $!";
+	delay_until(sub {
+		$pid != (eval { $read_pid->($pid_file) } // $pid)
+	});
+	$new_pid = $read_pid->($pid_file);
+	isnt($new_pid, $pid, 'new child started again');
+	$old_pid = $read_pid->("$pid_file.oldbin");
+	is($old_pid, $pid, '.oldbin pid file written');
+
+	# drop the old parent
+	kill('QUIT', $old_pid) or die "QUIT failed: $!";
+	delay_until(sub { !kill(0, $old_pid) });
+
+	# drop the new child
+	check_sock($unix);
+	kill('QUIT', $new_pid) or die "QUIT failed: $!";
+	delay_until(sub { !kill(0, $new_pid) });
+	ok(!-f $pid_file, 'PID file is gone');
+
+
+	# try USR2 without workers (-W0)
+	$td = start_script([$httpd, @args, '-W0', $psgi], undef, $opt);
+	$td->join;
+	is($?, 0, 'daemonized w/o workers');
+	check_sock($unix);
+	$pid = $read_pid->($pid_file);
+
+	# replace running process
+	kill('USR2', $pid) or die "USR2 failed: $!";
+	delay_until(sub { !kill(0, $pid) });
+
+	check_sock($unix);
+	$pid = $read_pid->($pid_file);
+	kill('QUIT', $pid) or die "USR2 failed: $!";
+	delay_until(sub { !kill(0, $pid) });
+	ok(!-f $pid_file, 'PID file is gone');
 }
 
 done_testing();

  reply	other threads:[~2020-03-22  8:58 UTC|newest]

Thread overview: 4+ messages / expand[flat|nested]  mbox.gz  Atom feed  top
2020-03-22  8:58 [PATCH 0/2] daemon: SIGUSR2 fixes Eric Wong
2020-03-22  8:58 ` Eric Wong [this message]
2020-03-24  7:24   ` [PATCH 3/2 (squash)] t/httpd-unix: fix race in SIGUSR2 test Eric Wong
2020-03-22  8:58 ` [PATCH 2/2] daemon: unlink .oldbin PID file correctly Eric Wong

Reply instructions:

You may reply publicly to this message via plain-text email
using any one of the following methods:

* Save the following mbox file, import it into your mail client,
  and reply-to-all from there: mbox

  Avoid top-posting and favor interleaved quoting:
  https://en.wikipedia.org/wiki/Posting_style#Interleaved_style

  List information: https://public-inbox.org/README

* Reply using the --to, --cc, and --in-reply-to
  switches of git-send-email(1):

  git send-email \
    --in-reply-to=20200322085849.14360-2-e@yhbt.net \
    --to=e@yhbt.net \
    --cc=meta@public-inbox.org \
    /path/to/YOUR_REPLY

  https://kernel.org/pub/software/scm/git/docs/git-send-email.html

* If your mail client supports setting the In-Reply-To header
  via mailto: links, try the mailto: link
Be sure your reply has a Subject: header at the top and a blank line before the message body.
Code repositories for project(s) associated with this public inbox

	https://80x24.org/public-inbox.git

This is a public inbox, see mirroring instructions
for how to clone and mirror all data and code used for this inbox;
as well as URLs for read-only IMAP folder(s) and NNTP newsgroup(s).