Temp

smack-experimental/src/main/java/org/jivesoftware/smackx/jet/JetManager.java

@@ -16,26 +16,54 @@
  */
 package org.jivesoftware.smackx.jet;
 
+import java.io.File;
+import java.io.FileInputStream;
+import java.security.SecureRandom;
+import java.util.Collections;
 import java.util.HashMap;
 import java.util.Map;
 import java.util.WeakHashMap;
+import java.util.logging.Logger;
+import javax.crypto.Cipher;
+import javax.crypto.CipherInputStream;
+import javax.crypto.KeyGenerator;
+import javax.crypto.SecretKey;
+import javax.crypto.spec.IvParameterSpec;
+import javax.crypto.spec.SecretKeySpec;
 
 import org.jivesoftware.smack.Manager;
 import org.jivesoftware.smack.XMPPConnection;
+import org.jivesoftware.smack.packet.Element;
+import org.jivesoftware.smack.packet.ExtensionElement;
+import org.jivesoftware.smackx.jingle.JingleManager;
+import org.jivesoftware.smackx.jingle.JingleUtil;
+import org.jivesoftware.smackx.jingle.element.Jingle;
+import org.jivesoftware.smackx.jingle.element.JingleContent;
+import org.jivesoftware.smackx.jingle.element.JingleContentDescriptionChildElement;
+import org.jivesoftware.smackx.jingle_filetransfer.OutgoingJingleFileOffer;
+import org.jivesoftware.smackx.jingle_filetransfer.element.JingleFileTransfer;
+import org.jivesoftware.smackx.jingle_filetransfer.element.JingleFileTransferChild;
+import org.jivesoftware.smackx.jingle_filetransfer.handler.FileTransferHandler;
+
+import org.jxmpp.jid.FullJid;
 
 /**
  * Manager for Jingle Encrypted Transfers (XEP-XXXX).
  */
 public final class JetManager extends Manager {
 
+    private static final Logger LOGGER = Logger.getLogger(JetManager.class.getName());
+
     public static final String NAMESPACE = "urn:xmpp:jingle:jet:0";
 
     private static final WeakHashMap<XMPPConnection, JetManager> INSTANCES = new WeakHashMap<>();
+    private final JingleUtil jutil;
 
-    private static final Map<String, JingleEncryptionMethod> encryptionProviders = new HashMap<>();
+    private static final Map<String, JingleEncryptionMethod> encryptionMethods = new HashMap<>();
 
     private JetManager(XMPPConnection connection) {
         super(connection);
+        jutil = new JingleUtil(connection);
     }
 
     public static JetManager getInstanceFor(XMPPConnection connection) {
@@ -49,16 +77,91 @@ public final class JetManager extends Manager {
         return manager;
     }
 
+    public FileTransferHandler sendEncryptedFile(FullJid recipient, File file, String encryptionMethodNamespace) throws Exception {
 
-    public void registerEncryptionProvider(String namespace, JingleEncryptionMethod provider) {
-        encryptionProviders.put(namespace, provider);
+        JingleEncryptionMethod encryptionMethod = getEncryptionMethod(encryptionMethodNamespace);
+        if (encryptionMethod == null) {
+            throw new IllegalStateException("No encryption method with namespace " + encryptionMethodNamespace + " registered.");
+        }
+
+        int keyLength = 256;
+        String keyType = "AES";
+        String cipherMode = "AES/GCM/NoPadding";
+
+        KeyGenerator keyGenerator = KeyGenerator.getInstance(keyType);
+        keyGenerator.init(keyLength);
+        byte[] key = keyGenerator.generateKey().getEncoded();
+
+        SecureRandom secureRandom = new SecureRandom();
+        byte[] iv = new byte[keyLength];
+        secureRandom.nextBytes(iv);
+
+        byte[] keyAndIv = new byte[2 * keyLength];
+        System.arraycopy(key, 0, keyAndIv, 0, keyLength);
+        System.arraycopy(iv, 0, keyAndIv, keyLength, keyLength);
+
+        SecretKey secretKey = new SecretKeySpec(key, keyType);
+        IvParameterSpec ivSpec = new IvParameterSpec(iv);
+        Cipher cipher = Cipher.getInstance(cipherMode, "BC");
+        cipher.init(Cipher.ENCRYPT_MODE, secretKey, ivSpec);
+
+        Exception ioe = null;
+        byte[] fileBuf = null;
+
+        FileInputStream fi = null;
+        CipherInputStream ci = null;
+
+        try {
+            fi = new FileInputStream(file);
+            ci = new CipherInputStream(fi, cipher);
+
+            fileBuf = new byte[(int) file.length()];
+            ci.read(fileBuf);
+
+        } catch (Exception e) {
+            ioe = e;
+
+        } finally {
+
+            if (ci != null) {
+                ci.close();
+            }
+
+            if (fi != null) {
+                fi.close();
+            }
+        }
+
+        if (ioe != null) {
+            throw ioe;
+        }
+
+        if (fileBuf == null) {
+            return null;
+        }
+
+        ExtensionElement encryptionExtension = encryptionMethod.encryptJingleTransfer(recipient, keyAndIv);
+
+        OutgoingJingleFileOffer offer = new OutgoingJingleFileOffer(connection(), recipient);
+
+        JingleFileTransferChild fileTransferChild = JingleFileTransferChild.getBuilder().setFile(file).build();
+        JingleFileTransfer fileTransfer = new JingleFileTransfer(Collections.<JingleContentDescriptionChildElement>singletonList(fileTransferChild));
+        Jingle initiate = jutil.createSessionInitiateFileOffer(recipient, JingleManager.randomId(), JingleContent.Creator.initiator,
+                JingleManager.randomId(), fileTransfer, offer.getTransportSession().createTransport(), Collections.<Element>singletonList(encryptionExtension));
+        return offer;
     }
 
-    public void unregisterEncryptionProvider(String namespace) {
-        encryptionProviders.remove(namespace);
+
+    public void registerEncryptionMethod(String namespace, JingleEncryptionMethod method) {
+        encryptionMethods.put(namespace, method);
     }
 
-    public JingleEncryptionMethod getEncryptionProvider(String namespace) {
-        return encryptionProviders.get(namespace);
+    public void unregisterEncryptionMethod(String namespace) {
+        encryptionMethods.remove(namespace);
     }
+
+    public JingleEncryptionMethod getEncryptionMethod(String namespace) {
+        return encryptionMethods.get(namespace);
+    }
+
 }

smack-experimental/src/main/java/org/jivesoftware/smackx/jet/OutgoingJetOffer.java

@@ -0,0 +1,27 @@
+package org.jivesoftware.smackx.jet;
+
+import java.io.File;
+
+import org.jivesoftware.smack.XMPPConnection;
+import org.jivesoftware.smackx.jingle_filetransfer.OutgoingJingleFileOffer;
+
+import org.jxmpp.jid.FullJid;
+
+/**
+ * Created by vanitas on 14.07.17.
+ */
+public class OutgoingJetOffer extends OutgoingJingleFileOffer {
+
+    public OutgoingJetOffer(XMPPConnection connection, FullJid responder, String sid) {
+        super(connection, responder, sid);
+    }
+
+    public OutgoingJetOffer(XMPPConnection connection, FullJid recipient) {
+        super(connection, recipient);
+    }
+
+    @Override
+    public void send(File file) {
+
+    }
+}

smack-experimental/src/main/java/org/jivesoftware/smackx/jingle_filetransfer/OutgoingJingleFileOffer.java

@@ -122,7 +122,7 @@ public class OutgoingJingleFileOffer extends JingleFileTransferSession {
 
         state = State.pending;
 
-        Jingle initiate = jutil.createSessionInitiateFileOffer(getResponder(), getSessionId(), creator, name, file, transportSession.createTransport());
+        Jingle initiate = jutil.createSessionInitiateFileOffer(getResponder(), getSessionId(), creator, name, file, transportSession.createTransport(), null);
         this.contents.addAll(initiate.getContents());
 
         connection.sendStanza(initiate);