mirror of https://github.com/dnomd343/tgbot
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
266 lines
11 KiB
266 lines
11 KiB
<?php
|
|
|
|
class ntpServer { // 获取NTP服务器列表
|
|
private $db, $ntpDB = './db/ntpServer.db'; // NTP服务器数据库
|
|
|
|
private function getListName($listId) { // 获取对应组的名称
|
|
$res = $this->db->query('SELECT * FROM `ntp_list` WHERE id=' . $listId . ';');
|
|
return $res->fetchArray(SQLITE3_ASSOC)['name'];
|
|
}
|
|
|
|
public function getList() { // 获取所有NTP服务器地址
|
|
$this->db = new SqliteDB($this->ntpDB);
|
|
$res = $this->db->query('SELECT * FROM `ntp_host`;');
|
|
while ($row = $res->fetchArray(SQLITE3_ASSOC)) {
|
|
$index = $row['list_id'];
|
|
unset($row['list_id']);
|
|
$list[$this->getListName($index)][] = $row;
|
|
}
|
|
return $list;
|
|
}
|
|
}
|
|
|
|
class ntpCheck { // NTP服务器检查
|
|
private function formatOffset($str) { // 格式化偏移时间
|
|
$num = number_format($str, 6) * 1000; // s -> ms
|
|
$str = sprintf("%1\$.3f", $num); // 补零到小数点后3位
|
|
if ($num > 0) {
|
|
$str = '+' . $str; // 正数前加+
|
|
}
|
|
return $str . 'ms';
|
|
}
|
|
|
|
private function sortByIp($servers) { // 排序算法
|
|
$temp = array();
|
|
foreach ($servers as $val){
|
|
$temp[] = $val['Server'];
|
|
}
|
|
sort($temp);
|
|
$temp = array_flip($temp);
|
|
$sort = array();
|
|
foreach ($servers as $val) {
|
|
$temp_1 = $val['Server'];
|
|
$temp_2 = $temp[$temp_1];
|
|
$sort[$temp_2] = $val;
|
|
}
|
|
asort($sort);
|
|
return $sort;
|
|
}
|
|
|
|
private function sortServer($servers) { // 按顺序排列服务器
|
|
foreach ($servers as $server) {
|
|
if(filter_var($server['Server'], FILTER_VALIDATE_IP, FILTER_FLAG_IPV4)) {
|
|
$ipv4[] = $server; // 提取IPv4服务器
|
|
}
|
|
if(filter_var($server['Server'], FILTER_VALIDATE_IP, FILTER_FLAG_IPV6)) {
|
|
$ipv6[] = $server; // 提取IPv6服务器
|
|
}
|
|
}
|
|
if (isset($ipv4)) { // 存在IPv4服务器
|
|
foreach ($ipv4 as $index => $ip) {
|
|
$ipv4[$index]['Server'] = ip2long($ip['Server']); // IPv4预处理
|
|
}
|
|
$ipv4 = $this->sortByIp($ipv4); // 排序IPv4服务器
|
|
foreach ($ipv4 as $index => $ip) {
|
|
$ip['Server'] = long2ip($ip['Server']); // IPv4恢复
|
|
$result[] = $ip;
|
|
}
|
|
}
|
|
if (isset($ipv6)) { // 存在IPv6服务器
|
|
foreach ($ipv6 as $index => $ip) {
|
|
$ipv6[$index]['Server'] = (new DNS)->ip2long6($ip['Server']); // IPv6预处理
|
|
}
|
|
$ipv6 = $this->sortByIp($ipv6); // 排序IPv6服务器
|
|
foreach ($ipv6 as $index => $ip) {
|
|
$ip['Server'] = (new DNS)->long2ip6($ip['Server']); // IPv6恢复
|
|
$result[] = $ip;
|
|
}
|
|
}
|
|
return (!isset($result)) ? array() : $result; // 无结果 返回空数组
|
|
}
|
|
|
|
private function getNtpStatus($host) { // 获取NTP服务器状态
|
|
$html = (new Curl)->post('https://servertest.online/ntp', array(
|
|
'a' => $host,
|
|
'c' => 'Query+both'
|
|
));
|
|
if ($html == '') { return null; } // 服务错误
|
|
preg_match('/<\/form>[\s\S]+<footer>/', $html, $match); // 切取数据部分
|
|
preg_match('/<label>[\s\S]+<footer>/', $match[0], $match); // 去除前部干扰
|
|
$match = substr($match[0], 0, strlen($match[0]) - 8); // 去除后部<footer>
|
|
$match = strtr($match, array('<h2>IPv6 test results</h2>' => '<br><br>')); // 去除中间干扰
|
|
$match = explode('<br>', $match); // 以<br>为界限切割
|
|
$match[] = ''; // 添加空记录方便分组
|
|
foreach ($match as $row) { // 匹配分组
|
|
if ($row == '') {
|
|
if (count($temp) === 5) {
|
|
$record[] = $temp;
|
|
$temp = array();
|
|
}
|
|
continue;
|
|
}
|
|
$temp[] = $row;
|
|
}
|
|
foreach ($record as $group) { // 检查组内容是否正常
|
|
$group['ok'] = '';
|
|
foreach ($group as $index => $row) { // 遍历组内每一行
|
|
if ($index === 'ok') { continue; }
|
|
$row = strtr($row, array('<label>' => ''));
|
|
$row = strtr($row, array(':</label>' => '|'));
|
|
$content = explode('|', $row); // 切取参数
|
|
if (count($content) !== 2) { unset($group['ok']); } // 格式不合格
|
|
$group[$content[0]] = $content[1];
|
|
unset($group[$index]); // 删除原始记录
|
|
}
|
|
if (!isset($group['ok'])) { continue; } // 剔除不合格组
|
|
if ($group['Result'] !== '<b class=com>OK</b>') { continue; }
|
|
if (!filter_var($group['Server'], FILTER_VALIDATE_IP)) { continue; }
|
|
unset($group['Result']);
|
|
unset($group['ok']);
|
|
$group['Offset'] = $this->formatOffset($group['Offset']); // 转换为ms
|
|
$data[] = $group;
|
|
}
|
|
$data = (!isset($data)) ? array() : $data; // 无结果时为空数组
|
|
return $this->sortServer($data); // 排序后返回
|
|
}
|
|
|
|
public function isCache($host) { // 检查NTP服务器是否已缓存
|
|
$redis = new RedisCache('ntp');
|
|
$servers = $redis->getData($host);
|
|
return (!$servers) ? false : true;
|
|
}
|
|
|
|
public function ntpStatus($host) { // 检测NTP服务器状态
|
|
$redis = new RedisCache('ntp');
|
|
$servers = $redis->getData($host); // 查询缓存数据
|
|
if (!$servers) { // 缓存未命中
|
|
$servers = $this->getNtpStatus($host); // 发起测试
|
|
if ($servers === null) { return null; } // 服务接口错误
|
|
$redis->setData($host, json_encode($servers), 900); // 缓存15min
|
|
} else { // 缓存命中
|
|
$servers = json_decode($servers, true); // 使用缓存数据
|
|
}
|
|
return $servers;
|
|
}
|
|
}
|
|
|
|
class ntpCheckEntry { // NTP功能入口
|
|
private function hashGroupName($str) { // 计算组名的哈希值 取前12位
|
|
return substr(md5($str), 0, 12);
|
|
}
|
|
|
|
private function showNtpList() { // 列出所有NTP服务器组
|
|
$ntpList = (new ntpServer)->getList();
|
|
foreach ($ntpList as $index => $ntpHosts) {
|
|
$buttons[] = array([ // 生成按钮列表
|
|
'text' => $index,
|
|
'callback_data' => '/ntp ' . $this->hashGroupName($index)
|
|
]);
|
|
}
|
|
return array(
|
|
'text' => 'Which one did you like?',
|
|
'reply_markup' => json_encode(array( // 按钮列表
|
|
'inline_keyboard' => $buttons
|
|
))
|
|
);
|
|
}
|
|
|
|
private function showNtpServer($targetGroup) { // 列出指定组的NTP服务器
|
|
$ntpList = (new ntpServer)->getList();
|
|
foreach ($ntpList as $index => $ntpHosts) { // 搜索目标组
|
|
if ($this->hashGroupName($index) === $targetGroup) { break; }
|
|
}
|
|
$msg = '*' . $index . '*' . PHP_EOL; // 显示组名
|
|
foreach ($ntpHosts as $ntpHost) {
|
|
if ($ntpHost['desc'] !== '') {
|
|
$msg .= $ntpHost['desc'] . '：'; // 服务器描述
|
|
}
|
|
$msg .= '`' . $ntpHost['host'] . '`' . PHP_EOL; // 服务器地址
|
|
}
|
|
return array(
|
|
'text' => $msg,
|
|
'parse_mode' => 'Markdown',
|
|
'reply_markup' => json_encode(array(
|
|
'inline_keyboard' => array([[ // 返回上一级 按钮
|
|
'text' => ' << Go back <<',
|
|
'callback_data' => '/ntp servers'
|
|
]])
|
|
))
|
|
);
|
|
}
|
|
|
|
private function genMessage($host, $servers) { // 生成返回信息
|
|
if (count($servers) === 0) { // NTP服务不可用
|
|
$msg = '`' . $host . '`' . PHP_EOL;
|
|
$msg .= 'NTP Server *Offline*' . PHP_EOL . PHP_EOL;
|
|
return $msg;
|
|
}
|
|
$msg = '`' . $host . '`' . PHP_EOL;
|
|
$msg .= 'NTP Server *Normally*' . PHP_EOL . PHP_EOL; // NTP服务器正常
|
|
foreach ($servers as $server) { // 显示所有服务器
|
|
$msg .= '`' . $server['Server'] . '`' . PHP_EOL;
|
|
$msg .= '_Stratum:_ ' . $server['Stratum'] . PHP_EOL;
|
|
$msg .= '_Offset:_ ' . $server['Offset'] . PHP_EOL;
|
|
$msg .= PHP_EOL;
|
|
}
|
|
return $msg;
|
|
}
|
|
|
|
private function sendNtpStatus($host) { // 检查并发送NTP服务器状态
|
|
if ((new ntpCheck)->isCache($host)) { // 状态已缓存
|
|
$servers = (new ntpCheck)->ntpStatus($host);
|
|
tgApi::sendMarkdown($this->genMessage($host, $servers)); // 发送服务器状态
|
|
return;
|
|
}
|
|
$message = tgApi::sendMarkdown('`' . $host . '`' . PHP_EOL . 'NTP Server Checking...');
|
|
$messageId = json_decode($message, true)['result']['message_id']; // 未缓存 发送缓冲消息
|
|
$servers = (new ntpCheck)->ntpStatus($host); // 发起查询
|
|
if ($servers === null) { // 上游接口错误
|
|
$message = 'Server error';
|
|
} else {
|
|
$message = $this->genMessage($host, $servers); // 生成返回信息
|
|
}
|
|
tgApi::editMessage(array( // 返回查询结果
|
|
'text' => $message,
|
|
'parse_mode' => 'Markdown',
|
|
'message_id' => $messageId
|
|
));
|
|
}
|
|
|
|
private function sendHelp() { // 显示使用说明
|
|
$helpMessage = array(
|
|
'parse_mode' => 'Markdown',
|
|
'text' => '*Usage:* `/ntp IP/Domain`',
|
|
'reply_markup' => json_encode(array( // 显示 NTP服务器列表 按钮
|
|
'inline_keyboard' => array([[
|
|
'text' => 'Get NTP Servers',
|
|
'callback_data' => '/ntp servers'
|
|
]])
|
|
))
|
|
);
|
|
tgApi::sendMessage($helpMessage); // 发送使用说明
|
|
}
|
|
|
|
public function query($rawParam) { // NTP测试查询入口
|
|
if ($rawParam == '' || $rawParam === 'help') {
|
|
$this->sendHelp(); // 显示使用说明
|
|
} else if (!(new Domain)->isHost($rawParam)) {
|
|
tgApi::sendText('Illegal host'); // 输入错误
|
|
} else {
|
|
$this->sendNtpStatus($rawParam); // 检查并发送NTP服务器状态
|
|
}
|
|
}
|
|
|
|
public function callback($rawParam) { // NTP测试回调入口
|
|
if ($rawParam === 'servers') {
|
|
$content = $this->showNtpList(); // 显示可选组
|
|
} else {
|
|
$content = $this->showNtpServer($rawParam); // 显示指定组的服务器列表
|
|
}
|
|
tgApi::editMessage(array(
|
|
'message_id' => $GLOBALS['tgEnv']['messageId']
|
|
) + $content); // 输出结果
|
|
}
|
|
}
|
|
|
|
?>
|
|
|