1package eu.siacs.conversations.http;
2
3import android.app.PendingIntent;
4import android.content.Intent;
5import android.net.Uri;
6import android.os.PowerManager;
7import android.util.Log;
8import android.util.Pair;
9
10import java.io.FileNotFoundException;
11import java.io.IOException;
12import java.io.InputStream;
13import java.io.OutputStream;
14import java.net.HttpURLConnection;
15import java.net.InetAddress;
16import java.net.InetSocketAddress;
17import java.net.MalformedURLException;
18import java.net.Proxy;
19import java.net.URL;
20
21import javax.net.ssl.HttpsURLConnection;
22
23import eu.siacs.conversations.Config;
24import eu.siacs.conversations.entities.Account;
25import eu.siacs.conversations.entities.DownloadableFile;
26import eu.siacs.conversations.entities.Message;
27import eu.siacs.conversations.entities.Transferable;
28import eu.siacs.conversations.persistance.FileBackend;
29import eu.siacs.conversations.services.AbstractConnectionManager;
30import eu.siacs.conversations.services.XmppConnectionService;
31import eu.siacs.conversations.ui.UiCallback;
32import eu.siacs.conversations.utils.CryptoHelper;
33import eu.siacs.conversations.utils.Xmlns;
34import eu.siacs.conversations.xml.Element;
35import eu.siacs.conversations.xmpp.OnIqPacketReceived;
36import eu.siacs.conversations.xmpp.jid.Jid;
37import eu.siacs.conversations.xmpp.stanzas.IqPacket;
38
39public class HttpUploadConnection implements Transferable {
40
41 private HttpConnectionManager mHttpConnectionManager;
42 private XmppConnectionService mXmppConnectionService;
43
44 private boolean canceled = false;
45 private boolean delayed = false;
46 private Account account;
47 private DownloadableFile file;
48 private Message message;
49 private String mime;
50 private URL mGetUrl;
51 private URL mPutUrl;
52 private boolean mUseTor = false;
53
54 private byte[] key = null;
55
56 private long transmitted = 0;
57
58 private InputStream mFileInputStream;
59
60 public HttpUploadConnection(HttpConnectionManager httpConnectionManager) {
61 this.mHttpConnectionManager = httpConnectionManager;
62 this.mXmppConnectionService = httpConnectionManager.getXmppConnectionService();
63 this.mUseTor = mXmppConnectionService.useTorToConnect();
64 }
65
66 @Override
67 public boolean start() {
68 return false;
69 }
70
71 @Override
72 public int getStatus() {
73 return STATUS_UPLOADING;
74 }
75
76 @Override
77 public long getFileSize() {
78 return file == null ? 0 : file.getExpectedSize();
79 }
80
81 @Override
82 public int getProgress() {
83 if (file == null) {
84 return 0;
85 }
86 return (int) ((((double) transmitted) / file.getExpectedSize()) * 100);
87 }
88
89 @Override
90 public void cancel() {
91 this.canceled = true;
92 }
93
94 private void fail() {
95 mHttpConnectionManager.finishUploadConnection(this);
96 message.setTransferable(null);
97 mXmppConnectionService.markMessage(message, Message.STATUS_SEND_FAILED);
98 FileBackend.close(mFileInputStream);
99 }
100
101 public void init(Message message, boolean delay) {
102 this.message = message;
103 this.account = message.getConversation().getAccount();
104 this.file = mXmppConnectionService.getFileBackend().getFile(message, false);
105 this.mime = this.file.getMimeType();
106 this.delayed = delay;
107 if (Config.ENCRYPT_ON_HTTP_UPLOADED
108 || message.getEncryption() == Message.ENCRYPTION_AXOLOTL
109 || message.getEncryption() == Message.ENCRYPTION_OTR) {
110 this.key = new byte[48];
111 mXmppConnectionService.getRNG().nextBytes(this.key);
112 this.file.setKeyAndIv(this.key);
113 }
114 Pair<InputStream,Integer> pair;
115 try {
116 pair = AbstractConnectionManager.createInputStream(file, true);
117 } catch (FileNotFoundException e) {
118 fail();
119 return;
120 }
121 this.file.setExpectedSize(pair.second);
122 this.mFileInputStream = pair.first;
123 Jid host = account.getXmppConnection().findDiscoItemByFeature(Xmlns.HTTP_UPLOAD);
124 IqPacket request = mXmppConnectionService.getIqGenerator().requestHttpUploadSlot(host,file,mime);
125 mXmppConnectionService.sendIqPacket(account, request, new OnIqPacketReceived() {
126 @Override
127 public void onIqPacketReceived(Account account, IqPacket packet) {
128 if (packet.getType() == IqPacket.TYPE.RESULT) {
129 Element slot = packet.findChild("slot",Xmlns.HTTP_UPLOAD);
130 if (slot != null) {
131 try {
132 mGetUrl = new URL(slot.findChildContent("get"));
133 mPutUrl = new URL(slot.findChildContent("put"));
134 if (!canceled) {
135 new Thread(new FileUploader()).start();
136 }
137 } catch (MalformedURLException e) {
138 fail();
139 }
140 } else {
141 fail();
142 }
143 } else {
144 fail();
145 }
146 }
147 });
148 message.setTransferable(this);
149 mXmppConnectionService.markMessage(message, Message.STATUS_UNSEND);
150 }
151
152 private class FileUploader implements Runnable {
153
154 @Override
155 public void run() {
156 this.upload();
157 }
158
159 private void upload() {
160 OutputStream os = null;
161 HttpURLConnection connection = null;
162 PowerManager.WakeLock wakeLock = mHttpConnectionManager.createWakeLock("http_upload_"+message.getUuid());
163 try {
164 wakeLock.acquire();
165 Log.d(Config.LOGTAG, "uploading to " + mPutUrl.toString());
166 if (mUseTor) {
167 connection = (HttpURLConnection) mPutUrl.openConnection(mHttpConnectionManager.getProxy());
168 } else {
169 connection = (HttpURLConnection) mPutUrl.openConnection();
170 }
171 if (connection instanceof HttpsURLConnection) {
172 mHttpConnectionManager.setupTrustManager((HttpsURLConnection) connection, true);
173 }
174 connection.setRequestMethod("PUT");
175 connection.setFixedLengthStreamingMode((int) file.getExpectedSize());
176 connection.setRequestProperty("Content-Type", mime == null ? "application/octet-stream" : mime);
177 connection.setRequestProperty("User-Agent",mXmppConnectionService.getIqGenerator().getIdentityName());
178 connection.setDoOutput(true);
179 connection.connect();
180 os = connection.getOutputStream();
181 transmitted = 0;
182 int count = -1;
183 byte[] buffer = new byte[4096];
184 while (((count = mFileInputStream.read(buffer)) != -1) && !canceled) {
185 transmitted += count;
186 os.write(buffer, 0, count);
187 mXmppConnectionService.updateConversationUi();
188 }
189 os.flush();
190 os.close();
191 mFileInputStream.close();
192 int code = connection.getResponseCode();
193 if (code == 200 || code == 201) {
194 Log.d(Config.LOGTAG, "finished uploading file");
195 if (key != null) {
196 mGetUrl = new URL(mGetUrl.toString() + "#" + CryptoHelper.bytesToHex(key));
197 }
198 mXmppConnectionService.getFileBackend().updateFileParams(message, mGetUrl);
199 mXmppConnectionService.getFileBackend().addImageFileToMedia(file);
200 message.setTransferable(null);
201 message.setCounterpart(message.getConversation().getJid().toBareJid());
202 if (message.getEncryption() == Message.ENCRYPTION_DECRYPTED) {
203 mXmppConnectionService.getPgpEngine().encrypt(message, new UiCallback<Message>() {
204 @Override
205 public void success(Message message) {
206 mXmppConnectionService.resendMessage(message,delayed);
207 }
208
209 @Override
210 public void error(int errorCode, Message object) {
211 fail();
212 }
213
214 @Override
215 public void userInputRequried(PendingIntent pi, Message object) {
216 fail();
217 }
218 });
219 } else {
220 mXmppConnectionService.resendMessage(message, delayed);
221 }
222 } else {
223 fail();
224 }
225 } catch (IOException e) {
226 e.printStackTrace();
227 Log.d(Config.LOGTAG,"http upload failed "+e.getMessage());
228 fail();
229 } finally {
230 FileBackend.close(mFileInputStream);
231 FileBackend.close(os);
232 if (connection != null) {
233 connection.disconnect();
234 }
235 wakeLock.release();
236 }
237 }
238 }
239}