| /** |
| * All rights reserved. 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; |
| |
| import org.jivesoftware.smack.PacketListener; |
| import org.jivesoftware.smack.filter.AndFilter; |
| import org.jivesoftware.smack.filter.IQTypeFilter; |
| import org.jivesoftware.smack.filter.PacketFilter; |
| import org.jivesoftware.smack.filter.PacketTypeFilter; |
| import org.jivesoftware.smack.packet.IQ; |
| import org.jivesoftware.smack.packet.Packet; |
| import org.jivesoftware.smackx.bytestreams.BytestreamListener; |
| import org.jivesoftware.smackx.bytestreams.socks5.packet.Bytestream; |
| |
| /** |
| * 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). |
| * |
| * @author Henning Staib |
| */ |
| final class InitiationListener implements PacketListener { |
| |
| /* manager containing the listeners and the XMPP connection */ |
| private final Socks5BytestreamManager manager; |
| |
| /* packet filter for all SOCKS5 Bytestream requests */ |
| private final PacketFilter initFilter = new AndFilter(new PacketTypeFilter(Bytestream.class), |
| new IQTypeFilter(IQ.Type.SET)); |
| |
| /* executor service to process incoming requests concurrently */ |
| private final ExecutorService initiationListenerExecutor; |
| |
| /** |
| * Constructor |
| * |
| * @param manager the SOCKS5 Bytestream manager |
| */ |
| protected InitiationListener(Socks5BytestreamManager manager) { |
| this.manager = manager; |
| initiationListenerExecutor = Executors.newCachedThreadPool(); |
| } |
| |
| public void processPacket(final Packet packet) { |
| initiationListenerExecutor.execute(new Runnable() { |
| |
| public void run() { |
| processRequest(packet); |
| } |
| }); |
| } |
| |
| private void processRequest(Packet packet) { |
| Bytestream byteStreamRequest = (Bytestream) packet; |
| |
| // 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); |
| } |
| } |
| |
| /** |
| * Returns the packet filter for SOCKS5 Bytestream initialization requests. |
| * |
| * @return the packet filter for SOCKS5 Bytestream initialization requests |
| */ |
| protected PacketFilter getFilter() { |
| return this.initFilter; |
| } |
| |
| /** |
| * Shuts down the listeners executor service. |
| */ |
| protected void shutdown() { |
| this.initiationListenerExecutor.shutdownNow(); |
| } |
| |
| } |