From f9f84c1e4389dc23e77b51dadd53410377128cfc Mon Sep 17 00:00:00 2001
From: Jerome Jutteau <mojo@couak.net>
Date: Tue, 1 Jan 2013 23:22:12 +0100
Subject: [PATCH] Changed path system to links and files to manage large
 volumes, max size fix, refactorised delete_link function name

---
 admin.php         |   2 +-
 file.php          |  19 ++-
 index.php         |   7 +-
 lib/functions.php | 329 ++++++++++++++++++++++++++++++----------------
 lib/lang/fr.php   |   2 +
 5 files changed, 228 insertions(+), 131 deletions(-)

diff --git a/admin.php b/admin.php
index e9e5342..bdd0267 100755
--- a/admin.php
+++ b/admin.php
@@ -219,7 +219,7 @@ if (isset ($_POST['action']))
     }
     elseif (strcmp ($_POST['action'], 'delete_link') == 0)
     {
-        jirafeau_delete ($_POST['link']);
+        jirafeau_delete_link ($_POST['link']);
         echo '<div class="message">' . NL;
         echo '<p>' . t('Link deleted') . '</p></div>';
     }
diff --git a/file.php b/file.php
index fefeffd..d01bee6 100644
--- a/file.php
+++ b/file.php
@@ -49,10 +49,10 @@ if (isset ($_GET['h']) && !empty ($_GET['h']))
         require (JIRAFEAU_ROOT.'lib/template/footer.php');
         exit;
     }
-    
-    if (!file_exists (VAR_FILES . $link['md5']))
+    $p = s2p ($link['md5']);
+    if (!file_exists (VAR_FILES . $p . $link['md5']))
     {
-        jirafeau_delete ($link_name);
+        jirafeau_delete_link ($link_name);
         require (JIRAFEAU_ROOT.'lib/template/header.php');
         echo '<div class="error"><p>'.t('File not available.').
         '</p></div>';
@@ -62,7 +62,7 @@ if (isset ($_GET['h']) && !empty ($_GET['h']))
 
     if (!empty ($delete_code) && $delete_code == $link['link_code'])
     {
-        jirafeau_delete ($link_name);
+        jirafeau_delete_link ($link_name);
         require (JIRAFEAU_ROOT.'lib/template/header.php');
         echo '<div class="message"><p>'.t('File has been deleted.').
          '</p></div>';
@@ -70,15 +70,15 @@ if (isset ($_GET['h']) && !empty ($_GET['h']))
         exit;
     }
 
-    if ($link['time'] != JIRAFEAU_INFINITY && time ()> $link['time'])
+    if ($link['time'] != JIRAFEAU_INFINITY && time () > $link['time'])
     {
-        jirafeau_delete ($link_name);
+        jirafeau_delete_link ($link_name);
         require (JIRAFEAU_ROOT.'lib/template/header.php');
         echo '<div class="error"><p>'.
         t('The time limit of this file has expired.') . ' ' .
         t('File has been deleted.') .
         '</p></div>';
-        require (JIRAFEAU_ROOT.'lib/template/footer.php');
+        require (JIRAFEAU_ROOT . 'lib/template/footer.php');
         exit;
     }
 
@@ -108,7 +108,6 @@ if (isset ($_GET['h']) && !empty ($_GET['h']))
         if ($link['key'] != md5 ($_POST['key']))
         {
             header ("Access denied");
-
             require (JIRAFEAU_ROOT.'lib/template/header.php');
             echo '<div class="error"><p>' . t('Access denied') .
             '</p></div>';
@@ -125,10 +124,10 @@ if (isset ($_GET['h']) && !empty ($_GET['h']))
         header ('Content-Disposition: attachment; filename="' .
             $link['file_name'] . '"');
     }
-    readfile (VAR_FILES . $link['md5']);
+    readfile (VAR_FILES . $p . $link['md5']);
 
     if ($link['onetime'] == 'O')
-        jirafeau_delete ($link_name);
+        jirafeau_delete_link ($link_name);
     exit;
 }
 else
