mirror of
https://github.com/fankes/komari-agent.git
synced 2025-10-19 02:59:23 +08:00
Refactor monitoring package: remove platform-specific files and consolidate OS detection and process counting logic
- Deleted os_windows.go and process_windows.go, replacing them with platform-agnostic implementations in unit directory. - Removed Linux-specific process counting logic from process_linux.go and integrated it into unit. - Consolidated uptime and OS name retrieval into unit files for better organization. - Updated update mechanism to use global variables for current version and repository. - Introduced command-line flags for configuration, including disabling auto-update and web SSH. - Implemented WebSocket connection handling and terminal interaction for both Unix and Windows systems. - Added basic info upload functionality to server package, enhancing monitoring capabilities.
This commit is contained in:
56
monitoring/unit/process_windows.go
Normal file
56
monitoring/unit/process_windows.go
Normal file
@@ -0,0 +1,56 @@
|
||||
//go:build windows
|
||||
// +build windows
|
||||
|
||||
package monitoring
|
||||
|
||||
import (
|
||||
"syscall"
|
||||
"unsafe"
|
||||
)
|
||||
|
||||
// ProcessCount returns the number of running processes
|
||||
func ProcessCount() (count int) {
|
||||
return processCountWindows()
|
||||
|
||||
}
|
||||
|
||||
// processCountWindows counts processes using Windows API
|
||||
func processCountWindows() (count int) {
|
||||
// Load kernel32.dll
|
||||
kernel32, err := syscall.LoadLibrary("kernel32.dll")
|
||||
if err != nil {
|
||||
return 0
|
||||
}
|
||||
defer syscall.FreeLibrary(kernel32)
|
||||
|
||||
// Get EnumProcesses function
|
||||
enumProcesses, err := syscall.GetProcAddress(kernel32, "K32EnumProcesses")
|
||||
if err != nil {
|
||||
return 0
|
||||
}
|
||||
|
||||
// Prepare buffer for process IDs
|
||||
const maxProcesses = 1024
|
||||
pids := make([]uint32, maxProcesses)
|
||||
var bytesReturned uint32
|
||||
|
||||
// Call EnumProcesses
|
||||
ret, _, _ := syscall.SyscallN(
|
||||
uintptr(enumProcesses),
|
||||
uintptr(unsafe.Pointer(&pids[0])),
|
||||
uintptr(len(pids)*4),
|
||||
uintptr(unsafe.Pointer(&bytesReturned)),
|
||||
)
|
||||
|
||||
if ret == 0 {
|
||||
return 0
|
||||
}
|
||||
|
||||
// Count valid PIDs
|
||||
count = int(bytesReturned) / 4 // bytesReturned is size in bytes, divide by 4 for uint32 count
|
||||
if count > maxProcesses {
|
||||
count = maxProcesses
|
||||
}
|
||||
|
||||
return count
|
||||
}
|
Reference in New Issue
Block a user