352 lines
		
	
	
		
			8.4 KiB
		
	
	
	
		
			PHP
		
	
	
	
	
	
			
		
		
	
	
			352 lines
		
	
	
		
			8.4 KiB
		
	
	
	
		
			PHP
		
	
	
	
	
	
<?php
 | 
						|
namespace W3TC;
 | 
						|
 | 
						|
/**
 | 
						|
 * Amazon S3 CDN engine
 | 
						|
 */
 | 
						|
 | 
						|
if ( !class_exists( 'S3Compatible' ) ) {
 | 
						|
	require_once W3TC_LIB_DIR . '/S3Compatible.php';
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * class CdnEngine_S3
 | 
						|
 */
 | 
						|
class CdnEngine_S3_Compatible extends CdnEngine_Base {
 | 
						|
	/**
 | 
						|
	 * S3 object
 | 
						|
	 */
 | 
						|
	private $_s3 = null;
 | 
						|
 | 
						|
	/**
 | 
						|
	 * PHP5 Constructor
 | 
						|
	 *
 | 
						|
	 * @param array   $config
 | 
						|
	 */
 | 
						|
	function __construct( $config = array() ) {
 | 
						|
		$config = array_merge( array(
 | 
						|
				'key' => '',
 | 
						|
				'secret' => '',
 | 
						|
				'bucket' => '',
 | 
						|
				'cname' => array(),
 | 
						|
			), $config );
 | 
						|
 | 
						|
		$this->_s3 = new \S3Compatible( $config['key'], $config['secret'], false,
 | 
						|
			$config['api_host'] );
 | 
						|
		$this->_s3->setSignatureVersion( 'v2' );
 | 
						|
 | 
						|
		parent::__construct( $config );
 | 
						|
	}
 | 
						|
 | 
						|
	/**
 | 
						|
	 * Formats URL
 | 
						|
	 *
 | 
						|
	 * @param string  $path
 | 
						|
	 * @return string
 | 
						|
	 */
 | 
						|
	function _format_url( $path ) {
 | 
						|
		$domain = $this->get_domain( $path );
 | 
						|
 | 
						|
		if ( $domain ) {
 | 
						|
			$scheme = $this->_get_scheme();
 | 
						|
 | 
						|
			// it does not support '+', requires '%2B'
 | 
						|
			$path = str_replace( '+', '%2B', $path );
 | 
						|
			$url = sprintf( '%s://%s/%s', $scheme, $domain, $path );
 | 
						|
 | 
						|
			return $url;
 | 
						|
		}
 | 
						|
 | 
						|
		return false;
 | 
						|
	}
 | 
						|
 | 
						|
	/**
 | 
						|
	 * Uploads files to S3
 | 
						|
	 *
 | 
						|
	 * @param array   $files
 | 
						|
	 * @param array   $results
 | 
						|
	 * @param boolean $force_rewrite
 | 
						|
	 * @return boolean
 | 
						|
	 */
 | 
						|
	function upload( $files, &$results, $force_rewrite = false,
 | 
						|
		$timeout_time = NULL ) {
 | 
						|
 | 
						|
		$error = null;
 | 
						|
 | 
						|
		foreach ( $files as $file ) {
 | 
						|
			$local_path = $file['local_path'];
 | 
						|
			$remote_path = $file['remote_path'];
 | 
						|
 | 
						|
			// process at least one item before timeout so that progress goes on
 | 
						|
			if ( !empty( $results ) ) {
 | 
						|
				if ( !is_null( $timeout_time ) && time() > $timeout_time ) {
 | 
						|
					return 'timeout';
 | 
						|
				}
 | 
						|
			}
 | 
						|
 | 
						|
			$results[] = $this->_upload( $file, $force_rewrite );
 | 
						|
 | 
						|
			if ( $this->_config['compression'] && $this->_may_gzip( $remote_path ) ) {
 | 
						|
				$file['remote_path_gzip'] = $remote_path . $this->_gzip_extension;
 | 
						|
				$results[] = $this->_upload_gzip( $file, $force_rewrite );
 | 
						|
			}
 | 
						|
		}
 | 
						|
 | 
						|
		return !$this->_is_error( $results );
 | 
						|
	}
 | 
						|
 | 
						|
	/**
 | 
						|
	 * Uploads single file to S3
 | 
						|
	 *
 | 
						|
	 * @param array   CDN file array
 | 
						|
	 * @param boolean $force_rewrite
 | 
						|
	 * @return array
 | 
						|
	 */
 | 
						|
	function _upload( $file, $force_rewrite = false ) {
 | 
						|
		$local_path = $file['local_path'];
 | 
						|
		$remote_path = $file['remote_path'];
 | 
						|
 | 
						|
		if ( !file_exists( $local_path ) ) {
 | 
						|
			return $this->_get_result( $local_path, $remote_path,
 | 
						|
				W3TC_CDN_RESULT_ERROR, 'Source file not found.', $file );
 | 
						|
		}
 | 
						|
 | 
						|
		if ( !$force_rewrite ) {
 | 
						|
			$this->_set_error_handler();
 | 
						|
			$info = @$this->_s3->getObjectInfo( $this->_config['bucket'],
 | 
						|
				$remote_path );
 | 
						|
			$this->_restore_error_handler();
 | 
						|
 | 
						|
			if ( $info ) {
 | 
						|
				$hash = @md5_file( $local_path );
 | 
						|
				$s3_hash = ( isset( $info['hash'] ) ? $info['hash'] : '' );
 | 
						|
 | 
						|
				if ( $hash === $s3_hash ) {
 | 
						|
					return $this->_get_result( $local_path, $remote_path,
 | 
						|
						W3TC_CDN_RESULT_OK, 'Object up-to-date.', $file );
 | 
						|
				}
 | 
						|
			}
 | 
						|
		}
 | 
						|
 | 
						|
		$headers = $this->get_headers_for_file( $file, array( 'ETag' => '*' ) );
 | 
						|
 | 
						|
		$this->_set_error_handler();
 | 
						|
		$result = @$this->_s3->putObjectFile( $local_path,
 | 
						|
			$this->_config['bucket'], $remote_path,
 | 
						|
			\S3Compatible::ACL_PUBLIC_READ, array(), $headers );
 | 
						|
		$this->_restore_error_handler();
 | 
						|
 | 
						|
		if ( $result ) {
 | 
						|
			return $this->_get_result( $local_path, $remote_path,
 | 
						|
				W3TC_CDN_RESULT_OK, 'OK', $file );
 | 
						|
		}
 | 
						|
 | 
						|
		return $this->_get_result( $local_path, $remote_path,
 | 
						|
			W3TC_CDN_RESULT_ERROR,
 | 
						|
			sprintf( 'Unable to put object (%s).', $this->_get_last_error() ),
 | 
						|
			$file );
 | 
						|
	}
 | 
						|
 | 
						|
	/**
 | 
						|
	 * Uploads gzip version of file
 | 
						|
	 *
 | 
						|
	 * @param string  $local_path
 | 
						|
	 * @param string  $remote_path
 | 
						|
	 * @param boolean $force_rewrite
 | 
						|
	 * @return array
 | 
						|
	 */
 | 
						|
	function _upload_gzip( $file, $force_rewrite = false ) {
 | 
						|
		$local_path = $file['local_path'];
 | 
						|
		$remote_path = $file['remote_path_gzip'];
 | 
						|
 | 
						|
		if ( !function_exists( 'gzencode' ) )
 | 
						|
			return $this->_get_result( $local_path, $remote_path,
 | 
						|
				W3TC_CDN_RESULT_ERROR, "GZIP library doesn't exist.", $file );
 | 
						|
 | 
						|
		if ( !file_exists( $local_path ) )
 | 
						|
			return $this->_get_result( $local_path, $remote_path,
 | 
						|
				W3TC_CDN_RESULT_ERROR, 'Source file not found.', $file );
 | 
						|
 | 
						|
		$contents = @file_get_contents( $local_path );
 | 
						|
		if ( $contents === false )
 | 
						|
			return $this->_get_result( $local_path, $remote_path,
 | 
						|
				W3TC_CDN_RESULT_ERROR, 'Unable to read file.', $file );
 | 
						|
 | 
						|
		$data = gzencode( $contents );
 | 
						|
 | 
						|
		if ( !$force_rewrite ) {
 | 
						|
			$this->_set_error_handler();
 | 
						|
			$info = @$this->_s3->getObjectInfo( $this->_config['bucket'],
 | 
						|
				$remote_path );
 | 
						|
			$this->_restore_error_handler();
 | 
						|
 | 
						|
			if ( $info ) {
 | 
						|
				$hash = md5( $data );
 | 
						|
				$s3_hash = ( isset( $info['hash'] ) ? $info['hash'] : '' );
 | 
						|
 | 
						|
				if ( $hash === $s3_hash ) {
 | 
						|
					return $this->_get_result( $local_path, $remote_path,
 | 
						|
						W3TC_CDN_RESULT_OK, 'Object up-to-date.', $file );
 | 
						|
				}
 | 
						|
			}
 | 
						|
		}
 | 
						|
 | 
						|
		$headers = $this->get_headers_for_file( $file, array( 'ETag' => '*' ) );
 | 
						|
		$headers = array_merge( $headers, array(
 | 
						|
				'Vary' => 'Accept-Encoding',
 | 
						|
				'Content-Encoding' => 'gzip'
 | 
						|
			) );
 | 
						|
 | 
						|
		$this->_set_error_handler();
 | 
						|
		$result = @$this->_s3->putObjectString( $data, $this->_config['bucket'],
 | 
						|
			$remote_path, \S3Compatible::ACL_PUBLIC_READ, array(), $headers );
 | 
						|
		$this->_restore_error_handler();
 | 
						|
 | 
						|
		if ( $result )
 | 
						|
			return $this->_get_result( $local_path, $remote_path,
 | 
						|
				W3TC_CDN_RESULT_OK, 'OK', $file );
 | 
						|
 | 
						|
		return $this->_get_result( $local_path, $remote_path,
 | 
						|
			W3TC_CDN_RESULT_ERROR, sprintf( 'Unable to put object (%s).',
 | 
						|
				$this->_get_last_error() ), $file );
 | 
						|
	}
 | 
						|
 | 
						|
	/**
 | 
						|
	 * Deletes files from S3
 | 
						|
	 *
 | 
						|
	 * @param array   $files
 | 
						|
	 * @param array   $results
 | 
						|
	 * @return boolean
 | 
						|
	 */
 | 
						|
	function delete( $files, &$results ) {
 | 
						|
		$error = null;
 | 
						|
 | 
						|
		foreach ( $files as $file ) {
 | 
						|
			$local_path = $file['local_path'];
 | 
						|
			$remote_path = $file['remote_path'];
 | 
						|
 | 
						|
			$this->_set_error_handler();
 | 
						|
			$result = @$this->_s3->deleteObject( $this->_config['bucket'],
 | 
						|
				$remote_path );
 | 
						|
			$this->_restore_error_handler();
 | 
						|
 | 
						|
			if ( $result ) {
 | 
						|
				$results[] = $this->_get_result( $local_path, $remote_path,
 | 
						|
					W3TC_CDN_RESULT_OK, 'OK', $file );
 | 
						|
			} else {
 | 
						|
				$results[] = $this->_get_result( $local_path, $remote_path,
 | 
						|
					W3TC_CDN_RESULT_ERROR,
 | 
						|
					sprintf( 'Unable to delete object (%s).',
 | 
						|
						$this->_get_last_error() ), $file );
 | 
						|
			}
 | 
						|
 | 
						|
			if ( $this->_config['compression'] ) {
 | 
						|
				$remote_path_gzip = $remote_path . $this->_gzip_extension;
 | 
						|
 | 
						|
				$this->_set_error_handler();
 | 
						|
				$result = @$this->_s3->deleteObject( $this->_config['bucket'],
 | 
						|
					$remote_path_gzip );
 | 
						|
				$this->_restore_error_handler();
 | 
						|
 | 
						|
				if ( $result ) {
 | 
						|
					$results[] = $this->_get_result( $local_path,
 | 
						|
						$remote_path_gzip, W3TC_CDN_RESULT_OK, 'OK', $file );
 | 
						|
				} else {
 | 
						|
					$results[] = $this->_get_result( $local_path,
 | 
						|
						$remote_path_gzip, W3TC_CDN_RESULT_ERROR,
 | 
						|
						sprintf( 'Unable to delete object (%s).',
 | 
						|
							$this->_get_last_error() ),
 | 
						|
						$file );
 | 
						|
				}
 | 
						|
			}
 | 
						|
		}
 | 
						|
 | 
						|
		return !$this->_is_error( $results );
 | 
						|
	}
 | 
						|
 | 
						|
	/**
 | 
						|
	 * Tests S3
 | 
						|
	 *
 | 
						|
	 * @param string  $error
 | 
						|
	 * @return boolean
 | 
						|
	 */
 | 
						|
	function test( &$error ) {
 | 
						|
		if ( !parent::test( $error ) ) {
 | 
						|
			return false;
 | 
						|
		}
 | 
						|
 | 
						|
		$string = 'test_s3_' . md5( time() );
 | 
						|
 | 
						|
		$this->_set_error_handler();
 | 
						|
 | 
						|
		if ( !@$this->_s3->putObjectString( $string, $this->_config['bucket'],
 | 
						|
				$string, \S3Compatible::ACL_PUBLIC_READ ) ) {
 | 
						|
			$error = sprintf( 'Unable to put object (%s).',
 | 
						|
				$this->_get_last_error() );
 | 
						|
 | 
						|
			$this->_restore_error_handler();
 | 
						|
 | 
						|
			return false;
 | 
						|
		}
 | 
						|
 | 
						|
		$object = @$this->_s3->getObject( $this->_config['bucket'], $string );
 | 
						|
		if ( !$object ) {
 | 
						|
			$error = sprintf( 'Unable to get object (%s).',
 | 
						|
				$this->_get_last_error() );
 | 
						|
 | 
						|
			$this->_restore_error_handler();
 | 
						|
			return false;
 | 
						|
		}
 | 
						|
 | 
						|
		if ( $object->body != $string ) {
 | 
						|
			$error = 'Objects are not equal.';
 | 
						|
 | 
						|
			@$this->_s3->deleteObject( $this->_config['bucket'], $string );
 | 
						|
			$this->_restore_error_handler();
 | 
						|
 | 
						|
			return false;
 | 
						|
		}
 | 
						|
 | 
						|
		if ( !@$this->_s3->deleteObject( $this->_config['bucket'], $string ) ) {
 | 
						|
			$error = sprintf( 'Unable to delete object (%s).',
 | 
						|
				$this->_get_last_error() );
 | 
						|
 | 
						|
			$this->_restore_error_handler();
 | 
						|
 | 
						|
			return false;
 | 
						|
		}
 | 
						|
 | 
						|
		$this->_restore_error_handler();
 | 
						|
 | 
						|
		return true;
 | 
						|
	}
 | 
						|
 | 
						|
	/**
 | 
						|
	 * Returns CDN domain
 | 
						|
	 *
 | 
						|
	 * @return array
 | 
						|
	 */
 | 
						|
	function get_domains() {
 | 
						|
		return (array) $this->_config['cname'];
 | 
						|
	}
 | 
						|
 | 
						|
	/**
 | 
						|
	 * Returns via string
 | 
						|
	 *
 | 
						|
	 * @return string
 | 
						|
	 */
 | 
						|
	function get_via() {
 | 
						|
		return sprintf( 'S3-compatible: %s', parent::get_via() );
 | 
						|
	}
 | 
						|
 | 
						|
	/**
 | 
						|
	 * How and if headers should be set
 | 
						|
	 *
 | 
						|
	 * @return string W3TC_CDN_HEADER_NONE, W3TC_CDN_HEADER_UPLOADABLE,
 | 
						|
	 *   W3TC_CDN_HEADER_MIRRORING
 | 
						|
	 */
 | 
						|
	function headers_support() {
 | 
						|
		return W3TC_CDN_HEADER_UPLOADABLE;
 | 
						|
	}
 | 
						|
}
 |