org.apache.activemq.transport.xmpp.ProtocolConverter Maven / Gradle / Ivy
/**
*
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You 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.apache.activemq.transport.xmpp;
import ietf.params.xml.ns.xmpp_sasl.Auth;
import ietf.params.xml.ns.xmpp_sasl.Challenge;
import ietf.params.xml.ns.xmpp_sasl.Success;
import ietf.params.xml.ns.xmpp_tls.Proceed;
import ietf.params.xml.ns.xmpp_tls.Starttls;
import jabber.client.Body;
import jabber.client.Error;
import jabber.client.Iq;
import jabber.client.Message;
import jabber.client.Presence;
import jabber.iq.auth.Query;
import org.apache.activemq.command.*;
import org.apache.activemq.transport.xmpp.command.Handler;
import org.apache.activemq.transport.xmpp.command.HandlerRegistry;
import org.apache.activemq.util.IdGenerator;
import org.apache.activemq.util.LongSequenceGenerator;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.jabber.protocol.disco_info.Feature;
import org.jabber.protocol.disco_info.Identity;
import org.jabber.protocol.disco_items.Item;
import org.jabber.protocol.muc_user.X;
import org.w3c.dom.Element;
import javax.jms.JMSException;
import java.io.IOException;
import java.io.PrintWriter;
import java.io.StringWriter;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.atomic.AtomicBoolean;
/**
* TODO lots of this code could be shared with Stomp
*/
public class ProtocolConverter {
private static final transient Log log = LogFactory.getLog(ProtocolConverter.class);
private HandlerRegistry registry = new HandlerRegistry();
private XmppTransport transport;
private static final IdGenerator connectionIdGenerator = new IdGenerator();
private static final IdGenerator clientIdGenerator = new IdGenerator("xmpp");
private final ConnectionId connectionId = new ConnectionId(connectionIdGenerator.generateId());
private final SessionId sessionId = new SessionId(connectionId, -1);
private final ProducerId producerId = new ProducerId(sessionId, 1);
private final ConnectionInfo connectionInfo = new ConnectionInfo(connectionId);
private final SessionInfo sessionInfo = new SessionInfo(sessionId);
private final ProducerInfo producerInfo = new ProducerInfo(producerId);
private final LongSequenceGenerator consumerIdGenerator = new LongSequenceGenerator();
private final LongSequenceGenerator messageIdGenerator = new LongSequenceGenerator();
private final LongSequenceGenerator transactionIdGenerator = new LongSequenceGenerator();
private final Map> resposeHandlers = new ConcurrentHashMap>();
private final Map> subscriptionsByConsumerId = new ConcurrentHashMap>();
private final Map jidToConsumerMap = new HashMap();
private final Map transactions = new ConcurrentHashMap();
private final Object commnadIdMutex = new Object();
private int lastCommandId;
private final AtomicBoolean connected = new AtomicBoolean(false);
public ProtocolConverter(XmppTransport transport) {
this.transport = transport;
initialiseRegistry();
}
protected int generateCommandId() {
synchronized (commnadIdMutex) {
return lastCommandId++;
}
}
protected void initialiseRegistry() {
// this kinda wiring muck is soooo much cleaner in C# :(
registry.registerHandler(Message.class, new Handler() {
public void handle(Message event) throws Exception {
onMessage(event);
}
});
registry.registerHandler(Auth.class, new Handler() {
public void handle(Auth event) throws Exception {
onAuth(event);
}
});
registry.registerHandler(Starttls.class, new Handler() {
public void handle(Starttls event) throws Exception {
onStarttls(event);
}
});
registry.registerHandler(Iq.class, new Handler() {
public void handle(Iq event) throws Exception {
onIq(event);
}
});
registry.registerHandler(Presence.class, new Handler() {
public void handle(Presence event) throws Exception {
onPresence(event);
}
});
}
public void onXmppCommand(Object command) throws Exception {
// TODO we could do some nice code generation to boost performance
// by autogenerating the bytecode to statically lookup a handler from a registry maybe?
Handler handler = registry.getHandler(command.getClass());
if (handler == null) {
unknownCommand(command);
}
else {
handler.handle(command);
}
}
public void onActiveMQCommad(Command command) throws Exception {
if (command.isResponse()) {
Response response = (Response) command;
Handler handler = resposeHandlers.remove(new Integer(response.getCorrelationId()));
if (handler != null) {
handler.handle(response);
}
else {
log.warn("No handler for response: " + response);
}
}
else if (command.isMessageDispatch()) {
MessageDispatch md = (MessageDispatch) command;
Handler handler = subscriptionsByConsumerId.get(md.getConsumerId());
if (handler != null) {
handler.handle(md);
}
else {
log.warn("No handler for message: " + md);
}
}
}
protected void unknownCommand(Object command) throws Exception {
log.warn("Unkown command: " + command + " of type: " + command.getClass().getName());
}
protected void onIq(final Iq iq) throws Exception {
Object any = iq.getAny();
if (any instanceof Query) {
onAuthQuery(any, iq);
}
else if (any instanceof jabber.iq._private.Query) {
jabber.iq._private.Query query = (jabber.iq._private.Query) any;
if (log.isDebugEnabled()) {
log.debug("Iq Private " + debugString(iq) + " any: " + query.getAny());
}
Iq result = createResult(iq);
jabber.iq._private.Query answer = new jabber.iq._private.Query();
result.setAny(answer);
transport.marshall(result);
}
else if (any instanceof jabber.iq.roster.Query) {
jabber.iq.roster.Query query = (jabber.iq.roster.Query) any;
if (log.isDebugEnabled()) {
log.debug("Iq Roster " + debugString(iq) + " item: " + query.getItem());
}
Iq result = createResult(iq);
jabber.iq.roster.Query roster = new jabber.iq.roster.Query();
result.setAny(roster);
transport.marshall(result);
}
else if (any instanceof org.jabber.protocol.disco_items.Query) {
onDiscoItems(iq, (org.jabber.protocol.disco_items.Query) any);
}
else if (any instanceof org.jabber.protocol.disco_info.Query) {
onDiscoInfo(iq, (org.jabber.protocol.disco_info.Query) any);
}
else {
if (any instanceof Element) {
Element element = (Element) any;
log.warn("Iq Unknown " + debugString(iq) + " element namespace: " + element.getNamespaceURI() + " localName: " + element.getLocalName());
}
else {
log.warn("Iq Unknown " + debugString(iq) + " any: " + any + " of type: " + any.getClass().getName());
}
Iq result = createResult(iq);
jabber.client.Error error = new Error();
error.setUnexpectedRequest("Don't understand: " + any.toString());
result.setAny(error);
transport.marshall(result);
}
}
protected void onAuthQuery(Object any, final Iq iq) throws IOException {
Query query = (Query) any;
if (log.isDebugEnabled()) {
log.debug("Iq Auth Query " + debugString(iq) + " resource: " + query.getResource() + " username: " + query.getUsername());
}
if (query.getPassword() == null) {
Iq result = createResult(iq);
Query required = new Query();
required.setPassword("");
required.setUsername("");
result.setAny(required);
transport.marshall(result);
return;
}
//connectionInfo.setClientId(query.getResource());
connectionInfo.setUserName(query.getUsername());
connectionInfo.setPassword(query.getPassword());
// TODO support digest?
if (connectionInfo.getClientId() == null) {
connectionInfo.setClientId(clientIdGenerator.generateId());
}
sendToActiveMQ(connectionInfo, new Handler() {
public void handle(Response response) throws Exception {
Iq result = createResult(iq);
if (response instanceof ExceptionResponse) {
ExceptionResponse exceptionResponse = (ExceptionResponse) response;
Throwable exception = exceptionResponse.getException();
log.warn("Failed to create connection: " + exception, exception);
Error error = new Error();
result.setError(error);
StringWriter buffer = new StringWriter();
exception.printStackTrace(new PrintWriter(buffer));
error.setInternalServerError(buffer.toString());
}
else {
connected.set(true);
}
transport.marshall(result);
sendToActiveMQ(sessionInfo, createErrorHandler("create sesssion"));
sendToActiveMQ(producerInfo, createErrorHandler("create producer"));
}
});
}
protected String debugString(Iq iq) {
return " to: " + iq.getTo() + " type: " + iq.getType() + " from: " + iq.getFrom() + " id: " + iq.getId();
}
protected void onDiscoItems(Iq iq, org.jabber.protocol.disco_items.Query query) throws IOException {
String to = iq.getTo();
if (log.isDebugEnabled()) {
log.debug("Iq Disco Items query " + debugString(iq) + " node: " + query.getNode() + " item: " + query.getItem());
}
Iq result = createResult(iq);
org.jabber.protocol.disco_items.Query answer = new org.jabber.protocol.disco_items.Query();
if (to == null || to.length() == 0) {
answer.getItem().add(createItem("queues", "Queues", "queues"));
answer.getItem().add(createItem("topics", "Topics", "topics"));
}
else {
// lets not add anything?
}
result.setAny(answer);
transport.marshall(result);
}
protected void onDiscoInfo(Iq iq, org.jabber.protocol.disco_info.Query query) throws IOException {
String to = iq.getTo();
// TODO lets create the topic 'to'
if (log.isDebugEnabled()) {
log.debug("Iq Disco Info query " + debugString(iq) + " node: " + query.getNode() + " features: " + query.getFeature() + " identity: " + query.getIdentity());
}
Iq result = createResult(iq);
org.jabber.protocol.disco_info.Query answer = new org.jabber.protocol.disco_info.Query();
answer.setNode(to);
answer.getFeature().add(createFeature("http://jabber.org/protocol/disco#info"));
answer.getFeature().add(createFeature("http://jabber.org/protocol/disco#items"));
if (to == null || to.length() == 0) {
answer.getIdentity().add(createIdentity("directory", "chatroom", "queues"));
answer.getIdentity().add(createIdentity("directory", "chatroom", "topics"));
/*
answer.getIdentity().add(createIdentity("hierarchy", "queues", "branch"));
answer.getIdentity().add(createIdentity("hierarchy", "topics", "branch"));
*/
}
else {
// for queues/topics
if (to.equals("queues")) {
answer.getIdentity().add(createIdentity("conference", "queue.a", "text"));
answer.getIdentity().add(createIdentity("conference", "queue.b", "text"));
}
else if (to.equals("topics")) {
answer.getIdentity().add(createIdentity("conference", "topic.x", "text"));
answer.getIdentity().add(createIdentity("conference", "topic.y", "text"));
answer.getIdentity().add(createIdentity("conference", "topic.z", "text"));
}
else {
// lets reply to an actual room
answer.getIdentity().add(createIdentity("conference", to, "text"));
answer.getFeature().add(createFeature("http://jabber.org/protocol/muc"));
answer.getFeature().add(createFeature("muc-open"));
}
}
result.setAny(answer);
transport.marshall(result);
}
protected void onPresence(Presence presence) throws IOException, JMSException {
if (log.isDebugEnabled()) {
log.debug("Presence: " + presence.getFrom() + " id: " + presence.getId() + " to: " + presence.getTo() + " type: " + presence.getType()
+ " showOrStatusOrPriority: " + presence.getShowOrStatusOrPriority() + " any: " + presence.getAny());
}
org.jabber.protocol.muc_user.Item item = new org.jabber.protocol.muc_user.Item();
item.setAffiliation("owner");
item.setRole("moderator");
item.setNick("broker");
sendPresence(presence, item);
/*
item = new org.jabber.protocol.muc_user.Item();
item.setAffiliation("admin");
item.setRole("moderator");
sendPresence(presence, item);
*/
// lets create a subscription
final String to = presence.getTo();
ActiveMQDestination destination = createActiveMQDestination(to);
if (destination == null) {
log.debug("No 'to' attribute specified for presence so not creating a JMS subscription");
return;
}
boolean createConsumer = false;
ConsumerInfo consumerInfo = null;
synchronized (jidToConsumerMap) {
consumerInfo = jidToConsumerMap.get(to);
if (consumerInfo == null) {
consumerInfo = new ConsumerInfo();
jidToConsumerMap.put(to, consumerInfo);
ConsumerId consumerId = new ConsumerId(sessionId, consumerIdGenerator.getNextSequenceId());
consumerInfo.setConsumerId(consumerId);
consumerInfo.setPrefetchSize(10);
consumerInfo.setNoLocal(true);
createConsumer = true;
}
}
if (!createConsumer) {
return;
}
consumerInfo.setDestination(destination);
subscriptionsByConsumerId.put(consumerInfo.getConsumerId(), new Handler() {
public void handle(MessageDispatch messageDispatch) throws Exception {
// processing the inbound message
if (log.isDebugEnabled()) {
log.debug("Receiving inbound: " + messageDispatch.getMessage());
}
// lets send back an ACK
MessageAck ack = new MessageAck(messageDispatch, MessageAck.STANDARD_ACK_TYPE, 1);
sendToActiveMQ(ack, createErrorHandler("Ack of message: " + messageDispatch.getMessage().getMessageId()));
Message message = createXmppMessage(to, messageDispatch);
if (message != null) {
if (log.isDebugEnabled()) {
log.debug("Sending message to XMPP client from: " + message.getFrom() + " to: " + message.getTo() + " type: " + message.getType() + " with body: " + message.getAny());
}
transport.marshall(message);
}
}
});
sendToActiveMQ(consumerInfo, createErrorHandler("subscribe to destination: " + destination));
}
protected Message createXmppMessage(String to, MessageDispatch messageDispatch) throws JMSException {
Message answer = new Message();
answer.setType("groupchat");
String from = to;
int idx= from.indexOf('/');
if (idx > 0) {
from = from.substring(0, idx) + "/broker";
}
answer.setFrom(from);
answer.setTo(to);
org.apache.activemq.command.Message message = messageDispatch.getMessage();
//answer.setType(message.getType());
if (message instanceof ActiveMQTextMessage) {
ActiveMQTextMessage activeMQTextMessage = (ActiveMQTextMessage) message;
Body body = new Body();
body.setValue(activeMQTextMessage.getText());
answer.getAny().add(body);
}
else {
// TODO support other message types
}
return answer;
}
protected void sendPresence(Presence presence, org.jabber.protocol.muc_user.Item item) throws IOException {
Presence answer = new Presence();
answer.setFrom(presence.getTo());
answer.setType(presence.getType());
answer.setTo(presence.getFrom());
X x = new X();
x.getDeclineOrDestroyOrInvite().add(item);
answer.getShowOrStatusOrPriority().add(x);
transport.marshall(answer);
}
protected Item createItem(String jid, String name, String node) {
Item answer = new Item();
answer.setJid(jid);
answer.setName(name);
answer.setNode(node);
return answer;
}
protected Identity createIdentity(String category, String type, String name) {
Identity answer = new Identity();
answer.setCategory(category);
answer.setName(name);
answer.setType(type);
return answer;
}
protected Feature createFeature(String var) {
Feature feature = new Feature();
feature.setVar(var);
return feature;
}
/**
* Creates a result command from the input
*/
protected Iq createResult(Iq iq) {
Iq result = new Iq();
result.setId(iq.getId());
result.setFrom(transport.getFrom());
result.setTo(iq.getFrom());
result.setLang(iq.getLang());
result.setType("result");
return result;
}
protected void sendToActiveMQ(Command command, Handler handler) {
command.setCommandId(generateCommandId());
if (handler != null) {
command.setResponseRequired(true);
resposeHandlers.put(command.getCommandId(), handler);
}
transport.getTransportListener().onCommand(command);
}
protected void onStarttls(Starttls starttls) throws Exception {
log.debug("Starttls");
transport.marshall(new Proceed());
}
protected void onMessage(Message message) throws Exception {
if (log.isDebugEnabled()) {
log.debug("Message from: " + message.getFrom() + " to: " + message.getTo() + " subjectOrBodyOrThread: " + message.getSubjectOrBodyOrThread());
}
final ActiveMQMessage activeMQMessage = createActiveMQMessage(message);
ActiveMQDestination destination = createActiveMQDestination(message.getTo());
activeMQMessage.setMessageId(new MessageId(producerInfo, messageIdGenerator.getNextSequenceId()));
activeMQMessage.setDestination(destination);
activeMQMessage.setProducerId(producerId);
activeMQMessage.setTimestamp(System.currentTimeMillis());
addActiveMQMessageHeaders(activeMQMessage, message);
/*
MessageDispatch dispatch = new MessageDispatch();
dispatch.setDestination(destination);
dispatch.setMessage(activeMQMessage);
*/
if (log.isDebugEnabled()) {
log.debug("Sending ActiveMQ message: " + activeMQMessage);
}
sendToActiveMQ(activeMQMessage, createErrorHandler("send message"));
}
protected Handler createErrorHandler(final String text) {
return new Handler() {
public void handle(Response event) throws Exception {
if (event instanceof ExceptionResponse) {
ExceptionResponse exceptionResponse = (ExceptionResponse) event;
Throwable exception = exceptionResponse.getException();
log.error("Failed to " + text + ". Reason: " + exception, exception);
}
else if (log.isDebugEnabled()) {
log.debug("Completed " + text);
}
}
};
}
/**
* Converts the Jabber destination name into a destination in ActiveMQ
*/
protected ActiveMQDestination createActiveMQDestination(String jabberDestination) throws JMSException {
if (jabberDestination == null) {
return null;
}
String name = jabberDestination;
int idx = jabberDestination.indexOf('@');
if (idx > 0) {
name = name.substring(0, idx);
}
return new ActiveMQTopic(name);
}
protected ActiveMQMessage createActiveMQMessage(Message message) throws JMSException {
ActiveMQTextMessage answer = new ActiveMQTextMessage();
String text = "";
List © 2015 - 2025 Weber Informatics LLC | Privacy Policy