nickAS21
5 years ago
6 changed files with 313 additions and 111 deletions
@ -0,0 +1,242 @@ |
|||
/** |
|||
* Copyright © 2016-2021 The Thingsboard Authors |
|||
* |
|||
* Licensed under the Apache License, Version 2.0 (the "License"); |
|||
* you may not use this file except in compliance with the License. |
|||
* You may obtain a copy of the License at |
|||
* |
|||
* http://www.apache.org/licenses/LICENSE-2.0
|
|||
* |
|||
* Unless required by applicable law or agreed to in writing, software |
|||
* distributed under the License is distributed on an "AS IS" BASIS, |
|||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
|||
* See the License for the specific language governing permissions and |
|||
* limitations under the License. |
|||
*/ |
|||
package org.thingsboard.server.transport.lwm2m.bootstrap.store; |
|||
|
|||
import lombok.extern.slf4j.Slf4j; |
|||
import org.eclipse.leshan.core.Link; |
|||
import org.eclipse.leshan.core.node.LwM2mObject; |
|||
import org.eclipse.leshan.core.node.LwM2mPath; |
|||
import org.eclipse.leshan.core.request.*; |
|||
import org.eclipse.leshan.core.response.BootstrapDiscoverResponse; |
|||
import org.eclipse.leshan.core.response.BootstrapReadResponse; |
|||
import org.eclipse.leshan.core.response.LwM2mResponse; |
|||
import org.eclipse.leshan.server.bootstrap.BootstrapConfig; |
|||
import org.eclipse.leshan.server.bootstrap.BootstrapConfigStore; |
|||
import org.eclipse.leshan.server.bootstrap.BootstrapSession; |
|||
import org.eclipse.leshan.server.bootstrap.BootstrapTaskProvider; |
|||
import org.eclipse.leshan.server.bootstrap.BootstrapUtil; |
|||
|
|||
import java.net.InetSocketAddress; |
|||
import java.util.*; |
|||
|
|||
import static org.eclipse.leshan.server.bootstrap.BootstrapUtil.toWriteRequest; |
|||
|
|||
@Slf4j |
|||
public class LwM2MBootstrapConfigStoreTaskProvider implements BootstrapTaskProvider { |
|||
|
|||
private BootstrapConfigStore store; |
|||
|
|||
private Map<Integer, String> supportedObjects; |
|||
|
|||
/** |
|||
* Map<serverId, InstanceId> |
|||
*/ |
|||
protected Map<Integer, Integer> securityInstances; |
|||
protected Map<Integer, Integer> serverInstances; |
|||
|
|||
public LwM2MBootstrapConfigStoreTaskProvider(BootstrapConfigStore store) { |
|||
this.store = store; |
|||
} |
|||
|
|||
@Override |
|||
public Tasks getTasks(BootstrapSession session, List<LwM2mResponse> previousResponse) { |
|||
|
|||
BootstrapConfig config = store.get(session.getEndpoint(), session.getIdentity(), session); |
|||
if (config == null) { |
|||
return null; |
|||
} |
|||
if (previousResponse == null && shouldStartWithDiscover(config)) { |
|||
Tasks tasks = new Tasks(); |
|||
tasks.requestsToSend = new ArrayList<>(1); |
|||
tasks.requestsToSend.add(new BootstrapDiscoverRequest()); |
|||
tasks.last = false; |
|||
return tasks; |
|||
} else { |
|||
Tasks tasks = new Tasks(); |
|||
if (this.supportedObjects == null) { |
|||
initSupportedObjectsDefault(); |
|||
|
|||
} |
|||
// add supportedObjects
|
|||
tasks.supportedObjects = this.supportedObjects; |
|||
// handle bootstrap discover response
|
|||
if (previousResponse != null) { |
|||
if (previousResponse.get(0) instanceof BootstrapDiscoverResponse) { |
|||
BootstrapDiscoverResponse discoverResponse = (BootstrapDiscoverResponse) previousResponse.get(0); |
|||
if (discoverResponse.isSuccess()) { |
|||
this.initAfterBootstrapDiscover(discoverResponse); |
|||
findSecurityInstanceId(discoverResponse.getObjectLinks()); |
|||
} else { |
|||
log.info( |
|||
"Bootstrap Discover return error {} : to continue bootstrap session without autoIdForSecurityObject mode. {}", |
|||
discoverResponse, session); |
|||
} |
|||
if (this.securityInstances.get(0) == null) { |
|||
log.error( |
|||
"Unable to find bootstrap server instance in Security Object (0) in response {}: unable to continue bootstrap session with autoIdForSecurityObject mode. {}", |
|||
discoverResponse, session); |
|||
return null; |
|||
} |
|||
tasks.requestsToSend = new ArrayList<>(1); |
|||
tasks.requestsToSend.add(new BootstrapReadRequest("/1")); |
|||
tasks.last = false; |
|||
return tasks; |
|||
} |
|||
BootstrapReadResponse readResponse = (BootstrapReadResponse) previousResponse.get(0); |
|||
findServerInstanceId(readResponse); |
|||
// create requests from config
|
|||
tasks.requestsToSend = this.toRequests(config, |
|||
config.contentFormat != null ? config.contentFormat : session.getContentFormat(), |
|||
this.securityInstances.get(0)); |
|||
} else { |
|||
// create requests from config
|
|||
tasks.requestsToSend = BootstrapUtil.toRequests(config, |
|||
config.contentFormat != null ? config.contentFormat : session.getContentFormat()); |
|||
|
|||
} |
|||
return tasks; |
|||
} |
|||
} |
|||
|
|||
protected boolean shouldStartWithDiscover(BootstrapConfig config) { |
|||
return config.autoIdForSecurityObject; |
|||
} |
|||
|
|||
protected void findSecurityInstanceId(Link[] objectLinks) { |
|||
this.securityInstances = new HashMap<>(); |
|||
for (Link link : objectLinks) { |
|||
if (link.getUrl().startsWith("/0/")) { |
|||
try { |
|||
LwM2mPath path = new LwM2mPath(link.getUrl()); |
|||
if (path.isObjectInstance()) { |
|||
if (link.getAttributes().containsKey("ssid")) { |
|||
int serverId = Integer.valueOf(link.getAttributes().get("ssid")); |
|||
if (!this.securityInstances.containsKey(serverId)) { |
|||
this.securityInstances.put(serverId, path.getObjectInstanceId()); |
|||
} else { |
|||
log.error(String.format("Invalid lwm2mSecurityInstance by [{}]", path.getObjectInstanceId())); |
|||
} |
|||
this.securityInstances.put(Integer.valueOf(link.getAttributes().get("ssid")), path.getObjectInstanceId()); |
|||
} else { |
|||
if (!this.securityInstances.containsKey(0)) { |
|||
this.securityInstances.put(0, path.getObjectInstanceId()); |
|||
} else { |
|||
log.error(String.format("Invalid bootstrapSecurityInstance by [{}]", path.getObjectInstanceId())); |
|||
} |
|||
} |
|||
} |
|||
} catch (Exception e) { |
|||
// ignore if this is not a LWM2M path
|
|||
log.error("Invalid LwM2MPath starting by \"/0/\""); |
|||
} |
|||
} |
|||
} |
|||
} |
|||
|
|||
protected void findServerInstanceId(BootstrapReadResponse readResponse) { |
|||
serverInstances = new HashMap<>(); |
|||
((LwM2mObject) readResponse.getContent()).getInstances().values().forEach(instance -> { |
|||
serverInstances.put(((Long) instance.getResource(0).getValue()).intValue(), instance.getId()); |
|||
}); |
|||
} |
|||
|
|||
public BootstrapConfigStore getStore() { |
|||
return this.store; |
|||
} |
|||
|
|||
private void initAfterBootstrapDiscover(BootstrapDiscoverResponse response) { |
|||
Link[] links = response.getObjectLinks(); |
|||
Arrays.stream(links).forEach(link -> { |
|||
LwM2mPath path = new LwM2mPath(link.getUrl()); |
|||
if (path != null && !path.isRoot() && path.getObjectId() < 3) { |
|||
if (path.isObject()) { |
|||
String ver = link.getAttributes().get("ver") != null ? link.getAttributes().get("ver") : "1.0"; |
|||
this.supportedObjects.put(path.getObjectId(), ver); |
|||
} |
|||
} |
|||
}); |
|||
} |
|||
|
|||
|
|||
public List<BootstrapDownlinkRequest<? extends LwM2mResponse>> toRequests(BootstrapConfig bootstrapConfig, |
|||
ContentFormat contentFormat, int bootstrapSecurityInstanceId) { |
|||
List<BootstrapDownlinkRequest<? extends LwM2mResponse>> requests = new ArrayList<>(); |
|||
boolean isBsServer = false; |
|||
boolean isLwServer = false; |
|||
/** Map<serverId, InstanceId> */ |
|||
Map<Integer, Integer> instances = new HashMap<>(); |
|||
// handle security
|
|||
int id = 0; |
|||
for (BootstrapConfig.ServerSecurity security : new TreeMap<>(bootstrapConfig.security).values()) { |
|||
if (security.bootstrapServer) { |
|||
requests.add(toWriteRequest(bootstrapSecurityInstanceId, security, contentFormat)); |
|||
isBsServer = true; |
|||
instances.put(security.serverId, bootstrapSecurityInstanceId); |
|||
} else { |
|||
if (id == bootstrapSecurityInstanceId) { |
|||
id++; |
|||
} |
|||
requests.add(toWriteRequest(id, security, contentFormat)); |
|||
instances.put(security.serverId, id); |
|||
id++; |
|||
} |
|||
} |
|||
// handle server
|
|||
for (Map.Entry<Integer, BootstrapConfig.ServerConfig> server : bootstrapConfig.servers.entrySet()) { |
|||
int securityInstanceId = instances.get(server.getValue().shortId); |
|||
requests.add(toWriteRequest(securityInstanceId, server.getValue(), contentFormat)); |
|||
if (!isBsServer && this.serverInstances.containsKey(server.getValue().shortId) && securityInstanceId != this.serverInstances.get(server.getValue().shortId)){ |
|||
requests.add(new BootstrapDeleteRequest("/0/" + securityInstanceId)); |
|||
requests.add(new BootstrapDeleteRequest("/1/" + this.serverInstances.get(server.getValue().shortId))); |
|||
} |
|||
isLwServer = true; |
|||
} |
|||
// handle acl
|
|||
for (Map.Entry<Integer, BootstrapConfig.ACLConfig> acl : bootstrapConfig.acls.entrySet()) { |
|||
requests.add(toWriteRequest(acl.getKey(), acl.getValue(), contentFormat)); |
|||
} |
|||
// handle delete
|
|||
if (isBsServer & isLwServer) { |
|||
requests.add(new BootstrapDeleteRequest("/0")); |
|||
requests.add(new BootstrapDeleteRequest("/1")); |
|||
} |
|||
|
|||
return (requests); |
|||
} |
|||
|
|||
private InetSocketAddress getSocketAddress(String uri) { |
|||
// String uri1 = "coap://localhost:5687";
|
|||
if (uri.contains("coap://")) { |
|||
uri = uri.replace("coap://", ""); |
|||
} else return null; |
|||
String[] uris = uri.split(":"); |
|||
if (uris.length == 2) { |
|||
try { |
|||
return new InetSocketAddress(uris[0], Integer.parseInt(uris[1])); |
|||
} catch (Exception e) { |
|||
return null; |
|||
} |
|||
|
|||
} else return null; |
|||
} |
|||
|
|||
private void initSupportedObjectsDefault() { |
|||
this.supportedObjects = new HashMap<>(); |
|||
this.supportedObjects.put(0, "1.1"); |
|||
this.supportedObjects.put(1, "1.1"); |
|||
this.supportedObjects.put(2, "1.0"); |
|||
} |
|||
} |
|||
Loading…
Reference in new issue