312 lines
		
	
	
		
			9.1 KiB
		
	
	
	
		
			PHP
		
	
	
		
		
			
		
	
	
			312 lines
		
	
	
		
			9.1 KiB
		
	
	
	
		
			PHP
		
	
	
|  | <?php | ||
|  | 
 | ||
|  | class HTMLPurifier_DefinitionCache_Serializer extends HTMLPurifier_DefinitionCache | ||
|  | { | ||
|  | 
 | ||
|  |     /** | ||
|  |      * @param HTMLPurifier_Definition $def | ||
|  |      * @param HTMLPurifier_Config $config | ||
|  |      * @return int|bool | ||
|  |      */ | ||
|  |     public function add($def, $config) | ||
|  |     { | ||
|  |         if (!$this->checkDefType($def)) { | ||
|  |             return; | ||
|  |         } | ||
|  |         $file = $this->generateFilePath($config); | ||
|  |         if (file_exists($file)) { | ||
|  |             return false; | ||
|  |         } | ||
|  |         if (!$this->_prepareDir($config)) { | ||
|  |             return false; | ||
|  |         } | ||
|  |         return $this->_write($file, serialize($def), $config); | ||
|  |     } | ||
|  | 
 | ||
|  |     /** | ||
|  |      * @param HTMLPurifier_Definition $def | ||
|  |      * @param HTMLPurifier_Config $config | ||
|  |      * @return int|bool | ||
|  |      */ | ||
|  |     public function set($def, $config) | ||
|  |     { | ||
|  |         if (!$this->checkDefType($def)) { | ||
|  |             return; | ||
|  |         } | ||
|  |         $file = $this->generateFilePath($config); | ||
|  |         if (!$this->_prepareDir($config)) { | ||
|  |             return false; | ||
|  |         } | ||
|  |         return $this->_write($file, serialize($def), $config); | ||
|  |     } | ||
|  | 
 | ||
|  |     /** | ||
|  |      * @param HTMLPurifier_Definition $def | ||
|  |      * @param HTMLPurifier_Config $config | ||
|  |      * @return int|bool | ||
|  |      */ | ||
|  |     public function replace($def, $config) | ||
|  |     { | ||
|  |         if (!$this->checkDefType($def)) { | ||
|  |             return; | ||
|  |         } | ||
|  |         $file = $this->generateFilePath($config); | ||
|  |         if (!file_exists($file)) { | ||
|  |             return false; | ||
|  |         } | ||
|  |         if (!$this->_prepareDir($config)) { | ||
|  |             return false; | ||
|  |         } | ||
|  |         return $this->_write($file, serialize($def), $config); | ||
|  |     } | ||
|  | 
 | ||
|  |     /** | ||
|  |      * @param HTMLPurifier_Config $config | ||
|  |      * @return bool|HTMLPurifier_Config | ||
|  |      */ | ||
|  |     public function get($config) | ||
|  |     { | ||
|  |         $file = $this->generateFilePath($config); | ||
|  |         if (!file_exists($file)) { | ||
|  |             return false; | ||
|  |         } | ||
|  |         return unserialize(file_get_contents($file)); | ||
|  |     } | ||
|  | 
 | ||
|  |     /** | ||
|  |      * @param HTMLPurifier_Config $config | ||
|  |      * @return bool | ||
|  |      */ | ||
|  |     public function remove($config) | ||
|  |     { | ||
|  |         $file = $this->generateFilePath($config); | ||
|  |         if (!file_exists($file)) { | ||
|  |             return false; | ||
|  |         } | ||
|  |         return unlink($file); | ||
|  |     } | ||
|  | 
 | ||
|  |     /** | ||
|  |      * @param HTMLPurifier_Config $config | ||
|  |      * @return bool | ||
|  |      */ | ||
|  |     public function flush($config) | ||
|  |     { | ||
|  |         if (!$this->_prepareDir($config)) { | ||
|  |             return false; | ||
|  |         } | ||
|  |         $dir = $this->generateDirectoryPath($config); | ||
|  |         $dh = opendir($dir); | ||
|  |         // Apparently, on some versions of PHP, readdir will return
 | ||
|  |         // an empty string if you pass an invalid argument to readdir.
 | ||
|  |         // So you need this test.  See #49.
 | ||
|  |         if (false === $dh) { | ||
|  |             return false; | ||
|  |         } | ||
|  |         while (false !== ($filename = readdir($dh))) { | ||
|  |             if (empty($filename)) { | ||
|  |                 continue; | ||
|  |             } | ||
|  |             if ($filename[0] === '.') { | ||
|  |                 continue; | ||
|  |             } | ||
|  |             unlink($dir . '/' . $filename); | ||
|  |         } | ||
|  |         closedir($dh); | ||
|  |         return true; | ||
|  |     } | ||
|  | 
 | ||
|  |     /** | ||
|  |      * @param HTMLPurifier_Config $config | ||
|  |      * @return bool | ||
|  |      */ | ||
|  |     public function cleanup($config) | ||
|  |     { | ||
|  |         if (!$this->_prepareDir($config)) { | ||
|  |             return false; | ||
|  |         } | ||
|  |         $dir = $this->generateDirectoryPath($config); | ||
|  |         $dh = opendir($dir); | ||
|  |         // See #49 (and above).
 | ||
|  |         if (false === $dh) { | ||
|  |             return false; | ||
|  |         } | ||
|  |         while (false !== ($filename = readdir($dh))) { | ||
|  |             if (empty($filename)) { | ||
|  |                 continue; | ||
|  |             } | ||
|  |             if ($filename[0] === '.') { | ||
|  |                 continue; | ||
|  |             } | ||
|  |             $key = substr($filename, 0, strlen($filename) - 4); | ||
|  |             if ($this->isOld($key, $config)) { | ||
|  |                 unlink($dir . '/' . $filename); | ||
|  |             } | ||
|  |         } | ||
|  |         closedir($dh); | ||
|  |         return true; | ||
|  |     } | ||
|  | 
 | ||
|  |     /** | ||
|  |      * Generates the file path to the serial file corresponding to | ||
|  |      * the configuration and definition name | ||
|  |      * @param HTMLPurifier_Config $config | ||
|  |      * @return string | ||
|  |      * @todo Make protected | ||
|  |      */ | ||
|  |     public function generateFilePath($config) | ||
|  |     { | ||
|  |         $key = $this->generateKey($config); | ||
|  |         return $this->generateDirectoryPath($config) . '/' . $key . '.ser'; | ||
|  |     } | ||
|  | 
 | ||
|  |     /** | ||
|  |      * Generates the path to the directory contain this cache's serial files | ||
|  |      * @param HTMLPurifier_Config $config | ||
|  |      * @return string | ||
|  |      * @note No trailing slash | ||
|  |      * @todo Make protected | ||
|  |      */ | ||
|  |     public function generateDirectoryPath($config) | ||
|  |     { | ||
|  |         $base = $this->generateBaseDirectoryPath($config); | ||
|  |         return $base . '/' . $this->type; | ||
|  |     } | ||
|  | 
 | ||
|  |     /** | ||
|  |      * Generates path to base directory that contains all definition type | ||
|  |      * serials | ||
|  |      * @param HTMLPurifier_Config $config | ||
|  |      * @return mixed|string | ||
|  |      * @todo Make protected | ||
|  |      */ | ||
|  |     public function generateBaseDirectoryPath($config) | ||
|  |     { | ||
|  |         $base = $config->get('Cache.SerializerPath'); | ||
|  |         $base = is_null($base) ? HTMLPURIFIER_PREFIX . '/HTMLPurifier/DefinitionCache/Serializer' : $base; | ||
|  |         return $base; | ||
|  |     } | ||
|  | 
 | ||
|  |     /** | ||
|  |      * Convenience wrapper function for file_put_contents | ||
|  |      * @param string $file File name to write to | ||
|  |      * @param string $data Data to write into file | ||
|  |      * @param HTMLPurifier_Config $config | ||
|  |      * @return int|bool Number of bytes written if success, or false if failure. | ||
|  |      */ | ||
|  |     private function _write($file, $data, $config) | ||
|  |     { | ||
|  |         $result = file_put_contents($file, $data); | ||
|  |         if ($result !== false) { | ||
|  |             // set permissions of the new file (no execute)
 | ||
|  |             $chmod = $config->get('Cache.SerializerPermissions'); | ||
|  |             if ($chmod !== null) { | ||
|  |                 chmod($file, $chmod & 0666); | ||
|  |             } | ||
|  |         } | ||
|  |         return $result; | ||
|  |     } | ||
|  | 
 | ||
|  |     /** | ||
|  |      * Prepares the directory that this type stores the serials in | ||
|  |      * @param HTMLPurifier_Config $config | ||
|  |      * @return bool True if successful | ||
|  |      */ | ||
|  |     private function _prepareDir($config) | ||
|  |     { | ||
|  |         $directory = $this->generateDirectoryPath($config); | ||
|  |         $chmod = $config->get('Cache.SerializerPermissions'); | ||
|  |         if ($chmod === null) { | ||
|  |             if (!@mkdir($directory) && !is_dir($directory)) { | ||
|  |                 trigger_error( | ||
|  |                     'Could not create directory ' . $directory . '', | ||
|  |                     E_USER_WARNING | ||
|  |                 ); | ||
|  |                 return false; | ||
|  |             } | ||
|  |             return true; | ||
|  |         } | ||
|  |         if (!is_dir($directory)) { | ||
|  |             $base = $this->generateBaseDirectoryPath($config); | ||
|  |             if (!is_dir($base)) { | ||
|  |                 trigger_error( | ||
|  |                     'Base directory ' . $base . ' does not exist, | ||
|  |                     please create or change using %Cache.SerializerPath', | ||
|  |                     E_USER_WARNING | ||
|  |                 ); | ||
|  |                 return false; | ||
|  |             } elseif (!$this->_testPermissions($base, $chmod)) { | ||
|  |                 return false; | ||
|  |             } | ||
|  |             if (!@mkdir($directory, $chmod) && !is_dir($directory)) { | ||
|  |                 trigger_error( | ||
|  |                     'Could not create directory ' . $directory . '', | ||
|  |                     E_USER_WARNING | ||
|  |                 ); | ||
|  |                 return false; | ||
|  |             } | ||
|  |             if (!$this->_testPermissions($directory, $chmod)) { | ||
|  |                 return false; | ||
|  |             } | ||
|  |         } elseif (!$this->_testPermissions($directory, $chmod)) { | ||
|  |             return false; | ||
|  |         } | ||
|  |         return true; | ||
|  |     } | ||
|  | 
 | ||
|  |     /** | ||
|  |      * Tests permissions on a directory and throws out friendly | ||
|  |      * error messages and attempts to chmod it itself if possible | ||
|  |      * @param string $dir Directory path | ||
|  |      * @param int $chmod Permissions | ||
|  |      * @return bool True if directory is writable | ||
|  |      */ | ||
|  |     private function _testPermissions($dir, $chmod) | ||
|  |     { | ||
|  |         // early abort, if it is writable, everything is hunky-dory
 | ||
|  |         if (is_writable($dir)) { | ||
|  |             return true; | ||
|  |         } | ||
|  |         if (!is_dir($dir)) { | ||
|  |             // generally, you'll want to handle this beforehand
 | ||
|  |             // so a more specific error message can be given
 | ||
|  |             trigger_error( | ||
|  |                 'Directory ' . $dir . ' does not exist', | ||
|  |                 E_USER_WARNING | ||
|  |             ); | ||
|  |             return false; | ||
|  |         } | ||
|  |         if (function_exists('posix_getuid') && $chmod !== null) { | ||
|  |             // POSIX system, we can give more specific advice
 | ||
|  |             if (fileowner($dir) === posix_getuid()) { | ||
|  |                 // we can chmod it ourselves
 | ||
|  |                 $chmod = $chmod | 0700; | ||
|  |                 if (chmod($dir, $chmod)) { | ||
|  |                     return true; | ||
|  |                 } | ||
|  |             } elseif (filegroup($dir) === posix_getgid()) { | ||
|  |                 $chmod = $chmod | 0070; | ||
|  |             } else { | ||
|  |                 // PHP's probably running as nobody, so we'll
 | ||
|  |                 // need to give global permissions
 | ||
|  |                 $chmod = $chmod | 0777; | ||
|  |             } | ||
|  |             trigger_error( | ||
|  |                 'Directory ' . $dir . ' not writable, ' . | ||
|  |                 'please chmod to ' . decoct($chmod), | ||
|  |                 E_USER_WARNING | ||
|  |             ); | ||
|  |         } else { | ||
|  |             // generic error message
 | ||
|  |             trigger_error( | ||
|  |                 'Directory ' . $dir . ' not writable, ' . | ||
|  |                 'please alter file permissions', | ||
|  |                 E_USER_WARNING | ||
|  |             ); | ||
|  |         } | ||
|  |         return false; | ||
|  |     } | ||
|  | } | ||
|  | 
 | ||
|  | // vim: et sw=4 sts=4
 |