# Copyright (C) 2015 all contributors <meta@public-inbox.org>
# License: AGPLv3 or later (https://www.gnu.org/licenses/agpl-3.0.txt)
-package PublicInbox::Daemon; # empty class :p
-
-# contains common daemon code for the nntpd server.
+# contains common daemon code for the nntpd and httpd servers.
# This may be used for read-only IMAP server if we decide to implement it.
-package main;
+package PublicInbox::Daemon;
use strict;
use warnings;
use Getopt::Long qw/:config gnu_getopt no_ignore_case auto_abbrev/;
+use IO::Handle;
+use IO::Socket;
+STDOUT->autoflush(1);
+STDERR->autoflush(1);
require Danga::Socket;
require POSIX;
require PublicInbox::Listener;
my @CMD;
my $set_user;
my (@cfg_listen, $stdout, $stderr, $group, $user, $pid_file, $daemonize);
-my $worker_processes = 0;
+my $worker_processes = 1;
my @listeners;
my %pids;
my %listener_names;
my $reexec_pid;
my $cleanup;
+my ($uid, $gid);
END { $cleanup->() if $cleanup };
sub daemon_prepare ($) {
foreach my $l (@cfg_listen) {
next if $listener_names{$l}; # already inherited
- require IO::Socket::INET6; # works for IPv4, too
- my %o = (
- LocalAddr => $l,
- ReuseAddr => 1,
- Proto => 'tcp',
- );
- if (my $s = IO::Socket::INET6->new(%o)) {
+ my (%o, $sock_pkg);
+ if (index($l, '/') == 0) {
+ $sock_pkg = 'IO::Socket::UNIX';
+ eval "use $sock_pkg";
+ die $@ if $@;
+ %o = (Type => SOCK_STREAM, Peer => $l);
+ if (-S $l) {
+ my $c = $sock_pkg->new(%o);
+ if (!defined($c) && $!{ECONNREFUSED}) {
+ unlink $l or die
+"failed to unlink stale socket=$l: $!\n";
+ } # else: let the bind fail
+ }
+ $o{Local} = delete $o{Peer};
+ } else {
+ $sock_pkg = 'IO::Socket::INET6'; # works for IPv4, too
+ eval "use $sock_pkg";
+ die $@ if $@;
+ %o = (LocalAddr => $l, ReuseAddr => 1, Proto => 'tcp');
+ }
+ $o{Listen} = 1024;
+ my $prev = umask 0000;
+ my $s = eval { $sock_pkg->new(%o) };
+ warn "error binding $l: $!\n" unless $s;
+ umask $prev;
+
+ if ($s) {
$listener_names{sockname($s)} = $s;
push @listeners, $s;
- } else {
- warn "error binding $l: $!\n";
}
}
- die 'No listeners bound' unless @listeners;
+ die "No listeners bound\n" unless @listeners;
}
sub daemonize () {
- chdir '/' or die "chdir failed: $!\n";
- open(STDIN, '+<', '/dev/null') or die "redirect stdin failed: $!\n";
+ chdir '/' or die "chdir failed: $!";
+ open(STDIN, '+<', '/dev/null') or die "redirect stdin failed: $!";
return unless (defined $pid_file || defined $group || defined $user
|| $daemonize);
require Net::Server::Daemonize;
Net::Server::Daemonize::check_pid_file($pid_file) if defined $pid_file;
- my $uid = Net::Server::Daemonize::get_uid($user) if defined $user;
- my $gid;
+ $uid = Net::Server::Daemonize::get_uid($user) if defined $user;
if (defined $group) {
$gid = Net::Server::Daemonize::get_gid($group);
$gid = (split /\s+/, $gid)[0];
exit if $pid;
}
if (defined $pid_file) {
- Net::Server::Daemonize::create_pid_file($pid_file);
- if ($uid and !chown($uid, $gid, $pid_file)) {
- warn "could not chown $pid_file: $!\n";
- }
+ write_pid($pid_file);
my $unlink_pid = $$;
$cleanup = sub {
unlink_pid_file_safe_ish($unlink_pid, $pid_file);
# killing again terminates immediately:
exit unless @listeners;
+ $_->close foreach @listeners; # call Danga::Socket::close
@listeners = ();
# give slow clients 30s to finish reading/writing whatever
if ($stdout) {
open STDOUT, '>>', $stdout or
warn "failed to redirect stdout to $stdout: $!\n";
+ STDOUT->autoflush(1);
+ do_chown($stdout);
}
if ($stderr) {
open STDERR, '>>', $stderr or
warn "failed to redirect stderr to $stderr: $!\n";
+ STDERR->autoflush(1);
+ do_chown($stderr);
}
}
sub sockname ($) {
my ($s) = @_;
- my $n = getsockname($s) or return;
- my ($port, $addr);
- if (length($n) >= 28) {
- require Socket6;
- ($port, $addr) = Socket6::unpack_sockaddr_in6($n);
- } else {
- ($port, $addr) = Socket::sockaddr_in($n);
- }
- if (length($addr) == 4) {
- $n = Socket::inet_ntoa($addr)
- } else {
- $n = '['.Socket6::inet_ntop(Socket6::AF_INET6(), $addr).']';
- }
- $n .= ":$port";
+ my $addr = getsockname($s) or return;
+ my ($host, $port) = host_with_port($addr);
+ "$host:$port";
+}
+
+sub host_with_port ($) {
+ my ($addr) = @_;
+ my ($port, $host);
+
+ # this eval will die on Unix sockets:
+ eval {
+ if (length($addr) >= 28) {
+ require Socket6;
+ ($port, $host) = Socket6::unpack_sockaddr_in6($addr);
+ $host = Socket6::inet_ntop(Socket6::AF_INET6(), $host);
+ $host = "[$host]";
+ } else {
+ ($port, $host) = Socket::sockaddr_in($addr);
+ $host = Socket::inet_ntoa($host);
+ }
+ };
+ $@ ? ('127.0.0.1', 0) : ($host, $port);
}
sub inherit () {
my $end = $fds + 2; # LISTEN_FDS_START - 1
my @rv = ();
foreach my $fd (3..$end) {
- my $s = IO::Handle->new;
- $s->fdopen($fd, 'r');
+ my $s = IO::Handle->new_from_fd($fd, 'r');
if (my $k = sockname($s)) {
$listener_names{$k} = $s;
push @rv, $s;
}
unlink_pid_file_safe_ish($$, $pid_file);
$pid_file .= '.oldbin';
- Net::Server::Daemonize::create_pid_file($pid_file);
+ write_pid($pid_file);
}
my ($pid, $err) = do_fork();
unless (defined $pid) {
my $new = POSIX::SigSet->new;
$new->fillset;
my $old = POSIX::SigSet->new;
- POSIX::sigprocmask(&POSIX::SIG_BLOCK, $new, $old) or
- die "SIG_BLOCK: $!\n";
+ POSIX::sigprocmask(&POSIX::SIG_BLOCK, $new, $old) or die "SIG_BLOCK: $!";
my $pid = fork;
my $err = $!;
- POSIX::sigprocmask(&POSIX::SIG_SETMASK, $old) or
- die "SIG_SETMASK: $!\n";
+ POSIX::sigprocmask(&POSIX::SIG_SETMASK, $old) or die "SIG_SETMASK: $!";
($pid, $err);
}
return unless $pid_file;
my $file = $pid_file;
- $file =~ s/\.oldbin\z// or die "BUG: no '.oldbin' suffix in $file\n";
+ $file =~ s/\.oldbin\z// or die "BUG: no '.oldbin' suffix in $file";
unlink_pid_file_safe_ish($$, $pid_file);
$pid_file = $file;
- eval { Net::Server::Daemonize::create_pid_file($pid_file) };
+ eval { write_pid($pid_file) };
warn $@, "\n" if $@;
}
}
}
-sub master_loop ($) {
- my ($refresh) = @_;
- pipe(my ($p0, $p1)) or die "failed to create parent-pipe: $!\n";
- my %pwatch = ( fileno($p0) => sub { kill('TERM', $$) } );
- pipe(my ($r, $w)) or die "failed to create self-pipe: $!\n";
+sub master_loop {
+ pipe(my ($p0, $p1)) or die "failed to create parent-pipe: $!";
+ pipe(my ($r, $w)) or die "failed to create self-pipe: $!";
IO::Handle::blocking($w, 0);
my $set_workers = $worker_processes;
my @caught;
$worker_processes = 0;
} elsif ($s eq 'HUP') {
$worker_processes = $set_workers;
- $refresh->();
kill_workers($s);
} elsif ($s eq 'TTIN') {
if ($set_workers > $worker_processes) {
warn "failed to fork worker[$i]: $err\n";
} elsif ($pid == 0) {
$set_user->() if $set_user;
- close($_) for ($w, $r, $p1);
- Danga::Socket->AddOtherFds(%pwatch);
- return; # run normal work code
+ return $p0; # run normal work code
} else {
warn "PID=$pid is worker[$i]\n";
$pids{$pid} = $i;
sub daemon_loop ($$) {
my ($refresh, $post_accept) = @_;
- $refresh->(); # load config before forking to save work
+ my $parent_pipe;
if ($worker_processes > 0) {
- master_loop($refresh); # returns if in child process
+ $refresh->(); # preload by default
+ $parent_pipe = master_loop(); # returns if in child process
+ my $fd = fileno($parent_pipe);
+ Danga::Socket->AddOtherFds($fd => sub { kill('TERM', $$) } );
} else {
+ reopen_logs();
$set_user->() if $set_user;
$SIG{USR2} = sub { worker_quit() if upgrade() };
+ $refresh->();
}
+ $uid = $gid = undef;
reopen_logs();
$SIG{QUIT} = $SIG{INT} = $SIG{TERM} = *worker_quit;
$SIG{USR1} = *reopen_logs;
$SIG{HUP} = $refresh;
# this calls epoll_create:
- PublicInbox::Listener->new($_, $post_accept) for @listeners;
+ @listeners = map {
+ PublicInbox::Listener->new($_, $post_accept)
+ } @listeners;
Danga::Socket->EventLoop;
+ $parent_pipe = undef;
}
-sub daemon_run ($$$) {
+sub run ($$$) {
my ($default, $refresh, $post_accept) = @_;
daemon_prepare($default);
+ daemonize();
daemon_loop($refresh, $post_accept);
}
+sub do_chown ($) {
+ my ($path) = @_;
+ if (defined $uid and !chown($uid, $gid, $path)) {
+ warn "could not chown $path: $!\n";
+ }
+}
+
+sub write_pid ($) {
+ my ($path) = @_;
+ Net::Server::Daemonize::create_pid_file($path);
+ do_chown($path);
+}
+
1;