mirror of
https://github.com/Arvanaghi/CheckPlease
synced 2026-06-08 10:25:16 +00:00
Brandon Arvanaghi
82 lines
2.0 KiB
Go
82 lines
2.0 KiB
Go
/*
|
|
Ensures there are more than N processes currently running on the system (default: 50), Go
|
|
Module written by Brandon Arvanaghi
|
|
Website: arvanaghi.com
|
|
Twitter: @arvanaghi
|
|
*/
|
|
|
|
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"strconv"
|
|
"unsafe"
|
|
"syscall"
|
|
"os"
|
|
)
|
|
|
|
var (
|
|
kernel32 = syscall.NewLazyDLL("kernel32.dll")
|
|
CreateToolhelp32Snapshot = kernel32.NewProc("CreateToolhelp32Snapshot")
|
|
Process32First = kernel32.NewProc("Process32FirstW")
|
|
Process32Next = kernel32.NewProc("Process32NextW")
|
|
CloseHandle = kernel32.NewProc("CloseHandle")
|
|
)
|
|
|
|
type PROCESSENTRY32 struct {
|
|
dwSize uint32
|
|
cntUsage uint32
|
|
th32ProcessID uint32
|
|
th32DefaultHeapID uintptr
|
|
th32ModuleID uint32
|
|
cntThreads uint32
|
|
th32ParentProcessID uint32
|
|
pcPriClassBase int32
|
|
dwFlags uint32
|
|
szExeFile [260]uint16
|
|
}
|
|
|
|
func main() {
|
|
minRunningProcesses := 50
|
|
|
|
if (len(os.Args) == 2) {
|
|
minRunningProcesses, _ = strconv.Atoi(os.Args[1])
|
|
}
|
|
|
|
// TH32CS_SNAPPROCESS == 0x00000002, meaning snapshot all processes
|
|
hProcessSnap, _, _ := CreateToolhelp32Snapshot.Call(2,0)
|
|
if hProcessSnap < 0 {
|
|
fmt.Println("[---] Unable to create Snapshot, exiting.")
|
|
os.Exit(-1)
|
|
}
|
|
defer CloseHandle.Call(hProcessSnap)
|
|
|
|
exeNames := make([]string, 0, 100)
|
|
var pe32 PROCESSENTRY32
|
|
pe32.dwSize = uint32(unsafe.Sizeof(pe32))
|
|
|
|
Process32First.Call(hProcessSnap, uintptr(unsafe.Pointer(&pe32)))
|
|
|
|
for {
|
|
|
|
exeNames = append(exeNames, syscall.UTF16ToString(pe32.szExeFile[:260]))
|
|
|
|
retVal, _, _ := Process32Next.Call(hProcessSnap, uintptr(unsafe.Pointer(&pe32)))
|
|
if retVal == 0 {
|
|
break
|
|
}
|
|
|
|
}
|
|
|
|
runningProcesses := 0
|
|
for range exeNames {
|
|
runningProcesses += 1
|
|
}
|
|
|
|
if (runningProcesses >= minRunningProcesses) {
|
|
fmt.Printf("There are %d processes running on the system, which satisfies the minimum you set of %d. Proceed!\n", runningProcesses, minRunningProcesses)
|
|
} else {
|
|
fmt.Printf("Only %d processes are running on the system, which is less than the minimum you set of %d. Do not proceed.\n", runningProcesses, minRunningProcesses)
|
|
}
|
|
|
|
} |