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
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
|
using Duplicati.Service;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Runtime.InteropServices;
using System.Text;
using System.Threading.Tasks;
namespace Duplicati.WindowsService
{
public class ServiceControl : System.ServiceProcess.ServiceBase
{
private const string LOG_SOURCE = "Duplicati";
private const string LOG_NAME = "Application";
public const string SERVICE_NAME = "Duplicati";
public const string DISPLAY_NAME = "Duplicati service";
public const string DESCRIPTION = "Runs Duplicati as a service";
private System.Diagnostics.EventLog m_eventLog;
private readonly object m_lock = new object();
private Runner m_runner = null;
private readonly string[] m_cmdargs;
private readonly bool m_verbose_messages;
public ServiceControl(string[] args)
{
this.ServiceName = SERVICE_NAME;
m_eventLog = new System.Diagnostics.EventLog();
if (!System.Diagnostics.EventLog.SourceExists(LOG_SOURCE))
System.Diagnostics.EventLog.CreateEventSource(LOG_SOURCE, LOG_NAME);
m_eventLog.Source = LOG_SOURCE;
m_eventLog.Log = LOG_NAME;
m_verbose_messages = args != null && args.Any(x => string.Equals("--debug-service", x, StringComparison.OrdinalIgnoreCase));
m_cmdargs = (args ?? new string[0]).Where(x => !string.Equals("--debug-service", x, StringComparison.OrdinalIgnoreCase)).ToArray();
}
protected override void OnStart(string[] args)
{
DoStart(args);
}
protected override void OnStop()
{
DoStop();
}
protected override void OnShutdown()
{
DoStop();
}
private void DoStart(string[] args)
{
var startargs = (args ?? new string[0]).Union(m_cmdargs ?? new string[0]).ToArray();
if (m_verbose_messages)
m_eventLog.WriteEntry("Starting...");
lock(m_lock)
if (m_runner == null)
{
if (m_verbose_messages)
m_eventLog.WriteEntry("Set start time to 30 seconds...");
var sv = new ServiceStatus()
{
dwCurrentState = ServiceState.SERVICE_START_PENDING,
dwWaitHint = (uint)TimeSpan.FromSeconds(30).TotalMilliseconds
};
SetServiceStatus(this.ServiceHandle, ref sv);
if (m_verbose_messages)
m_eventLog.WriteEntry("Starting runner...");
m_runner = new Runner(
startargs,
() =>
{
if (m_verbose_messages)
m_eventLog.WriteEntry("Started!");
var sv2 = new ServiceStatus()
{
dwCurrentState = ServiceState.SERVICE_RUNNING
};
SetServiceStatus(this.ServiceHandle, ref sv2);
},
() =>
{
if (m_verbose_messages)
m_eventLog.WriteEntry("Stopped!");
var sv2 = new ServiceStatus()
{
dwCurrentState = ServiceState.SERVICE_STOPPED
};
SetServiceStatus(this.ServiceHandle, ref sv2);
base.Stop();
},
(msg, important) =>
{
if (important || m_verbose_messages)
m_eventLog.WriteEntry(msg);
}
);
}
}
private void DoStop()
{
if (m_verbose_messages)
m_eventLog.WriteEntry("Stopping...");
lock (m_lock)
if (m_runner != null)
{
if (m_verbose_messages)
m_eventLog.WriteEntry("Calling stop...");
var sv = new ServiceStatus()
{
dwCurrentState = ServiceState.SERVICE_STOP_PENDING,
dwWaitHint = (uint)TimeSpan.FromSeconds(5).TotalMilliseconds
};
SetServiceStatus(this.ServiceHandle, ref sv);
if (m_verbose_messages)
m_eventLog.WriteEntry("Soft stop invoked...");
m_runner.Stop(false);
}
}
private enum ServiceState : uint
{
SERVICE_STOPPED = 0x00000001,
SERVICE_START_PENDING = 0x00000002,
SERVICE_STOP_PENDING = 0x00000003,
SERVICE_RUNNING = 0x00000004,
SERVICE_CONTINUE_PENDING = 0x00000005,
SERVICE_PAUSE_PENDING = 0x00000006,
SERVICE_PAUSED = 0x00000007,
}
[StructLayout(LayoutKind.Sequential)]
private struct ServiceStatus
{
public uint dwServiceType;
public ServiceState dwCurrentState;
public uint dwControlsAccepted;
public uint dwWin32ExitCode;
public uint dwServiceSpecificExitCode;
public uint dwCheckPoint;
public uint dwWaitHint;
};
[DllImport("advapi32.dll", SetLastError = true)]
private static extern bool SetServiceStatus(IntPtr handle, ref ServiceStatus serviceStatus);
}
}
|
