10h30/ultimatemember
1
0
Fork 0
mirror of https://github.com/10h30/ultimatemember.git synced 2026-06-05 15:09:37 +09:00
Code Issues Packages Projects Releases Wiki Activity

Fix ssl checker for load balancers

Browse Source
This commit is contained in:
jonfalcon
2016-01-13 15:41:52 -08:00
parent 90f3a4b9b6
commit 67635132b4
1 changed files with 128 additions and 122 deletions
Download Patch File Download Diff File Expand all files Collapse all files
core/um-files.php
+128 -122
View File
@@ -5,7 +5,7 @@ class UM_Files {
function __construct() {
add_action('init', array(&$this, 'setup_paths'), 1);
$this->fonticon = array(
'pdf' => array('icon' => 'um-faicon-file-pdf-o', 'color' => '#D24D4D' ),
'txt' => array('icon' => 'um-faicon-file-text-o' ),
@@ -20,30 +20,30 @@ class UM_Files {
'rar' => array('icon' => 'um-faicon-file-zip-o' ),
'mp3' => array('icon' => 'um-faicon-file-audio-o' ),
);
$this->default_file_fonticon = 'um-faicon-file-o';
}
/***
*** @allowed image types
***/
function allowed_image_types() {
$array['png'] = 'PNG';
$array['jpeg'] = 'JPEG';
$array['jpg'] = 'JPG';
$array['gif'] = 'GIF';
$array = apply_filters('um_allowed_image_types', $array);
return $array;
}
/***
*** @allowed file types
***/
function allowed_file_types() {
$array['pdf'] = 'PDF';
$array['txt'] = 'Text';
$array['csv'] = 'CSV';
@@ -56,11 +56,11 @@ class UM_Files {
$array['zip'] = 'ZIP';
$array['rar'] = 'RAR';
$array['mp3'] = 'MP3';
$array = apply_filters('um_allowed_file_types', $array);
return $array;
}
/***
*** @Get extension icon
***/
@@ -71,7 +71,7 @@ class UM_Files {
return $this->default_file_fonticon;
}
}
/***
*** @Get extension icon background
***/
@@ -82,21 +82,27 @@ class UM_Files {
return '#666';
}
}
/***
*** @Setup upload directory
***/
function setup_paths(){
$this->upload_dir = wp_upload_dir();
$this->upload_basedir = $this->upload_dir['basedir'] . '/ultimatemember/';
$this->upload_baseurl = $this->upload_dir['baseurl'] . '/ultimatemember/';
$this->upload_basedir = apply_filters('um_upload_basedir_filter', $this->upload_basedir );
$this->upload_baseurl = apply_filters('um_upload_baseurl_filter', $this->upload_baseurl );
if( is_ssl() ){
// @note : is_ssl() doesn't work properly for some sites running with load balancers
// Check the links for more info about this bug
// https://codex.wordpress.org/Function_Reference/is_ssl
// http://snippets.webaware.com.au/snippets/wordpress-is_ssl-doesnt-work-behind-some-load-balancers/
if( is_ssl() || stripos( get_option( 'siteurl' ), 'https://' ) !== false
|| ( isset( $_SERVER['HTTP_X_FORWARDED_PROTO'] )
&& $_SERVER['HTTP_X_FORWARDED_PROTO'] == 'https' ) ) {
$this->upload_baseurl = str_replace("http://", "https://", $this->upload_baseurl);
}
@@ -114,9 +120,9 @@ class UM_Files {
@mkdir( $this->upload_temp , 0755, true);
umask($old);
}
}
/***
*** @Generate unique temp directory
***/
@@ -127,14 +133,14 @@ class UM_Files {
$array['url'] = $this->upload_temp_url . $unique_number . '/';
return $array;
}
/***
*** @get path only without file name
***/
function path_only( $file ) {
return trailingslashit( dirname( $file ) );
}
/***
*** @fix image orientation
***/
@@ -160,103 +166,103 @@ class UM_Files {
}
return $rotate;
}
/***
*** @Process an image
***/
function create_and_copy_image($source, $destination, $quality = 100) {
$info = @getimagesize($source);
if ($info['mime'] == 'image/jpeg'){
$image = imagecreatefromjpeg($source);
} else if ($info['mime'] == 'image/gif'){
$image = imagecreatefromgif($source);
} else if ($info['mime'] == 'image/png'){
$image = imagecreatefrompng($source);
}
list($w, $h) = @getimagesize( $source );
if ( $w > um_get_option('image_max_width') ) {
$ratio = round( $w / $h, 2 );
$new_w = um_get_option('image_max_width');
$new_h = round( $new_w / $ratio, 2 );
$image_p = imagecreatetruecolor( $new_w, $new_h );
imagecopyresampled( $image_p, $image, 0, 0, 0, 0, $new_w, $new_h, $w, $h );
$image_p = $this->fix_image_orientation($image_p, $source);
imagejpeg( $image_p, $destination, $quality);
} else {
$image = $this->fix_image_orientation($image, $source);
imagejpeg( $image, $destination, $quality);
}
}
/***
*** @Process a file
***/
function upload_temp_file($source, $destination) {
move_uploaded_file($source, $destination);
}
/***
*** @Process a temp upload
***/
function new_image_upload_temp($source, $destination, $quality = 100){
$unique_dir = $this->unique_dir();
$this->make_dir( $unique_dir['dir'] );
$this->create_and_copy_image($source, $unique_dir['dir'] . $destination, $quality);
$url = $unique_dir['url'] . $destination;
return $url;
}
/***
*** @Process a temp upload for files
***/
function new_file_upload_temp($source, $destination ){
$unique_dir = $this->unique_dir();
$this->make_dir( $unique_dir['dir'] );
$this->upload_temp_file($source, $unique_dir['dir'] . $destination);
$url = $unique_dir['url'] . $destination;
return $url;
}
/***
*** @Make a Folder
***/
function make_dir( $dir ){
$old = umask(0);
@mkdir( $dir, 0755, true);
umask($old);
}
/***
*** @Get extension by mime type
***/
@@ -264,64 +270,64 @@ class UM_Files {
$split = explode('/',$mime);
return $split[1];
}
/***
*** @Get file data
***/
function get_file_data($file){
$array['size'] = filesize($file);
return $array;
}
/***
*** @Get image data
***/
function get_image_data($file){
$array['size'] = filesize($file);
$array['image'] = @getimagesize($file);
if ( $array['image'] > 0 ) {
$array['invalid_image'] = false;
list($width, $height, $type, $attr) = @getimagesize($file);
$array['width'] = $width;
$array['height'] = $height;
$array['ratio'] = $width / $height;
$array['extension'] = $this->get_extension_by_mime_type( $array['image']['mime'] );
} else {
$array['invalid_image'] = true;
}
return $array;
}
/***
*** @Check image upload and handle errors
***/
function check_image_upload($file, $field) {
global $ultimatemember;
$error = null;
$fileinfo = $this->get_image_data($file);
$data = $ultimatemember->fields->get_field($field);
if ( $data == null ) {
$data = apply_filters("um_custom_image_handle_{$field}", '' );
if ( !$data ) {
$error = __('This media type is not recognized.','ultimatemember');
}
}
if ( $fileinfo['invalid_image'] == true ) {
$error = sprintf(__('Your image is invalid or too large!','ultimatemember') );
} elseif ( isset( $data['allowed_types'] ) && !$this->in_array( $fileinfo['extension'], $data['allowed_types'] ) ) {
@@ -333,10 +339,10 @@ class UM_Files {
} elseif ( isset($data['min_height']) && ( $fileinfo['height'] < $data['min_height'] ) ) {
$error = sprintf(__('Your photo is too small. It must be at least %spx wide.','ultimatemember'), $data['min_height']);
}
return $error;
}
/***
*** @Check file upload and handle errors
***/
@@ -346,16 +352,16 @@ class UM_Files {
$fileinfo = $this->get_file_data($file);
$data = $ultimatemember->fields->get_field($field);
if ( !$this->in_array( $extension, $data['allowed_types'] ) ) {
$error = ( isset( $data['extension_error'] ) && !empty( $data['extension_error'] ) ) ? $data['extension_error'] : 'not allowed';
} elseif ( isset($data['min_size']) && ( $fileinfo['size'] < $data['min_size'] ) ) {
$error = $data['min_size_error'];
}
return $error;
}
/***
*** @If a value exists in comma seperated list
***/
@@ -364,17 +370,17 @@ class UM_Files {
return true;
return false;
}
/***
*** @This function will delete file upload from server
***/
function delete_file( $src ) {
if ( strstr( $src, '?' ) ){
$splitted = explode('?', $src );
$src = $splitted[0];
}
$is_temp = um_is_temp_upload( $src );
if ( $is_temp ) {
unlink( $is_temp );
@@ -383,38 +389,38 @@ class UM_Files {
die('Not a valid temp file');
}
}
/***
*** @delete a main user photo
***/
function delete_core_user_photo( $user_id, $type ) {
delete_user_meta( $user_id, $type );
do_action("um_after_remove_{$type}", $user_id);
$dir = $this->upload_basedir . $user_id . '/';
$prefix = $type;
chdir($dir);
$matches = glob($prefix.'*',GLOB_MARK);
if( is_array($matches) && !empty($matches)) {
foreach($matches as $match) {
if( is_file($dir.$match) ) unlink($dir.$match);
}
}
if ( count(glob("$dir/*")) === 0) {
rmdir( $dir );
}
}
/***
*** @resize a local image
***/
function resize_image( $file, $crop ) {
$targ_x1 = $crop[0];
$targ_y1 = $crop[1];
$targ_x2 = $crop[2];
@@ -425,12 +431,12 @@ class UM_Files {
imagecopy( $dst_r, $img_r, 0, 0, $targ_x1, $targ_y1, $targ_x2, $targ_y2 );
imagejpeg( $dst_r, $this->path_only( $file ) . basename( $file ), 100);
$split = explode('/ultimatemember/temp/', $file);
return $this->upload_temp_url . $split[1];
}
/***
*** @make a user folder for uploads
***/
@@ -441,34 +447,34 @@ class UM_Files {
umask($old);
}
}
/***
*** @new user upload
***/
function new_user_upload( $user_id, $source, $key ) {
// if he does not have uploads dir yet
$this->new_user( $user_id );
if ( is_user_logged_in() && ( get_current_user_id() != $user_id ) && !um_user_can('can_edit_everyone') ) {
wp_die( __('Unauthorized to do this attempt.','ultimatemember') );
}
if ( !is_user_logged_in() && ( $key == 'profile_photo' || $key == 'cover_photo' ) ) {
wp_die( __('Unauthorized to do this attempt.','ultimatemember') );
}
// name and extension stuff
$source_name = basename( $source );
if ( $key == 'profile_photo' ) {
$source_name = 'profile_photo.jpg';
}
if ( $key == 'cover_photo' ) {
$source_name = 'cover_photo.jpg';
}
$ext = '.' . pathinfo($source_name, PATHINFO_EXTENSION);
$name = str_replace( $ext, '', $source_name );
$filename = $name . $ext;
@@ -478,60 +484,60 @@ class UM_Files {
unlink( $this->upload_basedir . $user_id . '/' . $filename );
}
copy( $source, $this->upload_basedir . $user_id . '/' . $filename );
// thumbs
if ( $key == 'profile_photo' ) {
list($w, $h) = @getimagesize( $source );
$sizes = um_get_option('photo_thumb_sizes');
foreach( $sizes as $size ) {
if ( file_exists( $this->upload_basedir . $user_id . '/' . $name . '-' . $size . $ext ) ) {
unlink( $this->upload_basedir . $user_id . '/' . $name . '-' . $size . $ext );
}
if ( $size < $w ) {
$thumb_s = imagecreatefromjpeg( $source );
$thumb = imagecreatetruecolor( $size, $size );
imagecopyresampled( $thumb, $thumb_s, 0, 0, 0, 0, $size, $size, $w, $h );
imagejpeg( $thumb, $this->upload_basedir . $user_id . '/' . $name . '-' . $size . $ext, 100);
}
}
// removes a synced profile photo
delete_user_meta( $user_id, 'synced_profile_photo' );
}
if ( $key == 'cover_photo' ) {
list($w, $h) = @getimagesize( $source );
$sizes = um_get_option('cover_thumb_sizes');
foreach( $sizes as $size ) {
$ratio = round( $w / $h, 2 );
$height = round( $size / $ratio, 2 );
if ( file_exists( $this->upload_basedir . $user_id . '/' . $name . '-' . $size . $ext ) ) {
unlink( $this->upload_basedir . $user_id . '/' . $name . '-' . $size . $ext );
}
if ( $size < $w ) {
$thumb_s = imagecreatefromjpeg( $source );
$thumb = imagecreatetruecolor( $size, $height );
imagecopyresampled( $thumb, $thumb_s, 0, 0, 0, 0, $size, $height, $w, $h );
imagejpeg( $thumb, $this->upload_basedir . $user_id . '/' . $name . '-' . $size . $ext, 100);
}
}
}
// clean up temp
@@ -542,26 +548,26 @@ class UM_Files {
// update user's meta
do_action('um_before_upload_db_meta', $user_id, $key );
do_action("um_before_upload_db_meta_{$key}", $user_id );
update_user_meta( $user_id, $key, $filename );
do_action('um_after_upload_db_meta', $user_id, $key );
do_action("um_after_upload_db_meta_{$key}", $user_id );
// the url of upload
return $this->upload_baseurl . $user_id . '/' . $filename;
}
/***
*** @Remove a directory
***/
function remove_dir($dir) {
function remove_dir($dir) {
if ( file_exists( $dir ) ) {
foreach(glob($dir . '/*') as $file) {
if(is_dir($file)) $this->remove_dir($file); else unlink($file);
foreach(glob($dir . '/*') as $file) {
if(is_dir($file)) $this->remove_dir($file); else unlink($file);
} rmdir($dir);
}
}
}
}