notify_push/lib/SelfTest.php

231 строка
9.4 KiB
PHP

<?php
declare(strict_types=1);
/**
* @copyright Copyright (c) 2020 Robin Appelman <robin@icewind.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\NotifyPush;
use OCA\NotifyPush\Queue\IQueue;
use OCA\NotifyPush\Queue\RedisQueue;
use OCP\App\IAppManager;
use OCP\Http\Client\IClientService;
use OCP\IConfig;
use OCP\IDBConnection;
use Symfony\Component\Console\Output\OutputInterface;
class SelfTest {
public const ERROR_OTHER = 1;
public const ERROR_TRUSTED_PROXY = 2;
private $client;
private $cookie;
private $config;
private $queue;
private $connection;
private $appManager;
public function __construct(
IClientService $clientService,
IConfig $config,
IQueue $queue,
IDBConnection $connection,
IAppManager $appManager
) {
$this->client = $clientService->newClient();
$this->cookie = rand(1, pow(2, 30));
$this->queue = $queue;
$this->config = $config;
$this->connection = $connection;
$this->appManager = $appManager;
}
public function test(string $server, OutputInterface $output, bool $ignoreProxyError = false): int {
if ($this->queue instanceof RedisQueue) {
$output->writeln("<info>✓ redis is configured</info>");
} else {
$output->writeln("<error>🗴 redis is not configured</error>");
return self::ERROR_OTHER;
}
if (strpos($server, 'http://') === 0) {
$output->writeln("<comment>🗴 using unencrypted http for push server is strongly discouraged</comment>");
} elseif (strpos($server, 'https://') !== 0) {
$output->writeln("<error>🗴 malformed server url</error>");
return self::ERROR_OTHER;
}
if (strpos($server, 'localhost') !== false) {
$output->writeln("<comment>🗴 push server url is set to localhost, the push server will not be reachable from other machines</comment>");
}
$this->queue->push('notify_test_cookie', $this->cookie);
$this->config->setAppValue('notify_push', 'cookie', (string)$this->cookie);
try {
$retrievedCookie = (int)$this->client->get($server . '/test/cookie', ['nextcloud' => ['allow_local_address' => true], 'verify' => false])->getBody();
} catch (\Exception $e) {
$msg = $e->getMessage();
$output->writeln("<error>🗴 can't connect to push server: $msg</error>");
return self::ERROR_OTHER;
}
if ($this->cookie === $retrievedCookie) {
$output->writeln("<info>✓ push server is receiving redis messages</info>");
} else {
$expected = $this->cookie;
$output->writeln("<error>🗴 push server is not receiving redis messages (received $expected, got $retrievedCookie)</error>");
return self::ERROR_OTHER;
}
// test if the push server can load storage mappings from the db
[$storageId, $count] = $this->getStorageIdForTest();
try {
$retrievedCount = (int)$this->client->get($server . '/test/mapping/' . $storageId, ['nextcloud' => ['allow_local_address' => true], 'verify' => false])->getBody();
} catch (\Exception $e) {
$msg = $e->getMessage();
$output->writeln("<error>🗴 can't connect to push server: $msg</error>");
return self::ERROR_OTHER;
}
if ((int)$count === $retrievedCount) {
$output->writeln("<info>✓ push server can load mount info from database</info>");
} else {
$output->writeln("<error>🗴 push server can't load mount info from database</error>");
return self::ERROR_OTHER;
}
// test if the push server can reach nextcloud by having it request the cookie
try {
$response = $this->client->get($server . '/test/reverse_cookie', ['nextcloud' => ['allow_local_address' => true], 'verify' => false])->getBody();
$retrievedCookie = (int)$response;
if ($this->cookie === $retrievedCookie) {
$output->writeln("<info>✓ push server can connect to the Nextcloud server</info>");
} else {
$output->writeln("<error>🗴 push server can't connect to the Nextcloud server</error>");
if (!is_numeric($response)) {
$output->writeln("<error> $response</error>");
}
return self::ERROR_OTHER;
}
} catch (\Exception $e) {
$msg = $e->getMessage();
$output->writeln("<error>🗴 can't connect to push server: $msg</error>");
return self::ERROR_OTHER;
}
// test that the push server is a trusted proxy
try {
$remote = $this->client->get($server . '/test/remote/1.2.3.4', ['nextcloud' => ['allow_local_address' => true], 'verify' => false])->getBody();
} catch (\Exception $e) {
$msg = $e->getMessage();
$output->writeln("<error>🗴 can't connect to push server: $msg</error>");
return self::ERROR_OTHER;
}
if ($ignoreProxyError || $remote === '1.2.3.4') {
$output->writeln("<info>✓ push server is a trusted proxy</info>");
} else {
$trustedProxies = $this->config->getSystemValue('trusted_proxies', []);
$headers = $this->config->getSystemValue('forwarded_for_headers', ['HTTP_X_FORWARDED_FOR']);
$receivedHeader = $this->queue->getConnection()->get("notify_push_forwarded_header");
$remote = $this->queue->getConnection()->get("notify_push_remote");
if (array_search('HTTP_X_FORWARDED_FOR', $headers) === false) {
$output->writeln("<error>🗴 Nextcloud is configured to not use the `x-http-forwarded-for` header.</error>");
$output->writeln("<error> Please add 'HTTP_X_FORWARDED_FOR' the the 'forwarded_for_headers' in your config.php.</error>");
return self::ERROR_TRUSTED_PROXY;
}
$output->writeln("<error>🗴 push server is not a trusted proxy, please add '$remote' to the list of trusted proxies" .
" or configure any existing reverse proxy to forward the 'x-forwarded-for' send by the push server.</error>");
$output->writeln(" See https://docs.nextcloud.com/server/latest/admin_manual/configuration_server/reverse_proxy_configuration.html#defining-trusted-proxies for how to set trusted proxies.");
$output->writeln(" The following trusted proxies are currently configured: " . implode(', ', array_map(function (string $proxy) {
return '"' . $proxy . '"';
}, $trustedProxies)));
$invalidConfig = array_filter($trustedProxies, function (string $proxy) {
return !$this->isValidProxyConfig($proxy);
});
if ($invalidConfig) {
$output->writeln("<error> of which the following seem to be invalid: " . implode(', ', array_map(function (string $proxy) {
return '"' . $proxy . '"';
}, $invalidConfig)) . "</error>");
}
$output->writeln(" The following x-forwarded-for header was received by Nextcloud: $receivedHeader");
$output->writeln(" from the following remote: $remote");
$output->writeln("");
$output->writeln(" If you're having issues getting the trusted proxy setup working, you can try bypassing any existing reverse proxy");
$output->writeln(" in your setup by setting the `NEXTCLOUD_URL` environment variable to point directly to the internal Nextcloud webserver url");
$output->writeln(" (You will still need the ip address of the push server added as trusted proxy)");
return self::ERROR_TRUSTED_PROXY;
}
// test that the binary is up to date
try {
$this->queue->getConnection()->del("notify_push_version");
$response = $this->client->post($server . '/test/version', ['nextcloud' => ['allow_local_address' => true], 'verify' => false]);
if ($response === "error") {
$output->writeln("<error>🗴 failed to get binary version, check the push server output for more information</error>");
return self::ERROR_OTHER;
}
usleep(10 * 1000);
$binaryVersion = $this->queue->getConnection()->get("notify_push_version");
} catch (\Exception $e) {
$msg = $e->getMessage();
$output->writeln("<error>🗴 failed to get binary version: $msg</error>");
return self::ERROR_OTHER;
}
$appVersion = $this->appManager->getAppVersion('notify_push');
$appVersionNoMinor = substr($appVersion, 0, strrpos($appVersion, '.'));
$binaryVersionNoMinor = substr($binaryVersion, 0, strrpos($binaryVersion, '.'));
if ($appVersionNoMinor === $binaryVersionNoMinor) {
$output->writeln("<info>✓ push server is running the same version as the app</info>");
} else {
$output->writeln("<error>🗴 push server (version $binaryVersion) is not the same version as the app (version $appVersion).</error>");
return self::ERROR_OTHER;
}
return 0;
}
private function getStorageIdForTest() {
$query = $this->connection->getQueryBuilder();
$query->select('storage_id', $query->func()->count())
->from('mounts', 'm')
->innerJoin('m', 'filecache', 'f', $query->expr()->eq('root_id', 'fileid'))
->where($query->expr()->eq('path_hash', $query->createNamedParameter(md5(''))))
->groupBy('storage_id')
->setMaxResults(1);
return $query->execute()->fetch(\PDO::FETCH_NUM);
}
private function isValidProxyConfig(string $proxyConfig): bool {
$cidrre = '/^([0-9]{1,3}\.[0-9]{1,3}\.[0-9]{1,3}\.[0-9]{1,3})\/([0-9]{1,2})$/';
if (filter_var($proxyConfig, FILTER_VALIDATE_IP) !== false) {
return true;
} else {
return (bool)preg_match($cidrre, $proxyConfig);
}
}
}