mirror of
https://github.com/JanDeDobbeleer/oh-my-posh.git
synced 2024-09-20 05:27:31 -07:00
37 lines
1,004 B
Go
37 lines
1,004 B
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"log"
|
||
|
|
|
||
|
|
"golang.org/x/sys/windows"
|
||
|
|
)
|
||
|
|
|
||
|
|
func (env *environment) isRunningAsRoot() bool {
|
||
|
|
var sid *windows.SID
|
||
|
|
// Although this looks scary, it is directly copied from the
|
||
|
|
// official windows documentation. The Go API for this is a
|
||
|
|
// direct wrap around the official C++ API.
|
||
|
|
// See https://docs.microsoft.com/en-us/windows/desktop/api/securitybaseapi/nf-securitybaseapi-checktokenmembership
|
||
|
|
err := windows.AllocateAndInitializeSid(
|
||
|
|
&windows.SECURITY_NT_AUTHORITY,
|
||
|
|
2,
|
||
|
|
windows.SECURITY_BUILTIN_DOMAIN_RID,
|
||
|
|
windows.DOMAIN_ALIAS_RID_ADMINS,
|
||
|
|
0, 0, 0, 0, 0, 0,
|
||
|
|
&sid)
|
||
|
|
if err != nil {
|
||
|
|
log.Fatalf("SID Error: %s", err)
|
||
|
|
return false
|
||
|
|
}
|
||
|
|
|
||
|
|
// This appears to cast a null pointer so I'm not sure why this
|
||
|
|
// works, but this guy says it does and it Works for Me™:
|
||
|
|
// https://github.com/golang/go/issues/28804#issuecomment-438838144
|
||
|
|
token := windows.Token(0)
|
||
|
|
member, err := token.IsMember(sid)
|
||
|
|
if err != nil {
|
||
|
|
return false
|
||
|
|
}
|
||
|
|
return member && token.IsElevated()
|
||
|
|
}
|