mirror of
https://github.com/getnamingo/registry.git
synced 2025-05-11 01:08:34 +02:00
94 lines
No EOL
3.3 KiB
PHP
94 lines
No EOL
3.3 KiB
PHP
<?php
|
|
|
|
$c = require_once 'config.php';
|
|
require_once 'helpers.php';
|
|
|
|
$logFilePath = '/var/log/namingo/abusemonitor.log';
|
|
$log = setupLogger($logFilePath, 'Abuse_Monitor');
|
|
$log->info('job started.');
|
|
|
|
use Swoole\Coroutine;
|
|
use League\Flysystem\Local\LocalFilesystemAdapter;
|
|
use League\Flysystem\Filesystem;
|
|
use MatthiasMullie\Scrapbook\Adapters\Flysystem as ScrapbookFlysystem;
|
|
use MatthiasMullie\Scrapbook\Psr6\Pool;
|
|
use GuzzleHttp\Client;
|
|
|
|
// Initialize the PDO connection pool
|
|
$pool = new Swoole\Database\PDOPool(
|
|
(new Swoole\Database\PDOConfig())
|
|
->withDriver($c['db_type'])
|
|
->withHost($c['db_host'])
|
|
->withPort($c['db_port'])
|
|
->withDbName($c['db_database'])
|
|
->withUsername($c['db_username'])
|
|
->withPassword($c['db_password'])
|
|
->withCharset('utf8mb4')
|
|
);
|
|
|
|
Swoole\Runtime::enableCoroutine();
|
|
|
|
// Filesystem Cache setup
|
|
$cachePath = __DIR__ . '/cache';
|
|
|
|
// Check if the 'cache' directory exists
|
|
if (!is_dir($cachePath)) {
|
|
// Attempt to create the directory
|
|
if (!mkdir($cachePath, 0755, true)) {
|
|
$log->error("Unable to create cache directory at $cachePath. Please check permissions.");
|
|
throw new Exception("Unable to create cache directory at $cachePath. Please check permissions.");
|
|
}
|
|
}
|
|
|
|
$adapter = new LocalFilesystemAdapter($cachePath, null, LOCK_EX);
|
|
$filesystem = new Filesystem($adapter);
|
|
$cache = new Pool(new ScrapbookFlysystem($filesystem));
|
|
|
|
// Cache key and file URL for URLAbuse data
|
|
$urlAbuseCacheKey = 'urlabuse_blacklist';
|
|
$urlAbuseUrl = 'https://urlabuse.com/public/data/data.txt';
|
|
|
|
// Cache key and URL for URLHaus data
|
|
$urlhausCacheKey = 'urlhaus_blacklist';
|
|
$urlhausUrl = 'https://urlhaus.abuse.ch/downloads/json_recent/';
|
|
|
|
// Creating first coroutine
|
|
Coroutine::create(function () use ($pool, $log, $cache, $urlAbuseCacheKey, $urlAbuseUrl, $urlhausCacheKey, $urlhausUrl) {
|
|
try {
|
|
$pdo = $pool->get();
|
|
$stmt = $pdo->query('SELECT name, clid FROM domain');
|
|
|
|
// Retrieve cached URLAbuse data
|
|
$urlAbuseData = getUrlAbuseData($cache, $urlAbuseCacheKey, $urlAbuseUrl);
|
|
|
|
// Retrieve cached URLHaus data
|
|
$urlhausData = getUrlhausData($cache, $urlhausCacheKey, $urlhausUrl);
|
|
|
|
while ($row = $stmt->fetch(PDO::FETCH_ASSOC)) {
|
|
$domain = $row['name'];
|
|
|
|
// Check URLAbuse
|
|
if (checkUrlAbuse($domain, $urlAbuseData)) {
|
|
processAbuseDetection($pdo, $domain, $row['clid'], 'URLAbuse', 'https://urlabuse.com/public/data/data.txt', $log);
|
|
}
|
|
|
|
// Check URLhaus
|
|
if (checkUrlhaus($domain, $urlhausData)) {
|
|
processAbuseDetection($pdo, $domain, $row['clid'], 'URLHaus', 'https://urlhaus.abuse.ch/downloads/json_recent/', $log);
|
|
}
|
|
|
|
// Check Spamhaus
|
|
if (checkSpamhaus($domain)) {
|
|
processAbuseDetection($pdo, $domain, $row['clid'], 'Spamhaus', 'http://www.spamhaus.org/query/domain/'.$domain, $log);
|
|
}
|
|
}
|
|
$log->info('job finished successfully.');
|
|
} catch (PDOException $e) {
|
|
$log->error('Database error: ' . $e->getMessage());
|
|
} catch (Throwable $e) {
|
|
$log->error('Error: ' . $e->getMessage());
|
|
} finally {
|
|
// Return the connection to the pool
|
|
$pool->put($pdo);
|
|
}
|
|
}); |