diff --git a/index.php b/index.php
index 660c18e..ae25e7b 100644
--- a/index.php
+++ b/index.php
@@ -64,7 +64,7 @@ if ($writable && isset ($_POST['jirafeau']))
 
     $res =
         jirafeau_upload ($_FILES['file'], isset ($_POST['one_time_download']),
-                         $key, $time, $cfg, $_SERVER['REMOTE_ADDR']);
+                         $key, $time, $_SERVER['REMOTE_ADDR']);
 }
 
 require (JIRAFEAU_ROOT . 'lib/template/header.php');
@@ -138,9 +138,8 @@ if (!has_error () && $writable)
         <legend><?php echo t('Upload a file');
     ?></legend> <p><input type = "file" name = "file" size =
         "30" /></p> <p class =
-        "config"><?php printf ('%s: %dMB', t('Maximum file size'),
-                               jirafeau_get_max_upload_size () / (1024 *
-                                                                  1024));
+        "config"><?php printf ('%s: %s', t('Maximum file size'),
+                               jirafeau_get_max_upload_size ());
     ?></p><p>
     <input type = "submit" id='send' value ="<?php echo t('Send'); ?>"
     onclick="
diff --git a/lib/functions.php b/lib/functions.php
index 7b63148..c26c5bc 100644
--- a/lib/functions.php
+++ b/lib/functions.php
@@ -17,6 +17,68 @@
  *  You should have received a copy of the GNU Affero General Public License
  *  along with this program.  If not, see <http://www.gnu.org/licenses/>.
  */
+
+/**
+ * Transform a string in a path by seperating each letters by a '/'.
+  * @return path finishing with a '/'
+ */
+function
+s2p ($s)
+{
+    $p = '';
+    for ($i = 0; $i < strlen ($s); $i++)
+        $p .= $s{$i} . '/';
+    return $p;
+}
+
+function
+jirafeau_human_size ($octets)
+{
+    $u = array ('B', 'KB', 'MB', 'GB', 'TB');
+    $o = max ($octets, 0);
+    $p = min (floor (($o ? log ($o) : 0) / log (1024)), count ($u) - 1);
+    $o /= pow (1024, $p);
+    return round ($o, 1) . $u[$p];
+} 
+
+function
+jirafeau_clean_rm_link ($link)
+{
+    $p = s2p ("$link");
+    if (file_exists (VAR_LINKS . $p . $link))
+        unlink (VAR_LINKS . $p . $link);
+    $parse = VAR_LINKS . $p;
+    $scan = array();
+    while (file_exists ($parse)
+           && ($scan = scandir ($parse))
+           && count ($scan) == 2 // '.' and '..' folders => empty.
+           && basename ($parse) != basename (VAR_LINKS)) 
+    {
+        rmdir ($parse);
+        $parse = substr ($parse, 0, strlen($parse) - strlen(basename ($parse)) - 1);
+    }
+}
+
+function
+jirafeau_clean_rm_file ($md5)
+{
+    $p = s2p ("$md5");
+    if (file_exists (VAR_FILES . $p . $md5))
+        unlink (VAR_FILES . $p . $md5);
+    if (file_exists (VAR_FILES . $p . $md5 . '_count'))
+        unlink (VAR_FILES . $p . $md5 . '_count');
+    $parse = VAR_FILES . $p;
+    $scan = array();
+    while (file_exists ($parse)
+           && ($scan = scandir ($parse))
+           && count ($scan) == 2 // '.' and '..' folders => empty.
+           && basename ($parse) != basename (VAR_FILES)) 
+    {
+        rmdir ($parse);
+        $parse = substr ($parse, 0, strlen($parse) - strlen(basename ($parse)) - 1);
+    }
+}
+
 /**
  * transforms a php.ini string representing a value in an integer
  * @param $value the value from php.ini
@@ -46,13 +108,14 @@ function jirafeau_ini_to_bytes ($value)
 
 /**
  * gets the maximum upload size according to php.ini
- * @returns the maximum upload size
+ * @returns the maximum upload size string
  */
 function
 jirafeau_get_max_upload_size ()
 {
-    return min (jirafeau_ini_to_bytes (ini_get ('post_max_size')),
-                jirafeau_ini_to_bytes (ini_get ('upload_max_filesize')));
+    return jirafeau_human_size(
+            min (jirafeau_ini_to_bytes (ini_get ('post_max_size')),
+                 jirafeau_ini_to_bytes (ini_get ('upload_max_filesize'))));
 }
 
 /**
@@ -94,37 +157,34 @@ jirafeau_upload_errstr ($code)
  */
 
 function
