X-Git-Url: https://git.sesse.net/?p=pr0n;a=blobdiff_plain;f=perl%2FSesse%2Fpr0n%2FWebDAV.pm;h=ebef01022e54788e7136dcf0961633edfa8b6ac0;hp=49be2398aa4a5701c4e2ef720f63bb45af803b53;hb=290a2bdcb92997fd2537603dee5c67350b2bb7b1;hpb=6be9a26e71fc85e69cbbdbc95fc50d3569050921 diff --git a/perl/Sesse/pr0n/WebDAV.pm b/perl/Sesse/pr0n/WebDAV.pm index 49be239..ebef010 100644 --- a/perl/Sesse/pr0n/WebDAV.pm +++ b/perl/Sesse/pr0n/WebDAV.pm @@ -11,7 +11,7 @@ use Apache2::Upload; sub handler { my $r = shift; my $dbh = Sesse::pr0n::Common::get_dbh(); - + $r->headers_out->{'DAV'} = "1,2"; # We only handle depth=0, depth=1 (cf. the RFC) @@ -109,7 +109,7 @@ EOF dberror($r, "Couldn't get events"); while (my $ref = $q->fetchrow_hashref()) { - my $id = $ref->{'id'}; + my $id = $ref->{'event'}; my $name = $ref->{'name'}; $name =~ s/&/\&/g; # hack :-) @@ -247,7 +247,7 @@ EOF EOF return Apache2::Const::OK; - } elsif ($r->uri =~ m#^/webdav/upload/([a-zA-Z0-9-]+)/([a-zA-Z0-9._-]+)$#) { + } elsif ($r->uri =~ m#^/webdav/upload/([a-zA-Z0-9-]+)/([a-zA-Z0-9._()-]+)$#) { # stat a single file my ($event, $filename) = ($1, $2); my ($fname, $size, $mtime); @@ -368,7 +368,7 @@ EOF } else { # check if we have a "shadow file" for this if (defined($autorename) && $autorename eq "autorename/") { - my $ref = $dbh->selectrow_hashref('SELECT id FROM shadow_files WHERE host=? AND event=? AND filename=? AND expires_at > now()', + my $ref = $dbh->selectrow_hashref('SELECT id FROM shadow_files WHERE vhost=? AND event=? AND filename=? AND expires_at > now()', undef, $r->get_server_name, $event, $filename); if (defined($ref)) { ($fname, $size, $mtime) = Sesse::pr0n::Common::stat_image_from_id($r, $ref->{'id'}); @@ -405,12 +405,15 @@ EOF my ($event, $autorename, $filename) = ($1, $2, $3); my $size = $r->headers_in->{'content-length'}; + if (!defined($size)) { + $size = $r->headers_in->{'x-expected-entity-length'}; + } my $orig_filename = $filename; # Remove evil characters - if ($filename =~ /[^a-zA-Z0-9._-]/) { + if ($filename =~ /[^a-zA-Z0-9._()-]/) { if (defined($autorename) && $autorename eq "autorename/") { - $filename =~ tr/a-zA-Z0-9.-/_/c; + $filename =~ tr/a-zA-Z0-9.()-/_/c; } else { $r->status(403); $r->content_type('text/plain; charset=utf-8'); @@ -423,11 +426,11 @@ EOF # gnome-vfs and mac os x love to make zero-byte files, # make them happy # - if ($r->headers_in->{'content-length'} == 0) { + if ($size == 0 || $filename =~ /^\.(_|DS_Store)/) { $dbh->do('DELETE FROM fake_files WHERE expires_at <= now() OR (event=? AND vhost=? AND filename=?);', undef, $event, $r->get_server_name, $filename) or dberror($r, "Couldn't prune fake_files"); - $dbh->do('INSERT INTO fake_files (vhost,event,filename,expires_at) VALUES (?,?,?,now() + interval \'30 seconds\');', + $dbh->do('INSERT INTO fake_files (vhost,event,filename,expires_at) VALUES (?,?,?,now() + interval \'1 day\');', undef, $r->get_server_name, $event, $filename) or dberror($r, "Couldn't add file"); $r->content_type('text/plain; charset="utf-8"'); @@ -436,7 +439,7 @@ EOF $r->log->info("Fake upload of $event/$filename"); return Apache2::Const::OK; } - + # Get the new ID my $ref = $dbh->selectrow_hashref("SELECT NEXTVAL('imageid_seq') AS id;"); my $newid = $ref->{'id'}; @@ -445,13 +448,18 @@ EOF } # Autorename if we need to - if (defined($autorename) && $autorename eq "autorename/") { - my $ref = $dbh->selectrow_hashref("SELECT COUNT(*) AS numfiles FROM images WHERE vhost=? AND event=? AND filename=?", - undef, $r->get_server_name, $event, $filename) - or dberror($r, "Couldn't check for existing files"); - if ($ref->{'numfiles'} > 0) { + $ref = $dbh->selectrow_hashref("SELECT COUNT(*) AS numfiles FROM images WHERE vhost=? AND event=? AND filename=?", + undef, $r->get_server_name, $event, $filename) + or dberror($r, "Couldn't check for existing files"); + if ($ref->{'numfiles'} > 0) { + if (defined($autorename) && $autorename eq "autorename/") { $r->log->info("Renaming $filename to $newid.jpeg"); $filename = "$newid.jpeg"; + } else { + $r->status(403); + $r->content_type('text/plain; charset=utf-8'); + $r->print("File $filename already exists in event $event, cannot overwrite"); + return Apache2::Const::OK; } } @@ -468,17 +476,16 @@ EOF $dbh->do('INSERT INTO images (id,vhost,event,uploadedby,takenby,filename) VALUES (?,?,?,?,?,?)', undef, $newid, $r->get_server_name, $event, $user, $takenby, $filename); - $dbh->do('UPDATE events SET last_update=CURRENT_TIMESTAMP WHERE vhost=? AND event=?', - undef, $r->get_server_name, $event); + Sesse::pr0n::Common::purge_cache($r, "/$event/"); # Now save the file to disk + Sesse::pr0n::Common::ensure_disk_location_exists($r, $newid); $fname = Sesse::pr0n::Common::get_disk_location($r, $newid); open NEWFILE, ">$fname" or die "$fname: $!"; my $buf; - my $content_length = $r->headers_in->{'content-length'}; - if ($r->read($buf, $content_length)) { + if ($r->read($buf, $size)) { print NEWFILE $buf or die "write($fname): $!"; } @@ -490,8 +497,10 @@ EOF # Make cache while we're at it. # Don't do it for the resource forks Mac OS X loves to upload :-( - if ($filename !~ /^\._/) { - Sesse::pr0n::Common::ensure_cached($r, $filename, $newid, -1, -1, 1, 80, 64, 320, 256, -1, -1); + if ($filename !~ /^\.(_|DS_Store)/) { + # FIXME: Ideally we'd want to ensure cache of -1x-1 here as well (for NEFs), but that would + # preclude mipmapping in its current form. + Sesse::pr0n::Common::ensure_cached($r, $filename, $newid, undef, undef, "nobox", 80, 64, 320, 256); } # OK, we got this far, commit @@ -507,12 +516,12 @@ EOF } } - # Insert a `shadow file' we can stat the next 30 secs + # Insert a `shadow file' we can stat the next day or so if (defined($autorename) && $autorename eq "autorename/") { $dbh->do('DELETE FROM shadow_files WHERE expires_at <= now() OR (vhost=? AND event=? AND filename=?);', undef, $r->get_server_name, $event, $filename) or dberror($r, "Couldn't prune shadow_files"); - $dbh->do('INSERT INTO shadow_files (vhost,event,filename,id,expires_at) VALUES (?,?,?,?,now() + interval \'30 seconds\');', + $dbh->do('INSERT INTO shadow_files (vhost,event,filename,id,expires_at) VALUES (?,?,?,?,now() + interval \'1 day\');', undef, $r->get_server_name, $event, $orig_filename, $newid) or dberror($r, "Couldn't add shadow file"); $r->log->info("Added shadow entry for $event/$filename"); @@ -562,7 +571,7 @@ EOF my $ref = $q->fetchrow_hashref; if ($ref->{'cnt'} == 0) { - my @errors = Sesse::pr0n::Common::add_new_event($dbh, $ne_id, $ne_date, $ne_desc, $r->get_server_name); + my @errors = Sesse::pr0n::Common::add_new_event($r, $dbh, $ne_id, $ne_date, $ne_desc); if (scalar @errors > 0) { die "Couldn't add new event $ne_id: " . join(', ', @errors); } @@ -602,10 +611,8 @@ EOF # Try to insert this new file eval { - $dbh->do('INSERT INTO images (id,vhost,event,uploadedby,takenby,filename) VALUES (?,?,?,?,?);', + $dbh->do('INSERT INTO images (id,vhost,event,uploadedby,takenby,filename) VALUES (?,?,?,?,?,?)', undef, $newid, $r->get_server_name, $event, $user, $takenby, $filename); - $dbh->do('UPDATE events SET last_update=CURRENT_TIMESTAMP WHERE vhost=? AND event=?', - undef, $r->get_server_name, $event); # Now save the file to disk $fname = Sesse::pr0n::Common::get_disk_location($r, $newid); @@ -622,7 +629,7 @@ EOF or die "/usr/bin/exifautotran: $!"; # Make cache while we're at it. - Sesse::pr0n::Common::ensure_cached($r, $filename, $newid, -1, -1, 1, 80, 64, 320, 256, -1, -1); + Sesse::pr0n::Common::ensure_cached($r, $filename, $newid, undef, undef, 1, 80, 64, 320, 256, -1, -1); # OK, we got this far, commit $dbh->commit; @@ -658,7 +665,8 @@ EOF } my ($event, $autorename, $filename) = ($1, $2, $3); - my $sha1 = Digest::SHA1::sha1_base64("/$event/$autorename/$filename"); + $autorename = '' if (!defined($autorename)); + my $sha1 = Digest::SHA1::sha1_base64("/$event/$autorename$filename"); $r->status(200); $r->content_type('text/xml; charset=utf-8'); @@ -705,7 +713,7 @@ EOF $dbh->do('DELETE FROM images WHERE vhost=? AND event=? AND filename=?', undef, $r->get_server_name, $event, $filename) or dberror($r, "Couldn't remove file"); - $dbh->do('UPDATE events SET last_update=CURRENT_TIMESTAMP WHERE vhost=? AND event=?', + $dbh->do('UPDATE last_picture_cache SET last_update=CURRENT_TIMESTAMP WHERE vhost=? AND event=?', undef, $r->get_server_name, $event) or dberror($r, "Couldn't invalidate cache"); $r->status(200);