Drop UpdateGServer worker task if domain is blocked

This commit is contained in:
Hypolite Petovan 2023-01-04 11:39:22 -05:00
parent a907d6c87b
commit 1f3c07c06f
3 changed files with 44 additions and 10 deletions

View File

@ -44,6 +44,7 @@ use Friendica\Util\Network;
use Friendica\Util\Strings; use Friendica\Util\Strings;
use Friendica\Util\XML; use Friendica\Util\XML;
use Friendica\Network\HTTPException; use Friendica\Network\HTTPException;
use Friendica\Worker\UpdateGServer;
use GuzzleHttp\Psr7\Uri; use GuzzleHttp\Psr7\Uri;
use Psr\Http\Message\UriInterface; use Psr\Http\Message\UriInterface;
@ -100,11 +101,11 @@ class GServer
*/ */
public static function add(string $url, bool $only_nodeinfo = false) public static function add(string $url, bool $only_nodeinfo = false)
{ {
if (self::getID($url, false)) { if (self::getID($url)) {
return; return;
} }
Worker::add(Worker::PRIORITY_LOW, 'UpdateGServer', $url, $only_nodeinfo); UpdateGServer::add(Worker::PRIORITY_LOW, $url, $only_nodeinfo);
} }
/** /**
@ -192,8 +193,9 @@ class GServer
return false; return false;
} else { } else {
if (strtotime($gserver['next_contact']) < time()) { if (strtotime($gserver['next_contact']) < time()) {
Worker::add(Worker::PRIORITY_LOW, 'UpdateGServer', $gserver['url'], false); UpdateGServer::add(Worker::PRIORITY_LOW, $gserver['url']);
} }
return self::isDefunct($gserver); return self::isDefunct($gserver);
} }
} }
@ -211,8 +213,9 @@ class GServer
return true; return true;
} else { } else {
if (strtotime($gserver['next_contact']) < time()) { if (strtotime($gserver['next_contact']) < time()) {
Worker::add(Worker::PRIORITY_LOW, 'UpdateGServer', $gserver['url'], false); UpdateGServer::add(Worker::PRIORITY_LOW, $gserver['url']);
} }
return !$gserver['failed'] && in_array($gserver['network'], Protocol::FEDERATED); return !$gserver['failed'] && in_array($gserver['network'], Protocol::FEDERATED);
} }
} }
@ -253,7 +256,7 @@ class GServer
} }
if (!empty($server) && (empty($gserver) || strtotime($gserver['next_contact']) < time())) { if (!empty($server) && (empty($gserver) || strtotime($gserver['next_contact']) < time())) {
Worker::add(Worker::PRIORITY_LOW, 'UpdateGServer', $server, false); UpdateGServer::add(Worker::PRIORITY_LOW, $server);
} }
return $reachable; return $reachable;
@ -376,7 +379,7 @@ class GServer
Logger::info('Reset failed status for server', ['url' => $gserver['url']]); Logger::info('Reset failed status for server', ['url' => $gserver['url']]);
if (strtotime($gserver['next_contact']) < time()) { if (strtotime($gserver['next_contact']) < time()) {
Worker::add(Worker::PRIORITY_LOW, 'UpdateGServer', $gserver['url'], false); UpdateGServer::add(Worker::PRIORITY_LOW, $gserver['url']);
} }
} }
} }
@ -394,7 +397,7 @@ class GServer
Logger::info('Set failed status for server', ['url' => $gserver['url']]); Logger::info('Set failed status for server', ['url' => $gserver['url']]);
if (strtotime($gserver['next_contact']) < time()) { if (strtotime($gserver['next_contact']) < time()) {
Worker::add(Worker::PRIORITY_LOW, 'UpdateGServer', $gserver['url'], false); UpdateGServer::add(Worker::PRIORITY_LOW, $gserver['url']);
} }
} }
} }

View File

@ -22,9 +22,14 @@
namespace Friendica\Worker; namespace Friendica\Worker;
use Friendica\Core\Logger; use Friendica\Core\Logger;
use Friendica\Core\Worker;
use Friendica\Database\DBA; use Friendica\Database\DBA;
use Friendica\Model\GServer; use Friendica\Model\GServer;
use Friendica\Network\HTTPException\InternalServerErrorException;
use Friendica\Util\Network;
use Friendica\Util\Strings; use Friendica\Util\Strings;
use GuzzleHttp\Psr7\Uri;
use Psr\Http\Message\UriInterface;
class UpdateGServer class UpdateGServer
{ {
@ -34,8 +39,9 @@ class UpdateGServer
* @param string $server_url Server URL * @param string $server_url Server URL
* @param boolean $only_nodeinfo Only use nodeinfo for server detection * @param boolean $only_nodeinfo Only use nodeinfo for server detection
* @return void * @return void
* @throws \Exception
*/ */
public static function execute(string $server_url, bool $only_nodeinfo = false) public static function execute(string $server_url, bool $only_nodeinfo)
{ {
if (empty($server_url)) { if (empty($server_url)) {
return; return;
@ -47,6 +53,11 @@ class UpdateGServer
return; return;
} }
// Silently dropping the worker task if the server domain is blocked
if (Network::isUrlBlocked($filtered)) {
return;
}
if (($filtered != $server_url) && DBA::exists('gserver', ['nurl' => Strings::normaliseLink($server_url)])) { if (($filtered != $server_url) && DBA::exists('gserver', ['nurl' => Strings::normaliseLink($server_url)])) {
GServer::setFailureByUrl($server_url); GServer::setFailureByUrl($server_url);
return; return;
@ -61,4 +72,23 @@ class UpdateGServer
$ret = GServer::check($filtered, '', true, $only_nodeinfo); $ret = GServer::check($filtered, '', true, $only_nodeinfo);
Logger::info('Updated gserver', ['url' => $filtered, 'result' => $ret]); Logger::info('Updated gserver', ['url' => $filtered, 'result' => $ret]);
} }
/**
* @param array|int $run_parameters Priority constant or array of options described in Worker::add
* @param string $serverUrl
* @param bool $onlyNodeInfo Only use NodeInfo for server detection
* @return int
* @throws InternalServerErrorException
*/
public static function add($run_parameters, string $serverUrl, bool $onlyNodeInfo = false): int
{
// Dropping the worker task if the server domain is blocked
if (Network::isUrlBlocked($serverUrl)) {
return 0;
}
// We have to convert the Uri back to string because worker parameters are saved in JSON format which
// doesn't allow for structured objects.
return Worker::add($run_parameters, 'UpdateGServer', $serverUrl, $onlyNodeInfo);
}
} }

