-
Notifications
You must be signed in to change notification settings - Fork 402
/
DefaultBootstrapSessionManager.java
198 lines (169 loc) · 8.08 KB
/
DefaultBootstrapSessionManager.java
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
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
/*******************************************************************************
* Copyright (c) 2016 Sierra Wireless and others.
*
* All rights reserved. This program and the accompanying materials
* are made available under the terms of the Eclipse Public License v2.0
* and Eclipse Distribution License v1.0 which accompany this distribution.
*
* The Eclipse Public License is available at
* http://www.eclipse.org/legal/epl-v20.html
* and the Eclipse Distribution License is available at
* http://www.eclipse.org/org/documents/edl-v10.html.
*
* Contributors:
* Sierra Wireless - initial API and implementation
*******************************************************************************/
package org.eclipse.leshan.server.bootstrap;
import java.net.URI;
import java.util.ArrayList;
import java.util.List;
import org.eclipse.leshan.core.request.BootstrapDownlinkRequest;
import org.eclipse.leshan.core.request.BootstrapFinishRequest;
import org.eclipse.leshan.core.request.BootstrapRequest;
import org.eclipse.leshan.core.request.Identity;
import org.eclipse.leshan.core.response.LwM2mResponse;
import org.eclipse.leshan.core.util.Validate;
import org.eclipse.leshan.server.bootstrap.BootstrapTaskProvider.Tasks;
import org.eclipse.leshan.server.model.LwM2mBootstrapModelProvider;
import org.eclipse.leshan.server.model.StandardBootstrapModelProvider;
import org.eclipse.leshan.server.security.Authorization;
import org.eclipse.leshan.server.security.BootstrapAuthorizer;
import org.eclipse.leshan.server.security.BootstrapSecurityStore;
import org.eclipse.leshan.server.security.SecurityChecker;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
/**
* Default implementation of a {@link BootstrapSessionManager}.
* <p>
* Starting a session only checks credentials from BootstrapSecurityStore.
* <p>
* Nothing specific is done on session's end.
*/
public class DefaultBootstrapSessionManager implements BootstrapSessionManager {
private static final Logger LOG = LoggerFactory.getLogger(DefaultBootstrapSessionManager.class);
private final BootstrapTaskProvider tasksProvider;
private final LwM2mBootstrapModelProvider modelProvider;
private final BootstrapAuthorizer authorizer;
/**
* Create a {@link DefaultBootstrapSessionManager} using a default {@link SecurityChecker} to accept or refuse new
* {@link BootstrapSession}.
*/
public DefaultBootstrapSessionManager(BootstrapSecurityStore bsSecurityStore, BootstrapConfigStore configStore) {
this(new BootstrapConfigStoreTaskProvider(configStore), new StandardBootstrapModelProvider(),
new DefaultBootstrapAuthorizer(bsSecurityStore));
}
/**
* Create a {@link DefaultBootstrapSessionManager}.
*/
public DefaultBootstrapSessionManager(BootstrapTaskProvider tasksProvider,
LwM2mBootstrapModelProvider modelProvider, BootstrapAuthorizer authorizer) {
Validate.notNull(tasksProvider);
Validate.notNull(modelProvider);
Validate.notNull(authorizer);
this.tasksProvider = tasksProvider;
this.modelProvider = modelProvider;
this.authorizer = authorizer;
}
@Override
public BootstrapSession begin(BootstrapRequest request, Identity clientIdentity, URI endpointUsed) {
Authorization authorization = authorizer.isAuthorized(request, clientIdentity);
DefaultBootstrapSession session = new DefaultBootstrapSession(request, clientIdentity,
authorization.isApproved(), authorization.getApplicationData(), endpointUsed);
LOG.trace("Bootstrap session started : {}", session);
return session;
}
@Override
public boolean hasConfigFor(BootstrapSession session) {
Tasks firstTasks = tasksProvider.getTasks(session, null);
if (firstTasks == null)
return false;
initTasks(session, firstTasks);
return true;
}
protected void initTasks(BootstrapSession bssession, Tasks tasks) {
DefaultBootstrapSession session = (DefaultBootstrapSession) bssession;
// set models
if (tasks.supportedObjects != null)
session.setModel(modelProvider.getObjectModel(session, tasks.supportedObjects));
// set Requests to Send
session.setRequests(tasks.requestsToSend);
// prepare list where we will store Responses
session.setResponses(new ArrayList<LwM2mResponse>(tasks.requestsToSend.size()));
// is last Tasks ?
session.setMoreTasks(!tasks.last);
}
@Override
public BootstrapDownlinkRequest<? extends LwM2mResponse> getFirstRequest(BootstrapSession bsSession) {
return nextRequest(bsSession);
}
protected BootstrapDownlinkRequest<? extends LwM2mResponse> nextRequest(BootstrapSession bsSession) {
DefaultBootstrapSession session = (DefaultBootstrapSession) bsSession;
List<BootstrapDownlinkRequest<? extends LwM2mResponse>> requestsToSend = session.getRequests();
if (!requestsToSend.isEmpty()) {
// get next requests
return requestsToSend.remove(0);
} else {
if (session.hasMoreTasks()) {
Tasks nextTasks = tasksProvider.getTasks(session, session.getResponses());
if (nextTasks == null) {
session.setMoreTasks(false);
return new BootstrapFinishRequest();
}
initTasks(session, nextTasks);
return nextRequest(bsSession);
} else {
return new BootstrapFinishRequest();
}
}
}
@Override
public BootstrapPolicy onResponseSuccess(BootstrapSession bsSession,
BootstrapDownlinkRequest<? extends LwM2mResponse> request, LwM2mResponse response) {
if (LOG.isTraceEnabled())
LOG.trace("{} {} receives success response for {} : {}", request.getClass().getSimpleName(),
request.getPath(), bsSession, request);
if (!(request instanceof BootstrapFinishRequest)) {
// store response
DefaultBootstrapSession session = (DefaultBootstrapSession) bsSession;
session.getResponses().add(response);
// on success for NOT bootstrap finish request we send next request
return BootstrapPolicy.continueWith(nextRequest(bsSession));
} else {
// on success for bootstrap finish request we stop the session
return BootstrapPolicy.finished();
}
}
@Override
public BootstrapPolicy onResponseError(BootstrapSession bsSession,
BootstrapDownlinkRequest<? extends LwM2mResponse> request, LwM2mResponse response) {
if (LOG.isTraceEnabled())
LOG.trace("{} {} receives error response {} for {} : {}", request.getClass().getSimpleName(),
request.getPath(), response, bsSession, request);
if (!(request instanceof BootstrapFinishRequest)) {
// store response
DefaultBootstrapSession session = (DefaultBootstrapSession) bsSession;
session.getResponses().add(response);
// on response error for NOT bootstrap finish request we continue any sending next request
return BootstrapPolicy.continueWith(nextRequest(bsSession));
} else {
// on response error for bootstrap finish request we stop the session
return BootstrapPolicy.failed();
}
}
@Override
public BootstrapPolicy onRequestFailure(BootstrapSession bsSession,
BootstrapDownlinkRequest<? extends LwM2mResponse> request, Throwable cause) {
if (LOG.isTraceEnabled())
LOG.trace("{} {} failed because of {} for {} : {}", request.getClass().getSimpleName(), request.getPath(),
cause, bsSession, request);
return BootstrapPolicy.failed();
}
@Override
public void end(BootstrapSession bsSession) {
LOG.trace("Bootstrap session finished : {}", bsSession);
}
@Override
public void failed(BootstrapSession bsSession, BootstrapFailureCause cause) {
LOG.trace("Bootstrap session failed by {}: {}", cause, bsSession);
}
}