|
|
@ -17,37 +17,28 @@
|
|
|
|
*/
|
|
|
|
*/
|
|
|
|
package io.entgra.device.mgt.plugins.output.adapter.mqtt.util;
|
|
|
|
package io.entgra.device.mgt.plugins.output.adapter.mqtt.util;
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
import io.entgra.device.mgt.core.apimgt.keymgt.extension.DCRResponse;
|
|
|
|
|
|
|
|
import io.entgra.device.mgt.core.apimgt.keymgt.extension.TokenRequest;
|
|
|
|
|
|
|
|
import io.entgra.device.mgt.core.apimgt.keymgt.extension.TokenResponse;
|
|
|
|
|
|
|
|
import io.entgra.device.mgt.core.apimgt.keymgt.extension.exception.BadRequestException;
|
|
|
|
|
|
|
|
import io.entgra.device.mgt.core.apimgt.keymgt.extension.exception.KeyMgtException;
|
|
|
|
|
|
|
|
import io.entgra.device.mgt.core.apimgt.keymgt.extension.service.KeyMgtService;
|
|
|
|
|
|
|
|
import io.entgra.device.mgt.core.apimgt.keymgt.extension.service.KeyMgtServiceImpl;
|
|
|
|
|
|
|
|
import io.entgra.device.mgt.core.identity.jwt.client.extension.exception.JWTClientException;
|
|
|
|
|
|
|
|
import org.apache.commons.lang3.StringUtils;
|
|
|
|
import org.apache.commons.logging.Log;
|
|
|
|
import org.apache.commons.logging.Log;
|
|
|
|
import org.apache.commons.logging.LogFactory;
|
|
|
|
import org.apache.commons.logging.LogFactory;
|
|
|
|
import org.apache.commons.ssl.Base64;
|
|
|
|
import org.apache.commons.ssl.Base64;
|
|
|
|
import org.apache.http.HttpResponse;
|
|
|
|
|
|
|
|
import org.apache.http.client.HttpClient;
|
|
|
|
|
|
|
|
import org.apache.http.client.methods.HttpPost;
|
|
|
|
|
|
|
|
import org.apache.http.entity.ContentType;
|
|
|
|
|
|
|
|
import org.apache.http.entity.StringEntity;
|
|
|
|
|
|
|
|
import org.apache.http.message.BasicHeader;
|
|
|
|
|
|
|
|
import org.eclipse.paho.client.mqttv3.MqttClient;
|
|
|
|
import org.eclipse.paho.client.mqttv3.MqttClient;
|
|
|
|
import org.eclipse.paho.client.mqttv3.MqttConnectOptions;
|
|
|
|
import org.eclipse.paho.client.mqttv3.MqttConnectOptions;
|
|
|
|
import org.eclipse.paho.client.mqttv3.MqttException;
|
|
|
|
import org.eclipse.paho.client.mqttv3.MqttException;
|
|
|
|
import org.eclipse.paho.client.mqttv3.MqttMessage;
|
|
|
|
import org.eclipse.paho.client.mqttv3.MqttMessage;
|
|
|
|
import org.eclipse.paho.client.mqttv3.persist.MqttDefaultFilePersistence;
|
|
|
|
import org.eclipse.paho.client.mqttv3.persist.MqttDefaultFilePersistence;
|
|
|
|
import org.json.simple.JSONObject;
|
|
|
|
import org.jetbrains.annotations.NotNull;
|
|
|
|
import org.json.simple.parser.JSONParser;
|
|
|
|
|
|
|
|
import org.json.simple.parser.ParseException;
|
|
|
|
|
|
|
|
import io.entgra.device.mgt.core.apimgt.keymgt.extension.DCRResponse;
|
|
|
|
|
|
|
|
import io.entgra.device.mgt.core.apimgt.keymgt.extension.TokenRequest;
|
|
|
|
|
|
|
|
import io.entgra.device.mgt.core.apimgt.keymgt.extension.TokenResponse;
|
|
|
|
|
|
|
|
import io.entgra.device.mgt.core.apimgt.keymgt.extension.exception.BadRequestException;
|
|
|
|
|
|
|
|
import io.entgra.device.mgt.core.apimgt.keymgt.extension.exception.KeyMgtException;
|
|
|
|
|
|
|
|
import io.entgra.device.mgt.core.apimgt.keymgt.extension.service.KeyMgtService;
|
|
|
|
|
|
|
|
import io.entgra.device.mgt.core.apimgt.keymgt.extension.service.KeyMgtServiceImpl;
|
|
|
|
|
|
|
|
import org.wso2.carbon.context.PrivilegedCarbonContext;
|
|
|
|
import org.wso2.carbon.context.PrivilegedCarbonContext;
|
|
|
|
import org.wso2.carbon.event.output.adapter.core.exception.ConnectionUnavailableException;
|
|
|
|
import org.wso2.carbon.event.output.adapter.core.exception.ConnectionUnavailableException;
|
|
|
|
import org.wso2.carbon.event.output.adapter.core.exception.OutputEventAdapterException;
|
|
|
|
import org.wso2.carbon.event.output.adapter.core.exception.OutputEventAdapterException;
|
|
|
|
import org.wso2.carbon.event.output.adapter.core.exception.OutputEventAdapterRuntimeException;
|
|
|
|
import org.wso2.carbon.event.output.adapter.core.exception.OutputEventAdapterRuntimeException;
|
|
|
|
import io.entgra.device.mgt.core.identity.jwt.client.extension.dto.AccessTokenInfo;
|
|
|
|
|
|
|
|
import io.entgra.device.mgt.core.identity.jwt.client.extension.exception.JWTClientException;
|
|
|
|
|
|
|
|
import io.entgra.device.mgt.core.identity.jwt.client.extension.service.JWTClientManagerService;
|
|
|
|
|
|
|
|
import org.wso2.carbon.user.api.UserStoreException;
|
|
|
|
import org.wso2.carbon.user.api.UserStoreException;
|
|
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
/**
|
|
|
@ -153,6 +144,8 @@ public class MQTTAdapterPublisher {
|
|
|
|
String dcrUrlString = this.mqttBrokerConnectionConfiguration.getDcrUrl();
|
|
|
|
String dcrUrlString = this.mqttBrokerConnectionConfiguration.getDcrUrl();
|
|
|
|
|
|
|
|
|
|
|
|
if (dcrUrlString != null && !dcrUrlString.isEmpty()) {
|
|
|
|
if (dcrUrlString != null && !dcrUrlString.isEmpty()) {
|
|
|
|
|
|
|
|
PrivilegedCarbonContext.startTenantFlow();
|
|
|
|
|
|
|
|
PrivilegedCarbonContext.getThreadLocalCarbonContext().setTenantDomain(tenantDomain, true);
|
|
|
|
try {
|
|
|
|
try {
|
|
|
|
KeyMgtService keyMgtService = new KeyMgtServiceImpl();
|
|
|
|
KeyMgtService keyMgtService = new KeyMgtServiceImpl();
|
|
|
|
String applicationName = MQTTEventAdapterConstants.APPLICATION_NAME_PREFIX
|
|
|
|
String applicationName = MQTTEventAdapterConstants.APPLICATION_NAME_PREFIX
|
|
|
@ -171,6 +164,8 @@ public class MQTTAdapterPublisher {
|
|
|
|
} catch (KeyMgtException e) {
|
|
|
|
} catch (KeyMgtException e) {
|
|
|
|
log.error("Failed to create an application.", e);
|
|
|
|
log.error("Failed to create an application.", e);
|
|
|
|
throw new OutputEventAdapterRuntimeException("Failed to create an application.", e);
|
|
|
|
throw new OutputEventAdapterRuntimeException("Failed to create an application.", e);
|
|
|
|
|
|
|
|
} finally {
|
|
|
|
|
|
|
|
PrivilegedCarbonContext.endTenantFlow();
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
throw new OutputEventAdapterRuntimeException("Invalid configuration for mqtt publisher");
|
|
|
|
throw new OutputEventAdapterRuntimeException("Invalid configuration for mqtt publisher");
|
|
|
@ -178,27 +173,33 @@ public class MQTTAdapterPublisher {
|
|
|
|
|
|
|
|
|
|
|
|
private String getToken(String clientId, String clientSecret)
|
|
|
|
private String getToken(String clientId, String clientSecret)
|
|
|
|
throws UserStoreException, JWTClientException {
|
|
|
|
throws UserStoreException, JWTClientException {
|
|
|
|
PrivilegedCarbonContext.startTenantFlow();
|
|
|
|
|
|
|
|
PrivilegedCarbonContext.getThreadLocalCarbonContext().setTenantDomain(tenantDomain, true);
|
|
|
|
|
|
|
|
try {
|
|
|
|
try {
|
|
|
|
String scopes = mqttBrokerConnectionConfiguration.getScopes();
|
|
|
|
TokenRequest tokenRequest = getTokenRequest(clientId, clientSecret);
|
|
|
|
scopes += " perm:topic:pub:" + tenantDomain + ":+:+:operation";
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
TokenRequest tokenRequest = new TokenRequest(clientId, clientSecret,
|
|
|
|
|
|
|
|
null, scopes.toString(), "client_credentials", null,
|
|
|
|
|
|
|
|
null, null, null, Integer.MAX_VALUE);
|
|
|
|
|
|
|
|
KeyMgtService keyMgtService = new KeyMgtServiceImpl();
|
|
|
|
KeyMgtService keyMgtService = new KeyMgtServiceImpl();
|
|
|
|
TokenResponse tokenResponse = keyMgtService.generateAccessToken(tokenRequest);
|
|
|
|
TokenResponse tokenResponse = keyMgtService.generateAccessToken(tokenRequest);
|
|
|
|
|
|
|
|
|
|
|
|
return tokenResponse.getAccessToken();
|
|
|
|
return tokenResponse.getAccessToken();
|
|
|
|
} catch (KeyMgtException | BadRequestException e) {
|
|
|
|
} catch (KeyMgtException | BadRequestException e) {
|
|
|
|
log.error("Error while generating access token", e);
|
|
|
|
log.error("Error while generating access token", e);
|
|
|
|
} finally {
|
|
|
|
|
|
|
|
PrivilegedCarbonContext.endTenantFlow();
|
|
|
|
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return null;
|
|
|
|
return null;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@NotNull
|
|
|
|
|
|
|
|
private TokenRequest getTokenRequest(String clientId, String clientSecret) {
|
|
|
|
|
|
|
|
String scopes = mqttBrokerConnectionConfiguration.getScopes();
|
|
|
|
|
|
|
|
scopes += " perm:topic:pub:" + tenantDomain + ":+:+:operation";
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if (!StringUtils.isEmpty(mqttBrokerConnectionConfiguration.getTopic())) {
|
|
|
|
|
|
|
|
scopes += " perm:topic:pub:" + mqttBrokerConnectionConfiguration.getTopic().replace("/",":");
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
TokenRequest tokenRequest = new TokenRequest(clientId, clientSecret,
|
|
|
|
|
|
|
|
null, scopes.toString(), "client_credentials", null,
|
|
|
|
|
|
|
|
null, null, null, Integer.MAX_VALUE);
|
|
|
|
|
|
|
|
return tokenRequest;
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
private String getBase64Encode(String key, String value) {
|
|
|
|
private String getBase64Encode(String key, String value) {
|
|
|
|
return new String(Base64.encodeBase64((key + ":" + value).getBytes()));
|
|
|
|
return new String(Base64.encodeBase64((key + ":" + value).getBytes()));
|
|
|
|
}
|
|
|
|
}
|
|
|
|