<?php if (!defined('BASEPATH')) exit('No direct script access allowed'); /** * Code Igniter * * An open source application development framework for PHP 4.3.2 or newer * * @package CodeIgniter * @author Rick Ellis * @copyright Copyright (c) 2006, pMachine, Inc. * @license http://www.codeignitor.com/user_guide/license.html * @link http://www.codeigniter.com * @since Version 1.0 * @filesource */ // ------------------------------------------------------------------------ /** * File Uploading Class * * @package CodeIgniter * @subpackage Libraries * @category Uploads * @author Rick Ellis * @link http://www.codeigniter.com/user_guide/libraries/file_uploading.html */ class CI_Upload { var $max_size = 0; var $max_width = 0; var $max_height = 0; var $allowed_types = ""; var $file_temp = ""; var $file_name = ""; var $orig_name = ""; var $file_type = ""; var $file_size = ""; var $file_ext = ""; var $file_path = ""; var $overwrite = FALSE; var $encrypt_name = FALSE; var $is_image = FALSE; var $image_width = ''; var $image_height = ''; var $image_type = ''; var $image_size_str = ''; var $error_msg = array(); var $mimes = array(); var $remove_spaces = TRUE; var $xss_clean = FALSE; var $temp_prefix = "temp_file_"; /** * Constructor * * @access public */ function CI_Upload($props = array()) { if (count($props) > 0) { $this->initialize($props); } log_message('debug', "Upload Class Initialized"); } // -------------------------------------------------------------------- /** * Initialize preferences * * @access public * @param array * @return void */ function initialize($config = array()) { foreach ($config as $key => $val) { $method = 'set_'.$key; if (method_exists($this, $method)) { $this->$method($val); } else { $this->$key = $val; } } } // -------------------------------------------------------------------- /** * Perform the file upload * * @access public * @return bool */ function do_upload() { // Is $_FILES['userfile'] set? If not, no reason to continue. if ( ! isset($_FILES['userfile'])) { $this->set_error('upload_userfile_not_set'); return FALSE; } // Is the upload path valid? if ( ! $this->validate_upload_path()) { return FALSE; } // Was the file able to be uploaded? If not, determine the reason why. if ( ! is_uploaded_file($_FILES['userfile']['tmp_name'])) { $error = ( ! isset($_FILES['userfile']['error'])) ? 4 : $_FILES['userfile']['error']; switch($error) { case 1 : $this->set_error('upload_file_exceeds_limit'); break; case 3 : $this->set_error('upload_file_partial'); break; case 4 : $this->set_error('upload_no_file_selected'); break; default : $this->set_error('upload_no_file_selected'); break; } return FALSE; } // Set the uploaded data as class variables $this->file_temp = $_FILES['userfile']['tmp_name']; $this->file_name = $_FILES['userfile']['name']; $this->file_size = $_FILES['userfile']['size']; $this->file_type = preg_replace("/^(.+?);.*$/", "\\1", $_FILES['userfile']['type']); $this->file_type = strtolower($this->file_type); $this->file_ext = $this->get_extension($_FILES['userfile']['name']); // Convert the file size to kilobytes if ($this->file_size > 0) { $this->file_size = round($this->file_size/1024, 2); } // Is the file type allowed to be uploaded? if ( ! $this->is_allowed_filetype()) { $this->set_error('upload_invalid_filetype'); return FALSE; } // Is the file size within the allowed maximum? if ( ! $this->is_allowed_filesize()) { $this->set_error('upload_invalid_filesize'); return FALSE; } // Are the image dimensions within the allowed size? // Note: This can fail if the server has an open_basdir restriction. if ( ! $this->is_allowed_dimensions()) { $this->set_error('upload_invalid_dimensions'); return FALSE; } // Sanitize the file name for security $this->file_name = $this->clean_file_name($this->file_name); // Remove white spaces in the name if ($this->remove_spaces == TRUE) { $this->file_name = preg_replace("/\s+/", "_", $this->file_name); } /* * Validate the file name * This function appends an number onto the end of * the file if one with the same name already exists. * If it returns false there was a problem. */ $this->orig_name = $this->file_name; if ($this->overwrite == FALSE) { $this->file_name = $this->set_filename($this->file_path, $this->file_name); if ($this->file_name === FALSE) { return FALSE; } } /* * Move the file to the final destination * To deal with different server configurations * we'll attempt to use copy() first. If that fails * we'll use move_uploaded_file(). One of the two should * reliably work in most environments */ if ( ! @copy($this->file_temp, $this->file_path.$this->file_name)) { if ( ! @move_uploaded_file($this->file_temp, $this->file_path.$this->file_name)) { $this->set_error('upload_destination_error'); return FALSE; } } /* * Run the file through the XSS hacking filter * This helps prevent malicious code from being * embedded within a file. Scripts can easily * be disguised as images or other file types. */ if ($this->xss_clean == TRUE) { $this->do_xss_clean(); } /* * Set the finalized image dimensions * This sets the image width/height (assuming the * file was an image). We use this information * in the "data" function. */ $this->set_image_properties($this->file_path.$this->file_name); return TRUE; } // -------------------------------------------------------------------- /** * Finalized Data Array * * Returns an associative array containing all of the information * related to the upload, allowing the developer easy access in one array. * * @access public * @return array */ function data() { return array ( 'file_name' => $this->file_name, 'file_type' => $this->file_type, 'file_path' => $this->file_path, 'full_path' => $this->file_path.$this->file_name, 'raw_name' => str_replace($this->file_ext, '', $this->file_name), 'orig_name' => $this->orig_name, 'file_ext' => $this->file_ext, 'file_size' => $this->file_size, 'is_image' => $this->is_image(), 'image_width' => $this->image_width, 'image_height' => $this->image_height, 'image_type' => $this->image_type, 'image_size_str' => $this->image_size_str, ); } // -------------------------------------------------------------------- /** * Set Upload Path * * @access public * @param string * @return void */ function set_upload_path($path) { $this->file_path = $path; } // -------------------------------------------------------------------- /** * Set the file name * * This function takes a filename/path as input and looks for the * existnace of a file with the same name. If found, it will append a * number to the end of the filename to avoid overwritting a pre-existing file. * * @access public * @param string * @param string * @return string */ function set_filename($path, $filename) { if ($this->encrypt_name == TRUE) { mt_srand(); $filename = md5(uniqid(mt_rand())).$this->file_ext; } if ( ! file_exists($path.$filename)) { return $filename; } $filename = str_replace($this->file_ext, '', $filename); $new_filename = ''; for ($i = 1; $i < 100; $i++) { if ( ! file_exists($path.$filename.$i.$this->file_ext)) { $new_filename = $filename.$i.$this->file_ext; break; } } if ($new_filename == '') { $this->set_error('upload_bad_filename'); return FALSE; } else { return $new_filename; } } // -------------------------------------------------------------------- /** * Set Maximum File Size * * @access public * @param integer * @return void */ function set_max_filesize($n) { $this->max_size = ( ! eregi("^[[:digit:]]+$", $n)) ? 0 : $n; } // -------------------------------------------------------------------- /** * Set Maximum Image Width * * @access public * @param integer * @return void */ function set_max_width($n) { $this->max_width = ( ! eregi("^[[:digit:]]+$", $n)) ? 0 : $n; } // -------------------------------------------------------------------- /** * Set Maximum Image Height * * @access public * @param integer * @return void */ function set_max_height($n) { $this->max_height = ( ! eregi("^[[:digit:]]+$", $n)) ? 0 : $n; } // -------------------------------------------------------------------- /** * Set Allowed File Types * * @access public * @param string * @return void */ function set_allowed_types($types) { $this->allowed_types = explode('|', $types); } // -------------------------------------------------------------------- /** * Set Image Properties * * Uses GD to determine the width/height/type of image * * @access public * @param string * @return void */ function set_image_properties($path = '') { if ( ! $this->is_image()) { return; } if (function_exists('getimagesize')) { if (FALSE !== ($D = @getimagesize($path))) { $types = array(1 => 'gif', 2 => 'jpeg', 3 => 'png'); $this->image_width = $D['0']; $this->image_height = $D['1']; $this->image_type = ( ! isset($types[$D['2']])) ? 'unknown' : $types[$D['2']]; $this->image_size_str = $D['3']; // string containing height and width } } } // -------------------------------------------------------------------- /** * Set XSS Clean * * Enables the XSS flag so that the file that was uploaded * will be run through the XSS filter. * * @access public * @param bool * @return void */ function set_xss_clean($flag = FALSE) { $this->xss_clean = ($flag == TRUE) ? TRUE : FALSE; } // -------------------------------------------------------------------- /** * Validate the image * * @access public * @return bool */ function is_image() { $img_mimes = array( 'image/gif', 'image/jpg', 'image/jpe', 'image/jpeg', 'image/pjpeg', 'image/png', 'image/x-png' ); return (in_array($this->file_type, $img_mimes)) ? TRUE : FALSE; } // -------------------------------------------------------------------- /** * Verify that the filetype is allowed * * @access public * @return bool */ function is_allowed_filetype() { if (count($this->allowed_types) == 0) { $this->set_error('upload_no_file_types'); return FALSE; } foreach ($this->allowed_types as $val) { $mime = $this->mimes_types(strtolower($val)); if (is_array($mime)) { if (in_array($this->file_type, $mime)) { return TRUE; } } else { if ($mime == $this->file_type) { return TRUE; } } } return FALSE; } // -------------------------------------------------------------------- /** * Verify that the file is within the allowed size * * @access public * @return bool */ function is_allowed_filesize() { if ($this->max_size != 0 AND $this->file_size > $this->max_size) { return FALSE; } else { return TRUE; } } // -------------------------------------------------------------------- /** * Verify that the image is within the allowed width/height * * @access public * @return bool */ function is_allowed_dimensions() { if ( ! $this->is_image()) { return TRUE; } if (function_exists('getimagesize')) { $D = @getimagesize($this->file_temp); if ($this->max_width > 0 AND $D['0'] > $this->max_width) { return FALSE; } if ($this->max_height > 0 AND $D['1'] > $this->max_height) { return FALSE; } return TRUE; } return TRUE; } // -------------------------------------------------------------------- /** * VAlidate Upload Path * * Verifies that it is a valid upload path with proper permissions. * * * @access public * @return bool */ function validate_upload_path() { if ($this->file_path == '') { $this->set_error('upload_no_filepath'); return FALSE; } if (function_exists('realpath') AND @realpath($this->file_path) !== FALSE) { $this->file_path = str_replace("\\", "/", realpath($this->file_path)); } if ( ! @is_dir($this->file_path)) { $this->set_error('upload_no_filepath'); return FALSE; } if ( ! is_writable($this->file_path)) { $this->set_error('upload_not_writable'); return FALSE; } $this->file_path = preg_replace("/(.+?)\/*$/", "\\1/", $this->file_path); return TRUE; } // -------------------------------------------------------------------- /** * Extract the file extension * * @access public * @param string * @return string */ function get_extension($filename) { $x = explode('.', $filename); return '.'.end($x); } // -------------------------------------------------------------------- /** * Clean the file name for security * * @access public * @param string * @return string */ function clean_file_name($filename) { $bad = array( "<!--", "-->", "'", "<", ">", '"', '&', '$', '=', ';', '?', '/', "%20", "%22", "%3c", // < "%253c", // < "%3e", // > "%0e", // > "%28", // ( "%29", // ) "%2528", // ( "%26", // & "%24", // $ "%3f", // ? "%3b", // ; "%3d" // = ); foreach ($bad as $val) { $filename = str_replace($val, '', $filename); } return $filename; } // -------------------------------------------------------------------- /** * Runs the file through the XSS clean function * * This prevents people from embedding malicious code in their files. * I'm not sure that it won't negatively affect certain files in unexpected ways, * but so far I haven't found that it causes trouble. * * @access public * @return void */ function do_xss_clean() { $file = $this->file_path.$this->file_name; if (filesize($file) == 0) { return FALSE; } if ( ! $fp = @fopen($file, 'rb')) { return FALSE; } flock($fp, LOCK_EX); $data = fread($fp, filesize($file)); $obj =& get_instance(); $data = $obj->input->xss_clean($data); fwrite($fp, $data); flock($fp, LOCK_UN); fclose($fp); } // -------------------------------------------------------------------- /** * Set an error message * * @access public * @param string * @return void */ function set_error($msg) { $obj =& get_instance(); $obj->lang->load('upload'); if (is_array($msg)) { foreach ($msg as $val) { $msg = ($obj->lang->line($val) == FALSE) ? $val : $obj->lang->line($val); $this->error_msg[] = $msg; log_message('error', $msg); } } else { $msg = ($obj->lang->line($msg) == FALSE) ? $msg : $obj->lang->line($msg); $this->error_msg[] = $msg; log_message('error', $msg); } } // -------------------------------------------------------------------- /** * Display the error message * * @access public * @param string * @param string * @return string */ function display_errors($open = '<p>', $close = '</p>') { $str = ''; foreach ($this->error_msg as $val) { $str .= $open.$val.$close; } return $str; } // -------------------------------------------------------------------- /** * List of Mime Types * * This is a list of mime types. We use it to validate * the "allowed types" set by the developer * * @access public * @param string * @return string */ function mimes_types($mime) { if (count($this->mimes) == 0) { if (@include(APPPATH.'config/mimes'.EXT)) { $this->mimes = $mimes; unset($mimes); } } return ( ! isset($this->mimes[$mime])) ? FALSE : $this->mimes[$mime]; } } // END Upload Class ?>