blob: 7b09a3962eb9998cd81c4a17419846dcd230d941 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
|
package org.openslx.taskmanager.tasks;
import java.util.ArrayList;
import java.util.List;
import org.openslx.taskmanager.api.SystemCommandTask;
import com.google.gson.annotations.Expose;
public class Systemctl extends SystemCommandTask
{
/**
* What to do: start, stop...
*/
@Expose
private String operation;
@Expose
String service;
private StatusObject status = new StatusObject();
@Override
protected String[] initCommandLine()
{
boolean allowAll = false;
boolean asRoot = true;
// Check if operation is allowed
switch ( operation ) {
case "start":
case "stop":
case "enable":
case "disable":
case "restart":
case "status":
break;
case "show":
case "is-active":
allowAll = true;
asRoot = false;
break;
default:
status.addMsg( "Unknown operation: " + operation );
return null;
}
if ( !allowAll ) {
// Check if service is allowed
switch ( service ) {
case "dmsd":
case "dnbd3-server":
break;
case "taskmanager":
case "lighttpd":
if ( !operation.equals( "restart" ) ) {
status.addMsg( "Error: Service " + service + " can only be restarted." );
return null;
}
break;
default:
status.addMsg( "Unknown service: " + operation );
return null;
}
}
List<String> params = new ArrayList<>();
if ( asRoot ) {
params.add( "/usr/bin/sudo" );
params.add( "-n" );
params.add( "-u" );
params.add( "root" );
}
params.add( "/bin/systemctl" );
params.add( operation );
params.add( service );
this.timeoutSeconds = 15;
return params.toArray( new String[ params.size() ] );
}
private boolean errorIfNull( String check, String message )
{
if ( check == null )
status.addMsg( message );
return check == null;
}
@Override
protected boolean processEnded( int exitCode )
{
return exitCode == 0;
}
@Override
protected void processStdOut( String line )
{
status.addMsg( line );
}
@Override
protected void processStdErr( String line )
{
status.addMsg( line );
}
@Override
protected boolean initTask()
{
this.setStatusObject( this.status );
return !errorIfNull( operation, "No operation given" );
}
class StatusObject
{
private String messages = "";
public void addMsg( String str )
{
messages = messages + "\n" + str;
}
}
}
|