48c44c29ab
Updates #2763. Squashed commit of the following: commit bd2077c6569b53ae341a58aa73de6063d7037e8e Author: Ainar Garipov <A.Garipov@AdGuard.COM> Date: Fri Jun 4 16:25:17 2021 +0300 all: move rlimit_nofile, imp docs commit ba95d4ab7c722bf83300d626a598aface37539ad Author: Ainar Garipov <A.Garipov@AdGuard.COM> Date: Fri Jun 4 15:12:23 2021 +0300 all: support setgid, setuid on unix
76 lines
2.0 KiB
Go
76 lines
2.0 KiB
Go
// Package aghos contains utilities for functions requiring system calls.
|
|
package aghos
|
|
|
|
import (
|
|
"fmt"
|
|
"os/exec"
|
|
"runtime"
|
|
"syscall"
|
|
)
|
|
|
|
// UnsupportedError is returned by functions and methods when a particular
|
|
// operation Op cannot be performed on the current OS.
|
|
type UnsupportedError struct {
|
|
Op string
|
|
OS string
|
|
}
|
|
|
|
// Error implements the error interface for *UnsupportedError.
|
|
func (err *UnsupportedError) Error() (msg string) {
|
|
return fmt.Sprintf("%s is unsupported on %s", err.Op, err.OS)
|
|
}
|
|
|
|
// Unsupported is a helper that returns an *UnsupportedError with the Op field
|
|
// set to op and the OS field set to the current OS.
|
|
func Unsupported(op string) (err error) {
|
|
return &UnsupportedError{
|
|
Op: op,
|
|
OS: runtime.GOOS,
|
|
}
|
|
}
|
|
|
|
// CanBindPrivilegedPorts checks if current process can bind to privileged
|
|
// ports.
|
|
func CanBindPrivilegedPorts() (can bool, err error) {
|
|
return canBindPrivilegedPorts()
|
|
}
|
|
|
|
// SetRlimit sets user-specified limit of how many fd's we can use.
|
|
//
|
|
// See https://github.com/AdguardTeam/AdGuardHome/internal/issues/659.
|
|
func SetRlimit(val uint64) (err error) {
|
|
return setRlimit(val)
|
|
}
|
|
|
|
// HaveAdminRights checks if the current user has root (administrator) rights.
|
|
func HaveAdminRights() (bool, error) {
|
|
return haveAdminRights()
|
|
}
|
|
|
|
// SendProcessSignal sends signal to a process.
|
|
func SendProcessSignal(pid int, sig syscall.Signal) error {
|
|
return sendProcessSignal(pid, sig)
|
|
}
|
|
|
|
// MaxCmdOutputSize is the maximum length of performed shell command output.
|
|
const MaxCmdOutputSize = 2 * 1024
|
|
|
|
// RunCommand runs shell command.
|
|
func RunCommand(command string, arguments ...string) (int, string, error) {
|
|
cmd := exec.Command(command, arguments...)
|
|
out, err := cmd.Output()
|
|
if len(out) > MaxCmdOutputSize {
|
|
out = out[:MaxCmdOutputSize]
|
|
}
|
|
if err != nil {
|
|
return 1, "", fmt.Errorf("exec.Command(%s) failed: %v: %s", command, err, string(out))
|
|
}
|
|
|
|
return cmd.ProcessState.ExitCode(), string(out), nil
|
|
}
|
|
|
|
// IsOpenWrt returns true if host OS is OpenWrt.
|
|
func IsOpenWrt() (ok bool) {
|
|
return isOpenWrt()
|
|
}
|