b1.0

2026-06-22 11:33:18 +08:00
commit 9277e4a2b8
24 changed files with 3700 additions and 0 deletions
@@ -0,0 +1,95 @@
+// Copyright (c) 2026 S.A. (SNOWARE). Written for SCU Team.
+// Licensed under MulanPubL-2.0.
+// You may obtain a copy at: http://license.coscl.org.cn/MulanPubL-2.0
+// THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND.
+
+// DoConnect 模块：负责 EasyTier 子进程的完整生命周期管理（启动、监控、输出捕获、清理）。
+package main
+
+import (
+	"bytes"
+	"fmt"
+	"os"
+	"os/exec"
+	"path/filepath"
+	"strings"
+	"time"
+)
+
+// connectToVirtualHotspot 构建 EasyTier 命令行，清理旧进程，启动新进程，并在 500ms 内检测进程是否存活。
+// 返回子进程的 *os.Process 引用，失败返回 nil。
+func connectToVirtualHotspot() *os.Process {
+	log.Info().Msg("连接虚拟热点开始")
+
+	host := "magpie.snoware.sa-tale.eu.cc"
+	port := "11010"
+	netname := "scu_team"
+	token := "@5cut0t"
+
+	miner := Miner{}
+	cmd := miner.BuildTaskCommand(
+		fmt.Sprintf("tcp://%s:%s", host, port),
+		netname,
+		token,
+	)
+
+	log.Info().
+		Str("host", host).
+		Str("port", port).
+		Str("network", netname).
+		Msg("启动 EasyTier 进程")
+
+	exePath, err := os.Executable()
+	if err != nil {
+		log.Error().Err(err).Msg("获取自身可执行文件路径失败")
+		return nil
+	}
+	workdir := filepath.Dir(exePath)
+	log.Debug().Str("work_dir", workdir).Msg("工作目录")
+
+	cleanupOldEasyTier()
+
+	parts := strings.Fields(cmd)
+	easyTierPath := filepath.Join(workdir, parts[0][2:])
+	if _, err := os.Stat(easyTierPath); os.IsNotExist(err) {
+		log.Error().Str("path", easyTierPath).Msg("easytier-core.exe 不存在，请确保该文件与 srcc.exe 在同一目录")
+		return nil
+	}
+
+	c := exec.Command(easyTierPath, parts[1:]...)
+	c.Dir = workdir
+
+	var stderr, stdout bytes.Buffer
+	c.Stderr = &stderr
+	c.Stdout = &stdout
+
+	if err := c.Start(); err != nil {
+		log.Error().Err(err).Msg("EasyTier 启动失败")
+		return nil
+	}
+
+	done := make(chan error, 1)
+	go func() {
+		done <- c.Wait()
+	}()
+	select {
+	case err := <-done:
+		log.Error().Err(err).Str("stdout", stdout.String()).Str("stderr", stderr.String()).Msg("EasyTier 进程启动后立即退出")
+		return nil
+	case <-time.After(500 * time.Millisecond):
+	}
+
+	log.Info().Int("pid", c.Process.Pid).Msg("EasyTier 进程已启动")
+	return c.Process
+}
+
+// cleanupOldEasyTier 通过 taskkill 强制终止所有旧 easytier-core.exe 进程，避免端口冲突。
+func cleanupOldEasyTier() {
+	cmd := exec.Command("taskkill", "/f", "/im", "easytier-core.exe")
+	cmd.Stdout = nil
+	cmd.Stderr = nil
+	if err := cmd.Run(); err != nil {
+		return
+	}
+	log.Warn().Msg("检测到旧的 EasyTier 进程，已终止")
+}