2014-02-17 23:58:40 +01:00
|
|
|
/**
|
|
|
|
*
|
|
|
|
* Copyright the original author or 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.jivesoftware.smackx.bytestreams.socks5;
|
|
|
|
|
|
|
|
import java.util.concurrent.ExecutorService;
|
|
|
|
import java.util.concurrent.Executors;
|
2014-03-19 14:22:20 +01:00
|
|
|
import java.util.logging.Level;
|
|
|
|
import java.util.logging.Logger;
|
2014-02-17 23:58:40 +01:00
|
|
|
|
2014-03-19 14:22:20 +01:00
|
|
|
import org.jivesoftware.smack.SmackException.NotConnectedException;
|
2015-01-08 11:01:35 +01:00
|
|
|
import org.jivesoftware.smack.iqrequest.AbstractIqRequestHandler;
|
|
|
|
import org.jivesoftware.smack.packet.IQ;
|
2015-02-05 11:17:27 +01:00
|
|
|
import org.jivesoftware.smack.packet.Stanza;
|
2017-06-14 17:12:43 +02:00
|
|
|
|
2014-02-17 23:58:40 +01:00
|
|
|
import org.jivesoftware.smackx.bytestreams.BytestreamListener;
|
|
|
|
import org.jivesoftware.smackx.bytestreams.socks5.packet.Bytestream;
|
2015-02-27 10:02:48 +01:00
|
|
|
import org.jivesoftware.smackx.filetransfer.StreamNegotiator;
|
2014-02-17 23:58:40 +01:00
|
|
|
|
|
|
|
/**
|
|
|
|
* InitiationListener handles all incoming SOCKS5 Bytestream initiation requests. If there are no
|
|
|
|
* listeners for a SOCKS5 bytestream request InitiationListener will always refuse the request and
|
|
|
|
* reply with a <not-acceptable/> error (<a
|
|
|
|
* href="http://xmpp.org/extensions/xep-0065.html#usecase-alternate">XEP-0065</a> Section 5.2.A2).
|
2018-05-09 23:06:12 +02:00
|
|
|
*
|
2014-02-17 23:58:40 +01:00
|
|
|
* @author Henning Staib
|
|
|
|
*/
|
2015-01-08 11:01:35 +01:00
|
|
|
final class InitiationListener extends AbstractIqRequestHandler {
|
2014-03-19 14:22:20 +01:00
|
|
|
private static final Logger LOGGER = Logger.getLogger(InitiationListener.class.getName());
|
2014-02-17 23:58:40 +01:00
|
|
|
|
|
|
|
/* manager containing the listeners and the XMPP connection */
|
|
|
|
private final Socks5BytestreamManager manager;
|
|
|
|
|
|
|
|
/* executor service to process incoming requests concurrently */
|
|
|
|
private final ExecutorService initiationListenerExecutor;
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Constructor
|
2018-05-09 23:06:12 +02:00
|
|
|
*
|
2014-02-17 23:58:40 +01:00
|
|
|
* @param manager the SOCKS5 Bytestream manager
|
|
|
|
*/
|
|
|
|
protected InitiationListener(Socks5BytestreamManager manager) {
|
2015-01-08 11:01:35 +01:00
|
|
|
super(Bytestream.ELEMENT, Bytestream.NAMESPACE, IQ.Type.set, Mode.async);
|
2014-02-17 23:58:40 +01:00
|
|
|
this.manager = manager;
|
|
|
|
initiationListenerExecutor = Executors.newCachedThreadPool();
|
|
|
|
}
|
|
|
|
|
2015-01-08 11:01:35 +01:00
|
|
|
|
|
|
|
@Override
|
|
|
|
public IQ handleIQRequest(final IQ packet) {
|
2014-02-17 23:58:40 +01:00
|
|
|
initiationListenerExecutor.execute(new Runnable() {
|
|
|
|
|
2017-02-11 16:16:41 +01:00
|
|
|
@Override
|
2014-02-17 23:58:40 +01:00
|
|
|
public void run() {
|
2014-03-19 14:22:20 +01:00
|
|
|
try {
|
|
|
|
processRequest(packet);
|
|
|
|
}
|
2015-02-14 09:43:44 +01:00
|
|
|
catch (InterruptedException | NotConnectedException e) {
|
2014-03-19 14:22:20 +01:00
|
|
|
LOGGER.log(Level.WARNING, "process request", e);
|
|
|
|
}
|
2014-02-17 23:58:40 +01:00
|
|
|
}
|
|
|
|
});
|
2015-01-08 11:01:35 +01:00
|
|
|
|
|
|
|
return null;
|
2014-02-17 23:58:40 +01:00
|
|
|
}
|
|
|
|
|
2015-02-14 09:43:44 +01:00
|
|
|
private void processRequest(Stanza packet) throws NotConnectedException, InterruptedException {
|
2014-02-17 23:58:40 +01:00
|
|
|
Bytestream byteStreamRequest = (Bytestream) packet;
|
|
|
|
|
2015-03-04 22:42:36 +01:00
|
|
|
StreamNegotiator.signal(byteStreamRequest.getFrom().toString() + '\t' + byteStreamRequest.getSessionID(), byteStreamRequest);
|
2015-02-27 10:02:48 +01:00
|
|
|
|
2014-02-17 23:58:40 +01:00
|
|
|
// ignore request if in ignore list
|
|
|
|
if (this.manager.getIgnoredBytestreamRequests().remove(byteStreamRequest.getSessionID())) {
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
// build bytestream request from packet
|
|
|
|
Socks5BytestreamRequest request = new Socks5BytestreamRequest(this.manager,
|
|
|
|
byteStreamRequest);
|
|
|
|
|
|
|
|
// notify listeners for bytestream initiation from a specific user
|
|
|
|
BytestreamListener userListener = this.manager.getUserListener(byteStreamRequest.getFrom());
|
|
|
|
if (userListener != null) {
|
|
|
|
userListener.incomingBytestreamRequest(request);
|
|
|
|
|
|
|
|
}
|
|
|
|
else if (!this.manager.getAllRequestListeners().isEmpty()) {
|
|
|
|
/*
|
|
|
|
* if there is no user specific listener inform listeners for all initiation requests
|
|
|
|
*/
|
|
|
|
for (BytestreamListener listener : this.manager.getAllRequestListeners()) {
|
|
|
|
listener.incomingBytestreamRequest(request);
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
else {
|
|
|
|
/*
|
|
|
|
* if there is no listener for this initiation request, reply with reject message
|
|
|
|
*/
|
|
|
|
this.manager.replyRejectPacket(byteStreamRequest);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Shuts down the listeners executor service.
|
|
|
|
*/
|
|
|
|
protected void shutdown() {
|
|
|
|
this.initiationListenerExecutor.shutdownNow();
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|