947 lines
		
	
	
		
			23 KiB
		
	
	
	
		
			PHP
		
	
	
	
	
	
			
		
		
	
	
			947 lines
		
	
	
		
			23 KiB
		
	
	
	
		
			PHP
		
	
	
	
	
	
| <?php
 | |
| /**
 | |
|  * File: DbCache_WpdbInjection_QueryCaching.php
 | |
|  *
 | |
|  * @package W3TC
 | |
|  */
 | |
| 
 | |
| namespace W3TC;
 | |
| 
 | |
| /**
 | |
|  * Class: DbCache_WpdbInjection_QueryCaching
 | |
|  *
 | |
|  * phpcs:disable PSR2.Classes.PropertyDeclaration.Underscore
 | |
|  * phpcs:disable PSR2.Methods.MethodDeclaration.Underscore
 | |
|  */
 | |
| class DbCache_WpdbInjection_QueryCaching extends DbCache_WpdbInjection {
 | |
| 	/**
 | |
| 	 * Queries total.
 | |
| 	 *
 | |
| 	 * @var int
 | |
| 	 */
 | |
| 	public $query_total = 0;
 | |
| 
 | |
| 	/**
 | |
| 	 * Query cache hits.
 | |
| 	 *
 | |
| 	 * @var int
 | |
| 	 */
 | |
| 	public $query_hits = 0;
 | |
| 
 | |
| 	/**
 | |
| 	 * Query cache misses.
 | |
| 	 *
 | |
| 	 * @var int
 | |
| 	 */
 | |
| 	public $query_misses = 0;
 | |
| 
 | |
| 	/**
 | |
| 	 * Time total taken by queries, in microsecs.
 | |
| 	 *
 | |
| 	 * @var int
 | |
| 	 */
 | |
| 	public $time_total = 0;
 | |
| 
 | |
| 	/**
 | |
| 	 * Config.
 | |
| 	 *
 | |
| 	 * @var Config
 | |
| 	 */
 | |
| 	public $_config = null;
 | |
| 
 | |
| 	/**
 | |
| 	 * Lifetime.
 | |
| 	 *
 | |
| 	 * @var int
 | |
| 	 */
 | |
| 	public $_lifetime = null;
 | |
| 
 | |
| 	/**
 | |
| 	 * Number of cache flushes during http request processing.
 | |
| 	 *
 | |
| 	 * @var int
 | |
| 	 */
 | |
| 	private $cache_flushes = 0;
 | |
| 
 | |
| 	/**
 | |
| 	 * Request-global cache reject reason.
 | |
| 	 *
 | |
| 	 * @var string
 | |
| 	 */
 | |
| 	private $cache_reject_reason = null;
 | |
| 
 | |
| 	/**
 | |
| 	 * Request-global check reject scope.
 | |
| 	 *
 | |
| 	 * @var bool
 | |
| 	 */
 | |
| 	private $cache_reject_request_wide = false;
 | |
| 
 | |
| 	/**
 | |
| 	 * Debug flag.
 | |
| 	 *
 | |
| 	 * @var bool
 | |
| 	 */
 | |
| 	private $debug = false;
 | |
| 
 | |
| 	/**
 | |
| 	 * Reject log flag.
 | |
| 	 *
 | |
| 	 * @var bool
 | |
| 	 */
 | |
| 	private $reject_logged = false;
 | |
| 
 | |
| 	/**
 | |
| 	 * Log filehandle flag.
 | |
| 	 *
 | |
| 	 * @var bool
 | |
| 	 */
 | |
| 	private $log_filehandle = false;
 | |
| 
 | |
| 	/**
 | |
| 	 * Reject constants flag.
 | |
| 	 *
 | |
| 	 * @var bool
 | |
| 	 */
 | |
| 	private $reject_constants = false;
 | |
| 
 | |
| 	/**
 | |
| 	 * Use filters flag.
 | |
| 	 *
 | |
| 	 * @var bool
 | |
| 	 */
 | |
| 	private $use_filters = false;
 | |
| 
 | |
| 	/**
 | |
| 	 * Result of check if caching is possible at the level of current http request.
 | |
| 	 *
 | |
| 	 * @var bool
 | |
| 	 */
 | |
| 	private $can_cache_once_per_request_result = null;
 | |
| 
 | |
| 	/**
 | |
| 	 * Constructor.
 | |
| 	 */
 | |
| 	public function __construct() {
 | |
| 		$c                      = Dispatcher::config();
 | |
| 		$this->_config          = $c;
 | |
| 		$this->_lifetime        = $c->get_integer( 'dbcache.lifetime' );
 | |
| 		$this->debug            = $c->get_boolean( 'dbcache.debug' );
 | |
| 		$this->reject_logged    = $c->get_boolean( 'dbcache.reject.logged' );
 | |
| 		$this->reject_constants = $c->get_array( 'dbcache.reject.constants' );
 | |
| 		$this->use_filters      = $this->_config->get_boolean( 'dbcache.use_filters' );
 | |
| 	}
 | |
| 
 | |
| 	/**
 | |
| 	 * Executes query.
 | |
| 	 *
 | |
| 	 * @param string $query Query.
 | |
| 	 * @return int
 | |
| 	 */
 | |
| 	public function query( $query ) {
 | |
| 		if ( ! $this->wpdb_mixin->ready ) {
 | |
| 			return $this->next_injection->query( $query );
 | |
| 		}
 | |
| 
 | |
| 		$reject_reason     = '';
 | |
| 		$is_cache_hit      = false;
 | |
| 		$data              = false;
 | |
| 		$time_total        = 0;
 | |
| 		$group             = '';
 | |
| 		$flush_after_query = false;
 | |
| 
 | |
| 		$this->query_total++;
 | |
| 
 | |
| 		$caching = $this->_can_cache( $query, $reject_reason );
 | |
| 		if ( preg_match( '~^\s*start transaction\b~is', $query ) ) {
 | |
| 			$this->cache_reject_reason = 'transaction';
 | |
| 			$reject_reason             = $this->cache_reject_reason;
 | |
| 			$caching                   = false;
 | |
| 		}
 | |
| 
 | |
| 		if ( preg_match( '~^\s*insert\b|^\s*delete\b|^\s*update\b|^\s*replace\b|^\s*commit\b|^\s*truncate\b|^\s*drop\b|^\s*create\b~is', $query ) ) {
 | |
| 			$this->cache_reject_reason = 'modification query';
 | |
| 			$reject_reason             = $this->cache_reject_reason;
 | |
| 			$caching                   = false;
 | |
| 			$flush_after_query         = true;
 | |
| 		}
 | |
| 
 | |
| 		if ( $this->use_filters && function_exists( 'apply_filters' ) ) {
 | |
| 			$reject_reason = apply_filters(
 | |
| 				'w3tc_dbcache_can_cache_sql',
 | |
| 				( $caching ? '' : $reject_reason ),
 | |
| 				$query
 | |
| 			);
 | |
| 
 | |
| 			$caching = empty( $reject_reason );
 | |
| 		}
 | |
| 
 | |
| 		if ( $caching ) {
 | |
| 			$this->wpdb_mixin->timer_start();
 | |
| 			$cache      = $this->_get_cache();
 | |
| 			$group      = $this->_get_group( $query );
 | |
| 			$data       = $cache->get( md5( $query ), $group );
 | |
| 			$time_total = $this->wpdb_mixin->timer_stop();
 | |
| 		}
 | |
| 
 | |
| 		if ( is_array( $data ) ) {
 | |
| 			$is_cache_hit = true;
 | |
| 			$this->query_hits++;
 | |
| 
 | |
| 			$this->wpdb_mixin->last_error  = $data['last_error'];
 | |
| 			$this->wpdb_mixin->last_query  = $data['last_query'];
 | |
| 			$this->wpdb_mixin->last_result = $data['last_result'];
 | |
| 			$this->wpdb_mixin->col_info    = $data['col_info'];
 | |
| 			$this->wpdb_mixin->num_rows    = $data['num_rows'];
 | |
| 			$return_val                    = $data['return_val'];
 | |
| 		} else {
 | |
| 			$this->query_misses++;
 | |
| 
 | |
| 			$this->wpdb_mixin->timer_start();
 | |
| 			$return_val = $this->next_injection->query( $query );
 | |
| 			$time_total = $this->wpdb_mixin->timer_stop();
 | |
| 
 | |
| 			if ( $flush_after_query ) {
 | |
| 				$group = $this->_get_group( $query );
 | |
| 
 | |
| 				$this->_flush_cache_for_sql_group(
 | |
| 					$group,
 | |
| 					array( 'modification_query' => $query )
 | |
| 				);
 | |
| 			}
 | |
| 
 | |
| 			if ( $caching ) {
 | |
| 				$data = array(
 | |
| 					'last_error'  => $this->wpdb_mixin->last_error,
 | |
| 					'last_query'  => $this->wpdb_mixin->last_query,
 | |
| 					'last_result' => $this->wpdb_mixin->last_result,
 | |
| 					'col_info'    => $this->wpdb_mixin->col_info,
 | |
| 					'num_rows'    => $this->wpdb_mixin->num_rows,
 | |
| 					'return_val'  => $return_val,
 | |
| 				);
 | |
| 
 | |
| 				$cache = $this->_get_cache();
 | |
| 				$group = $this->_get_group( $query );
 | |
| 
 | |
| 				$filter_data = array(
 | |
| 					'query'      => $query,
 | |
| 					'group'      => $group,
 | |
| 					'content'    => $data,
 | |
| 					'expiration' => $this->_lifetime,
 | |
| 				);
 | |
| 
 | |
| 				if ( $this->use_filters && function_exists( 'apply_filters' ) ) {
 | |
| 					$filter_data = apply_filters( 'w3tc_dbcache_cache_set', $filter_data );
 | |
| 				}
 | |
| 
 | |
| 				$cache->set(
 | |
| 					md5( $filter_data['query'] ),
 | |
| 					$filter_data['content'],
 | |
| 					$filter_data['expiration'],
 | |
| 					$filter_data['group']
 | |
| 				);
 | |
| 			}
 | |
| 		}
 | |
| 
 | |
| 		if ( $this->debug ) {
 | |
| 			$this->log_query(
 | |
| 				array(
 | |
| 					gmdate( 'r' ),
 | |
| 					strtr(
 | |
| 						isset( $_SERVER['REQUEST_URI'] ) ?
 | |
| 							filter_var( stripslashes( $_SERVER['REQUEST_URI'] ), FILTER_SANITIZE_URL ) : '', // phpcs:ignore WordPress.Security.ValidatedSanitizedInput
 | |
| 						"<>\r\n",
 | |
| 						'..  '
 | |
| 					),
 | |
| 					strtr( $query, "<>\r\n", '..  ' ), // query.
 | |
| 					(int) ( $time_total * 1000000 ), // time_total in seconds.
 | |
| 					$reject_reason, // reason.
 | |
| 					$is_cache_hit, // cached.
 | |
| 					( $data ? strlen( serialize( $data ) ) : 0 ), // phpcs:ignore WordPress.PHP.DiscouragedPHPFunctions.serialize_serialize -- data size
 | |
| 					strtr( $group, "<>\r\n", '..  ' ), // group.
 | |
| 				)
 | |
| 			);
 | |
| 		}
 | |
| 
 | |
| 		$this->time_total += $time_total;
 | |
| 
 | |
| 		return $return_val;
 | |
| 	}
 | |
| 
 | |
| 	/**
 | |
| 	 * Escape.
 | |
| 	 *
 | |
| 	 * @param array $data Data.
 | |
| 	 */
 | |
| 	public function _escape( $data ) {
 | |
| 		return $this->next_injection->_escape( $data );
 | |
| 	}
 | |
| 
 | |
| 	/**
 | |
| 	 * Prepare.
 | |
| 	 *
 | |
| 	 * @param string $query Query.
 | |
| 	 * @param array  $args  Arguments.
 | |
| 	 */
 | |
| 	public function prepare( $query, $args ) {
 | |
| 		return $this->next_injection->prepare( $query, $args );
 | |
| 	}
 | |
| 
 | |
| 	/**
 | |
| 	 * Initializes object, calls underlying processor.
 | |
| 	 */
 | |
| 	public function initialize() {
 | |
| 		return $this->next_injection->initialize();
 | |
| 	}
 | |
| 
 | |
| 	/**
 | |
| 	 * Insert a row into a table.
 | |
| 	 *
 | |
| 	 * @param string       $table  Table.
 | |
| 	 * @param array        $data   Data.
 | |
| 	 * @param array|string $format Format.
 | |
| 	 *
 | |
| 	 * @return int|false
 | |
| 	 */
 | |
| 	public function insert( $table, $data, $format = null ) {
 | |
| 		return $this->next_injection->insert( $table, $data, $format );
 | |
| 	}
 | |
| 
 | |
| 	/**
 | |
| 	 * Replace a row into a table.
 | |
| 	 *
 | |
| 	 * @param string       $table  Table.
 | |
| 	 * @param array        $data   Data.
 | |
| 	 * @param array|string $format Format.
 | |
| 	 *
 | |
| 	 * @return int|false
 | |
| 	 */
 | |
| 	public function replace( $table, $data, $format = null ) {
 | |
| 		$group = $this->_get_group( $table );
 | |
| 		$this->_flush_cache_for_sql_group(
 | |
| 			$group,
 | |
| 			array( 'wpdb_replace' => $table )
 | |
| 		);
 | |
| 
 | |
| 		return $this->next_injection->replace( $table, $data, $format );
 | |
| 	}
 | |
| 
 | |
| 	/**
 | |
| 	 * Update a row in the table
 | |
| 	 *
 | |
| 	 * @param string       $table        Table.
 | |
| 	 * @param array        $data         Data.
 | |
| 	 * @param array        $where        Where.
 | |
| 	 * @param array|string $format       Format.
 | |
| 	 * @param array|string $where_format Format where.
 | |
| 	 *
 | |
| 	 * @return int|false
 | |
| 	 */
 | |
| 	public function update( $table, $data, $where, $format = null, $where_format = null ) {
 | |
| 		$group = $this->_get_group( $table );
 | |
| 		$this->_flush_cache_for_sql_group( $group, array( 'wpdb_update' => $table ) );
 | |
| 		return $this->next_injection->update( $table, $data, $where, $format, $where_format );
 | |
| 	}
 | |
| 
 | |
| 	/**
 | |
| 	 * Deletes from table.
 | |
| 	 *
 | |
| 	 * @param string       $table        Table.
 | |
| 	 * @param array        $where        Where.
 | |
| 	 * @param array|string $where_format Format where.
 | |
| 	 *
 | |
| 	 * @return int|false
 | |
| 	 */
 | |
| 	public function delete( $table, $where, $where_format = null ) {
 | |
| 		$group = $this->_get_group( $table );
 | |
| 		$this->_flush_cache_for_sql_group( $group, array( 'wpdb_delete' => $table ) );
 | |
| 		return $this->next_injection->delete( $table, $where, $where_format );
 | |
| 	}
 | |
| 
 | |
| 	/**
 | |
| 	 * Flushes cache.
 | |
| 	 *
 | |
| 	 * @param array $extras Extra arguments.
 | |
| 	 *
 | |
| 	 * @return bool
 | |
| 	 */
 | |
| 	public function flush_cache( $extras = array() ) {
 | |
| 		return $this->_flush_cache_for_sql_group( 'remaining', $extras );
 | |
| 	}
 | |
| 
 | |
| 	/**
 | |
| 	 * Flush cache for SQL groups.
 | |
| 	 *
 | |
| 	 * @access private
 | |
| 	 *
 | |
| 	 * @param string $group  Group.
 | |
| 	 * @param array  $extras Extra arguments.
 | |
| 	 *
 | |
| 	 * @return bool
 | |
| 	 */
 | |
| 	private function _flush_cache_for_sql_group( $group, $extras = array() ) {
 | |
| 		$this->wpdb_mixin->timer_start();
 | |
| 
 | |
| 		if ( $this->debug ) {
 | |
| 			$filename = Util_Debug::log(
 | |
| 				'dbcache',
 | |
| 				'flushing based on sqlquery group ' . $group .
 | |
| 				' with extras ' . wp_json_encode( $extras )
 | |
| 			);
 | |
| 		}
 | |
| 		if ( $this->_config->get_boolean( 'dbcache.debug_purge' ) ) {
 | |
| 			Util_Debug::log_purge(
 | |
| 				'dbcache',
 | |
| 				'_flush_cache_for_sql_group',
 | |
| 				array( $group, $extras )
 | |
| 			);
 | |
| 		}
 | |
| 
 | |
| 		$cache        = $this->_get_cache();
 | |
| 		$flush_groups = $this->_get_flush_groups( $group, $extras );
 | |
| 		$v            = true;
 | |
| 
 | |
| 		$this->cache_flushes++;
 | |
| 
 | |
| 		foreach ( $flush_groups as $f_group => $nothing ) {
 | |
| 			if ( $this->debug ) {
 | |
| 				$filename = Util_Debug::log( 'dbcache', 'flush group ' . $f_group );
 | |
| 			}
 | |
| 			$v &= $cache->flush( $f_group );
 | |
| 		}
 | |
| 
 | |
| 		$this->time_total += $this->wpdb_mixin->timer_stop();
 | |
| 
 | |
| 		return $v;
 | |
| 	}
 | |
| 
 | |
| 	/**
 | |
| 	 * Returns cache object.
 | |
| 	 *
 | |
| 	 * @return W3_Cache_Base
 | |
| 	 */
 | |
| 	public function _get_cache() {
 | |
| 		static $cache = array();
 | |
| 
 | |
| 		if ( ! isset( $cache[0] ) ) {
 | |
| 			$engine = $this->_config->get_string( 'dbcache.engine' );
 | |
| 
 | |
| 			switch ( $engine ) {
 | |
| 				case 'memcached':
 | |
| 					$engine_config = array(
 | |
| 						'servers'           => $this->_config->get_array( 'dbcache.memcached.servers' ),
 | |
| 						'persistent'        => $this->_config->get_boolean( 'dbcache.memcached.persistent' ),
 | |
| 						'aws_autodiscovery' => $this->_config->get_boolean( 'dbcache.memcached.aws_autodiscovery' ),
 | |
| 						'username'          => $this->_config->get_string( 'dbcache.memcached.username' ),
 | |
| 						'password'          => $this->_config->get_string( 'dbcache.memcached.password' ),
 | |
| 						'binary_protocol'   => $this->_config->get_boolean( 'dbcache.memcached.binary_protocol' ),
 | |
| 					);
 | |
| 					break;
 | |
| 
 | |
| 				case 'redis':
 | |
| 					$engine_config = array(
 | |
| 						'servers'                 => $this->_config->get_array( 'dbcache.redis.servers' ),
 | |
| 						'verify_tls_certificates' => $this->_config->get_boolean( 'dbcache.redis.verify_tls_certificates' ),
 | |
| 						'persistent'              => $this->_config->get_boolean( 'dbcache.redis.persistent' ),
 | |
| 						'timeout'                 => $this->_config->get_integer( 'dbcache.redis.timeout' ),
 | |
| 						'retry_interval'          => $this->_config->get_integer( 'dbcache.redis.retry_interval' ),
 | |
| 						'read_timeout'            => $this->_config->get_integer( 'dbcache.redis.read_timeout' ),
 | |
| 						'dbid'                    => $this->_config->get_integer( 'dbcache.redis.dbid' ),
 | |
| 						'password'                => $this->_config->get_string( 'dbcache.redis.password' ),
 | |
| 					);
 | |
| 					break;
 | |
| 
 | |
| 				case 'file':
 | |
| 					$engine_config = array(
 | |
| 						'use_wp_hash'     => true,
 | |
| 						'section'         => 'db',
 | |
| 						'locking'         => $this->_config->get_boolean( 'dbcache.file.locking' ),
 | |
| 						'flush_timelimit' => $this->_config->get_integer( 'timelimit.cache_flush' ),
 | |
| 					);
 | |
| 					break;
 | |
| 
 | |
| 				default:
 | |
| 					$engine_config = array();
 | |
| 			}
 | |
| 			$engine_config['module']      = 'dbcache';
 | |
| 			$engine_config['host']        = Util_Environment::host();
 | |
| 			$engine_config['instance_id'] = Util_Environment::instance_id();
 | |
| 
 | |
| 			$cache[0] = Cache::instance( $engine, $engine_config );
 | |
| 		}
 | |
| 
 | |
| 		return $cache[0];
 | |
| 	}
 | |
| 
 | |
| 	/**
 | |
| 	 * Check if can cache sql.
 | |
| 	 *
 | |
| 	 * @param string $sql                 SQL query.
 | |
| 	 * @param string $cache_reject_reason Cache reject reason.
 | |
| 	 *
 | |
| 	 * @return boolean
 | |
| 	 */
 | |
| 	public function _can_cache( $sql, &$cache_reject_reason ) {
 | |
| 		/**
 | |
| 		 * Skip if request-wide reject reason specified.
 | |
| 		 * Note - as a result requedt-wide checks are done only once per request.
 | |
| 		 */
 | |
| 		if ( ! is_null( $this->cache_reject_reason ) ) {
 | |
| 			$cache_reject_reason             = $this->cache_reject_reason;
 | |
| 			$this->cache_reject_request_wide = true;
 | |
| 			return false;
 | |
| 		}
 | |
| 
 | |
| 		/**
 | |
| 		 * Do once-per-request check if needed.
 | |
| 		 */
 | |
| 		if ( is_null( $this->can_cache_once_per_request_result ) ) {
 | |
| 			$this->can_cache_once_per_request_result = $this->_can_cache_once_per_request();
 | |
| 			if ( ! $this->can_cache_once_per_request_result ) {
 | |
| 				$this->cache_reject_request_wide = true;
 | |
| 				return false;
 | |
| 			}
 | |
| 		}
 | |
| 
 | |
| 		/**
 | |
| 		 * Check for constants.
 | |
| 		 */
 | |
| 		foreach ( $this->reject_constants as $name ) {
 | |
| 			if ( defined( $name ) && constant( $name ) ) {
 | |
| 				$this->cache_reject_reason = $name . ' constant defined';
 | |
| 				$cache_reject_reason       = $this->cache_reject_reason;
 | |
| 
 | |
| 				return false;
 | |
| 			}
 | |
| 		}
 | |
| 
 | |
| 		/**
 | |
| 		 * Check for AJAX requests.
 | |
| 		 */
 | |
| 		$ajax_skip = false;
 | |
| 
 | |
| 		if ( defined( 'DOING_AJAX' ) ) {
 | |
| 			$http_referer = isset( $_SERVER['HTTP_REFERER'] ) ?
 | |
| 				filter_var( stripslashes( $_SERVER['HTTP_REFERER'] ), FILTER_SANITIZE_URL ) : ''; // phpcs:ignore WordPress.Security.ValidatedSanitizedInput
 | |
| 
 | |
| 			// wp_admin is always defined for ajax requests, check by referrer.
 | |
| 			if ( strpos( $http_referer, '/wp-admin/' ) === false ) {
 | |
| 				$ajax_skip = true;
 | |
| 			}
 | |
| 		}
 | |
| 
 | |
| 		/**
 | |
| 		 * Skip if admin.
 | |
| 		 */
 | |
| 		if ( defined( 'WP_ADMIN' ) && ! $ajax_skip ) {
 | |
| 			$this->cache_reject_reason = 'WP_ADMIN';
 | |
| 			$cache_reject_reason       = $this->cache_reject_reason;
 | |
| 
 | |
| 			return false;
 | |
| 		}
 | |
| 
 | |
| 		/**
 | |
| 		 * Skip if SQL is rejected.
 | |
| 		 */
 | |
| 		if ( ! $this->_check_sql( $sql ) ) {
 | |
| 			$cache_reject_reason = 'query not cacheable';
 | |
| 
 | |
| 			return false;
 | |
| 		}
 | |
| 
 | |
| 		/**
 | |
| 		 * Skip if user is logged in.
 | |
| 		 */
 | |
| 		if ( $this->reject_logged && ! $this->_check_logged_in() ) {
 | |
| 			$this->cache_reject_reason = 'user.logged_in';
 | |
| 			$cache_reject_reason       = $this->cache_reject_reason;
 | |
| 
 | |
| 			return false;
 | |
| 		}
 | |
| 
 | |
| 		return true;
 | |
| 	}
 | |
| 
 | |
| 	/**
 | |
| 	 * Check if can cache sql, checks which have constant results during whole request.
 | |
| 	 *
 | |
| 	 * @return bool
 | |
| 	 */
 | |
| 	public function _can_cache_once_per_request() {
 | |
| 		/**
 | |
| 		 * Skip if disabled
 | |
| 		 */
 | |
| 		if ( ! $this->_config->get_boolean( 'dbcache.enabled' ) ) {
 | |
| 			$this->cache_reject_reason = 'dbcache.disabled';
 | |
| 
 | |
| 			return false;
 | |
| 		}
 | |
| 
 | |
| 		/**
 | |
| 		 * Skip if request URI is rejected
 | |
| 		 */
 | |
| 		if ( ! $this->_check_request_uri() ) {
 | |
| 			$this->cache_reject_reason = 'request';
 | |
| 			return false;
 | |
| 		}
 | |
| 
 | |
| 		/**
 | |
| 		 * Skip if cookie is rejected
 | |
| 		 */
 | |
| 		if ( ! $this->_check_cookies() ) {
 | |
| 			$this->cache_reject_reason = 'cookie';
 | |
| 			return false;
 | |
| 		}
 | |
| 
 | |
| 		return true;
 | |
| 	}
 | |
| 
 | |
| 	/**
 | |
| 	 * Check SQL
 | |
| 	 *
 | |
| 	 * @param string $sql SQL query.
 | |
| 	 *
 | |
| 	 * @return bool
 | |
| 	 */
 | |
| 	public function _check_sql( $sql ) {
 | |
| 
 | |
| 		$auto_reject_strings = $this->_config->get_array( 'dbcache.reject.words' );
 | |
| 
 | |
| 		if ( preg_match( '~' . implode( '|', $auto_reject_strings ) . '~is', $sql ) ) {
 | |
| 			return false;
 | |
| 		}
 | |
| 
 | |
| 		$reject_sql = $this->_config->get_array( 'dbcache.reject.sql' );
 | |
| 
 | |
| 		foreach ( $reject_sql as $expr ) {
 | |
| 			$expr = trim( $expr );
 | |
| 			$expr = str_replace( '{prefix}', $this->wpdb_mixin->prefix, $expr );
 | |
| 			if ( ! empty( $expr ) && preg_match( '~' . $expr . '~i', $sql ) ) {
 | |
| 				return false;
 | |
| 			}
 | |
| 		}
 | |
| 
 | |
| 		return true;
 | |
| 	}
 | |
| 
 | |
| 	/**
 | |
| 	 * Check request URI
 | |
| 	 *
 | |
| 	 * @return boolean
 | |
| 	 */
 | |
| 	public function _check_request_uri() {
 | |
| 		$auto_reject_uri = array(
 | |
| 			'wp-login',
 | |
| 			'wp-register',
 | |
| 		);
 | |
| 
 | |
| 		$request_uri = isset( $_SERVER['REQUEST_URI'] ) ?
 | |
| 			filter_var( stripslashes( $_SERVER['REQUEST_URI'] ), FILTER_SANITIZE_URL ) : ''; // phpcs:ignore WordPress.Security.ValidatedSanitizedInput
 | |
| 
 | |
| 		foreach ( $auto_reject_uri as $uri ) {
 | |
| 			if ( strstr( $request_uri, $uri ) !== false ) {
 | |
| 				return false;
 | |
| 			}
 | |
| 		}
 | |
| 
 | |
| 		$reject_uri = $this->_config->get_array( 'dbcache.reject.uri' );
 | |
| 		$reject_uri = array_map( array( '\W3TC\Util_Environment', 'parse_path' ), $reject_uri );
 | |
| 
 | |
| 		foreach ( $reject_uri as $expr ) {
 | |
| 			$expr = trim( $expr );
 | |
| 			if ( ! empty( $expr ) && preg_match( '~' . $expr . '~i', $request_uri ) ) {
 | |
| 				return false;
 | |
| 			}
 | |
| 		}
 | |
| 
 | |
| 		return true;
 | |
| 	}
 | |
| 
 | |
| 	/**
 | |
| 	 * Checks for WordPress cookies.
 | |
| 	 *
 | |
| 	 * @return bool
 | |
| 	 */
 | |
| 	public function _check_cookies() {
 | |
| 		foreach ( array_keys( $_COOKIE ) as $cookie_name ) {
 | |
| 			if ( 'wordpress_test_cookie' === $cookie_name ) {
 | |
| 				continue;
 | |
| 			}
 | |
| 			if ( preg_match( '/^wp-postpass|^comment_author/', $cookie_name ) ) {
 | |
| 				return false;
 | |
| 			}
 | |
| 		}
 | |
| 
 | |
| 		foreach ( $this->_config->get_array( 'dbcache.reject.cookie' ) as $reject_cookie ) {
 | |
| 			foreach ( array_keys( $_COOKIE ) as $cookie_name ) {
 | |
| 				if ( strstr( $cookie_name, $reject_cookie ) !== false ) {
 | |
| 					return false;
 | |
| 				}
 | |
| 			}
 | |
| 		}
 | |
| 
 | |
| 		return true;
 | |
| 	}
 | |
| 
 | |
| 	/**
 | |
| 	 * Check if user is logged in.
 | |
| 	 *
 | |
| 	 * @return bool
 | |
| 	 */
 | |
| 	public function _check_logged_in() {
 | |
| 		foreach ( array_keys( $_COOKIE ) as $cookie_name ) {
 | |
| 			if ( strpos( $cookie_name, 'wordpress_logged_in' ) === 0 ) {
 | |
| 				return false;
 | |
| 			}
 | |
| 		}
 | |
| 
 | |
| 		return true;
 | |
| 	}
 | |
| 
 | |
| 	/**
 | |
| 	 * Get group.
 | |
| 	 *
 | |
| 	 * @access private
 | |
| 	 *
 | |
| 	 * @param string $sql SQL query.
 | |
| 	 *
 | |
| 	 * @return string
 | |
| 	 */
 | |
| 	private function _get_group( $sql ) {
 | |
| 		$sql = strtolower( $sql );
 | |
| 
 | |
| 		// Collect list of tables used in query.
 | |
| 		if ( preg_match_all( '~(^|[\s,`])' . $this->wpdb_mixin->prefix . '([0-9a-zA-Z_]+)~i', $sql, $m ) ) {
 | |
| 			$tables = array_unique( $m[2] );
 | |
| 		} else {
 | |
| 			$tables = array();
 | |
| 		}
 | |
| 
 | |
| 		if ( $this->contains_only_tables( $tables, array( 'options' => '*' ) ) ) {
 | |
| 			$group = 'options';
 | |
| 		} elseif (
 | |
| 			$this->contains_only_tables(
 | |
| 				$tables,
 | |
| 				array(
 | |
| 					'comments'     => '*',
 | |
| 					'commentsmeta' => '*',
 | |
| 				)
 | |
| 			) ) {
 | |
| 			$group = 'comments';
 | |
| 		} elseif ( count( $tables ) <= 1 ) {
 | |
| 			$group = 'singletables';   // Request with single table affected.
 | |
| 		} else {
 | |
| 			$group = 'remaining';
 | |
| 		}
 | |
| 
 | |
| 		if ( $this->use_filters && function_exists( 'apply_filters' ) ) {
 | |
| 			$group = apply_filters( 'w3tc_dbcache_get_sql_group', $group, $sql, $tables );
 | |
| 		}
 | |
| 
 | |
| 		return $group;
 | |
| 	}
 | |
| 
 | |
| 	/**
 | |
| 	 * Contains only tables.
 | |
| 	 *
 | |
| 	 * @accress private
 | |
| 	 *
 | |
| 	 * @param array $tables  Tables.
 | |
| 	 *
 | |
| 	 * @param array $allowed Allowed.
 | |
| 	 */
 | |
| 	private function contains_only_tables( $tables, $allowed ) {
 | |
| 		if ( empty( $tables ) ) {
 | |
| 			return false;
 | |
| 		}
 | |
| 
 | |
| 		foreach ( $tables as $t ) {
 | |
| 			if ( ! isset( $allowed[ $t ] ) ) {
 | |
| 				return false;
 | |
| 			}
 | |
| 		}
 | |
| 
 | |
| 		return true;
 | |
| 	}
 | |
| 
 | |
| 	/**
 | |
| 	 * Get flush groups
 | |
| 	 *
 | |
| 	 * @access private
 | |
| 	 *
 | |
| 	 * @param string $group  Group.
 | |
| 	 *
 | |
| 	 * @param array  $extras Extra arguments.
 | |
| 	 */
 | |
| 	private function _get_flush_groups( $group, $extras = array() ) {
 | |
| 		$groups_to_flush = array();
 | |
| 
 | |
| 		switch ( $group ) {
 | |
| 			case 'remaining':
 | |
| 			case 'singletables':
 | |
| 				$groups_to_flush = array(
 | |
| 					'remaining'    => '*',
 | |
| 					'options'      => '*',
 | |
| 					'comments'     => '*',
 | |
| 					'singletables' => '*',
 | |
| 				);
 | |
| 				break;
 | |
| 			/**
 | |
| 			 * Options are updated on each second request,
 | |
| 			 * ignore by default probability that SELECTs with joins with options are critical and don't flush "remaining".
 | |
| 			 * That can be changed by w3tc_dbcache_get_flush_groups filter.
 | |
| 			 */
 | |
| 			case 'options':
 | |
| 				$groups_to_flush = array( $group => '*' );
 | |
| 				break;
 | |
| 			default:
 | |
| 				$groups_to_flush = array(
 | |
| 					$group      => '*',
 | |
| 					'remaining' => '*',
 | |
| 				);
 | |
| 		}
 | |
| 
 | |
| 		if ( $this->use_filters && function_exists( 'apply_filters' ) ) {
 | |
| 			$groups_to_flush = apply_filters( 'w3tc_dbcache_get_flush_groups', $groups_to_flush, $group, $extras );
 | |
| 		}
 | |
| 
 | |
| 		return $groups_to_flush;
 | |
| 	}
 | |
| 
 | |
| 	/**
 | |
| 	 * Get reject reason.
 | |
| 	 *
 | |
| 	 * @return string
 | |
| 	 */
 | |
| 	public function get_reject_reason() {
 | |
| 		if ( is_null( $this->cache_reject_reason ) ) {
 | |
| 			return '';
 | |
| 		}
 | |
| 
 | |
| 		$request_wide_string = $this->cache_reject_request_wide ?
 | |
| 			( function_exists( '__' ) ? __( 'Request-wide ', 'w3-total-cache' ) : 'Request ' ) : '';
 | |
| 
 | |
| 		return $request_wide_string . $this->_get_reject_reason_message( $this->cache_reject_reason );
 | |
| 	}
 | |
| 
 | |
| 	/**
 | |
| 	 * Get reject reason message.
 | |
| 	 *
 | |
| 	 * @param string $key Key.
 | |
| 	 *
 | |
| 	 * @return string|void
 | |
| 	 */
 | |
| 	private function _get_reject_reason_message( $key ) {
 | |
| 		if ( ! function_exists( '__' ) ) {
 | |
| 			return $key;
 | |
| 		}
 | |
| 
 | |
| 		switch ( $key ) {
 | |
| 			case 'dbcache.disabled':
 | |
| 				return __( 'Database caching is disabled', 'w3-total-cache' );
 | |
| 			case 'DONOTCACHEDB':
 | |
| 				return __( 'DONOTCACHEDB constant is defined', 'w3-total-cache' );
 | |
| 			case 'DOING_AJAX':
 | |
| 				return __( 'Doing AJAX', 'w3-total-cache' );
 | |
| 			case 'request':
 | |
| 				return __( 'Request URI is rejected', 'w3-total-cache' );
 | |
| 			case 'cookie':
 | |
| 				return __( 'Cookie is rejected', 'w3-total-cache' );
 | |
| 			case 'DOING_CRONG':
 | |
| 				return __( 'Doing cron', 'w3-total-cache' );
 | |
| 			case 'APP_REQUEST':
 | |
| 				return __( 'Application request', 'w3-total-cache' );
 | |
| 			case 'XMLRPC_REQUEST':
 | |
| 				return __( 'XMLRPC request', 'w3-total-cache' );
 | |
| 			case 'WP_ADMIN':
 | |
| 				return __( 'wp-admin', 'w3-total-cache' );
 | |
| 			case 'SHORTINIT':
 | |
| 				return __( 'Short init', 'w3-total-cache' );
 | |
| 			case 'query':
 | |
| 				return __( 'Query is rejected', 'w3-total-cache' );
 | |
| 			case 'user.logged_in':
 | |
| 				return __( 'User is logged in', 'w3-total-cache' );
 | |
| 			default:
 | |
| 				return $key;
 | |
| 		}
 | |
| 	}
 | |
| 
 | |
| 	/**
 | |
| 	 * Footer comment.
 | |
| 	 *
 | |
| 	 * @param array $strings Strings.
 | |
| 	 *
 | |
| 	 * @return array
 | |
| 	 */
 | |
| 	public function w3tc_footer_comment( $strings ) {
 | |
| 		$reject_reason = $this->get_reject_reason();
 | |
| 		$append        = empty( $reject_reason ) ? '' : sprintf( ' (%1$s)', $reject_reason );
 | |
| 
 | |
| 		if ( $this->query_hits ) {
 | |
| 			$strings[] = sprintf(
 | |
| 				// translators: 1: Query hits, 2: Total queries, 3: Total time, 4: Engine name, 5: Reject reason.
 | |
| 				__( 'Database Caching %1$d/%2$d queries in %3$.3f seconds using %4$s%5$s', 'w3-total-cache' ),
 | |
| 				$this->query_hits,
 | |
| 				$this->query_total,
 | |
| 				$this->time_total,
 | |
| 				Cache::engine_name( $this->_config->get_string( 'dbcache.engine' ) ),
 | |
| 				$append
 | |
| 			);
 | |
| 		} else {
 | |
| 			$strings[] = sprintf(
 | |
| 				// translators: 1: Engine name, 2: Reject reason.
 | |
| 				__( 'Database Caching using %1$s%2$s', 'w3-total-cache' ),
 | |
| 				Cache::engine_name( $this->_config->get_string( 'dbcache.engine' ) ),
 | |
| 				$append
 | |
| 			);
 | |
| 		}
 | |
| 
 | |
| 		if ( $this->debug ) {
 | |
| 			$strings[] = '';
 | |
| 			$strings[] = __( 'Db cache debug info:', 'w3-total-cache' );
 | |
| 			$strings[] = sprintf( '%1$s%2$d', str_pad( __( 'Total queries: ', 'w3-total-cache' ), 20 ), $this->query_total );
 | |
| 			$strings[] = sprintf( '%1$s%2$d', str_pad( __( 'Cached queries: ', 'w3-total-cache' ), 20 ), $this->query_hits );
 | |
| 			$strings[] = sprintf( '%1$s%2$.4f', str_pad( __( 'Total query time: ', 'w3-total-cache' ), 20 ), $this->time_total );
 | |
| 		}
 | |
| 
 | |
| 		if ( $this->log_filehandle ) {
 | |
| 			fclose( $this->log_filehandle ); // phpcs:ignore WordPress.WP.AlternativeFunctions.file_system_read_fclose
 | |
| 			$this->log_filehandle = false;
 | |
| 		}
 | |
| 		return $strings;
 | |
| 	}
 | |
| 
 | |
| 	/**
 | |
| 	 * Usage statistics of request.
 | |
| 	 *
 | |
| 	 * @param object $storage Storage object.
 | |
| 	 *
 | |
| 	 * @return void
 | |
| 	 */
 | |
| 	public function w3tc_usage_statistics_of_request( $storage ) {
 | |
| 		$storage->counter_add( 'dbcache_calls_total', $this->query_total );
 | |
| 		$storage->counter_add( 'dbcache_calls_hits', $this->query_hits );
 | |
| 		$storage->counter_add( 'dbcache_flushes', $this->cache_flushes );
 | |
| 		$time_ms = (int) ( $this->time_total * 1000 );
 | |
| 		$storage->counter_add( 'dbcache_time_ms', $time_ms );
 | |
| 	}
 | |
| 
 | |
| 	/**
 | |
| 	 * Log query.
 | |
| 	 *
 | |
| 	 * @access private
 | |
| 	 *
 | |
| 	 * @param string $line Line to add.
 | |
| 	 *
 | |
| 	 * @return void
 | |
| 	 */
 | |
| 	private function log_query( $line ) {
 | |
| 		if ( ! $this->log_filehandle ) {
 | |
| 			$filename             = Util_Debug::log_filename( 'dbcache-queries' );
 | |
| 			$this->log_filehandle = fopen( $filename, 'a' ); // phpcs:ignore WordPress.WP.AlternativeFunctions.file_system_read_fopen
 | |
| 		}
 | |
| 
 | |
| 		fputcsv( $this->log_filehandle, $line, "\t" );
 | |
| 	}
 | |
| }
 |