mirror of
https://github.com/oneclickvirt/ecs.git
synced 2025-10-05 15:27:08 +08:00
381 lines
14 KiB
Go
381 lines
14 KiB
Go
package main
|
||
|
||
import (
|
||
"fmt"
|
||
"github.com/oneclickvirt/CommonMediaTests/commediatests"
|
||
backtraceori "github.com/oneclickvirt/backtrace/bk"
|
||
basicmodel "github.com/oneclickvirt/basics/model"
|
||
"github.com/oneclickvirt/ecs/backtrace"
|
||
"github.com/oneclickvirt/ecs/commediatest"
|
||
"github.com/oneclickvirt/ecs/cputest"
|
||
"github.com/oneclickvirt/ecs/disktest"
|
||
"github.com/oneclickvirt/ecs/memorytest"
|
||
"github.com/oneclickvirt/ecs/ntrace"
|
||
"github.com/oneclickvirt/ecs/speedtest"
|
||
"github.com/oneclickvirt/ecs/unlocktest"
|
||
"github.com/oneclickvirt/ecs/utils"
|
||
gostunmodel "github.com/oneclickvirt/gostun/model"
|
||
"github.com/oneclickvirt/portchecker/email"
|
||
speedtestmodel "github.com/oneclickvirt/speedtest/model"
|
||
"gopkg.in/alecthomas/kingpin.v2"
|
||
"os"
|
||
"os/signal"
|
||
"regexp"
|
||
"runtime"
|
||
"strings"
|
||
"sync"
|
||
"syscall"
|
||
"time"
|
||
)
|
||
|
||
var (
|
||
ecsVersion = "v0.0.21"
|
||
input, choice string
|
||
width = 84
|
||
filePath = "goecs.txt"
|
||
enabelUpload = true
|
||
)
|
||
|
||
var (
|
||
showVersion = kingpin.Flag("v", "Display version information").Bool()
|
||
menuMode = kingpin.Flag("menu", "Enable/Disable menu mode, disable example: -menu=false").Default("true").Bool()
|
||
language = kingpin.Flag("l", "Set language (supported: en, zh)").Default("zh").String()
|
||
basicStatus = kingpin.Flag("basic", "Enable/Disable basic test").Default("true").Bool()
|
||
cpuTestStatus = kingpin.Flag("cpu", "Enable/Disable CPU test").Default("true").Bool()
|
||
memoryTestStatus = kingpin.Flag("memory", "Enable/Disable memory test").Default("true").Bool()
|
||
diskTestStatus = kingpin.Flag("disk", "Enable/Disable disk test").Default("true").Bool()
|
||
commTestStatus = kingpin.Flag("comm", "Enable/Disable common media test").Default("true").Bool()
|
||
utTestStatus = kingpin.Flag("ut", "Enable/Disable unlock media test").Default("true").Bool()
|
||
securityTestStatus = kingpin.Flag("security", "Enable/Disable security test").Default("true").Bool()
|
||
emailTestStatus = kingpin.Flag("email", "Enable/Disable email port test").Default("true").Bool()
|
||
backtraceStatus = kingpin.Flag("backtrace", "Enable/Disable backtrace test (in 'en' language or on `windows` it always false)").Default("true").Bool()
|
||
nt3Status = kingpin.Flag("nt3", "Enable/Disable NT3 test (in 'en' language or on `windows` it always false)").Default("true").Bool()
|
||
speedTestStatus = kingpin.Flag("speed", "Enable/Disable speed test").Default("true").Bool()
|
||
cpuTestMethod = kingpin.Flag("cpum", "Set CPU test method (supported: sysbench, geekbench, winsat)").Default("sysbench").String()
|
||
cpuTestThreadMode = kingpin.Flag("cput", "Set CPU test thread mode (supported: single, multi)").Default("multi").String()
|
||
memoryTestMethod = kingpin.Flag("memorym", "Set memory test method (supported: sysbench, dd, winsat)").Default("dd").String()
|
||
diskTestMethod = kingpin.Flag("diskm", "Set disk test method (supported: fio, dd, winsat)").Default("fio").String()
|
||
diskTestPath = kingpin.Flag("diskp", "Set disk test path, e.g., -diskp /root").String()
|
||
diskMultiCheck = kingpin.Flag("diskmc", "Enable/Disable multiple disk checks, e.g., -diskmc=false").Default("false").Bool()
|
||
nt3Location = kingpin.Flag("nt3loc", "Specify NT3 test location (supported: GZ, SH, BJ, CD for Guangzhou, Shanghai, Beijing, Chengdu)").Default("GZ").String()
|
||
nt3CheckType = kingpin.Flag("nt3t", "Set NT3 test type (supported: both, ipv4, ipv6)").Default("ipv4").String()
|
||
spNum = kingpin.Flag("spnum", "Set the number of servers per operator for speed test").Default("2").Int()
|
||
enableLogger = kingpin.Flag("log", "Enable/Disable logging in the current path").Default("false").Bool()
|
||
)
|
||
|
||
func main() {
|
||
kingpin.Parse()
|
||
sig := make(chan os.Signal, 1)
|
||
signal.Notify(sig, os.Interrupt, syscall.SIGINT, syscall.SIGTERM)
|
||
if *showVersion {
|
||
fmt.Println(ecsVersion)
|
||
return
|
||
}
|
||
if *enableLogger {
|
||
basicmodel.EnableLoger = true
|
||
speedtestmodel.EnableLoger = true
|
||
gostunmodel.EnableLoger = true
|
||
commediatests.EnableLoger = true
|
||
backtraceori.EnableLoger = true
|
||
}
|
||
if *menuMode {
|
||
*basicStatus, *cpuTestStatus, *memoryTestStatus, *diskTestStatus = false, false, false, false
|
||
*commTestStatus, *utTestStatus, *securityTestStatus, *emailTestStatus = false, false, false, false
|
||
*backtraceStatus, *nt3Status, *speedTestStatus = false, false, false
|
||
switch *language {
|
||
case "zh":
|
||
fmt.Println("1. 融合怪完全体")
|
||
fmt.Println("2. 极简版(系统信息+CPU+内存+磁盘+测速节点5个)")
|
||
fmt.Println("3. 精简版(系统信息+CPU+内存+磁盘+御三家+常用流媒体+回程+路由+测速节点5个)")
|
||
fmt.Println("4. 精简网络版(系统信息+CPU+内存+磁盘+回程+路由+测速节点5个)")
|
||
fmt.Println("5. 精简解锁版(系统信息+CPU+内存+磁盘IO+御三家+常用流媒体+测速节点5个)")
|
||
fmt.Println("6. 网络单项(IP质量检测+三网回程+三网路由与延迟+测速节点11个)")
|
||
fmt.Println("7. 解锁单项(御三家解锁+常用流媒体解锁)")
|
||
fmt.Println("8. 硬件单项(基础系统信息+CPU+内存+dd磁盘测试+fio磁盘测试)")
|
||
fmt.Println("9. IP质量检测(15个数据库的IP检测+邮件端口检测)")
|
||
fmt.Println("10. 三网回程线路+广州三网路由+全国三网延迟")
|
||
case "en":
|
||
fmt.Println("1. VPS Fusion Monster Test Comprehensive Test Suite")
|
||
fmt.Println("2. Minimal Test Suite (System Info + CPU + Memory + Disk + 5 Speed Test Nodes)")
|
||
fmt.Println("3. Standard Test Suite (System Info + CPU + Memory + Disk + Basic Unlock Tests + Common Streaming Services + 5 Speed Test Nodes)")
|
||
fmt.Println("4. Network-Focused Test Suite (System Info + CPU + Memory + Disk + 5 Speed Test Nodes)")
|
||
fmt.Println("5. Unlock-Focused Test Suite (System Info + CPU + Memory + Disk IO + Basic Unlock Tests + Common Streaming Services + 5 Speed Test Nodes)")
|
||
fmt.Println("6. Network-Only Test (IP Quality Test + 5 Speed Test Nodes)")
|
||
fmt.Println("7. Unlock-Only Test (Basic Unlock Tests + Common Streaming Services Unlock)")
|
||
fmt.Println("8. Hardware-Only Test (Basic System Info + CPU + Memory + dd Disk Test + fio Disk Test)")
|
||
fmt.Println("9. IP Quality Test (IP Test with 15 Databases + Email Port Test)")
|
||
}
|
||
Loop:
|
||
for {
|
||
fmt.Print("请输入选项 / Please enter your choice: ")
|
||
fmt.Scanln(&input)
|
||
input = strings.TrimSpace(input)
|
||
input = strings.TrimRight(input, "\n")
|
||
re := regexp.MustCompile(`^\d+$`) // 正则表达式匹配纯数字
|
||
if re.MatchString(input) {
|
||
choice = input
|
||
switch choice {
|
||
case "1":
|
||
*basicStatus = true
|
||
*cpuTestStatus = true
|
||
*memoryTestStatus = true
|
||
*diskTestStatus = true
|
||
*commTestStatus = true
|
||
*utTestStatus = true
|
||
*securityTestStatus = true
|
||
*emailTestStatus = true
|
||
*backtraceStatus = true
|
||
*nt3Status = true
|
||
*speedTestStatus = true
|
||
break Loop
|
||
case "2":
|
||
*basicStatus = true
|
||
*cpuTestStatus = true
|
||
*memoryTestStatus = true
|
||
*diskTestStatus = true
|
||
*speedTestStatus = true
|
||
break Loop
|
||
case "3":
|
||
*basicStatus = true
|
||
*cpuTestStatus = true
|
||
*memoryTestStatus = true
|
||
*diskTestStatus = true
|
||
*commTestStatus = true
|
||
*utTestStatus = true
|
||
*securityTestStatus = true
|
||
*backtraceStatus = true
|
||
*nt3Status = true
|
||
*speedTestStatus = true
|
||
break Loop
|
||
case "4":
|
||
*basicStatus = true
|
||
*cpuTestStatus = true
|
||
*memoryTestStatus = true
|
||
*diskTestStatus = true
|
||
*backtraceStatus = true
|
||
*nt3Status = true
|
||
*speedTestStatus = true
|
||
break Loop
|
||
case "5":
|
||
*basicStatus = true
|
||
*cpuTestStatus = true
|
||
*memoryTestStatus = true
|
||
*diskTestStatus = true
|
||
*securityTestStatus = true
|
||
*speedTestStatus = true
|
||
break Loop
|
||
case "6":
|
||
*speedTestStatus = true
|
||
*backtraceStatus = true
|
||
*nt3Status = true
|
||
break Loop
|
||
case "7":
|
||
*securityTestStatus = true
|
||
*commTestStatus = true
|
||
break Loop
|
||
case "8":
|
||
*basicStatus = true
|
||
*cpuTestStatus = true
|
||
*memoryTestStatus = true
|
||
*diskTestStatus = true
|
||
break Loop
|
||
case "9":
|
||
*emailTestStatus = true
|
||
break Loop
|
||
case "10":
|
||
*backtraceStatus = true
|
||
*nt3Status = true
|
||
*speedTestStatus = true
|
||
break Loop
|
||
default:
|
||
if *language == "zh" {
|
||
fmt.Println("无效的选项")
|
||
} else {
|
||
fmt.Println("Invalid choice")
|
||
}
|
||
}
|
||
} else {
|
||
if *language == "zh" {
|
||
fmt.Println("输入错误，请输入一个纯数字")
|
||
} else {
|
||
fmt.Println("Invalid input, please enter a number")
|
||
}
|
||
}
|
||
}
|
||
}
|
||
if *language == "en" {
|
||
*backtraceStatus = false
|
||
*nt3Status = false
|
||
}
|
||
startTime := time.Now()
|
||
var (
|
||
wg1, wg2 sync.WaitGroup
|
||
basicInfo, securityInfo, emailInfo, mediaInfo string
|
||
output, tempOutput string
|
||
)
|
||
// 启动一个goroutine来等待信号
|
||
go func() {
|
||
// 等待信号
|
||
<-sig
|
||
utils.ProcessAndUpload(output, filePath, enabelUpload)
|
||
os.Exit(1) // 使用非零状态码退出，表示意外退出
|
||
}()
|
||
output = utils.PrintAndCapture(func() {
|
||
switch *language {
|
||
case "zh":
|
||
utils.PrintHead(*language, width, ecsVersion)
|
||
if *basicStatus || *securityTestStatus {
|
||
if *basicStatus {
|
||
utils.PrintCenteredTitle("基础信息", width)
|
||
}
|
||
basicInfo, securityInfo, *nt3CheckType = utils.SecurityCheck(*language, *nt3CheckType, *securityTestStatus)
|
||
if *basicStatus {
|
||
fmt.Printf(basicInfo)
|
||
}
|
||
}
|
||
if *cpuTestStatus {
|
||
utils.PrintCenteredTitle(fmt.Sprintf("CPU测试-通过%s测试", *cpuTestMethod), width)
|
||
cputest.CpuTest(*language, *cpuTestMethod, *cpuTestThreadMode)
|
||
}
|
||
if *memoryTestStatus {
|
||
utils.PrintCenteredTitle(fmt.Sprintf("内存测试-通过%s测试", *cpuTestMethod), width)
|
||
memorytest.MemoryTest(*language, *memoryTestMethod)
|
||
}
|
||
if *diskTestStatus {
|
||
utils.PrintCenteredTitle(fmt.Sprintf("硬盘测试-通过%s测试", *diskTestMethod), width)
|
||
disktest.DiskTest(*language, *diskTestMethod, *diskTestPath, *diskMultiCheck)
|
||
}
|
||
if *emailTestStatus {
|
||
wg2.Add(1)
|
||
go func() {
|
||
defer wg2.Done()
|
||
emailInfo = email.EmailCheck()
|
||
}()
|
||
}
|
||
if *utTestStatus {
|
||
wg1.Add(1)
|
||
go func() {
|
||
defer wg1.Done()
|
||
mediaInfo = unlocktest.MediaTest(*language)
|
||
}()
|
||
}
|
||
if *commTestStatus {
|
||
utils.PrintCenteredTitle("御三家流媒体解锁", width)
|
||
commediatest.ComMediaTest(*language)
|
||
}
|
||
if *utTestStatus {
|
||
utils.PrintCenteredTitle("跨国流媒体解锁", width)
|
||
wg1.Wait()
|
||
fmt.Printf(mediaInfo)
|
||
}
|
||
if *securityTestStatus {
|
||
utils.PrintCenteredTitle("IP质量检测", width)
|
||
fmt.Printf(securityInfo)
|
||
}
|
||
if *emailTestStatus {
|
||
utils.PrintCenteredTitle("邮件端口检测", width)
|
||
wg2.Wait()
|
||
fmt.Println(emailInfo)
|
||
}
|
||
if runtime.GOOS != "windows" {
|
||
if *backtraceStatus {
|
||
utils.PrintCenteredTitle("三网回程", width)
|
||
backtrace.BackTrace()
|
||
}
|
||
// nexttrace 在win上不支持检测，报错 bind: An invalid argument was supplied.
|
||
if *nt3Status {
|
||
utils.PrintCenteredTitle("路由检测", width)
|
||
ntrace.TraceRoute3(*language, *nt3Location, *nt3CheckType)
|
||
}
|
||
}
|
||
if *speedTestStatus {
|
||
utils.PrintCenteredTitle("就近节点测速", width)
|
||
speedtest.ShowHead(*language)
|
||
if (*menuMode && choice == "1") || !*menuMode {
|
||
speedtest.NearbySP()
|
||
speedtest.CustomSP("net", "global", 2)
|
||
speedtest.CustomSP("net", "cu", *spNum)
|
||
speedtest.CustomSP("net", "ct", *spNum)
|
||
speedtest.CustomSP("net", "cmcc", *spNum)
|
||
} else if *menuMode && choice == "2" || choice == "3" || choice == "4" || choice == "5" {
|
||
speedtest.CustomSP("net", "global", 4)
|
||
}
|
||
}
|
||
utils.PrintCenteredTitle("", width)
|
||
endTime := time.Now()
|
||
duration := endTime.Sub(startTime)
|
||
minutes := int(duration.Minutes())
|
||
seconds := int(duration.Seconds()) % 60
|
||
currentTime := time.Now().Format("Mon Jan 2 15:04:05 MST 2006")
|
||
fmt.Printf("花费 : %d 分 %d 秒\n", minutes, seconds)
|
||
fmt.Printf("时间 : %s\n", currentTime)
|
||
utils.PrintCenteredTitle("", width)
|
||
case "en":
|
||
utils.PrintHead(*language, width, ecsVersion)
|
||
if *basicStatus || *securityTestStatus {
|
||
if *basicStatus {
|
||
utils.PrintCenteredTitle("Basic Information", width)
|
||
}
|
||
basicInfo, securityInfo, *nt3CheckType = utils.SecurityCheck(*language, *nt3CheckType, *securityTestStatus)
|
||
if *basicStatus {
|
||
fmt.Printf(basicInfo)
|
||
}
|
||
}
|
||
if *cpuTestStatus {
|
||
utils.PrintCenteredTitle(fmt.Sprintf("CPU Test - %s Method", *cpuTestMethod), width)
|
||
cputest.CpuTest(*language, *cpuTestMethod, *cpuTestThreadMode)
|
||
}
|
||
if *memoryTestStatus {
|
||
utils.PrintCenteredTitle(fmt.Sprintf("Memory Test - %s Method", *memoryTestMethod), width)
|
||
memorytest.MemoryTest(*language, *memoryTestMethod)
|
||
}
|
||
if *diskTestStatus {
|
||
utils.PrintCenteredTitle(fmt.Sprintf("Disk Test - %s Method", *diskTestMethod), width)
|
||
disktest.DiskTest(*language, *diskTestMethod, *diskTestPath, *diskMultiCheck)
|
||
}
|
||
if *emailTestStatus {
|
||
wg1.Add(1)
|
||
go func() {
|
||
defer wg1.Done()
|
||
emailInfo = email.EmailCheck()
|
||
}()
|
||
}
|
||
if *commTestStatus {
|
||
utils.PrintCenteredTitle("The Three Families Streaming Media Unlock", width)
|
||
commediatest.ComMediaTest(*language)
|
||
}
|
||
if *utTestStatus {
|
||
utils.PrintCenteredTitle("Cross-Border Streaming Media Unlock", width)
|
||
unlocktest.MediaTest(*language)
|
||
}
|
||
if *securityTestStatus {
|
||
utils.PrintCenteredTitle("IP Quality Check", width)
|
||
fmt.Printf(securityInfo)
|
||
}
|
||
if *emailTestStatus {
|
||
utils.PrintCenteredTitle("Email Port Check", width)
|
||
wg1.Wait()
|
||
fmt.Println(emailInfo)
|
||
}
|
||
if *speedTestStatus {
|
||
utils.PrintCenteredTitle("Nearby Node Speed Test", width)
|
||
speedtest.ShowHead(*language)
|
||
speedtest.NearbySP()
|
||
speedtest.CustomSP("net", "global", -1)
|
||
}
|
||
utils.PrintCenteredTitle("", width)
|
||
endTime := time.Now()
|
||
duration := endTime.Sub(startTime)
|
||
minutes := int(duration.Minutes())
|
||
seconds := int(duration.Seconds()) % 60
|
||
currentTime := time.Now().Format("Mon Jan 2 15:04:05 MST 2006")
|
||
fmt.Printf("Cost Time : %d 分 %d 秒\n", minutes, seconds)
|
||
fmt.Printf("Current Time : %s\n", currentTime)
|
||
utils.PrintCenteredTitle("", width)
|
||
default:
|
||
fmt.Println("Unsupported language")
|
||
}
|
||
}, tempOutput, output)
|
||
utils.ProcessAndUpload(output, filePath, enabelUpload)
|
||
}
|