feat: Implement a check if clid run using admin privileges

Author: noxymon

go.mod

@@ -6,6 +6,7 @@ require (
 	github.com/baneeishaque/adoptium_jdk_go v0.0.0-20221214163615-0b2cfe72098d
 	github.com/codegangsta/cli v1.20.0
 	github.com/tucnak/store v0.0.0-20170905113834-b02ecdcc6dfb
+	golang.org/x/sys v0.40.0
 	gopkg.in/cheggaaa/pb.v1 v1.0.28
 )
 
@@ -18,7 +19,6 @@ require (
 	github.com/itchyny/timefmt-go v0.1.7 // indirect
 	github.com/mattn/go-colorable v0.1.12 // indirect
 	github.com/mattn/go-runewidth v0.0.19 // indirect
-	golang.org/x/sys v0.40.0 // indirect
 	gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c // indirect
 	gopkg.in/yaml.v2 v2.4.0 // indirect
 )

internal/cmdCli/init.go

@@ -1,13 +1,15 @@
 package cmdCli
 
 import (
+	"errors"
 	"fmt"
 	"os"
 	"os/exec"
 	"path/filepath"
 
 	"github.com/codegangsta/cli"
 	"github.com/ystyle/jvms/internal/entity"
+	"github.com/ystyle/jvms/utils/admin"
 	"github.com/ystyle/jvms/utils/file"
 )
 
@@ -29,6 +31,9 @@ func init_(config *entity.Config) *cli.Command {
 			},
 		},
 		Action: func(c *cli.Context) error {
+			if !admin.IsAdmin() {
+				return errors.New("jvms init requires administrator privileges. Please run as administrator")
+			}
 			if c.IsSet("java_home") || config.JavaHome == "" {
 				config.JavaHome = c.String("java_home")
 			}

internal/cmdCli/switch.go

@@ -11,6 +11,7 @@ import (
 
 	"github.com/codegangsta/cli"
 	"github.com/ystyle/jvms/internal/entity"
+	"github.com/ystyle/jvms/utils/admin"
 	"github.com/ystyle/jvms/utils/file"
 	"github.com/ystyle/jvms/utils/jdk"
 )
@@ -43,6 +44,9 @@ func switch_(config *entity.Config) *cli.Command {
 // SwitchFunc is used by both switch and use commands
 func switchFunc(config *entity.Config) func(*cli.Context) error {
 	return func(c *cli.Context) error {
+		if !admin.IsAdmin() {
+			return errors.New("jvms switch requires administrator privileges. Please run as administrator")
+		}
 		v := strings.TrimSpace(c.Args().Get(0))
 		if v == "" {
 			return errors.New("you should input a version or index number, Type \"jvms list\" to see what is installed")

utils/admin/admin.go

@@ -0,0 +1,34 @@
+package admin
+
+import (
+	"golang.org/x/sys/windows"
+)
+
+// IsAdmin checks if the current process has administrative privileges
+func IsAdmin() bool {
+	var sid *windows.SID
+
+	// Although this looks huge, it's the equivalent of
+	// set "sid=S-1-5-32-544"
+	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 {
+		return false
+	}
+	defer windows.FreeSid(sid)
+
+	// This appears to cast a null pointer, but that is actually fine.
+	// CheckTokenMembership allows the token handle to be 0 (current process token).
+	// It's checked against the sid created above.
+	token := windows.Token(0)
+	member, err := token.IsMember(sid)
+	if err != nil {
+		return false
+	}
+	return member
+}