Total Complexity | 54 |
Total Lines | 345 |
Duplicated Lines | 0 % |
Changes | 0 |
Complex classes like PhotoUploader often do a lot of different things. To break such a class down, we need to identify a cohesive component within that class. A common approach to find such a component is to look for fields/methods that share the same prefixes, or suffixes.
Once you have determined the fields that belong together, you can apply the Extract Class refactoring. If the component makes sense as a sub-class, Extract Subclass is also a candidate, and is often faster.
While breaking up the class, it is a good idea to analyze how other classes use PhotoUploader, and based on these observations, apply Extract Interface, too.
1 | <?php |
||
28 | class PhotoUploader |
||
29 | { |
||
30 | public $uploadDir; |
||
31 | public $savedDestination; |
||
32 | public $savedFilename; |
||
33 | public $maxFileSize; |
||
34 | public $maxWidth; |
||
35 | public $maxHeight; |
||
36 | public $isError; |
||
37 | public $error; |
||
38 | public $checkMd5; |
||
39 | |||
40 | /** |
||
41 | * Extgallery\PhotoUploader constructor. |
||
42 | * @param $uploadDir |
||
43 | * @param int $maxFileSize |
||
44 | * @param null $maxWidth |
||
|
|||
45 | * @param null $maxHeight |
||
46 | */ |
||
47 | public function __construct($uploadDir, $maxFileSize = 0, $maxWidth = null, $maxHeight = null) |
||
48 | { |
||
49 | $this->uploadDir = $uploadDir; |
||
50 | $this->maxFileSize = (int)$maxFileSize; |
||
51 | if (isset($maxWidth)) { |
||
52 | $this->maxWidth = (int)$maxWidth; |
||
53 | } |
||
54 | if (isset($maxHeight)) { |
||
55 | $this->maxHeight = (int)$maxHeight; |
||
56 | } |
||
57 | |||
58 | $this->isError = false; |
||
59 | $this->error = ''; |
||
60 | $this->checkMd5 = true; |
||
61 | } |
||
62 | |||
63 | /** |
||
64 | * @param $file |
||
65 | * |
||
66 | * @return bool |
||
67 | */ |
||
68 | public function fetchPhoto($file) |
||
69 | { |
||
70 | $jupart = Request::getInt('jupart', 0, 'POST'); |
||
71 | $jufinal = Request::getInt('jufinal', 1, 'POST'); |
||
72 | $md5sums = $_POST['md5sum'][0] ?? null; |
||
73 | |||
74 | if ('' == $this->uploadDir) { |
||
75 | $this->abort('upload dir not defined'); |
||
76 | |||
77 | return false; |
||
78 | } |
||
79 | |||
80 | if (!\is_dir($this->uploadDir)) { |
||
81 | $this->abort('fail to open upload dir'); |
||
82 | |||
83 | return false; |
||
84 | } |
||
85 | |||
86 | if (!\is_writable($this->uploadDir)) { |
||
87 | $this->abort('upload dir not writable'); |
||
88 | |||
89 | return false; |
||
90 | } |
||
91 | |||
92 | if ($this->checkMd5 && !isset($md5sums)) { |
||
93 | $this->abort('Expecting an MD5 checksum'); |
||
94 | |||
95 | return false; |
||
96 | } |
||
97 | |||
98 | $dstdir = $this->uploadDir; |
||
99 | $dstname = $dstdir . '/juvar.' . \session_id(); |
||
100 | $tmpname = $dstdir . '/juvar.tmp' . \session_id(); |
||
101 | |||
102 | if (!\move_uploaded_file($file['tmp_name'], $tmpname)) { |
||
103 | $this->abort('Unable to move uploaded file'); |
||
104 | |||
105 | return false; |
||
106 | } |
||
107 | |||
108 | if ($jupart) { |
||
109 | // got a chunk of a multi-part upload |
||
110 | $len = \filesize($tmpname); |
||
111 | $_SESSION['juvar.tmpsize'] += $len; |
||
112 | if ($len > 0) { |
||
113 | $src = \fopen($tmpname, 'rb'); |
||
114 | $dst = \fopen($dstname, (1 == $jupart) ? 'wb' : 'ab'); |
||
115 | while ($len > 0) { |
||
116 | $rlen = ($len > 8192) ? 8192 : $len; |
||
117 | $buf = \fread($src, $rlen); |
||
118 | if (!$buf) { |
||
119 | \fclose($src); |
||
120 | \fclose($dst); |
||
121 | \unlink($dstname); |
||
122 | $this->abort('read IO error'); |
||
123 | |||
124 | return false; |
||
125 | } |
||
126 | if (!\fwrite($dst, $buf, $rlen)) { |
||
127 | \fclose($src); |
||
128 | \fclose($dst); |
||
129 | \unlink($dstname); |
||
130 | $this->abort('write IO error'); |
||
131 | |||
132 | return false; |
||
133 | } |
||
134 | $len -= $rlen; |
||
135 | } |
||
136 | \fclose($src); |
||
137 | \fclose($dst); |
||
138 | \unlink($tmpname); |
||
139 | } |
||
140 | if ($jufinal) { |
||
141 | // This is the last chunk. Check total lenght and rename it to it's final name. |
||
142 | $dlen = \filesize($dstname); |
||
143 | if ($dlen != $_SESSION['juvar.tmpsize']) { |
||
144 | $this->abort('file size mismatch'); |
||
145 | |||
146 | return false; |
||
147 | } |
||
148 | if ($this->checkMd5 && ($md5sums != \md5_file($dstname))) { |
||
149 | $this->abort('MD5 checksum mismatch'); |
||
150 | |||
151 | return false; |
||
152 | } |
||
153 | // remove zero sized files |
||
154 | if ($dlen > 0) { |
||
155 | if (!$this->_saveFile($dstname, $file['name'])) { |
||
156 | return false; |
||
157 | } |
||
158 | } else { |
||
159 | $this->abort('0 file size'); |
||
160 | |||
161 | return false; |
||
162 | } |
||
163 | // reset session var |
||
164 | $_SESSION['juvar.tmpsize'] = 0; |
||
165 | } |
||
166 | } else { |
||
167 | // Got a single file upload. Trivial. |
||
168 | if ($this->checkMd5 && $md5sums != \md5_file($tmpname)) { |
||
169 | $this->abort('MD5 checksum mismatch'); |
||
170 | |||
171 | return false; |
||
172 | } |
||
173 | if (!$this->_saveFile($tmpname, $file['name'])) { |
||
174 | return false; |
||
175 | } |
||
176 | } |
||
177 | |||
178 | return true; |
||
179 | } |
||
180 | |||
181 | /** |
||
182 | * @param $tmpDestination |
||
183 | * @param $fileName |
||
184 | * |
||
185 | * @return bool |
||
186 | */ |
||
187 | public function _saveFile($tmpDestination, $fileName) |
||
188 | { |
||
189 | $this->savedFilename = $fileName; |
||
190 | $this->savedDestination = $this->uploadDir . $fileName; |
||
191 | |||
192 | if (!$this->_checkFile($tmpDestination)) { |
||
193 | return false; |
||
194 | } |
||
195 | |||
196 | if (!\rename($tmpDestination, $this->savedDestination)) { |
||
197 | $this->abort('error renaming file'); |
||
198 | |||
199 | return false; |
||
200 | } |
||
201 | |||
202 | @\chmod($this->savedDestination, 0644); |
||
203 | |||
204 | return true; |
||
205 | } |
||
206 | |||
207 | /** |
||
208 | * @param $tmpDestination |
||
209 | * |
||
210 | * @return bool |
||
211 | */ |
||
212 | public function _checkFile($tmpDestination) |
||
213 | { |
||
214 | // $imageExtensions = array(IMAGETYPE_GIF => 'gif', IMAGETYPE_JPEG => 'jpeg', IMAGETYPE_JPG => 'jpg', IMAGETYPE_PNG => 'png'); |
||
215 | |||
216 | $valid_types = [IMAGETYPE_GIF, IMAGETYPE_JPEG, IMAGETYPE_PNG, IMAGETYPE_BMP]; |
||
217 | |||
218 | $imageExtensions = ['gif', 'jpg', 'jpeg', 'png']; |
||
219 | |||
220 | // Check IE XSS before returning success |
||
221 | $ext = mb_strtolower(mb_substr(mb_strrchr($this->savedDestination, '.'), 1)); |
||
222 | $photoInfo = \getimagesize($tmpDestination); |
||
223 | if (false === $photoInfo || $imageExtensions[(int)$photoInfo[2]] != $ext) { |
||
224 | $this->abort('Suspicious image upload refused'); |
||
225 | |||
226 | return false; |
||
227 | } |
||
228 | |||
229 | if (!$this->checkMaxFileSize($tmpDestination)) { |
||
230 | $this->abort('Max file size error'); |
||
231 | |||
232 | return false; |
||
233 | } |
||
234 | |||
235 | if (!$this->checkMaxWidth($photoInfo)) { |
||
236 | $this->abort('Max width error'); |
||
237 | |||
238 | return false; |
||
239 | } |
||
240 | |||
241 | if (!$this->checkMaxHeight($photoInfo)) { |
||
242 | $this->abort('Max height error'); |
||
243 | |||
244 | return false; |
||
245 | } |
||
246 | |||
247 | if (!$this->checkImageType($photoInfo)) { |
||
248 | $this->abort('File type not allowed'); |
||
249 | |||
250 | return false; |
||
251 | } |
||
252 | |||
253 | return true; |
||
254 | } |
||
255 | |||
256 | /** |
||
257 | * @param $file |
||
258 | * |
||
259 | * @return bool |
||
260 | */ |
||
261 | public function checkMaxFileSize($file) |
||
262 | { |
||
263 | if (!isset($this->maxFileSize)) { |
||
264 | return true; |
||
265 | } |
||
266 | |||
267 | if (\filesize($file) > $this->maxFileSize) { |
||
268 | return false; |
||
269 | } |
||
270 | |||
271 | return true; |
||
272 | } |
||
273 | |||
274 | /** |
||
275 | * @param $photoInfo |
||
276 | * |
||
277 | * @return bool |
||
278 | */ |
||
279 | public function checkMaxWidth($photoInfo) |
||
280 | { |
||
281 | if (!isset($this->maxWidth)) { |
||
282 | return true; |
||
283 | } |
||
284 | |||
285 | if ($photoInfo[0] > $this->maxWidth) { |
||
286 | return false; |
||
287 | } |
||
288 | |||
289 | return true; |
||
290 | } |
||
291 | |||
292 | /** |
||
293 | * @param $photoInfo |
||
294 | * |
||
295 | * @return bool |
||
296 | */ |
||
297 | public function checkMaxHeight($photoInfo) |
||
298 | { |
||
299 | if (!isset($this->maxHeight)) { |
||
300 | return true; |
||
301 | } |
||
302 | |||
303 | if ($photoInfo[1] > $this->maxHeight) { |
||
304 | return false; |
||
305 | } |
||
306 | |||
307 | return true; |
||
308 | } |
||
309 | |||
310 | /** |
||
311 | * @param $photoInfo |
||
312 | * |
||
313 | * @return bool |
||
314 | */ |
||
315 | public function checkImageType($photoInfo) |
||
324 | } |
||
325 | |||
326 | /** |
||
327 | * @param string $msg |
||
328 | */ |
||
329 | public function abort($msg = '') |
||
330 | { |
||
331 | // remove all uploaded files of *this* request |
||
332 | if (isset($_FILES)) { |
||
333 | foreach ($_FILES as $key => $val) { |
||
334 | //@unlink($val['tmp_name']); |
||
335 | } |
||
336 | } |
||
337 | |||
338 | // remove accumulated file, if any. |
||
339 | //@unlink($this->uploadDir .'/juvar.'.session_id()); |
||
340 | //@unlink($this->uploadDir .'/juvar.tmp'.session_id()); |
||
341 | |||
342 | // reset session var |
||
343 | $_SESSION['juvar.tmpsize'] = 0; |
||
344 | |||
345 | $this->isError = true; |
||
346 | $this->error = $msg; |
||
347 | } |
||
348 | |||
349 | /** |
||
350 | * @return bool |
||
351 | */ |
||
352 | public function isError() |
||
353 | { |
||
354 | return $this->isError; |
||
355 | } |
||
356 | |||
357 | /** |
||
358 | * @return string |
||
359 | */ |
||
360 | public function getError() |
||
363 | } |
||
364 | |||
365 | public function getSavedDestination() |
||
366 | { |
||
367 | return $this->savedDestination; |
||
368 | } |
||
369 | |||
370 | public function getSavedFilename() |
||
373 | } |
||
374 | } |
||
375 |