-jirafeau_delete ($link)
+jirafeau_delete_link ($link)
 {
-    if (!file_exists ( VAR_LINKS . $link))
+    $l = jirafeau_get_link ($link);
+    if (!count ($l))
         return;
 
-    $content = file (VAR_LINKS . $link);
-    $md5 = trim ($content[5]);
-    unlink (VAR_LINKS . $link);
+    jirafeau_clean_rm_link ($link);
+
+    $md5 = $l['md5'];
+    $p = s2p ("$md5");
 
     $counter = 1;
-    if (file_exists ( VAR_FILES . $md5. '_count'))
+    if (file_exists (VAR_FILES . $p . $md5. '_count'))
     {
-        $content = file ( VAR_FILES . $md5. '_count');
+        $content = file (VAR_FILES . $p . $md5. '_count');
         $counter = trim ($content[0]);
     }
     $counter--;
 
     if ($counter >= 1)
     {
-        $handle = fopen ( VAR_FILES . $md5. '_count', 'w');
+        $handle = fopen (VAR_FILES . $p . $md5. '_count', 'w');
         fwrite ($handle, $counter);
         fclose ($handle);
     }
 
     if ($counter == 0)
-    {
-        if (file_exists (VAR_FILES . $md5))
-            unlink ( VAR_FILES . $md5);
-        if (file_exists (VAR_FILES . $md5 . '_count'))
-        unlink ( VAR_FILES . $md5. '_count');
-    }
+        jirafeau_clean_rm_file ($md5);
 }
 
 /**
@@ -134,27 +194,38 @@ function
 jirafeau_delete_file ($md5)
 {
     $count = 0;
-    $links_dir = scandir (VAR_LINKS);
-    
-    foreach ($links_dir as $link)
+    /* Get all links files. */
+    $stack = array (VAR_LINKS);
+    while (($d = array_shift ($stack)) && $d != NULL)
     {
-        if (strcmp ($link, '.') == 0 || strcmp ($link, '..') == 0 ||
-            preg_match ('/\.tmp/i', "$link"))
-            continue;
-        /* Read link informations. */
-        $l = jirafeau_get_link ($link);
-        if ($l['md5'] == $md5)
+        $dir = scandir ($d);
+
+        foreach ($dir as $node)
         {
-            $count++;
-            jirafeau_delete ($link);
+            if (strcmp ($node, '.') == 0 || strcmp ($node, '..') == 0 ||
+                preg_match ('/\.tmp/i', "$node"))
+                continue;
+            
+            if (is_dir ($d . $node))
+            {
+                /* Push new found directory. */
+                $stack[] = $d . $node . '/';
+            }
+            elseif (is_file ($d . $node))
+            {
+                /* Read link informations. */
+                $l = jirafeau_get_link (basename ($node));
+                if (!count ($l))
+                    continue;
+                if ($l['md5'] == $md5)
+                {
+                    $count++;
+                    jirafeau_delete_link ($node);
+                }   
+            }
         }
     }
-
-    if (file_exists (VAR_FILES . $md5 . '_count'))
-        unlink (VAR_FILES . $md5. '_count');
-    if (file_exists (VAR_FILES . $md5))
-        unlink (VAR_FILES . $md5);
-
+    jirafeau_clean_rm_file ($md5);
     return $count;
 }
 
@@ -164,7 +235,6 @@ jirafeau_delete_file ($md5)
  * @param $one_time_download is the file a one time download ?
  * @param $key if not empty, protect the file with this key
  * @param $time the time of validity of the file
- * @param $cfg the current configuration
  * @param $ip uploader's ip
  * @returns an array containing some information
  *   'error' => information on possible errors
@@ -172,7 +242,7 @@ jirafeau_delete_file ($md5)
  *   'delete_link' => the link code to delete file
  */
 function
