OCA\Files_Antivirus\RequestHelper

Complexity

Total Complexity

12

Size/Duplication

Total Lines	84
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	0

Test Coverage

Coverage

72.41%

Importance

Changes	2
Bugs	0	Features	0

4 Methods

Rating	Name	Duplication	Size	Complexity
A	setSizeForPath()	0	3	1
A	isScriptName()	0	4	1
A	__construct()	0	4	1
D	getUploadSize()	0	33	9

<?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;
		}

		$requestMethod = $this->request->getMethod();
		// Are we uploading anything?
		if ($requestMethod !== 'PUT') {
			return null;
		}
		$isRemoteScript = $this->isScriptName('remote.php');
		$isPublicScript = $this->isScriptName('public.php');
		if (!$isRemoteScript && !$isPublicScript) {
			return null;
		}

		if ($isRemoteScript) {
			// v1 && v2 Chunks are not scanned
			if (\strpos($path, 'uploads/') === 0) {
				return null;
			}

			if (\OC_FileChunking::isWebdavChunk()
				&& \strpos($path, 'cache/') === 0
			) {
				return null;
			}
		}
		$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;
	}
}