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
|
//-----------------------------------------------------------------------------
// Copyright (c) Microsoft Corporation. All rights reserved.
//-----------------------------------------------------------------------------
namespace System.ServiceModel.Administration
{
using System;
using System.Collections.Generic;
using System.Runtime;
using System.Security;
using System.ServiceModel;
using System.ServiceModel.Configuration;
sealed class ManagementExtension
{
static Dictionary<ServiceHostBase, DateTime> services;
static bool activated = false;
static object syncRoot = new object();
static bool isEnabled = GetIsWmiProviderEnabled();
internal static bool IsActivated
{
get { return ManagementExtension.activated; }
}
internal static bool IsEnabled
{
get { return ManagementExtension.isEnabled; }
}
[Fx.Tag.SecurityNote(Critical = "Calls critical method DiagnosticSection.UnsafeGetSection.",
Safe = "Doesn't leak config section instance, just returns a bool.")]
[SecuritySafeCritical]
static bool GetIsWmiProviderEnabled()
{
return DiagnosticSection.UnsafeGetSection().WmiProviderEnabled;
}
static Dictionary<ServiceHostBase, DateTime> GetServices()
{
if (ManagementExtension.services == null)
{
lock (ManagementExtension.syncRoot)
{
if (ManagementExtension.services == null)
{
ManagementExtension.services = new Dictionary<ServiceHostBase, DateTime>();
}
}
}
return ManagementExtension.services;
}
internal static ICollection<ServiceHostBase> Services
{
get
{
return GetServices().Keys;
}
}
internal static DateTime GetTimeOpened(ServiceHostBase service)
{
return GetServices()[service];
}
public static void OnServiceOpened(ServiceHostBase serviceHostBase)
{
EnsureManagementProvider();
Add(serviceHostBase);
}
public static void OnServiceClosing(ServiceHostBase serviceHostBase)
{
Remove(serviceHostBase);
}
static void Add(ServiceHostBase service)
{
Dictionary<ServiceHostBase, DateTime> services = GetServices();
lock (services)
{
if (!services.ContainsKey(service))
{
services.Add(service, DateTime.Now);
}
}
}
static void Remove(ServiceHostBase service)
{
Dictionary<ServiceHostBase, DateTime> services = GetServices();
lock (services)
{
if (services.ContainsKey(service))
{
services.Remove(service);
}
}
}
static void EnsureManagementProvider()
{
if (!ManagementExtension.activated)
{
lock (ManagementExtension.syncRoot)
{
if (!ManagementExtension.activated)
{
Activate();
ManagementExtension.activated = true;
}
}
}
}
static void Activate()
{
WbemProvider wmi = new WbemProvider(AdministrationStrings.IndigoNamespace, AdministrationStrings.IndigoAppName);
wmi.Register(AdministrationStrings.AppDomainInfo, new AppDomainInstanceProvider());
wmi.Register(AdministrationStrings.Service, new ServiceInstanceProvider());
wmi.Register(AdministrationStrings.Contract, new ContractInstanceProvider());
wmi.Register(AdministrationStrings.Endpoint, new EndpointInstanceProvider());
wmi.Register(AdministrationStrings.ServiceAppDomain, new ServiceAppDomainAssociationProvider());
wmi.Register(AdministrationStrings.ServiceToEndpointAssociation, new ServiceEndpointAssociationProvider());
}
}
}
|
