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
|
/*
* Copyright (c) 2014 Cisco Systems, Inc. and others. All rights reserved.
*
* This program and the accompanying materials are made available under the
* terms of the Eclipse Public License v1.0 which accompanies this distribution,
* and is available at http://www.eclipse.org/legal/epl-v10.html
*/
package org.opendaylight.aaa.authz.srv;
import java.util.Collection;
import org.opendaylight.aaa.api.AuthenticationService;
import org.opendaylight.controller.md.sal.dom.api.DOMDataBroker;
import org.opendaylight.controller.sal.core.api.Broker;
import org.opendaylight.controller.sal.core.api.Consumer;
import org.opendaylight.controller.sal.core.api.Provider;
import org.osgi.framework.BundleContext;
/**
* Created by wdec on 26/08/2014.
*/
public class AuthzBrokerImpl implements Broker, AutoCloseable, Provider {
private Broker broker;
private ProviderSession providerSession;
private AuthenticationService authenticationService;
public void setBroker(Broker broker) {
this.broker = broker;
}
@Override
public void close() throws Exception {
}
// Implements AuthzBroker handling of registering consumers or providers.
@Override
public ConsumerSession registerConsumer(Consumer consumer) {
ConsumerSession realSession = broker.registerConsumer(new ConsumerWrapper(consumer));
AuthzConsumerContextImpl authzConsumerContext = new AuthzConsumerContextImpl(realSession,
this);
consumer.onSessionInitiated(authzConsumerContext);
return authzConsumerContext;
}
@Override
public ConsumerSession registerConsumer(Consumer consumer, BundleContext bundleContext) {
ConsumerSession realSession = broker.registerConsumer(new ConsumerWrapper(consumer),
bundleContext);
AuthzConsumerContextImpl authzConsumerContext = new AuthzConsumerContextImpl(realSession,
this);
consumer.onSessionInitiated(authzConsumerContext);
return authzConsumerContext;
}
@Override
public ProviderSession registerProvider(Provider provider) {
ProviderSession realSession = broker.registerProvider(new ProviderWrapper(provider));
AuthzProviderContextImpl authzProviderContext = new AuthzProviderContextImpl(realSession,
this);
provider.onSessionInitiated(authzProviderContext);
return authzProviderContext;
}
@Override
public ProviderSession registerProvider(Provider provider, BundleContext bundleContext) {
// Allow the real broker to do its thing, while providing a wrapped
// callback
ProviderSession realSession = broker.registerProvider(new ProviderWrapper(provider),
bundleContext);
// Create Authz ProviderContext
AuthzProviderContextImpl authzProviderContext = new AuthzProviderContextImpl(realSession,
this);
// Run onsessionInitiated on injected provider with the AuthZ provider
// context.
provider.onSessionInitiated(authzProviderContext);
return authzProviderContext;
}
// Handle the AuthZBroker registration with the real broker
@Override
public void onSessionInitiated(ProviderSession providerSession) {
// Get now the real DOMDataBroker and register it with the
// AuthzDOMBroker together with the provider session
final DOMDataBroker domDataBroker = providerSession.getService(DOMDataBroker.class);
AuthzDomDataBroker.getInstance().setProviderSession(providerSession);
AuthzDomDataBroker.getInstance().setDomDataBroker(domDataBroker);
AuthzDomDataBroker.getInstance().setAuthService(this.authenticationService);
}
@Override
public Collection<ProviderFunctionality> getProviderFunctionality() {
return null;
}
public void setAuthenticationService(AuthenticationService authenticationService) {
this.authenticationService = authenticationService;
}
// Wrapper for Provider
public static class ProviderWrapper implements Provider {
private final Provider provider;
public ProviderWrapper(Provider provider) {
this.provider = provider;
}
@Override
public void onSessionInitiated(ProviderSession providerSession) {
// Do a Noop when the real broker calls back
}
@Override
public Collection<ProviderFunctionality> getProviderFunctionality() {
// Allow the RestconfImpl to respond to this
return provider.getProviderFunctionality();
}
}
// Wrapper for Consumer
public static class ConsumerWrapper implements Consumer {
private final Consumer consumer;
public ConsumerWrapper(Consumer consumer) {
this.consumer = consumer;
}
@Override
public void onSessionInitiated(ConsumerSession consumerSession) {
// Do a Noop when the real broker calls back
}
@Override
public Collection<ConsumerFunctionality> getConsumerFunctionality() {
return consumer.getConsumerFunctionality();
}
}
}
|
