narayana/anotherim
narayana/anotherim
3
2
Fork 0
Code Issues 3 Pull requests Actions Packages Projects Releases 4 Wiki Activity

execute status code check on HEAD

Browse source
This commit is contained in:
Daniel Gultsch 2021-05-16 16:17:06 +02:00
parent 0f3181555a
commit b025265f91
1 changed files with 46 additions and 38 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

84
src/main/java/eu/siacs/conversations/http/HttpDownloadConnection.java
View file

@ -318,6 +318,7 @@ public class HttpDownloadConnection implements Transferable {
mostRecentCall = client.newCall(request); mostRecentCall = client.newCall(request);
try { try {
final Response response = mostRecentCall.execute(); final Response response = mostRecentCall.execute();
throwOnInvalidCode(response);
final String contentLength = response.header("Content-Length"); final String contentLength = response.header("Content-Length");
final String contentType = response.header("Content-Type"); final String contentType = response.header("Content-Type");
final AbstractConnectionManager.Extension extension = AbstractConnectionManager.Extension.of(mUrl.encodedPath()); final AbstractConnectionManager.Extension extension = AbstractConnectionManager.Extension.of(mUrl.encodedPath());
@ -332,7 +333,11 @@ public class HttpDownloadConnection implements Transferable {
if (Strings.isNullOrEmpty(contentLength)) { if (Strings.isNullOrEmpty(contentLength)) {
throw new IOException("no content-length found in HEAD response"); throw new IOException("no content-length found in HEAD response");
} }
return Long.parseLong(contentLength, 10); final long size = Long.parseLong(contentLength, 10);
if (size < 0) {
throw new IOException("Server reported negative file size");
}
return size;
} catch (IOException e) { } catch (IOException e) {
Log.d(Config.LOGTAG, "io exception during HEAD " + e.getMessage()); Log.d(Config.LOGTAG, "io exception during HEAD " + e.getMessage());
throw e; throw e;
@ -395,45 +400,41 @@ public class HttpDownloadConnection implements Transferable {
final Request request = requestBuilder.build(); final Request request = requestBuilder.build();
mostRecentCall = client.newCall(request); mostRecentCall = client.newCall(request);
final Response response = mostRecentCall.execute(); final Response response = mostRecentCall.execute();
final int code = response.code(); throwOnInvalidCode(response);
if (code >= 200 && code <= 299) { final String contentRange = response.header("Content-Range");
final String contentRange = response.header("Content-Range"); final boolean serverResumed = tryResume && contentRange != null && contentRange.startsWith("bytes " + resumeSize + "-");
final boolean serverResumed = tryResume && contentRange != null && contentRange.startsWith("bytes " + resumeSize + "-"); final InputStream inputStream = response.body().byteStream();
final InputStream inputStream = response.body().byteStream(); final OutputStream outputStream;
final OutputStream outputStream; long transmitted = 0;
long transmitted = 0; if (tryResume && serverResumed) {
if (tryResume && serverResumed) { Log.d(Config.LOGTAG, "server resumed");
Log.d(Config.LOGTAG, "server resumed"); transmitted = file.getSize();
transmitted = file.getSize(); updateProgress(Math.round(((double) transmitted / expected) * 100));
updateProgress(Math.round(((double) transmitted / expected) * 100)); outputStream = AbstractConnectionManager.createOutputStream(file, true, false);
outputStream = AbstractConnectionManager.createOutputStream(file, true, false);
} else {
final String contentLength = response.header("Content-Length");
final long size = Strings.isNullOrEmpty(contentLength) ? 0 : Longs.tryParse(contentLength);
if (expected != size) {
Log.d(Config.LOGTAG, "content-length reported on GET (" + size + ") did not match Content-Length reported on HEAD (" + expected + ")");
}
file.getParentFile().mkdirs();
if (!file.exists() && !file.createNewFile()) {
throw new FileWriterException();
}
outputStream = AbstractConnectionManager.createOutputStream(file, false, false);
}
int count;
final byte[] buffer = new byte[4096];
while ((count = inputStream.read(buffer)) != -1) {
transmitted += count;
try {
outputStream.write(buffer, 0, count);
} catch (IOException e) {
throw new FileWriterException();
}
updateProgress(Math.round(((double) transmitted / expected) * 100));
}
outputStream.flush();
} else { } else {
throw new IOException(String.format(Locale.ENGLISH, "HTTP Status code was %d", code)); final String contentLength = response.header("Content-Length");
final long size = Strings.isNullOrEmpty(contentLength) ? 0 : Longs.tryParse(contentLength);
if (expected != size) {
Log.d(Config.LOGTAG, "content-length reported on GET (" + size + ") did not match Content-Length reported on HEAD (" + expected + ")");
}
file.getParentFile().mkdirs();
if (!file.exists() && !file.createNewFile()) {
throw new FileWriterException();
}
outputStream = AbstractConnectionManager.createOutputStream(file, false, false);
} }
int count;
final byte[] buffer = new byte[4096];
while ((count = inputStream.read(buffer)) != -1) {
transmitted += count;
try {
outputStream.write(buffer, 0, count);
} catch (IOException e) {
throw new FileWriterException();
}
updateProgress(Math.round(((double) transmitted / expected) * 100));
}
outputStream.flush();
} }
private void updateImageBounds() { private void updateImageBounds() {
@ -451,4 +452,11 @@ public class HttpDownloadConnection implements Transferable {
} }
} }
private static void throwOnInvalidCode(final Response response) throws IOException {
final int code = response.code();
if (code < 200 || code >= 300) {
throw new IOException(String.format(Locale.ENGLISH, "HTTP Status code was %d", code));
}
}
} }