aboutsummaryrefslogtreecommitdiff
path: root/ikiwiki-mass-rebuild
diff options
context:
space:
mode:
Diffstat (limited to 'ikiwiki-mass-rebuild')
-rwxr-xr-xikiwiki-mass-rebuild117
1 files changed, 46 insertions, 71 deletions
diff --git a/ikiwiki-mass-rebuild b/ikiwiki-mass-rebuild
index f13033e7f..bbef4946b 100755
--- a/ikiwiki-mass-rebuild
+++ b/ikiwiki-mass-rebuild
@@ -2,80 +2,45 @@
use warnings;
use strict;
-sub supplemental_groups {
- my $user=shift;
-
- my @list;
- while (my @fields=getgrent()) {
- if (grep { $_ eq $user } split(' ', $fields[3])) {
- push @list, $fields[2];
- }
- }
+my $etcfile="/etc/ikiwiki/wikilist";
- return @list;
+sub root {
+ $> == 0;
}
-sub samelists {
- my %a=map { $_ => 1 } split(' ', shift());
- my %b=map { $_ => 1 } split(' ', shift());
-
- foreach my $i (keys %b) {
- if (! exists $a{$i}) {
- return 0;
- }
- }
- foreach my $i (keys %a) {
- if (! exists $b{$i}) {
- return 0;
- }
- }
- return 1;
+sub username {
+ (getpwuid($>))[0];
}
sub processline {
- my $user=shift;
my $setup=shift;
- if (! getpwnam("$user")) {
- print STDERR "warning: user $user does not exist\n";
- return
- }
if (! -f "$setup") {
print STDERR "warning: $setup does not exist, skipping\n";
return;
}
- print "Processing $setup as user $user ...\n";
- # su is not used because it passes arguments through the shell,
- # which is not safe for untrusted setup file names.
- defined(my $pid = fork) or die "Can’t fork: $!";
- if (! $pid) {
- my ($uuid, $ugid) = (getpwnam($user))[2, 3];
- my $grouplist=join(" ", $ugid, sort {$a <=> $b} $ugid, supplemental_groups($user));
- if (! samelists(($)=$grouplist), $grouplist)) {
- die "failed to set egid $grouplist (got back $))";
- }
- $(=$ugid;
- $<=$uuid;
- $>=$uuid;
- if ($< != $uuid || $> != $uuid || $( != $ugid) {
- die "failed to drop permissions to $user";
- }
- %ENV=(
- PATH => $ENV{PATH},
- HOME => (getpwnam($user))[7],
- );
- exec("ikiwiki", "-setup", $setup, @ARGV);
- die "failed to run ikiwiki: $!";
+ print "Processing $setup as user ".username()." ...\n";
+ my $ret=system("ikiwiki", "-setup", $setup, @ARGV);
+ if ($ret != 0) {
+ print STDERR "warning: processing $setup failed with code $ret\n";
}
- waitpid($pid,0);
- if ($?) {
- print STDERR "Processing $setup as user $user failed with code $?\n";
+}
+
+my %users;
+sub processuser {
+ my $user=shift;
+ next if $user=~/^-/ || $users{$user};
+ $users{$user}=1;
+ my $ret=system("su", $user, "-s", "/bin/sh", "-c", "--", "$0 --nonglobal @ARGV");
+ if ($ret != 0) {
+ print STDERR "warning: processing for $user failed with code $ret\n";
}
}
sub processlist {
my $file=shift;
- my $forceuser=shift;
+
+ return unless -e $file;
my $list;
open ($list, "<$file") || die "$file: $!";
@@ -84,22 +49,28 @@ sub processlist {
s/^\s+//;
s/\s+$//;
next if /^#/ || ! length;
-
- if (/^([^\s]+)\s+([^\s]+)$/) {
+ if (/^([-\w]+)\s+([^\s]+)$/) {
my $user=$1;
my $setup=$2;
- if (defined $forceuser && $forceuser ne $user) {
- print STDERR "warning: in $file line $., attempt to set user to $user, but user forced to $forceuser. Skipping\n";
+ if (root()) {
+ processuser($user);
+ }
+ else {
+ if (username() eq $user) {
+ processline($setup);
+ }
}
- processline($user, $setup);
}
- elsif (/^([^\s]+)$/) {
+ elsif (/^([-\w]+)$/) {
my $user=$1;
- my $home=(getpwnam($user))[7];
- if (defined $home && -d $home) {
- my $dotfile="$home/.ikiwiki/wikilist";
- if (-e $dotfile) {
- processlist($dotfile, $user);
+ if (root()) {
+ processuser($user);
+ }
+ else {
+ my $home=(getpwnam($user))[7];
+ if (defined $home && -d $home) {
+ my $dotfile="$home/.ikiwiki/wikilist";
+ processlist($dotfile);
}
}
}
@@ -107,9 +78,13 @@ sub processlist {
close $list;
}
-my $wikilist="/etc/ikiwiki/wikilist";
-
-if (-e $wikilist) {
- processlist($wikilist);
+if (@ARGV && $ARGV[0] eq "--nonglobal") {
+ shift;
+ # avoid recursively processing if the wikilist file has a root
+ # user in it
+ if (root()) {
+ exit 1;
+ }
}
+processlist($etcfile);