mirror of
https://github.com/taigrr/systemctl.git
synced 2026-04-02 02:28:50 -07:00
Compare commits
12 Commits
v1.0.7
...
cd/ci-impr
| Author | SHA1 | Date | |
|---|---|---|---|
| c966da674a | |||
| 1b451ee8b2 | |||
| adf3c36632 | |||
| 22132919e5 | |||
| d38136c0dc | |||
|
|
14c9f0f70d | ||
| 5f1537f8bc | |||
|
|
d38d347cc6 | ||
| 14a2ca2acd | |||
| 451a949ace | |||
| 21fce7918e | |||
| 54f4f7a235 |
55
.github/workflows/ci.yml
vendored
Normal file
55
.github/workflows/ci.yml
vendored
Normal file
@@ -0,0 +1,55 @@
|
||||
name: CI
|
||||
|
||||
on:
|
||||
push:
|
||||
branches: [master]
|
||||
pull_request:
|
||||
branches: [master]
|
||||
|
||||
permissions:
|
||||
contents: read
|
||||
|
||||
jobs:
|
||||
test:
|
||||
name: Test (Go ${{ matrix.go-version }})
|
||||
runs-on: ubuntu-latest
|
||||
strategy:
|
||||
matrix:
|
||||
go-version: ["1.26"]
|
||||
steps:
|
||||
- uses: actions/checkout@v4
|
||||
|
||||
- name: Set up Go ${{ matrix.go-version }}
|
||||
uses: actions/setup-go@v5
|
||||
with:
|
||||
go-version: ${{ matrix.go-version }}
|
||||
cache: true
|
||||
|
||||
- name: Run tests with race detection and coverage
|
||||
run: go test -race -coverprofile=coverage.out -covermode=atomic ./...
|
||||
|
||||
- name: Upload coverage to Codecov
|
||||
if: matrix.go-version == '1.26'
|
||||
uses: codecov/codecov-action@v5
|
||||
with:
|
||||
files: coverage.out
|
||||
token: ${{ secrets.CODECOV_TOKEN }}
|
||||
fail_ci_if_error: false
|
||||
|
||||
lint:
|
||||
name: Lint
|
||||
runs-on: ubuntu-latest
|
||||
steps:
|
||||
- uses: actions/checkout@v4
|
||||
|
||||
- name: Set up Go
|
||||
uses: actions/setup-go@v5
|
||||
with:
|
||||
go-version: "1.26"
|
||||
cache: true
|
||||
|
||||
- name: Run staticcheck
|
||||
uses: dominikh/staticcheck-action@v1
|
||||
with:
|
||||
version: latest
|
||||
install-go: false
|
||||
@@ -32,7 +32,7 @@ If your system isn't running (or targeting another system running) `systemctl`,
|
||||
## Helper functionality
|
||||
|
||||
- [x] Get start time of a service (`ExecMainStartTimestamp`) as a `Time` type
|
||||
- [x] Get current memory in bytes (`MemoryCurrent`) an an int
|
||||
- [x] Get current memory in bytes (`MemoryCurrent`) as an int
|
||||
- [x] Get the PID of the main process (`MainPID`) as an int
|
||||
- [x] Get the restart count of a unit (`NRestarts`) as an int
|
||||
|
||||
|
||||
@@ -13,10 +13,10 @@ import (
|
||||
|
||||
func TestErrorFuncs(t *testing.T) {
|
||||
errFuncs := []func(ctx context.Context, unit string, opts Options) error{
|
||||
Enable,
|
||||
Disable,
|
||||
Restart,
|
||||
Start,
|
||||
func(ctx context.Context, unit string, opts Options) error { return Enable(ctx, unit, opts) },
|
||||
func(ctx context.Context, unit string, opts Options) error { return Disable(ctx, unit, opts) },
|
||||
func(ctx context.Context, unit string, opts Options) error { return Restart(ctx, unit, opts) },
|
||||
func(ctx context.Context, unit string, opts Options) error { return Start(ctx, unit, opts) },
|
||||
}
|
||||
errCases := []struct {
|
||||
unit string
|
||||
|
||||
119
filtererr_test.go
Normal file
119
filtererr_test.go
Normal file
@@ -0,0 +1,119 @@
|
||||
package systemctl
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"testing"
|
||||
)
|
||||
|
||||
func TestFilterErr(t *testing.T) {
|
||||
tests := []struct {
|
||||
name string
|
||||
stderr string
|
||||
want error
|
||||
}{
|
||||
{
|
||||
name: "empty stderr",
|
||||
stderr: "",
|
||||
want: nil,
|
||||
},
|
||||
{
|
||||
name: "unit does not exist",
|
||||
stderr: "Unit foo.service does not exist, proceeding anyway.",
|
||||
want: ErrDoesNotExist,
|
||||
},
|
||||
{
|
||||
name: "unit not found",
|
||||
stderr: "Unit foo.service not found.",
|
||||
want: ErrDoesNotExist,
|
||||
},
|
||||
{
|
||||
name: "unit not loaded",
|
||||
stderr: "Unit foo.service not loaded.",
|
||||
want: ErrUnitNotLoaded,
|
||||
},
|
||||
{
|
||||
name: "no such file or directory",
|
||||
stderr: "No such file or directory",
|
||||
want: ErrDoesNotExist,
|
||||
},
|
||||
{
|
||||
name: "interactive authentication required",
|
||||
stderr: "Interactive authentication required.",
|
||||
want: ErrInsufficientPermissions,
|
||||
},
|
||||
{
|
||||
name: "access denied",
|
||||
stderr: "Access denied",
|
||||
want: ErrInsufficientPermissions,
|
||||
},
|
||||
{
|
||||
name: "dbus session bus address",
|
||||
stderr: "Failed to connect to bus: $DBUS_SESSION_BUS_ADDRESS not set",
|
||||
want: ErrBusFailure,
|
||||
},
|
||||
{
|
||||
name: "unit is masked",
|
||||
stderr: "Unit foo.service is masked.",
|
||||
want: ErrMasked,
|
||||
},
|
||||
{
|
||||
name: "generic failed",
|
||||
stderr: "Failed to do something unknown",
|
||||
want: ErrUnspecified,
|
||||
},
|
||||
{
|
||||
name: "unrecognized warning",
|
||||
stderr: "Warning: something benign happened",
|
||||
want: nil,
|
||||
},
|
||||
}
|
||||
|
||||
for _, tt := range tests {
|
||||
t.Run(tt.name, func(t *testing.T) {
|
||||
got := filterErr(tt.stderr)
|
||||
if tt.want == nil {
|
||||
if got != nil {
|
||||
t.Errorf("filterErr(%q) = %v, want nil", tt.stderr, got)
|
||||
}
|
||||
return
|
||||
}
|
||||
if !errors.Is(got, tt.want) {
|
||||
t.Errorf("filterErr(%q) = %v, want error wrapping %v", tt.stderr, got, tt.want)
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func TestHasValidUnitSuffix(t *testing.T) {
|
||||
tests := []struct {
|
||||
unit string
|
||||
want bool
|
||||
}{
|
||||
{"nginx.service", true},
|
||||
{"sshd.socket", true},
|
||||
{"backup.timer", true},
|
||||
{"dev-sda1.device", true},
|
||||
{"home.mount", true},
|
||||
{"dev-sda1.swap", true},
|
||||
{"user.slice", true},
|
||||
{"multi-user.target", true},
|
||||
{"session-1.scope", true},
|
||||
{"foo.automount", true},
|
||||
{"backup.path", true},
|
||||
{"foo.snapshot", true},
|
||||
{"nginx", false},
|
||||
{"", false},
|
||||
{"foo.bar", false},
|
||||
{"foo.services", false},
|
||||
{".service", true},
|
||||
}
|
||||
|
||||
for _, tt := range tests {
|
||||
t.Run(tt.unit, func(t *testing.T) {
|
||||
got := HasValidUnitSuffix(tt.unit)
|
||||
if got != tt.want {
|
||||
t.Errorf("HasValidUnitSuffix(%q) = %v, want %v", tt.unit, got, tt.want)
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
64
helpers.go
64
helpers.go
@@ -3,6 +3,7 @@ package systemctl
|
||||
import (
|
||||
"context"
|
||||
"errors"
|
||||
"os"
|
||||
"strconv"
|
||||
"strings"
|
||||
"time"
|
||||
@@ -31,10 +32,26 @@ func GetNumRestarts(ctx context.Context, unit string, opts Options) (int, error)
|
||||
if err != nil {
|
||||
return -1, err
|
||||
}
|
||||
return strconv.Atoi(value)
|
||||
if value == "[not set]" {
|
||||
return -1, ErrValueNotSet
|
||||
}
|
||||
restarts, err := strconv.Atoi(value)
|
||||
if err != nil {
|
||||
return -1, err
|
||||
}
|
||||
// systemd returns NRestarts=0 for both genuinely zero-restart units and
|
||||
// nonexistent/unloaded units. Disambiguate by checking LoadState: if the
|
||||
// unit isn't loaded, the value is meaningless.
|
||||
if restarts == 0 {
|
||||
loadState, loadErr := Show(ctx, unit, properties.LoadState, opts)
|
||||
if loadErr == nil && loadState == "not-found" {
|
||||
return -1, ErrValueNotSet
|
||||
}
|
||||
}
|
||||
return restarts, nil
|
||||
}
|
||||
|
||||
// Get current memory in bytes (`systemctl show [unit] --property MemoryCurrent`) an an int
|
||||
// Get current memory in bytes (`systemctl show [unit] --property MemoryCurrent`) as an int
|
||||
func GetMemoryUsage(ctx context.Context, unit string, opts Options) (int, error) {
|
||||
value, err := Show(ctx, unit, properties.MemoryCurrent, opts)
|
||||
if err != nil {
|
||||
@@ -55,6 +72,33 @@ func GetPID(ctx context.Context, unit string, opts Options) (int, error) {
|
||||
return strconv.Atoi(value)
|
||||
}
|
||||
|
||||
// GetSocketsForServiceUnit returns the socket units associated with a given service unit.
|
||||
func GetSocketsForServiceUnit(ctx context.Context, unit string, opts Options) ([]string, error) {
|
||||
args := []string{"list-sockets", "--all", "--no-legend", "--no-pager"}
|
||||
if opts.UserMode {
|
||||
args = append(args, "--user")
|
||||
}
|
||||
stdout, _, _, err := execute(ctx, args)
|
||||
if err != nil {
|
||||
return []string{}, err
|
||||
}
|
||||
lines := strings.Split(stdout, "\n")
|
||||
sockets := []string{}
|
||||
for _, line := range lines {
|
||||
fields := strings.Fields(line)
|
||||
if len(fields) < 3 {
|
||||
continue
|
||||
}
|
||||
socketUnit := fields[1]
|
||||
serviceUnit := fields[2]
|
||||
if serviceUnit == unit+".service" {
|
||||
sockets = append(sockets, socketUnit)
|
||||
}
|
||||
}
|
||||
return sockets, nil
|
||||
}
|
||||
|
||||
// GetUnits returns a list of all loaded units and their states.
|
||||
func GetUnits(ctx context.Context, opts Options) ([]Unit, error) {
|
||||
args := []string{"list-units", "--all", "--no-legend", "--full", "--no-pager"}
|
||||
if opts.UserMode {
|
||||
@@ -83,6 +127,7 @@ func GetUnits(ctx context.Context, opts Options) ([]Unit, error) {
|
||||
return units, nil
|
||||
}
|
||||
|
||||
// GetMaskedUnits returns a list of all masked unit names.
|
||||
func GetMaskedUnits(ctx context.Context, opts Options) ([]string, error) {
|
||||
args := []string{"list-unit-files", "--state=masked"}
|
||||
if opts.UserMode {
|
||||
@@ -112,7 +157,16 @@ func GetMaskedUnits(ctx context.Context, opts Options) ([]string, error) {
|
||||
return units, nil
|
||||
}
|
||||
|
||||
// check if a service is masked
|
||||
// IsSystemd checks if systemd is the current init system by reading /proc/1/comm.
|
||||
func IsSystemd() (bool, error) {
|
||||
b, err := os.ReadFile("/proc/1/comm")
|
||||
if err != nil {
|
||||
return false, err
|
||||
}
|
||||
return strings.TrimSpace(string(b)) == "systemd", nil
|
||||
}
|
||||
|
||||
// IsMasked checks if a unit is masked.
|
||||
func IsMasked(ctx context.Context, unit string, opts Options) (bool, error) {
|
||||
units, err := GetMaskedUnits(ctx, opts)
|
||||
if err != nil {
|
||||
@@ -126,8 +180,8 @@ func IsMasked(ctx context.Context, unit string, opts Options) (bool, error) {
|
||||
return false, nil
|
||||
}
|
||||
|
||||
// check if a service is running
|
||||
// https://unix.stackexchange.com/a/396633
|
||||
// IsRunning checks if a unit's sub-state is "running".
|
||||
// See https://unix.stackexchange.com/a/396633 for details.
|
||||
func IsRunning(ctx context.Context, unit string, opts Options) (bool, error) {
|
||||
status, err := Show(ctx, unit, properties.SubState, opts)
|
||||
return status == "running", err
|
||||
|
||||
@@ -105,8 +105,8 @@ func TestGetNumRestarts(t *testing.T) {
|
||||
|
||||
// try nonexistant unit in user mode as user
|
||||
{"nonexistant", ErrValueNotSet, Options{UserMode: false}, true},
|
||||
// try existing unit in user mode as user
|
||||
{"syncthing", ErrValueNotSet, Options{UserMode: true}, true},
|
||||
// try existing unit in user mode as user (loaded, so NRestarts=0 is valid)
|
||||
{"syncthing", nil, Options{UserMode: true}, true},
|
||||
// try existing unit in system mode as user
|
||||
{"nginx", nil, Options{UserMode: false}, true},
|
||||
|
||||
|
||||
@@ -3,6 +3,7 @@ package properties
|
||||
type Property string
|
||||
|
||||
const (
|
||||
Accept Property = "Accept"
|
||||
ActiveEnterTimestamp Property = "ActiveEnterTimestamp"
|
||||
ActiveEnterTimestampMonotonic Property = "ActiveEnterTimestampMonotonic"
|
||||
ActiveExitTimestampMonotonic Property = "ActiveExitTimestampMonotonic"
|
||||
@@ -12,9 +13,13 @@ const (
|
||||
AssertResult Property = "AssertResult"
|
||||
AssertTimestamp Property = "AssertTimestamp"
|
||||
AssertTimestampMonotonic Property = "AssertTimestampMonotonic"
|
||||
Backlog Property = "Backlog"
|
||||
Before Property = "Before"
|
||||
BindIPv6Only Property = "BindIPv6Only"
|
||||
BindLogSockets Property = "BindLogSockets"
|
||||
BlockIOAccounting Property = "BlockIOAccounting"
|
||||
BlockIOWeight Property = "BlockIOWeight"
|
||||
Broadcast Property = "Broadcast"
|
||||
CPUAccounting Property = "CPUAccounting"
|
||||
CPUAffinityFromNUMA Property = "CPUAffinityFromNUMA"
|
||||
CPUQuotaPerSecUSec Property = "CPUQuotaPerSecUSec"
|
||||
@@ -28,6 +33,7 @@ const (
|
||||
CacheDirectoryMode Property = "CacheDirectoryMode"
|
||||
CanFreeze Property = "CanFreeze"
|
||||
CanIsolate Property = "CanIsolate"
|
||||
CanLiveMount Property = "CanLiveMount"
|
||||
CanReload Property = "CanReload"
|
||||
CanStart Property = "CanStart"
|
||||
CanStop Property = "CanStop"
|
||||
@@ -40,17 +46,26 @@ const (
|
||||
ConfigurationDirectoryMode Property = "ConfigurationDirectoryMode"
|
||||
Conflicts Property = "Conflicts"
|
||||
ControlGroup Property = "ControlGroup"
|
||||
ControlGroupId Property = "ControlGroupId"
|
||||
ControlPID Property = "ControlPID"
|
||||
CoredumpFilter Property = "CoredumpFilter"
|
||||
CoredumpReceive Property = "CoredumpReceive"
|
||||
DebugInvocation Property = "DebugInvocation"
|
||||
DefaultDependencies Property = "DefaultDependencies"
|
||||
DefaultMemoryLow Property = "DefaultMemoryLow"
|
||||
DefaultMemoryMin Property = "DefaultMemoryMin"
|
||||
DefaultStartupMemoryLow Property = "DefaultStartupMemoryLow"
|
||||
DeferAcceptUSec Property = "DeferAcceptUSec"
|
||||
Delegate Property = "Delegate"
|
||||
Description Property = "Description"
|
||||
DevicePolicy Property = "DevicePolicy"
|
||||
DirectoryMode Property = "DirectoryMode"
|
||||
DynamicUser Property = "DynamicUser"
|
||||
EffectiveCPUs Property = "EffectiveCPUs"
|
||||
EffectiveMemoryHigh Property = "EffectiveMemoryHigh"
|
||||
EffectiveMemoryMax Property = "EffectiveMemoryMax"
|
||||
EffectiveMemoryNodes Property = "EffectiveMemoryNodes"
|
||||
EffectiveTasksMax Property = "EffectiveTasksMax"
|
||||
ExecMainCode Property = "ExecMainCode"
|
||||
ExecMainExitTimestampMonotonic Property = "ExecMainExitTimestampMonotonic"
|
||||
ExecMainPID Property = "ExecMainPID"
|
||||
@@ -61,10 +76,14 @@ const (
|
||||
ExecReloadEx Property = "ExecReloadEx"
|
||||
ExecStart Property = "ExecStart"
|
||||
ExecStartEx Property = "ExecStartEx"
|
||||
ExtensionImagePolicy Property = "ExtensionImagePolicy"
|
||||
FailureAction Property = "FailureAction"
|
||||
FileDescriptorName Property = "FileDescriptorName"
|
||||
FileDescriptorStoreMax Property = "FileDescriptorStoreMax"
|
||||
FinalKillSignal Property = "FinalKillSignal"
|
||||
FlushPending Property = "FlushPending"
|
||||
FragmentPath Property = "FragmentPath"
|
||||
FreeBind Property = "FreeBind"
|
||||
FreezerState Property = "FreezerState"
|
||||
GID Property = "GID"
|
||||
GuessMainPID Property = "GuessMainPID"
|
||||
@@ -81,6 +100,8 @@ const (
|
||||
IPEgressPackets Property = "IPEgressPackets"
|
||||
IPIngressBytes Property = "IPIngressBytes"
|
||||
IPIngressPackets Property = "IPIngressPackets"
|
||||
IPTOS Property = "IPTOS"
|
||||
IPTTL Property = "IPTTL"
|
||||
Id Property = "Id"
|
||||
IgnoreOnIsolate Property = "IgnoreOnIsolate"
|
||||
IgnoreSIGPIPE Property = "IgnoreSIGPIPE"
|
||||
@@ -91,6 +112,10 @@ const (
|
||||
JobRunningTimeoutUSec Property = "JobRunningTimeoutUSec"
|
||||
JobTimeoutAction Property = "JobTimeoutAction"
|
||||
JobTimeoutUSec Property = "JobTimeoutUSec"
|
||||
KeepAlive Property = "KeepAlive"
|
||||
KeepAliveIntervalUSec Property = "KeepAliveIntervalUSec"
|
||||
KeepAliveProbes Property = "KeepAliveProbes"
|
||||
KeepAliveTimeUSec Property = "KeepAliveTimeUSec"
|
||||
KeyringMode Property = "KeyringMode"
|
||||
KillMode Property = "KillMode"
|
||||
KillSignal Property = "KillSignal"
|
||||
@@ -126,6 +151,7 @@ const (
|
||||
LimitSIGPENDINGSoft Property = "LimitSIGPENDINGSoft"
|
||||
LimitSTACK Property = "LimitSTACK"
|
||||
LimitSTACKSoft Property = "LimitSTACKSoft"
|
||||
Listen Property = "Listen"
|
||||
LoadState Property = "LoadState"
|
||||
LockPersonality Property = "LockPersonality"
|
||||
LogLevelMax Property = "LogLevelMax"
|
||||
@@ -134,42 +160,76 @@ const (
|
||||
LogsDirectoryMode Property = "LogsDirectoryMode"
|
||||
MainPID Property = "MainPID"
|
||||
ManagedOOMMemoryPressure Property = "ManagedOOMMemoryPressure"
|
||||
ManagedOOMMemoryPressureDurationUSec Property = "ManagedOOMMemoryPressureDurationUSec"
|
||||
ManagedOOMMemoryPressureLimit Property = "ManagedOOMMemoryPressureLimit"
|
||||
ManagedOOMPreference Property = "ManagedOOMPreference"
|
||||
ManagedOOMSwap Property = "ManagedOOMSwap"
|
||||
Mark Property = "Mark"
|
||||
MaxConnections Property = "MaxConnections"
|
||||
MaxConnectionsPerSource Property = "MaxConnectionsPerSource"
|
||||
MemoryAccounting Property = "MemoryAccounting"
|
||||
MemoryAvailable Property = "MemoryAvailable"
|
||||
MemoryCurrent Property = "MemoryCurrent"
|
||||
MemoryDenyWriteExecute Property = "MemoryDenyWriteExecute"
|
||||
MemoryHigh Property = "MemoryHigh"
|
||||
MemoryKSM Property = "MemoryKSM"
|
||||
MemoryLimit Property = "MemoryLimit"
|
||||
MemoryLow Property = "MemoryLow"
|
||||
MemoryMax Property = "MemoryMax"
|
||||
MemoryMin Property = "MemoryMin"
|
||||
MemoryPeak Property = "MemoryPeak"
|
||||
MemoryPressureThresholdUSec Property = "MemoryPressureThresholdUSec"
|
||||
MemoryPressureWatch Property = "MemoryPressureWatch"
|
||||
MemorySwapCurrent Property = "MemorySwapCurrent"
|
||||
MemorySwapMax Property = "MemorySwapMax"
|
||||
MemorySwapPeak Property = "MemorySwapPeak"
|
||||
MemoryZSwapCurrent Property = "MemoryZSwapCurrent"
|
||||
MemoryZSwapMax Property = "MemoryZSwapMax"
|
||||
MemoryZSwapWriteback Property = "MemoryZSwapWriteback"
|
||||
MessageQueueMaxMessages Property = "MessageQueueMaxMessages"
|
||||
MessageQueueMessageSize Property = "MessageQueueMessageSize"
|
||||
MountAPIVFS Property = "MountAPIVFS"
|
||||
MountImagePolicy Property = "MountImagePolicy"
|
||||
NAccepted Property = "NAccepted"
|
||||
NConnections Property = "NConnections"
|
||||
NFileDescriptorStore Property = "NFileDescriptorStore"
|
||||
NRefused Property = "NRefused"
|
||||
NRestarts Property = "NRestarts"
|
||||
NUMAPolicy Property = "NUMAPolicy"
|
||||
Names Property = "Names"
|
||||
NeedDaemonReload Property = "NeedDaemonReload"
|
||||
Nice Property = "Nice"
|
||||
NoDelay Property = "NoDelay"
|
||||
NoNewPrivileges Property = "NoNewPrivileges"
|
||||
NonBlocking Property = "NonBlocking"
|
||||
NotifyAccess Property = "NotifyAccess"
|
||||
OOMPolicy Property = "OOMPolicy"
|
||||
OOMScoreAdjust Property = "OOMScoreAdjust"
|
||||
OnFailureJobMode Property = "OnFailureJobMode"
|
||||
OnSuccessJobMode Property = "OnSuccessJobMode"
|
||||
PIDFile Property = "PIDFile"
|
||||
PassCredentials Property = "PassCredentials"
|
||||
PassFileDescriptorsToExec Property = "PassFileDescriptorsToExec"
|
||||
PassPacketInfo Property = "PassPacketInfo"
|
||||
PassSecurity Property = "PassSecurity"
|
||||
Perpetual Property = "Perpetual"
|
||||
PipeSize Property = "PipeSize"
|
||||
PollLimitBurst Property = "PollLimitBurst"
|
||||
PollLimitIntervalUSec Property = "PollLimitIntervalUSec"
|
||||
Priority Property = "Priority"
|
||||
PrivateDevices Property = "PrivateDevices"
|
||||
PrivateIPC Property = "PrivateIPC"
|
||||
PrivateMounts Property = "PrivateMounts"
|
||||
PrivateNetwork Property = "PrivateNetwork"
|
||||
PrivatePIDs Property = "PrivatePIDs"
|
||||
PrivateTmp Property = "PrivateTmp"
|
||||
PrivateTmpEx Property = "PrivateTmpEx"
|
||||
PrivateUsers Property = "PrivateUsers"
|
||||
PrivateUsersEx Property = "PrivateUsersEx"
|
||||
ProcSubset Property = "ProcSubset"
|
||||
ProtectClock Property = "ProtectClock"
|
||||
ProtectControlGroups Property = "ProtectControlGroups"
|
||||
ProtectControlGroupsEx Property = "ProtectControlGroupsEx"
|
||||
ProtectHome Property = "ProtectHome"
|
||||
ProtectHostname Property = "ProtectHostname"
|
||||
ProtectKernelLogs Property = "ProtectKernelLogs"
|
||||
@@ -177,12 +237,16 @@ const (
|
||||
ProtectKernelTunables Property = "ProtectKernelTunables"
|
||||
ProtectProc Property = "ProtectProc"
|
||||
ProtectSystem Property = "ProtectSystem"
|
||||
ReceiveBuffer Property = "ReceiveBuffer"
|
||||
RefuseManualStart Property = "RefuseManualStart"
|
||||
RefuseManualStop Property = "RefuseManualStop"
|
||||
ReloadResult Property = "ReloadResult"
|
||||
RemainAfterExit Property = "RemainAfterExit"
|
||||
RemoveIPC Property = "RemoveIPC"
|
||||
RemoveOnStop Property = "RemoveOnStop"
|
||||
RequiredBy Property = "RequiredBy"
|
||||
Requires Property = "Requires"
|
||||
RequiresMountsFor Property = "RequiresMountsFor"
|
||||
Restart Property = "Restart"
|
||||
RestartKillSignal Property = "RestartKillSignal"
|
||||
RestartUSec Property = "RestartUSec"
|
||||
@@ -190,15 +254,22 @@ const (
|
||||
RestrictRealtime Property = "RestrictRealtime"
|
||||
RestrictSUIDSGID Property = "RestrictSUIDSGID"
|
||||
Result Property = "Result"
|
||||
ReusePort Property = "ReusePort"
|
||||
RootDirectoryStartOnly Property = "RootDirectoryStartOnly"
|
||||
RootEphemeral Property = "RootEphemeral"
|
||||
RootImagePolicy Property = "RootImagePolicy"
|
||||
RuntimeDirectoryMode Property = "RuntimeDirectoryMode"
|
||||
RuntimeDirectoryPreserve Property = "RuntimeDirectoryPreserve"
|
||||
RuntimeMaxUSec Property = "RuntimeMaxUSec"
|
||||
SameProcessGroup Property = "SameProcessGroup"
|
||||
SecureBits Property = "SecureBits"
|
||||
SendBuffer Property = "SendBuffer"
|
||||
SendSIGHUP Property = "SendSIGHUP"
|
||||
SendSIGKILL Property = "SendSIGKILL"
|
||||
SetLoginEnvironment Property = "SetLoginEnvironment"
|
||||
Slice Property = "Slice"
|
||||
SocketMode Property = "SocketMode"
|
||||
SocketProtocol Property = "SocketProtocol"
|
||||
StandardError Property = "StandardError"
|
||||
StandardInput Property = "StandardInput"
|
||||
StandardOutput Property = "StandardOutput"
|
||||
@@ -209,6 +280,11 @@ const (
|
||||
StartupCPUShares Property = "StartupCPUShares"
|
||||
StartupCPUWeight Property = "StartupCPUWeight"
|
||||
StartupIOWeight Property = "StartupIOWeight"
|
||||
StartupMemoryHigh Property = "StartupMemoryHigh"
|
||||
StartupMemoryLow Property = "StartupMemoryLow"
|
||||
StartupMemoryMax Property = "StartupMemoryMax"
|
||||
StartupMemorySwapMax Property = "StartupMemorySwapMax"
|
||||
StartupMemoryZSwapMax Property = "StartupMemoryZSwapMax"
|
||||
StateChangeTimestamp Property = "StateChangeTimestamp"
|
||||
StateChangeTimestampMonotonic Property = "StateChangeTimestampMonotonic"
|
||||
StateDirectoryMode Property = "StateDirectoryMode"
|
||||
@@ -216,6 +292,7 @@ const (
|
||||
StopWhenUnneeded Property = "StopWhenUnneeded"
|
||||
SubState Property = "SubState"
|
||||
SuccessAction Property = "SuccessAction"
|
||||
SurviveFinalKillSignal Property = "SurviveFinalKillSignal"
|
||||
SyslogFacility Property = "SyslogFacility"
|
||||
SyslogLevel Property = "SyslogLevel"
|
||||
SyslogLevelPrefix Property = "SyslogLevelPrefix"
|
||||
@@ -233,8 +310,14 @@ const (
|
||||
TimeoutStartUSec Property = "TimeoutStartUSec"
|
||||
TimeoutStopFailureMode Property = "TimeoutStopFailureMode"
|
||||
TimeoutStopUSec Property = "TimeoutStopUSec"
|
||||
TimeoutUSec Property = "TimeoutUSec"
|
||||
TimerSlackNSec Property = "TimerSlackNSec"
|
||||
Timestamping Property = "Timestamping"
|
||||
Transient Property = "Transient"
|
||||
Transparent Property = "Transparent"
|
||||
TriggerLimitBurst Property = "TriggerLimitBurst"
|
||||
TriggerLimitIntervalUSec Property = "TriggerLimitIntervalUSec"
|
||||
Triggers Property = "Triggers"
|
||||
Type Property = "Type"
|
||||
UID Property = "UID"
|
||||
UMask Property = "UMask"
|
||||
@@ -245,4 +328,5 @@ const (
|
||||
WatchdogSignal Property = "WatchdogSignal"
|
||||
WatchdogTimestampMonotonic Property = "WatchdogTimestampMonotonic"
|
||||
WatchdogUSec Property = "WatchdogUSec"
|
||||
Writable Property = "Writable"
|
||||
)
|
||||
|
||||
@@ -1,6 +1,7 @@
|
||||
package properties
|
||||
|
||||
var Properties = []Property{
|
||||
Accept,
|
||||
ActiveEnterTimestamp,
|
||||
ActiveEnterTimestampMonotonic,
|
||||
ActiveExitTimestampMonotonic,
|
||||
@@ -10,9 +11,13 @@ var Properties = []Property{
|
||||
AssertResult,
|
||||
AssertTimestamp,
|
||||
AssertTimestampMonotonic,
|
||||
Backlog,
|
||||
Before,
|
||||
BindIPv6Only,
|
||||
BindLogSockets,
|
||||
BlockIOAccounting,
|
||||
BlockIOWeight,
|
||||
Broadcast,
|
||||
CPUAccounting,
|
||||
CPUAffinityFromNUMA,
|
||||
CPUQuotaPerSecUSec,
|
||||
@@ -26,6 +31,7 @@ var Properties = []Property{
|
||||
CacheDirectoryMode,
|
||||
CanFreeze,
|
||||
CanIsolate,
|
||||
CanLiveMount,
|
||||
CanReload,
|
||||
CanStart,
|
||||
CanStop,
|
||||
@@ -38,17 +44,26 @@ var Properties = []Property{
|
||||
ConfigurationDirectoryMode,
|
||||
Conflicts,
|
||||
ControlGroup,
|
||||
ControlGroupId,
|
||||
ControlPID,
|
||||
CoredumpFilter,
|
||||
CoredumpReceive,
|
||||
DebugInvocation,
|
||||
DefaultDependencies,
|
||||
DefaultMemoryLow,
|
||||
DefaultMemoryMin,
|
||||
DefaultStartupMemoryLow,
|
||||
DeferAcceptUSec,
|
||||
Delegate,
|
||||
Description,
|
||||
DevicePolicy,
|
||||
DirectoryMode,
|
||||
DynamicUser,
|
||||
EffectiveCPUs,
|
||||
EffectiveMemoryHigh,
|
||||
EffectiveMemoryMax,
|
||||
EffectiveMemoryNodes,
|
||||
EffectiveTasksMax,
|
||||
ExecMainCode,
|
||||
ExecMainExitTimestampMonotonic,
|
||||
ExecMainPID,
|
||||
@@ -59,10 +74,14 @@ var Properties = []Property{
|
||||
ExecReloadEx,
|
||||
ExecStart,
|
||||
ExecStartEx,
|
||||
ExtensionImagePolicy,
|
||||
FailureAction,
|
||||
FileDescriptorName,
|
||||
FileDescriptorStoreMax,
|
||||
FinalKillSignal,
|
||||
FlushPending,
|
||||
FragmentPath,
|
||||
FreeBind,
|
||||
FreezerState,
|
||||
GID,
|
||||
GuessMainPID,
|
||||
@@ -79,6 +98,8 @@ var Properties = []Property{
|
||||
IPEgressPackets,
|
||||
IPIngressBytes,
|
||||
IPIngressPackets,
|
||||
IPTOS,
|
||||
IPTTL,
|
||||
Id,
|
||||
IgnoreOnIsolate,
|
||||
IgnoreSIGPIPE,
|
||||
@@ -89,6 +110,10 @@ var Properties = []Property{
|
||||
JobRunningTimeoutUSec,
|
||||
JobTimeoutAction,
|
||||
JobTimeoutUSec,
|
||||
KeepAlive,
|
||||
KeepAliveIntervalUSec,
|
||||
KeepAliveProbes,
|
||||
KeepAliveTimeUSec,
|
||||
KeyringMode,
|
||||
KillMode,
|
||||
KillSignal,
|
||||
@@ -124,6 +149,7 @@ var Properties = []Property{
|
||||
LimitSIGPENDINGSoft,
|
||||
LimitSTACK,
|
||||
LimitSTACKSoft,
|
||||
Listen,
|
||||
LoadState,
|
||||
LockPersonality,
|
||||
LogLevelMax,
|
||||
@@ -132,42 +158,76 @@ var Properties = []Property{
|
||||
LogsDirectoryMode,
|
||||
MainPID,
|
||||
ManagedOOMMemoryPressure,
|
||||
ManagedOOMMemoryPressureDurationUSec,
|
||||
ManagedOOMMemoryPressureLimit,
|
||||
ManagedOOMPreference,
|
||||
ManagedOOMSwap,
|
||||
Mark,
|
||||
MaxConnections,
|
||||
MaxConnectionsPerSource,
|
||||
MemoryAccounting,
|
||||
MemoryAvailable,
|
||||
MemoryCurrent,
|
||||
MemoryDenyWriteExecute,
|
||||
MemoryHigh,
|
||||
MemoryKSM,
|
||||
MemoryLimit,
|
||||
MemoryLow,
|
||||
MemoryMax,
|
||||
MemoryMin,
|
||||
MemoryPeak,
|
||||
MemoryPressureThresholdUSec,
|
||||
MemoryPressureWatch,
|
||||
MemorySwapCurrent,
|
||||
MemorySwapMax,
|
||||
MemorySwapPeak,
|
||||
MemoryZSwapCurrent,
|
||||
MemoryZSwapMax,
|
||||
MemoryZSwapWriteback,
|
||||
MessageQueueMaxMessages,
|
||||
MessageQueueMessageSize,
|
||||
MountAPIVFS,
|
||||
MountImagePolicy,
|
||||
NAccepted,
|
||||
NConnections,
|
||||
NFileDescriptorStore,
|
||||
NRefused,
|
||||
NRestarts,
|
||||
NUMAPolicy,
|
||||
Names,
|
||||
NeedDaemonReload,
|
||||
Nice,
|
||||
NoDelay,
|
||||
NoNewPrivileges,
|
||||
NonBlocking,
|
||||
NotifyAccess,
|
||||
OOMPolicy,
|
||||
OOMScoreAdjust,
|
||||
OnFailureJobMode,
|
||||
OnSuccessJobMode,
|
||||
PIDFile,
|
||||
PassCredentials,
|
||||
PassFileDescriptorsToExec,
|
||||
PassPacketInfo,
|
||||
PassSecurity,
|
||||
Perpetual,
|
||||
PipeSize,
|
||||
PollLimitBurst,
|
||||
PollLimitIntervalUSec,
|
||||
Priority,
|
||||
PrivateDevices,
|
||||
PrivateIPC,
|
||||
PrivateMounts,
|
||||
PrivateNetwork,
|
||||
PrivatePIDs,
|
||||
PrivateTmp,
|
||||
PrivateTmpEx,
|
||||
PrivateUsers,
|
||||
PrivateUsersEx,
|
||||
ProcSubset,
|
||||
ProtectClock,
|
||||
ProtectControlGroups,
|
||||
ProtectControlGroupsEx,
|
||||
ProtectHome,
|
||||
ProtectHostname,
|
||||
ProtectKernelLogs,
|
||||
@@ -175,12 +235,16 @@ var Properties = []Property{
|
||||
ProtectKernelTunables,
|
||||
ProtectProc,
|
||||
ProtectSystem,
|
||||
ReceiveBuffer,
|
||||
RefuseManualStart,
|
||||
RefuseManualStop,
|
||||
ReloadResult,
|
||||
RemainAfterExit,
|
||||
RemoveIPC,
|
||||
RemoveOnStop,
|
||||
RequiredBy,
|
||||
Requires,
|
||||
RequiresMountsFor,
|
||||
Restart,
|
||||
RestartKillSignal,
|
||||
RestartUSec,
|
||||
@@ -188,15 +252,22 @@ var Properties = []Property{
|
||||
RestrictRealtime,
|
||||
RestrictSUIDSGID,
|
||||
Result,
|
||||
ReusePort,
|
||||
RootDirectoryStartOnly,
|
||||
RootEphemeral,
|
||||
RootImagePolicy,
|
||||
RuntimeDirectoryMode,
|
||||
RuntimeDirectoryPreserve,
|
||||
RuntimeMaxUSec,
|
||||
SameProcessGroup,
|
||||
SecureBits,
|
||||
SendBuffer,
|
||||
SendSIGHUP,
|
||||
SendSIGKILL,
|
||||
SetLoginEnvironment,
|
||||
Slice,
|
||||
SocketMode,
|
||||
SocketProtocol,
|
||||
StandardError,
|
||||
StandardInput,
|
||||
StandardOutput,
|
||||
@@ -207,6 +278,11 @@ var Properties = []Property{
|
||||
StartupCPUShares,
|
||||
StartupCPUWeight,
|
||||
StartupIOWeight,
|
||||
StartupMemoryHigh,
|
||||
StartupMemoryLow,
|
||||
StartupMemoryMax,
|
||||
StartupMemorySwapMax,
|
||||
StartupMemoryZSwapMax,
|
||||
StateChangeTimestamp,
|
||||
StateChangeTimestampMonotonic,
|
||||
StateDirectoryMode,
|
||||
@@ -214,6 +290,7 @@ var Properties = []Property{
|
||||
StopWhenUnneeded,
|
||||
SubState,
|
||||
SuccessAction,
|
||||
SurviveFinalKillSignal,
|
||||
SyslogFacility,
|
||||
SyslogLevel,
|
||||
SyslogLevelPrefix,
|
||||
@@ -231,8 +308,14 @@ var Properties = []Property{
|
||||
TimeoutStartUSec,
|
||||
TimeoutStopFailureMode,
|
||||
TimeoutStopUSec,
|
||||
TimeoutUSec,
|
||||
TimerSlackNSec,
|
||||
Timestamping,
|
||||
Transient,
|
||||
Transparent,
|
||||
TriggerLimitBurst,
|
||||
TriggerLimitIntervalUSec,
|
||||
Triggers,
|
||||
Type,
|
||||
UID,
|
||||
UMask,
|
||||
@@ -243,4 +326,5 @@ var Properties = []Property{
|
||||
WatchdogSignal,
|
||||
WatchdogTimestampMonotonic,
|
||||
WatchdogUSec,
|
||||
Writable,
|
||||
}
|
||||
|
||||
29
structs.go
29
structs.go
@@ -1,5 +1,7 @@
|
||||
package systemctl
|
||||
|
||||
import "strings"
|
||||
|
||||
type Options struct {
|
||||
UserMode bool
|
||||
}
|
||||
@@ -11,3 +13,30 @@ type Unit struct {
|
||||
Sub string
|
||||
Description string
|
||||
}
|
||||
|
||||
// UnitTypes contains all valid systemd unit type suffixes.
|
||||
var UnitTypes = []string{
|
||||
"automount",
|
||||
"device",
|
||||
"mount",
|
||||
"path",
|
||||
"scope",
|
||||
"service",
|
||||
"slice",
|
||||
"snapshot",
|
||||
"socket",
|
||||
"swap",
|
||||
"target",
|
||||
"timer",
|
||||
}
|
||||
|
||||
// HasValidUnitSuffix checks whether the given unit name ends with a valid
|
||||
// systemd unit type suffix (e.g. ".service", ".timer").
|
||||
func HasValidUnitSuffix(unit string) bool {
|
||||
for _, t := range UnitTypes {
|
||||
if strings.HasSuffix(unit, "."+t) {
|
||||
return true
|
||||
}
|
||||
}
|
||||
return false
|
||||
}
|
||||
|
||||
209
systemctl.go
209
systemctl.go
@@ -2,8 +2,6 @@ package systemctl
|
||||
|
||||
import (
|
||||
"context"
|
||||
"regexp"
|
||||
"strings"
|
||||
|
||||
"github.com/taigrr/systemctl/properties"
|
||||
)
|
||||
@@ -14,13 +12,10 @@ import (
|
||||
// files, and recreate the entire dependency tree. While the daemon is being
|
||||
// reloaded, all sockets systemd listens on behalf of user configuration will
|
||||
// stay accessible.
|
||||
func DaemonReload(ctx context.Context, opts Options) error {
|
||||
args := []string{"daemon-reload", "--system"}
|
||||
if opts.UserMode {
|
||||
args[1] = "--user"
|
||||
}
|
||||
_, _, _, err := execute(ctx, args)
|
||||
return err
|
||||
//
|
||||
// Any additional arguments are passed directly to the systemctl command.
|
||||
func DaemonReload(ctx context.Context, opts Options, args ...string) error {
|
||||
return daemonReload(ctx, opts, args...)
|
||||
}
|
||||
|
||||
// Reenables one or more units.
|
||||
@@ -28,13 +23,10 @@ func DaemonReload(ctx context.Context, opts Options) error {
|
||||
// This removes all symlinks to the unit files backing the specified units from
|
||||
// the unit configuration directory, then recreates the symlink to the unit again,
|
||||
// atomically. Can be used to change the symlink target.
|
||||
func Reenable(ctx context.Context, unit string, opts Options) error {
|
||||
args := []string{"reenable", "--system", unit}
|
||||
if opts.UserMode {
|
||||
args[1] = "--user"
|
||||
}
|
||||
_, _, _, err := execute(ctx, args)
|
||||
return err
|
||||
//
|
||||
// Any additional arguments are passed directly to the systemctl command.
|
||||
func Reenable(ctx context.Context, unit string, opts Options, args ...string) error {
|
||||
return reenable(ctx, unit, opts, args...)
|
||||
}
|
||||
|
||||
// Disables one or more units.
|
||||
@@ -42,13 +34,10 @@ func Reenable(ctx context.Context, unit string, opts Options) error {
|
||||
// This removes all symlinks to the unit files backing the specified units from
|
||||
// the unit configuration directory, and hence undoes any changes made by
|
||||
// enable or link.
|
||||
func Disable(ctx context.Context, unit string, opts Options) error {
|
||||
args := []string{"disable", "--system", unit}
|
||||
if opts.UserMode {
|
||||
args[1] = "--user"
|
||||
}
|
||||
_, _, _, err := execute(ctx, args)
|
||||
return err
|
||||
//
|
||||
// Any additional arguments are passed directly to the systemctl command.
|
||||
func Disable(ctx context.Context, unit string, opts Options, args ...string) error {
|
||||
return disable(ctx, unit, opts, args...)
|
||||
}
|
||||
|
||||
// Enable one or more units or unit instances.
|
||||
@@ -57,38 +46,20 @@ func Disable(ctx context.Context, unit string, opts Options) error {
|
||||
// the indicated unit files. After the symlinks have been created, the system
|
||||
// manager configuration is reloaded (in a way equivalent to daemon-reload),
|
||||
// in order to ensure the changes are taken into account immediately.
|
||||
func Enable(ctx context.Context, unit string, opts Options) error {
|
||||
args := []string{"enable", "--system", unit}
|
||||
if opts.UserMode {
|
||||
args[1] = "--user"
|
||||
}
|
||||
_, _, _, err := execute(ctx, args)
|
||||
return err
|
||||
//
|
||||
// Any additional arguments are passed directly to the systemctl command.
|
||||
func Enable(ctx context.Context, unit string, opts Options, args ...string) error {
|
||||
return enable(ctx, unit, opts, args...)
|
||||
}
|
||||
|
||||
// Check whether any of the specified units are active (i.e. running).
|
||||
//
|
||||
// Returns true if the unit is active, false if inactive or failed.
|
||||
// Also returns false in an error case.
|
||||
func IsActive(ctx context.Context, unit string, opts Options) (bool, error) {
|
||||
args := []string{"is-active", "--system", unit}
|
||||
if opts.UserMode {
|
||||
args[1] = "--user"
|
||||
}
|
||||
stdout, _, _, err := execute(ctx, args)
|
||||
stdout = strings.TrimSuffix(stdout, "\n")
|
||||
switch stdout {
|
||||
case "inactive":
|
||||
return false, nil
|
||||
case "active":
|
||||
return true, nil
|
||||
case "failed":
|
||||
return false, nil
|
||||
case "activating":
|
||||
return false, nil
|
||||
default:
|
||||
return false, err
|
||||
}
|
||||
//
|
||||
// Any additional arguments are passed directly to the systemctl command.
|
||||
func IsActive(ctx context.Context, unit string, opts Options, args ...string) (bool, error) {
|
||||
return isActive(ctx, unit, opts, args...)
|
||||
}
|
||||
|
||||
// Checks whether any of the specified unit files are enabled (as with enable).
|
||||
@@ -100,60 +71,17 @@ func IsActive(ctx context.Context, unit string, opts Options) (bool, error) {
|
||||
//
|
||||
// See https://www.freedesktop.org/software/systemd/man/systemctl.html#is-enabled%20UNIT%E2%80%A6
|
||||
// for more information
|
||||
func IsEnabled(ctx context.Context, unit string, opts Options) (bool, error) {
|
||||
args := []string{"is-enabled", "--system", unit}
|
||||
if opts.UserMode {
|
||||
args[1] = "--user"
|
||||
}
|
||||
stdout, _, _, err := execute(ctx, args)
|
||||
stdout = strings.TrimSuffix(stdout, "\n")
|
||||
switch stdout {
|
||||
case "enabled":
|
||||
return true, nil
|
||||
case "enabled-runtime":
|
||||
return true, nil
|
||||
case "linked":
|
||||
return false, ErrLinked
|
||||
case "linked-runtime":
|
||||
return false, ErrLinked
|
||||
case "alias":
|
||||
return true, nil
|
||||
case "masked":
|
||||
return false, ErrMasked
|
||||
case "masked-runtime":
|
||||
return false, ErrMasked
|
||||
case "static":
|
||||
return true, nil
|
||||
case "indirect":
|
||||
return true, nil
|
||||
case "disabled":
|
||||
return false, nil
|
||||
case "generated":
|
||||
return true, nil
|
||||
case "transient":
|
||||
return true, nil
|
||||
}
|
||||
if err != nil {
|
||||
return false, err
|
||||
}
|
||||
return false, ErrUnspecified
|
||||
//
|
||||
// Any additional arguments are passed directly to the systemctl command.
|
||||
func IsEnabled(ctx context.Context, unit string, opts Options, args ...string) (bool, error) {
|
||||
return isEnabled(ctx, unit, opts, args...)
|
||||
}
|
||||
|
||||
// Check whether any of the specified units are in a "failed" state.
|
||||
func IsFailed(ctx context.Context, unit string, opts Options) (bool, error) {
|
||||
args := []string{"is-failed", "--system", unit}
|
||||
if opts.UserMode {
|
||||
args[1] = "--user"
|
||||
}
|
||||
stdout, _, _, err := execute(ctx, args)
|
||||
if matched, _ := regexp.MatchString(`inactive`, stdout); matched {
|
||||
return false, nil
|
||||
} else if matched, _ := regexp.MatchString(`active`, stdout); matched {
|
||||
return false, nil
|
||||
} else if matched, _ := regexp.MatchString(`failed`, stdout); matched {
|
||||
return true, nil
|
||||
}
|
||||
return false, err
|
||||
//
|
||||
// Any additional arguments are passed directly to the systemctl command.
|
||||
func IsFailed(ctx context.Context, unit string, opts Options, args ...string) (bool, error) {
|
||||
return isFailed(ctx, unit, opts, args...)
|
||||
}
|
||||
|
||||
// Mask one or more units, as specified on the command line. This will link
|
||||
@@ -162,71 +90,51 @@ func IsFailed(ctx context.Context, unit string, opts Options) (bool, error) {
|
||||
// Notably, Mask may return ErrDoesNotExist if a unit doesn't exist, but it will
|
||||
// continue masking anyway. Calling Mask on a non-existing masked unit does not
|
||||
// return an error. Similarly, see Unmask.
|
||||
func Mask(ctx context.Context, unit string, opts Options) error {
|
||||
args := []string{"mask", "--system", unit}
|
||||
if opts.UserMode {
|
||||
args[1] = "--user"
|
||||
}
|
||||
_, _, _, err := execute(ctx, args)
|
||||
return err
|
||||
//
|
||||
// Any additional arguments are passed directly to the systemctl command.
|
||||
func Mask(ctx context.Context, unit string, opts Options, args ...string) error {
|
||||
return mask(ctx, unit, opts, args...)
|
||||
}
|
||||
|
||||
// Stop and then start one or more units specified on the command line.
|
||||
// If the units are not running yet, they will be started.
|
||||
func Restart(ctx context.Context, unit string, opts Options) error {
|
||||
args := []string{"restart", "--system", unit}
|
||||
if opts.UserMode {
|
||||
args[1] = "--user"
|
||||
}
|
||||
_, _, _, err := execute(ctx, args)
|
||||
return err
|
||||
//
|
||||
// Any additional arguments are passed directly to the systemctl command.
|
||||
func Restart(ctx context.Context, unit string, opts Options, args ...string) error {
|
||||
return restart(ctx, unit, opts, args...)
|
||||
}
|
||||
|
||||
// Show a selected property of a unit. Accepted properties are predefined in the
|
||||
// properties subpackage to guarantee properties are valid and assist code-completion.
|
||||
func Show(ctx context.Context, unit string, property properties.Property, opts Options) (string, error) {
|
||||
args := []string{"show", "--system", unit, "--property", string(property)}
|
||||
if opts.UserMode {
|
||||
args[1] = "--user"
|
||||
}
|
||||
stdout, _, _, err := execute(ctx, args)
|
||||
stdout = strings.TrimPrefix(stdout, string(property)+"=")
|
||||
stdout = strings.TrimSuffix(stdout, "\n")
|
||||
return stdout, err
|
||||
//
|
||||
// Any additional arguments are passed directly to the systemctl command.
|
||||
func Show(ctx context.Context, unit string, property properties.Property, opts Options, args ...string) (string, error) {
|
||||
return show(ctx, unit, property, opts, args...)
|
||||
}
|
||||
|
||||
// Start (activate) a given unit
|
||||
func Start(ctx context.Context, unit string, opts Options) error {
|
||||
args := []string{"start", "--system", unit}
|
||||
if opts.UserMode {
|
||||
args[1] = "--user"
|
||||
}
|
||||
_, _, _, err := execute(ctx, args)
|
||||
return err
|
||||
//
|
||||
// Any additional arguments are passed directly to the systemctl command.
|
||||
func Start(ctx context.Context, unit string, opts Options, args ...string) error {
|
||||
return start(ctx, unit, opts, args...)
|
||||
}
|
||||
|
||||
// Get back the status string which would be returned by running
|
||||
// `systemctl status [unit]`.
|
||||
//
|
||||
// Generally, it makes more sense to programatically retrieve the properties
|
||||
// Generally, it makes more sense to programmatically retrieve the properties
|
||||
// using Show, but this command is provided for the sake of completeness
|
||||
func Status(ctx context.Context, unit string, opts Options) (string, error) {
|
||||
args := []string{"status", "--system", unit}
|
||||
if opts.UserMode {
|
||||
args[1] = "--user"
|
||||
}
|
||||
stdout, _, _, err := execute(ctx, args)
|
||||
return stdout, err
|
||||
//
|
||||
// Any additional arguments are passed directly to the systemctl command.
|
||||
func Status(ctx context.Context, unit string, opts Options, args ...string) (string, error) {
|
||||
return status(ctx, unit, opts, args...)
|
||||
}
|
||||
|
||||
// Stop (deactivate) a given unit
|
||||
func Stop(ctx context.Context, unit string, opts Options) error {
|
||||
args := []string{"stop", "--system", unit}
|
||||
if opts.UserMode {
|
||||
args[1] = "--user"
|
||||
}
|
||||
_, _, _, err := execute(ctx, args)
|
||||
return err
|
||||
//
|
||||
// Any additional arguments are passed directly to the systemctl command.
|
||||
func Stop(ctx context.Context, unit string, opts Options, args ...string) error {
|
||||
return stop(ctx, unit, opts, args...)
|
||||
}
|
||||
|
||||
// Unmask one or more unit files, as specified on the command line.
|
||||
@@ -236,11 +144,8 @@ func Stop(ctx context.Context, unit string, opts Options) error {
|
||||
// doesn't exist, but only if it's not already masked.
|
||||
// If the unit doesn't exist but it's masked anyway, no error will be
|
||||
// returned. Gross, I know. Take it up with Poettering.
|
||||
func Unmask(ctx context.Context, unit string, opts Options) error {
|
||||
args := []string{"unmask", "--system", unit}
|
||||
if opts.UserMode {
|
||||
args[1] = "--user"
|
||||
}
|
||||
_, _, _, err := execute(ctx, args)
|
||||
return err
|
||||
//
|
||||
// Any additional arguments are passed directly to the systemctl command.
|
||||
func Unmask(ctx context.Context, unit string, opts Options, args ...string) error {
|
||||
return unmask(ctx, unit, opts, args...)
|
||||
}
|
||||
|
||||
65
systemctl_darwin.go
Normal file
65
systemctl_darwin.go
Normal file
@@ -0,0 +1,65 @@
|
||||
//go:build !linux
|
||||
|
||||
package systemctl
|
||||
|
||||
import (
|
||||
"context"
|
||||
|
||||
"github.com/taigrr/systemctl/properties"
|
||||
)
|
||||
|
||||
func daemonReload(_ context.Context, _ Options, _ ...string) error {
|
||||
return nil
|
||||
}
|
||||
|
||||
func reenable(_ context.Context, _ string, _ Options, _ ...string) error {
|
||||
return nil
|
||||
}
|
||||
|
||||
func disable(_ context.Context, _ string, _ Options, _ ...string) error {
|
||||
return nil
|
||||
}
|
||||
|
||||
func enable(_ context.Context, _ string, _ Options, _ ...string) error {
|
||||
return nil
|
||||
}
|
||||
|
||||
func isActive(_ context.Context, _ string, _ Options, _ ...string) (bool, error) {
|
||||
return false, nil
|
||||
}
|
||||
|
||||
func isEnabled(_ context.Context, _ string, _ Options, _ ...string) (bool, error) {
|
||||
return false, nil
|
||||
}
|
||||
|
||||
func isFailed(_ context.Context, _ string, _ Options, _ ...string) (bool, error) {
|
||||
return false, nil
|
||||
}
|
||||
|
||||
func mask(_ context.Context, _ string, _ Options, _ ...string) error {
|
||||
return nil
|
||||
}
|
||||
|
||||
func restart(_ context.Context, _ string, _ Options, _ ...string) error {
|
||||
return nil
|
||||
}
|
||||
|
||||
func show(_ context.Context, _ string, _ properties.Property, _ Options, _ ...string) (string, error) {
|
||||
return "", nil
|
||||
}
|
||||
|
||||
func start(_ context.Context, _ string, _ Options, _ ...string) error {
|
||||
return nil
|
||||
}
|
||||
|
||||
func status(_ context.Context, _ string, _ Options, _ ...string) (string, error) {
|
||||
return "", nil
|
||||
}
|
||||
|
||||
func stop(_ context.Context, _ string, _ Options, _ ...string) error {
|
||||
return nil
|
||||
}
|
||||
|
||||
func unmask(_ context.Context, _ string, _ Options, _ ...string) error {
|
||||
return nil
|
||||
}
|
||||
149
systemctl_linux.go
Normal file
149
systemctl_linux.go
Normal file
@@ -0,0 +1,149 @@
|
||||
//go:build linux
|
||||
|
||||
package systemctl
|
||||
|
||||
import (
|
||||
"context"
|
||||
"strings"
|
||||
|
||||
"github.com/taigrr/systemctl/properties"
|
||||
)
|
||||
|
||||
func daemonReload(ctx context.Context, opts Options, args ...string) error {
|
||||
a := prepareArgs("daemon-reload", opts, args...)
|
||||
_, _, _, err := execute(ctx, a)
|
||||
return err
|
||||
}
|
||||
|
||||
func reenable(ctx context.Context, unit string, opts Options, args ...string) error {
|
||||
a := prepareArgs("reenable", opts, append([]string{unit}, args...)...)
|
||||
_, _, _, err := execute(ctx, a)
|
||||
return err
|
||||
}
|
||||
|
||||
func disable(ctx context.Context, unit string, opts Options, args ...string) error {
|
||||
a := prepareArgs("disable", opts, append([]string{unit}, args...)...)
|
||||
_, _, _, err := execute(ctx, a)
|
||||
return err
|
||||
}
|
||||
|
||||
func enable(ctx context.Context, unit string, opts Options, args ...string) error {
|
||||
a := prepareArgs("enable", opts, append([]string{unit}, args...)...)
|
||||
_, _, _, err := execute(ctx, a)
|
||||
return err
|
||||
}
|
||||
|
||||
func isActive(ctx context.Context, unit string, opts Options, args ...string) (bool, error) {
|
||||
a := prepareArgs("is-active", opts, append([]string{unit}, args...)...)
|
||||
stdout, _, _, err := execute(ctx, a)
|
||||
stdout = strings.TrimSuffix(stdout, "\n")
|
||||
switch stdout {
|
||||
case "inactive":
|
||||
return false, nil
|
||||
case "active":
|
||||
return true, nil
|
||||
case "failed":
|
||||
return false, nil
|
||||
case "activating":
|
||||
return false, nil
|
||||
default:
|
||||
return false, err
|
||||
}
|
||||
}
|
||||
|
||||
func isEnabled(ctx context.Context, unit string, opts Options, args ...string) (bool, error) {
|
||||
a := prepareArgs("is-enabled", opts, append([]string{unit}, args...)...)
|
||||
stdout, _, _, err := execute(ctx, a)
|
||||
stdout = strings.TrimSuffix(stdout, "\n")
|
||||
switch stdout {
|
||||
case "enabled":
|
||||
return true, nil
|
||||
case "enabled-runtime":
|
||||
return true, nil
|
||||
case "linked":
|
||||
return false, ErrLinked
|
||||
case "linked-runtime":
|
||||
return false, ErrLinked
|
||||
case "alias":
|
||||
return true, nil
|
||||
case "masked":
|
||||
return false, ErrMasked
|
||||
case "masked-runtime":
|
||||
return false, ErrMasked
|
||||
case "static":
|
||||
return true, nil
|
||||
case "indirect":
|
||||
return true, nil
|
||||
case "disabled":
|
||||
return false, nil
|
||||
case "generated":
|
||||
return true, nil
|
||||
case "transient":
|
||||
return true, nil
|
||||
}
|
||||
if err != nil {
|
||||
return false, err
|
||||
}
|
||||
return false, ErrUnspecified
|
||||
}
|
||||
|
||||
func isFailed(ctx context.Context, unit string, opts Options, args ...string) (bool, error) {
|
||||
a := prepareArgs("is-failed", opts, append([]string{unit}, args...)...)
|
||||
stdout, _, _, err := execute(ctx, a)
|
||||
stdout = strings.TrimSuffix(stdout, "\n")
|
||||
switch stdout {
|
||||
case "inactive":
|
||||
return false, nil
|
||||
case "active":
|
||||
return false, nil
|
||||
case "failed":
|
||||
return true, nil
|
||||
default:
|
||||
return false, err
|
||||
}
|
||||
}
|
||||
|
||||
func mask(ctx context.Context, unit string, opts Options, args ...string) error {
|
||||
a := prepareArgs("mask", opts, append([]string{unit}, args...)...)
|
||||
_, _, _, err := execute(ctx, a)
|
||||
return err
|
||||
}
|
||||
|
||||
func restart(ctx context.Context, unit string, opts Options, args ...string) error {
|
||||
a := prepareArgs("restart", opts, append([]string{unit}, args...)...)
|
||||
_, _, _, err := execute(ctx, a)
|
||||
return err
|
||||
}
|
||||
|
||||
func show(ctx context.Context, unit string, property properties.Property, opts Options, args ...string) (string, error) {
|
||||
extra := append([]string{unit, "--property", string(property)}, args...)
|
||||
a := prepareArgs("show", opts, extra...)
|
||||
stdout, _, _, err := execute(ctx, a)
|
||||
stdout = strings.TrimPrefix(stdout, string(property)+"=")
|
||||
stdout = strings.TrimSuffix(stdout, "\n")
|
||||
return stdout, err
|
||||
}
|
||||
|
||||
func start(ctx context.Context, unit string, opts Options, args ...string) error {
|
||||
a := prepareArgs("start", opts, append([]string{unit}, args...)...)
|
||||
_, _, _, err := execute(ctx, a)
|
||||
return err
|
||||
}
|
||||
|
||||
func status(ctx context.Context, unit string, opts Options, args ...string) (string, error) {
|
||||
a := prepareArgs("status", opts, append([]string{unit}, args...)...)
|
||||
stdout, _, _, err := execute(ctx, a)
|
||||
return stdout, err
|
||||
}
|
||||
|
||||
func stop(ctx context.Context, unit string, opts Options, args ...string) error {
|
||||
a := prepareArgs("stop", opts, append([]string{unit}, args...)...)
|
||||
_, _, _, err := execute(ctx, a)
|
||||
return err
|
||||
}
|
||||
|
||||
func unmask(ctx context.Context, unit string, opts Options, args ...string) error {
|
||||
a := prepareArgs("unmask", opts, append([]string{unit}, args...)...)
|
||||
_, _, _, err := execute(ctx, a)
|
||||
return err
|
||||
}
|
||||
19
util.go
19
util.go
@@ -11,6 +11,7 @@ import (
|
||||
|
||||
var systemctl string
|
||||
|
||||
// killed is the exit code returned when a process is terminated by SIGINT.
|
||||
const killed = 130
|
||||
|
||||
func init() {
|
||||
@@ -39,6 +40,10 @@ func execute(ctx context.Context, args []string) (string, string, int, error) {
|
||||
warnings = stderr.String()
|
||||
code = cmd.ProcessState.ExitCode()
|
||||
|
||||
if code == killed {
|
||||
return output, warnings, code, ErrExecTimeout
|
||||
}
|
||||
|
||||
customErr := filterErr(warnings)
|
||||
if customErr != nil {
|
||||
err = customErr
|
||||
@@ -50,6 +55,20 @@ func execute(ctx context.Context, args []string) (string, string, int, error) {
|
||||
return output, warnings, code, err
|
||||
}
|
||||
|
||||
// prepareArgs builds the systemctl command arguments from a base command,
|
||||
// options, and any additional arguments the caller wants to pass through.
|
||||
func prepareArgs(base string, opts Options, extra ...string) []string {
|
||||
args := make([]string, 0, 2+len(extra))
|
||||
args = append(args, base)
|
||||
if opts.UserMode {
|
||||
args = append(args, "--user")
|
||||
} else {
|
||||
args = append(args, "--system")
|
||||
}
|
||||
args = append(args, extra...)
|
||||
return args
|
||||
}
|
||||
|
||||
func filterErr(stderr string) error {
|
||||
switch {
|
||||
case strings.Contains(stderr, `does not exist`):
|
||||
|
||||
62
util_test.go
Normal file
62
util_test.go
Normal file
@@ -0,0 +1,62 @@
|
||||
package systemctl
|
||||
|
||||
import (
|
||||
"reflect"
|
||||
"testing"
|
||||
)
|
||||
|
||||
func TestPrepareArgs(t *testing.T) {
|
||||
tests := []struct {
|
||||
name string
|
||||
base string
|
||||
opts Options
|
||||
extra []string
|
||||
expected []string
|
||||
}{
|
||||
{
|
||||
name: "system mode no extra",
|
||||
base: "start",
|
||||
opts: Options{},
|
||||
extra: nil,
|
||||
expected: []string{"start", "--system"},
|
||||
},
|
||||
{
|
||||
name: "user mode no extra",
|
||||
base: "start",
|
||||
opts: Options{UserMode: true},
|
||||
extra: nil,
|
||||
expected: []string{"start", "--user"},
|
||||
},
|
||||
{
|
||||
name: "system mode with unit",
|
||||
base: "start",
|
||||
opts: Options{},
|
||||
extra: []string{"nginx.service"},
|
||||
expected: []string{"start", "--system", "nginx.service"},
|
||||
},
|
||||
{
|
||||
name: "user mode with unit and extra args",
|
||||
base: "restart",
|
||||
opts: Options{UserMode: true},
|
||||
extra: []string{"foo.service", "--no-block"},
|
||||
expected: []string{"restart", "--user", "foo.service", "--no-block"},
|
||||
},
|
||||
{
|
||||
name: "daemon-reload no extra",
|
||||
base: "daemon-reload",
|
||||
opts: Options{},
|
||||
extra: nil,
|
||||
expected: []string{"daemon-reload", "--system"},
|
||||
},
|
||||
}
|
||||
|
||||
for _, tt := range tests {
|
||||
t.Run(tt.name, func(t *testing.T) {
|
||||
got := prepareArgs(tt.base, tt.opts, tt.extra...)
|
||||
if !reflect.DeepEqual(got, tt.expected) {
|
||||
t.Errorf("prepareArgs(%q, %+v, %v) = %v, want %v",
|
||||
tt.base, tt.opts, tt.extra, got, tt.expected)
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user