X-Git-Url: https://git.sesse.net/?p=pr0n;a=blobdiff_plain;f=perl%2FSesse%2Fpr0n%2FWebDAV.pm;h=71cd8ba13ea8f45763e8360d62e7ab814b67b314;hp=8c90e2fcd3fc9f94eb400ff9990e458b0b095a16;hb=a8507a180fdcb9e5e78dd689d1be1fa08401441e;hpb=38e2fbe36a79ffe51fee5e150dbb014aca7fb810 diff --git a/perl/Sesse/pr0n/WebDAV.pm b/perl/Sesse/pr0n/WebDAV.pm index 8c90e2f..71cd8ba 100644 --- a/perl/Sesse/pr0n/WebDAV.pm +++ b/perl/Sesse/pr0n/WebDAV.pm @@ -3,7 +3,7 @@ use strict; use warnings; use Sesse::pr0n::Common qw(error dberror); -use Digest::SHA1; +use Digest::SHA; use MIME::Base64; use Apache2::Request; use Apache2::Upload; @@ -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'}); @@ -408,7 +408,6 @@ EOF if (!defined($size)) { $size = $r->headers_in->{'x-expected-entity-length'}; } - $r->log->info("sz=$size"); my $orig_filename = $filename; # Remove evil characters @@ -449,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; } } @@ -472,11 +476,10 @@ 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 last_picture_cache 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: $!"; @@ -610,8 +613,6 @@ EOF eval { $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 last_picture_cache 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); @@ -665,7 +666,7 @@ EOF my ($event, $autorename, $filename) = ($1, $2, $3); $autorename = '' if (!defined($autorename)); - my $sha1 = Digest::SHA1::sha1_base64("/$event/$autorename$filename"); + my $sha1 = Digest::SHA::sha1_base64("/$event/$autorename$filename"); $r->status(200); $r->content_type('text/xml; charset=utf-8');