View File

@ -27,6 +27,7 @@ use Friendica\Database\DBA;
use Friendica\DI; use Friendica\DI;
use Friendica\Util\DateTimeFormat; use Friendica\Util\DateTimeFormat;
use Friendica\Util\Strings; use Friendica\Util\Strings;
use GuzzleHttp\Psr7\Uri;
class UpdateGServers class UpdateGServers
{ {
@ -63,12 +64,12 @@ class UpdateGServers
// There are duplicated "url" but not "nurl". So we check both addresses instead of just overwriting them, // There are duplicated "url" but not "nurl". So we check both addresses instead of just overwriting them,
// since that would mean loosing data. // since that would mean loosing data.
if (!empty($gserver['url'])) { if (!empty($gserver['url'])) {
if (Worker::add(Worker::PRIORITY_LOW, 'UpdateGServer', $gserver['url'])) { if (UpdateGServer::add(Worker::PRIORITY_LOW, $gserver['url'])) {
$count++; $count++;
} }
} }
if (!empty($gserver['nurl']) && ($gserver['nurl'] != Strings::normaliseLink($gserver['url']))) { if (!empty($gserver['nurl']) && ($gserver['nurl'] != Strings::normaliseLink($gserver['url']))) {
if (Worker::add(Worker::PRIORITY_LOW, 'UpdateGServer', $gserver['nurl'])) { if (UpdateGServer::add(Worker::PRIORITY_LOW, $gserver['nurl'])) {
$count++; $count++;
} }
} }