OCA\Files_Antivirus\RequestHelper

Complexity

Total Complexity

14

Size/Duplication

Total Lines	83
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	0

Test Coverage

Coverage

68.97%

Importance

Changes	3
Bugs	0	Features	0

4 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	4	1
A	setSizeForPath()	0	3	1
C	getUploadSize()	0	32	11
A	isScriptName()	0	4	1

<?php
/**
 * ownCloud - files_antivirus
 *
 * This file is licensed under the Affero General Public License version 3 or
 * later. See the COPYING file.
 *
 * @author Viktar Dubiniuk <dubiniuk@owncloud.com>
 *
 * @copyright Viktar Dubiniuk 2017-2018
 * @license AGPL-3.0
 */

namespace OCA\Files_Antivirus;

use OC\Cache\CappedMemoryCache;
use \OCP\IRequest;

/**
 * Used to detect the size of the uploaded file
 *
 * @package OCA\Files_Antivirus
 */
class RequestHelper {
	/**
	 * @var  IRequest
	 */
	private $request;

	/**
	 * @var CappedMemoryCache
	 */
	private $fileSizeCache;

	/**
	 * RequestHelper constructor.
	 *
	 * @param IRequest $request
	 */
	public function __construct(IRequest $request) {
		$this->request = $request;
		$this->fileSizeCache = new CappedMemoryCache();
	}

	/**
	 * @param string $path
	 * @param string $size
	 *
	 * @return void
	 */
	public function setSizeForPath($path, $size) {
		$this->fileSizeCache->set($path, $size);
	}

	/**
	 * Get current upload size
	 * returns null for chunks and when there is no upload
	 *
	 * @param string $path
	 *
	 * @return int|null
	 */
	public function getUploadSize($path) {
		$cachedSize = $this->fileSizeCache->get($path);
		if ($cachedSize > 0) {
			return $cachedSize;
		}

		$uploadSize = null;
		$requestMethod = $this->request->getMethod();
		$isRemoteScript = $this->isScriptName('remote.php');
		$isPublicScript = $this->isScriptName('public.php');
		// Are we uploading anything?
		if (\in_array($requestMethod, ['PUT']) && $isRemoteScript) {
			// v1 && v2 Chunks are not scanned
			if ($requestMethod === 'PUT' &&  \strpos($path, 'uploads/') === 0) {
				return null;
			}

			if (\OC_FileChunking::isWebdavChunk() && \strpos($path, 'cache/') === 0) {
				return null;
			}

			if ($requestMethod === 'PUT') {
				$uploadSize = (int)$this->request->getHeader('CONTENT_LENGTH');
			} else {
				$uploadSize = (int)$this->request->getHeader('OC_TOTAL_LENGTH');
			}
		} else if ($requestMethod === 'PUT' && $isPublicScript) {
			$uploadSize = (int)$this->request->getHeader('CONTENT_LENGTH');
		}

		return $uploadSize;
	}

	/**
	 *
	 * @param string $string
	 *
	 * @return bool
	 */
	public function isScriptName($string) {
		$pattern = \sprintf('|/%s|', \preg_quote($string));
		return \preg_match($pattern, $this->request->getScriptName()) === 1;
	}
}