435 lines
15 KiB
C
435 lines
15 KiB
C
|
#ifndef CONFIGURATION_H
|
|||
|
|
#define CONFIGURATION_H
|
|||
|
|
#include <CApplication.h>
|
|||
|
|
#include <CCFile.h>
|
|||
|
|
#include <CCHttpClient.h>
|
|||
|
|
#include <CCString.h>
|
|||
|
|
#include <QNetworkInterface>
|
|||
|
|
#include <CCThread.h>
|
|||
|
|
#include <QMessageBox>
|
|||
|
|
#include <QProcess>
|
|||
|
|
|
|||
|
|
#include "Tools/CText.h"
|
|||
|
|
#include "CCSocket.h"
|
|||
|
|
#include "CCProcess.h"
|
|||
|
|
|
|||
|
|
#define SubControl_Jar "SubControl/WebSub/SpringWeb.jar"
|
|||
|
|
#define SubControl_Back_Cfg "SubControl/src/main/resources/ServerSettings/ServerSettings.CText"
|
|||
|
|
#define SubControl_Web_Cfg "SubControl/conf/nginx.conf"
|
|||
|
|
#define Sub_Logo "SubControl/html/assets/logo-CuXNbvLG.png"
|
|||
|
|
#define Sub_ImgBuffer "ImgBuffer/"
|
|||
|
|
|
|||
|
|
#ifdef _WIN32
|
|||
|
|
#include <tchar.h>
|
|||
|
|
#include <windows.h>
|
|||
|
|
#include <tlhelp32.h>
|
|||
|
|
#include <shlobj.h> // 包含 ShellLink 和 IPersistFile 接口
|
|||
|
|
#define SubControl_Java "SubControl/WebSub/jreWin/bin/Sub_Server.exe"
|
|||
|
|
#define SubControl_Web_Exe "SubControl/Sub_Web.exe"
|
|||
|
|
#define IPBS_NSSM_Exe "IPBS_NSSM.exe"
|
|||
|
|
#elif __linux__
|
|||
|
|
#define SubControl_Java "SubControl/WebSub/jreWin/bin/Sub_Server"
|
|||
|
|
#define SubControl_Web_Exe "SubControl/Sub_WebL"
|
|||
|
|
#endif
|
|||
|
|
|
|||
|
|
class Configuration
|
|||
|
|
{
|
|||
|
|
private:
|
|||
|
|
inline static std::mutex _mutex;
|
|||
|
|
public:
|
|||
|
|
Configuration() = default;
|
|||
|
|
inline static CCString ExePath = CApplication::GetTheProgramDirectory() + "/";
|
|||
|
|
inline static CCString jarPath = CCFile::GetnormalizePath(ExePath + SubControl_Jar);
|
|||
|
|
inline static CCString JavaPath = CCFile::GetnormalizePath(ExePath + SubControl_Java);
|
|||
|
|
inline static CCString CfgPath = ExePath + SubControl_Back_Cfg;
|
|||
|
|
inline static CCString WebCfgPath = ExePath + SubControl_Web_Cfg;
|
|||
|
|
inline static CCString WebPath = CCFile::GetnormalizePath(ExePath + SubControl_Web_Exe);
|
|||
|
|
inline static CCString IPBS_NSSMPath = CCFile::GetnormalizePath(ExePath + IPBS_NSSM_Exe);
|
|||
|
|
inline static CCString CS_IP = "";
|
|||
|
|
inline static CCString CS_Port = "";
|
|||
|
|
inline static CCString Web_IP = "";
|
|||
|
|
inline static CCString Web_Port = "";
|
|||
|
|
inline static CCString Sub_IP = CCSocket::GetLocalIP()[0];
|
|||
|
|
inline static CCString Sub_Port = "";
|
|||
|
|
inline static CCString Sub_Audio_Port = "";
|
|||
|
|
inline static CCString Sub_Org = "";
|
|||
|
|
inline static CCString Language = "";
|
|||
|
|
inline static CCString State = "Close";
|
|||
|
|
inline static int OpenSleep = 20;
|
|||
|
|
inline static CCString CORSEn = "false";
|
|||
|
|
inline static int FirstRun = 1;
|
|||
|
|
inline static CCString Web_URL = "";
|
|||
|
|
inline static std::vector<CCInt> CSPortList;
|
|||
|
|
inline static std::vector<CCString> CSIPList;
|
|||
|
|
inline static std::vector<CCString> SubIPList;
|
|||
|
|
inline static CCProcess WebRun,SubRun;
|
|||
|
|
inline static int PostMax = 5417;
|
|||
|
|
public:
|
|||
|
|
static int GetInt(char h1, char h2)
|
|||
|
|
{
|
|||
|
|
int lowUnsigned = h1 & 0xFF;
|
|||
|
|
int highUnsigned = h2 & 0xFF;
|
|||
|
|
return (highUnsigned << 8) | lowUnsigned;
|
|||
|
|
}
|
|||
|
|
static void ReadConfigInit(){
|
|||
|
|
CCVar A = CText::ReadConfig(CfgPath.c_str());
|
|||
|
|
if(!A.empty()){
|
|||
|
|
CS_IP = A["C/S_IP"];
|
|||
|
|
CS_Port = A["C/S_PORT"];
|
|||
|
|
Web_IP = A["Sub_Control_IP"];
|
|||
|
|
Web_Port = A["Sub_Web_PORT"];
|
|||
|
|
Sub_IP = Web_IP;
|
|||
|
|
Sub_Port = A["Sub_Control_PORT"];
|
|||
|
|
Sub_Audio_Port = A["Sub_Control_Audio_PORT"];
|
|||
|
|
Sub_Org = A["Origins"];
|
|||
|
|
Language = A["Language"];
|
|||
|
|
OpenSleep = A["OpenSleep"].to_int();
|
|||
|
|
State = A["State"];
|
|||
|
|
CORSEn = A["CORSEn"];
|
|||
|
|
FirstRun = A["FirstRun"].to_int();
|
|||
|
|
}
|
|||
|
|
Configuration::Web_URL = "http://" + Web_IP + ":" + Web_Port;
|
|||
|
|
}
|
|||
|
|
static void ReadCSInfo(const std::function<void()>& fun = nullptr,QWidget *parent = nullptr){
|
|||
|
|
CCUniqueLock lock(_mutex);
|
|||
|
|
CSPortList.clear();
|
|||
|
|
CSIPList.clear();
|
|||
|
|
CCSocket UDP;
|
|||
|
|
UDP.Socket(IPVX::IPV4,TORU::UDP,TYPE::DGRAM);
|
|||
|
|
bool A = UDP.SetSockOpt(CCOpt::BROADCAST);
|
|||
|
|
const CCString IP = Sub_IP;
|
|||
|
|
if(bool F = UDP.Bind(IP.c_str(),0)){
|
|||
|
|
constexpr char str[] = {0x55,static_cast<char>(0xAA),0x48,0x00,0x00,0x00,0x00,0x00};
|
|||
|
|
sockaddr_in serverAddr{};
|
|||
|
|
serverAddr.sin_family = AF_INET;
|
|||
|
|
serverAddr.sin_port = htons(10060);
|
|||
|
|
serverAddr.sin_addr.s_addr = INADDR_BROADCAST;
|
|||
|
|
UDP.SendData(str,serverAddr);
|
|||
|
|
int AS = 0;
|
|||
|
|
while (true){
|
|||
|
|
char buf[1024] = {0};
|
|||
|
|
CCHostInfo info;
|
|||
|
|
if(UDP.isDataAvailable()){
|
|||
|
|
CCVar length = UDP.UDPRecvData(buf, 1024,&info);
|
|||
|
|
if(length > 0){
|
|||
|
|
if(buf[2] == 0x48){
|
|||
|
|
int P = GetInt(buf[4],buf[5]);
|
|||
|
|
CSIPList.emplace_back(info.IPAddress);
|
|||
|
|
CSPortList.emplace_back(P);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
else{
|
|||
|
|
if(length == -1)
|
|||
|
|
{
|
|||
|
|
break;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
else{
|
|||
|
|
AS++;
|
|||
|
|
CCThread::Sleep(1000 * 10);
|
|||
|
|
if(AS == 10){
|
|||
|
|
break;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
UDP.Close();
|
|||
|
|
if(fun){
|
|||
|
|
fun();
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
else{
|
|||
|
|
// const CCString IPA = "网络初始化失败 -> " + Sub_IP;
|
|||
|
|
// QMessageBox::information(parent,"错误",IPA.c_str());
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
static void ReadSub_IP(){
|
|||
|
|
SubIPList.clear();
|
|||
|
|
foreach (QHostAddress ptr , QNetworkInterface::allAddresses()) {
|
|||
|
|
// 获取ipv4地址
|
|||
|
|
if (ptr.protocol() == QAbstractSocket::IPv4Protocol) {
|
|||
|
|
// 过滤本地回环127.0.0.1
|
|||
|
|
if (!ptr.isLoopback()) {
|
|||
|
|
CCString A = ptr.toString().toStdString();
|
|||
|
|
SubIPList.emplace_back(A);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
static void CfgInit(){
|
|||
|
|
while (true){
|
|||
|
|
CCThread::Sleep(1000 * 500);
|
|||
|
|
ReadSub_IP();
|
|||
|
|
if(!SubIPList.empty()){
|
|||
|
|
Sub_IP = SubIPList[0];
|
|||
|
|
break;
|
|||
|
|
}
|
|||
|
|
else{
|
|||
|
|
QMessageBox::StandardButton reply = QMessageBox::question(nullptr, "提示",
|
|||
|
|
"未检测到合法IPV4地址，请检查网络,是否继续检测?",
|
|||
|
|
QMessageBox::Yes | QMessageBox::No);
|
|||
|
|
if (reply == QMessageBox::Yes) {
|
|||
|
|
continue;
|
|||
|
|
} else {
|
|||
|
|
Sub_IP = "";
|
|||
|
|
break;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
while (true){
|
|||
|
|
CCThread::Sleep(1000 * 500);
|
|||
|
|
ReadCSInfo();
|
|||
|
|
if(!CSIPList.empty()){
|
|||
|
|
CS_IP = CSIPList[0];
|
|||
|
|
CS_Port = CSPortList[0].to_String();
|
|||
|
|
break;
|
|||
|
|
}
|
|||
|
|
else{
|
|||
|
|
CS_IP = "";
|
|||
|
|
CS_Port = "";
|
|||
|
|
QMessageBox::StandardButton reply = QMessageBox::question(nullptr, "提示",
|
|||
|
|
"在局域网未检测到可用的C/S服务器，请先启动C/S服务器,是否继续检测?",
|
|||
|
|
QMessageBox::Yes | QMessageBox::No);
|
|||
|
|
if (reply == QMessageBox::Yes) {
|
|||
|
|
continue;
|
|||
|
|
} else {
|
|||
|
|
CS_IP = "";
|
|||
|
|
CS_Port = "";
|
|||
|
|
break;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
if(PostMax == Sub_Port.to_int()){
|
|||
|
|
PostMax = PostMax + 3;
|
|||
|
|
}
|
|||
|
|
CCVar Ports = getAvailablePorts(3,PostMax);
|
|||
|
|
Sub_Port = Ports[0].to_String();
|
|||
|
|
Sub_Audio_Port = Ports[1].to_String();
|
|||
|
|
Web_Port = Ports[2].to_String();
|
|||
|
|
Web_IP = Sub_IP;
|
|||
|
|
Sub_Org = "['http://" + Web_IP + ":" + Web_Port + "']";
|
|||
|
|
}
|
|||
|
|
static bool isPortAvailable(int port,TORU ipvx = TCP)
|
|||
|
|
{
|
|||
|
|
bool F = false;
|
|||
|
|
if(ipvx == TCP){
|
|||
|
|
CCSocket sock;
|
|||
|
|
sock.Socket(IPV4,TCP,TYPE::STREAM);
|
|||
|
|
F = sock.Bind(Sub_IP.c_str(), port);
|
|||
|
|
try{
|
|||
|
|
sock.Close();
|
|||
|
|
}
|
|||
|
|
catch (CCException& e){}
|
|||
|
|
}
|
|||
|
|
else{
|
|||
|
|
CCSocket sock;
|
|||
|
|
sock.Socket(IPV4,UDP,TYPE::DGRAM);
|
|||
|
|
F = sock.Bind(Sub_IP.c_str(), port);
|
|||
|
|
try{
|
|||
|
|
sock.Close();
|
|||
|
|
}
|
|||
|
|
catch (CCException& e){}
|
|||
|
|
}
|
|||
|
|
return F;
|
|||
|
|
}
|
|||
|
|
static std::vector<CCInt> getAvailablePorts(int count, int startPort = 1024, int endPort = 65535)
|
|||
|
|
{
|
|||
|
|
std::vector<CCInt> availablePorts;
|
|||
|
|
for (int port = startPort; port <= endPort && availablePorts.size() < count; ++port) {
|
|||
|
|
if(availablePorts.size() < count - 1){
|
|||
|
|
if (isPortAvailable(port)) {
|
|||
|
|
availablePorts.push_back(port);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
else{
|
|||
|
|
if(isPortAvailable(port)){
|
|||
|
|
availablePorts.push_back(port);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
return availablePorts;
|
|||
|
|
}
|
|||
|
|
static void OpenSub_Server(){
|
|||
|
|
CCString cmd = "cd " + ExePath + "SubControl/" + " && ";
|
|||
|
|
SubRun.AddCommand(cmd);
|
|||
|
|
// cmd = "start "" " + JavaPath + " -jar " + jarPath;
|
|||
|
|
cmd = JavaPath + " -jar " + jarPath;
|
|||
|
|
SubRun.AddCommand(cmd);
|
|||
|
|
SubRun.Start();
|
|||
|
|
}
|
|||
|
|
static void CloseSub_Server(){
|
|||
|
|
// CCProcess Close;
|
|||
|
|
// Close.AddCommand("taskkill /f /t /im Sub_Server.exe");
|
|||
|
|
// Close.Start();
|
|||
|
|
QProcess::startDetached("taskkill /f /t /im Sub_Server.exe");
|
|||
|
|
}
|
|||
|
|
static void OpenSub_Web(){
|
|||
|
|
CCString cmd = "cd " + ExePath + "SubControl/" + " && ";
|
|||
|
|
WebRun.AddCommand(cmd);
|
|||
|
|
cmd = WebPath;
|
|||
|
|
WebRun.AddCommand(cmd);
|
|||
|
|
WebRun.Start();
|
|||
|
|
Configuration::Web_URL = "http://" + Web_IP + ":" + Web_Port + "/";
|
|||
|
|
}
|
|||
|
|
static void CloseSub_Web(){
|
|||
|
|
// CCProcess Close;
|
|||
|
|
// Close.AddCommand("taskkill /f /t /im Sub_Web.exe");
|
|||
|
|
// Close.Start();
|
|||
|
|
QProcess::startDetached("taskkill /f /t /im Sub_Web.exe");
|
|||
|
|
}
|
|||
|
|
static bool GetSub_Status(){
|
|||
|
|
bool F = false;
|
|||
|
|
CCProcess P;
|
|||
|
|
#ifdef _WIN32
|
|||
|
|
P.AddCommand("tasklist | findstr Sub_Server.exe");
|
|||
|
|
#elif __linux__
|
|||
|
|
#endif
|
|||
|
|
P.Start(true);
|
|||
|
|
while (true){
|
|||
|
|
CCString str = P.ReadLineBuffer();
|
|||
|
|
if(str.empty()){
|
|||
|
|
break;
|
|||
|
|
}
|
|||
|
|
else{
|
|||
|
|
F = str.find("Sub_Server") != CCString::npos;
|
|||
|
|
if(F){
|
|||
|
|
break;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
P.Stop();
|
|||
|
|
return F;
|
|||
|
|
}
|
|||
|
|
static bool GetSub_Web_Status(){
|
|||
|
|
bool F = false;
|
|||
|
|
CCProcess P;
|
|||
|
|
#ifdef _WIN32
|
|||
|
|
P.AddCommand("tasklist | findstr Sub_Web.exe");
|
|||
|
|
#elif __linux__
|
|||
|
|
#endif
|
|||
|
|
P.Start(true);
|
|||
|
|
while (true){
|
|||
|
|
CCString str = P.ReadLineBuffer();
|
|||
|
|
if(str.empty()){
|
|||
|
|
break;
|
|||
|
|
}
|
|||
|
|
else{
|
|||
|
|
F = str.find("Sub_Web") != CCString::npos;
|
|||
|
|
if(F){
|
|||
|
|
break;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
P.Stop();
|
|||
|
|
return F;
|
|||
|
|
}
|
|||
|
|
static bool SaveJarCfg() {
|
|||
|
|
CCString File = CCFile::GetnormalizePath(CfgPath);
|
|||
|
|
FILE *fp = fopen(File.c_str(), "w");
|
|||
|
|
if(fp == nullptr){
|
|||
|
|
return false;
|
|||
|
|
}
|
|||
|
|
fprintf(fp, "C/S_IP:%s\n",CS_IP.c_str());
|
|||
|
|
fprintf(fp, "C/S_PORT:%s\n",CS_Port.c_str());
|
|||
|
|
fprintf(fp, "Sub_Control_IP:%s\n",Sub_IP.c_str());
|
|||
|
|
fprintf(fp, "Sub_Control_PORT:%s\n",Sub_Port.c_str());
|
|||
|
|
fprintf(fp, "Sub_Control_Audio_PORT:%s\n",Sub_Audio_Port.c_str());
|
|||
|
|
fprintf(fp, "Sub_Web_PORT:%s\n",Web_Port.c_str());
|
|||
|
|
fprintf(fp, "Origins:%s\n",Sub_Org.c_str());
|
|||
|
|
fprintf(fp, "#中文简体,English\n");
|
|||
|
|
fprintf(fp, "Language:%s\n",Language.c_str());
|
|||
|
|
fprintf(fp, "OpenSleep:%d\n",OpenSleep);
|
|||
|
|
fprintf(fp, "State:%s\n",Configuration::State.c_str());
|
|||
|
|
fprintf(fp, "CORSEn:%s\n",Configuration::CORSEn.c_str());
|
|||
|
|
fprintf(fp, "FirstRun:%d\n",0);
|
|||
|
|
fclose(fp);
|
|||
|
|
return true;
|
|||
|
|
}
|
|||
|
|
static bool SaveWebCfg(){
|
|||
|
|
CCString File = CCFile::GetnormalizePath(WebCfgPath);
|
|||
|
|
CCVar fp = fopen(File.c_str(), "w");
|
|||
|
|
if (fp == nullptr) {
|
|||
|
|
return false;
|
|||
|
|
}
|
|||
|
|
// 写入配置数据
|
|||
|
|
fprintf(fp, "worker_processes 1;\n"
|
|||
|
|
"events {\n"
|
|||
|
|
" worker_connections 1024;\n"
|
|||
|
|
"}\n"
|
|||
|
|
"http {\n"
|
|||
|
|
" include mime.types;\n"
|
|||
|
|
" default_type application/octet-stream;\n"
|
|||
|
|
" sendfile on;\n"
|
|||
|
|
" keepalive_timeout 65;\n"
|
|||
|
|
" client_max_body_size 500m;\n"
|
|||
|
|
" server {\n"
|
|||
|
|
" listen %s;\n"
|
|||
|
|
" server_name %s;\n"
|
|||
|
|
" location / {\n"
|
|||
|
|
" root html;\n"
|
|||
|
|
" index index.html index.htm;\n"
|
|||
|
|
" try_files $uri $uri/ /index.html;\n"
|
|||
|
|
" # try_files $uri $uri/ @router;\n"
|
|||
|
|
" }\n"
|
|||
|
|
" location /api/ {\n"
|
|||
|
|
" #rewrite ^.+api/?(.*)$ /$1 break; \n"
|
|||
|
|
" rewrite ^/api/(.*)$ /$1 break; \n"
|
|||
|
|
" proxy_pass http://%s:%s/;\n"
|
|||
|
|
" proxy_redirect off;\n"
|
|||
|
|
" proxy_set_header Host $host;\n"
|
|||
|
|
" proxy_set_header X-Real-IP $remote_addr;\n"
|
|||
|
|
" proxy_set_header X-Forwarded-For $proxy_add_x_forwarded_for;\n"
|
|||
|
|
" }\n"
|
|||
|
|
" error_page 500 502 503 504 /50x.html;\n"
|
|||
|
|
" location = /50x.html {\n"
|
|||
|
|
" root html;\n"
|
|||
|
|
" }\n"
|
|||
|
|
" }\n"
|
|||
|
|
"}",Web_Port.c_str(),Web_IP.c_str(),
|
|||
|
|
Sub_IP.c_str(),Sub_Port.c_str());
|
|||
|
|
fclose(fp);
|
|||
|
|
return true;
|
|||
|
|
}
|
|||
|
|
static bool terminateProcessByName(const char *processName) {
|
|||
|
|
#ifdef _WIN32
|
|||
|
|
CCString str = processName;
|
|||
|
|
str.append(".exe");
|
|||
|
|
HANDLE hSnapshot = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0);
|
|||
|
|
if (hSnapshot == INVALID_HANDLE_VALUE) {
|
|||
|
|
return false;
|
|||
|
|
}
|
|||
|
|
PROCESSENTRY32 pe32;
|
|||
|
|
pe32.dwSize = sizeof(PROCESSENTRY32);
|
|||
|
|
if (Process32First(hSnapshot, &pe32)) {
|
|||
|
|
do {
|
|||
|
|
if (_tcscmp(pe32.szExeFile, str.c_str()) == 0) {
|
|||
|
|
HANDLE hProcess = OpenProcess(PROCESS_TERMINATE, FALSE, pe32.th32ProcessID);
|
|||
|
|
if (hProcess != NULL) {
|
|||
|
|
TerminateProcess(hProcess, 0);
|
|||
|
|
CloseHandle(hProcess);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
} while (Process32Next(hSnapshot, &pe32));
|
|||
|
|
}
|
|||
|
|
CloseHandle(hSnapshot);
|
|||
|
|
return true;
|
|||
|
|
#elif __linux__
|
|||
|
|
|
|||
|
|
#endif
|
|||
|
|
}
|
|||
|
|
static void IPBS_NSSMF(bool F){
|
|||
|
|
if(F){
|
|||
|
|
const CCString str = CCString("./") + IPBS_NSSM_Exe;
|
|||
|
|
QProcess::startDetached(str.c_str());
|
|||
|
|
}
|
|||
|
|
else{
|
|||
|
|
#ifdef _WIN32
|
|||
|
|
terminateProcessByName("IPBS_NSSM");
|
|||
|
|
#elif __linux__
|
|||
|
|
#endif
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
};
|
|||
|
|
|
|||
|
|
#endif
|