code refactoring

permission/README.md

@@ -0,0 +1,2 @@
+# permissionutils
+The package contains various helpers about permissions/privileges

permission/error.go

@@ -0,0 +1,5 @@
+package permissionutil
+
+import "errors"
+
+var ErrNotImplemented = errors.New("not implemented")

permission/permission.go

@@ -0,0 +1,11 @@
+package permissionutil
+
+var (
+	IsRoot       bool
+	HasCapNetRaw bool
+)
+
+func init() {
+	IsRoot, _ = checkCurrentUserRoot()
+	HasCapNetRaw, _ = checkCurrentUserCapNetRaw()
+}

permission/permission_darwin.go

@@ -0,0 +1,17 @@
+//go:build darwin
+
+package permissionutil
+
+import (
+	"os"
+)
+
+// checkCurrentUserRoot checks if the current user is root
+func checkCurrentUserRoot() (bool, error) {
+	return os.Geteuid() == 0, nil
+}
+
+// checkCurrentUserCapNetRaw checks if the current user has the CAP_NET_RAW capability
+func checkCurrentUserCapNetRaw() (bool, error) {
+	return false, ErrNotImplemented
+}

permission/permission_linux.go

@@ -0,0 +1,42 @@
+//go:build linux || unix
+
+package permissionutil
+
+import (
+	"os"
+	"runtime"
+
+	"github.com/projectdiscovery/naabu/v2/pkg/israce"
+	"golang.org/x/sys/unix"
+)
+
+// checkCurrentUserRoot checks if the current user is root
+func checkCurrentUserRoot() (bool, error) {
+	return os.Geteuid() == 0, nil
+}
+
+// checkCurrentUserCapNetRaw checks if the current user has the CAP_NET_RAW capability
+func checkCurrentUserCapNetRaw() (bool, error) {
+	if raceutil.Enabled {
+		return false, errors.New("race detector enabled")
+	}
+	// runtime.LockOSThread interferes with race detection
+	header := unix.CapUserHeader{
+		Version: unix.LINUX_CAPABILITY_VERSION_3,
+		Pid:     int32(os.Getpid()),
+	}
+	data := unix.CapUserData{}
+	runtime.LockOSThread()
+	defer runtime.UnlockOSThread()
+
+	err := unix.Capget(&header, &data)
+	if err != nil {
+		return false, err
+	}
+	data.Inheritable = (1 << unix.CAP_NET_RAW)
+	err := unix.Capset(&header, &data)
+	if err != nil {
+		return false, err
+	}
+	return true, nil
+}

permission/permission_win.go

@@ -0,0 +1,13 @@
+//go:build windows
+
+package permissionutil
+
+// checkCurrentUserRoot on windows is not implemented
+func checkCurrentUserRoot() (bool, error) {
+	return false, ErrNotImplemented
+}
+
+// checkCurrentUserCapNetRaw on windows is not implemented
+func checkCurrentUserCapNetRaw() (bool, error) {
+	return false, ErrNotImplemented
+}