laipower/wp-content/plugins/w3-total-cache/CdnEngine_S3_Compatible.php

352 lines
8.4 KiB
PHP
Raw Normal View History

<?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;
}
}