aboutsummaryrefslogtreecommitdiffstats
path: root/src/eu/siacs/conversations/http/HttpConnection.java
blob: 003007d8508f060991283861a572cb2bdec42b8a (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
package eu.siacs.conversations.http;

import java.io.BufferedInputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.net.HttpURLConnection;
import java.net.MalformedURLException;
import java.net.URL;

import javax.net.ssl.HttpsURLConnection;

import android.util.Log;

import eu.siacs.conversations.Config;
import eu.siacs.conversations.entities.Downloadable;
import eu.siacs.conversations.entities.DownloadableFile;
import eu.siacs.conversations.entities.Message;
import eu.siacs.conversations.services.XmppConnectionService;

public class HttpConnection implements Downloadable {

	private HttpConnectionManager mHttpConnectionManager;
	private XmppConnectionService mXmppConnectionService;

	private URL mUrl;
	private Message message;
	private DownloadableFile file;

	public HttpConnection(HttpConnectionManager manager) {
		this.mHttpConnectionManager = manager;
		this.mXmppConnectionService = manager.getXmppConnectionService();
	}

	@Override
	public void start() {
		new Thread(new FileDownloader()).start();
	}

	public void init(Message message) {
		this.message = message;
		this.message.setDownloadable(this);
		try {
			mUrl = new URL(message.getBody());
			this.file = mXmppConnectionService.getFileBackend().getConversationsFile(message,false);
			message.setType(Message.TYPE_IMAGE);
			mXmppConnectionService.markMessage(message, Message.STATUS_RECEIVED_OFFER);
			checkFileSize();
		} catch (MalformedURLException e) {
			this.cancel();
		}
	}
	
	private void checkFileSize() {
		new Thread(new FileSizeChecker()).start();
	}

	public void cancel() {
		mXmppConnectionService.markMessage(message, Message.STATUS_RECEPTION_FAILED);
		Log.d(Config.LOGTAG,"canceled download");
	}

	private class FileSizeChecker implements Runnable {

		@Override
		public void run() {
			try {
				long size = retrieveFileSize();
				file.setExpectedSize(size);
				if (size <= mHttpConnectionManager.getAutoAcceptFileSize()) {
					start();
				}
				Log.d(Config.LOGTAG,"file size: "+size);
			} catch (IOException e) {
				cancel();
			}
		}

		private long retrieveFileSize() throws IOException {
			HttpURLConnection connection = (HttpURLConnection) mUrl.openConnection();
			connection.setRequestMethod("HEAD");
			if (connection instanceof HttpsURLConnection) {
				
			}
			String contentLength = connection.getHeaderField("Content-Length");
			if (contentLength == null) {
				throw new IOException();
			}
			try {
				return Long.parseLong(contentLength, 10);
			} catch (NumberFormatException e) {
				throw new IOException();
			}
		}

	}
	
	private class FileDownloader implements Runnable {

		@Override
		public void run() {
			try {
				mXmppConnectionService.markMessage(message, Message.STATUS_RECEIVING);
				download();
				mXmppConnectionService.markMessage(message, Message.STATUS_RECEIVED);
			} catch (IOException e) {
				cancel();
			}
		}
		
		private void download() throws IOException {
			HttpURLConnection connection = (HttpURLConnection) mUrl.openConnection();
			if (connection instanceof HttpsURLConnection) {
				
			}
			BufferedInputStream is = new BufferedInputStream(connection.getInputStream());
			OutputStream os = file.createOutputStream();
			int count = -1;
			byte[] buffer = new byte[1024];
			while ((count = is.read(buffer)) != -1) {
				os.write(buffer, 0, count);
			}
			os.flush();
			os.close();
			is.close();
			Log.d(Config.LOGTAG,"finished downloading "+file.getAbsolutePath().toString());
		}
		
	}
}