From 97baf75a86b8bf875c3918931fb0207d4dd29da3 Mon Sep 17 00:00:00 2001
From: vanitasvitae <vanitasvitae@fsfe.org>
Date: Wed, 23 Aug 2017 14:09:41 +0200
Subject: [PATCH] Fix cancelling file transfers and create int tests

---
 .../component/JingleFileTransfer.java         |   8 +
 .../component/JingleIncomingFileOffer.java    |   1 +
 .../component/JingleOutgoingFileOffer.java    |   6 +
 ...ngleFileTransferCancelIntegrationTest.java | 165 ++++++++++++++++++
 4 files changed, 180 insertions(+)
 create mode 100644 smack-integration-test/src/main/java/org/jivesoftware/smackx/jingle_filetransfer/JingleFileTransferCancelIntegrationTest.java

diff --git a/smack-experimental/src/main/java/org/jivesoftware/smackx/jingle_filetransfer/component/JingleFileTransfer.java b/smack-experimental/src/main/java/org/jivesoftware/smackx/jingle_filetransfer/component/JingleFileTransfer.java
index c7dab6234..e84a756c3 100644
--- a/smack-experimental/src/main/java/org/jivesoftware/smackx/jingle_filetransfer/component/JingleFileTransfer.java
+++ b/smack-experimental/src/main/java/org/jivesoftware/smackx/jingle_filetransfer/component/JingleFileTransfer.java
@@ -19,6 +19,8 @@ package org.jivesoftware.smackx.jingle_filetransfer.component;
 import java.util.ArrayList;
 import java.util.Collections;
 import java.util.List;
+import java.util.logging.Level;
+import java.util.logging.Logger;
 
 import org.jivesoftware.smack.SmackException;
 import org.jivesoftware.smack.XMPPConnection;
@@ -35,6 +37,8 @@ import org.jivesoftware.smackx.jingle_filetransfer.listener.ProgressListener;
  */
 public abstract class JingleFileTransfer extends JingleDescription<JingleFileTransferElement> implements JingleFileTransferController {
 
+    private static final Logger LOGGER = Logger.getLogger(JingleFileTransfer.class.getName());
+
     public static final String NAMESPACE_V5 = "urn:xmpp:jingle:apps:file-transfer:5";
     public static final String NAMESPACE = NAMESPACE_V5;
 
@@ -69,19 +73,23 @@ public abstract class JingleFileTransfer extends JingleDescription<JingleFileTra
         switch (state) {
             case pending:
                 if (session.isResponder()) {
+                    LOGGER.log(Level.INFO, "Cancel pending file transfer as responder.");
                     connection.createStanzaCollectorAndSend(JingleElement.createSessionTerminate(session.getPeer(), session.getSessionId(), JingleReasonElement.Reason.decline));
                 } else {
+                    LOGGER.log(Level.INFO, "Cancel pending file transfer as initiator.");
                     connection.createStanzaCollectorAndSend(JingleElement.createSessionTerminate(session.getPeer(), session.getSessionId(), JingleReasonElement.Reason.cancel));
                 }
                 break;
 
             case active:
+                LOGGER.log(Level.INFO, "Cancel active file transfer.");
                 connection.createStanzaCollectorAndSend(JingleElement.createSessionTerminate(session.getPeer(), session.getSessionId(), JingleReasonElement.Reason.cancel));
                 break;
 
             default: break;
         }
         getParent().onContentCancel();
+        state = State.cancelled;
     }
 
     public void notifyProgressListenersStarted() {
diff --git a/smack-experimental/src/main/java/org/jivesoftware/smackx/jingle_filetransfer/component/JingleIncomingFileOffer.java b/smack-experimental/src/main/java/org/jivesoftware/smackx/jingle_filetransfer/component/JingleIncomingFileOffer.java
index fe08e6de8..f27af1f51 100644
--- a/smack-experimental/src/main/java/org/jivesoftware/smackx/jingle_filetransfer/component/JingleIncomingFileOffer.java
+++ b/smack-experimental/src/main/java/org/jivesoftware/smackx/jingle_filetransfer/component/JingleIncomingFileOffer.java
@@ -67,6 +67,7 @@ public class JingleIncomingFileOffer extends AbstractJingleFileOffer implements
 
         if (state == State.negotiating) {
             state = State.active;
+            notifyProgressListenersStarted();
         } else {
             return;
         }
diff --git a/smack-experimental/src/main/java/org/jivesoftware/smackx/jingle_filetransfer/component/JingleOutgoingFileOffer.java b/smack-experimental/src/main/java/org/jivesoftware/smackx/jingle_filetransfer/component/JingleOutgoingFileOffer.java
index 93a21cc18..6686fd0bd 100644
--- a/smack-experimental/src/main/java/org/jivesoftware/smackx/jingle_filetransfer/component/JingleOutgoingFileOffer.java
+++ b/smack-experimental/src/main/java/org/jivesoftware/smackx/jingle_filetransfer/component/JingleOutgoingFileOffer.java
@@ -42,11 +42,13 @@ public class JingleOutgoingFileOffer extends AbstractJingleFileOffer implements
     public JingleOutgoingFileOffer(File file, JingleFile metadata) throws FileNotFoundException {
         super(metadata);
         this.source = new FileInputStream(file);
+        this.state = State.pending;
     }
 
     public JingleOutgoingFileOffer(InputStream inputStream, JingleFile metadata) {
         super(metadata);
         this.source = inputStream;
+        this.state = State.pending;
     }
 
     @Override
@@ -60,6 +62,10 @@ public class JingleOutgoingFileOffer extends AbstractJingleFileOffer implements
             throw new IllegalStateException("Source InputStream is null!");
         }
 
