update v-advance
This commit is contained in:
lzx
@@ -2,12 +2,14 @@ package linux
|
||||
|
||||
// lspci | grep -i vga
|
||||
import (
|
||||
"client/internal/db"
|
||||
message "client/internal/msg"
|
||||
"client/internal/utils"
|
||||
"fmt"
|
||||
"log"
|
||||
"os"
|
||||
"os/exec"
|
||||
"os/user"
|
||||
"path/filepath"
|
||||
"strings"
|
||||
"sync"
|
||||
@@ -17,12 +19,32 @@ import (
|
||||
)
|
||||
|
||||
type LinuxClient struct {
|
||||
mu sync.Mutex
|
||||
Auth string
|
||||
MachineCode string
|
||||
ID string
|
||||
MiningConfig message.MiningConfig
|
||||
Status int // 当前client状态，1正在工作，2空闲
|
||||
mu sync.Mutex
|
||||
Auth string
|
||||
MachineCode string
|
||||
ID string
|
||||
MiningConfig message.MiningConfig
|
||||
db *db.SQLiteServer
|
||||
Status int // 当前client状态，1正在工作，2空闲
|
||||
currentProcess *currentProcess // 当前挖矿进程及类型
|
||||
}
|
||||
|
||||
type currentProcess struct {
|
||||
process *exec.Cmd
|
||||
miner string
|
||||
}
|
||||
|
||||
// CheckPermission 检查当前进程是否具备运行本客户端所需的权限（Linux）
|
||||
// 要求使用 root 运行，否则很多命令（如 dmidecode、nvidia-smi 等）可能失败。
|
||||
func CheckPermission() error {
|
||||
u, err := user.Current()
|
||||
if err != nil {
|
||||
return fmt.Errorf("获取当前用户信息失败: %v", err)
|
||||
}
|
||||
if u.Uid != "0" {
|
||||
return fmt.Errorf("当前用户(%s)不是 root 用户，请使用 sudo 或 root 账号运行本程序", u.Username)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func NewLinuxClient(auth string) *LinuxClient {
|
||||
@@ -80,6 +102,12 @@ func NewLinuxClient(auth string) *LinuxClient {
|
||||
client.Auth = auth
|
||||
client.MiningConfig = miningConfig
|
||||
client.Status = 2
|
||||
|
||||
// 初始化sqlite3数据库
|
||||
db := db.NewSQLiteServer()
|
||||
client.db = db
|
||||
client.initHistoryTask()
|
||||
// 获取主机身份
|
||||
mac, err := client.GetMACAddress()
|
||||
if err != nil {
|
||||
log.Fatalln(err)
|
||||
@@ -87,9 +115,29 @@ func NewLinuxClient(auth string) *LinuxClient {
|
||||
}
|
||||
client.MachineCode = mac
|
||||
client.ID = auth + "." + mac
|
||||
|
||||
return client
|
||||
}
|
||||
|
||||
func (l *LinuxClient) initHistoryTask() {
|
||||
exsits, task := l.db.LoadMiningTask()
|
||||
if exsits {
|
||||
currentTs := time.Now().Unix()
|
||||
if currentTs < int64(task.EndTimestamp) {
|
||||
l.mu.Lock()
|
||||
l.Status = 1
|
||||
l.mu.Unlock()
|
||||
err := l.Mining(task)
|
||||
if err != nil {
|
||||
log.Fatalf("重新开启挖矿任务失败，请手动检查：%v", err)
|
||||
return
|
||||
}
|
||||
} else {
|
||||
l.db.FinishMiningTask(task)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// 获取 NVIDIA 显卡信息
|
||||
func getNvidiaGPUInfo() (map[int]message.GPU, error) {
|
||||
// 使用 nvidia-smi 列出所有 GPU
|
||||
@@ -189,24 +237,33 @@ func (l *LinuxClient) GetGPUInfo() (map[int]message.GPU, error) {
|
||||
}
|
||||
|
||||
// 获取 MAC 地址
|
||||
// 获取“机器码”：Linux 下改为使用主机 UUID
|
||||
// 优先从 /sys/class/dmi/id/product_uuid 读取，失败时再尝试 dmidecode
|
||||
func (l *LinuxClient) GetMACAddress() (string, error) {
|
||||
// 执行 ifconfig 或 ip link 命令
|
||||
cmd := exec.Command("ifconfig", "-a")
|
||||
out, err := cmd.Output()
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
|
||||
// 查找 MAC 地址
|
||||
lines := strings.Split(string(out), "\n")
|
||||
for _, line := range lines {
|
||||
if strings.Contains(line, "ether") {
|
||||
parts := strings.Fields(line)
|
||||
return parts[1], nil
|
||||
// 1. 优先读取内核提供的 product_uuid 文件（大多数物理机/虚拟机都支持）
|
||||
const uuidPath = "/sys/class/dmi/id/product_uuid"
|
||||
if data, err := os.ReadFile(uuidPath); err == nil {
|
||||
uuid := strings.TrimSpace(string(data))
|
||||
if uuid != "" && !strings.EqualFold(uuid, "FFFFFFFF-FFFF-FFFF-FFFF-FFFFFFFFFFFF") {
|
||||
uuid = strings.Trim(uuid, "{}")
|
||||
uuid = strings.ToUpper(uuid)
|
||||
return uuid, nil
|
||||
}
|
||||
}
|
||||
|
||||
return "", fmt.Errorf("MAC address not found")
|
||||
// 2. 退回使用 dmidecode（需要有权限）
|
||||
cmd := exec.Command("dmidecode", "-s", "system-uuid")
|
||||
out, err := cmd.Output()
|
||||
if err != nil {
|
||||
return "", fmt.Errorf("获取主机 UUID 失败: %v", err)
|
||||
}
|
||||
uuid := strings.TrimSpace(string(out))
|
||||
if uuid == "" || strings.EqualFold(uuid, "FFFFFFFF-FFFF-FFFF-FFFF-FFFFFFFFFFFF") {
|
||||
return "", fmt.Errorf("获取到的主机 UUID 无效: %q", uuid)
|
||||
}
|
||||
uuid = strings.Trim(uuid, "{}")
|
||||
uuid = strings.ToUpper(uuid)
|
||||
return uuid, nil
|
||||
}
|
||||
|
||||
/*
|
||||
@@ -249,27 +306,47 @@ func (l *LinuxClient) lolminer(cfg message.ConfigurationMiningMsg) {
|
||||
log.Fatalf("Error starting lolMiner: %v", err)
|
||||
return
|
||||
}
|
||||
// 添加执行记录
|
||||
go func() {
|
||||
err := l.db.InsertMiningTask(cfg)
|
||||
if err != nil {
|
||||
log.Fatalf("本次挖矿任务记录失败：%v", err)
|
||||
}
|
||||
}()
|
||||
// 获取 lolMiner 的进程 ID
|
||||
fmt.Printf("lolMiner started with PID: %d\n", cmd.Process.Pid)
|
||||
|
||||
// 记录当前挖矿进程
|
||||
l.mu.Lock()
|
||||
l.currentProcess = ¤tProcess{
|
||||
process: cmd,
|
||||
miner: "lolminer",
|
||||
}
|
||||
l.mu.Unlock()
|
||||
// 获取当前时间戳（秒级）
|
||||
currentTimestamp := time.Now().Unix()
|
||||
endTimestamp := int64(cfg.EndTimestamp)
|
||||
// 计算目标时间戳和当前时间戳之间的差值
|
||||
if endTimestamp <= currentTimestamp {
|
||||
fmt.Println("目标时间已经到达，直接执行操作")
|
||||
// 如果目标时间已经到达，立即结束挖矿进程
|
||||
fmt.Println("目标时间已经到达，立即结束 lolMiner 挖矿进程")
|
||||
l.StopMining()
|
||||
} else {
|
||||
// 计算需要等待的秒数
|
||||
waitDuration := time.Second * time.Duration(endTimestamp-currentTimestamp)
|
||||
fmt.Printf("当前时间戳：%d，目标时间戳：%d，剩余时间：%v\n", currentTimestamp, endTimestamp, waitDuration)
|
||||
// 使用 time.Sleep 等待直到目标时间戳
|
||||
time.Sleep(waitDuration)
|
||||
fmt.Println("目标时间到达，开始执行操作")
|
||||
// 杀掉进程
|
||||
err = cmd.Process.Kill()
|
||||
if err != nil {
|
||||
log.Fatalf("Error killing lolMiner: %v", err)
|
||||
}
|
||||
|
||||
fmt.Println("目标时间到达，开始执行操作，停止 lolMiner 挖矿进程")
|
||||
// 通过 StopMining 统一停止挖矿进程
|
||||
l.StopMining()
|
||||
// 修改执行记录
|
||||
go func() {
|
||||
err := l.db.FinishMiningTask(cfg)
|
||||
if err != nil {
|
||||
log.Fatalf("修改执行记录失败：%v", err)
|
||||
}
|
||||
}()
|
||||
// 输出进程被杀死的信息
|
||||
fmt.Println("lolMiner process killed.")
|
||||
}
|
||||
@@ -277,6 +354,7 @@ func (l *LinuxClient) lolminer(cfg message.ConfigurationMiningMsg) {
|
||||
l.mu.Lock()
|
||||
l.Status = 2
|
||||
l.mu.Unlock()
|
||||
|
||||
}
|
||||
|
||||
/*
|
||||
@@ -317,27 +395,46 @@ func (l *LinuxClient) bzminer(cfg message.ConfigurationMiningMsg) {
|
||||
log.Fatalf("Error starting bzminer: %v", err)
|
||||
return
|
||||
}
|
||||
// 添加执行记录
|
||||
go func() {
|
||||
err := l.db.InsertMiningTask(cfg)
|
||||
if err != nil {
|
||||
log.Fatalf("本次挖矿任务记录失败：%v", err)
|
||||
}
|
||||
}()
|
||||
// 获取 bzminer 的进程 ID
|
||||
fmt.Printf("bzminer started with PID: %d\n", cmd.Process.Pid)
|
||||
|
||||
// 记录当前挖矿进程
|
||||
l.mu.Lock()
|
||||
l.currentProcess = ¤tProcess{
|
||||
process: cmd,
|
||||
miner: "bzminer",
|
||||
}
|
||||
l.mu.Unlock()
|
||||
// 获取当前时间戳（秒级）
|
||||
currentTimestamp := time.Now().Unix()
|
||||
endTimestamp := int64(cfg.EndTimestamp)
|
||||
// 计算目标时间戳和当前时间戳之间的差值
|
||||
if endTimestamp <= currentTimestamp {
|
||||
fmt.Println("目标时间已经到达，直接执行操作")
|
||||
fmt.Println("目标时间已经到达，停止 bzminer 挖矿进程")
|
||||
l.StopMining()
|
||||
} else {
|
||||
// 计算需要等待的秒数
|
||||
waitDuration := time.Second * time.Duration(endTimestamp-currentTimestamp)
|
||||
fmt.Printf("当前时间戳：%d，目标时间戳：%d，剩余时间：%v\n", currentTimestamp, endTimestamp, waitDuration)
|
||||
// 使用 time.Sleep 等待直到目标时间戳
|
||||
time.Sleep(waitDuration)
|
||||
fmt.Println("目标时间到达，开始执行操作")
|
||||
// 杀掉进程
|
||||
err = cmd.Process.Kill()
|
||||
if err != nil {
|
||||
log.Fatalf("Error killing bzminer: %v", err)
|
||||
}
|
||||
|
||||
fmt.Println("目标时间到达，开始执行操作，停止 bzminer 挖矿进程")
|
||||
// 通过 StopMining 统一停止挖矿进程
|
||||
l.StopMining()
|
||||
// 修改执行记录
|
||||
go func() {
|
||||
err := l.db.FinishMiningTask(cfg)
|
||||
if err != nil {
|
||||
log.Fatalf("修改执行记录失败：%v", err)
|
||||
}
|
||||
}()
|
||||
// 输出进程被杀死的信息
|
||||
fmt.Println("bzminer process killed.")
|
||||
}
|
||||
@@ -374,7 +471,8 @@ func (l *LinuxClient) rigel(cfg message.ConfigurationMiningMsg) {
|
||||
}
|
||||
dir := l.MiningConfig.RigelPath
|
||||
name := filepath.Join(dir, "rigel")
|
||||
args := []string{"-a", strings.ToLower(cfg.Algo), "-o", cfg.PoolUrl, "-u", address, "-w", cfg.WorkerID, "--log-file", "logs/miner.log"}
|
||||
// 禁用 rigel 内置 watchdog，避免自动拉起子进程
|
||||
args := []string{"--no-watchdog", "-a", strings.ToLower(cfg.Algo), "-o", cfg.PoolUrl, "-u", address, "-w", cfg.WorkerID, "--log-file", "logs/miner.log"}
|
||||
cmd := exec.Command(name, args...)
|
||||
cmd.Dir = dir
|
||||
cmd.Stdout = os.Stdout
|
||||
@@ -386,28 +484,48 @@ func (l *LinuxClient) rigel(cfg message.ConfigurationMiningMsg) {
|
||||
log.Fatalf("Error starting rigel: %v", err)
|
||||
return
|
||||
}
|
||||
// 添加执行记录
|
||||
go func() {
|
||||
err := l.db.InsertMiningTask(cfg)
|
||||
if err != nil {
|
||||
log.Fatalf("本次挖矿任务记录失败：%v", err)
|
||||
}
|
||||
}()
|
||||
// 获取 rigel 的进程 ID
|
||||
fmt.Printf("rigel started with PID: %d\n", cmd.Process.Pid)
|
||||
|
||||
// 记录当前挖矿进程
|
||||
l.mu.Lock()
|
||||
l.currentProcess = ¤tProcess{
|
||||
process: cmd,
|
||||
miner: "rigel",
|
||||
}
|
||||
l.mu.Unlock()
|
||||
// 获取当前时间戳（秒级）
|
||||
currentTimestamp := time.Now().Unix()
|
||||
endTimestamp := int64(cfg.EndTimestamp)
|
||||
// 计算目标时间戳和当前时间戳之间的差值
|
||||
if endTimestamp <= currentTimestamp {
|
||||
fmt.Println("目标时间已经到达，直接执行操作")
|
||||
// 如果目标时间已经到达，立即结束 rigel 挖矿进程
|
||||
fmt.Println("目标时间已经到达，立即结束 rigel 挖矿进程")
|
||||
l.StopMining()
|
||||
} else {
|
||||
// 计算需要等待的秒数
|
||||
waitDuration := time.Second * time.Duration(endTimestamp-currentTimestamp)
|
||||
fmt.Printf("当前时间戳：%d，目标时间戳：%d，剩余时间：%v\n", currentTimestamp, endTimestamp, waitDuration)
|
||||
// 使用 time.Sleep 等待直到目标时间戳
|
||||
time.Sleep(waitDuration)
|
||||
fmt.Println("目标时间到达，开始执行操作")
|
||||
// 杀掉进程
|
||||
err = cmd.Process.Kill()
|
||||
if err != nil {
|
||||
log.Fatalf("Error killing rigel: %v", err)
|
||||
}
|
||||
|
||||
// 输出进程被杀死的信息
|
||||
fmt.Println("目标时间到达，开始执行操作，结束 rigel 挖矿进程")
|
||||
// 通过 StopMining 统一停止挖矿进程
|
||||
l.StopMining()
|
||||
// 修改执行记录
|
||||
go func() {
|
||||
err := l.db.FinishMiningTask(cfg)
|
||||
if err != nil {
|
||||
log.Fatalf("修改执行记录失败：%v", err)
|
||||
}
|
||||
}()
|
||||
// 输出进程被结束的信息
|
||||
fmt.Println("rigel process killed.")
|
||||
}
|
||||
log.Printf("当前挖矿任务已执行完毕：币=%s, 算法=%s, 矿池=%s, 截止时间=%d", cfg.Coin, cfg.Algo, cfg.Pool, cfg.EndTimestamp)
|
||||
@@ -445,3 +563,42 @@ func (l *LinuxClient) Mining(cfg message.ConfigurationMiningMsg) error {
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// 主动终止当前挖矿进程
|
||||
func (l *LinuxClient) StopMining() {
|
||||
l.mu.Lock()
|
||||
cp := l.currentProcess
|
||||
l.mu.Unlock()
|
||||
|
||||
if cp == nil || cp.process == nil || cp.process.Process == nil {
|
||||
log.Println("当前没有正在进行的挖矿任务")
|
||||
return
|
||||
}
|
||||
|
||||
log.Printf("准备停止当前挖矿任务，miner=%s, pid=%d", cp.miner, cp.process.Process.Pid)
|
||||
|
||||
// 根据挖矿软件类型选择合适的停止方式
|
||||
switch strings.ToLower(cp.miner) {
|
||||
case "rigel":
|
||||
// 优先尝试优雅退出（Interrupt），失败再 Kill
|
||||
if err := cp.process.Process.Signal(os.Interrupt); err != nil {
|
||||
log.Printf("向 rigel 发送 Interrupt 失败，尝试 Kill: %v", err)
|
||||
if errKill := cp.process.Process.Kill(); errKill != nil {
|
||||
log.Printf("Kill rigel 失败：%v", errKill)
|
||||
}
|
||||
}
|
||||
default:
|
||||
// 其它挖矿软件直接 Kill
|
||||
if err := cp.process.Process.Kill(); err != nil {
|
||||
log.Printf("停止挖矿进程失败：%v", err)
|
||||
}
|
||||
}
|
||||
|
||||
// 清理状态
|
||||
l.mu.Lock()
|
||||
l.currentProcess = nil
|
||||
l.Status = 2
|
||||
l.mu.Unlock()
|
||||
|
||||
log.Println("当前挖矿任务已被手动停止")
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user