7d650e5bcf
* Move the curl calls to http::request() so we can at least start to make this work for the file_get_contents() fallback" Sticking CURL-specific options in the Http::get call seems unwise as that ties you to a specific backend, assuming Http::get won't change. Further, it seems that if it does use the backend, it just plain won't work -- the returned data will be thrown away instead of saved to the target file.
92 lines
2.3 KiB
PHP
92 lines
2.3 KiB
PHP
<?php
|
|
|
|
|
|
class UploadFromUrl extends UploadBase {
|
|
static function isAllowed( $user ) {
|
|
if( !$user->isAllowed( 'upload_by_url' ) )
|
|
return 'upload_by_url';
|
|
return parent::isAllowed( $user );
|
|
}
|
|
static function isEnabled() {
|
|
global $wgAllowCopyUploads;
|
|
return $wgAllowCopyUploads && parent::isEnabled();
|
|
}
|
|
|
|
function initialize( $name, $url ) {
|
|
global $wgTmpDirectory;
|
|
$local_file = tempnam( $wgTmpDirectory, 'WEBUPLOAD' );
|
|
$this-initialize( $name, $local_file, 0, true );
|
|
|
|
$this->mUrl = trim( $url );
|
|
}
|
|
|
|
/**
|
|
* Do the real fetching stuff
|
|
*/
|
|
function fetchFile() {
|
|
if( stripos($this->mUrl, 'http://') !== 0 && stripos($this->mUrl, 'ftp://') !== 0 ) {
|
|
return array(
|
|
'status' => self::BEFORE_PROCESSING,
|
|
'error' => 'upload-proto-error',
|
|
);
|
|
}
|
|
$res = $this->curlCopy();
|
|
if( $res !== true ) {
|
|
return array(
|
|
'status' => self::BEFORE_PROCESSING,
|
|
'error' => $res,
|
|
);
|
|
}
|
|
return self::OK;
|
|
}
|
|
|
|
/**
|
|
* Safe copy from URL
|
|
* Returns true if there was an error, false otherwise
|
|
*/
|
|
private function curlCopy() {
|
|
global $wgUser, $wgOut;
|
|
|
|
# Open temporary file
|
|
$this->mCurlDestHandle = @fopen( $this->mTempPath, "wb" );
|
|
if( $this->mCurlDestHandle === false ) {
|
|
# Could not open temporary file to write in
|
|
return 'upload-file-error';
|
|
}
|
|
|
|
$ch = curl_init();
|
|
curl_setopt( $ch, CURLOPT_HTTP_VERSION, 1.0); # Probably not needed, but apparently can work around some bug
|
|
curl_setopt( $ch, CURLOPT_TIMEOUT, 10); # 10 seconds timeout
|
|
curl_setopt( $ch, CURLOPT_LOW_SPEED_LIMIT, 512); # 0.5KB per second minimum transfer speed
|
|
curl_setopt( $ch, CURLOPT_URL, $this->mUrl);
|
|
curl_setopt( $ch, CURLOPT_WRITEFUNCTION, array( $this, 'uploadCurlCallback' ) );
|
|
curl_exec( $ch );
|
|
$error = curl_errno( $ch );
|
|
curl_close( $ch );
|
|
|
|
fclose( $this->mCurlDestHandle );
|
|
unset( $this->mCurlDestHandle );
|
|
|
|
if( $error )
|
|
return "upload-curl-error$errornum";
|
|
|
|
return true;
|
|
}
|
|
|
|
/**
|
|
* Callback function for CURL-based web transfer
|
|
* Write data to file unless we've passed the length limit;
|
|
* if so, abort immediately.
|
|
* @access private
|
|
*/
|
|
function uploadCurlCallback( $ch, $data ) {
|
|
global $wgMaxUploadSize;
|
|
$length = strlen( $data );
|
|
$this->mFileSize += $length;
|
|
if( $this->mFileSize > $wgMaxUploadSize ) {
|
|
return 0;
|
|
}
|
|
fwrite( $this->mCurlDestHandle, $data );
|
|
return $length;
|
|
}
|
|
}
|