+        state = State.active;
+
+        notifyProgressListenersStarted();
+
         OutputStream outputStream = null;
 
         try {
diff --git a/smack-integration-test/src/main/java/org/jivesoftware/smackx/jingle_filetransfer/JingleFileTransferCancelIntegrationTest.java b/smack-integration-test/src/main/java/org/jivesoftware/smackx/jingle_filetransfer/JingleFileTransferCancelIntegrationTest.java
new file mode 100644
index 000000000..909965c83
--- /dev/null
+++ b/smack-integration-test/src/main/java/org/jivesoftware/smackx/jingle_filetransfer/JingleFileTransferCancelIntegrationTest.java
@@ -0,0 +1,165 @@
+/**
+ *
+ * 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;
+
+import static junit.framework.TestCase.fail;
+
+import java.io.ByteArrayInputStream;
+import java.io.ByteArrayOutputStream;
+import java.io.IOException;
+import java.util.Random;
+
+import org.jivesoftware.smack.SmackException;
+import org.jivesoftware.smack.XMPPException;
+import org.jivesoftware.smackx.disco.ServiceDiscoveryManager;
+import org.jivesoftware.smackx.jingle.element.JingleReasonElement;
+import org.jivesoftware.smackx.jingle.transport.jingle_ibb.JingleIBBTransport;
+import org.jivesoftware.smackx.jingle.transport.jingle_ibb.JingleIBBTransportManager;
+import org.jivesoftware.smackx.jingle_filetransfer.component.JingleFile;
+import org.jivesoftware.smackx.jingle_filetransfer.controller.IncomingFileOfferController;
+import org.jivesoftware.smackx.jingle_filetransfer.controller.OutgoingFileOfferController;
+import org.jivesoftware.smackx.jingle_filetransfer.listener.IncomingFileOfferListener;
+import org.jivesoftware.smackx.jingle_filetransfer.listener.ProgressListener;
+
+import org.igniterealtime.smack.inttest.AbstractSmackIntegrationTest;
+import org.igniterealtime.smack.inttest.SmackIntegrationTest;
+import org.igniterealtime.smack.inttest.SmackIntegrationTestEnvironment;
+import org.igniterealtime.smack.inttest.util.SimpleResultSyncPoint;
+import org.junit.BeforeClass;
+
+public class JingleFileTransferCancelIntegrationTest extends AbstractSmackIntegrationTest {
+
+    public JingleFileTransferCancelIntegrationTest(SmackIntegrationTestEnvironment environment) {
+        super(environment);
+    }
+
+    @BeforeClass
+    public void setup() {
+        JingleFileTransferManager.getInstanceFor(conOne);
+        JingleIBBTransportManager.getInstanceFor(conOne);
+        ServiceDiscoveryManager.getInstanceFor(conOne).addFeature(JingleIBBTransport.NAMESPACE);
+        JingleFileTransferManager.getInstanceFor(conThree);
+        JingleIBBTransportManager.getInstanceFor(conThree);
+        ServiceDiscoveryManager.getInstanceFor(conThree).addFeature(JingleIBBTransport.NAMESPACE);
+    }
+
+    @SmackIntegrationTest
+    public void senderCancelTest() throws Exception {
+        final SimpleResultSyncPoint s1 = new SimpleResultSyncPoint();
+
+        byte[] payload = new byte[320000];
+        new Random().nextBytes(payload);
+
+        JingleFileTransferManager.getInstanceFor(conThree).addIncomingFileOfferListener(new IncomingFileOfferListener() {
+            @Override
+            public void onIncomingFileOffer(IncomingFileOfferController offer) {
+                offer.addProgressListener(new ProgressListener() {
+                    @Override
+                    public void started() {
+
+                    }
+
+                    @Override
+                    public void terminated(JingleReasonElement.Reason reason) {
+                        if (reason == JingleReasonElement.Reason.cancel) {
+                            s1.signal();
+                        } else {
+                            s1.signalFailure();
+                        }
+                    }
+                });
+                try {
+                    offer.accept(conThree, new ByteArrayOutputStream());
+                } catch (InterruptedException | IOException | SmackException.NoResponseException | SmackException.NotConnectedException | XMPPException.XMPPErrorException e) {
+                    fail(e.toString());
+                }
+            }
+        });
+
+        final OutgoingFileOfferController out = JingleFileTransferManager.getInstanceFor(conOne).sendStream(new ByteArrayInputStream(payload), new JingleFile("name", null, 320000, null, null, null), conThree.getUser().asFullJidOrThrow());
+        out.addProgressListener(new ProgressListener() {
+            @Override
+            public void started() {
+                try {
+                    out.cancel(conOne);
+                } catch (SmackException.NotConnectedException | InterruptedException e) {
+                    fail(e.toString());
+                }
+            }
+
+            @Override
+            public void terminated(JingleReasonElement.Reason reason) {
+
+            }
+        });
+
+        s1.waitForResult(60 * 1000);
+    }
+
+    @SmackIntegrationTest
+    public void receiverCancelTest() throws Exception {
+        final SimpleResultSyncPoint s1 = new SimpleResultSyncPoint();
+
+        byte[] payload = new byte[320000];
+        new Random().nextBytes(payload);
+
+        JingleFileTransferManager.getInstanceFor(conThree).addIncomingFileOfferListener(new IncomingFileOfferListener() {
+            @Override
+            public void onIncomingFileOffer(final IncomingFileOfferController offer) {
+                offer.addProgressListener(new ProgressListener() {
+                    @Override
+                    public void started() {
+                        try {
+                            offer.cancel(conThree);
+                        } catch (SmackException.NotConnectedException | InterruptedException e) {
+                            fail(e.toString());
+                        }
+                    }
+
+                    @Override
+                    public void terminated(JingleReasonElement.Reason reason) {
+
+                    }
+                });
+                try {
+                    offer.accept(conThree, new ByteArrayOutputStream());
+                } catch (InterruptedException | IOException | SmackException.NoResponseException | SmackException.NotConnectedException | XMPPException.XMPPErrorException e) {
+                    fail(e.toString());
+                }
+            }
+        });
+
+        final OutgoingFileOfferController out = JingleFileTransferManager.getInstanceFor(conOne).sendStream(new ByteArrayInputStream(payload), new JingleFile("name", null, 320000, null, null, null), conThree.getUser().asFullJidOrThrow());
+        out.addProgressListener(new ProgressListener() {
+            @Override
+            public void started() {
+
+            }
+
+            @Override
+            public void terminated(JingleReasonElement.Reason reason) {
+                if (reason == JingleReasonElement.Reason.cancel) {
+                    s1.signal();
+                } else {
+                    s1.signalFailure();
+                }
+            }
+        });
+
+        s1.waitForResult(60 * 1000);
+    }
+}