(no commit message)

[git.ikiwiki.info.git] / ikiwiki-mass-rebuild

diff --git a/ikiwiki-mass-rebuild b/ikiwiki-mass-rebuild
index 1ec90b3c0d34e05dd6092e542523e0bf6d81646a..ce4e084e8f0daff8e99e221d6e0027fac70423fb 100755 (executable)
--- a/ikiwiki-mass-rebuild
+++ b/ikiwiki-mass-rebuild
@@ -1,48 +1,47 @@
 #!/usr/bin/perl
 #!/usr/bin/perl

+no lib '.';

 use warnings;
 use strict;
 
 use warnings;
 use strict;
 

+my $etcfile="/etc/ikiwiki/wikilist";
+
+sub root {
+       $> == 0;
+}
+
+sub username {
+       (getpwuid($>))[0];
+}
+

 sub processline {
 sub processline {

-       my $user=shift;

        my $setup=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;
        }
        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];
-               $)="$ugid $ugid";
-               $(=$ugid;
-               $<=$uuid;
-               $>=$uuid;
-               if ($< != $uuid || $> != $uuid || $( != $ugid || $) ne "$ugid $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;
+       return 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;
        }
 }
 
 sub processlist {
        my $file=shift;

-       my $forceuser=shift;
+
+       return unless -e $file;

 
        my $list;
        open ($list, "<$file") || die "$file: $!";
 
        my $list;
        open ($list, "<$file") || die "$file: $!";


@@ -51,22 +50,28 @@ sub processlist {
                s/^\s+//;
                s/\s+$//;
                next if /^#/ || ! length;
                s/^\s+//;
                s/\s+$//;
                next if /^#/ || ! length;

-
-               if (/^([^\s]+)\s+([^\s]+)$/) {
+               if (/^([-\w]+)\s+([^\s]+)$/) {

                        my $user=$1;
                        my $setup=$2;
                        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 $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);

                                }
                        }
                }
                                }
                        }
                }


@@ -74,9 +79,13 @@ sub processlist {
        close $list;
 }
 
        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);