woocommerce/tests/includes/code-hacking/code-hacker.php

<?php

/**
 * CodeHacker - allows to hack (alter on the fly) the content of PHP code files.
 *
 * Based on BypassFinals: https://github.com/dg/bypass-finals
 *
 * How to use:
 *
 * 1. Register hacks using CodeHacker::add_hack(hack). A hack is either:
 *    - A function with 'hack($code, $path)' signature, or
 *    - An object having a public 'hack($code, $path)' method.
 *
 *    Where $code is a string containing the code to hack, and $path is the full path of the file
 *    containing the code. The function/method must return a string with the code already hacked.
 *
 * 2. Run CodeHacker::enable()
 *
 * For using with PHPUnit, see CodeHackerTestHook.
 */
class CodeHacker {

	const PROTOCOL = 'file';

	/** @var resource|null */
	public $context;

	/** @var resource|null */
	private $handle;

	/** @var array|null */
	private static $pathWhitelist = array();

	private static $hacks = array();

	private static $enabled = false;

	public static function enable() {
		if ( ! self::$enabled ) {
			stream_wrapper_unregister( self::PROTOCOL );
			stream_wrapper_register( self::PROTOCOL, __CLASS__ );
			self::$enabled = true;
		}
	}

	public static function restore() {
		if ( self::$enabled ) {
			stream_wrapper_restore( self::PROTOCOL );
			self::$enabled = false;
		}
	}

	public static function clear_hacks() {
		self::$hacks = array();
	}

	public static function is_enabled() {
		return self::$enabled;
	}

	public static function add_hack( $hack ) {
		if ( ! is_callable( $hack ) && ! is_object( $hack ) ) {
			throw new Exception( "hacks must be either functions, or objects having a 'process(\$text, \$path)' method." );
		}

		// TODO: Check that callbacks have at least two parameters (if they have more, they must be optional); and that objects have a "process" method with the same condition.

		self::$hacks[] = $hack;
	}

	public static function setWhitelist( array $pathWhitelist ) {
		self::$pathWhitelist = $pathWhitelist;
	}


	public function dir_closedir() {
		closedir( $this->handle );
	}


	public function dir_opendir( $path, $options ) {
		$this->handle = $this->context
			? $this->native( 'opendir', $path, $this->context )
			: $this->native( 'opendir', $path );
		return (bool) $this->handle;
	}


	public function dir_readdir() {
		 return readdir( $this->handle );
	}


	public function dir_rewinddir() {
		return rewinddir( $this->handle );
	}


	public function mkdir( $path, $mode, $options ) {
		$recursive = (bool) ( $options & STREAM_MKDIR_RECURSIVE );
		return $this->native( 'mkdir', $path, $mode, $recursive, $this->context );
	}


	public function rename( $pathFrom, $pathTo ) {
		return $this->native( 'rename', $pathFrom, $pathTo, $this->context );
	}


	public function rmdir( $path, $options ) {
		return $this->native( 'rmdir', $path, $this->context );
	}


	public function stream_cast( $castAs ) {
		return $this->handle;
	}


	public function stream_close() {
		// echo "***** CLOSE HANDLE: " . $this->handle . " \n";
		fclose( $this->handle );
	}


	public function stream_eof() {
		return feof( $this->handle );
	}


	public function stream_flush() {
		return fflush( $this->handle );
	}


	public function stream_lock( $operation ) {
		return $operation
			? flock( $this->handle, $operation )
			: true;
	}


	public function stream_metadata( $path, $option, $value ) {
		switch ( $option ) {
			case STREAM_META_TOUCH:
				$value += array( null, null );
				return $this->native( 'touch', $path, $value[0], $value[1] );
			case STREAM_META_OWNER_NAME:
			case STREAM_META_OWNER:
				return $this->native( 'chown', $path, $value );
			case STREAM_META_GROUP_NAME:
			case STREAM_META_GROUP:
				return $this->native( 'chgrp', $path, $value );
			case STREAM_META_ACCESS:
				return $this->native( 'chmod', $path, $value );
		}
	}


	public function stream_open( $path, $mode, $options, &$openedPath ) {
		$usePath = (bool) ( $options & STREAM_USE_PATH );
		if ( $mode === 'rb' && self::pathInWhitelist( $path ) && pathinfo( $path, PATHINFO_EXTENSION ) === 'php' ) {
			$content = $this->native( 'file_get_contents', $path, $usePath, $this->context );
			if ( $content === false ) {
				return false;
			}
			$modified = self::hack( $content, $path );
			if ( $modified !== $content ) {
				$this->handle = tmpfile();
				$this->native( 'fwrite', $this->handle, $modified );
				$this->native( 'fseek', $this->handle, 0 );
				return true;
			}
		}
		$this->handle = $this->context
			? $this->native( 'fopen', $path, $mode, $usePath, $this->context )
			: $this->native( 'fopen', $path, $mode, $usePath );
		return (bool) $this->handle;
	}


	public function stream_read( $count ) {
		return fread( $this->handle, $count );
	}


	public function stream_seek( $offset, $whence = SEEK_SET ) {
		return fseek( $this->handle, $offset, $whence ) === 0;
	}


	public function stream_set_option( $option, $arg1, $arg2 ) {
	}


	public function stream_stat() {
		 return fstat( $this->handle );
	}


	public function stream_tell() {
		 return ftell( $this->handle );
	}


	public function stream_truncate( $newSize ) {
		return ftruncate( $this->handle, $newSize );
	}


	public function stream_write( $data ) {
		return fwrite( $this->handle, $data );
	}


	public function unlink( $path ) {
		return $this->native( 'unlink', $path );
	}


	public function url_stat( $path, $flags ) {
		$func = $flags & STREAM_URL_STAT_LINK ? 'lstat' : 'stat';
		return $flags & STREAM_URL_STAT_QUIET
			? @$this->native( $func, $path )
			: $this->native( $func, $path );
	}


	private function native( $func ) {
		stream_wrapper_restore( self::PROTOCOL );
		$res = call_user_func_array( $func, array_slice( func_get_args(), 1 ) );
		stream_wrapper_unregister( self::PROTOCOL );
		stream_wrapper_register( self::PROTOCOL, __CLASS__ );
		return $res;
	}


	private static function hack( $code, $path ) {
		foreach ( self::$hacks as $hack ) {
			if ( is_callable( $hack ) ) {
				$code = call_user_func( $hack, $code, $path );
			} else {
				$code = $hack->hack( $code, $path );
			}
		}

		return $code;
	}


	private static function pathInWhitelist( $path ) {
		if ( empty( self::$pathWhitelist ) ) {
			return true;
		}
		foreach ( self::$pathWhitelist as $whitelistItem ) {
			if ( substr( $path, -strlen( $whitelistItem ) ) === $whitelistItem ) {
				return true;
			}
		}
		return false;
	}
}