Complete rcs_preprevert and lightly test.
[ikiwiki] / IkiWiki / Receive.pm
1 #!/usr/bin/perl
2 package IkiWiki::Receive;
3
4 use warnings;
5 use strict;
6 use IkiWiki;
7
8 sub getuser () {
9         my $user=(getpwuid(exists $ENV{CALLER_UID} ? $ENV{CALLER_UID} : $<))[0];
10         if (! defined $user) {
11                 error("cannot determine username for $<");
12         }
13         return $user;
14 }
15
16 sub trusted () {
17         my $user=getuser();
18         return ! ref $config{untrusted_committers} ||
19                 ! grep { $_ eq $user } @{$config{untrusted_committers}};
20 }
21
22 sub genwrapper () {
23         # Test for commits from untrusted committers in the wrapper, to
24         # avoid starting ikiwiki proper at all for trusted commits.
25
26         my $ret=<<"EOF";
27         {
28                 int u=getuid();
29 EOF
30         $ret.="\t\tif ( ".
31                 join("&&", map {
32                         my $uid=getpwnam($_);
33                         if (! defined $uid) {
34                                 error(sprintf(gettext("cannot determine id of untrusted committer %s"), $_));
35                         }
36                         "u != $uid";
37                 } @{$config{untrusted_committers}}).
38                 ") exit(0);\n";
39
40         
41         $ret.=<<"EOF";
42                 asprintf(&s, "CALLER_UID=%i", u);
43                 newenviron[i++]=s;
44         }
45 EOF
46         return $ret;
47 }
48
49 sub test () {
50         exit 0 if trusted();
51
52         IkiWiki::lockwiki();
53         IkiWiki::loadindex();
54
55         # Dummy up a cgi environment to use when calling check_canedit
56         # and friends.
57         eval q{use CGI};
58         error($@) if $@;
59         my $cgi=CGI->new;
60
61         # And dummy up a session object.
62         require IkiWiki::CGI;
63         my $session=IkiWiki::cgi_getsession($cgi);
64         $session->param("name", getuser());
65         # Make sure whatever user was authed is in the
66         # userinfo db.
67         require IkiWiki::UserInfo;
68         if (! IkiWiki::userinfo_get($session->param("name"), "regdate")) {
69                 IkiWiki::userinfo_setall($session->param("name"), {
70                         email => "",
71                         password => "",
72                         regdate => time,
73                 }) || error("failed adding user");
74         }
75
76         test_changes(cgi => $cgi,
77                      session => $session,
78                      changes => [IkiWiki::rcs_receive()]
79                     );
80         exit 0;
81 }
82
83 sub test_changes {
84         my %params = @_;
85         my $cgi = $params{cgi};
86         my $session = $params{session};
87         my @changes = @{$params{changes}};
88
89         my %newfiles;
90         foreach my $change (@changes) {
91                 # This untaint is safe because we check file_pruned and
92                 # wiki_file_regexp.
93                 my ($file)=$change->{file}=~/$config{wiki_file_regexp}/;
94                 $file=IkiWiki::possibly_foolish_untaint($file);
95                 if (! defined $file || ! length $file ||
96                     IkiWiki::file_pruned($file)) {
97                         error(gettext("bad file name %s"), $file);
98                 }
99
100                 my $type=pagetype($file);
101                 my $page=pagename($file) if defined $type;
102
103                 if ($change->{action} eq 'add') {
104                         $newfiles{$file}=1;
105                 }
106
107                 if ($change->{action} eq 'change' ||
108                     $change->{action} eq 'add') {
109                         if (defined $page) {
110                                 IkiWiki::check_canedit($page, $cgi, $session);
111                                 next;
112                         }
113                         else {
114                                 if (IkiWiki::Plugin::attachment->can("check_canattach")) {
115                                         IkiWiki::Plugin::attachment::check_canattach($session, $file, $change->{path});
116                                         IkiWiki::check_canedit($file, $cgi, $session);
117                                         next;
118                                 }
119                                 else {
120                                 use Data::Dumper;
121                                 die "fall through test_changes add: " . Data::Dumper::Dumper($change);
122                                 }
123                         }
124                 }
125                 elsif ($change->{action} eq 'remove') {
126                         # check_canremove tests to see if the file is present
127                         # on disk. This will fail when a single commit adds a
128                         # file and then removes it again. Avoid the problem
129                         # by not testing the removal in such pairs of changes.
130                         # (The add is still tested, just to make sure that
131                         # no data is added to the repo that a web edit
132                         # could not add.)
133                         next if $newfiles{$file};
134
135                         if (IkiWiki::Plugin::remove->can("check_canremove")) {
136                                 IkiWiki::Plugin::remove::check_canremove(defined $page ? $page : $file, $cgi, $session);
137                                 IkiWiki::check_canedit(defined $page ? $page : $file, $cgi, $session);
138                                 next;
139                         }
140                 }
141                 else {
142                         error "unknown action ".$change->{action};
143                 }
144
145                 error sprintf(gettext("you are not allowed to change %s"), $file);
146         }
147 }
148
149 1