Skip to content

Instantly share code, notes, and snippets.

@ptrsmk
Forked from hakre/dl-file.php
Last active October 24, 2018 13:52
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save ptrsmk/a851ecdb8d820b242cfb5813bfb8a2a8 to your computer and use it in GitHub Desktop.
Save ptrsmk/a851ecdb8d820b242cfb5813bfb8a2a8 to your computer and use it in GitHub Desktop.
Wordpress login to download uploaded files
<?php
/* dl-file.php
* Protect uploaded files with login.
* @link http://wordpress.stackexchange.com/questions/37144/protect-wordpress-uploads-if-user-is-not-logged-in
*
* @author hakre
* @license GPL-3.0+
* @registry SPDX
*/
require_once('wp-load.php');
require_once ABSPATH . WPINC . '/formatting.php';
require_once ABSPATH . WPINC . '/capabilities.php';
require_once ABSPATH . WPINC . '/user.php';
require_once ABSPATH . WPINC . '/meta.php';
require_once ABSPATH . WPINC . '/post.php';
require_once ABSPATH . WPINC . '/pluggable.php';
wp_cookie_constants();
ob_end_clean();
ob_end_flush();
is_user_logged_in() || auth_redirect();
list($basedir) = array_values(array_intersect_key(wp_upload_dir(), array('basedir' => 1)))+array(NULL);
//$file = rtrim($basedir,'/') . '/' . str_replace('..', '', isset($_GET[ 'file' ])?$_GET[ 'file' ]:'');
$file = rtrim($basedir, '/') . '/' . (isset($_GET['file']) ? $_GET['file'] : '');
//echo $file;
$file = realpath($file);
//print_r( $file );
if ($file === FALSE || !$basedir || !is_file($file)) {
status_header(404);
die('404 &#8212; File not found.');
}
if (strpos($file, $basedir) !== 0) {
status_header(403);
die('403 &#8212; Access denied.');
}
$privileged_roles = array( 'subscriber', 'administrator' );
$user = wp_get_current_user();
$role = ( array ) $user->roles;
$role_result = array_intersect( $privileged_roles, $role );
if( empty( $role_result ) ) {
status_header(403);
die('403 &#8212; Access denied.');
}
$mime = wp_check_filetype($file);
if( false === $mime[ 'type' ] && function_exists( 'mime_content_type' ) )
$mime[ 'type' ] = mime_content_type( $file );
if( $mime[ 'type' ] )
$mimetype = $mime[ 'type' ];
else
$mimetype = 'image/' . substr( $file, strrpos( $file, '.' ) + 1 );
header( 'Content-Type: ' . $mimetype ); // always send this
if ( false === strpos( $_SERVER['SERVER_SOFTWARE'], 'Microsoft-IIS' ) )
header( 'Content-Length: ' . filesize( $file ) );
$last_modified = gmdate( 'D, d M Y H:i:s', filemtime( $file ) );
$etag = '"' . md5( $last_modified ) . '"';
header( "Last-Modified: $last_modified GMT" );
header( 'ETag: ' . $etag );
header( 'Expires: ' . gmdate( 'D, d M Y H:i:s', time() + 100000000 ) . ' GMT' );
header("Content-Transfer-Encoding: Binary");
$filename = pathinfo( $file, PATHINFO_BASENAME );
header("Content-disposition: attachment; filename=".$filename);
// Support for Conditional GET
$client_etag = isset( $_SERVER['HTTP_IF_NONE_MATCH'] ) ? stripslashes( $_SERVER['HTTP_IF_NONE_MATCH'] ) : false;
if( ! isset( $_SERVER['HTTP_IF_MODIFIED_SINCE'] ) )
$_SERVER['HTTP_IF_MODIFIED_SINCE'] = false;
$client_last_modified = trim( $_SERVER['HTTP_IF_MODIFIED_SINCE'] );
// If string is empty, return 0. If not, attempt to parse into a timestamp
$client_modified_timestamp = $client_last_modified ? strtotime( $client_last_modified ) : 0;
// Make a timestamp for our most recent modification...
$modified_timestamp = strtotime($last_modified);
if ( ( $client_last_modified && $client_etag )
? ( ( $client_modified_timestamp >= $modified_timestamp) && ( $client_etag == $etag ) )
: ( ( $client_modified_timestamp >= $modified_timestamp) || ( $client_etag == $etag ) )
) {
status_header( 304 );
exit;
}
// If we made it this far, just serve the file
readfile( $file );
exit;
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment