Init

go.mod

@@ -0,0 +1,11 @@
+module gitee.com/snoware/swlib
+
+go 1.26.4
+
+require github.com/rs/zerolog v1.35.1
+
+require (
+	github.com/mattn/go-colorable v0.1.14 // indirect
+	github.com/mattn/go-isatty v0.0.20 // indirect
+	golang.org/x/sys v0.29.0 // indirect
+)

go.sum

@@ -0,0 +1,9 @@
+github.com/mattn/go-colorable v0.1.14 h1:9A9LHSqF/7dyVVX6g0U9cwm9pG3kP9gSzcuIPHPsaIE=
+github.com/mattn/go-colorable v0.1.14/go.mod h1:6LmQG8QLFO4G5z1gPvYEzlUgJ2wF+stgPZH1UqBm1s8=
+github.com/mattn/go-isatty v0.0.20 h1:xfD0iDuEKnDkl03q4limB+vH+GxLEtL/jb4xVJSWWEY=
+github.com/mattn/go-isatty v0.0.20/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y=
+github.com/rs/zerolog v1.35.1 h1:m7xQeoiLIiV0BCEY4Hs+j2NG4Gp2o2KPKmhnnLiazKI=
+github.com/rs/zerolog v1.35.1/go.mod h1:EjML9kdfa/RMA7h/6z6pYmq1ykOuA8/mjWaEvGI+jcw=
+golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
+golang.org/x/sys v0.29.0 h1:TPYlXGxvx1MGTn2GiZDhnjPA9wZzZeGKHHmKhHYvgaU=
+golang.org/x/sys v0.29.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA=

logger/logger.go

@@ -0,0 +1,104 @@
+// 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.
+
+// Package logger 基于 zerolog 的日志初始化，支持控制台输出（可读格式）+ 文件输出（JSON 结构化）。
+// 日志目录优先级：exe 同目录 log/ → %APPDATA%/SRemoteConnectClient/log/ → 当前工作目录 log/ → 仅控制台输出。
+package logger
+
+import (
+	"io"
+	"os"
+	"path/filepath"
+	"time"
+
+	"github.com/rs/zerolog"
+)
+
+// Log 包级全局日志实例，供外部包使用。
+var Log zerolog.Logger
+
+// Init 初始化全局日志。
+// 尝试在 exe 目录下创建 log/，若无写权限（多用户安装场景）则回退到 %APPDATA%。
+// 日志按日期命名，控制台使用可读格式（带颜色），文件使用 JSON 结构化格式。
+func Init() {
+	consoleWriter := zerolog.ConsoleWriter{
+		Out:        os.Stdout,
+		TimeFormat: "2006-01-02 15:04:05",
+		NoColor:    false,
+	}
+
+	logFile := openLogFile()
+	if logFile == nil {
+		// 连 %APPDATA% 都写不了，仅控制台输出
+		Log = zerolog.New(consoleWriter).
+			With().Timestamp().Caller().Logger().
+			Level(zerolog.DebugLevel)
+		return
+	}
+
+	multi := io.MultiWriter(consoleWriter, logFile)
+	Log = zerolog.New(multi).
+		With().Timestamp().Caller().Logger().
+		Level(zerolog.DebugLevel)
+}
+
+// openLogFile 按优先级尝试打开日志文件。
+// 返回 nil 表示无法创建日志文件，此时程序仅输出到控制台。
+func openLogFile() *os.File {
+	logDir, err := resolveLogDir()
+	if err != nil {
+		return nil
+	}
+
+	today := time.Now().Format("2006-01-02")
+	logPath := filepath.Join(logDir, today+".log")
+
+	f, err := os.OpenFile(logPath, os.O_CREATE|os.O_WRONLY|os.O_APPEND, 0644)
+	if err != nil {
+		return nil
+	}
+	return f
+}
+
+// resolveLogDir 解析日志目录。
+// 优先使用 exe 同目录下的 log/；若不可写则使用 %APPDATA%/SRemoteConnectClient/log/。
+// [PORTBARR]
+func resolveLogDir() (string, error) {
+	// 1. 尝试 exe 同目录
+	exePath, _ := os.Executable()
+	if exePath != "" {
+		dir := filepath.Join(filepath.Dir(exePath), "log")
+		if err := os.MkdirAll(dir, 0755); err == nil {
+			// 验证确实可写
+			testFile := filepath.Join(dir, ".write_test")
+			if f, err := os.Create(testFile); err == nil {
+				f.Close()
+				os.Remove(testFile)
+				return dir, nil
+			}
+		}
+	}
+
+	// 2. 回退到 %APPDATA%/SRemoteConnectClient/log
+	appData := os.Getenv("APPDATA")
+	if appData == "" {
+		appData = os.Getenv("LOCALAPPDATA")
+	}
+	if appData != "" {
+		dir := filepath.Join(appData, "SRemoteConnectClient", "log")
+		if err := os.MkdirAll(dir, 0755); err == nil {
+			return dir, nil
+		}
+	}
+
+	// 3. 最后尝试当前工作目录
+	cwd, _ := os.Getwd()
+	dir := filepath.Join(cwd, "log")
+	if err := os.MkdirAll(dir, 0755); err == nil {
+		return dir, nil
+	}
+
+	return "", os.ErrPermission
+}

