tools/upload-file.pl

   1 #!/usr/bin/perl
   2
   3 # Copyright (C) 2007 LibLime
   4 #
   5 # This file is part of Koha.
   6 #
   7 # Koha is free software; you can redistribute it and/or modify it under the
   8 # terms of the GNU General Public License as published by the Free Software
   9 # Foundation; either version 2 of the License, or (at your option) any later
  10 # version.
  11 #
  12 # Koha is distributed in the hope that it will be useful, but WITHOUT ANY
  13 # WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
  14 # A PARTICULAR PURPOSE.  See the GNU General Public License for more details.
  15 #
  16 # You should have received a copy of the GNU General Public License along
  17 # with Koha; if not, write to the Free Software Foundation, Inc.,
  18 # 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  19
  20 use strict;
  21 #use warnings; FIXME - Bug 2505
  22
  23 # standard or CPAN modules used
  24 use IO::File;
  25 use CGI;
  26 use CGI::Session;
  27 use C4::Context;
  28 use C4::Auth qw/check_cookie_auth/;
  29 use CGI::Cookie; # need to check cookies before
  30                  # having CGI parse the POST request
  31 use C4::UploadedFile;
  32
  33 # upload-file.pl must authenticate the user
  34 # before processing the POST request,
  35 # and quickly bounce if the user is
  36 # not authorized.  Consequently, unlike
  37 # most of the other CGI scripts, upload-file.pl
  38 # requires that the session cookie already
  39 # have been created.
  40
  41 my %cookies = fetch CGI::Cookie;
  42 my ($auth_status, $sessionID) = check_cookie_auth($cookies{'CGISESSID'}->value, { tools => '*' });
  43 if ($auth_status ne "ok") {
  44     $auth_status = 'denied' if $auth_status eq 'failed';
  45     send_reply($auth_status, "");
  46     exit 0;
  47 }
  48
  49 my $uploaded_file = C4::UploadedFile->new($sessionID);
  50 unless (defined $uploaded_file) {
  51     # FIXME - failed to create file for some reason
  52     send_reply('failed', '');
  53     exit 0;
  54 }
  55 $uploaded_file->max_size($ENV{'CONTENT_LENGTH'}); # may not be the file size, exactly
  56
  57 my $first_chunk = 1;
  58
  59 my $query;
  60 $query = new CGI \&upload_hook;
  61 $uploaded_file->done();
  62 send_reply('done', $uploaded_file->id());
  63
  64 # FIXME - if possible, trap signal caused by user cancelling upload
  65 # FIXME - something is wrong during cleanup: \t(in cleanup) Can't call method "commit" on unblessed reference at /usr/local/share/perl/5.8.8/CGI/Session/Driver/DBI.pm line 130 during global destruction.
  66 exit 0;
  67
  68 sub upload_hook {
  69     my ($file_name, $buffer, $bytes_read, $session) = @_;
  70     $uploaded_file->stash(\$buffer, $bytes_read);
  71     if ($first_chunk) {
  72         $uploaded_file->name($file_name);
  73         $first_chunk = 0;
  74     }
  75 }
  76
  77 sub send_reply {
  78     my ($upload_status, $fileid) = @_;
  79
  80     my $reply = CGI->new("");
  81     print $reply->header(-type => 'text/html');
  82     # response will be sent back as JSON
  83     print '{"status":"' . $upload_status . '","fileid":"' . $fileid . '"}';
  84 }