1551bd5dad
Emberwake spread/waterhole UI, campaign DB, spread handler, spread-kit web publisher, and SPREAD_TECHNIQUES doc. Crucible Phase A-C: expanded ops, port-forward matrix, remote dir browser, crucible help/tests. Linux agent hardening: credential vault, persistence audit, firewall/defender deploy, SMB spread status, CPU stats, screenshots/crypt/file-ops split. Docker compose and agent/server images with e2e validation script and docs. Musical dashboard: ambient music player, hover SFX, SoundContext/AmbientMusicContext, steampunk polish. Public builds API, dropper handler updates, SessionGate and fleet UX. README and PROBLEMS.md refresh.
54 lines
1.1 KiB
Go
54 lines
1.1 KiB
Go
package stats
|
|
|
|
import (
|
|
"unsafe"
|
|
)
|
|
|
|
var (
|
|
procGetSystemTimes = kernel32.NewProc("GetSystemTimes")
|
|
)
|
|
|
|
type filetime struct {
|
|
LowDateTime uint32
|
|
HighDateTime uint32
|
|
}
|
|
|
|
func filetimeToUint64(ft filetime) uint64 {
|
|
return (uint64(ft.HighDateTime) << 32) | uint64(ft.LowDateTime)
|
|
}
|
|
|
|
func (r *Reporter) SystemCPUPercent() float64 {
|
|
r.mu.Lock()
|
|
defer r.mu.Unlock()
|
|
|
|
var idle, kernel, user filetime
|
|
ret, _, _ := procGetSystemTimes.Call(
|
|
uintptr(unsafe.Pointer(&idle)),
|
|
uintptr(unsafe.Pointer(&kernel)),
|
|
uintptr(unsafe.Pointer(&user)),
|
|
)
|
|
if ret == 0 {
|
|
return 0
|
|
}
|
|
|
|
idleTicks := filetimeToUint64(idle)
|
|
kernelTicks := filetimeToUint64(kernel)
|
|
userTicks := filetimeToUint64(user)
|
|
|
|
if !r.hasSample {
|
|
r.lastIdle = idleTicks
|
|
r.lastKernel = kernelTicks
|
|
r.lastUser = userTicks
|
|
r.hasSample = true
|
|
return 0
|
|
}
|
|
|
|
idleDelta := float64(idleTicks - r.lastIdle)
|
|
totalDelta := float64((kernelTicks - r.lastKernel) + (userTicks - r.lastUser))
|
|
r.lastIdle = idleTicks
|
|
r.lastKernel = kernelTicks
|
|
r.lastUser = userTicks
|
|
|
|
return cpuBusyPercentFromDeltas(idleDelta, totalDelta)
|
|
}
|