lib/functions.php

   1 <?php
   2 /*
   3  *  Jirafeau, your web file repository
   4  *  Copyright (C) 2008  Julien "axolotl" BERNARD <axolotl@magieeternelle.org>
   5  *  Copyright (C) 2012  Jerome Jutteau <j.jutteau@gmail.com>
   6  *
   7  *  This program is free software: you can redistribute it and/or modify
   8  *  it under the terms of the GNU Affero General Public License as
   9  *  published by the Free Software Foundation, either version 3 of the
  10  *  License, or (at your option) any later version.
  11  *
  12  *  This program is distributed in the hope that it will be useful,
  13  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15  *  GNU Affero General Public License for more details.
  16  *
  17  *  You should have received a copy of the GNU Affero General Public License
  18  *  along with this program.  If not, see <http://www.gnu.org/licenses/>.
  19  */
  20
  21 /**
  22  * transforms a php.ini string representing a value in an integer
  23  * @param $value the value from php.ini
  24  * @returns an integer for this value
  25  */
  26 function jirafeau_ini_to_bytes($value) {
  27   $modifier = substr($value, -1);
  28   $bytes = substr($value, 0, -1);
  29   switch(strtoupper($modifier)) {
  30   case 'P':
  31     $bytes *= 1024;
  32   case 'T':
  33     $bytes *= 1024;
  34   case 'G':
  35     $bytes *= 1024;
  36   case 'M':
  37     $bytes *= 1024;
  38   case 'K':
  39     $bytes *= 1024;
  40   default:
  41     break;
  42   }
  43   return $bytes;
  44 }
  45
  46 /**
  47  * gets the maximum upload size according to php.ini
  48  * @returns the maximum upload size
  49  */
  50 function jirafeau_get_max_upload_size() {
  51   return min(jirafeau_ini_to_bytes(ini_get('post_max_size')), jirafeau_ini_to_bytes(ini_get('upload_max_filesize')));
  52 }
  53
  54 /**
  55  * detects if a given filename is present in a directory and find an alternate filename
  56  * @param $name the initial filename
  57  * @param $dir the directory to explore (finishing with a '/')
  58  * @returns an alternate filename, possibly the initial filename
  59  */
  60 function jirafeau_detect_collision($name, $dir) {
  61   if(!file_exists($dir . $name)) {
  62     return $name;
  63   }
  64
  65   $dot = strpos($name, '.');
  66   $dot = ($dot === false) ? strlen($name) : $dot;
  67   $first = substr($name, 0, $dot);
  68   $second = substr($name, $dot);
  69   $i = 1;
  70   do {
  71     $new_name = $first . '-' . $i . $second;
  72     $i++;
  73   } while(file_exists($dir . $new_name));
  74
  75   return $new_name;
  76 }
  77
  78 /**
  79  * gets a string explaining the error
  80  * @param $code the error code
  81  * @returns a string explaining the error
  82  */
  83 function jirafeau_upload_errstr($code) {
  84   switch($code) {
  85   case UPLOAD_ERR_INI_SIZE:
  86   case UPLOAD_ERR_FORM_SIZE:
  87     return _('Your file exceeds the maximum authorized file size.');
  88     break;
  89
  90   case UPLOAD_ERR_PARTIAL:
  91   case UPLOAD_ERR_NO_FILE:
  92     return _('Your file was not uploaded correctly. You may succeed in retrying.');
  93     break;
  94
  95   case UPLOAD_ERR_NO_TMP_DIR:
  96   case UPLOAD_ERR_CANT_WRITE:
  97   case UPLOAD_ERR_EXTENSION:
  98     return _('Internal error. You may not succeed in retrying.');
  99     break;
 100
 101   default:
 102     break;
 103   }
 104   return _('Unknown error.');
 105 }
 106
 107 /**
 108  * handles an uploaded file
 109  * @param $file the file struct given by $_FILE[]
 110  * @param $one_time_download is the file a one time download ?
 111  * @param $key if not empty, protect the file with this key
 112  * @param $time the time of validity of the file
 113  * @param $cfg the current configuration
 114  * @param $ip uploader's ip
 115  * @returns an array containing some information
 116  *   'error' => information on possible errors
 117  *   'link' => the link name of the uploaded file
 118  */
 119 function jirafeau_upload($file, $one_time_download, $key, $time, $cfg, $ip) {
 120   if(empty($file['tmp_name']) || !is_uploaded_file($file['tmp_name'])) {
 121     return(array('error' => array('has_error' => true, 'why' => jirafeau_upload_errstr($file['error'])), 'link' => ''));
 122   }
 123
 124   /* array representing no error */
 125   $noerr = array('has_error' => false, 'why' => '');
 126
 127   /* file informations */
 128   $md5 = md5_file($file['tmp_name']);
 129   $name = trim($file['name']);
 130   $mime_type = $file['type'];
 131   $size = $file['size'];
 132
 133   /* does file already exist ? */
 134   $rc = false;
 135   if(file_exists(VAR_FILES . $md5)) {
 136     $rc = unlink($file['tmp_name']);
 137   }
 138   elseif(move_uploaded_file($file['tmp_name'], VAR_FILES . $md5)) {
 139     $rc = true;
 140   }
 141   if(!$rc)
 142   {
 143     return(array(
 144       'error' => array(
 145         'has_error' => true,
 146         'why' => _('Internal error during file creation.')),
 147       'link' => '')
 148     );
 149   }
 150
 151   /* increment or create count file */
 152   $counter=0;
 153   if(file_exists(VAR_FILES . $md5 . '_count')) {
 154     $content = file(VAR_FILES . $md5 . '_count');
 155     $counter = trim($content[0]);
 156   }
 157   $counter++;
 158   $handle = fopen(VAR_FILES . $md5 . '_count', 'w');
 159   fwrite($handle, $counter);
 160   fclose($handle);
 161
 162   /* create link file */
 163   $link_tmp_name = VAR_LINKS . $md5 . rand(0, 10000) . '.tmp';
 164   $handle = fopen($link_tmp_name, 'w');
 165   fwrite($handle, $name . NL . $mime_type . NL . $size . NL . $key . NL . $time . NL . $md5 . NL . ($one_time_download ? 'O' : 'R') . NL . date('U') . NL . $ip . NL);
 166   fclose($handle);
 167   $md5_link = md5_file($link_tmp_name);
 168   if(!rename($link_tmp_name, VAR_LINKS . $md5_link)) {
 169     if ($counter > 1) {
 170       $counter--;
 171       $handle = fopen(VAR_FILES . $md5 . '_count', 'w');
 172       fwrite($handle, $counter);
 173       fclose($handle);
 174     }
 175     else {
 176       unlink($link_tmp_name);
 177       unlink(VAR_FILE . $md5 . '_count');
 178       unlink(VAR_FILE . $md5);
 179     }
 180     return(array(
 181       'error' => array(
 182         'has_error' => true,
 183         'why' => _('Internal error during file creation.')),
 184       'link' => '')
 185     );
 186   }
 187   return(array('error' => $noerr, 'link' => $md5_link));
 188 }
 189
 190 /**
 191  * tells if a mime-type is viewable in a browser
 192  * @param $mime the mime type
 193  * @returns a boolean telling if a mime type is viewable
 194  */
 195 function jirafeau_is_viewable($mime) {
 196   if(!empty($mime)) {
 197     // actually, verify if mime-type is an image or a text
 198     $viewable = array('image', 'text');
 199     $decomposed = explode('/', $mime);
 200     return in_array($decomposed[0], $viewable);
 201   }
 202   return false;
 203 }
 204
 205
 206 // Error handling functions.
 207 //! Global array that contains all registered errors.
 208 $error_list = array ();
 209
 210 /**
 211  * Adds an error to the list of errors.
 212  * @param $title the error's title
 213  * @param $description is a human-friendly description of the problem.
 214  */
 215 function add_error ($title, $description) {
 216     global $error_list;
 217     $error_list[] = '<p>' . $title . '<br />' . $description . '</p>';
 218 }
 219
 220 /**
 221  * Informs whether any error has been registered yet.
 222  * @return true if there are errors.
 223  */
 224 function has_error () {
 225     global $error_list;
 226     return !empty ($error_list);
 227 }
 228
 229 /**
 230  * Displays all the errors.
 231  */
 232 function show_errors () {
 233     if (has_error ()) {
 234         global $error_list;
 235         echo '<div class="error">';
 236         foreach ($error_list as $error) {
 237             echo $error;
 238         }
 239         echo '</div>';
 240     }
 241 }
 242
 243 ?>