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