parent
603d7dab5e
commit
2bbb848c31
@ -0,0 +1,140 @@ |
||||
<?php |
||||
|
||||
declare(strict_types=1); |
||||
/** |
||||
* @copyright Copyright (c) 2020, Roeland Jago Douma <roeland@famdouma.nl> |
||||
* |
||||
* @author Roeland Jago Douma <roeland@famdouma.nl> |
||||
* |
||||
* @license GNU AGPL version 3 or any later version |
||||
* |
||||
* This program is free software: you can redistribute it and/or modify |
||||
* it under the terms of the GNU Affero General Public License as |
||||
* published by the Free Software Foundation, either version 3 of the |
||||
* License, or (at your option) any later version. |
||||
* |
||||
* This program is distributed in the hope that it will be useful, |
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
||||
* GNU Affero General Public License for more details. |
||||
* |
||||
* You should have received a copy of the GNU Affero General Public License |
||||
* along with this program. If not, see <http://www.gnu.org/licenses/>. |
||||
* |
||||
*/ |
||||
|
||||
namespace OCA\Encryption\Command; |
||||
|
||||
use OC\Files\View; |
||||
use OCA\Encryption\Util; |
||||
use OCP\IConfig; |
||||
use OCP\IUserManager; |
||||
use Symfony\Component\Console\Command\Command; |
||||
use Symfony\Component\Console\Helper\QuestionHelper; |
||||
use Symfony\Component\Console\Input\InputInterface; |
||||
use Symfony\Component\Console\Output\OutputInterface; |
||||
|
||||
class ScanLegacyFormat extends Command { |
||||
|
||||
/** @var Util */ |
||||
protected $util; |
||||
|
||||
/** @var IConfig */ |
||||
protected $config; |
||||
|
||||
/** @var QuestionHelper */ |
||||
protected $questionHelper; |
||||
|
||||
/** @var IUserManager */ |
||||
private $userManager; |
||||
|
||||
/** @var View */ |
||||
private $rootView; |
||||
|
||||
/** |
||||
* @param Util $util |
||||
* @param IConfig $config |
||||
* @param QuestionHelper $questionHelper |
||||
*/ |
||||
public function __construct(Util $util, |
||||
IConfig $config, |
||||
QuestionHelper $questionHelper, |
||||
IUserManager $userManager) { |
||||
parent::__construct(); |
||||
|
||||
$this->util = $util; |
||||
$this->config = $config; |
||||
$this->questionHelper = $questionHelper; |
||||
$this->userManager = $userManager; |
||||
$this->rootView = new View(); |
||||
} |
||||
|
||||
protected function configure() { |
||||
$this |
||||
->setName('encryption:scan:legacy-format') |
||||
->setDescription('Scan the files for the legacy format'); |
||||
} |
||||
|
||||
protected function execute(InputInterface $input, OutputInterface $output): int { |
||||
$result = true; |
||||
|
||||
$output->writeln('Scanning all files for legacy encryption'); |
||||
|
||||
foreach ($this->userManager->getBackends() as $backend) { |
||||
$limit = 500; |
||||
$offset = 0; |
||||
do { |
||||
$users = $backend->getUsers('', $limit, $offset); |
||||
foreach ($users as $user) { |
||||
$output->writeln('Scanning all files for ' . $user); |
||||
$this->setupUserFS($user); |
||||
$result &= $this->scanFolder($output, '/' . $user); |
||||
} |
||||
$offset += $limit; |
||||
} while (count($users) >= $limit); |
||||
} |
||||
|
||||
if ($result) { |
||||
$output->writeln('All scanned files are propperly encrypted. You can disable the legacy compatibility mode.'); |
||||
return 0; |
||||
} |
||||
|
||||
return 1; |
||||
} |
||||
|
||||
private function scanFolder(OutputInterface $output, string $folder): bool { |
||||
$clean = true; |
||||
|
||||
foreach ($this->rootView->getDirectoryContent($folder) as $item) { |
||||
$path = $folder . '/' . $item['name']; |
||||
if ($this->rootView->is_dir($path)) { |
||||
if ($this->scanFolder($output, $path) === false) { |
||||
$clean = false; |
||||
} |
||||
} else { |
||||
if (!$item->isEncrypted()) { |
||||
// ignore |
||||
continue; |
||||
} |
||||
|
||||
$stats = $this->rootView->stat($path); |
||||
if (!isset($stats['hasHeader']) || $stats['hasHeader'] === false) { |
||||
$clean = false; |
||||
$output->writeln($path . ' does not have a proper header'); |
||||
} |
||||
} |
||||
} |
||||
|
||||
return $clean; |
||||
} |
||||
|
||||
/** |
||||
* setup user file system |
||||
* |
||||
* @param string $uid |
||||
*/ |
||||
protected function setupUserFS($uid) { |
||||
\OC_Util::tearDownFS(); |
||||
\OC_Util::setupFS($uid); |
||||
} |
||||
} |
||||
Loading…
Reference in new issue