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.
272 lines
9.6 KiB
272 lines
9.6 KiB
<?php
|
|
|
|
class kmsDB extends SQLite3 {
|
|
function __construct() {
|
|
$this->open('./db/kmsKeys.db'); // KMS密钥数据库
|
|
}
|
|
}
|
|
|
|
class kmsKeys {
|
|
private function getVersionName($type, $version_id) { // 获取对应版本的名称
|
|
$db = new kmsDB;
|
|
$res = $db->query('SELECT * FROM `' . $type . '_version` WHERE version_id=' . $version_id . ';');
|
|
return $res->fetchArray(SQLITE3_ASSOC)['version_name'];
|
|
}
|
|
|
|
private function getKmsKeys($type) { // 获取所有版本的KMS密钥
|
|
$db = new kmsDB;
|
|
$res = $db->query('SELECT * FROM `' . $type . '`;');
|
|
while ($row = $res->fetchArray(SQLITE3_ASSOC)) {
|
|
$index = $row['version'];
|
|
unset($row['version']);
|
|
$data[$this->getVersionName($type, $index)][] = $row;
|
|
}
|
|
return $data;
|
|
}
|
|
|
|
public function getKeys($type) { // 获取指定类型KMS密钥
|
|
switch ($type) {
|
|
case '':
|
|
return $this->getKmsKeys('win') + $this->getKmsKeys('win-server');
|
|
case 'win':
|
|
return $this->getKmsKeys('win');
|
|
case 'win-server':
|
|
return $this->getKmsKeys('win-server');
|
|
default:
|
|
return array();
|
|
}
|
|
}
|
|
}
|
|
|
|
class kmsCheck {
|
|
private $api = 'https://kms.343.re/';
|
|
|
|
public function kmsStatus($host, $port) {
|
|
$server = $host . ':' . $port;
|
|
$redis = new redisCache('kms');
|
|
$info = $redis->getData($server); // 查询缓存数据
|
|
if (!$info) { // 缓存未命中
|
|
$url = $this->api . 'check?host=' . $host . '&port=' . $port;
|
|
$info = json_decode(file_get_contents($url), true); // 请求上游接口
|
|
$info['server'] = $server;
|
|
switch ($info['status']) {
|
|
case 'ok':
|
|
$info['online'] = true;
|
|
break;
|
|
case 'error':
|
|
$info['online'] = false;
|
|
break;
|
|
default:
|
|
return array(
|
|
'status' => 'error',
|
|
'message' => 'Server error'
|
|
);
|
|
}
|
|
$info['status'] = 'ok';
|
|
unset($info['message']);
|
|
$redis->setData($server, json_encode($info), 300); // 缓存5min
|
|
} else { // 缓存命中
|
|
$info = json_decode($info, true); // 使用缓存数据
|
|
}
|
|
return $info;
|
|
}
|
|
}
|
|
|
|
class kmsCheckEntry {
|
|
private function isHost($host) { // 判断host是否合法
|
|
preg_match('/^(?=^.{3,255}$)[a-zA-Z0-9][-a-zA-Z0-9]{0,62}(\.[a-zA-Z0-9][-a-zA-Z0-9]{0,62})+$/', $host, $match);
|
|
if (count($match) !== 0) { // 域名
|
|
if (!is_numeric(substr($host, -1))) { return true; } // 域名最后一位不为数字
|
|
}
|
|
if (filter_var($host, \FILTER_VALIDATE_IP, \FILTER_FLAG_IPV4)) { // IPv4地址
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
private function formatCheck($server) {
|
|
$temp = explode(':', $server);
|
|
if (count($temp) === 1) { // 不带:的请求
|
|
if ($this->isHost($temp[0])) {
|
|
$host = $server;
|
|
$port = 1688;
|
|
} else {
|
|
return array(
|
|
'status' => 'error',
|
|
'message' => 'Illegal host'
|
|
);
|
|
}
|
|
} else if (count($temp) === 2) { // 带一个:的请求
|
|
if ($this->isHost($temp[0])) {
|
|
$host = $temp[0];
|
|
} else {
|
|
return array(
|
|
'status' => 'error',
|
|
'message' => 'Illegal host'
|
|
);
|
|
}
|
|
$port = $temp[1];
|
|
if ($port < 0 || $port > 65535) {
|
|
return array(
|
|
'status' => 'error',
|
|
'message' => 'Illegal port'
|
|
);
|
|
}
|
|
} else { // 带多个:的请求
|
|
return array(
|
|
'status' => 'error',
|
|
'message' => 'Illegal request'
|
|
);
|
|
}
|
|
return array(
|
|
'status' => 'ok',
|
|
'host' => $host,
|
|
'port' => $port
|
|
);
|
|
}
|
|
|
|
private function simpStr($str) { // 简化版本名称
|
|
preg_match_all('/[0-9a-zA-Z]/', $str, $match);
|
|
return implode('', $match[0]);
|
|
}
|
|
|
|
private function getKmsVersions($type) { // 获取win或win-server的版本列表
|
|
$kmsKeys = (new kmsKeys)->getKeys($type);
|
|
foreach ($kmsKeys as $version => $kmsKey) {
|
|
$buttons[] = array([ // 生成按钮列表
|
|
'text' => $version,
|
|
'callback_data' => '/kms ' . $this->simpStr($version)
|
|
]);
|
|
}
|
|
$buttons[] = array([
|
|
'text' => '<< Go back <<',
|
|
'callback_data' => '/kms keys'
|
|
]);
|
|
return array(
|
|
'text' => 'Which version did you need?',
|
|
'reply_markup' => json_encode(array( // 显示列表按钮
|
|
'inline_keyboard' => $buttons
|
|
))
|
|
);
|
|
}
|
|
|
|
private function getKmsKeys($targetVersion) { // 显示指定版本的KMS密钥列表
|
|
$kmsKeys = (new kmsKeys)->getKeys('');
|
|
foreach ($kmsKeys as $version => $kmsKey) { // 比对压缩以后的名称
|
|
if ($this->simpStr($version) === $targetVersion) { break; } // 匹配成功
|
|
}
|
|
$msg = '*' . $version . ' KMS Keys*' . PHP_EOL . PHP_EOL;
|
|
foreach ($kmsKey as $row) {
|
|
$msg .= $row['name'] . '：`' . $row['key'] . '`' . PHP_EOL . PHP_EOL;
|
|
}
|
|
return array(
|
|
'parse_mode' => 'Markdown',
|
|
'text' => $msg
|
|
);
|
|
}
|
|
|
|
private function checkKms($host, $port) { // 检查KMS服务器状态
|
|
$content = (new kmsCheck)->kmsStatus($host, $port);
|
|
if ($content['status'] === 'ok') {
|
|
if ($content['online'] === true) {
|
|
return array(
|
|
'parse_mode' => 'Markdown',
|
|
'text' => '`' . $content['server'] . '`' . PHP_EOL . 'KMS服务*正常运行*'
|
|
);
|
|
} else {
|
|
return array(
|
|
'parse_mode' => 'Markdown',
|
|
'text' => '`' . $content['server'] . '`' . PHP_EOL . 'KMS服务*无法使用*'
|
|
);
|
|
}
|
|
} else {
|
|
return array(
|
|
'text' => $content['message']
|
|
);
|
|
}
|
|
}
|
|
|
|
public function query($rawParam) { // kmsCheck查询入口
|
|
if ($rawParam == '' || $rawParam === 'help') { // 显示使用说明
|
|
tgApi::sendMessage(array(
|
|
'parse_mode' => 'Markdown',
|
|
'text' => '*Usage:* `/kms IP/Domain[:port]`',
|
|
'reply_markup' => json_encode(array( // 获取KMS密钥按钮
|
|
'inline_keyboard' => array([[
|
|
'text' => 'Get KMS Keys',
|
|
'callback_data' => '/kms keys'
|
|
]])
|
|
))
|
|
));
|
|
return;
|
|
}
|
|
$check = $this->formatCheck($rawParam);
|
|
if ($check['status'] === 'error') { // 输入格式有误
|
|
tgApi::sendText($check['message']);
|
|
return;
|
|
}
|
|
$message = tgApi::sendMarkdown('`' . $rawParam . '`' . PHP_EOL . 'KMS服务检测中...');
|
|
$message = json_decode($message, true);
|
|
fastcgi_finish_request(); // 断开连接
|
|
tgApi::editMessage(array(
|
|
'message_id' => $message['result']['message_id'],
|
|
) + $this->checkKms($check['host'], $check['port'])); // 发起查询并返回结果
|
|
}
|
|
|
|
public function callback($rawParam) { // kmsCheck回调入口
|
|
global $tgEnv;
|
|
$selectMsg = array(
|
|
'text' => 'Which one did you need?',
|
|
'reply_markup' => json_encode(array(
|
|
'inline_keyboard' => array(
|
|
array([
|
|
'text' => 'Windows',
|
|
'callback_data' => '/kms win'
|
|
]),
|
|
array([
|
|
'text' => 'Windows Server',
|
|
'callback_data' => '/kms win-server'
|
|
]),
|
|
array([
|
|
'text' => 'Activation Command',
|
|
'callback_data' => '/kms cmd'
|
|
])
|
|
)
|
|
))
|
|
);
|
|
$actiCmd = '```' . PHP_EOL . 'slmgr /upk' . PHP_EOL . 'slmgr /ipk {KMS_KEY}' . PHP_EOL;
|
|
$actiCmd .= 'slmgr /skms {KMS_HOST}' . PHP_EOL . 'slmgr /ato' . PHP_EOL . 'slmgr /dlv';
|
|
$actiCmd .= PHP_EOL . '```';
|
|
switch ($rawParam) {
|
|
case 'keys':
|
|
tgApi::editMessage(array(
|
|
'message_id' => $tgEnv['messageId'],
|
|
) + $selectMsg);
|
|
return;
|
|
case 'cmd':
|
|
tgApi::editMessage(array(
|
|
'message_id' => $tgEnv['messageId'],
|
|
'parse_mode' => 'Markdown',
|
|
'text' => $actiCmd,
|
|
'reply_markup' => json_encode(array(
|
|
'inline_keyboard' => array([[
|
|
'text' => '<< Go back <<',
|
|
'callback_data' => '/kms keys'
|
|
]])
|
|
))
|
|
));
|
|
return;
|
|
case 'win':
|
|
case 'win-server':
|
|
tgApi::editMessage(array(
|
|
'message_id' => $tgEnv['messageId'],
|
|
) + $this->getKmsVersions($rawParam));
|
|
return;
|
|
}
|
|
tgApi::editMessage(array(
|
|
'message_id' => $tgEnv['messageId']
|
|
) + $this->getKmsKeys($rawParam));
|
|
}
|
|
}
|
|
|
|
?>
|
|
|