win/dbgsole/dbgsole.go

@@ -0,0 +1,52 @@
+// 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.
+
+// console_windows 模块：通过 WinAPI（GetConsoleWindow + ShowWindowAsync）控制控制台窗口的显示/隐藏。
+package console
+
+import (
+	"syscall"
+
+	"github.com/rs/zerolog/log"
+)
+
+// 预加载 WinAPI 所需的 DLL 和函数。
+var (
+	kernel32         = syscall.NewLazyDLL("kernel32.dll")
+	user32           = syscall.NewLazyDLL("user32.dll")
+	getConsoleWindow = kernel32.NewProc("GetConsoleWindow")
+	showWindowAsync  = user32.NewProc("ShowWindowAsync")
+)
+
+// ShowWindow 常量。
+const (
+	SW_HIDE = 0 // 隐藏窗口
+	SW_SHOW = 5 // 显示窗口
+)
+
+// cachedConsoleHwnd 在 initConsoleHwnd() 中缓存，避免 systray.Run() 后 GetConsoleWindow 返回 0。
+var cachedConsoleHwnd uintptr
+
+// initConsoleHwnd 在 main() 中提前调用，缓存控制台窗口句柄。
+func initConsoleHwnd() {
+	hwnd, _, _ := getConsoleWindow.Call()
+	cachedConsoleHwnd = hwnd
+}
+
+// hideConsoleWindow 隐藏控制台窗口。若无控制台则无操作。
+func hideConsoleWindow() {
+	if cachedConsoleHwnd != 0 {
+		showWindowAsync.Call(cachedConsoleHwnd, SW_HIDE)
+		log.Debug().Msg("控制台窗口已隐藏")
+	}
+}
+
+// showConsoleWindow 显示控制台窗口。若无控制台则无操作。
+func showConsoleWindow() {
+	if cachedConsoleHwnd != 0 {
+		showWindowAsync.Call(cachedConsoleHwnd, SW_SHOW)
+		log.Debug().Msg("控制台窗口已显示")
+	}
+}

win/perm/perm.go

@@ -0,0 +1,96 @@
+// 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.
+
+// Package winperm 提供 Windows UAC 管理员权限检测与提权。
+package perm
+
+import (
+	"os"
+	"syscall"
+	"unsafe"
+
+	"gitee.com/snoware/swlib/logger"
+)
+
+// Permission 权限管理器，封装管理员检测和提权操作。
+type Permission struct{}
+
+// 预加载 shell32.dll 中的 Windows API。
+var (
+	shell32       = syscall.NewLazyDLL("shell32.dll")
+	isUserAnAdmin = shell32.NewProc("IsUserAnAdmin")
+	shellExecuteW = shell32.NewProc("ShellExecuteW")
+)
+
+// NewPermission 创建 Permission 实例。
+func NewPermission() *Permission {
+	return &Permission{}
+}
+
+// HasUAC 调用 IsUserAnAdmin 检查当前进程是否以管理员权限运行。
+// 返回 true 表示已拥有管理员权限。
+func (p *Permission) HasUAC() bool {
+	ret, _, _ := isUserAnAdmin.Call()
+	hasAdmin := ret != 0
+	logger.Log.Debug().Bool("has_admin", hasAdmin).Msg("检查管理员权限")
+	return hasAdmin
+}
+
+// RunExeAsAdmin 通过 ShellExecuteW("runas") 以管理员身份启动指定的可执行文件。
+//
+// 参数 exe: 目标可执行文件的绝对路径。
+// 成功时当前进程应退出，由新的管理员进程接管。
+func (p *Permission) RunExeAsAdmin(exe string) error {
+	logger.Log.Info().Str("exe", exe).Msg("尝试以管理员身份运行")
+
+	verb := "runas"
+	cwd, _ := os.Getwd()
+
+	verbPtr, _ := syscall.UTF16PtrFromString(verb)
+	exePtr, _ := syscall.UTF16PtrFromString(exe)
+	cwdPtr, _ := syscall.UTF16PtrFromString(cwd)
+
+	var showCmd int32 = 1 // SW_NORMAL
+
+	ret, _, _ := shellExecuteW.Call(
+		0,
+		uintptr(unsafe.Pointer(verbPtr)),
+		uintptr(unsafe.Pointer(exePtr)),
+		0,
+		uintptr(unsafe.Pointer(cwdPtr)),
+		uintptr(showCmd),
+	)
+
+	if ret <= 32 {
+		err := syscall.GetLastError()
+		logger.Log.Error().Err(err).Int32("ret", int32(ret)).Msg("管理员提权失败")
+		return err
+	}
+
+	logger.Log.Info().Msg("管理员提权成功，当前进程将退出")
+	return nil
+}
+
+// GetSelf 获取当前可执行文件的绝对路径。
+func (p *Permission) GetSelf() (string, error) {
+	exe, err := os.Executable()
+	if err != nil {
+		logger.Log.Error().Err(err).Msg("获取自身可执行文件路径失败")
+		return "", err
+	}
+	logger.Log.Debug().Str("exe_path", exe).Msg("获取自身路径")
+	return exe, nil
+}
+
+// RunSelfAsAdmin 以管理员身份重新执行自身。
+// 返回当前可执行文件路径和可能的错误。
+func (p *Permission) RunSelfAsAdmin() (string, error) {
+	exe, err := p.GetSelf()
+	if err != nil {
+		return "", err
+	}
+	err = p.RunExeAsAdmin(exe)
+	return exe, err
+}

