includes/libs/ParamValidator/Util/UploadedFile.php

   1 <?php
   2
   3 namespace Wikimedia\ParamValidator\Util;
   4
   5 use Psr\Http\Message\UploadedFileInterface;
   6 use RuntimeException;
   7 use Wikimedia\AtEase\AtEase;
   8
   9 /**
  10  * A simple implementation of UploadedFileInterface
  11  *
  12  * This exists so ParamValidator needn't depend on any specific PSR-7
  13  * implementation for a class implementing UploadedFileInterface. It shouldn't
  14  * be used directly by other code, other than perhaps when implementing
  15  * Callbacks::getUploadedFile() when another PSR-7 library is not already in use.
  16  *
  17  * @since 1.34
  18  * @unstable
  19  */
  20 class UploadedFile implements UploadedFileInterface {
  21
  22         /** @var array File data */
  23         private $data;
  24
  25         /** @var bool */
  26         private $fromUpload;
  27
  28         /** @var UploadedFileStream|null */
  29         private $stream = null;
  30
  31         /** @var bool Whether moveTo() was called */
  32         private $moved = false;
  33
  34         /**
  35          * @param array $data Data from $_FILES
  36          * @param bool $fromUpload Set false if using this task with data not from
  37          *  $_FILES. Intended for unit testing.
  38          */
  39         public function __construct( array $data, $fromUpload = true ) {
  40                 $this->data = $data;
  41                 $this->fromUpload = $fromUpload;
  42         }
  43
  44         /**
  45          * Throw if there was an error
  46          * @throws RuntimeException
  47          */
  48         private function checkError() {
  49                 switch ( $this->data['error'] ) {
  50                         case UPLOAD_ERR_OK:
  51                                 break;
  52
  53                         case UPLOAD_ERR_INI_SIZE:
  54                                 throw new RuntimeException( 'Upload exceeded maximum size' );
  55
  56                         case UPLOAD_ERR_FORM_SIZE:
  57                                 throw new RuntimeException( 'Upload exceeded form-specified maximum size' );
  58
  59                         case UPLOAD_ERR_PARTIAL:
  60                                 throw new RuntimeException( 'File was only partially uploaded' );
  61
  62                         case UPLOAD_ERR_NO_FILE:
  63                                 throw new RuntimeException( 'No file was uploaded' );
  64
  65                         case UPLOAD_ERR_NO_TMP_DIR:
  66                                 throw new RuntimeException( 'PHP has no temporary folder for storing uploaded files' );
  67
  68                         case UPLOAD_ERR_CANT_WRITE:
  69                                 throw new RuntimeException( 'PHP was unable to save the uploaded file' );
  70
  71                         case UPLOAD_ERR_EXTENSION:
  72                                 throw new RuntimeException( 'A PHP extension stopped the file upload' );
  73
  74                         default:
  75                                 throw new RuntimeException( 'Unknown upload error code ' . $this->data['error'] );
  76                 }
  77
  78                 if ( $this->moved ) {
  79                         throw new RuntimeException( 'File has already been moved' );
  80                 }
  81                 if ( !isset( $this->data['tmp_name'] ) || !file_exists( $this->data['tmp_name'] ) ) {
  82                         throw new RuntimeException( 'Uploaded file is missing' );
  83                 }
  84         }
  85
  86         public function getStream() {
  87                 if ( $this->stream ) {
  88                         return $this->stream;
  89                 }
  90
  91                 $this->checkError();
  92                 $this->stream = new UploadedFileStream( $this->data['tmp_name'] );
  93                 return $this->stream;
  94         }
  95
  96         public function moveTo( $targetPath ) {
  97                 $this->checkError();
  98
  99                 if ( $this->fromUpload && !is_uploaded_file( $this->data['tmp_name'] ) ) {
 100                         throw new RuntimeException( 'Specified file is not an uploaded file' );
 101                 }
 102
 103                 error_clear_last();
 104                 $ret = AtEase::quietCall(
 105                         $this->fromUpload ? 'move_uploaded_file' : 'rename',
 106                         $this->data['tmp_name'],
 107                         $targetPath
 108                 );
 109                 if ( $ret === false ) {
 110                         $err = error_get_last();
 111                         throw new RuntimeException( "Move failed: " . ( $err['message'] ?? 'Unknown error' ) );
 112                 }
 113
 114                 $this->moved = true;
 115                 if ( $this->stream ) {
 116                         $this->stream->close();
 117                         $this->stream = null;
 118                 }
 119         }
 120
 121         public function getSize() {
 122                 return $this->data['size'] ?? null;
 123         }
 124
 125         public function getError() {
 126                 return $this->data['error'] ?? UPLOAD_ERR_NO_FILE;
 127         }
 128
 129         public function getClientFilename() {
 130                 $ret = $this->data['name'] ?? null;
 131                 return $ret === '' ? null : $ret;
 132         }
 133
 134         public function getClientMediaType() {
 135                 $ret = $this->data['type'] ?? null;
 136                 return $ret === '' ? null : $ret;
 137         }
 138
 139 }