IkiWiki/Wrapper.pm

   1 #!/usr/bin/perl
   2
   3 use warnings;
   4 use strict;
   5
   6 package IkiWiki;
   7
   8 sub gen_wrapper () { #{{{
   9         eval q{use Cwd 'abs_path'};
  10         $config{srcdir}=abs_path($config{srcdir});
  11         $config{destdir}=abs_path($config{destdir});
  12         my $this=abs_path($0);
  13         if (! -x $this) {
  14                 error("$this doesn't seem to be executable");
  15         }
  16
  17         if ($config{setup}) {
  18                 error("cannot create a wrapper that uses a setup file");
  19         }
  20
  21         my @params=($config{srcdir}, $config{destdir},
  22                 "--wikiname=$config{wikiname}",
  23                 "--templatedir=$config{templatedir}");
  24         push @params, "--verbose" if $config{verbose};
  25         push @params, "--rebuild" if $config{rebuild};
  26         push @params, "--nosvn" if !$config{svn};
  27         push @params, "--cgi" if $config{cgi};
  28         push @params, "--url=$config{url}" if length $config{url};
  29         push @params, "--cgiurl=$config{cgiurl}" if length $config{cgiurl};
  30         push @params, "--historyurl=$config{historyurl}" if length $config{historyurl};
  31         push @params, "--diffurl=$config{diffurl}" if length $config{diffurl};
  32         push @params, "--anonok" if $config{anonok};
  33         push @params, "--adminuser=$_" foreach @{$config{adminuser}};
  34         my $params=join(" ", @params);
  35         my $call='';
  36         foreach my $p ($this, $this, @params) {
  37                 $call.=qq{"$p", };
  38         }
  39         $call.="NULL";
  40
  41         my @envsave;
  42         push @envsave, qw{REMOTE_ADDR QUERY_STRING REQUEST_METHOD REQUEST_URI
  43                        CONTENT_TYPE CONTENT_LENGTH GATEWAY_INTERFACE
  44                        HTTP_COOKIE} if $config{cgi};
  45         my $envsave="";
  46         foreach my $var (@envsave) {
  47                 $envsave.=<<"EOF"
  48         if ((s=getenv("$var")))
  49                 asprintf(&newenviron[i++], "%s=%s", "$var", s);
  50 EOF
  51         }
  52
  53         open(OUT, ">ikiwiki-wrap.c") || error("failed to write ikiwiki-wrap.c: $!");;
  54         print OUT <<"EOF";
  55 /* A wrapper for ikiwiki, can be safely made suid. */
  56 #define _GNU_SOURCE
  57 #include <stdio.h>
  58 #include <unistd.h>
  59 #include <stdlib.h>
  60 #include <string.h>
  61
  62 extern char **environ;
  63
  64 int main (int argc, char **argv) {
  65         /* Sanitize environment. */
  66         char *s;
  67         char *newenviron[$#envsave+3];
  68         int i=0;
  69 $envsave
  70         newenviron[i++]="HOME=$ENV{HOME}";
  71         newenviron[i]=NULL;
  72         environ=newenviron;
  73
  74         if (argc == 2 && strcmp(argv[1], "--params") == 0) {
  75                 printf("$params\\n");
  76                 exit(0);
  77         }
  78
  79         execl($call);
  80         perror("failed to run $this");
  81         exit(1);
  82 }
  83 EOF
  84         close OUT;
  85         if (system("gcc", "ikiwiki-wrap.c", "-o", possibly_foolish_untaint($config{wrapper})) != 0) {
  86                 error("failed to compile ikiwiki-wrap.c");
  87         }
  88         unlink("ikiwiki-wrap.c");
  89         if (defined $config{wrappermode} &&
  90             ! chmod(oct($config{wrappermode}), possibly_foolish_untaint($config{wrapper}))) {
  91                 error("chmod $config{wrapper}: $!");
  92         }
  93         print "successfully generated $config{wrapper}\n";
  94 } #}}}
  95
  96 1