-jirafeau_upload ($file, $one_time_download, $key, $time, $cfg, $ip)
+jirafeau_upload ($file, $one_time_download, $key, $time, $ip)
 {
     if (empty ($file['tmp_name']) || !is_uploaded_file ($file['tmp_name']))
     {
@@ -195,11 +265,13 @@ jirafeau_upload ($file, $one_time_download, $key, $time, $cfg, $ip)
 
     /* does file already exist ? */
     $rc = false;
-    if (file_exists ( VAR_FILES . $md5))
+    $p = s2p ("$md5");
+    if (file_exists (VAR_FILES . $p .  $md5))
     {
         $rc = unlink ($file['tmp_name']);
     }
-    elseif (move_uploaded_file ($file['tmp_name'],  VAR_FILES . $md5))
+    elseif ((file_exists (VAR_FILES . $p) || @mkdir (VAR_FILES . $p, 0755, true))
+            && move_uploaded_file ($file['tmp_name'], VAR_FILES . $p . $md5))
     {
         $rc = true;
     }
@@ -208,20 +280,20 @@ jirafeau_upload ($file, $one_time_download, $key, $time, $cfg, $ip)
         return (array(
                  'error' =>
                    array ('has_error' => true,
-                          'why' => t('Internal error during file creation. ')),
+                          'why' => t('Internal error during file creation.')),
                  'link' =>'',
                  'delete_link' => ''));
     }
 
     /* increment or create count file */
     $counter = 0;
-    if (file_exists (VAR_FILES . $md5 . '_count'))
+    if (file_exists (VAR_FILES . $p . $md5 . '_count'))
     {
-        $content = file ( VAR_FILES . $md5. '_count');
+        $content = file (VAR_FILES . $p . $md5. '_count');
         $counter = trim ($content[0]);
     }
     $counter++;
-    $handle = fopen ( VAR_FILES . $md5. '_count', 'w');
+    $handle = fopen (VAR_FILES . $p . $md5. '_count', 'w');
     fwrite ($handle, $counter);
     fclose ($handle);
 
@@ -236,28 +308,31 @@ jirafeau_upload ($file, $one_time_download, $key, $time, $cfg, $ip)
         $password = md5 ($key);
 
     /* create link file */
-    $link_tmp_name =  VAR_LINKS . $md5.rand (0, 10000) . ' .tmp';
+    $link_tmp_name =  VAR_LINKS . $md5 . rand (0, 10000) . ' .tmp';
     $handle = fopen ($link_tmp_name, 'w');
     fwrite ($handle,
-            $name . NL. $mime_type . NL. $size . NL. $password . NL. $time . NL . $md5.
-            NL.($one_time_download ? 'O' : 'R') . NL.date ('U') . NL. $ip . NL.
-            $delete_link_code . NL);
+            $name . NL. $mime_type . NL. $size . NL. $password . NL. $time .
+            NL . $md5. NL . ($one_time_download ? 'O' : 'R') . NL.date ('U') .
+            NL. $ip . NL. $delete_link_code . NL);
     fclose ($handle);
     $md5_link = md5_file ($link_tmp_name);
