<?php
class Dnbd3Rpc {
const QUERY_UNREACHABLE = 1;
const QUERY_NOT_200 = 2;
const QUERY_NOT_JSON = 3;
private static function translateServer($server)
{
// Special case - local server
if ($server === '<self>') {
$server = '127.0.0.1:5003';
} elseif (($out = Dnbd3Util::matchAddress($server))) {
if (isset($out['v4'])) {
$server = $out['v4'];
} else {
$server = '[' . $out['v6'] . ']';
}
if (isset($out['port'])) {
$server .= $out['port'];
} else {
$server .= ':5003';
}
}
return $server;
}
/**
* Query given DNBD3 server for status information.
*
* @param string $server server address
* @param bool $stats include general stats
* @param bool $clients include client list
* @param bool $images include image list
* @param bool $diskSpace include disk space stats
* @param bool $config get config
* @param bool $altservers list of alt servers with status
* @return int|array the queried data as an array, or false on error
*/
public static function query($server, $stats, $clients, $images, $diskSpace = false, $config = false, $altservers = false)
{
$server = self::translateServer($server);
$url = 'http://' . $server . '/query?';
if ($stats) {
$url .= 'q=stats&q=version&';
}
if ($clients) {
$url .= 'q=clients&';
}
if ($images) {
$url .= 'q=images&';
}
if ($diskSpace) {
$url .= 'q=space&';
}
if ($config) {
$url .= 'q=config&';
}
if ($altservers) {
$url .= 'q=altservers&';
}
$str = Download::asString($url, 3, $code);
if ($str === false)
return self::QUERY_UNREACHABLE;
if ($code !== 200)
return self::QUERY_NOT_200;
$ret = json_decode($str, true);
if (!is_array($ret))
return self::QUERY_NOT_JSON;
return $ret;
}
public static function getCacheMap($server, $imgId)
{
$server = self::translateServer($server);
$str = Download::asString('http://' . $server . '/cachemap?id=' . $imgId, 3, $code);
if ($str === false)
return self::QUERY_UNREACHABLE;
if ($code !== 200)
return self::QUERY_NOT_200;
return $str;
}
/**
* Get statistics for multiple servers at once.
* @param string[] $servers
* @return array
*/
public static function getStatsMulti(array $servers, int $timeout = 2): array
{
if (empty($servers))
return [];
$active = [];
$mh = curl_multi_init();
curl_multi_setopt($mh, CURLMOPT_MAXCONNECTS, 8);
curl_multi_setopt($mh, CURLMOPT_MAX_HOST_CONNECTIONS, 2);
curl_multi_setopt($mh, CURLMOPT_MAX_TOTAL_CONNECTIONS, 8);
foreach ($servers as $server) {
$url = 'http://' . self::translateServer($server) . '/query?q=stats';
$res = curl_init($url);
if ($res === false) {
error_log("curl_init($url) failed with $res");
continue;
}
curl_setopt_array($res, [
CURLOPT_CONNECTTIMEOUT => $timeout,
CURLOPT_TIMEOUT => $timeout,
CURLOPT_FOLLOWLOCATION => 0,
CURLOPT_ACCEPT_ENCODING => '', // Use everything libcurl supports
CURLOPT_PROTOCOLS => CURLPROTO_HTTP | CURLPROTO_HTTPS,
CURLOPT_RETURNTRANSFER => 1,
]);
$err = curl_multi_add_handle($mh, $res);
if ($err !== 0) {
error_log("curl_multi_add_handle() failed with $err");
curl_close($res);
} else {
$active[(int)$res] = $server;
}
}
// Wait
$running = 1;
$result = [];
$startTime = microtime(true);
for (;;) {
$ret = curl_multi_exec($mh, $running);
while (($info = curl_multi_info_read($mh)) !== false) {
if ($info['msg'] === CURLMSG_DONE) {
if (isset($active[(int)$info['handle']])) {
$server = $active[(int)$info['handle']];
unset($active[(int)$info['handle']]);
if ($info['result'] === CURLE_OK) {
$data = json_decode(curl_multi_getcontent($info['handle']), true);
if (is_array($data)) {
$data['ts'] = microtime(true);
$result[$server] = $data;
}
}
}
curl_multi_remove_handle($mh, $info['handle']);
curl_close($info['handle']);
}
}
$delay = ($startTime + $timeout) - microtime(true);
if ($ret !== CURLM_OK || !$running || $delay <= 0)
break;
$sret = curl_multi_select($mh, $delay);
if ($sret < 0) {
error_log("curl_multi_select returned $sret");
break;
}
}
curl_multi_close($mh);
return $result;
}
}
