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.
256 lines
10 KiB
256 lines
10 KiB
<?php
|
|
|
|
class kmsKeys { // KMS密钥获取
|
|
private $db, $kmsDB = './db/kmsKeys.db'; // KMS密钥数据库
|
|
|
|
private function getVersionName($type, $versionId) { // 获取对应版本的名称
|
|
$res = $this->db->query('SELECT * FROM `' . $type . '_version` WHERE version_id=' . $versionId . ';');
|
|
return $res->fetchArray(SQLITE3_ASSOC)['version_name'];
|
|
}
|
|
|
|
private function getKmsKeys($type) { // 获取指定类型的密钥信息 win/win-server
|
|
$this->db = new SqliteDB($this->kmsDB);
|
|
$res = $this->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 '': // win和win-server系列
|
|
return $this->getKmsKeys('win') + $this->getKmsKeys('win-server');
|
|
case 'win': // win系列
|
|
return $this->getKmsKeys('win');
|
|
case 'win-server': // win-server系列
|
|
return $this->getKmsKeys('win-server');
|
|
default:
|
|
return array(); // 未知类型 返回空数组
|
|
}
|
|
}
|
|
}
|
|
|
|
class kmsCheck { // KMS服务器检查
|
|
private function getKmsStatus($host, $port) { // 获取KMS服务器状态
|
|
if(filter_var($host, FILTER_VALIDATE_IP, FILTER_FLAG_IPV6)) {
|
|
$host = '[' . $host . ']'; // IPv6地址需用中括号包围
|
|
}
|
|
$cmd = $GLOBALS['env']['KMS_VLMCS'] . ' ';
|
|
$cmd .= $host . ':' . $port; // 加入目标服务器信息
|
|
$raw = shell_exec($cmd . ' -G temp'); // 执行vlmcs测试
|
|
preg_match_all('/Sending activation request \(KMS V6\)/', $raw, $match);
|
|
return (count($match[0]) === 6) ? true : false; // 返回KMS服务器状态
|
|
}
|
|
|
|
public function isCache($server) { // 检查KMS服务器是否已缓存
|
|
$redis = new RedisCache('kms');
|
|
$status = $redis->getData($server);
|
|
return (!$status) ? false : true;
|
|
}
|
|
|
|
public function kmsStatus($host, $port) { // 检测KMS服务器状态
|
|
$redis = new RedisCache('kms');
|
|
$server = $host . ':' . $port;
|
|
$status = $redis->getData($server);
|
|
if (!$status) { // 缓存未命中
|
|
if ($this->getKmsStatus($host, $port)) { // 测试服务器状态
|
|
$status = [ 'status' => 'online' ]; // 服务正常
|
|
} else {
|
|
$status = [ 'status' => 'offline' ]; // 服务掉线
|
|
}
|
|
$redis->setData($server, json_encode($status), 300); // 缓存5min
|
|
} else { // 缓存命中
|
|
$status = json_decode($status, true); // 使用缓存数据
|
|
}
|
|
return $status;
|
|
}
|
|
}
|
|
|
|
class kmsCheckEntry { // KMS功能入口
|
|
private function formatCheck($server) { // 输入参数格式检查
|
|
$temp = explode(':', $server);
|
|
if (count($temp) === 1) { // 不带:的请求
|
|
if (!(new Domain)->isHost($temp[0])) { return null; } // 错误请求
|
|
return array(
|
|
'host' => $temp[0],
|
|
'port' => 1688
|
|
);
|
|
} else if (count($temp) === 2) { // 带一个:的请求
|
|
if (!(new Domain)->isHost($temp[0])) { return null; } // 错误请求
|
|
if ($temp[1] < 0 || $temp[1] > 65535) { return null; } // 错误请求
|
|
return array(
|
|
'host' => $temp[0],
|
|
'port' => $temp[1]
|
|
);
|
|
} else { // 带多个:的请求
|
|
return null; // 错误请求
|
|
}
|
|
}
|
|
|
|
private function genMessage($server, $status) { // 生成KMS状态消息
|
|
$msg = '`' . $server . '`' . PHP_EOL;
|
|
if ($status['status'] === 'online') {
|
|
return $msg . 'KMS服务*正常运行*'; // 服务正常
|
|
} else {
|
|
return $msg . 'KMS服务*无法使用*'; // 服务异常
|
|
}
|
|
}
|
|
|
|
private function sendKmsStatus($host, $port) { // 检查并发送KMS服务器状态
|
|
$server = $host . ':' . $port;
|
|
if ((new kmsCheck)->isCache($server)) { // 状态已缓存
|
|
$status = (new kmsCheck)->kmsStatus($host, $port);
|
|
tgApi::sendMarkdown($this->genMessage($server, $status)); // 发送服务器状态
|
|
return;
|
|
}
|
|
$message = tgApi::sendMarkdown('`' . $server . '`' . PHP_EOL . 'KMS服务检测中...');
|
|
$messageId = json_decode($message, true)['result']['message_id']; // 未缓存 发送缓冲消息
|
|
$status = (new kmsCheck)->kmsStatus($host, $port); // 发起查询
|
|
tgApi::editMessage(array( // 返回查询结果
|
|
'parse_mode' => 'Markdown',
|
|
'message_id' => $messageId,
|
|
'text' => $this->genMessage($server, $status)
|
|
));
|
|
}
|
|
|
|
private function simpStr($str) { // 简化版本名称
|
|
preg_match_all('/[0-9a-zA-Z]/', $str, $match);
|
|
return implode('', $match[0]);
|
|
}
|
|
|
|
private function genKmsKeys($targetVersion) { // 显示指定版本的KMS密钥列表
|
|
$content = explode('|', $targetVersion);
|
|
$type = $content[0]; // 获取类型 win/win-server
|
|
$targetVersion = $content[1]; // 获取目标版本
|
|
$kmsKeys = (new kmsKeys)->getKeys($type); // 获取该类型的所有版本
|
|
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(
|
|
'text' => $msg,
|
|
'parse_mode' => 'Markdown',
|
|
'reply_markup' => json_encode(array( // 返回按钮
|
|
'inline_keyboard' => array([[
|
|
'text' => '<< Go back <<',
|
|
'callback_data' => '/kms ' . $type
|
|
]])
|
|
))
|
|
);
|
|
}
|
|
|
|
private function genKmsVersions($type) { // 获取版本列表 win/win-server
|
|
$kmsKeys = (new kmsKeys)->getKeys($type);
|
|
foreach ($kmsKeys as $version => $kmsKey) {
|
|
$buttons[] = array([ // 生成按钮列表
|
|
'text' => $version,
|
|
'callback_data' => '/kms ' . $type . '|' . $this->simpStr($version)
|
|
]);
|
|
}
|
|
$buttons[] = array([
|
|
'text' => '<< Go back <<',
|
|
'callback_data' => '/kms menu' // 加入返回按钮
|
|
]);
|
|
return array(
|
|
'text' => 'Which version did you need?',
|
|
'reply_markup' => json_encode(array( // 显示列表按钮
|
|
'inline_keyboard' => $buttons
|
|
))
|
|
);
|
|
}
|
|
|
|
private function genActiveCmd() { // 生成KMS激活命令
|
|
$kmsHost = $GLOBALS['env']['KMS_HOST'];
|
|
$actiCmd = '```' . PHP_EOL . 'slmgr /upk' . PHP_EOL . 'slmgr /ipk {KMS_KEY}' . PHP_EOL;
|
|
$actiCmd .= 'slmgr /skms ' . $kmsHost . PHP_EOL . 'slmgr /ato' . PHP_EOL . 'slmgr /dlv';
|
|
$actiCmd .= PHP_EOL . '```';
|
|
return array(
|
|
'text' => $actiCmd,
|
|
'parse_mode' => 'Markdown',
|
|
'reply_markup' => json_encode(array( // 返回按钮
|
|
'inline_keyboard' => array([[
|
|
'text' => '<< Go back <<',
|
|
'callback_data' => '/kms menu'
|
|
]])
|
|
))
|
|
);
|
|
}
|
|
|
|
private function genSelectMsg() { // 生成KMS版本选择消息
|
|
return array(
|
|
'text' => 'Which one did you need?',
|
|
'reply_markup' => json_encode(array(
|
|
'inline_keyboard' => array( // 功能选择按钮
|
|
array([
|
|
'text' => 'Windows',
|
|
'callback_data' => '/kms win' // Windows密钥
|
|
]),
|
|
array([
|
|
'text' => 'Windows Server',
|
|
'callback_data' => '/kms win-server' // Windows Server密钥
|
|
]),
|
|
array([
|
|
'text' => 'Activation Command',
|
|
'callback_data' => '/kms cmd' // 激活命令
|
|
])
|
|
)
|
|
))
|
|
);
|
|
}
|
|
|
|
private function sendHelp() { // 发送使用说明
|
|
$helpMessage = 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 menu'
|
|
]])
|
|
))
|
|
);
|
|
tgApi::sendMessage($helpMessage);
|
|
}
|
|
|
|
public function query($rawParam) { // KMS测试查询入口
|
|
if ($rawParam == '' || $rawParam === 'help') { // 显示使用说明
|
|
$this->sendHelp();
|
|
return;
|
|
}
|
|
$server = $this->formatCheck($rawParam);
|
|
if ($server === null) {
|
|
tgApi::sendText('Illegal request'); // 输入格式错误
|
|
return;
|
|
}
|
|
$this->sendKmsStatus($server['host'], $server['port']); // 检查并发送KMS服务器状态
|
|
}
|
|
|
|
public function callback($rawParam) { // KMS测试回调入口
|
|
$messageId = $GLOBALS['tgEnv']['messageId'];
|
|
switch ($rawParam) {
|
|
case 'menu': // 选择菜单
|
|
$message = $this->genSelectMsg();
|
|
break;
|
|
case 'cmd': // KMS激活命令
|
|
$message = $this->genActiveCmd();
|
|
break;
|
|
case 'win': // Windows激活密钥
|
|
case 'win-server': // Windows Server激活密钥
|
|
$message = $this->genKmsVersions($rawParam);
|
|
break;
|
|
default:
|
|
$message = $this->genKmsKeys($rawParam); // 显示密钥列表
|
|
}
|
|
tgApi::editMessage(array(
|
|
'message_id' => $messageId // 修改源消息内容
|
|
) + $message);
|
|
}
|
|
}
|
|
|
|
?>
|
|
|