mirror of
https://github.com/fatedier/frp.git
synced 2024-12-15 02:38:59 +08:00
47 lines
1.2 KiB
Go
47 lines
1.2 KiB
Go
|
// +build windows
|
||
|
// +build go1.4
|
||
|
|
||
|
package mousetrap
|
||
|
|
||
|
import (
|
||
|
"os"
|
||
|
"syscall"
|
||
|
"unsafe"
|
||
|
)
|
||
|
|
||
|
func getProcessEntry(pid int) (*syscall.ProcessEntry32, error) {
|
||
|
snapshot, err := syscall.CreateToolhelp32Snapshot(syscall.TH32CS_SNAPPROCESS, 0)
|
||
|
if err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
defer syscall.CloseHandle(snapshot)
|
||
|
var procEntry syscall.ProcessEntry32
|
||
|
procEntry.Size = uint32(unsafe.Sizeof(procEntry))
|
||
|
if err = syscall.Process32First(snapshot, &procEntry); err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
for {
|
||
|
if procEntry.ProcessID == uint32(pid) {
|
||
|
return &procEntry, nil
|
||
|
}
|
||
|
err = syscall.Process32Next(snapshot, &procEntry)
|
||
|
if err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// StartedByExplorer returns true if the program was invoked by the user double-clicking
|
||
|
// on the executable from explorer.exe
|
||
|
//
|
||
|
// It is conservative and returns false if any of the internal calls fail.
|
||
|
// It does not guarantee that the program was run from a terminal. It only can tell you
|
||
|
// whether it was launched from explorer.exe
|
||
|
func StartedByExplorer() bool {
|
||
|
pe, err := getProcessEntry(os.Getppid())
|
||
|
if err != nil {
|
||
|
return false
|
||
|
}
|
||
|
return "explorer.exe" == syscall.UTF16ToString(pe.ExeFile[:])
|
||
|
}
|