de.fraunhofer.iosb.ilt.frostserver.mqtt.moquette.MoquetteMqttServer Maven / Gradle / Ivy
/*
* Copyright (C) 2024 Fraunhofer Institut IOSB, Fraunhoferstr. 1, D 76131
* Karlsruhe, Germany.
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU Lesser General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public License
* along with this program. If not, see > clientSubscriptions = new HashMap<>();
private AuthWrapper authWrapper;
/**
* The MQTT Id used by the FROST server to connect to the MQTT broker.
*/
private final String frostClientId;
public MoquetteMqttServer() {
frostClientId = "SensorThings API Server (" + UUID.randomUUID() + ")";
}
@Override
public void addEntityCreateListener(EntityCreateListener listener) {
entityCreateListeners.add(EntityCreateListener.class, listener);
}
@Override
public void publish(String topic, String message, int qos) {
if (mqttBroker != null) {
final ByteBuf payload = ByteBufUtil.writeUtf8(UnpooledByteBufAllocator.DEFAULT, message);
MqttFixedHeader fixedHeader = new MqttFixedHeader(MqttMessageType.PUBLISH, false, MqttQoS.valueOf(qos), false, 0);
MqttPublishVariableHeader varHeader = new MqttPublishVariableHeader(topic, 0);
MqttPublishMessage mqttPublishMessage = new MqttPublishMessage(fixedHeader, varHeader, payload);
mqttBroker.internalPublish(mqttPublishMessage, frostClientId);
}
}
@Override
public void addSubscriptionListener(SubscriptionListener listener) {
subscriptionListeners.add(SubscriptionListener.class, listener);
}
@Override
public void removeEntityCreateListener(EntityCreateListener listener) {
entityCreateListeners.remove(EntityCreateListener.class, listener);
}
@Override
public void removeSubscriptionListener(SubscriptionListener listener) {
subscriptionListeners.remove(SubscriptionListener.class, listener);
}
protected void fireEntityCreate(EntityCreateEvent e) {
Object[] listeners = entityCreateListeners.getListenerList();
for (int i = 0; i < listeners.length; i = i + 2) {
if (listeners[i] == EntityCreateListener.class) {
((EntityCreateListener) listeners[i + 1]).onEntityCreate(e);
}
}
}
protected void fireSubscribe(SubscriptionEvent e) {
Object[] listeners = subscriptionListeners.getListenerList();
for (int i = 0; i < listeners.length; i = i + 2) {
if (listeners[i] == SubscriptionListener.class) {
((SubscriptionListener) listeners[i + 1]).onSubscribe(e);
}
}
}
protected void fireUnsubscribe(SubscriptionEvent e) {
Object[] listeners = subscriptionListeners.getListenerList();
for (int i = 0; i < listeners.length; i = i + 2) {
if (listeners[i] == SubscriptionListener.class) {
((SubscriptionListener) listeners[i + 1]).onUnsubscribe(e);
}
}
}
@Override
public void start() {
mqttBroker = new Server();
final List userHandlers = Arrays.asList(new AbstractInterceptHandlerImpl());
MqttSettings mqttSettings = settings.getMqttSettings();
Settings customSettings = mqttSettings.getCustomSettings();
IConfig config = new ConfigWrapper(customSettings);
// Ensure the immediate_flush property has a default of true.
config.intProp(IConfig.BUFFER_FLUSH_MS_PROPERTY_NAME, 0);
config.boolProp(IConfig.ENABLE_TELEMETRY_NAME, false);
config.setProperty(IConfig.PORT_PROPERTY_NAME, Integer.toString(mqttSettings.getPort()));
config.setProperty(IConfig.HOST_PROPERTY_NAME, mqttSettings.getHost());
config.setProperty(IConfig.ALLOW_ANONYMOUS_PROPERTY_NAME, Boolean.TRUE.toString());
String persistentStoreType = customSettings.get(TAG_PERSISTENT_STORE_TYPE, getClass());
if (VALUE_STORE_TYPE_H2.equalsIgnoreCase(persistentStoreType)) {
String tempPath = Paths.get(settings.getTempPath()).toString();
String persistentStore = customSettings.get(IConfig.DATA_PATH_PROPERTY_NAME, tempPath);
config.setProperty(IConfig.DATA_PATH_PROPERTY_NAME, persistentStore);
}
config.setProperty(IConfig.WEB_SOCKET_PORT_PROPERTY_NAME, customSettings.get(TAG_WEBSOCKET_PORT, getClass()));
String keystorePath = customSettings.get(TAG_KEYSTORE_PATH, getClass());
if (!keystorePath.isEmpty()) {
LOGGER.info("Configuring keystore for ssl");
config.setProperty(IConfig.JKS_PATH_PROPERTY_NAME, keystorePath);
config.setProperty(IConfig.KEY_STORE_PASSWORD_PROPERTY_NAME, customSettings.get(TAG_KEYSTORE_PASS, getClass()));
config.setProperty(IConfig.KEY_MANAGER_PASSWORD_PROPERTY_NAME, customSettings.get(TAG_KEYMANAGER_PASS, getClass()));
config.setProperty(IConfig.SSL_PORT_PROPERTY_NAME, customSettings.get(TAG_SSL_PORT, getClass()));
config.setProperty(IConfig.WSS_PORT_PROPERTY_NAME, customSettings.get(TAG_SSL_WEBSOCKET_PORT, getClass()));
}
authWrapper = createAuthWrapper();
try {
mqttBroker.startServer(config, userHandlers, null, authWrapper, authWrapper);
} catch (IOException ex) {
LOGGER.error("Failed to start MQTT Broker!", ex);
}
}
private AuthWrapper createAuthWrapper() {
Settings authSettings = settings.getAuthSettings();
String authProviderClassName = authSettings.get(CoreSettings.TAG_AUTH_PROVIDER, "");
if (!StringHelper.isNullOrEmpty(authProviderClassName)) {
return new AuthWrapper(settings, authProviderClassName, frostClientId);
}
return null;
}
@Override
public void stop() {
if (mqttBroker != null) {
mqttBroker.stopServer();
}
}
@Override
public void init(CoreSettings settings) {
if (settings == null) {
throw new IllegalArgumentException("MqttSettings must be non-null");
}
this.settings = settings;
}
private class AbstractInterceptHandlerImpl extends AbstractInterceptHandler {
@Override
public void onPublish(InterceptPublishMessage msg) {
try {
if (frostClientId.equals(msg.getClientID())) {
return;
}
LOGGER.trace(" Moquette -> FROST on {}", msg.getTopicName());
String payload = msg.getPayload().toString(StringHelper.UTF8);
PrincipalExtended userPrincipal;
if (authWrapper == null) {
userPrincipal = PrincipalExtended.ANONYMOUS_PRINCIPAL;
} else {
userPrincipal = authWrapper.getUserPrincipal(msg.getClientID());
}
fireEntityCreate(new EntityCreateEvent(msg.getTopicName(), payload, userPrincipal));
} finally {
super.onPublish(msg);
}
}
@Override
public void onConnect(InterceptConnectMessage msg) {
final String clientId = msg.getClientID();
if (clientId.equalsIgnoreCase(frostClientId)) {
return;
}
LOGGER.trace(" Client connected: {}", clientId);
clientSubscriptions.put(clientId, new ArrayList<>());
}
@Override
public void onDisconnect(InterceptDisconnectMessage msg) {
final String clientId = msg.getClientID();
if (clientId.equalsIgnoreCase(frostClientId)) {
return;
}
LOGGER.trace(" Client disconnected: {}", clientId);
clientSubscriptions
.getOrDefault(clientId, new ArrayList<>())
.stream()
.forEach(subscribedTopic -> fireUnsubscribe(new SubscriptionEvent(subscribedTopic)));
clientSubscriptions.remove(clientId);
}
@Override
public void onSubscribe(InterceptSubscribeMessage msg) {
final String clientId = msg.getClientID();
if (clientId.equalsIgnoreCase(frostClientId)) {
return;
}
final String topicFilter = msg.getTopicFilter();
LOGGER.trace(" Client {} subscribed to {}", clientId, topicFilter);
clientSubscriptions
.computeIfAbsent(clientId, t -> new ArrayList<>())
.add(topicFilter);
fireSubscribe(new SubscriptionEvent(topicFilter));
}
@Override
public void onUnsubscribe(InterceptUnsubscribeMessage msg) {
final String clientId = msg.getClientID();
if (frostClientId.equals(clientId)) {
return;
}
final String topicFilter = msg.getTopicFilter();
LOGGER.trace(" Client {} unsubscribed from {}", clientId, topicFilter);
boolean removed = clientSubscriptions.getOrDefault(clientId, new ArrayList<>())
.remove(topicFilter);
if (removed) {
fireUnsubscribe(new SubscriptionEvent(topicFilter));
}
}
@Override
public String getID() {
return frostClientId;
}
@Override
public void onSessionLoopError(Throwable thrwbl) {
LOGGER.error("MQTT Session Loop caused an exception!", thrwbl);
}
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy