ikiwiki-mass-rebuild

   1 #!/usr/bin/perl
   2 use warnings;
   3 use strict;
   4
   5 sub processline {
   6         my $user=shift;
   7         my $setup=shift;
   8
   9         if (! getpwnam("$user")) {
  10                 print STDERR "warning: user $user does not exist\n";
  11                 return
  12         }
  13         if (! -f "$setup") {
  14                 print STDERR "warning: $setup does not exist, skipping\n";
  15                 return;
  16         }
  17         print "Processing $setup as user $user ...\n";
  18         # su is not used because it passes arguments through the shell,
  19         # which is not safe for untrusted setup file names.
  20         defined(my $pid = fork) or die "Can’t fork: $!";
  21         if (! $pid) {
  22                 my ($uuid, $ugid) = (getpwnam($user))[2, 3];
  23                 $)="$ugid $ugid";
  24                 $(=$ugid;
  25                 $<=$uuid;
  26                 $>=$uuid;
  27                 if ($< != $uuid || $> != $uuid || $( != $ugid || $) ne "$ugid $ugid") {
  28                         die "failed to drop permissions to $user";
  29                 }
  30                 %ENV=(
  31                         PATH => $ENV{PATH},
  32                         HOME => (getpwnam($user))[7],
  33                 );
  34                 exec("ikiwiki", "-setup", $setup, @ARGV);
  35                 die "failed to run ikiwiki: $!";
  36         }
  37         waitpid($pid,0);
  38         if ($?) {
  39                 print STDERR "Processing $setup as user $user failed with code $?\n";
  40         }
  41 }
  42
  43 sub processlist {
  44         my $file=shift;
  45         my $forceuser=shift;
  46
  47         my $list;
  48         open ($list, "<$file") || die "$file: $!";
  49         while (<$list>) {
  50                 chomp;
  51                 s/^\s+//;
  52                 s/\s+$//;
  53                 next if /^#/ || ! length;
  54
  55                 if (/^([^\s]+)\s+([^\s]+)$/) {
  56                         my $user=$1;
  57                         my $setup=$2;
  58                         if (defined $forceuser && $forceuser ne $user) {
  59                                 print STDERR "warning: in $file line $., attempt to set user to $user, but user forced to $forceuser. Skipping\n";
  60                         }
  61                         processline($user, $setup);
  62                 }
  63                 elsif (/^([^\s]+)$/) {
  64                         my $user=$1;
  65                         my $home=(getpwnam($user))[7];
  66                         if (defined $home && -d $home) {
  67                                 my $dotfile="$home/.ikiwiki/wikilist";
  68                                 if (-e $dotfile) {
  69                                         processlist($dotfile, $user);
  70                                 }
  71                         }
  72                 }
  73         }
  74         close $list;
  75 }
  76
  77 my $wikilist="/etc/ikiwiki/wikilist";
  78
  79 if (-e $wikilist) {
  80         processlist($wikilist);
  81 }
  82