feat(remote): 自 go-hua/remote 并入远程控制服务管理
- RDP/VNC(TightVNC)/Radmin 服务状态查询与启停重置、VNC 安装/配置读写/密码 加解密、GetRemoteInfo 汇总、CAD/SAS 注册表开关 - 与本库 file/shell/process 同属被控端远程运维能力; 新增依赖 kardianos/service、KarpelesLab/vncpasswd - 附 remote_test.go 与包 README(含平台支持矩阵与权限注意)
This commit is contained in:
+192
@@ -0,0 +1,192 @@
|
||||
//go:build windows
|
||||
|
||||
package remote
|
||||
|
||||
import (
|
||||
"github.com/KarpelesLab/vncpasswd"
|
||||
"github.com/kardianos/service"
|
||||
"golang.org/x/sys/windows/registry"
|
||||
)
|
||||
|
||||
var VncService service.Service
|
||||
|
||||
// ReadVncServerConfig 从注册表读取 TightVNC 服务端配置
|
||||
func ReadVncServerConfig() (VncServerConfig, error) {
|
||||
config := VncServerConfig{
|
||||
ExtraPorts: "",
|
||||
QueryTimeout: 0,
|
||||
QueryAcceptOnTimeout: 0,
|
||||
LocalInputPriorityTimeout: 0,
|
||||
LocalInputPriority: 0,
|
||||
BlockRemoteInput: 0,
|
||||
BlockLocalInput: 0,
|
||||
IpAccessControl: "",
|
||||
RfbPort: 0,
|
||||
HttpPort: 0,
|
||||
DisconnectAction: 0,
|
||||
AcceptRfbConnections: 0,
|
||||
UseVncAuthentication: 0,
|
||||
UseControlAuthentication: 0,
|
||||
RepeatControlAuthentication: 0,
|
||||
LoopbackOnly: 0,
|
||||
AcceptHttpConnections: 0,
|
||||
LogLevel: 0,
|
||||
EnableFileTransfers: 0,
|
||||
RemoveWallpaper: 0,
|
||||
UseD3D: 0,
|
||||
UseMirrorDriver: 0,
|
||||
EnableUrlParams: 0,
|
||||
Password: []byte{},
|
||||
AlwaysShared: 0,
|
||||
NeverShared: 0,
|
||||
DisconnectClients: 0,
|
||||
PollingInterval: 0,
|
||||
AllowLoopback: 0,
|
||||
VideoRecognitionInterval: 0,
|
||||
GrabTransparentWindows: 0,
|
||||
SaveLogToAllUsersPath: 0,
|
||||
RunControlInterface: 0,
|
||||
IdleTimeout: 0,
|
||||
VideoClasses: "",
|
||||
VideoRects: "",
|
||||
ConnectToRdp: 0,
|
||||
}
|
||||
|
||||
key, err := registry.OpenKey(registry.LOCAL_MACHINE, "SOFTWARE\\TightVNC\\Server", registry.ALL_ACCESS)
|
||||
if err != nil {
|
||||
return config, err
|
||||
}
|
||||
defer key.Close()
|
||||
|
||||
config.ExtraPorts, _, _ = key.GetStringValue("ExtraPorts")
|
||||
config.QueryTimeout, _, _ = GetDwordValue(key, "QueryTimeout")
|
||||
config.QueryAcceptOnTimeout, _, _ = GetDwordValue(key, "QueryAcceptOnTimeout")
|
||||
config.LocalInputPriorityTimeout, _, _ = GetDwordValue(key, "LocalInputPriorityTimeout")
|
||||
config.LocalInputPriority, _, _ = GetDwordValue(key, "LocalInputPriority")
|
||||
config.BlockRemoteInput, _, _ = GetDwordValue(key, "BlockRemoteInput")
|
||||
config.BlockLocalInput, _, _ = GetDwordValue(key, "BlockLocalInput")
|
||||
config.IpAccessControl, _, _ = key.GetStringValue("IpAccessControl")
|
||||
config.RfbPort, _, _ = GetDwordValue(key, "RfbPort")
|
||||
config.HttpPort, _, _ = GetDwordValue(key, "HttpPort")
|
||||
config.DisconnectAction, _, _ = GetDwordValue(key, "DisconnectAction")
|
||||
config.AcceptRfbConnections, _, _ = GetDwordValue(key, "AcceptRfbConnections")
|
||||
config.UseVncAuthentication, _, _ = GetDwordValue(key, "UseVncAuthentication")
|
||||
config.UseControlAuthentication, _, _ = GetDwordValue(key, "UseControlAuthentication")
|
||||
config.RepeatControlAuthentication, _, _ = GetDwordValue(key, "RepeatControlAuthentication")
|
||||
config.LoopbackOnly, _, _ = GetDwordValue(key, "LoopbackOnly")
|
||||
config.AcceptHttpConnections, _, _ = GetDwordValue(key, "AcceptHttpConnections")
|
||||
config.EnableFileTransfers, _, _ = GetDwordValue(key, "EnableFileTransfers")
|
||||
config.RemoveWallpaper, _, _ = GetDwordValue(key, "RemoveWallpaper")
|
||||
config.UseD3D, _, _ = GetDwordValue(key, "UseD3D")
|
||||
config.UseMirrorDriver, _, _ = GetDwordValue(key, "UseMirrorDriver")
|
||||
config.EnableUrlParams, _, _ = GetDwordValue(key, "EnableUrlParams")
|
||||
config.Password, _, _ = key.GetBinaryValue("Password")
|
||||
config.AlwaysShared, _, _ = GetDwordValue(key, "AlwaysShared")
|
||||
config.NeverShared, _, _ = GetDwordValue(key, "NeverShared")
|
||||
config.DisconnectClients, _, _ = GetDwordValue(key, "DisconnectClients")
|
||||
config.PollingInterval, _, _ = GetDwordValue(key, "PollingInterval")
|
||||
config.AllowLoopback, _, _ = GetDwordValue(key, "AllowLoopback")
|
||||
config.VideoRecognitionInterval, _, _ = GetDwordValue(key, "VideoRecognitionInterval")
|
||||
config.GrabTransparentWindows, _, _ = GetDwordValue(key, "GrabTransparentWindows")
|
||||
config.SaveLogToAllUsersPath, _, _ = GetDwordValue(key, "SaveLogToAllUsersPath")
|
||||
config.RunControlInterface, _, _ = GetDwordValue(key, "RunControlInterface")
|
||||
config.IdleTimeout, _, _ = GetDwordValue(key, "IdleTimeout")
|
||||
config.VideoClasses, _, _ = key.GetStringValue("VideoClasses")
|
||||
config.VideoRects, _, _ = key.GetStringValue("VideoRects")
|
||||
config.ConnectToRdp, _, _ = GetDwordValue(key, "ExtraPorts")
|
||||
|
||||
return config, nil
|
||||
}
|
||||
|
||||
// GetDwordValue 读取 REG_DWORD 类型注册表值
|
||||
func GetDwordValue(k registry.Key, name string) (val uint32, valtype uint32, err error) {
|
||||
val64 := uint64(0)
|
||||
val64, valtype, err = k.GetIntegerValue(name)
|
||||
val = uint32(val64)
|
||||
return
|
||||
}
|
||||
|
||||
var vncSvcConfig = &service.Config{
|
||||
Name: "tvnserver",
|
||||
DisplayName: "TightVNC Server",
|
||||
Description: "",
|
||||
Arguments: []string{"-service"},
|
||||
}
|
||||
|
||||
// InstallVncServer 安装VNC服务端 返回:结果,密码,错误
|
||||
func InstallVncServer() (bool, string, error) {
|
||||
//安装服务
|
||||
//导入注册表
|
||||
//生成随机密码
|
||||
return true, "12345678", nil
|
||||
}
|
||||
|
||||
// VncInit 初始化 VNC 服务对象，VncInstall/VncStart 等操作前必须调用
|
||||
func VncInit() error {
|
||||
var err error
|
||||
VncService, err = service.New(nil, vncSvcConfig)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// VncInstall 安装 TightVNC 服务
|
||||
func VncInstall() error {
|
||||
err := VncService.Install()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// VncUninstall 停止并卸载 TightVNC 服务
|
||||
func VncUninstall() error {
|
||||
_ = VncService.Stop()
|
||||
err := VncService.Uninstall()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// VncStart 启动 TightVNC 服务
|
||||
func VncStart() error {
|
||||
err := VncService.Start()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// VncStop 停止 TightVNC 服务
|
||||
func VncStop() error {
|
||||
err := VncService.Stop()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// VncRestart 重启 TightVNC 服务
|
||||
func VncRestart() error {
|
||||
err := VncService.Stop()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
err = VncService.Start()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// VncPasswordCrypt 加密 VNC 密码（TightVNC 注册表存储格式）
|
||||
func VncPasswordCrypt(password string) []byte {
|
||||
return vncpasswd.Crypt(password)
|
||||
}
|
||||
|
||||
// VncPasswordDecrypt 解密 VNC 密码
|
||||
func VncPasswordDecrypt(data []byte) string {
|
||||
return vncpasswd.Decrypt(data)
|
||||
}
|
||||
Reference in New Issue
Block a user