Revize e6a01bd8
Přidáno uživatelem Pultak před více než 2 roky(ů)
ld_client/LDClient/detection/ProcessDetection.cs | ||
---|---|---|
1 |
using System.Diagnostics; |
|
2 |
using LDClient.network; |
|
1 |
using LDClient.network; |
|
3 | 2 |
using LDClient.network.data; |
4 | 3 |
|
5 |
namespace LDClient.detection { |
|
4 |
namespace LDClient.detection; |
|
5 |
|
|
6 |
/// <summary> |
|
7 |
/// This class takes care of process detection. When t32mtc (process) |
|
8 |
/// is detected, it means that the debugger is currently being used. |
|
9 |
/// The class keeps track of the current state of a debugger. |
|
10 |
/// </summary> |
|
11 |
public sealed class ProcessDetection : IProcessDetection { |
|
6 | 12 |
|
7 | 13 |
/// <summary> |
8 |
/// This class takes care of process detection. When t32mtc (process) |
|
9 |
/// is detected, it means that the debugger is currently being used. |
|
10 |
/// The class keeps track of the current state of a debugger. |
|
14 |
/// Datetime format used when sending payloads to the server. |
|
11 | 15 |
/// </summary> |
12 |
public sealed class ProcessDetection : IProcessDetection { |
|
13 |
|
|
14 |
/// <summary> |
|
15 |
/// Datetime format used when sending payloads to the server. |
|
16 |
/// </summary> |
|
17 |
private const string DatetimeFormat = "yyyy-MM-dd hh:mm:ss"; |
|
16 |
private const string DatetimeFormat = "yyyy-MM-dd hh:mm:ss"; |
|
18 | 17 |
|
19 |
/// <summary>
|
|
20 |
/// Name of the process the application detects.
|
|
21 |
/// </summary>
|
|
22 |
private readonly string _processName;
|
|
18 |
/// <summary> |
|
19 |
/// Name of the process the application detects. |
|
20 |
/// </summary> |
|
21 |
private readonly string _processName; |
|
23 | 22 |
|
24 |
/// <summary>
|
|
25 |
/// How often the application check the current status of the process (cunning / not running).
|
|
26 |
/// </summary>
|
|
27 |
private readonly uint _detectionPeriodMs;
|
|
23 |
/// <summary> |
|
24 |
/// How often the application check the current status of the process (cunning / not running). |
|
25 |
/// </summary> |
|
26 |
private readonly uint _detectionPeriodMs; |
|
28 | 27 |
|
29 |
/// <summary>
|
|
30 |
/// Instance of InfoFetcher used to fetch information from the debugger
|
|
31 |
/// (when the process is detected).
|
|
32 |
/// </summary>
|
|
33 |
private readonly IInfoFetcher _infoFetcher;
|
|
28 |
/// <summary> |
|
29 |
/// Instance of InfoFetcher used to fetch information from the debugger |
|
30 |
/// (when the process is detected). |
|
31 |
/// </summary> |
|
32 |
private readonly IInfoFetcher _infoFetcher; |
|
34 | 33 |
|
35 |
/// <summary>
|
|
36 |
/// Instance of API clients used for sending data off to the server.
|
|
37 |
/// </summary>
|
|
38 |
private readonly IApiClient _apiClient;
|
|
34 |
/// <summary> |
|
35 |
/// Instance of API clients used for sending data off to the server. |
|
36 |
/// </summary> |
|
37 |
private readonly IApiClient _apiClient; |
|
39 | 38 |
|
40 |
/// <summary>
|
|
41 |
/// Instance of ProcessUtils which encapsulates common functionality
|
|
42 |
/// when it comes to dealing with processes (limited by the needs of this application).
|
|
43 |
/// </summary>
|
|
44 |
private readonly IProcessUtils _processUtils;
|
|
39 |
/// <summary> |
|
40 |
/// Instance of ProcessUtils which encapsulates common functionality |
|
41 |
/// when it comes to dealing with processes (limited by the needs of this application). |
|
42 |
/// </summary> |
|
43 |
private readonly IProcessUtils _processUtils; |
|
45 | 44 |
|
46 |
/// <summary>
|
|
47 |
/// Flag indicating whether the process is currently running or not.
|
|
48 |
/// </summary>
|
|
49 |
private bool _processIsActive;
|
|
45 |
/// <summary> |
|
46 |
/// Flag indicating whether the process is currently running or not. |
|
47 |
/// </summary> |
|
48 |
private bool _processIsActive; |
|
50 | 49 |
|
51 |
/// <summary>
|
|
52 |
/// Flag if the application failed to retrieve data when the process was detected.
|
|
53 |
/// </summary>
|
|
54 |
private bool _failedToRetrieveData;
|
|
50 |
/// <summary> |
|
51 |
/// Flag if the application failed to retrieve data when the process was detected. |
|
52 |
/// </summary> |
|
53 |
private bool _failedToRetrieveData; |
|
55 | 54 |
|
56 |
/// <summary>
|
|
57 |
/// Last payload that was sent to the server.
|
|
58 |
/// </summary>
|
|
59 |
private Payload? _lastConnectedPayload;
|
|
55 |
/// <summary> |
|
56 |
/// Last payload that was sent to the server. |
|
57 |
/// </summary> |
|
58 |
private Payload? _lastConnectedPayload; |
|
60 | 59 |
|
61 |
/// <summary>
|
|
62 |
/// Flag used to stop the thread (process detection).
|
|
63 |
/// </summary>
|
|
64 |
public bool DetectionRunning = false;
|
|
60 |
/// <summary> |
|
61 |
/// Flag used to stop the thread (process detection). |
|
62 |
/// </summary> |
|
63 |
public bool DetectionRunning = false; |
|
65 | 64 |
|
66 |
/// <summary>
|
|
67 |
/// Creates an instance of this class.
|
|
68 |
/// </summary>
|
|
69 |
/// <param name="processName">Name of the process the application detects</param>
|
|
70 |
/// <param name="detectionPeriodMs">How often the application check the current status of the process (cunning / not running)</param>
|
|
71 |
/// <param name="infoFetcher">Instance of InfoFetcher used to fetch information from the debugger</param>
|
|
72 |
/// <param name="apiClient">Instance of API clients used for sending data off to the server</param>
|
|
73 |
/// <param name="processUtils">Instance of ProcessUtils which encapsulates common functionality when it comes to dealing with processes (limited by the needs of this application)</param>
|
|
74 |
public ProcessDetection(string processName, uint detectionPeriodMs, IInfoFetcher infoFetcher,
|
|
75 |
IApiClient apiClient, IProcessUtils processUtils) {
|
|
76 |
_processName = processName;
|
|
77 |
_detectionPeriodMs = detectionPeriodMs;
|
|
78 |
_infoFetcher = infoFetcher;
|
|
79 |
_apiClient = apiClient;
|
|
80 |
_failedToRetrieveData = false;
|
|
81 |
_processUtils = processUtils;
|
|
82 |
}
|
|
65 |
/// <summary> |
|
66 |
/// Creates an instance of this class. |
|
67 |
/// </summary> |
|
68 |
/// <param name="processName">Name of the process the application detects</param> |
|
69 |
/// <param name="detectionPeriodMs">How often the application check the current status of the process (cunning / not running)</param> |
|
70 |
/// <param name="infoFetcher">Instance of InfoFetcher used to fetch information from the debugger</param> |
|
71 |
/// <param name="apiClient">Instance of API clients used for sending data off to the server</param> |
|
72 |
/// <param name="processUtils">Instance of ProcessUtils which encapsulates common functionality when it comes to dealing with processes (limited by the needs of this application)</param> |
|
73 |
public ProcessDetection(string processName, uint detectionPeriodMs, IInfoFetcher infoFetcher, |
|
74 |
IApiClient apiClient, IProcessUtils processUtils) { |
|
75 |
_processName = processName; |
|
76 |
_detectionPeriodMs = detectionPeriodMs; |
|
77 |
_infoFetcher = infoFetcher; |
|
78 |
_apiClient = apiClient; |
|
79 |
_failedToRetrieveData = false; |
|
80 |
_processUtils = processUtils; |
|
81 |
} |
|
83 | 82 |
|
84 |
/// <summary>
|
|
85 |
/// Retrieves data from the debugger.
|
|
86 |
/// </summary>
|
|
87 |
/// <returns>True, if the data was fetched successfully. False, otherwise.</returns>
|
|
88 |
private async Task<bool> RetrieveDataFromDebugger() {
|
|
89 |
// Try to fetch data from the debugger.
|
|
90 |
var success = await _infoFetcher.FetchDataAsync();
|
|
83 |
/// <summary> |
|
84 |
/// Retrieves data from the debugger. |
|
85 |
/// </summary> |
|
86 |
/// <returns>True, if the data was fetched successfully. False, otherwise.</returns> |
|
87 |
private async Task<bool> RetrieveDataFromDebugger() { |
|
88 |
// Try to fetch data from the debugger. |
|
89 |
var success = await _infoFetcher.FetchDataAsync(); |
|
91 | 90 |
|
92 |
// If the data was fetched successfully, send a payload off to the server. |
|
93 |
if (success) { |
|
94 |
_lastConnectedPayload = await SendDataToServerAsync(_infoFetcher.HeadSerialNumber, |
|
95 |
_infoFetcher.BodySerialNumber, DatetimeFormat); |
|
96 |
} |
|
97 |
return success; |
|
91 |
// If the data was fetched successfully, send a payload off to the server. |
|
92 |
if (success) { |
|
93 |
_lastConnectedPayload = await SendDataToServerAsync(_infoFetcher.HeadSerialNumber, |
|
94 |
_infoFetcher.BodySerialNumber, DatetimeFormat); |
|
98 | 95 |
} |
96 |
return success; |
|
97 |
} |
|
99 | 98 |
|
100 |
/// <summary>
|
|
101 |
/// Sends a payload to the server when a debugger gets disconnected.
|
|
102 |
/// </summary>
|
|
103 |
private async Task DebuggerDisconnected() {
|
|
104 |
// Make sure the debugger was connected in the first place.
|
|
105 |
if (_lastConnectedPayload is not null) {
|
|
106 |
// Update the status and timestamp of the last payload
|
|
107 |
// (the serial numbers remain the same).
|
|
108 |
_lastConnectedPayload.Status = ConnectionStatus.Disconnected;
|
|
109 |
_lastConnectedPayload.TimeStamp = DateTime.Now.ToString(DatetimeFormat);
|
|
99 |
/// <summary> |
|
100 |
/// Sends a payload to the server when a debugger gets disconnected. |
|
101 |
/// </summary> |
|
102 |
private async Task DebuggerDisconnected() { |
|
103 |
// Make sure the debugger was connected in the first place. |
|
104 |
if (_lastConnectedPayload is not null) { |
|
105 |
// Update the status and timestamp of the last payload |
|
106 |
// (the serial numbers remain the same). |
|
107 |
_lastConnectedPayload.Status = ConnectionStatus.Disconnected; |
|
108 |
_lastConnectedPayload.TimeStamp = DateTime.Now.ToString(DatetimeFormat); |
|
110 | 109 |
|
111 |
// Send the data to the server.
|
|
112 |
await _apiClient.SendPayloadAsync(_lastConnectedPayload);
|
|
110 |
// Send the data to the server. |
|
111 |
await _apiClient.SendPayloadAsync(_lastConnectedPayload); |
|
113 | 112 |
|
114 |
// Clear the last payload. |
|
115 |
_lastConnectedPayload = null; |
|
116 |
} |
|
113 |
// Clear the last payload. |
|
114 |
_lastConnectedPayload = null; |
|
117 | 115 |
} |
116 |
} |
|
118 | 117 |
|
119 |
/// <summary>
|
|
120 |
/// Checks if the t32mtc process is running or not.
|
|
121 |
/// </summary>
|
|
122 |
private async Task DetectProcessAsync() {
|
|
123 |
// Check if the process is running.
|
|
124 |
var processExists = _processUtils.IsProcessRunning(_processName);
|
|
118 |
/// <summary> |
|
119 |
/// Checks if the t32mtc process is running or not. |
|
120 |
/// </summary> |
|
121 |
private async Task DetectProcessAsync() { |
|
122 |
// Check if the process is running. |
|
123 |
var processExists = _processUtils.IsProcessRunning(_processName); |
|
125 | 124 |
|
126 |
// Check if the process was not running but now it is (flip flop ON). |
|
127 |
if (processExists && !_processIsActive) { |
|
128 |
Program.DefaultLogger.Info($"Process started: {_processName}"); |
|
129 |
if (!_failedToRetrieveData) { |
|
130 |
_failedToRetrieveData = !await RetrieveDataFromDebugger(); |
|
131 |
} |
|
132 |
} |
|
133 |
// Check if the process was running but now it is not (fli flop OFF). |
|
134 |
else if (!processExists && _processIsActive) { |
|
135 |
Program.DefaultLogger.Info($"Process stopped: {_processName}"); |
|
136 |
_failedToRetrieveData = false; |
|
137 |
await DebuggerDisconnected(); |
|
125 |
// Check if the process was not running but now it is (flip flop ON). |
|
126 |
if (processExists && !_processIsActive) { |
|
127 |
Program.DefaultLogger.Info($"Process started: {_processName}"); |
|
128 |
if (!_failedToRetrieveData) { |
|
129 |
_failedToRetrieveData = !await RetrieveDataFromDebugger(); |
|
138 | 130 |
} |
139 |
|
|
140 |
// Keep track of the current state of the debugger. |
|
141 |
_processIsActive = processExists; |
|
131 |
} |
|
132 |
// Check if the process was running but now it is not (fli flop OFF). |
|
133 |
else if (!processExists && _processIsActive) { |
|
134 |
Program.DefaultLogger.Info($"Process stopped: {_processName}"); |
|
135 |
_failedToRetrieveData = false; |
|
136 |
await DebuggerDisconnected(); |
|
142 | 137 |
} |
143 | 138 |
|
144 |
/// <summary> |
|
145 |
/// Creates a payload and sends it to the server. |
|
146 |
/// </summary> |
|
147 |
/// <param name="headSerialNumber">serial number of the head of the debugger</param> |
|
148 |
/// <param name="bodySerialNumber">serial number of the body of the debugger</param> |
|
149 |
/// <param name="datetimeFormat">datetime format (timestamp)</param> |
|
150 |
/// <returns>the newly-created payload</returns> |
|
151 |
private async Task<Payload> SendDataToServerAsync(string headSerialNumber, string bodySerialNumber, string datetimeFormat) { |
|
152 |
// Create a new payload. |
|
153 |
Payload payload = new() { |
|
154 |
UserName = Environment.UserName, |
|
155 |
HostName = Environment.MachineName, |
|
156 |
TimeStamp = DateTime.Now.ToString(datetimeFormat), |
|
157 |
HeadDevice = new DebuggerInfo { |
|
158 |
SerialNumber = headSerialNumber |
|
159 |
}, |
|
160 |
BodyDevice = new DebuggerInfo { |
|
161 |
SerialNumber = bodySerialNumber |
|
162 |
}, |
|
163 |
Status = ConnectionStatus.Connected |
|
164 |
}; |
|
139 |
// Keep track of the current state of the debugger. |
|
140 |
_processIsActive = processExists; |
|
141 |
} |
|
142 |
|
|
143 |
/// <summary> |
|
144 |
/// Creates a payload and sends it to the server. |
|
145 |
/// </summary> |
|
146 |
/// <param name="headSerialNumber">serial number of the head of the debugger</param> |
|
147 |
/// <param name="bodySerialNumber">serial number of the body of the debugger</param> |
|
148 |
/// <param name="datetimeFormat">datetime format (timestamp)</param> |
|
149 |
/// <returns>the newly-created payload</returns> |
|
150 |
private async Task<Payload> SendDataToServerAsync(string headSerialNumber, string bodySerialNumber, string datetimeFormat) { |
|
151 |
// Create a new payload. |
|
152 |
Payload payload = new() { |
|
153 |
UserName = Environment.UserName, |
|
154 |
HostName = Environment.MachineName, |
|
155 |
TimeStamp = DateTime.Now.ToString(datetimeFormat), |
|
156 |
HeadDevice = new DebuggerInfo { |
|
157 |
SerialNumber = headSerialNumber |
|
158 |
}, |
|
159 |
BodyDevice = new DebuggerInfo { |
|
160 |
SerialNumber = bodySerialNumber |
|
161 |
}, |
|
162 |
Status = ConnectionStatus.Connected |
|
163 |
}; |
|
165 | 164 |
|
166 |
// Send it to the server and return it.
|
|
167 |
await _apiClient.SendPayloadAsync(payload);
|
|
168 |
return payload;
|
|
169 |
}
|
|
165 |
// Send it to the server and return it. |
|
166 |
await _apiClient.SendPayloadAsync(payload); |
|
167 |
return payload; |
|
168 |
} |
|
170 | 169 |
|
171 |
/// <summary>
|
|
172 |
/// Periodically runs process detection. This method is instantiated
|
|
173 |
/// as a thread from the main class (Program.cs).
|
|
174 |
/// </summary>
|
|
175 |
public async void RunPeriodicDetection() {
|
|
176 |
Program.DefaultLogger.Info("Process periodic detector has started");
|
|
177 |
DetectionRunning = true;
|
|
170 |
/// <summary> |
|
171 |
/// Periodically runs process detection. This method is instantiated |
|
172 |
/// as a thread from the main class (Program.cs). |
|
173 |
/// </summary> |
|
174 |
public async void RunPeriodicDetection() { |
|
175 |
Program.DefaultLogger.Info("Process periodic detector has started"); |
|
176 |
DetectionRunning = true; |
|
178 | 177 |
|
179 |
while (DetectionRunning) { |
|
180 |
await DetectProcessAsync(); |
|
181 |
Thread.Sleep((int) _detectionPeriodMs); |
|
182 |
} |
|
178 |
while (DetectionRunning) { |
|
179 |
await DetectProcessAsync(); |
|
180 |
Thread.Sleep((int) _detectionPeriodMs); |
|
183 | 181 |
} |
184 | 182 |
} |
185 |
} |
|
183 |
} |
Také k dispozici: Unified diff
re #9712 Added documentation of the newly implemented methods + refactoring