From ceba706df4847e160cafc8da5a9e2aba38a857ac Mon Sep 17 00:00:00 2001
From: Christian Schneppe <christian@pix-art.de>
Date: Sun, 29 Jan 2017 20:08:29 +0100
Subject: use base64 encoding for file names uploaded with http

---
 .../de/pixart/messenger/generator/IqGenerator.java  | 21 ++++++++++++++++++++-
 1 file changed, 20 insertions(+), 1 deletion(-)

(limited to 'src/main/java/de/pixart/messenger/generator/IqGenerator.java')

diff --git a/src/main/java/de/pixart/messenger/generator/IqGenerator.java b/src/main/java/de/pixart/messenger/generator/IqGenerator.java
index e2b462f76..adcf2b04f 100644
--- a/src/main/java/de/pixart/messenger/generator/IqGenerator.java
+++ b/src/main/java/de/pixart/messenger/generator/IqGenerator.java
@@ -10,12 +10,14 @@ import org.whispersystems.libaxolotl.ecc.ECPublicKey;
 import org.whispersystems.libaxolotl.state.PreKeyRecord;
 import org.whispersystems.libaxolotl.state.SignedPreKeyRecord;
 
+import java.nio.ByteBuffer;
 import java.security.cert.CertificateEncodingException;
 import java.security.cert.X509Certificate;
 import java.util.ArrayList;
 import java.util.List;
 import java.util.Set;
 import java.util.TimeZone;
+import java.util.UUID;
 
 import de.pixart.messenger.Config;
 import de.pixart.messenger.R;
@@ -316,7 +318,7 @@ public class IqGenerator extends AbstractGenerator {
         IqPacket packet = new IqPacket(IqPacket.TYPE.GET);
         packet.setTo(host);
         Element request = packet.addChild("request", Xmlns.HTTP_UPLOAD);
-        request.addChild("filename").setContent(file.getName());
+        request.addChild("filename").setContent(convertFilename(file.getName()));
         request.addChild("size").setContent(String.valueOf(file.getExpectedSize()));
         if (mime != null) {
             request.addChild("content-type").setContent(mime);
@@ -324,6 +326,23 @@ public class IqGenerator extends AbstractGenerator {
         return packet;
     }
 
+    private static String convertFilename(String name) {
+        int pos = name.indexOf('.');
+        if (pos != -1) {
+            try {
+                UUID uuid = UUID.fromString(name.substring(0, pos));
+                ByteBuffer bb = ByteBuffer.wrap(new byte[16]);
+                bb.putLong(uuid.getMostSignificantBits());
+                bb.putLong(uuid.getLeastSignificantBits());
+                return Base64.encodeToString(bb.array(), Base64.URL_SAFE) + name.substring(pos, name.length());
+            } catch (Exception e) {
+                return name;
+            }
+        } else {
+            return name;
+        }
+    }
+
     public IqPacket generateCreateAccountWithCaptcha(Account account, String id, Data data) {
         final IqPacket register = new IqPacket(IqPacket.TYPE.SET);
         register.setFrom(account.getJid().toBareJid());
-- 
cgit v1.2.3