快速部署的KMS服务
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.

215 lines
6.1 KiB

#!/usr/bin/env php8
<?php
$VERSION = 'v1.2.3-rc2';
require_once './src/Daemon.php';
require_once './src/Logger.php';
require_once './src/Process.php';
$KMS_PORT = 1688; // kms expose port
$HTTP_PORT = 1689; // http server port
$ENABLE_HTTP = true; // http interface
$NGINX = array( // nginx process
'name' => 'nginx',
'command' => ['/usr/sbin/nginx'],
'pidFile' => '/run/nginx.pid',
);
$PHP_FPM = array( // php-fpm process
'name' => 'php-fpm8',
'command' => ['/usr/sbin/php-fpm8'],
'pidFile' => '/run/php-fpm8.pid',
);
$VLMCSD = array( // vlmcsd process
'name' => 'vlmcsd',
'command' => ['/usr/bin/vlmcsd', '-e', '-p', '/run/vlmcsd.pid'],
'pidFile' => '/run/vlmcsd.pid',
);
$HELP_MSG = "
kms-server ($VERSION)
Project: https://github.com/dnomd343/kms-server.git
--debug Enable debug mode
--kms-port Specify kms listen port
--http-port Specify http listen port
--disable-http Disable http service
--version Show version info
--help Show help message
\n";
function load_nginx_config(int $kms_port, int $http_port): void {
global $VERSION;
$nginx_config = "server {
listen $http_port;
listen [::]:$http_port ipv6only=on;
set_real_ip_from ::/0;
set_real_ip_from 0.0.0.0/0;
real_ip_header X-Real-IP;
real_ip_recursive on;
location /assets {
root /kms-server;
}
location / {
set \$cli_mode false;
include fastcgi_params;
fastcgi_pass unix:/run/php-fpm.sock;
if (\$http_user_agent ~* (curl|wget)) {
set \$cli_mode true;
}
fastcgi_param KMS_PORT $kms_port;
fastcgi_param KMS_CLI \$cli_mode;
fastcgi_param KMS_VER $VERSION;
fastcgi_param SCRIPT_FILENAME /kms-server/src/Route.php;
}\n}\n";
logging::debug("Nginx configure ->\n" . $nginx_config);
$nginx_file = fopen('/etc/nginx/kms.conf', 'w');
fwrite($nginx_file, $nginx_config); // save nginx configure file
fclose($nginx_file);
}
function load_vlmcsd_command(int $kms_port): void { // add port option for vlmcsd
global $VLMCSD;
if ($kms_port != 1688) { // not default kms port
array_push($VLMCSD['command'], '-P', strval($kms_port));
}
}
function get_param(string $field, string $default): string {
global $argv;
if (!in_array($field, $argv)) { // field not exist
return $default;
}
$index = array_search($field, $argv) + 1;
if ($index == sizeof($argv)) { // reach arguments end
return $default;
}
return $argv[$index]; // return next argument
}
function load_params(): void {
global $argv;
if (strtolower(getenv('DEBUG')) === 'true' || in_array('--debug', $argv)) {
logging::$logLevel = logging::DEBUG; // enter debug mode
}
global $ENABLE_HTTP;
if (strtolower(getenv('DISABLE_HTTP')) === 'true' || in_array('--disable-http', $argv)) {
logging::warning('Disable http service');
$ENABLE_HTTP = false; // disable http service
}
global $KMS_PORT;
if (getenv('KMS_PORT')) {
$KMS_PORT = intval(getenv('KMS_PORT'));
logging::debug('Get KMS_PORT from env -> ' . $KMS_PORT);
}
$KMS_PORT = intval(get_param('--kms-port', strval($KMS_PORT)));
if ($KMS_PORT < 1 || $KMS_PORT > 65535) { // 1 ~ 65535
logging::critical('Illegal KMS Port -> ' . $KMS_PORT);
exit;
}
if ($KMS_PORT != 1688) { // not default kms port
logging::warning('KMS Server Port -> ' . $KMS_PORT);
} else {
logging::debug('KMS Server Port -> ' . $KMS_PORT);
}
global $HTTP_PORT;
if (getenv('HTTP_PORT')) {
$HTTP_PORT = intval(getenv('HTTP_PORT'));
logging::debug('Get HTTP_PORT from env -> ' . $HTTP_PORT);
}
$HTTP_PORT = intval(get_param('--http-port', strval($HTTP_PORT)));
if ($HTTP_PORT < 1 || $HTTP_PORT > 65535) { // 1 ~ 65535
logging::critical('Illegal HTTP Port -> ' . $HTTP_PORT);
exit;
}
if ($HTTP_PORT != 1689) { // not default http port
logging::warning('HTTP Server Port -> ' . $HTTP_PORT);
} else {
logging::debug('HTTP Server Port -> ' . $HTTP_PORT);
}
}
function start_process(): void { // start sub processes
global $ENABLE_HTTP;
global $NGINX, $PHP_FPM, $VLMCSD;
if ($ENABLE_HTTP) { // http server process
new Process($NGINX['command']);
logging::info('Start nginx server...OK');
new Process($PHP_FPM['command']);
logging::info('Start php-fpm server...OK');
}
new Process($VLMCSD['command']);
logging::info('Start vlmcsd server...OK');
}
function exit_process(): void { // kill sub processes
global $ENABLE_HTTP;
global $NGINX, $PHP_FPM, $VLMCSD;
if ($ENABLE_HTTP) {
subExit($NGINX, $PHP_FPM, $VLMCSD); // with http service
} else {
subExit($VLMCSD);
}
}
declare(ticks = 1);
pcntl_signal(SIGCHLD, function() { // receive SIGCHLD signal
pcntl_wait($status, WNOHANG); // avoid zombie process
});
pcntl_signal(SIGTERM, function() { // receive SIGTERM signal
logging::info('Get SIGTERM -> exit');
exit_process();
exit;
});
pcntl_signal(SIGINT, function() { // receive SIGINT signal
logging::info('Get SIGINT -> exit');
exit_process();
exit;
});
pcntl_signal(SIGQUIT, function() { // receive SIGQUIT signal
logging::info('Get SIGQUIT -> exit');
exit_process();
exit;
});
if (in_array('-v', $argv) || in_array('--version', $argv)) {
echo "\033[33mkms-server\033[0m => \033[36m$VERSION\033[0m\n"; // show version info
exit;
}
if (in_array('-h', $argv) || in_array('--help', $argv)) {
echo $HELP_MSG; // show help message
exit;
}
logging::info('Loading kms-server (' . $VERSION . ')');
load_params();
load_vlmcsd_command($KMS_PORT);
load_nginx_config($KMS_PORT, $HTTP_PORT);
start_process();
logging::info('Enter daemon process');
while (true) { // start daemon
for ($i = 0; $i < 500; $i++) { // sleep 5s
msDelay(10); // return main loop every 10ms
}
if ($ENABLE_HTTP) { // with http service
daemon($NGINX);
daemon($PHP_FPM);
}
daemon($VLMCSD);
}