Nothing clumsy about it in the first place; having the file name in the title is...
[lhc/web/wiklou.git] / includes / SpecialUpload.php
1 <?php
2 /**
3 *
4 * @addtogroup SpecialPage
5 */
6
7
8 /**
9 * Entry point
10 */
11 function wfSpecialUpload() {
12 global $wgRequest;
13 $form = new UploadForm( $wgRequest );
14 $form->execute();
15 }
16
17 /**
18 * implements Special:Upload
19 * @addtogroup SpecialPage
20 */
21 class UploadForm {
22 /**#@+
23 * @access private
24 */
25 var $mComment, $mLicense, $mIgnoreWarning, $mCurlError;
26 var $mDestName, $mTempPath, $mFileSize, $mFileProps;
27 var $mCopyrightStatus, $mCopyrightSource, $mReUpload, $mAction, $mUploadClicked;
28 var $mSrcName, $mSessionKey, $mStashed, $mDesiredDestName, $mRemoveTempFile, $mSourceType;
29 var $mDestWarningAck, $mCurlDestHandle;
30 var $mLocalFile;
31
32 # Placeholders for text injection by hooks (must be HTML)
33 # extensions should take care to _append_ to the present value
34 var $uploadFormTextTop;
35 var $uploadFormTextAfterSummary;
36
37 const SESSION_VERSION = 1;
38 /**#@-*/
39
40 /**
41 * Constructor : initialise object
42 * Get data POSTed through the form and assign them to the object
43 * @param $request Data posted.
44 */
45 function UploadForm( &$request ) {
46 global $wgAllowCopyUploads;
47 $this->mDesiredDestName = $request->getText( 'wpDestFile' );
48 $this->mIgnoreWarning = $request->getCheck( 'wpIgnoreWarning' );
49 $this->mComment = $request->getText( 'wpUploadDescription' );
50
51 if( !$request->wasPosted() ) {
52 # GET requests just give the main form; no data except destination
53 # filename and description
54 return;
55 }
56
57 # Placeholders for text injection by hooks (empty per default)
58 $this->uploadFormTextTop = "";
59 $this->uploadFormTextAfterSummary = "";
60
61 $this->mReUpload = $request->getCheck( 'wpReUpload' );
62 $this->mUploadClicked = $request->getCheck( 'wpUpload' );
63
64 $this->mLicense = $request->getText( 'wpLicense' );
65 $this->mCopyrightStatus = $request->getText( 'wpUploadCopyStatus' );
66 $this->mCopyrightSource = $request->getText( 'wpUploadSource' );
67 $this->mWatchthis = $request->getBool( 'wpWatchthis' );
68 $this->mSourceType = $request->getText( 'wpSourceType' );
69 $this->mDestWarningAck = $request->getText( 'wpDestFileWarningAck' );
70
71 $this->mAction = $request->getVal( 'action' );
72
73 $this->mSessionKey = $request->getInt( 'wpSessionKey' );
74 if( !empty( $this->mSessionKey ) &&
75 isset( $_SESSION['wsUploadData'][$this->mSessionKey]['version'] ) &&
76 $_SESSION['wsUploadData'][$this->mSessionKey]['version'] == self::SESSION_VERSION ) {
77 /**
78 * Confirming a temporarily stashed upload.
79 * We don't want path names to be forged, so we keep
80 * them in the session on the server and just give
81 * an opaque key to the user agent.
82 */
83 $data = $_SESSION['wsUploadData'][$this->mSessionKey];
84 $this->mTempPath = $data['mTempPath'];
85 $this->mFileSize = $data['mFileSize'];
86 $this->mSrcName = $data['mSrcName'];
87 $this->mFileProps = $data['mFileProps'];
88 $this->mCurlError = 0/*UPLOAD_ERR_OK*/;
89 $this->mStashed = true;
90 $this->mRemoveTempFile = false;
91 } else {
92 /**
93 *Check for a newly uploaded file.
94 */
95 if( $wgAllowCopyUploads && $this->mSourceType == 'web' ) {
96 $this->initializeFromUrl( $request );
97 } else {
98 $this->initializeFromUpload( $request );
99 }
100 }
101 }
102
103 /**
104 * Initialize the uploaded file from PHP data
105 * @access private
106 */
107 function initializeFromUpload( $request ) {
108 $this->mTempPath = $request->getFileTempName( 'wpUploadFile' );
109 $this->mFileSize = $request->getFileSize( 'wpUploadFile' );
110 $this->mSrcName = $request->getFileName( 'wpUploadFile' );
111 $this->mCurlError = $request->getUploadError( 'wpUploadFile' );
112 $this->mSessionKey = false;
113 $this->mStashed = false;
114 $this->mRemoveTempFile = false; // PHP will handle this
115 }
116
117 /**
118 * Copy a web file to a temporary file
119 * @access private
120 */
121 function initializeFromUrl( $request ) {
122 global $wgTmpDirectory;
123 $url = $request->getText( 'wpUploadFileURL' );
124 $local_file = tempnam( $wgTmpDirectory, 'WEBUPLOAD' );
125
126 $this->mTempPath = $local_file;
127 $this->mFileSize = 0; # Will be set by curlCopy
128 $this->mCurlError = $this->curlCopy( $url, $local_file );
129 $this->mSrcName = array_pop( explode( '/', $url ) );
130 $this->mSessionKey = false;
131 $this->mStashed = false;
132
133 // PHP won't auto-cleanup the file
134 $this->mRemoveTempFile = file_exists( $local_file );
135 }
136
137 /**
138 * Safe copy from URL
139 * Returns true if there was an error, false otherwise
140 */
141 private function curlCopy( $url, $dest ) {
142 global $wgUser, $wgOut;
143
144 if( !$wgUser->isAllowed( 'upload_by_url' ) ) {
145 $wgOut->permissionRequired( 'upload_by_url' );
146 return true;
147 }
148
149 # Maybe remove some pasting blanks :-)
150 $url = trim( $url );
151 if( stripos($url, 'http://') !== 0 && stripos($url, 'ftp://') !== 0 ) {
152 # Only HTTP or FTP URLs
153 $wgOut->errorPage( 'upload-proto-error', 'upload-proto-error-text' );
154 return true;
155 }
156
157 # Open temporary file
158 $this->mCurlDestHandle = @fopen( $this->mTempPath, "wb" );
159 if( $this->mCurlDestHandle === false ) {
160 # Could not open temporary file to write in
161 $wgOut->errorPage( 'upload-file-error', 'upload-file-error-text');
162 return true;
163 }
164
165 $ch = curl_init();
166 curl_setopt( $ch, CURLOPT_HTTP_VERSION, 1.0); # Probably not needed, but apparently can work around some bug
167 curl_setopt( $ch, CURLOPT_TIMEOUT, 10); # 10 seconds timeout
168 curl_setopt( $ch, CURLOPT_LOW_SPEED_LIMIT, 512); # 0.5KB per second minimum transfer speed
169 curl_setopt( $ch, CURLOPT_URL, $url);
170 curl_setopt( $ch, CURLOPT_WRITEFUNCTION, array( $this, 'uploadCurlCallback' ) );
171 curl_exec( $ch );
172 $error = curl_errno( $ch ) ? true : false;
173 $errornum = curl_errno( $ch );
174 // if ( $error ) print curl_error ( $ch ) ; # Debugging output
175 curl_close( $ch );
176
177 fclose( $this->mCurlDestHandle );
178 unset( $this->mCurlDestHandle );
179 if( $error ) {
180 unlink( $dest );
181 if( wfEmptyMsg( "upload-curl-error$errornum", wfMsg("upload-curl-error$errornum") ) )
182 $wgOut->errorPage( 'upload-misc-error', 'upload-misc-error-text' );
183 else
184 $wgOut->errorPage( "upload-curl-error$errornum", "upload-curl-error$errornum-text" );
185 }
186
187 return $error;
188 }
189
190 /**
191 * Callback function for CURL-based web transfer
192 * Write data to file unless we've passed the length limit;
193 * if so, abort immediately.
194 * @access private
195 */
196 function uploadCurlCallback( $ch, $data ) {
197 global $wgMaxUploadSize;
198 $length = strlen( $data );
199 $this->mFileSize += $length;
200 if( $this->mFileSize > $wgMaxUploadSize ) {
201 return 0;
202 }
203 fwrite( $this->mCurlDestHandle, $data );
204 return $length;
205 }
206
207 /**
208 * Start doing stuff
209 * @access public
210 */
211 function execute() {
212 global $wgUser, $wgOut;
213 global $wgEnableUploads;
214
215 # Check uploading enabled
216 if( !$wgEnableUploads ) {
217 $wgOut->showErrorPage( 'uploaddisabled', 'uploaddisabledtext', array( $this->mDesiredDestName ) );
218 return;
219 }
220
221 # Check permissions
222 if( !$wgUser->isAllowed( 'upload' ) ) {
223 if( !$wgUser->isLoggedIn() ) {
224 $wgOut->showErrorPage( 'uploadnologin', 'uploadnologintext' );
225 } else {
226 $wgOut->permissionRequired( 'upload' );
227 }
228 return;
229 }
230
231 # Check blocks
232 if( $wgUser->isBlocked() ) {
233 $wgOut->blockedPage();
234 return;
235 }
236
237 if( wfReadOnly() ) {
238 $wgOut->readOnlyPage();
239 return;
240 }
241
242 if( $this->mReUpload ) {
243 if( !$this->unsaveUploadedFile() ) {
244 return;
245 }
246 $this->mainUploadForm();
247 } else if( 'submit' == $this->mAction || $this->mUploadClicked ) {
248 $this->processUpload();
249 } else {
250 $this->mainUploadForm();
251 }
252
253 $this->cleanupTempFile();
254 }
255
256 /* -------------------------------------------------------------- */
257
258 /**
259 * Really do the upload
260 * Checks are made in SpecialUpload::execute()
261 * @access private
262 */
263 function processUpload() {
264 global $wgUser, $wgOut;
265
266 if( !wfRunHooks( 'UploadForm:BeforeProcessing', array( &$this ) ) )
267 {
268 wfDebug( "Hook 'UploadForm:BeforeProcessing' broke processing the file." );
269 return false;
270 }
271
272 /* Check for PHP error if any, requires php 4.2 or newer */
273 if( $this->mCurlError == 1/*UPLOAD_ERR_INI_SIZE*/ ) {
274 $this->mainUploadForm( wfMsgHtml( 'largefileserver' ) );
275 return;
276 }
277
278 /**
279 * If there was no filename or a zero size given, give up quick.
280 */
281 if( trim( $this->mSrcName ) == '' || empty( $this->mFileSize ) ) {
282 $this->mainUploadForm( wfMsgHtml( 'emptyfile' ) );
283 return;
284 }
285
286 # Chop off any directories in the given filename
287 if( $this->mDesiredDestName ) {
288 $basename = $this->mDesiredDestName;
289 } else {
290 $basename = $this->mSrcName;
291 }
292 $filtered = wfBaseName( $basename );
293
294 /**
295 * We'll want to blacklist against *any* 'extension', and use
296 * only the final one for the whitelist.
297 */
298 list( $partname, $ext ) = $this->splitExtensions( $filtered );
299
300 if( count( $ext ) ) {
301 $finalExt = $ext[count( $ext ) - 1];
302 } else {
303 $finalExt = '';
304 }
305
306 # If there was more than one "extension", reassemble the base
307 # filename to prevent bogus complaints about length
308 if( count( $ext ) > 1 ) {
309 for( $i = 0; $i < count( $ext ) - 1; $i++ )
310 $partname .= '.' . $ext[$i];
311 }
312
313 if( strlen( $partname ) < 1 ) {
314 $this->mainUploadForm( wfMsgHtml( 'minlength1' ) );
315 return;
316 }
317
318 /**
319 * Filter out illegal characters, and try to make a legible name
320 * out of it. We'll strip some silently that Title would die on.
321 */
322 $filtered = preg_replace ( "/[^".Title::legalChars()."]|:/", '-', $filtered );
323 $nt = Title::makeTitleSafe( NS_IMAGE, $filtered );
324 if( is_null( $nt ) ) {
325 $this->uploadError( wfMsgWikiHtml( 'illegalfilename', htmlspecialchars( $filtered ) ) );
326 return;
327 }
328 $this->mLocalFile = wfLocalFile( $nt );
329 $this->mDestName = $this->mLocalFile->getName();
330
331 /**
332 * If the image is protected, non-sysop users won't be able
333 * to modify it by uploading a new revision.
334 */
335 if( !$nt->userCan( 'edit' ) ) {
336 return $this->uploadError( wfMsgWikiHtml( 'protectedpage' ) );
337 }
338
339 /**
340 * In some cases we may forbid overwriting of existing files.
341 */
342 $overwrite = $this->checkOverwrite( $this->mDestName );
343 if( WikiError::isError( $overwrite ) ) {
344 return $this->uploadError( $overwrite->toString() );
345 }
346
347 /* Don't allow users to override the blacklist (check file extension) */
348 global $wgStrictFileExtensions;
349 global $wgFileExtensions, $wgFileBlacklist;
350 if ($finalExt == '') {
351 return $this->uploadError( wfMsgExt( 'filetype-missing', array ( 'parseinline' ) ) );
352 } elseif ( $this->checkFileExtensionList( $ext, $wgFileBlacklist ) ||
353 ($wgStrictFileExtensions && !$this->checkFileExtension( $finalExt, $wgFileExtensions ) ) ) {
354 return $this->uploadError( wfMsgExt( 'filetype-badtype', array ( 'parseinline' ),
355 htmlspecialchars( $finalExt ), implode ( ', ', $wgFileExtensions ) ) );
356 }
357
358 /**
359 * Look at the contents of the file; if we can recognize the
360 * type but it's corrupt or data of the wrong type, we should
361 * probably not accept it.
362 */
363 if( !$this->mStashed ) {
364 $this->mFileProps = File::getPropsFromPath( $this->mTempPath, $finalExt );
365 $this->checkMacBinary();
366 $veri = $this->verify( $this->mTempPath, $finalExt );
367
368 if( $veri !== true ) { //it's a wiki error...
369 return $this->uploadError( $veri->toString() );
370 }
371
372 /**
373 * Provide an opportunity for extensions to add further checks
374 */
375 $error = '';
376 if( !wfRunHooks( 'UploadVerification',
377 array( $this->mDestName, $this->mTempPath, &$error ) ) ) {
378 return $this->uploadError( $error );
379 }
380 }
381
382
383 /**
384 * Check for non-fatal conditions
385 */
386 if ( ! $this->mIgnoreWarning ) {
387 $warning = '';
388
389 global $wgCapitalLinks;
390 if( $wgCapitalLinks ) {
391 $filtered = ucfirst( $filtered );
392 }
393 if( $basename != $filtered ) {
394 $warning .= '<li>'.wfMsgHtml( 'badfilename', htmlspecialchars( $this->mDestName ) ).'</li>';
395 }
396
397 global $wgCheckFileExtensions;
398 if ( $wgCheckFileExtensions ) {
399 if ( ! $this->checkFileExtension( $finalExt, $wgFileExtensions ) ) {
400 $warning .= '<li>'.wfMsgExt( 'filetype-badtype', array ( 'parseinline' ),
401 htmlspecialchars( $finalExt ), implode ( ', ', $wgFileExtensions ) ).'</li>';
402 }
403 }
404
405 global $wgUploadSizeWarning;
406 if ( $wgUploadSizeWarning && ( $this->mFileSize > $wgUploadSizeWarning ) ) {
407 $skin = $wgUser->getSkin();
408 $wsize = $skin->formatSize( $wgUploadSizeWarning );
409 $asize = $skin->formatSize( $this->mFileSize );
410 $warning .= '<li>' . wfMsgHtml( 'large-file', $wsize, $asize ) . '</li>';
411 }
412 if ( $this->mFileSize == 0 ) {
413 $warning .= '<li>'.wfMsgHtml( 'emptyfile' ).'</li>';
414 }
415
416 if ( !$this->mDestWarningAck ) {
417 $warning .= self::getExistsWarning( $this->mLocalFile );
418 }
419 if( $warning != '' ) {
420 /**
421 * Stash the file in a temporary location; the user can choose
422 * to let it through and we'll complete the upload then.
423 */
424 return $this->uploadWarning( $warning );
425 }
426 }
427
428 /**
429 * Try actually saving the thing...
430 * It will show an error form on failure.
431 */
432 $pageText = self::getInitialPageText( $this->mComment, $this->mLicense,
433 $this->mCopyrightStatus, $this->mCopyrightSource );
434
435 $status = $this->mLocalFile->upload( $this->mTempPath, $this->mComment, $pageText,
436 File::DELETE_SOURCE, $this->mFileProps );
437 if ( !$status->isGood() ) {
438 $this->showError( $status->getWikiText() );
439 } else {
440 if ( $this->mWatchthis ) {
441 global $wgUser;
442 $wgUser->addWatch( $this->mLocalFile->getTitle() );
443 }
444 // Success, redirect to description page
445 $wgOut->redirect( $this->mLocalFile->getTitle()->getFullURL() );
446 wfRunHooks( 'UploadComplete', array( &$img ) );
447 }
448 }
449
450 /**
451 * Do existence checks on a file and produce a warning
452 * This check is static and can be done pre-upload via AJAX
453 * Returns an HTML fragment consisting of one or more LI elements if there is a warning
454 * Returns an empty string if there is no warning
455 */
456 static function getExistsWarning( $file ) {
457 global $wgUser;
458 // Check for uppercase extension. We allow these filenames but check if an image
459 // with lowercase extension exists already
460 $warning = '';
461 $ext = $file->getExtension();
462 $sk = $wgUser->getSkin();
463 if ( $ext !== '' ) {
464 $partname = substr( $file->getName(), 0, -strlen( $ext ) - 1 );
465 } else {
466 $partname = $file->getName();
467 }
468
469 if ( $ext != strtolower( $ext ) ) {
470 $nt_lc = Title::newFromText( $partname . '.' . strtolower( $ext ) );
471 $file_lc = wfLocalFile( $nt_lc );
472 } else {
473 $file_lc = false;
474 }
475
476 if( $file->exists() ) {
477 $dlink = $sk->makeKnownLinkObj( $file->getTitle() );
478 if ( $file->allowInlineDisplay() ) {
479 $dlink2 = $sk->makeImageLinkObj( $file->getTitle(), wfMsgExt( 'fileexists-thumb', 'parseinline', $dlink ),
480 $file->getName(), 'right', array(), false, true );
481 } elseif ( !$file->allowInlineDisplay() && $file->isSafeFile() ) {
482 $icon = $file->iconThumb();
483 $dlink2 = '<div style="float:right" id="mw-media-icon"><a href="' . $file->getURL() . '">' .
484 $icon->toHtml() . '</a><br />' . $dlink . '</div>';
485 } else {
486 $dlink2 = '';
487 }
488
489 $warning .= '<li>' . wfMsgExt( 'fileexists', 'parseline', $dlink ) . '</li>' . $dlink2;
490
491 } elseif ( $file_lc && $file_lc->exists() ) {
492 # Check if image with lowercase extension exists.
493 # It's not forbidden but in 99% it makes no sense to upload the same filename with uppercase extension
494 $dlink = $sk->makeKnownLinkObj( $nt_lc );
495 if ( $file_lc->allowInlineDisplay() ) {
496 $dlink2 = $sk->makeImageLinkObj( $nt_lc, wfMsgExt( 'fileexists-thumb', 'parseinline', $dlink ),
497 $nt_lc->getText(), 'right', array(), false, true );
498 } elseif ( !$file_lc->allowInlineDisplay() && $file_lc->isSafeFile() ) {
499 $icon = $file_lc->iconThumb();
500 $dlink2 = '<div style="float:right" id="mw-media-icon"><a href="' . $file_lc->getURL() . '">' .
501 $icon->toHtml() . '</a><br />' . $dlink . '</div>';
502 } else {
503 $dlink2 = '';
504 }
505
506 $warning .= '<li>' . wfMsgExt( 'fileexists-extension', 'parsemag' , $partname . '.'
507 . $ext , $dlink ) . '</li>' . $dlink2;
508
509 } elseif ( ( substr( $partname , 3, 3 ) == 'px-' || substr( $partname , 2, 3 ) == 'px-' )
510 && ereg( "[0-9]{2}" , substr( $partname , 0, 2) ) )
511 {
512 # Check for filenames like 50px- or 180px-, these are mostly thumbnails
513 $nt_thb = Title::newFromText( substr( $partname , strpos( $partname , '-' ) +1 ) . '.' . $ext );
514 $file_thb = wfLocalFile( $nt_thb );
515 if ($file_thb->exists() ) {
516 # Check if an image without leading '180px-' (or similiar) exists
517 $dlink = $sk->makeKnownLinkObj( $nt_thb);
518 if ( $file_thb->allowInlineDisplay() ) {
519 $dlink2 = $sk->makeImageLinkObj( $nt_thb,
520 wfMsgExt( 'fileexists-thumb', 'parseinline', $dlink ),
521 $nt_thb->getText(), 'right', array(), false, true );
522 } elseif ( !$file_thb->allowInlineDisplay() && $file_thb->isSafeFile() ) {
523 $icon = $file_thb->iconThumb();
524 $dlink2 = '<div style="float:right" id="mw-media-icon"><a href="' .
525 $file_thb->getURL() . '">' . $icon->toHtml() . '</a><br />' .
526 $dlink . '</div>';
527 } else {
528 $dlink2 = '';
529 }
530
531 $warning .= '<li>' . wfMsgExt( 'fileexists-thumbnail-yes', 'parsemag', $dlink ) .
532 '</li>' . $dlink2;
533 } else {
534 # Image w/o '180px-' does not exists, but we do not like these filenames
535 $warning .= '<li>' . wfMsgExt( 'file-thumbnail-no', 'parseinline' ,
536 substr( $partname , 0, strpos( $partname , '-' ) +1 ) ) . '</li>';
537 }
538 }
539 if ( $file->wasDeleted() ) {
540 # If the file existed before and was deleted, warn the user of this
541 # Don't bother doing so if the image exists now, however
542 $ltitle = SpecialPage::getTitleFor( 'Log' );
543 $llink = $sk->makeKnownLinkObj( $ltitle, wfMsgHtml( 'deletionlog' ),
544 'type=delete&page=' . $file->getTitle()->getPrefixedUrl() );
545 $warning .= '<li>' . wfMsgWikiHtml( 'filewasdeleted', $llink ) . '</li>';
546 }
547 return $warning;
548 }
549
550 static function ajaxGetExistsWarning( $filename ) {
551 $file = wfFindFile( $filename );
552 $s = '&nbsp;';
553 if ( $file ) {
554 $warning = self::getExistsWarning( $file );
555 if ( $warning !== '' ) {
556 $s = "<ul>$warning</ul>";
557 }
558 }
559 return $s;
560 }
561
562 /**
563 * Render a preview of a given license for the AJAX preview on upload
564 *
565 * @param string $license
566 * @return string
567 */
568 public static function ajaxGetLicensePreview( $license ) {
569 global $wgParser, $wgUser;
570 $text = '{{' . $license . '}}';
571 $title = Title::makeTitle( NS_IMAGE, 'Sample.jpg' );
572 $options = ParserOptions::newFromUser( $wgUser );
573
574 // Expand subst: first, then live templates...
575 $text = $wgParser->preSaveTransform( $text, $title, $wgUser, $options );
576 $output = $wgParser->parse( $text, $title, $options );
577
578 return $output->getText();
579 }
580
581 /**
582 * Stash a file in a temporary directory for later processing
583 * after the user has confirmed it.
584 *
585 * If the user doesn't explicitly cancel or accept, these files
586 * can accumulate in the temp directory.
587 *
588 * @param string $saveName - the destination filename
589 * @param string $tempName - the source temporary file to save
590 * @return string - full path the stashed file, or false on failure
591 * @access private
592 */
593 function saveTempUploadedFile( $saveName, $tempName ) {
594 global $wgOut;
595 $repo = RepoGroup::singleton()->getLocalRepo();
596 $status = $repo->storeTemp( $saveName, $tempName );
597 if ( !$status->isGood() ) {
598 $this->showError( $status->getWikiText() );
599 return false;
600 } else {
601 return $status->value;
602 }
603 }
604
605 /**
606 * Stash a file in a temporary directory for later processing,
607 * and save the necessary descriptive info into the session.
608 * Returns a key value which will be passed through a form
609 * to pick up the path info on a later invocation.
610 *
611 * @return int
612 * @access private
613 */
614 function stashSession() {
615 $stash = $this->saveTempUploadedFile( $this->mDestName, $this->mTempPath );
616
617 if( !$stash ) {
618 # Couldn't save the file.
619 return false;
620 }
621
622 $key = mt_rand( 0, 0x7fffffff );
623 $_SESSION['wsUploadData'][$key] = array(
624 'mTempPath' => $stash,
625 'mFileSize' => $this->mFileSize,
626 'mSrcName' => $this->mSrcName,
627 'mFileProps' => $this->mFileProps,
628 'version' => self::SESSION_VERSION,
629 );
630 return $key;
631 }
632
633 /**
634 * Remove a temporarily kept file stashed by saveTempUploadedFile().
635 * @access private
636 * @return success
637 */
638 function unsaveUploadedFile() {
639 global $wgOut;
640 $repo = RepoGroup::singleton()->getLocalRepo();
641 $success = $repo->freeTemp( $this->mTempPath );
642 if ( ! $success ) {
643 $wgOut->showFileDeleteError( $this->mTempPath );
644 return false;
645 } else {
646 return true;
647 }
648 }
649
650 /* -------------------------------------------------------------- */
651
652 /**
653 * @param string $error as HTML
654 * @access private
655 */
656 function uploadError( $error ) {
657 global $wgOut;
658 $wgOut->addHTML( "<h2>" . wfMsgHtml( 'uploadwarning' ) . "</h2>\n" );
659 $wgOut->addHTML( "<span class='error'>{$error}</span>\n" );
660 }
661
662 /**
663 * There's something wrong with this file, not enough to reject it
664 * totally but we require manual intervention to save it for real.
665 * Stash it away, then present a form asking to confirm or cancel.
666 *
667 * @param string $warning as HTML
668 * @access private
669 */
670 function uploadWarning( $warning ) {
671 global $wgOut, $wgContLang;
672 global $wgUseCopyrightUpload;
673
674 $this->mSessionKey = $this->stashSession();
675 if( !$this->mSessionKey ) {
676 # Couldn't save file; an error has been displayed so let's go.
677 return;
678 }
679
680 $wgOut->addHTML( "<h2>" . wfMsgHtml( 'uploadwarning' ) . "</h2>\n" );
681 $wgOut->addHTML( "<ul class='warning'>{$warning}</ul><br />\n" );
682
683 $save = wfMsgHtml( 'savefile' );
684 $reupload = wfMsgHtml( 'reupload' );
685 $iw = wfMsgWikiHtml( 'ignorewarning' );
686 $reup = wfMsgWikiHtml( 'reuploaddesc' );
687 $titleObj = SpecialPage::getTitleFor( 'Upload' );
688 $action = $titleObj->escapeLocalURL( 'action=submit' );
689 $align1 = $wgContLang->isRTL() ? 'left' : 'right';
690 $align2 = $wgContLang->isRTL() ? 'right' : 'left';
691
692 if ( $wgUseCopyrightUpload )
693 {
694 $copyright = "
695 <input type='hidden' name='wpUploadCopyStatus' value=\"" . htmlspecialchars( $this->mCopyrightStatus ) . "\" />
696 <input type='hidden' name='wpUploadSource' value=\"" . htmlspecialchars( $this->mCopyrightSource ) . "\" />
697 ";
698 } else {
699 $copyright = "";
700 }
701
702 $wgOut->addHTML( "
703 <form id='uploadwarning' method='post' enctype='multipart/form-data' action='$action'>
704 <input type='hidden' name='wpIgnoreWarning' value='1' />
705 <input type='hidden' name='wpSessionKey' value=\"" . htmlspecialchars( $this->mSessionKey ) . "\" />
706 <input type='hidden' name='wpUploadDescription' value=\"" . htmlspecialchars( $this->mComment ) . "\" />
707 <input type='hidden' name='wpLicense' value=\"" . htmlspecialchars( $this->mLicense ) . "\" />
708 <input type='hidden' name='wpDestFile' value=\"" . htmlspecialchars( $this->mDesiredDestName ) . "\" />
709 <input type='hidden' name='wpWatchthis' value=\"" . htmlspecialchars( intval( $this->mWatchthis ) ) . "\" />
710 {$copyright}
711 <table border='0'>
712 <tr>
713 <tr>
714 <td align='$align1'>
715 <input tabindex='2' type='submit' name='wpUpload' value=\"$save\" />
716 </td>
717 <td align='$align2'>$iw</td>
718 </tr>
719 <tr>
720 <td align='$align1'>
721 <input tabindex='2' type='submit' name='wpReUpload' value=\"{$reupload}\" />
722 </td>
723 <td align='$align2'>$reup</td>
724 </tr>
725 </tr>
726 </table></form>\n" );
727 }
728
729 /**
730 * Displays the main upload form, optionally with a highlighted
731 * error message up at the top.
732 *
733 * @param string $msg as HTML
734 * @access private
735 */
736 function mainUploadForm( $msg='' ) {
737 global $wgOut, $wgUser, $wgContLang;
738 global $wgUseCopyrightUpload, $wgUseAjax, $wgAjaxUploadDestCheck, $wgAjaxLicensePreview;
739 global $wgRequest, $wgAllowCopyUploads, $wgEnableAPI;
740 global $wgStylePath, $wgStyleVersion;
741
742 $useAjaxDestCheck = $wgUseAjax && $wgAjaxUploadDestCheck;
743 $useAjaxLicensePreview = $wgUseAjax && $wgAjaxLicensePreview;
744
745 $adc = wfBoolToStr( $useAjaxDestCheck );
746 $alp = wfBoolToStr( $useAjaxLicensePreview );
747
748 $wgOut->addScript( "<script type=\"text/javascript\">
749 wgAjaxUploadDestCheck = {$adc};
750 wgAjaxLicensePreview = {$alp};
751 </script>
752 <script type=\"text/javascript\" src=\"{$wgStylePath}/common/upload.js?{$wgStyleVersion}\"></script>
753 " );
754
755 if( !wfRunHooks( 'UploadForm:initial', array( &$this ) ) )
756 {
757 wfDebug( "Hook 'UploadForm:initial' broke output of the upload form" );
758 return false;
759 }
760
761 $cols = intval($wgUser->getOption( 'cols' ));
762 $ew = $wgUser->getOption( 'editwidth' );
763 if ( $ew ) $ew = " style=\"width:100%\"";
764 else $ew = '';
765
766 if ( '' != $msg ) {
767 $sub = wfMsgHtml( 'uploaderror' );
768 $wgOut->addHTML( "<h2>{$sub}</h2>\n" .
769 "<span class='error'>{$msg}</span>\n" );
770 }
771 $wgOut->addHTML( '<div id="uploadtext">' );
772 $wgOut->addWikiText( wfMsgNoTrans( 'uploadtext', $this->mDesiredDestName ) );
773 $wgOut->addHTML( '</div>' );
774
775 $sourcefilename = wfMsgHtml( 'sourcefilename' );
776 $destfilename = wfMsgHtml( 'destfilename' );
777 $summary = wfMsgWikiHtml( 'fileuploadsummary' );
778
779 $licenses = new Licenses();
780 $license = wfMsgExt( 'license', array( 'parseinline' ) );
781 $nolicense = wfMsgHtml( 'nolicense' );
782 $licenseshtml = $licenses->getHtml();
783
784 $ulb = wfMsgHtml( 'uploadbtn' );
785
786
787 $titleObj = SpecialPage::getTitleFor( 'Upload' );
788 $action = $titleObj->escapeLocalURL();
789
790 $encDestName = htmlspecialchars( $this->mDesiredDestName );
791
792 $watchChecked =
793 ( $wgUser->getOption( 'watchdefault' ) ||
794 ( $wgUser->getOption( 'watchcreations' ) && $this->mDesiredDestName == '' ) )
795 ? 'checked="checked"'
796 : '';
797 $warningChecked = $this->mIgnoreWarning ? 'checked' : '';
798
799 // Prepare form for upload or upload/copy
800 if( $wgAllowCopyUploads && $wgUser->isAllowed( 'upload_by_url' ) ) {
801 $filename_form =
802 "<input type='radio' id='wpSourceTypeFile' name='wpSourceType' value='file' " .
803 "onchange='toggle_element_activation(\"wpUploadFileURL\",\"wpUploadFile\")' checked />" .
804 "<input tabindex='1' type='file' name='wpUploadFile' id='wpUploadFile' " .
805 "onfocus='" .
806 "toggle_element_activation(\"wpUploadFileURL\",\"wpUploadFile\");" .
807 "toggle_element_check(\"wpSourceTypeFile\",\"wpSourceTypeURL\")'" .
808 ($this->mDesiredDestName?"":"onchange='fillDestFilename(\"wpUploadFile\")' ") . "size='40' />" .
809 wfMsgHTML( 'upload_source_file' ) . "<br/>" .
810 "<input type='radio' id='wpSourceTypeURL' name='wpSourceType' value='web' " .
811 "onchange='toggle_element_activation(\"wpUploadFile\",\"wpUploadFileURL\")' />" .
812 "<input tabindex='1' type='text' name='wpUploadFileURL' id='wpUploadFileURL' " .
813 "onfocus='" .
814 "toggle_element_activation(\"wpUploadFile\",\"wpUploadFileURL\");" .
815 "toggle_element_check(\"wpSourceTypeURL\",\"wpSourceTypeFile\")'" .
816 ($this->mDesiredDestName?"":"onchange='fillDestFilename(\"wpUploadFileURL\")' ") . "size='40' DISABLED />" .
817 wfMsgHtml( 'upload_source_url' ) ;
818 } else {
819 $filename_form =
820 "<input tabindex='1' type='file' name='wpUploadFile' id='wpUploadFile' " .
821 ($this->mDesiredDestName?"":"onchange='fillDestFilename(\"wpUploadFile\")' ") .
822 "size='40' />" .
823 "<input type='hidden' name='wpSourceType' value='file' />" ;
824 }
825 if ( $useAjaxDestCheck ) {
826 $warningRow = "<tr><td colspan='2' id='wpDestFile-warning'>&nbsp</td></tr>";
827 $destOnkeyup = 'onkeyup="wgUploadWarningObj.keypress();"';
828 } else {
829 $warningRow = '';
830 $destOnkeyup = '';
831 }
832
833 $encComment = htmlspecialchars( $this->mComment );
834 $align1 = $wgContLang->isRTL() ? 'left' : 'right';
835 $align2 = $wgContLang->isRTL() ? 'right' : 'left';
836
837 $wgOut->addHTML( <<<EOT
838 <form id='upload' method='post' enctype='multipart/form-data' action="$action">
839 <table border='0'>
840 <tr>
841 {$this->uploadFormTextTop}
842 <td align='$align1' valign='top'><label for='wpUploadFile'>{$sourcefilename}:</label></td>
843 <td align='$align2'>
844 {$filename_form}
845 </td>
846 </tr>
847 <tr>
848 <td align='$align1'><label for='wpDestFile'>{$destfilename}:</label></td>
849 <td align='$align2'>
850 <input tabindex='2' type='text' name='wpDestFile' id='wpDestFile' size='40'
851 value="$encDestName" $destOnkeyup />
852 </td>
853 </tr>
854 <tr>
855 <td align='$align1'><label for='wpUploadDescription'>{$summary}</label></td>
856 <td align='$align2'>
857 <textarea tabindex='3' name='wpUploadDescription' id='wpUploadDescription' rows='6'
858 cols='{$cols}'{$ew}>$encComment</textarea>
859 {$this->uploadFormTextAfterSummary}
860 </td>
861 </tr>
862 <tr>
863 EOT
864 );
865
866 if ( $licenseshtml != '' ) {
867 global $wgStylePath;
868 $wgOut->addHTML( "
869 <td align='$align1'><label for='wpLicense'>$license:</label></td>
870 <td align='$align2'>
871 <select name='wpLicense' id='wpLicense' tabindex='4'
872 onchange='licenseSelectorCheck()'>
873 <option value=''>$nolicense</option>
874 $licenseshtml
875 </select>
876 </td>
877 </tr>
878 <tr>" );
879 if( $useAjaxLicensePreview ) {
880 $wgOut->addHtml( "
881 <td></td>
882 <td id=\"mw-license-preview\"></td>
883 </tr>
884 <tr>" );
885 }
886 }
887
888 if ( $wgUseCopyrightUpload ) {
889 $filestatus = wfMsgHtml ( 'filestatus' );
890 $copystatus = htmlspecialchars( $this->mCopyrightStatus );
891 $filesource = wfMsgHtml ( 'filesource' );
892 $uploadsource = htmlspecialchars( $this->mCopyrightSource );
893
894 $wgOut->addHTML( "
895 <td align='$align1' nowrap='nowrap'><label for='wpUploadCopyStatus'>$filestatus:</label></td>
896 <td><input tabindex='5' type='text' name='wpUploadCopyStatus' id='wpUploadCopyStatus'
897 value=\"$copystatus\" size='40' /></td>
898 </tr>
899 <tr>
900 <td align='$align1'><label for='wpUploadCopyStatus'>$filesource:</label></td>
901 <td><input tabindex='6' type='text' name='wpUploadSource' id='wpUploadCopyStatus'
902 value=\"$uploadsource\" size='40' /></td>
903 </tr>
904 <tr>
905 ");
906 }
907
908 $wgOut->addHtml( "
909 <td></td>
910 <td>
911 <input tabindex='7' type='checkbox' name='wpWatchthis' id='wpWatchthis' $watchChecked value='true' />
912 <label for='wpWatchthis'>" . wfMsgHtml( 'watchthisupload' ) . "</label>
913 <input tabindex='8' type='checkbox' name='wpIgnoreWarning' id='wpIgnoreWarning' value='true' $warningChecked/>
914 <label for='wpIgnoreWarning'>" . wfMsgHtml( 'ignorewarnings' ) . "</label>
915 </td>
916 </tr>
917 $warningRow
918 <tr>
919 <td></td>
920 <td align='$align2'><input tabindex='9' type='submit' name='wpUpload' value=\"{$ulb}\" /></td>
921 </tr>
922 <tr>
923 <td></td>
924 <td align='$align2'>
925 " );
926 $wgOut->addWikiText( wfMsgForContent( 'edittools' ) );
927 $wgOut->addHTML( "
928 </td>
929 </tr>
930
931 </table>
932 <input type='hidden' name='wpDestFileWarningAck' id='wpDestFileWarningAck' value=''/>
933 </form>" );
934 }
935
936 /* -------------------------------------------------------------- */
937
938 /**
939 * Split a file into a base name and all dot-delimited 'extensions'
940 * on the end. Some web server configurations will fall back to
941 * earlier pseudo-'extensions' to determine type and execute
942 * scripts, so the blacklist needs to check them all.
943 *
944 * @return array
945 */
946 function splitExtensions( $filename ) {
947 $bits = explode( '.', $filename );
948 $basename = array_shift( $bits );
949 return array( $basename, $bits );
950 }
951
952 /**
953 * Perform case-insensitive match against a list of file extensions.
954 * Returns true if the extension is in the list.
955 *
956 * @param string $ext
957 * @param array $list
958 * @return bool
959 */
960 function checkFileExtension( $ext, $list ) {
961 return in_array( strtolower( $ext ), $list );
962 }
963
964 /**
965 * Perform case-insensitive match against a list of file extensions.
966 * Returns true if any of the extensions are in the list.
967 *
968 * @param array $ext
969 * @param array $list
970 * @return bool
971 */
972 function checkFileExtensionList( $ext, $list ) {
973 foreach( $ext as $e ) {
974 if( in_array( strtolower( $e ), $list ) ) {
975 return true;
976 }
977 }
978 return false;
979 }
980
981 /**
982 * Verifies that it's ok to include the uploaded file
983 *
984 * @param string $tmpfile the full path of the temporary file to verify
985 * @param string $extension The filename extension that the file is to be served with
986 * @return mixed true of the file is verified, a WikiError object otherwise.
987 */
988 function verify( $tmpfile, $extension ) {
989 #magically determine mime type
990 $magic=& MimeMagic::singleton();
991 $mime= $magic->guessMimeType($tmpfile,false);
992
993 #check mime type, if desired
994 global $wgVerifyMimeType;
995 if ($wgVerifyMimeType) {
996
997 wfDebug ( "\n\nmime: <$mime> extension: <$extension>\n\n");
998 #check mime type against file extension
999 if( !$this->verifyExtension( $mime, $extension ) ) {
1000 return new WikiErrorMsg( 'uploadcorrupt' );
1001 }
1002
1003 #check mime type blacklist
1004 global $wgMimeTypeBlacklist;
1005 if( isset($wgMimeTypeBlacklist) && !is_null($wgMimeTypeBlacklist)
1006 && $this->checkFileExtension( $mime, $wgMimeTypeBlacklist ) ) {
1007 return new WikiErrorMsg( 'filetype-badmime', htmlspecialchars( $mime ) );
1008 }
1009 }
1010
1011 #check for htmlish code and javascript
1012 if( $this->detectScript ( $tmpfile, $mime, $extension ) ) {
1013 return new WikiErrorMsg( 'uploadscripted' );
1014 }
1015
1016 /**
1017 * Scan the uploaded file for viruses
1018 */
1019 $virus= $this->detectVirus($tmpfile);
1020 if ( $virus ) {
1021 return new WikiErrorMsg( 'uploadvirus', htmlspecialchars($virus) );
1022 }
1023
1024 wfDebug( __METHOD__.": all clear; passing.\n" );
1025 return true;
1026 }
1027
1028 /**
1029 * Checks if the mime type of the uploaded file matches the file extension.
1030 *
1031 * @param string $mime the mime type of the uploaded file
1032 * @param string $extension The filename extension that the file is to be served with
1033 * @return bool
1034 */
1035 function verifyExtension( $mime, $extension ) {
1036 $magic =& MimeMagic::singleton();
1037
1038 if ( ! $mime || $mime == 'unknown' || $mime == 'unknown/unknown' )
1039 if ( ! $magic->isRecognizableExtension( $extension ) ) {
1040 wfDebug( __METHOD__.": passing file with unknown detected mime type; " .
1041 "unrecognized extension '$extension', can't verify\n" );
1042 return true;
1043 } else {
1044 wfDebug( __METHOD__.": rejecting file with unknown detected mime type; ".
1045 "recognized extension '$extension', so probably invalid file\n" );
1046 return false;
1047 }
1048
1049 $match= $magic->isMatchingExtension($extension,$mime);
1050
1051 if ($match===NULL) {
1052 wfDebug( __METHOD__.": no file extension known for mime type $mime, passing file\n" );
1053 return true;
1054 } elseif ($match===true) {
1055 wfDebug( __METHOD__.": mime type $mime matches extension $extension, passing file\n" );
1056
1057 #TODO: if it's a bitmap, make sure PHP or ImageMagic resp. can handle it!
1058 return true;
1059
1060 } else {
1061 wfDebug( __METHOD__.": mime type $mime mismatches file extension $extension, rejecting file\n" );
1062 return false;
1063 }
1064 }
1065
1066 /**
1067 * Heuristic for detecting files that *could* contain JavaScript instructions or
1068 * things that may look like HTML to a browser and are thus
1069 * potentially harmful. The present implementation will produce false positives in some situations.
1070 *
1071 * @param string $file Pathname to the temporary upload file
1072 * @param string $mime The mime type of the file
1073 * @param string $extension The extension of the file
1074 * @return bool true if the file contains something looking like embedded scripts
1075 */
1076 function detectScript($file, $mime, $extension) {
1077 global $wgAllowTitlesInSVG;
1078
1079 #ugly hack: for text files, always look at the entire file.
1080 #For binarie field, just check the first K.
1081
1082 if (strpos($mime,'text/')===0) $chunk = file_get_contents( $file );
1083 else {
1084 $fp = fopen( $file, 'rb' );
1085 $chunk = fread( $fp, 1024 );
1086 fclose( $fp );
1087 }
1088
1089 $chunk= strtolower( $chunk );
1090
1091 if (!$chunk) return false;
1092
1093 #decode from UTF-16 if needed (could be used for obfuscation).
1094 if (substr($chunk,0,2)=="\xfe\xff") $enc= "UTF-16BE";
1095 elseif (substr($chunk,0,2)=="\xff\xfe") $enc= "UTF-16LE";
1096 else $enc= NULL;
1097
1098 if ($enc) $chunk= iconv($enc,"ASCII//IGNORE",$chunk);
1099
1100 $chunk= trim($chunk);
1101
1102 #FIXME: convert from UTF-16 if necessarry!
1103
1104 wfDebug("SpecialUpload::detectScript: checking for embedded scripts and HTML stuff\n");
1105
1106 #check for HTML doctype
1107 if (eregi("<!DOCTYPE *X?HTML",$chunk)) return true;
1108
1109 /**
1110 * Internet Explorer for Windows performs some really stupid file type
1111 * autodetection which can cause it to interpret valid image files as HTML
1112 * and potentially execute JavaScript, creating a cross-site scripting
1113 * attack vectors.
1114 *
1115 * Apple's Safari browser also performs some unsafe file type autodetection
1116 * which can cause legitimate files to be interpreted as HTML if the
1117 * web server is not correctly configured to send the right content-type
1118 * (or if you're really uploading plain text and octet streams!)
1119 *
1120 * Returns true if IE is likely to mistake the given file for HTML.
1121 * Also returns true if Safari would mistake the given file for HTML
1122 * when served with a generic content-type.
1123 */
1124
1125 $tags = array(
1126 '<body',
1127 '<head',
1128 '<html', #also in safari
1129 '<img',
1130 '<pre',
1131 '<script', #also in safari
1132 '<table'
1133 );
1134 if( ! $wgAllowTitlesInSVG && $extension !== 'svg' && $mime !== 'image/svg' ) {
1135 $tags[] = '<title';
1136 }
1137
1138 foreach( $tags as $tag ) {
1139 if( false !== strpos( $chunk, $tag ) ) {
1140 return true;
1141 }
1142 }
1143
1144 /*
1145 * look for javascript
1146 */
1147
1148 #resolve entity-refs to look at attributes. may be harsh on big files... cache result?
1149 $chunk = Sanitizer::decodeCharReferences( $chunk );
1150
1151 #look for script-types
1152 if (preg_match('!type\s*=\s*[\'"]?\s*(?:\w*/)?(?:ecma|java)!sim',$chunk)) return true;
1153
1154 #look for html-style script-urls
1155 if (preg_match('!(?:href|src|data)\s*=\s*[\'"]?\s*(?:ecma|java)script:!sim',$chunk)) return true;
1156
1157 #look for css-style script-urls
1158 if (preg_match('!url\s*\(\s*[\'"]?\s*(?:ecma|java)script:!sim',$chunk)) return true;
1159
1160 wfDebug("SpecialUpload::detectScript: no scripts found\n");
1161 return false;
1162 }
1163
1164 /**
1165 * Generic wrapper function for a virus scanner program.
1166 * This relies on the $wgAntivirus and $wgAntivirusSetup variables.
1167 * $wgAntivirusRequired may be used to deny upload if the scan fails.
1168 *
1169 * @param string $file Pathname to the temporary upload file
1170 * @return mixed false if not virus is found, NULL if the scan fails or is disabled,
1171 * or a string containing feedback from the virus scanner if a virus was found.
1172 * If textual feedback is missing but a virus was found, this function returns true.
1173 */
1174 function detectVirus($file) {
1175 global $wgAntivirus, $wgAntivirusSetup, $wgAntivirusRequired, $wgOut;
1176
1177 if ( !$wgAntivirus ) {
1178 wfDebug( __METHOD__.": virus scanner disabled\n");
1179 return NULL;
1180 }
1181
1182 if ( !$wgAntivirusSetup[$wgAntivirus] ) {
1183 wfDebug( __METHOD__.": unknown virus scanner: $wgAntivirus\n" );
1184 # @TODO: localise
1185 $wgOut->addHTML( "<div class='error'>Bad configuration: unknown virus scanner: <i>$wgAntivirus</i></div>\n" );
1186 return "unknown antivirus: $wgAntivirus";
1187 }
1188
1189 # look up scanner configuration
1190 $command = $wgAntivirusSetup[$wgAntivirus]["command"];
1191 $exitCodeMap = $wgAntivirusSetup[$wgAntivirus]["codemap"];
1192 $msgPattern = isset( $wgAntivirusSetup[$wgAntivirus]["messagepattern"] ) ?
1193 $wgAntivirusSetup[$wgAntivirus]["messagepattern"] : null;
1194
1195 if ( strpos( $command,"%f" ) === false ) {
1196 # simple pattern: append file to scan
1197 $command .= " " . wfEscapeShellArg( $file );
1198 } else {
1199 # complex pattern: replace "%f" with file to scan
1200 $command = str_replace( "%f", wfEscapeShellArg( $file ), $command );
1201 }
1202
1203 wfDebug( __METHOD__.": running virus scan: $command \n" );
1204
1205 # execute virus scanner
1206 $exitCode = false;
1207
1208 #NOTE: there's a 50 line workaround to make stderr redirection work on windows, too.
1209 # that does not seem to be worth the pain.
1210 # Ask me (Duesentrieb) about it if it's ever needed.
1211 $output = array();
1212 if ( wfIsWindows() ) {
1213 exec( "$command", $output, $exitCode );
1214 } else {
1215 exec( "$command 2>&1", $output, $exitCode );
1216 }
1217
1218 # map exit code to AV_xxx constants.
1219 $mappedCode = $exitCode;
1220 if ( $exitCodeMap ) {
1221 if ( isset( $exitCodeMap[$exitCode] ) ) {
1222 $mappedCode = $exitCodeMap[$exitCode];
1223 } elseif ( isset( $exitCodeMap["*"] ) ) {
1224 $mappedCode = $exitCodeMap["*"];
1225 }
1226 }
1227
1228 if ( $mappedCode === AV_SCAN_FAILED ) {
1229 # scan failed (code was mapped to false by $exitCodeMap)
1230 wfDebug( __METHOD__.": failed to scan $file (code $exitCode).\n" );
1231
1232 if ( $wgAntivirusRequired ) {
1233 return "scan failed (code $exitCode)";
1234 } else {
1235 return NULL;
1236 }
1237 } else if ( $mappedCode === AV_SCAN_ABORTED ) {
1238 # scan failed because filetype is unknown (probably imune)
1239 wfDebug( __METHOD__.": unsupported file type $file (code $exitCode).\n" );
1240 return NULL;
1241 } else if ( $mappedCode === AV_NO_VIRUS ) {
1242 # no virus found
1243 wfDebug( __METHOD__.": file passed virus scan.\n" );
1244 return false;
1245 } else {
1246 $output = join( "\n", $output );
1247 $output = trim( $output );
1248
1249 if ( !$output ) {
1250 $output = true; #if there's no output, return true
1251 } elseif ( $msgPattern ) {
1252 $groups = array();
1253 if ( preg_match( $msgPattern, $output, $groups ) ) {
1254 if ( $groups[1] ) {
1255 $output = $groups[1];
1256 }
1257 }
1258 }
1259
1260 wfDebug( __METHOD__.": FOUND VIRUS! scanner feedback: $output" );
1261 return $output;
1262 }
1263 }
1264
1265 /**
1266 * Check if the temporary file is MacBinary-encoded, as some uploads
1267 * from Internet Explorer on Mac OS Classic and Mac OS X will be.
1268 * If so, the data fork will be extracted to a second temporary file,
1269 * which will then be checked for validity and either kept or discarded.
1270 *
1271 * @access private
1272 */
1273 function checkMacBinary() {
1274 $macbin = new MacBinary( $this->mTempPath );
1275 if( $macbin->isValid() ) {
1276 $dataFile = tempnam( wfTempDir(), "WikiMacBinary" );
1277 $dataHandle = fopen( $dataFile, 'wb' );
1278
1279 wfDebug( "SpecialUpload::checkMacBinary: Extracting MacBinary data fork to $dataFile\n" );
1280 $macbin->extractData( $dataHandle );
1281
1282 $this->mTempPath = $dataFile;
1283 $this->mFileSize = $macbin->dataForkLength();
1284
1285 // We'll have to manually remove the new file if it's not kept.
1286 $this->mRemoveTempFile = true;
1287 }
1288 $macbin->close();
1289 }
1290
1291 /**
1292 * If we've modified the upload file we need to manually remove it
1293 * on exit to clean up.
1294 * @access private
1295 */
1296 function cleanupTempFile() {
1297 if ( $this->mRemoveTempFile && file_exists( $this->mTempPath ) ) {
1298 wfDebug( "SpecialUpload::cleanupTempFile: Removing temporary file {$this->mTempPath}\n" );
1299 unlink( $this->mTempPath );
1300 }
1301 }
1302
1303 /**
1304 * Check if there's an overwrite conflict and, if so, if restrictions
1305 * forbid this user from performing the upload.
1306 *
1307 * @return mixed true on success, WikiError on failure
1308 * @access private
1309 */
1310 function checkOverwrite( $name ) {
1311 $img = wfFindFile( $name );
1312
1313 $error = '';
1314 if( $img ) {
1315 global $wgUser, $wgOut;
1316 if( $img->isLocal() ) {
1317 if( !self::userCanReUpload( $wgUser, $img->name ) ) {
1318 $error = 'fileexists-forbidden';
1319 }
1320 } else {
1321 if( !$wgUser->isAllowed( 'reupload' ) ||
1322 !$wgUser->isAllowed( 'reupload-shared' ) ) {
1323 $error = "fileexists-shared-forbidden";
1324 }
1325 }
1326 }
1327
1328 if( $error ) {
1329 $errorText = wfMsg( $error, wfEscapeWikiText( $img->getName() ) );
1330 return new WikiError( $wgOut->parse( $errorText ) );
1331 }
1332
1333 // Rockin', go ahead and upload
1334 return true;
1335 }
1336
1337 /**
1338 * Check if a user is the last uploader
1339 *
1340 * @param User $user
1341 * @param string $img, image name
1342 * @return bool
1343 */
1344 public static function userCanReUpload( User $user, $img ) {
1345 if( $user->isAllowed( 'reupload' ) )
1346 return true; // non-conditional
1347 if( !$user->isAllowed( 'reupload-own' ) )
1348 return false;
1349
1350 $dbr = wfGetDB( DB_SLAVE );
1351 $row = $dbr->selectRow('image',
1352 /* SELECT */ 'img_user',
1353 /* WHERE */ array( 'img_name' => $img )
1354 );
1355 if ( !$row )
1356 return false;
1357
1358 return $user->getID() == $row->img_user;
1359 }
1360
1361 /**
1362 * Display an error with a wikitext description
1363 */
1364 function showError( $description ) {
1365 global $wgOut;
1366 $wgOut->setPageTitle( wfMsg( "internalerror" ) );
1367 $wgOut->setRobotpolicy( "noindex,nofollow" );
1368 $wgOut->setArticleRelated( false );
1369 $wgOut->enableClientCache( false );
1370 $wgOut->addWikiText( $description );
1371 }
1372
1373 /**
1374 * Get the initial image page text based on a comment and optional file status information
1375 */
1376 static function getInitialPageText( $comment, $license, $copyStatus, $source ) {
1377 global $wgUseCopyrightUpload;
1378 if ( $wgUseCopyrightUpload ) {
1379 if ( $license != '' ) {
1380 $licensetxt = '== ' . wfMsgForContent( 'license' ) . " ==\n" . '{{' . $license . '}}' . "\n";
1381 }
1382 $pageText = '== ' . wfMsg ( 'filedesc' ) . " ==\n" . $comment . "\n" .
1383 '== ' . wfMsgForContent ( 'filestatus' ) . " ==\n" . $copyStatus . "\n" .
1384 "$licensetxt" .
1385 '== ' . wfMsgForContent ( 'filesource' ) . " ==\n" . $source ;
1386 } else {
1387 if ( $license != '' ) {
1388 $filedesc = $comment == '' ? '' : '== ' . wfMsg ( 'filedesc' ) . " ==\n" . $comment . "\n";
1389 $pageText = $filedesc .
1390 '== ' . wfMsgForContent ( 'license' ) . " ==\n" . '{{' . $license . '}}' . "\n";
1391 } else {
1392 $pageText = $comment;
1393 }
1394 }
1395 return $pageText;
1396 }
1397 }
1398
1399