Your IP : 216.73.216.44
<?php
namespace WP_Defender\Traits;
use WP_Defender\Component\Logger\Rotation_Logger as Logger;
trait IO {
/**
* A simple function to create & return the folder that we can use to write tmp files.
*
* @return string
*/
protected function get_tmp_path() {
$upload_dir = wp_upload_dir()['basedir'];
$tmp_dir = $upload_dir . DIRECTORY_SEPARATOR . 'wp-defender';
if ( ! is_dir( $tmp_dir ) ) {
wp_mkdir_p( $tmp_dir );
}
if ( ! is_file( $tmp_dir . DIRECTORY_SEPARATOR . 'index.php' ) ) {
file_put_contents( $tmp_dir . DIRECTORY_SEPARATOR . 'index.php', '' );
}
return $tmp_dir;
}
/**
* @param $category
*
* @return string
*/
public function get_log_path( $category = '' ): string {
$file = empty( $category ) ? 'defender.log' : $category;
$logger = new Logger();
$file_name = $logger->generate_file_name( $file );
return $this->get_tmp_path() . DIRECTORY_SEPARATOR . $file_name;
}
/**
* Create a lock. This will be used in scanning.
*
* @return string
*/
protected function get_lock_path(): string {
return $this->get_tmp_path() . DIRECTORY_SEPARATOR . 'scan.lock';
}
/**
* Create a lock. This will be used for 2FA.
*
* @return string
*/
protected function get_2fa_lock_path(): string {
return $this->get_tmp_path() . DIRECTORY_SEPARATOR . 'two-fa.lock';
}
/**
* Delete a folder with every content inside.
*
* @param $dir
*
* @return bool
*/
public function delete_dir( $dir ): bool {
if ( ! is_dir( $dir ) ) {
return false;
}
$it = new \RecursiveDirectoryIterator( $dir, \RecursiveDirectoryIterator::SKIP_DOTS );
$files = new \RecursiveIteratorIterator(
$it,
\RecursiveIteratorIterator::CHILD_FIRST
);
foreach ( $files as $file ) {
if ( $file->isDir() ) {
$ret = rmdir( $file->getPathname() );
} else {
$ret = unlink( $file->getPathname() );
}
if ( false === $ret ) {
return false;
}
}
rmdir( $dir );
return true;
}
/**
* Not remove double quotes inside str_replace().
*
* @param string $data
*
* @return array|string
*/
protected function convert_end_lines_dos_to_linux( $data ) {
return str_replace( [ "\r\n", "\r" ], "\n", $data );
}
/**
* Not remove double quotes inside str_replace().
*
* @param string $data
*
* @return array|string
*/
protected function convert_end_lines_linux_to_dos( $data ) {
return str_replace( "\n", "\r\n", $this->convert_end_lines_dos_to_linux( $data ) );
}
/**
* Compare hashes on different OS.
*
* @param string $file_path
* @param string|array $file_hash
*
* @return bool
*/
protected function compare_hashes_on_different_os( $file_path, $file_hash ) {
if ( hash_equals( md5_file( $file_path ), $file_hash ) ) {
return true;
}
if ( hash_equals( $this->hash_file( $file_path, 'linux' ), $file_hash ) ) {
return true;
}
if ( hash_equals( $this->hash_file( $file_path, 'dos' ), $file_hash ) ) {
return true;
}
return false;
}
/**
* @param string $file_path Path to file.
* @param string|array $file_hash Hash or some hashes of file2, e.g. for readme.txt.
*
* @return bool
*/
public function compare_hashes( $file_path, $file_hash ) {
if ( is_string( $file_hash ) ) {
return $this->compare_hashes_on_different_os( $file_path, $file_hash );
} elseif ( is_array( $file_hash ) ) {
// Sometimes file has some hashes.
foreach ( $file_hash as $hash_value ) {
if ( $this->compare_hashes_on_different_os( $file_path, $hash_value ) ) {
return true;
}
}
return false;
} else {
return false;
}
}
/**
* Hash a file in chunks.
*
* @param string $file_path Path to a file.
* @param string $convert_to Convert end of lines characters to linux or dos.
*
* @since 3.10.0
* @return bool|string
*/
protected function hash_file( string $file_path, string $convert_to = '' ) {
$handle = @fopen( $file_path, 'rb' );
if ( ! $handle ) {
return false;
}
$context = hash_init( 'md5' );
while ( ! feof( $handle ) ) {
$data = fread( $handle, 65536 );
if ( false === $data ) {
return false;
}
if ( 'linux' === $convert_to ) {
$data = $this->convert_end_lines_dos_to_linux( $data );
} else if ( 'dos' === $convert_to ) {
$data = $this->convert_end_lines_linux_to_dos( $data );
}
hash_update( $context, $data );
}
return hash_final( $context, false );
}
}