Page MenuHomeWickedGov Phorge
Files F5976761

DnsSrvDiscoverer.php
No OneTemporary
Actions

Size
2 KB
Referenced Files
None
Subscribers
None

DnsSrvDiscoverer.php
View Options

<?php
/**
* Service discovery using DNS SRV records
*
* @license GPL-2.0-or-later
* @file
*/
/**
* @since 1.29
*/
class DnsSrvDiscoverer {
/**
* @var string
*/
private $service;
/**
* @var string
*/
private $protocol;
/**
* @var string|null
*/
private $domain;
/**
* @var callable
*/
private $resolver;
/**
* Construct a new discoverer for the given domain, service, and protocol.
*
* @param string $service Name of the service to discover.
* @param string $protocol Service protocol. Defaults to 'tcp'
* @param ?string $domain The hostname/domain on which to perform discovery
* of the given service and protocol. Defaults to null which effectively
* performs a query relative to the host's configured search domain.
* @param ?callable $resolver Resolver function. Defaults to using
* dns_get_record. Primarily useful in testing.
*/
public function __construct(
string $service,
string $protocol = 'tcp',
?string $domain = null,
?callable $resolver = null
) {
$this->service = $service;
$this->protocol = $protocol;
$this->domain = $domain;
$this->resolver = $resolver ?? static function ( $srv ) {
return dns_get_record( $srv, DNS_SRV );
};
}
/**
* Queries the resolver for an SRV resource record matching the service,
* protocol, and domain and returns all target/port/priority/weight
* records.
*
* @return array
*/
public function getRecords() {
$result = [];
$records = ( $this->resolver )( $this->getSrvName() );
// Respect RFC 2782 with regard to a single '.' entry denoting a valid
// empty response
if (
!$records
|| ( count( $records ) === 1 && $records[0]['target'] === '.' )
) {
return $result;
}
foreach ( $records as $record ) {
$result[] = [
'target' => $record['target'],
'port' => (int)$record['port'],
'pri' => (int)$record['pri'],
'weight' => (int)$record['weight'],
];
}
return $result;
}
/**
* Performs discovery for the domain, service, and protocol, and returns a
* list of resolved server name/ip and port number pairs sorted by each
* record's priority, with servers of the same priority randomly shuffled.
*
* @return array
*/
public function getServers() {
$records = $this->getRecords();
usort( $records, static function ( $a, $b ) {
if ( $a['pri'] === $b['pri'] ) {
return mt_rand( 0, 1 ) ? 1 : -1;
}
return $a['pri'] - $b['pri'];
} );
$serversAndPorts = [];
foreach ( $records as $record ) {
$serversAndPorts[] = [ $record['target'], $record['port'] ];
}
return $serversAndPorts;
}
/**
* Returns the SRV resource record name.
*/
public function getSrvName(): string {
$srv = "_{$this->service}._{$this->protocol}";
if ( $this->domain === null || $this->domain === '' ) {
return $srv;
}
return "$srv.{$this->domain}";
}
}

File Metadata

Mime Type
text/x-php
Expires
Sat, Oct 3, 19:24 (4 d, 3 h ago)
Storage Engine
local-disk
Storage Format
Raw Data
Storage Handle
0e/82/68e5a0e813cd7d2ca9829dcc2360
Default Alt Text
DnsSrvDiscoverer.php (2 KB)

Event Timeline