win/trayicon/trayicon.go

@@ -0,0 +1,103 @@
+// 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.
+
+// trayicon_windows 模块：托盘图标加载，将 PNG 解码、缩放并包装为 Windows ICO 格式。
+// getlantern/systray 在 Windows 上存在 GDI+ → CreateIconIndirect 兼容性 bug，
+// 解法是将 PNG 包装成 ICO 格式（现代 Windows 原生支持 ICO 内嵌 PNG）。
+package main
+
+import (
+	"bytes"
+	"encoding/binary"
+	"image"
+	"image/png"
+	"os"
+	"path/filepath"
+
+	"github.com/rs/zerolog/log"
+)
+
+// loadIcon 加载托盘图标。
+// 优先从 image.png 文件读取，回退到内嵌 iconData。
+// 解码 PNG → 缩放到 32x32 → 重新编码 → 包装为 ICO 格式返回。
+func loadIcon(iconData []byte) []byte {
+	raw := iconData
+
+	exePath, err := os.Executable()
+	if err == nil {
+		iconFile := filepath.Join(filepath.Dir(exePath), "image.png")
+		if data, err := os.ReadFile(iconFile); err == nil {
+			raw = data
+			log.Debug().Str("file", iconFile).Msg("从文件读取图标")
+		}
+	}
+
+	img, err := png.Decode(bytes.NewReader(raw))
+	if err != nil {
+		log.Warn().Err(err).Msg("PNG 解码失败")
+		return nil
+	}
+
+	icon := resizeIcon(img, 32, 32)
+
+	var pngBuf bytes.Buffer
+	if err := png.Encode(&pngBuf, icon); err != nil {
+		log.Warn().Err(err).Msg("PNG 重新编码失败")
+		return nil
+	}
+
+	ico := pngToICO(pngBuf.Bytes(), 32, 32)
+	log.Debug().Int("size", len(ico)).Msg("托盘图标已生成 (ICO 32x32)")
+	return ico
+}
+
+// pngToICO 将 PNG 字节数据包装为 Windows ICO 文件格式。
+// ICO 结构：6 字节文件头 + 16 字节目录项 + PNG 图像数据。
+func pngToICO(pngData []byte, w, h int) []byte {
+	var buf bytes.Buffer
+
+	// ICO 文件头
+	binary.Write(&buf, binary.LittleEndian, uint16(0)) // 保留
+	binary.Write(&buf, binary.LittleEndian, uint16(1)) // 类型: ICO
+	binary.Write(&buf, binary.LittleEndian, uint16(1)) // 图标数量
+
+	// 处理 256 边界的宽度/高度
+	width := uint8(w)
+	if w >= 256 {
+		width = 0
+	}
+	height := uint8(h)
+	if h >= 256 {
+		height = 0
+	}
+
+	// ICO 目录项
+	binary.Write(&buf, binary.LittleEndian, width)
+	binary.Write(&buf, binary.LittleEndian, height)
+	binary.Write(&buf, binary.LittleEndian, uint8(0))
+	binary.Write(&buf, binary.LittleEndian, uint8(0))
+	binary.Write(&buf, binary.LittleEndian, uint16(1))
+	binary.Write(&buf, binary.LittleEndian, uint16(32))
+	binary.Write(&buf, binary.LittleEndian, uint32(len(pngData)))
+	binary.Write(&buf, binary.LittleEndian, uint32(22))
+
+	buf.Write(pngData)
+	return buf.Bytes()
+}
+
+// resizeIcon 使用最近邻插值将图像缩放到指定尺寸。
+// 返回 *image.RGBA 避免与原图共享底层数据。
+func resizeIcon(src image.Image, w, h int) *image.RGBA {
+	dst := image.NewRGBA(image.Rect(0, 0, w, h))
+	sb := src.Bounds()
+	for y := range h {
+		for x := range w {
+			sx := sb.Min.X + x*sb.Dx()/w
+			sy := sb.Min.Y + y*sb.Dy()/h
+			dst.Set(x, y, src.At(sx, sy))
+		}
+	}
+	return dst
+}