-    if (!rename ($link_tmp_name,  VAR_LINKS . $md5_link))
+    $l = s2p ("$md5_link");
+    if (!@mkdir (VAR_LINKS . $l, 0755, true) ||
+        !rename ($link_tmp_name,  VAR_LINKS . $l . $md5_link))
     {
-        unlink ($link_tmp_name);
+        if (file_exists ($link_tmp_name))
+            unlink ($link_tmp_name);
+        
         $counter--;
         if ($counter >= 1)
         {
-            $handle = fopen ( VAR_FILES . $md5. '_count', 'w');
+            $handle = fopen (VAR_FILES . $p . $md5. '_count', 'w');
             fwrite ($handle, $counter);
             fclose ($handle);
         }
         else
         {
-            unlink ( VAR_FILES . $md5. '_count');
-            unlink ( VAR_FILES . $md5);
+            jirafeau_clean_rm_file ($md5_link);
         }
         return (array(
                  'error' =>
@@ -343,7 +418,7 @@ function
 jirafeau_get_link ($hash)
 {
     $out = array ();
-    $link = VAR_LINKS . $hash;
+    $link = VAR_LINKS . s2p ("$hash") . $hash;
 
     if (!file_exists ($link))
         return $out;
@@ -363,23 +438,12 @@ jirafeau_get_link ($hash)
     return $out;
 }
 
-function
-jirafeau_human_size ($octets)
-{
-    $u = array ('B', 'KB', 'MB', 'GB', 'TB');
-    $o = max ($octets, 0);
-    $p = min (floor (($o ? log ($o) : 0) / log (1024)), count ($u) - 1);
-    $o /= pow (1024, $p);
-    return round ($o, 1) . $u[$p];
-} 
-
 /**
  * List files in admin interface.
  */
 function
 jirafeau_admin_list ($name, $file_hash, $link_hash)
 {
-    $links_dir = scandir (VAR_LINKS);
     echo '<fieldset><legend>';
     if (!empty ($name))
         echo $name . ' ';
@@ -401,45 +465,61 @@ jirafeau_admin_list ($name, $file_hash, $link_hash)
     echo '<td>' . t('Origin') . '</td>';
     echo '<td>' . t('Action') . '</td>';
     echo '</tr>';
-    foreach ($links_dir as $link)
+
+    /* Get all links files. */
+    $stack = array (VAR_LINKS);
+    while (($d = array_shift ($stack)) && $d != NULL)
     {
-        if (strcmp ($link, '.') == 0 || strcmp ($link, '..') == 0 ||
-            preg_match ('/\.tmp/i', "$link"))
-            continue;
-        /* Read link informations. */
-        $l = jirafeau_get_link ($link);
-        
-        /* Filter. */
-        if (!empty ($name) && $name != $l['file_name'])
-            continue;
-        if (!empty ($file_hash) && $file_hash != $l['md5'])
-            continue;
-        if (!empty ($link_hash) && $link_hash != $link)
-            continue;
-        
-        /* Print link informations. */
-        echo '<tr>';
-        echo '<td>' . $l['file_name'] . '</td>';
-        echo '<td>' . $l['mime_type'] . '</td>';
-        echo '<td>' . jirafeau_human_size ($l['file_size']) . '</td>';
-        echo '<td>' . ($l['time'] == -1 ? '' : strftime ('%c', $l['time'])) .
-             '</td>';
-        echo '<td>' . $l['onetime'] . '</td>';
-        echo '<td>' . strftime ('%c', $l['upload_date']) . '</td>';
-        echo '<td>' . $l['ip'] . '</td>';
-        echo '<td>' .
-        '<form action = "admin.php" method = "post">' .
-        '<input type = "hidden" name = "action" value = "delete_link"/>' .
-        '<input type = "hidden" name = "link" value = "' . $link . '"/>' .
-        '<input type = "submit" value = "' . t('Del link') . '" />' .
-        '</form>' .
-        '<form action = "admin.php" method = "post">' .
-        '<input type = "hidden" name = "action" value = "delete_file"/>' .
-        '<input type = "hidden" name = "md5" value = "' . $l['md5'] . '"/>' .
-        '<input type = "submit" value = "' . t('Del file and links') . '" />' .
-        '</form>' .
-        '</td>';
-        echo '</tr>';
+        $dir = scandir ($d);
+        foreach ($dir as $node)
+        {
+            if (strcmp ($node, '.') == 0 || strcmp ($node, '..') == 0 ||
+                preg_match ('/\.tmp/i', "$node"))
+                continue;
+            if (is_dir ($d . $node))
+            {
+                /* Push new found directory. */
+                $stack[] = $d . $node . '/';
+            }
+            elseif (is_file ($d . $node))
+            {
+                /* Read link informations. */
+                $l = jirafeau_get_link ($node);
+                if (!count ($l))
+                    continue;
+
+                /* Filter. */
+                if (!empty ($name) && $name != $l['file_name'])
+                    continue;
+                if (!empty ($file_hash) && $file_hash != $l['md5'])
+                    continue;
+                if (!empty ($link_hash) && $link_hash != $link)
+                    continue;
+                /* Print link informations. */
+                echo '<tr>';
+                echo '<td>' . $l['file_name'] . '</td>';
+                echo '<td>' . $l['mime_type'] . '</td>';
+                echo '<td>' . jirafeau_human_size ($l['file_size']) . '</td>';
+                echo '<td>' . ($l['time'] == -1 ? '' : strftime ('%c', $l['time'])) .
+                     '</td>';
+                echo '<td>' . $l['onetime'] . '</td>';
+                echo '<td>' . strftime ('%c', $l['upload_date']) . '</td>';
+                echo '<td>' . $l['ip'] . '</td>';
+                echo '<td>' .
+                '<form action = "admin.php" method = "post">' .
+                '<input type = "hidden" name = "action" value = "delete_link"/>' .
+                '<input type = "hidden" name = "link" value = "' . $node . '"/>' .
+                '<input type = "submit" value = "' . t('Del link') . '" />' .
+                '</form>' .
+                '<form action = "admin.php" method = "post">' .
+                '<input type = "hidden" name = "action" value = "delete_file"/>' .
+                '<input type = "hidden" name = "md5" value = "' . $l['md5'] . '"/>' .
+                '<input type = "submit" value = "' . t('Del file and links') . '" />' .
+                '</form>' .
+                '</td>';
+                echo '</tr>';
+            }
+        }
     }
     echo '</table></fieldset>';
 }
@@ -451,24 +531,41 @@ jirafeau_admin_list ($name, $file_hash, $link_hash)
 function
 jirafeau_admin_clean ()
 {
-    $c = 0;
-    $links_dir = scandir (VAR_LINKS);
-
-    foreach ($links_dir as $link)
+    $count = 0;
+    /* Get all links files. */
+    $stack = array (VAR_LINKS);
+    while (($d = array_shift ($stack)) && $d != NULL)
     {
-        if (strcmp ($link, '.') == 0 || strcmp ($link, '..') == 0 ||
-            preg_match ('/\.tmp/i', "$link"))
-            continue;
-        /* Read link informations. */
-        $l = jirafeau_get_link ($link);
-        if ($l['time'] > 0 && $l['time'] < time () || // expired
-            !file_exists (VAR_FILES . $l['md5']) || // invalid
-            !file_exists (VAR_FILES . $l['md5'] . '_count')) // invalid
+        $dir = scandir ($d);
+
+        foreach ($dir as $node)
         {
-            jirafeau_delete ($link);
-            $c++;
+            if (strcmp ($node, '.') == 0 || strcmp ($node, '..') == 0 ||
+                preg_match ('/\.tmp/i', "$node"))
+                continue;
+            
+            if (is_dir ($d . $node))
+            {
+                /* Push new found directory. */
+                $stack[] = $d . $node . '/';
+            }
+            elseif (is_file ($d . $node))
+            {
+                /* Read link informations. */
+                $l = jirafeau_get_link (basename ($node));
+                if (!count ($l))
+                    continue;
+                $p = s2p ($l['md5']);
+                if ($l['time'] > 0 && $l['time'] < time () || // expired
+                    !file_exists (VAR_FILES . $p . $l['md5']) || // invalid
+                    !file_exists (VAR_FILES . $p . $l['md5'] . '_count')) // invalid
+                {
+                    jirafeau_delete_link ($node);
+                    $count++;
+                }
+            }
         }
     }
-    return $c;
+    return $count;
 }
 ?>
diff --git a/lib/lang/fr.php b/lib/lang/fr.php
index 9148c07..d266c1f 100755
--- a/lib/lang/fr.php
+++ b/lib/lang/fr.php
@@ -52,6 +52,8 @@ $tr = array (
              'Password protection' => 'Protection par mot de passe',
              'Give the password of this file' => 'Donnez le mot de passe pour ce fichier',
              'Access denied' => 'Accès interdit',
+             /* functions.php */
+             'Internal error during file creation.' => 'Erreur interne durant la creation de fichier.',
              /* install.hpp */
              'This file was generated by the install process. You can edit it. Please see config.php to understand the configuration items.' => 'Ce fichier a été généré par le processus d\'installation. Vous pouvez l\'éditer. Merci de se référer à config.php pour comprendre les éléments de configuration.',
              'The following directory could not be created' => 'Le dossier suivant ne peut être créé',
-- 
2.34.1