2017-05-30 18:20:54 +02:00
|
|
|
/**
|
|
|
|
*
|
|
|
|
* Copyright 2017 Paul Schaub
|
|
|
|
*
|
|
|
|
* 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.jivesoftware.smackx.jingle_filetransfer;
|
|
|
|
|
2017-06-04 17:18:40 +02:00
|
|
|
import java.io.File;
|
|
|
|
import java.io.IOException;
|
|
|
|
import java.util.ArrayList;
|
2017-06-06 16:51:45 +02:00
|
|
|
import java.util.Collections;
|
2017-06-04 17:18:40 +02:00
|
|
|
import java.util.HashSet;
|
|
|
|
import java.util.WeakHashMap;
|
|
|
|
import java.util.logging.Logger;
|
|
|
|
|
2017-05-30 18:20:54 +02:00
|
|
|
import org.jivesoftware.smack.Manager;
|
2017-05-31 19:12:38 +02:00
|
|
|
import org.jivesoftware.smack.SmackException;
|
2017-05-30 18:20:54 +02:00
|
|
|
import org.jivesoftware.smack.XMPPConnection;
|
2017-06-08 17:32:37 +02:00
|
|
|
import org.jivesoftware.smack.XMPPException;
|
2017-05-31 19:12:38 +02:00
|
|
|
import org.jivesoftware.smack.packet.IQ;
|
2017-06-06 16:51:45 +02:00
|
|
|
import org.jivesoftware.smack.util.StringUtils;
|
2017-05-31 22:31:54 +02:00
|
|
|
import org.jivesoftware.smackx.disco.ServiceDiscoveryManager;
|
2017-05-31 19:12:38 +02:00
|
|
|
import org.jivesoftware.smackx.hash.HashManager;
|
2017-06-01 13:43:17 +02:00
|
|
|
import org.jivesoftware.smackx.hash.element.HashElement;
|
2017-06-08 17:32:37 +02:00
|
|
|
import org.jivesoftware.smackx.jingle.AbstractJingleContentTransportManager;
|
|
|
|
import org.jivesoftware.smackx.jingle.JingleContentProviderManager;
|
2017-06-01 16:49:58 +02:00
|
|
|
import org.jivesoftware.smackx.jingle.JingleHandler;
|
2017-05-31 19:12:38 +02:00
|
|
|
import org.jivesoftware.smackx.jingle.JingleManager;
|
|
|
|
import org.jivesoftware.smackx.jingle.element.Jingle;
|
|
|
|
import org.jivesoftware.smackx.jingle.element.JingleContent;
|
2017-06-03 23:38:46 +02:00
|
|
|
import org.jivesoftware.smackx.jingle.element.JingleContentDescriptionChildElement;
|
2017-06-02 18:45:11 +02:00
|
|
|
import org.jivesoftware.smackx.jingle_filetransfer.callback.IncomingJingleFileTransferCallback;
|
2017-06-04 00:06:37 +02:00
|
|
|
import org.jivesoftware.smackx.jingle_filetransfer.element.JingleFileTransferChildElement;
|
2017-06-04 17:18:40 +02:00
|
|
|
import org.jivesoftware.smackx.jingle_filetransfer.element.JingleFileTransferContentDescription;
|
2017-06-01 16:49:58 +02:00
|
|
|
import org.jivesoftware.smackx.jingle_filetransfer.listener.IncomingJingleFileTransferListener;
|
2017-06-04 00:06:37 +02:00
|
|
|
import org.jivesoftware.smackx.jingle_filetransfer.provider.JingleFileTransferContentDescriptionProvider;
|
2017-06-06 16:51:45 +02:00
|
|
|
import org.jivesoftware.smackx.jingle_ibb.JingleInBandBytestreamTransportManager;
|
2017-05-31 19:12:38 +02:00
|
|
|
import org.jxmpp.jid.FullJid;
|
2017-05-30 18:20:54 +02:00
|
|
|
|
|
|
|
/**
|
|
|
|
* Manager for Jingle File Transfers.
|
|
|
|
*
|
|
|
|
* @author Paul Schaub
|
|
|
|
*/
|
2017-06-06 16:51:45 +02:00
|
|
|
public final class JingleFileTransferManager extends Manager implements JingleHandler {
|
2017-05-30 18:20:54 +02:00
|
|
|
|
2017-05-31 19:12:38 +02:00
|
|
|
private static final Logger LOGGER = Logger.getLogger(JingleFileTransferManager.class.getName());
|
|
|
|
|
2017-05-30 23:42:31 +02:00
|
|
|
public static final String NAMESPACE_V5 = "urn:xmpp:jingle:apps:file-transfer:5";
|
2017-05-30 18:20:54 +02:00
|
|
|
|
2017-06-06 16:51:45 +02:00
|
|
|
private final JingleManager jingleManager;
|
2017-05-30 18:20:54 +02:00
|
|
|
private static final WeakHashMap<XMPPConnection, JingleFileTransferManager> INSTANCES = new WeakHashMap<>();
|
2017-06-01 16:49:58 +02:00
|
|
|
private final HashSet<IncomingJingleFileTransferListener> incomingJingleFileTransferListeners = new HashSet<>();
|
2017-05-30 18:20:54 +02:00
|
|
|
|
|
|
|
/**
|
2017-05-31 13:33:37 +02:00
|
|
|
* Private constructor. This registers a JingleContentDescriptionFileTransferProvider with the
|
|
|
|
* JingleContentProviderManager.
|
2017-05-30 18:20:54 +02:00
|
|
|
* @param connection connection
|
|
|
|
*/
|
|
|
|
private JingleFileTransferManager(XMPPConnection connection) {
|
|
|
|
super(connection);
|
2017-05-31 22:31:54 +02:00
|
|
|
ServiceDiscoveryManager sdm = ServiceDiscoveryManager.getInstanceFor(connection);
|
|
|
|
sdm.addFeature(NAMESPACE_V5);
|
2017-06-06 16:51:45 +02:00
|
|
|
jingleManager = JingleManager.getInstanceFor(connection);
|
|
|
|
jingleManager.registerDescriptionHandler(
|
2017-06-01 16:49:58 +02:00
|
|
|
NAMESPACE_V5, this);
|
2017-05-31 13:33:37 +02:00
|
|
|
JingleContentProviderManager.addJingleContentDescriptionProvider(
|
2017-06-04 00:06:37 +02:00
|
|
|
NAMESPACE_V5, new JingleFileTransferContentDescriptionProvider());
|
2017-06-06 16:51:45 +02:00
|
|
|
JingleInBandBytestreamTransportManager.getInstanceFor(connection);
|
2017-05-30 18:20:54 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Return a new instance of the FileTransferManager for the given connection.
|
|
|
|
*
|
|
|
|
* @param connection XMPPConnection we wish to get a FileTransferManager for.
|
|
|
|
* @return manager instance.
|
|
|
|
*/
|
|
|
|
public static JingleFileTransferManager getInstanceFor(XMPPConnection connection) {
|
|
|
|
JingleFileTransferManager manager = INSTANCES.get(connection);
|
|
|
|
if (manager == null) {
|
|
|
|
manager = new JingleFileTransferManager(connection);
|
|
|
|
INSTANCES.put(connection, manager);
|
|
|
|
}
|
|
|
|
return manager;
|
|
|
|
}
|
2017-05-31 19:12:38 +02:00
|
|
|
|
2017-06-01 16:49:58 +02:00
|
|
|
public void addIncomingFileTransferListener(IncomingJingleFileTransferListener listener) {
|
|
|
|
incomingJingleFileTransferListeners.add(listener);
|
|
|
|
}
|
|
|
|
|
|
|
|
public void removeIncomingFileTransferListener(IncomingJingleFileTransferListener listener) {
|
|
|
|
incomingJingleFileTransferListeners.remove(listener);
|
|
|
|
}
|
|
|
|
|
2017-06-06 16:51:45 +02:00
|
|
|
void notifyIncomingFileTransferListeners(Jingle jingle, IncomingJingleFileTransferCallback callback) {
|
2017-06-04 17:18:40 +02:00
|
|
|
for (IncomingJingleFileTransferListener l : incomingJingleFileTransferListeners) {
|
2017-06-06 16:51:45 +02:00
|
|
|
l.onIncomingJingleFileTransfer(jingle, callback);
|
2017-06-04 17:18:40 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2017-05-31 19:12:38 +02:00
|
|
|
/**
|
|
|
|
* QnD method.
|
|
|
|
* @param file
|
|
|
|
*/
|
2017-06-08 17:32:37 +02:00
|
|
|
public void sendFile(File file, final FullJid recipient) throws IOException, SmackException.NotConnectedException, InterruptedException, XMPPException.XMPPErrorException, SmackException.NoResponseException {
|
|
|
|
AbstractJingleContentTransportManager<?> preferedTransportManager = JingleInBandBytestreamTransportManager.getInstanceFor(connection());
|
|
|
|
|
2017-06-06 16:51:45 +02:00
|
|
|
JingleFileTransferSession session = new JingleFileTransferSession(connection(), recipient, connection().getUser(), recipient);
|
|
|
|
JingleFileTransferChildElement.Builder b = JingleFileTransferChildElement.getBuilder();
|
|
|
|
b.setFile(file);
|
|
|
|
byte[] buf = new byte[(int) file.length()];
|
|
|
|
HashElement hashElement = FileAndHashReader.readAndCalculateHash(file, buf, HashManager.ALGORITHM.SHA_256);
|
|
|
|
b.setHash(hashElement);
|
|
|
|
b.setDescription("File");
|
|
|
|
b.setMediaType("text/plain");
|
|
|
|
|
|
|
|
session.setBytes(buf);
|
|
|
|
JingleManager.getInstanceFor(connection()).registerJingleSession(session);
|
|
|
|
|
|
|
|
ArrayList<JingleContentDescriptionChildElement> payloads = new ArrayList<>();
|
|
|
|
payloads.add(b.build());
|
|
|
|
|
|
|
|
JingleContent.Builder bb = JingleContent.getBuilder();
|
|
|
|
bb.setDescription(new JingleFileTransferContentDescription(payloads))
|
2017-05-31 19:12:38 +02:00
|
|
|
.setCreator(JingleContent.Creator.initiator)
|
2017-06-06 16:51:45 +02:00
|
|
|
.setName(StringUtils.randomString(24))
|
2017-06-08 17:32:37 +02:00
|
|
|
.addTransport(preferedTransportManager.createJingleContentTransport());
|
2017-05-31 19:12:38 +02:00
|
|
|
|
2017-06-06 16:51:45 +02:00
|
|
|
Jingle jingle = (Jingle) session.initiate(Collections.singletonList(bb.build()));
|
2017-06-04 20:33:44 +02:00
|
|
|
jingle.setTo(recipient);
|
2017-06-06 16:51:45 +02:00
|
|
|
connection().sendStanza(jingle);
|
|
|
|
}
|
2017-05-31 19:12:38 +02:00
|
|
|
|
2017-06-06 16:51:45 +02:00
|
|
|
public FullJid ourJid() {
|
|
|
|
return connection().getUser();
|
|
|
|
}
|
2017-05-31 19:12:38 +02:00
|
|
|
|
2017-06-06 16:51:45 +02:00
|
|
|
@Override
|
|
|
|
public IQ handleJingleRequest(Jingle jingle) {
|
|
|
|
JingleFileTransferSession session = new JingleFileTransferSession(connection(), jingle);
|
|
|
|
JingleManager.getInstanceFor(connection()).registerJingleSession(session);
|
|
|
|
return session.handleRequest(jingle);
|
2017-05-31 19:12:38 +02:00
|
|
|
}
|
2017-05-30 18:20:54 +02:00
|
|
|
}
|