X-Git-Url: http://git.heureux-cyclage.org/?a=blobdiff_plain;f=includes%2FSpecialUpload.php;h=9f8b63f120ba2177a5e30bd68b8ffbf07575e68d;hb=09da7fb96c45255899fa0f4c62412d14b8575b3e;hp=03041de63e402e7dbe5ce6d9d6a88c8dddf52609;hpb=9dda120350bf7b267d92ca0d551c4c4998256e92;p=lhc%2Fweb%2Fwiklou.git diff --git a/includes/SpecialUpload.php b/includes/SpecialUpload.php index 03041de63e..9f8b63f120 100644 --- a/includes/SpecialUpload.php +++ b/includes/SpecialUpload.php @@ -1,290 +1,1063 @@ -addWikiText( wfMsg( "uploaddisabled" ) ); - return; - } - if ( ( 0 == $wgUser->getID() ) - or $wgUser->isBlocked() ) { - $wgOut->errorpage( "uploadnologin", "uploadnologintext" ); - return; - } - if ( wfReadOnly() ) { - $wgOut->readOnlyPage(); - return; - } - if ( isset( $wpReUpload) ) { - unsaveUploadedFile(); - mainUploadForm( "" ); - } else if ( "submit" == $action || isset( $wpUpload ) ) { - processUpload(); - } else { - mainUploadForm( "" ); - } +/** + * + */ +require_once 'Image.php'; +require_once 'MacBinary.php'; +require_once 'Licenses.php'; +/** + * Entry point + */ +function wfSpecialUpload() { + global $wgRequest; + $form = new UploadForm( $wgRequest ); + $form->execute(); } -function processUpload() -{ - global $wgUser, $wgOut, $wgLang, $wpUploadAffirm, $wpUploadFile; - global $wpUploadDescription, $wpIgnoreWarning; - global $HTTP_POST_FILES, $wgUploadDirectory; - global $wpUploadSaveName, $wpUploadTempName, $wpUploadSize; - global $wgSavedFile, $wgUploadOldVersion, $wpUploadOldVersion; - global $wgUseCopyrightUpload , $wpUploadCopyStatus , $wpUploadSource ; - global $wgCheckFileExtensions, $wgFileExtensions; - - if ( $wgUseCopyrightUpload ) - { - $wpUploadAffirm = 1 ; - if ( trim ( $wpUploadCopyStatus ) == "" || trim ( $wpUploadSource ) == "" ) - $wpUploadAffirm = 0 ; - } - - if ( 1 != $wpUploadAffirm ) { - mainUploadForm( WfMsg( "noaffirmation" ) ); - return; - } - if ( ! $wpUploadTempName ) { - $wpUploadTempName = $HTTP_POST_FILES['wpUploadFile']['tmp_name']; +/** + * + * @package MediaWiki + * @subpackage SpecialPage + */ +class UploadForm { + /**#@+ + * @access private + */ + var $mUploadFile, $mUploadDescription, $mLicense ,$mIgnoreWarning, $mUploadError; + var $mUploadSaveName, $mUploadTempName, $mUploadSize, $mUploadOldVersion; + var $mUploadCopyStatus, $mUploadSource, $mReUpload, $mAction, $mUpload; + var $mOname, $mSessionKey, $mStashed, $mDestFile, $mRemoveTempFile; + /**#@-*/ + + /** + * Constructor : initialise object + * Get data POSTed through the form and assign them to the object + * @param $request Data posted. + */ + function UploadForm( &$request ) { + $this->mDestFile = $request->getText( 'wpDestFile' ); + + if( !$request->wasPosted() ) { + # GET requests just give the main form; no data except wpDestfile. + return; + } + + $this->mIgnoreWarning = $request->getCheck( 'wpIgnoreWarning'); + $this->mReUpload = $request->getCheck( 'wpReUpload' ); + $this->mUpload = $request->getCheck( 'wpUpload' ); + + $this->mUploadDescription = $request->getText( 'wpUploadDescription' ); + $this->mLicense = $request->getText( 'wpLicense' ); + $this->mUploadCopyStatus = $request->getText( 'wpUploadCopyStatus' ); + $this->mUploadSource = $request->getText( 'wpUploadSource' ); + $this->mWatchthis = $request->getBool( 'wpWatchthis' ); + wfDebug( "UploadForm: watchthis is: '$this->mWatchthis'\n" ); + + $this->mAction = $request->getVal( 'action' ); + + $this->mSessionKey = $request->getInt( 'wpSessionKey' ); + if( !empty( $this->mSessionKey ) && + isset( $_SESSION['wsUploadData'][$this->mSessionKey] ) ) { + /** + * Confirming a temporarily stashed upload. + * We don't want path names to be forged, so we keep + * them in the session on the server and just give + * an opaque key to the user agent. + */ + $data = $_SESSION['wsUploadData'][$this->mSessionKey]; + $this->mUploadTempName = $data['mUploadTempName']; + $this->mUploadSize = $data['mUploadSize']; + $this->mOname = $data['mOname']; + $this->mUploadError = 0/*UPLOAD_ERR_OK*/; + $this->mStashed = true; + $this->mRemoveTempFile = false; + } else { + /** + *Check for a newly uploaded file. + */ + $this->mUploadTempName = $request->getFileTempName( 'wpUploadFile' ); + $this->mUploadSize = $request->getFileSize( 'wpUploadFile' ); + $this->mOname = $request->getFileName( 'wpUploadFile' ); + $this->mUploadError = $request->getUploadError( 'wpUploadFile' ); + $this->mSessionKey = false; + $this->mStashed = false; + $this->mRemoveTempFile = false; // PHP will handle this + } } - if ( ! $wpUploadSize ) { - $wpUploadSize = $HTTP_POST_FILES['wpUploadFile']['size']; + + /** + * Start doing stuff + * @access public + */ + function execute() { + global $wgUser, $wgOut; + global $wgEnableUploads, $wgUploadDirectory; + + /** Show an error message if file upload is disabled */ + if( ! $wgEnableUploads ) { + $wgOut->addWikiText( wfMsg( 'uploaddisabled' ) ); + return; + } + + /** Various rights checks */ + if( !$wgUser->isAllowed( 'upload' ) || $wgUser->isBlocked() ) { + $wgOut->errorpage( 'uploadnologin', 'uploadnologintext' ); + return; + } + if( wfReadOnly() ) { + $wgOut->readOnlyPage(); + return; + } + + /** Check if the image directory is writeable, this is a common mistake */ + if ( !is_writeable( $wgUploadDirectory ) ) { + $wgOut->addWikiText( wfMsg( 'upload_directory_read_only', $wgUploadDirectory ) ); + return; + } + + if( $this->mReUpload ) { + $this->unsaveUploadedFile(); + $this->mainUploadForm(); + } else if ( 'submit' == $this->mAction || $this->mUpload ) { + $this->processUpload(); + } else { + $this->mainUploadForm(); + } + + $this->cleanupTempFile(); } - $prev = error_reporting( E_ALL & ~( E_NOTICE | E_WARNING ) ); - $oname = wfCleanQueryVar( $HTTP_POST_FILES['wpUploadFile']['name'] ); - if ( $wpUploadSaveName != "" ) $wpUploadSaveName = wfCleanQueryVar( $wpUploadSaveName ); - error_reporting( $prev ); - if ( "" != $oname ) { - $basename = strrchr( $oname, "/" ); - if ( false === $basename ) { $basename = $oname; } - else ( $basename = substr( $basename, 1 ) ); + /* -------------------------------------------------------------- */ - $ext = strrchr( $basename, "." ); - if ( false === $ext ) { $ext = ""; } - else { $ext = substr( $ext, 1 ); } + /** + * Really do the upload + * Checks are made in SpecialUpload::execute() + * @access private + */ + function processUpload() { + global $wgUser, $wgOut, $wgUploadDirectory; + + /* Check for PHP error if any, requires php 4.2 or newer */ + if ( $this->mUploadError == 1/*UPLOAD_ERR_INI_SIZE*/ ) { + $this->mainUploadForm( wfMsgHtml( 'largefileserver' ) ); + return; + } - if ( "" == $ext ) { $xl = 0; } else { $xl = strlen( $ext ) + 1; } - $partname = substr( $basename, 0, strlen( $basename ) - $xl ); + /** + * If there was no filename or a zero size given, give up quick. + */ + if( trim( $this->mOname ) == '' || empty( $this->mUploadSize ) ) { + $this->mainUploadForm( wfMsgHtml( 'emptyfile' ) ); + return; + } + + # Chop off any directories in the given filename + if ( $this->mDestFile ) { + $basename = basename( $this->mDestFile ); + } else { + $basename = basename( $this->mOname ); + } + + /** + * We'll want to blacklist against *any* 'extension', and use + * only the final one for the whitelist. + */ + list( $partname, $ext ) = $this->splitExtensions( $basename ); + if( count( $ext ) ) { + $finalExt = $ext[count( $ext ) - 1]; + } else { + $finalExt = ''; + } + $fullExt = implode( '.', $ext ); if ( strlen( $partname ) < 3 ) { - mainUploadForm( WfMsg( "minlength" ) ); + $this->mainUploadForm( wfMsgHtml( 'minlength' ) ); return; } - $nt = Title::newFromText( $basename ); - $wpUploadSaveName = $nt->getDBkey(); - saveUploadedFile(); - if ( ( ! $wpIgnoreWarning ) && - ( 0 != strcmp( ucfirst( $basename ), $wpUploadSaveName ) ) ) { - return uploadWarning( wfMsg( "badfilename", $wpUploadSaveName ) ); + /** + * Filter out illegal characters, and try to make a legible name + * out of it. We'll strip some silently that Title would die on. + */ + $filtered = preg_replace ( "/[^".Title::legalChars()."]|:/", '-', $basename ); + $nt = Title::newFromText( $filtered ); + if( is_null( $nt ) ) { + $this->uploadError( wfMsgWikiHtml( 'illegalfilename', htmlspecialchars( $filtered ) ) ); + return; + } + $nt =& Title::makeTitle( NS_IMAGE, $nt->getDBkey() ); + $this->mUploadSaveName = $nt->getDBkey(); + + /** + * If the image is protected, non-sysop users won't be able + * to modify it by uploading a new revision. + */ + if( !$nt->userCanEdit() ) { + return $this->uploadError( wfMsgWikiHtml( 'protectedpage' ) ); + } + + /** + * In some cases we may forbid overwriting of existing files. + */ + $overwrite = $this->checkOverwrite( $this->mUploadSaveName ); + if( WikiError::isError( $overwrite ) ) { + return $this->uploadError( $overwrite->toString() ); + } + + /* Don't allow users to override the blacklist (check file extension) */ + global $wgStrictFileExtensions; + global $wgFileExtensions, $wgFileBlacklist; + if( $this->checkFileExtensionList( $ext, $wgFileBlacklist ) || + ($wgStrictFileExtensions && + !$this->checkFileExtension( $finalExt, $wgFileExtensions ) ) ) { + return $this->uploadError( wfMsgHtml( 'badfiletype', htmlspecialchars( $fullExt ) ) ); } - - if ( $wgCheckFileExtensions ) { - if ( ( ! $wpIgnoreWarning ) && - ( ! in_array( strtolower( $ext ), $wgFileExtensions ) ) ) { - return uploadWarning( wfMsg( "badfiletype", $ext ) ); - } + + /** + * Look at the contents of the file; if we can recognize the + * type but it's corrupt or data of the wrong type, we should + * probably not accept it. + */ + if( !$this->mStashed ) { + $this->checkMacBinary(); + $veri = $this->verify( $this->mUploadTempName, $finalExt ); + + if( $veri !== true ) { //it's a wiki error... + return $this->uploadError( $veri->toString() ); + } } - if ( ( ! $wpIgnoreWarning ) && ( $wpUploadSize > 150000 ) ) { - return uploadWarning( WfMsg( "largefile" ) ); + + /** + * Provide an opportunity for extensions to add futher checks + */ + $error = ''; + if( !wfRunHooks( 'UploadVerification', + array( $this->mUploadSaveName, $this->mUploadTempName, &$error ) ) ) { + return $this->uploadError( $error ); + } + + /** + * Check for non-fatal conditions + */ + if ( ! $this->mIgnoreWarning ) { + $warning = ''; + + global $wgCapitalLinks; + if( $wgCapitalLinks ) { + $filtered = ucfirst( $filtered ); + } + if( $this->mUploadSaveName != $filtered ) { + $warning .= '
  • '.wfMsgHtml( 'badfilename', htmlspecialchars( $this->mUploadSaveName ) ).'
    • '; + } + + global $wgCheckFileExtensions; + if ( $wgCheckFileExtensions ) { + if ( ! $this->checkFileExtension( $finalExt, $wgFileExtensions ) ) { + $warning .= '
  • '.wfMsgHtml( 'badfiletype', htmlspecialchars( $fullExt ) ).'
    • '; + } + } + + global $wgUploadSizeWarning; + if ( $wgUploadSizeWarning && ( $this->mUploadSize > $wgUploadSizeWarning ) ) { + # TODO: Format $wgUploadSizeWarning to something that looks better than the raw byte + # value, perhaps add GB,MB and KB suffixes? + $warning .= '
  • '.wfMsgHtml( 'largefile', $wgUploadSizeWarning, $this->mUploadSize ).'
    • '; + } + if ( $this->mUploadSize == 0 ) { + $warning .= '
  • '.wfMsgHtml( 'emptyfile' ).'
    • '; + } + + if( $nt->getArticleID() ) { + global $wgUser; + $sk = $wgUser->getSkin(); + $dlink = $sk->makeKnownLinkObj( $nt ); + $warning .= '
  • '.wfMsgHtml( 'fileexists', $dlink ).'
    • '; + } + + if( $warning != '' ) { + /** + * Stash the file in a temporary location; the user can choose + * to let it through and we'll complete the upload then. + */ + return $this->uploadWarning( $warning ); + } + } + + /** + * Try actually saving the thing... + * It will show an error form on failure. + */ + $hasBeenMunged = !empty( $this->mSessionKey ) || $this->mRemoveTempFile; + if( $this->saveUploadedFile( $this->mUploadSaveName, + $this->mUploadTempName, + $hasBeenMunged ) ) { + /** + * Update the upload log and create the description page + * if it's a new file. + */ + $img = Image::newFromName( $this->mUploadSaveName ); + $success = $img->recordUpload( $this->mUploadOldVersion, + $this->mUploadDescription, + $this->mLicense, + $this->mUploadCopyStatus, + $this->mUploadSource, + $this->mWatchthis ); + + if ( $success ) { + $this->showSuccess(); + } else { + // Image::recordUpload() fails if the image went missing, which is + // unlikely, hence the lack of a specialised message + $wgOut->fileNotFoundError( $this->mUploadSaveName ); + } } } - if ( isset( $wpUploadOldVersion ) ) { - $wgUploadOldVersion = $wpUploadOldVersion; + + /** + * Move the uploaded file from its temporary location to the final + * destination. If a previous version of the file exists, move + * it into the archive subdirectory. + * + * @todo If the later save fails, we may have disappeared the original file. + * + * @param string $saveName + * @param string $tempName full path to the temporary file + * @param bool $useRename if true, doesn't check that the source file + * is a PHP-managed upload temporary + */ + function saveUploadedFile( $saveName, $tempName, $useRename = false ) { + global $wgUploadDirectory, $wgOut; + + $fname= "SpecialUpload::saveUploadedFile"; + + $dest = wfImageDir( $saveName ); + $archive = wfImageArchiveDir( $saveName ); + $this->mSavedFile = "{$dest}/{$saveName}"; + + if( is_file( $this->mSavedFile ) ) { + $this->mUploadOldVersion = gmdate( 'YmdHis' ) . "!{$saveName}"; + wfSuppressWarnings(); + $success = rename( $this->mSavedFile, "${archive}/{$this->mUploadOldVersion}" ); + wfRestoreWarnings(); + + if( ! $success ) { + $wgOut->fileRenameError( $this->mSavedFile, + "${archive}/{$this->mUploadOldVersion}" ); + return false; + } + else wfDebug("$fname: moved file ".$this->mSavedFile." to ${archive}/{$this->mUploadOldVersion}\n"); + } + else { + $this->mUploadOldVersion = ''; + } + + wfSuppressWarnings(); + $success = $useRename + ? rename( $tempName, $this->mSavedFile ) + : move_uploaded_file( $tempName, $this->mSavedFile ); + wfRestoreWarnings(); + + if( ! $success ) { + $wgOut->fileCopyError( $tempName, $this->mSavedFile ); + return false; + } else { + wfDebug("$fname: wrote tempfile $tempName to ".$this->mSavedFile."\n"); + } + + chmod( $this->mSavedFile, 0644 ); + return true; + } + + /** + * Stash a file in a temporary directory for later processing + * after the user has confirmed it. + * + * If the user doesn't explicitly cancel or accept, these files + * can accumulate in the temp directory. + * + * @param string $saveName - the destination filename + * @param string $tempName - the source temporary file to save + * @return string - full path the stashed file, or false on failure + * @access private + */ + function saveTempUploadedFile( $saveName, $tempName ) { + global $wgOut; + $archive = wfImageArchiveDir( $saveName, 'temp' ); + $stash = $archive . '/' . gmdate( "YmdHis" ) . '!' . $saveName; + + $success = $this->mRemoveTempFile + ? rename( $tempName, $stash ) + : move_uploaded_file( $tempName, $stash ); + if ( !$success ) { + $wgOut->fileCopyError( $tempName, $stash ); + return false; + } + + return $stash; } - wfRecordUpload( $wpUploadSaveName, $wgUploadOldVersion, - $wpUploadSize, $wpUploadDescription ); - - $sk = $wgUser->getSkin(); - $ilink = $sk->makeMediaLink( $wpUploadSaveName, wfImageUrl( - $wpUploadSaveName ) ); - $dname = $wgLang->getNsText( Namespace::getImage() ) . ":{$wpUploadSaveName}"; - $dlink = $sk->makeKnownLink( $dname, $dname ); - - $wgOut->addHTML( "

    " . wfMsg( "successfulupload" ) . "

    \n" ); - $text = wfMsg( "fileuploaded", $ilink, $dlink ); - $wgOut->addHTML( "

    {$text}\n" ); - $wgOut->returnToMain( false ); -} -function saveUploadedFile() -{ - global $wpUploadSaveName, $wpUploadTempName; - global $wgSavedFile, $wgUploadOldVersion; - global $wgUploadDirectory, $wgOut; + /** + * Stash a file in a temporary directory for later processing, + * and save the necessary descriptive info into the session. + * Returns a key value which will be passed through a form + * to pick up the path info on a later invocation. + * + * @return int + * @access private + */ + function stashSession() { + $stash = $this->saveTempUploadedFile( + $this->mUploadSaveName, $this->mUploadTempName ); - $dest = wfImageDir( $wpUploadSaveName ); - $archive = wfImageArchiveDir( $wpUploadSaveName ); - $wgSavedFile = "{$dest}/{$wpUploadSaveName}"; + if( !$stash ) { + # Couldn't save the file. + return false; + } - if ( is_file( $wgSavedFile ) ) { - $wgUploadOldVersion = gmdate( "YmdHis" ) . "!{$wpUploadSaveName}"; + $key = mt_rand( 0, 0x7fffffff ); + $_SESSION['wsUploadData'][$key] = array( + 'mUploadTempName' => $stash, + 'mUploadSize' => $this->mUploadSize, + 'mOname' => $this->mOname ); + return $key; + } - if ( ! rename( $wgSavedFile, "${archive}/{$wgUploadOldVersion}" ) ) { - $wgOut->fileRenameError( $wgSavedFile, - "${archive}/{$wgUploadOldVersion}" ); + /** + * Remove a temporarily kept file stashed by saveTempUploadedFile(). + * @access private + */ + function unsaveUploadedFile() { + global $wgOut; + wfSuppressWarnings(); + $success = unlink( $this->mUploadTempName ); + wfRestoreWarnings(); + if ( ! $success ) { + $wgOut->fileDeleteError( $this->mUploadTempName ); + } + } + + /* -------------------------------------------------------------- */ + + /** + * Show some text and linkage on successful upload. + * @access private + */ + function showSuccess() { + global $wgUser, $wgOut, $wgContLang; + + $sk = $wgUser->getSkin(); + $ilink = $sk->makeMediaLink( $this->mUploadSaveName, Image::imageUrl( $this->mUploadSaveName ) ); + $dname = $wgContLang->getNsText( NS_IMAGE ) . ':'.$this->mUploadSaveName; + $dlink = $sk->makeKnownLink( $dname, $dname ); + + $wgOut->addHTML( '

    ' . wfMsgHtml( 'successfulupload' ) . "

    \n" ); + $text = wfMsgWikiHtml( 'fileuploaded', $ilink, $dlink ); + $wgOut->addHTML( $text ); + $wgOut->returnToMain( false ); + } + + /** + * @param string $error as HTML + * @access private + */ + function uploadError( $error ) { + global $wgOut; + $wgOut->addHTML( "

    " . wfMsgHtml( 'uploadwarning' ) . "

    \n" ); + $wgOut->addHTML( "{$error}\n" ); + } + + /** + * There's something wrong with this file, not enough to reject it + * totally but we require manual intervention to save it for real. + * Stash it away, then present a form asking to confirm or cancel. + * + * @param string $warning as HTML + * @access private + */ + function uploadWarning( $warning ) { + global $wgOut, $wgUser, $wgUploadDirectory, $wgRequest; + global $wgUseCopyrightUpload; + + $this->mSessionKey = $this->stashSession(); + if( !$this->mSessionKey ) { + # Couldn't save file; an error has been displayed so let's go. return; } - } else { - $wgUploadOldVersion = ""; + + $wgOut->addHTML( "

    " . wfMsgHtml( 'uploadwarning' ) . "

    \n" ); + $wgOut->addHTML( "
    \n" ); + + $save = wfMsgHtml( 'savefile' ); + $reupload = wfMsgHtml( 'reupload' ); + $iw = wfMsgWikiHtml( 'ignorewarning' ); + $reup = wfMsgWikiHtml( 'reuploaddesc' ); + $titleObj = Title::makeTitle( NS_SPECIAL, 'Upload' ); + $action = $titleObj->escapeLocalURL( 'action=submit' ); + + if ( $wgUseCopyrightUpload ) + { + $copyright = " + mUploadCopyStatus ) . "\" /> + mUploadSource ) . "\" /> + "; + } else { + $copyright = ""; + } + + $wgOut->addHTML( " +
    + + mSessionKey ) . "\" /> + mUploadDescription ) . "\" /> + mLicense ) . "\" /> + mDestFile ) . "\" /> + mWatchthis ) ) . "\" /> + {$copyright} + + + + + + + + + + + +
    + + $iw
    + + $reup
    \n" ); } - if ( ! move_uploaded_file( $wpUploadTempName, $wgSavedFile ) ) { - $wgOut->fileCopyError( $wpUploadTempName, $wgSavedFile ); + + /** + * Displays the main upload form, optionally with a highlighted + * error message up at the top. + * + * @param string $msg as HTML + * @access private + */ + function mainUploadForm( $msg='' ) { + global $wgOut, $wgUser, $wgUploadDirectory, $wgRequest; + global $wgUseCopyrightUpload; + + $cols = intval($wgUser->getOption( 'cols' )); + $ew = $wgUser->getOption( 'editwidth' ); + if ( $ew ) $ew = " style=\"width:100%\""; + else $ew = ''; + + if ( '' != $msg ) { + $sub = wfMsgHtml( 'uploaderror' ); + $wgOut->addHTML( "

    {$sub}

    \n" . + "{$msg}\n" ); + } + $wgOut->addHTML( '
    ' ); + $wgOut->addWikiText( wfMsg( 'uploadtext' ) ); + $wgOut->addHTML( '
    ' ); + $sk = $wgUser->getSkin(); + + + $sourcefilename = wfMsgHtml( 'sourcefilename' ); + $destfilename = wfMsgHtml( 'destfilename' ); + $summary = wfMsgWikiHtml( 'fileuploadsummary' ); + + $licenses = new Licenses(); + $license = wfMsgHtml( 'license' ); + $nolicense = wfMsgHtml( 'nolicense' ); + $licenseshtml = $licenses->getHtml(); + + $ulb = wfMsgHtml( 'uploadbtn' ); + + + $titleObj = Title::makeTitle( NS_SPECIAL, 'Upload' ); + $action = $titleObj->escapeLocalURL(); + + $encDestFile = htmlspecialchars( $this->mDestFile ); + + $watchChecked = $wgUser->getOption( 'watchdefault' ) + ? 'checked="checked"' + : ''; + + $wgOut->addHTML( " +
    + + + + + + + + + + + + + + " ); + + if ( $licenseshtml != '' ) { + $wgOut->addHTML( " + + + + + "); } - chmod( $wgSavedFile, 0644 ); -} -function unsaveUploadedFile() -{ - global $wpSessionKey, $wpUploadOldVersion; - global $wgUploadDirectory, $wgOut, $wsUploadFiles; + if ( $wgUseCopyrightUpload ) { + $filestatus = wfMsgHtml ( 'filestatus' ); + $copystatus = htmlspecialchars( $this->mUploadCopyStatus ); + $filesource = wfMsgHtml ( 'filesource' ); + $uploadsource = htmlspecialchars( $this->mUploadSource ); + + $wgOut->addHTML( " + + + + + + + + + "); + } + - $wgSavedFile = $wsUploadFiles[$wpSessionKey]; - $wgUploadOldVersion = $wpUploadOldVersion; + $wgOut->addHtml( " + + + + + + + + + + + + + + + + +
    + mDestFile?"":"onchange='fillDestFilename()' ") . "size='40' /> +
    + +
    + +
    + +
    + + + + +
    + " ); + $wgOut->addWikiText( wfMsgForContent( 'edittools' ) ); + $wgOut->addHTML( " +
    +
    " ); + } + + /* -------------------------------------------------------------- */ + + /** + * Split a file into a base name and all dot-delimited 'extensions' + * on the end. Some web server configurations will fall back to + * earlier pseudo-'extensions' to determine type and execute + * scripts, so the blacklist needs to check them all. + * + * @return array + */ + function splitExtensions( $filename ) { + $bits = explode( '.', $filename ); + $basename = array_shift( $bits ); + return array( $basename, $bits ); + } - if ( ! @unlink( $wgSavedFile ) ) { - $wgOut->fileDeleteError( $wgSavedFile ); - return; + /** + * Perform case-insensitive match against a list of file extensions. + * Returns true if the extension is in the list. + * + * @param string $ext + * @param array $list + * @return bool + */ + function checkFileExtension( $ext, $list ) { + return in_array( strtolower( $ext ), $list ); } - if ( "" != $wgUploadOldVersion ) { - $hash = md5( substr( $wgUploadOldVersion, 15 ) ); - $archive = "{$wgUploadDirectory}/archive/" . $hash{0} . - "/" . substr( $hash, 0, 2 ); - if ( ! rename( "{$archive}/{$wgUploadOldVersion}", $wgSavedFile ) ) { - $wgOut->fileRenameError( "{$archive}/{$wgUploadOldVersion}", - $wgSavedFile ); + /** + * Perform case-insensitive match against a list of file extensions. + * Returns true if any of the extensions are in the list. + * + * @param array $ext + * @param array $list + * @return bool + */ + function checkFileExtensionList( $ext, $list ) { + foreach( $ext as $e ) { + if( in_array( strtolower( $e ), $list ) ) { + return true; + } } + return false; } -} -function uploadWarning( $warning ) -{ - global $wgOut, $wgUser, $wgLang, $wgUploadDirectory; - global $wpUpload, $wpReUpload, $wpUploadAffirm, $wpUploadFile; - global $wpUploadDescription, $wpIgnoreWarning; - global $wpUploadSaveName, $wpUploadTempName, $wpUploadSize; - global $wgSavedFile, $wgUploadOldVersion; - global $wpSessionKey, $wpUploadOldVersion, $wsUploadFiles; - global $wgUseCopyrightUpload , $wpUploadCopyStatus , $wpUploadSource ; - - # wgSavedFile is stored in the session not the form, for security - $wpSessionKey = mt_rand( 0, 0x7fffffff ); - $wsUploadFiles[$wpSessionKey] = $wgSavedFile; - - $sub = wfMsg( "uploadwarning" ); - $wgOut->addHTML( "

    {$sub}

    \n" ); - $wgOut->addHTML( "

    {$warning}

    \n" ); - - $save = wfMsg( "savefile" ); - $reupload = wfMsg( "reupload" ); - $iw = wfMsg( "ignorewarning" ); - $reup = wfMsg( "reuploaddesc" ); - $action = wfLocalUrlE( $wgLang->specialPage( "Upload" ), - "action=submit" ); - - if ( $wgUseCopyrightUpload ) - { - $copyright = " - - -"; + /** + * Verifies that it's ok to include the uploaded file + * + * @param string $tmpfile the full path of the temporary file to verify + * @param string $extension The filename extension that the file is to be served with + * @return mixed true of the file is verified, a WikiError object otherwise. + */ + function verify( $tmpfile, $extension ) { + #magically determine mime type + $magic=& wfGetMimeMagic(); + $mime= $magic->guessMimeType($tmpfile,false); + + $fname= "SpecialUpload::verify"; + + #check mime type, if desired + global $wgVerifyMimeType; + if ($wgVerifyMimeType) { + + #check mime type against file extension + if( !$this->verifyExtension( $mime, $extension ) ) { + return new WikiErrorMsg( 'uploadcorrupt' ); + } + + #check mime type blacklist + global $wgMimeTypeBlacklist; + if( isset($wgMimeTypeBlacklist) && !is_null($wgMimeTypeBlacklist) + && $this->checkFileExtension( $mime, $wgMimeTypeBlacklist ) ) { + return new WikiErrorMsg( 'badfiletype', htmlspecialchars( $mime ) ); + } + } + + #check for htmlish code and javascript + if( $this->detectScript ( $tmpfile, $mime ) ) { + return new WikiErrorMsg( 'uploadscripted' ); + } + + /** + * Scan the uploaded file for viruses + */ + $virus= $this->detectVirus($tmpfile); + if ( $virus ) { + return new WikiErrorMsg( 'uploadvirus', htmlspecialchars($virus) ); + } + + wfDebug( "$fname: all clear; passing.\n" ); + return true; } - $wgOut->addHTML( " -
    - - - -{$copyright} - - - - - - - -
    - -{$iw}
    - -{$reup}
    \n" ); -} + /** + * Checks if the mime type of the uploaded file matches the file extension. + * + * @param string $mime the mime type of the uploaded file + * @param string $extension The filename extension that the file is to be served with + * @return bool + */ + function verifyExtension( $mime, $extension ) { + $fname = 'SpecialUpload::verifyExtension'; + + $magic =& wfGetMimeMagic(); + + if ( ! $mime || $mime == 'unknown' || $mime == 'unknown/unknown' ) + if ( ! $magic->isRecognizableExtension( $extension ) ) { + wfDebug( "$fname: passing file with unknown detected mime type; unrecognized extension '$extension', can't verify\n" ); + return true; + } else { + wfDebug( "$fname: rejecting file with unknown detected mime type; recognized extension '$extension', so probably invalid file\n" ); + return false; + } + + $match= $magic->isMatchingExtension($extension,$mime); + + if ($match===NULL) { + wfDebug( "$fname: no file extension known for mime type $mime, passing file\n" ); + return true; + } elseif ($match===true) { + wfDebug( "$fname: mime type $mime matches extension $extension, passing file\n" ); + + #TODO: if it's a bitmap, make sure PHP or ImageMagic resp. can handle it! + return true; -function mainUploadForm( $msg ) -{ - global $wgOut, $wgUser, $wgLang, $wgUploadDirectory; - global $wpUpload, $wpUploadAffirm, $wpUploadFile; - global $wpUploadDescription, $wpIgnoreWarning; - global $wgUseCopyrightUpload , $wpUploadSource , $wpUploadCopyStatus ; - - if ( "" != $msg ) { - $sub = wfMsg( "uploaderror" ); - $wgOut->addHTML( "

    {$sub}

    \n" . - "

    {$msg}

    \n" ); - } else { - $sub = wfMsg( "uploadfile" ); - $wgOut->addHTML( "

    {$sub}

    \n" ); + } else { + wfDebug( "$fname: mime type $mime mismatches file extension $extension, rejecting file\n" ); + return false; + } } - $wgOut->addHTML( "

    " . wfMsg( "uploadtext" ) ); - $sk = $wgUser->getSkin(); - - $fn = wfMsg( "filename" ); - $fd = wfMsg( "filedesc" ); - $ulb = wfMsg( "uploadbtn" ); - - $clink = $sk->makeKnownLink( wfMsg( "copyrightpage" ), - wfMsg( "copyrightpagename" ) ); - $ca = wfMsg( "affirmation", $clink ); - $iw = wfMsg( "ignorewarning" ); - - $action = wfLocalUrl( $wgLang->specialPage( "Upload" ) ); - - $source = " - - - -" ; - if ( $wgUseCopyrightUpload ) - { - $source = " -" . wfMsg ( "filestatus" ) . ": - - -". wfMsg ( "filesource" ) . ": - -" ; - } - $wgOut->addHTML( " -

    - - - -{$source} - -
    {$fn}: - -
    {$fd}: - -
      - -
    \n" ); -} + /** Heuristig for detecting files that *could* contain JavaScript instructions or + * things that may look like HTML to a browser and are thus + * potentially harmful. The present implementation will produce false positives in some situations. + * + * @param string $file Pathname to the temporary upload file + * @param string $mime The mime type of the file + * @return bool true if the file contains something looking like embedded scripts + */ + function detectScript($file,$mime) { + + #ugly hack: for text files, always look at the entire file. + #For binarie field, just check the first K. + + if (strpos($mime,'text/')===0) $chunk = file_get_contents( $file ); + else { + $fp = fopen( $file, 'rb' ); + $chunk = fread( $fp, 1024 ); + fclose( $fp ); + } + + $chunk= strtolower( $chunk ); + + if (!$chunk) return false; + + #decode from UTF-16 if needed (could be used for obfuscation). + if (substr($chunk,0,2)=="\xfe\xff") $enc= "UTF-16BE"; + elseif (substr($chunk,0,2)=="\xff\xfe") $enc= "UTF-16LE"; + else $enc= NULL; + + if ($enc) $chunk= iconv($enc,"ASCII//IGNORE",$chunk); + + $chunk= trim($chunk); + + #FIXME: convert from UTF-16 if necessarry! + + wfDebug("SpecialUpload::detectScript: checking for embedded scripts and HTML stuff\n"); + + #check for HTML doctype + if (eregi("addHTML( "
    Bad configuration: unknown virus scanner: $wgAntivirus
    \n" ); #LOCALIZE + + return "unknown antivirus: $wgAntivirus"; + } + + #look up scanner configuration + $virus_scanner= $wgAntivirusSetup[$wgAntivirus]["command"]; #command pattern + $virus_scanner_codes= $wgAntivirusSetup[$wgAntivirus]["codemap"]; #exit-code map + $msg_pattern= $wgAntivirusSetup[$wgAntivirus]["messagepattern"]; #message pattern + + $scanner= $virus_scanner; #copy, so we can resolve the pattern + + if (strpos($scanner,"%f")===false) $scanner.= " ".wfEscapeShellArg($file); #simple pattern: append file to scan + else $scanner= str_replace("%f",wfEscapeShellArg($file),$scanner); #complex pattern: replace "%f" with file to scan + + wfDebug("$fname: running virus scan: $scanner \n"); + + #execute virus scanner + $code= false; + + #NOTE: there's a 50 line workaround to make stderr redirection work on windows, too. + # that does not seem to be worth the pain. + # Ask me (Duesentrieb) about it if it's ever needed. + if (wfIsWindows()) exec("$scanner",$output,$code); + else exec("$scanner 2>&1",$output,$code); + + $exit_code= $code; #remeber for user feedback + + if ($virus_scanner_codes) { #map exit code to AV_xxx constants. + if (isset($virus_scanner_codes[$code])) $code= $virus_scanner_codes[$code]; #explicite mapping + else if (isset($virus_scanner_codes["*"])) $code= $virus_scanner_codes["*"]; #fallback mapping + } + + if ($code===AV_SCAN_FAILED) { #scan failed (code was mapped to false by $virus_scanner_codes) + wfDebug("$fname: failed to scan $file (code $exit_code).\n"); + + if ($wgAntivirusRequired) return "scan failed (code $exit_code)"; + else return NULL; + } + else if ($code===AV_SCAN_ABORTED) { #scan failed because filetype is unknown (probably imune) + wfDebug("$fname: unsupported file type $file (code $exit_code).\n"); + return NULL; + } + else if ($code===AV_NO_VIRUS) { + wfDebug("$fname: file passed virus scan.\n"); + return false; #no virus found + } + else { + $output= join("\n",$output); + $output= trim($output); + + if (!$output) $output= true; #if ther's no output, return true + else if ($msg_pattern) { + $groups= array(); + if (preg_match($msg_pattern,$output,$groups)) { + if ($groups[1]) $output= $groups[1]; + } + } + + wfDebug("$fname: FOUND VIRUS! scanner feedback: $output"); + return $output; + } + } + + /** + * Check if the temporary file is MacBinary-encoded, as some uploads + * from Internet Explorer on Mac OS Classic and Mac OS X will be. + * If so, the data fork will be extracted to a second temporary file, + * which will then be checked for validity and either kept or discarded. + * + * @access private + */ + function checkMacBinary() { + $macbin = new MacBinary( $this->mUploadTempName ); + if( $macbin->isValid() ) { + $dataFile = tempnam( wfTempDir(), "WikiMacBinary" ); + $dataHandle = fopen( $dataFile, 'wb' ); + + wfDebug( "SpecialUpload::checkMacBinary: Extracting MacBinary data fork to $dataFile\n" ); + $macbin->extractData( $dataHandle ); + + $this->mUploadTempName = $dataFile; + $this->mUploadSize = $macbin->dataForkLength(); + + // We'll have to manually remove the new file if it's not kept. + $this->mRemoveTempFile = true; + } + $macbin->close(); + } + + /** + * If we've modified the upload file we need to manually remove it + * on exit to clean up. + * @access private + */ + function cleanupTempFile() { + if( $this->mRemoveTempFile && file_exists( $this->mUploadTempName ) ) { + wfDebug( "SpecialUpload::cleanupTempFile: Removing temporary file $this->mUploadTempName\n" ); + unlink( $this->mUploadTempName ); + } + } + + /** + * Check if there's an overwrite conflict and, if so, if restrictions + * forbid this user from performing the upload. + * + * @return mixed true on success, WikiError on failure + * @access private + */ + function checkOverwrite( $name ) { + $img = Image::newFromName( $name ); + if( is_null( $img ) ) { + // Uh... this shouldn't happen ;) + // But if it does, fall through to previous behavior + return false; + } + + $error = ''; + if( $img->exists() ) { + global $wgUser, $wgOut; + if( $img->isLocal() ) { + if( !$wgUser->isAllowed( 'reupload' ) ) { + $error = 'fileexists-forbidden'; + } + } else { + if( !$wgUser->isAllowed( 'reupload' ) || + !$wgUser->isAllowed( 'reupload-shared' ) ) { + $error = "fileexists-shared-forbidden"; + } + } + } + + if( $error ) { + $errorText = wfMsg( $error, wfEscapeWikiText( $img->getName() ) ); + return new WikiError( $wgOut->parse( $errorText ) ); + } + + // Rockin', go ahead and upload + return true; + } + +} ?>