| | | 1 | | // Copyright (c) ZeroC, Inc. |
| | | 2 | | |
| | | 3 | | using System.Diagnostics; |
| | | 4 | | |
| | | 5 | | namespace Ice.Internal; |
| | | 6 | | |
| | | 7 | | internal sealed class LoggerAdminI : Ice.LoggerAdminDisp_ |
| | | 8 | | { |
| | | 9 | | public override void |
| | | 10 | | attachRemoteLogger( |
| | | 11 | | Ice.RemoteLoggerPrx prx, |
| | | 12 | | Ice.LogMessageType[] messageTypes, |
| | | 13 | | string[] categories, |
| | | 14 | | int messageMax, |
| | | 15 | | Ice.Current current) |
| | | 16 | | { |
| | 1 | 17 | | if (prx == null) |
| | | 18 | | { |
| | 0 | 19 | | return; // can't send this null RemoteLogger anything! |
| | | 20 | | } |
| | | 21 | | |
| | 1 | 22 | | Ice.RemoteLoggerPrx remoteLogger = Ice.RemoteLoggerPrxHelper.uncheckedCast(prx.ice_twoway()); |
| | | 23 | | |
| | 1 | 24 | | var filters = new Filters(messageTypes, categories); |
| | 1 | 25 | | LinkedList<Ice.LogMessage> initLogMessages = null; |
| | | 26 | | |
| | 1 | 27 | | lock (_mutex) |
| | | 28 | | { |
| | 1 | 29 | | if (_sendLogCommunicator == null) |
| | | 30 | | { |
| | 1 | 31 | | if (_destroyed) |
| | | 32 | | { |
| | 0 | 33 | | throw new Ice.ObjectNotExistException(); |
| | | 34 | | } |
| | | 35 | | |
| | 1 | 36 | | _sendLogCommunicator = |
| | 1 | 37 | | createSendLogCommunicator(current.adapter.getCommunicator(), _logger.getLocalLogger()); |
| | | 38 | | } |
| | | 39 | | |
| | 1 | 40 | | Ice.Identity remoteLoggerId = remoteLogger.ice_getIdentity(); |
| | | 41 | | |
| | 1 | 42 | | if (_remoteLoggerMap.ContainsKey(remoteLoggerId)) |
| | | 43 | | { |
| | 1 | 44 | | if (_traceLevel > 0) |
| | | 45 | | { |
| | 0 | 46 | | _logger.trace(_traceCategory, "rejecting `" + remoteLogger.ToString() + |
| | 0 | 47 | | "' with RemoteLoggerAlreadyAttachedException"); |
| | | 48 | | } |
| | | 49 | | |
| | 1 | 50 | | throw new Ice.RemoteLoggerAlreadyAttachedException(); |
| | | 51 | | } |
| | | 52 | | |
| | 1 | 53 | | _remoteLoggerMap.Add( |
| | 1 | 54 | | remoteLoggerId, |
| | 1 | 55 | | new RemoteLoggerData(changeCommunicator(remoteLogger, _sendLogCommunicator), filters)); |
| | | 56 | | |
| | 1 | 57 | | if (messageMax != 0) |
| | | 58 | | { |
| | 1 | 59 | | initLogMessages = new LinkedList<Ice.LogMessage>(_queue); // copy |
| | | 60 | | } |
| | | 61 | | else |
| | | 62 | | { |
| | 0 | 63 | | initLogMessages = new LinkedList<Ice.LogMessage>(); |
| | | 64 | | } |
| | 0 | 65 | | } |
| | | 66 | | |
| | 1 | 67 | | if (_traceLevel > 0) |
| | | 68 | | { |
| | 0 | 69 | | _logger.trace(_traceCategory, "attached `" + remoteLogger.ToString() + "'"); |
| | | 70 | | } |
| | | 71 | | |
| | 1 | 72 | | if (initLogMessages.Count > 0) |
| | | 73 | | { |
| | 1 | 74 | | filterLogMessages(initLogMessages, filters.messageTypes, filters.traceCategories, messageMax); |
| | | 75 | | } |
| | | 76 | | |
| | 1 | 77 | | _ = performInitAsync(); |
| | | 78 | | |
| | | 79 | | async Task performInitAsync() |
| | | 80 | | { |
| | | 81 | | try |
| | | 82 | | { |
| | 1 | 83 | | await remoteLogger.initAsync(_logger.getPrefix(), initLogMessages.ToArray()).ConfigureAwait(false); |
| | 1 | 84 | | if (_traceLevel > 1) |
| | | 85 | | { |
| | 0 | 86 | | _logger.trace( |
| | 0 | 87 | | _traceCategory, |
| | 0 | 88 | | $"init on `{remoteLogger}' completed successfully"); |
| | | 89 | | } |
| | 1 | 90 | | } |
| | 0 | 91 | | catch (Ice.LocalException ex) |
| | | 92 | | { |
| | 0 | 93 | | deadRemoteLogger(remoteLogger, _logger, ex, "init"); |
| | 0 | 94 | | } |
| | 0 | 95 | | catch (System.Exception ex) |
| | | 96 | | { |
| | | 97 | | Debug.Fail($"unexpected exception {ex}"); |
| | 0 | 98 | | throw; |
| | | 99 | | } |
| | 1 | 100 | | } |
| | 1 | 101 | | } |
| | | 102 | | |
| | | 103 | | public override bool |
| | | 104 | | detachRemoteLogger(Ice.RemoteLoggerPrx remoteLogger, Ice.Current current) |
| | | 105 | | { |
| | 1 | 106 | | if (remoteLogger == null) |
| | | 107 | | { |
| | 0 | 108 | | return false; |
| | | 109 | | } |
| | | 110 | | |
| | | 111 | | // |
| | | 112 | | // No need to convert the proxy as we only use its identity |
| | | 113 | | // |
| | 1 | 114 | | bool found = removeRemoteLogger(remoteLogger); |
| | | 115 | | |
| | 1 | 116 | | if (_traceLevel > 0) |
| | | 117 | | { |
| | 0 | 118 | | if (found) |
| | | 119 | | { |
| | 0 | 120 | | _logger.trace(_traceCategory, "detached `" + remoteLogger.ToString() + "'"); |
| | | 121 | | } |
| | | 122 | | else |
| | | 123 | | { |
| | 0 | 124 | | _logger.trace(_traceCategory, "cannot detach `" + remoteLogger.ToString() + "': not found"); |
| | | 125 | | } |
| | | 126 | | } |
| | | 127 | | |
| | 1 | 128 | | return found; |
| | | 129 | | } |
| | | 130 | | |
| | | 131 | | public override Ice.LogMessage[] |
| | | 132 | | getLog( |
| | | 133 | | Ice.LogMessageType[] messageTypes, |
| | | 134 | | string[] categories, |
| | | 135 | | int messageMax, |
| | | 136 | | out string prefix, |
| | | 137 | | Ice.Current current) |
| | | 138 | | { |
| | 1 | 139 | | LinkedList<Ice.LogMessage> logMessages = null; |
| | 1 | 140 | | lock (_mutex) |
| | | 141 | | { |
| | 1 | 142 | | if (messageMax != 0) |
| | | 143 | | { |
| | 1 | 144 | | logMessages = new LinkedList<Ice.LogMessage>(_queue); |
| | | 145 | | } |
| | | 146 | | else |
| | | 147 | | { |
| | 0 | 148 | | logMessages = new LinkedList<Ice.LogMessage>(); |
| | | 149 | | } |
| | 0 | 150 | | } |
| | | 151 | | |
| | 1 | 152 | | prefix = _logger.getPrefix(); |
| | | 153 | | |
| | 1 | 154 | | if (logMessages.Count > 0) |
| | | 155 | | { |
| | 1 | 156 | | var filters = new Filters(messageTypes, categories); |
| | 1 | 157 | | filterLogMessages(logMessages, filters.messageTypes, filters.traceCategories, messageMax); |
| | | 158 | | } |
| | 1 | 159 | | return logMessages.ToArray(); |
| | | 160 | | } |
| | | 161 | | |
| | 1 | 162 | | internal LoggerAdminI(Ice.Properties props, LoggerAdminLoggerI logger) |
| | | 163 | | { |
| | 1 | 164 | | _maxLogCount = props.getIcePropertyAsInt("Ice.Admin.Logger.KeepLogs"); |
| | 1 | 165 | | _maxTraceCount = props.getIcePropertyAsInt("Ice.Admin.Logger.KeepTraces"); |
| | 1 | 166 | | _traceLevel = props.getIcePropertyAsInt("Ice.Trace.Admin.Logger"); |
| | 1 | 167 | | _logger = logger; |
| | 1 | 168 | | } |
| | | 169 | | |
| | | 170 | | internal void destroy() |
| | | 171 | | { |
| | 1 | 172 | | Ice.Communicator sendLogCommunicator = null; |
| | | 173 | | |
| | 1 | 174 | | lock (_mutex) |
| | | 175 | | { |
| | 1 | 176 | | if (!_destroyed) |
| | | 177 | | { |
| | 1 | 178 | | _destroyed = true; |
| | 1 | 179 | | sendLogCommunicator = _sendLogCommunicator; |
| | 1 | 180 | | _sendLogCommunicator = null; |
| | | 181 | | } |
| | 1 | 182 | | } |
| | | 183 | | |
| | | 184 | | // |
| | | 185 | | // Destroy outside lock to avoid deadlock when there are outstanding two-way log calls sent to |
| | | 186 | | // remote loggers |
| | | 187 | | // |
| | 1 | 188 | | sendLogCommunicator?.destroy(); |
| | 1 | 189 | | } |
| | | 190 | | |
| | | 191 | | internal List<Ice.RemoteLoggerPrx> log(Ice.LogMessage logMessage) |
| | | 192 | | { |
| | 1 | 193 | | lock (_mutex) |
| | | 194 | | { |
| | 1 | 195 | | List<Ice.RemoteLoggerPrx> remoteLoggers = null; |
| | | 196 | | |
| | | 197 | | // |
| | | 198 | | // Put message in _queue |
| | | 199 | | // |
| | 1 | 200 | | if ((logMessage.type != Ice.LogMessageType.TraceMessage && _maxLogCount > 0) || |
| | 1 | 201 | | (logMessage.type == Ice.LogMessageType.TraceMessage && _maxTraceCount > 0)) |
| | | 202 | | { |
| | 1 | 203 | | _queue.AddLast(logMessage); |
| | | 204 | | |
| | 1 | 205 | | if (logMessage.type != Ice.LogMessageType.TraceMessage) |
| | | 206 | | { |
| | | 207 | | Debug.Assert(_maxLogCount > 0); |
| | 1 | 208 | | if (_logCount == _maxLogCount) |
| | | 209 | | { |
| | | 210 | | // |
| | | 211 | | // Need to remove the oldest log from the queue |
| | | 212 | | // |
| | | 213 | | Debug.Assert(_oldestLog != null); |
| | 0 | 214 | | LinkedListNode<LogMessage> next = _oldestLog.Next; |
| | 0 | 215 | | _queue.Remove(_oldestLog); |
| | 0 | 216 | | _oldestLog = next; |
| | | 217 | | |
| | 0 | 218 | | while (_oldestLog != null && _oldestLog.Value.type == Ice.LogMessageType.TraceMessage) |
| | | 219 | | { |
| | 0 | 220 | | _oldestLog = _oldestLog.Next; |
| | | 221 | | } |
| | | 222 | | Debug.Assert(_oldestLog != null); // remember: we just added a Log at the end |
| | | 223 | | } |
| | | 224 | | else |
| | | 225 | | { |
| | | 226 | | Debug.Assert(_logCount < _maxLogCount); |
| | 1 | 227 | | _logCount++; |
| | 1 | 228 | | _oldestLog ??= _queue.Last; |
| | | 229 | | } |
| | | 230 | | } |
| | | 231 | | else |
| | | 232 | | { |
| | | 233 | | Debug.Assert(_maxTraceCount > 0); |
| | 1 | 234 | | if (_traceCount == _maxTraceCount) |
| | | 235 | | { |
| | | 236 | | // |
| | | 237 | | // Need to remove the oldest trace from the queue |
| | | 238 | | // |
| | | 239 | | Debug.Assert(_oldestTrace != null); |
| | 1 | 240 | | LinkedListNode<LogMessage> next = _oldestTrace.Next; |
| | 1 | 241 | | _queue.Remove(_oldestTrace); |
| | 1 | 242 | | _oldestTrace = next; |
| | | 243 | | |
| | 1 | 244 | | while (_oldestTrace != null && _oldestTrace.Value.type != Ice.LogMessageType.TraceMessage) |
| | | 245 | | { |
| | 0 | 246 | | _oldestTrace = _oldestTrace.Next; |
| | | 247 | | } |
| | | 248 | | Debug.Assert(_oldestTrace != null); // remember: we just added a Log at the end |
| | | 249 | | } |
| | | 250 | | else |
| | | 251 | | { |
| | | 252 | | Debug.Assert(_traceCount < _maxTraceCount); |
| | 1 | 253 | | _traceCount++; |
| | 1 | 254 | | _oldestTrace ??= _queue.Last; |
| | | 255 | | } |
| | | 256 | | } |
| | | 257 | | } |
| | | 258 | | |
| | | 259 | | // |
| | | 260 | | // Queue updated, now find which remote loggers want this message |
| | | 261 | | // |
| | 1 | 262 | | foreach (RemoteLoggerData p in _remoteLoggerMap.Values) |
| | | 263 | | { |
| | 1 | 264 | | Filters filters = p.filters; |
| | | 265 | | |
| | 1 | 266 | | if (filters.messageTypes.Count == 0 || filters.messageTypes.Contains(logMessage.type)) |
| | | 267 | | { |
| | 1 | 268 | | if (logMessage.type != Ice.LogMessageType.TraceMessage || filters.traceCategories.Count == 0 || |
| | 1 | 269 | | filters.traceCategories.Contains(logMessage.traceCategory)) |
| | | 270 | | { |
| | 1 | 271 | | remoteLoggers ??= new List<Ice.RemoteLoggerPrx>(); |
| | 1 | 272 | | remoteLoggers.Add(p.remoteLogger); |
| | | 273 | | } |
| | | 274 | | } |
| | | 275 | | } |
| | | 276 | | |
| | 1 | 277 | | return remoteLoggers; |
| | | 278 | | } |
| | 1 | 279 | | } |
| | | 280 | | |
| | | 281 | | internal void |
| | | 282 | | deadRemoteLogger(Ice.RemoteLoggerPrx remoteLogger, Ice.Logger logger, Ice.LocalException ex, string operation) |
| | | 283 | | { |
| | | 284 | | // |
| | | 285 | | // No need to convert remoteLogger as we only use its identity |
| | | 286 | | // |
| | 0 | 287 | | if (removeRemoteLogger(remoteLogger)) |
| | | 288 | | { |
| | 0 | 289 | | if (_traceLevel > 0) |
| | | 290 | | { |
| | 0 | 291 | | logger.trace(_traceCategory, "detached `" + remoteLogger.ToString() + "' because " |
| | 0 | 292 | | + operation + " raised:\n" + ex.ToString()); |
| | | 293 | | } |
| | | 294 | | } |
| | 0 | 295 | | } |
| | | 296 | | |
| | 1 | 297 | | internal int getTraceLevel() => _traceLevel; |
| | | 298 | | |
| | | 299 | | private static void filterLogMessages( |
| | | 300 | | LinkedList<Ice.LogMessage> logMessages, |
| | | 301 | | HashSet<Ice.LogMessageType> messageTypes, |
| | | 302 | | HashSet<string> traceCategories, |
| | | 303 | | int messageMax) |
| | | 304 | | { |
| | | 305 | | Debug.Assert(logMessages.Count > 0 && messageMax != 0); |
| | | 306 | | |
| | | 307 | | // |
| | | 308 | | // Filter only if one of the 3 filters is set; messageMax < 0 means "give me all" |
| | | 309 | | // that match the other filters, if any. |
| | | 310 | | // |
| | 1 | 311 | | if (messageTypes.Count > 0 || traceCategories.Count > 0 || messageMax > 0) |
| | | 312 | | { |
| | 1 | 313 | | int count = 0; |
| | 1 | 314 | | LinkedListNode<LogMessage> p = logMessages.Last; |
| | 1 | 315 | | while (p != null) |
| | | 316 | | { |
| | 1 | 317 | | bool keepIt = false; |
| | 1 | 318 | | Ice.LogMessage msg = p.Value; |
| | 1 | 319 | | if (messageTypes.Count == 0 || messageTypes.Contains(msg.type)) |
| | | 320 | | { |
| | 1 | 321 | | if (msg.type != Ice.LogMessageType.TraceMessage || traceCategories.Count == 0 || |
| | 1 | 322 | | traceCategories.Contains(msg.traceCategory)) |
| | | 323 | | { |
| | 1 | 324 | | keepIt = true; |
| | | 325 | | } |
| | | 326 | | } |
| | | 327 | | |
| | 1 | 328 | | if (keepIt) |
| | | 329 | | { |
| | 1 | 330 | | ++count; |
| | 1 | 331 | | if (messageMax > 0 && count >= messageMax) |
| | | 332 | | { |
| | | 333 | | // Remove all older messages |
| | 1 | 334 | | p = p.Previous; |
| | 1 | 335 | | while (p != null) |
| | | 336 | | { |
| | 1 | 337 | | LinkedListNode<LogMessage> previous = p.Previous; |
| | 1 | 338 | | logMessages.Remove(p); |
| | 1 | 339 | | p = previous; |
| | | 340 | | } |
| | 1 | 341 | | break; // while |
| | | 342 | | } |
| | | 343 | | else |
| | | 344 | | { |
| | 1 | 345 | | p = p.Previous; |
| | | 346 | | } |
| | | 347 | | } |
| | | 348 | | else |
| | | 349 | | { |
| | 1 | 350 | | LinkedListNode<LogMessage> previous = p.Previous; |
| | 1 | 351 | | logMessages.Remove(p); |
| | 1 | 352 | | p = previous; |
| | | 353 | | } |
| | | 354 | | } |
| | | 355 | | } |
| | | 356 | | // else, don't need any filtering |
| | 1 | 357 | | } |
| | | 358 | | |
| | | 359 | | // |
| | | 360 | | // Change this proxy's communicator, while keeping its invocation timeout |
| | | 361 | | // |
| | | 362 | | private static Ice.RemoteLoggerPrx changeCommunicator(Ice.RemoteLoggerPrx prx, Ice.Communicator communicator) |
| | | 363 | | { |
| | 1 | 364 | | if (prx == null) |
| | | 365 | | { |
| | 0 | 366 | | return null; |
| | | 367 | | } |
| | | 368 | | |
| | 1 | 369 | | ObjectPrx result = ObjectPrxHelper.createProxy(communicator, prx.ToString()); |
| | 1 | 370 | | return RemoteLoggerPrxHelper.uncheckedCast(result.ice_invocationTimeout(prx.ice_getInvocationTimeout())); |
| | | 371 | | } |
| | | 372 | | |
| | | 373 | | private static void copyProperties(string prefix, Ice.Properties from, Ice.Properties to) |
| | | 374 | | { |
| | 1 | 375 | | foreach (KeyValuePair<string, string> p in from.getPropertiesForPrefix(prefix)) |
| | | 376 | | { |
| | 0 | 377 | | to.setProperty(p.Key, p.Value); |
| | | 378 | | } |
| | 1 | 379 | | } |
| | | 380 | | |
| | | 381 | | private static Ice.Communicator createSendLogCommunicator(Ice.Communicator communicator, Ice.Logger logger) |
| | | 382 | | { |
| | 1 | 383 | | var initData = new Ice.InitializationData(); |
| | 1 | 384 | | initData.logger = logger; |
| | 1 | 385 | | initData.properties = new Ice.Properties(); |
| | | 386 | | |
| | 1 | 387 | | Ice.Properties mainProps = communicator.getProperties(); |
| | | 388 | | |
| | 1 | 389 | | copyProperties("Ice.Default.Locator", mainProps, initData.properties); |
| | 1 | 390 | | copyProperties("IceSSL.", mainProps, initData.properties); |
| | | 391 | | |
| | 1 | 392 | | string[] extraProps = mainProps.getIcePropertyAsList("Ice.Admin.Logger.Properties"); |
| | | 393 | | |
| | 1 | 394 | | if (extraProps.Length > 0) |
| | | 395 | | { |
| | 0 | 396 | | for (int i = 0; i < extraProps.Length; ++i) |
| | | 397 | | { |
| | 0 | 398 | | string p = extraProps[i]; |
| | 0 | 399 | | if (!p.StartsWith("--", StringComparison.Ordinal)) |
| | | 400 | | { |
| | 0 | 401 | | extraProps[i] = "--" + p; |
| | | 402 | | } |
| | | 403 | | } |
| | 0 | 404 | | initData.properties.parseCommandLineOptions("", extraProps); |
| | | 405 | | } |
| | 1 | 406 | | return Ice.Util.initialize(initData); |
| | | 407 | | } |
| | | 408 | | |
| | | 409 | | private bool removeRemoteLogger(Ice.RemoteLoggerPrx remoteLogger) |
| | | 410 | | { |
| | 1 | 411 | | lock (_mutex) |
| | | 412 | | { |
| | 1 | 413 | | return _remoteLoggerMap.Remove(remoteLogger.ice_getIdentity()); |
| | | 414 | | } |
| | 1 | 415 | | } |
| | | 416 | | |
| | 1 | 417 | | private readonly LinkedList<Ice.LogMessage> _queue = new LinkedList<Ice.LogMessage>(); |
| | | 418 | | private int _logCount; // non-trace messages |
| | | 419 | | private readonly int _maxLogCount; |
| | | 420 | | private int _traceCount; |
| | | 421 | | private readonly int _maxTraceCount; |
| | | 422 | | private readonly int _traceLevel; |
| | | 423 | | |
| | | 424 | | private LinkedListNode<Ice.LogMessage> _oldestTrace; |
| | | 425 | | private LinkedListNode<Ice.LogMessage> _oldestLog; |
| | 1 | 426 | | private readonly object _mutex = new(); |
| | | 427 | | |
| | | 428 | | private class Filters |
| | | 429 | | { |
| | 1 | 430 | | internal Filters(Ice.LogMessageType[] m, string[] c) |
| | | 431 | | { |
| | 1 | 432 | | messageTypes = new HashSet<Ice.LogMessageType>(m); |
| | 1 | 433 | | traceCategories = new HashSet<string>(c); |
| | 1 | 434 | | } |
| | | 435 | | |
| | | 436 | | internal readonly HashSet<Ice.LogMessageType> messageTypes; |
| | | 437 | | internal readonly HashSet<string> traceCategories; |
| | | 438 | | } |
| | | 439 | | |
| | | 440 | | private class RemoteLoggerData |
| | | 441 | | { |
| | 1 | 442 | | internal RemoteLoggerData(Ice.RemoteLoggerPrx prx, Filters f) |
| | | 443 | | { |
| | 1 | 444 | | remoteLogger = prx; |
| | 1 | 445 | | filters = f; |
| | 1 | 446 | | } |
| | | 447 | | |
| | | 448 | | internal readonly Ice.RemoteLoggerPrx remoteLogger; |
| | | 449 | | internal readonly Filters filters; |
| | | 450 | | } |
| | | 451 | | |
| | 1 | 452 | | private readonly Dictionary<Ice.Identity, RemoteLoggerData> _remoteLoggerMap = new(); |
| | | 453 | | private readonly LoggerAdminLoggerI _logger; |
| | | 454 | | |
| | | 455 | | private Ice.Communicator _sendLogCommunicator; |
| | | 456 | | private bool _destroyed; |
| | | 457 | | private const string _traceCategory = "Admin.Logger"; |
| | | 458 | | } |