ensure downloaded file does not exceed Content-Length reported by HEAD

This commit is contained in:
Daniel Gultsch 2022-03-30 09:03:19 +02:00
parent 93c5916346
commit 7e762eb799
2 changed files with 21 additions and 3 deletions

View file

@ -102,11 +102,15 @@ public class HttpDownloadConnection implements Transferable {
if (this.message.getEncryption() == Message.ENCRYPTION_AXOLOTL && this.file.getKey() == null) { if (this.message.getEncryption() == Message.ENCRYPTION_AXOLOTL && this.file.getKey() == null) {
this.message.setEncryption(Message.ENCRYPTION_NONE); this.message.setEncryption(Message.ENCRYPTION_NONE);
} }
//TODO add auth tag size to knownFileSize
final Long knownFileSize = message.getFileParams().size; final Long knownFileSize = message.getFileParams().size;
Log.d(Config.LOGTAG,"knownFileSize: "+knownFileSize+", body="+message.getBody()); Log.d(Config.LOGTAG,"knownFileSize: "+knownFileSize+", body="+message.getBody());
if (knownFileSize != null && interactive) { if (knownFileSize != null && interactive) {
if (message.getEncryption() == Message.ENCRYPTION_AXOLOTL
&& this.file.getKey() != null) {
this.file.setExpectedSize(knownFileSize + 16);
} else {
this.file.setExpectedSize(knownFileSize); this.file.setExpectedSize(knownFileSize);
}
download(true); download(true);
} else { } else {
checkFileSize(interactive); checkFileSize(interactive);
@ -216,6 +220,8 @@ public class HttpDownloadConnection implements Transferable {
mXmppConnectionService.showErrorToastInUi(R.string.download_failed_could_not_connect); mXmppConnectionService.showErrorToastInUi(R.string.download_failed_could_not_connect);
} else if (e instanceof FileWriterException) { } else if (e instanceof FileWriterException) {
mXmppConnectionService.showErrorToastInUi(R.string.download_failed_could_not_write_file); mXmppConnectionService.showErrorToastInUi(R.string.download_failed_could_not_write_file);
} else if (e instanceof InvalidFileException) {
mXmppConnectionService.showErrorToastInUi(R.string.download_failed_invalid_file);
} else { } else {
mXmppConnectionService.showErrorToastInUi(R.string.download_failed_file_not_found); mXmppConnectionService.showErrorToastInUi(R.string.download_failed_file_not_found);
} }
@ -428,9 +434,12 @@ public class HttpDownloadConnection implements Transferable {
transmitted += count; transmitted += count;
try { try {
outputStream.write(buffer, 0, count); outputStream.write(buffer, 0, count);
} catch (IOException e) { } catch (final IOException e) {
throw new FileWriterException(file); throw new FileWriterException(file);
} }
if (transmitted > expected) {
throw new InvalidFileException(String.format("File exceeds expected size of %d", expected));
}
updateProgress(Math.round(((double) transmitted / expected) * 100)); updateProgress(Math.round(((double) transmitted / expected) * 100));
} }
outputStream.flush(); outputStream.flush();
@ -458,4 +467,12 @@ public class HttpDownloadConnection implements Transferable {
throw new IOException(String.format(Locale.ENGLISH, "HTTP Status code was %d", code)); throw new IOException(String.format(Locale.ENGLISH, "HTTP Status code was %d", code));
} }
} }
private static class InvalidFileException extends IOException {
private InvalidFileException(final String message) {
super(message);
}
}
} }

View file

@ -463,6 +463,7 @@
<string name="download_failed_file_not_found">Download failed: File not found</string> <string name="download_failed_file_not_found">Download failed: File not found</string>
<string name="download_failed_could_not_connect">Download failed: Could not connect to host</string> <string name="download_failed_could_not_connect">Download failed: Could not connect to host</string>
<string name="download_failed_could_not_write_file">Download failed: Could not write file</string> <string name="download_failed_could_not_write_file">Download failed: Could not write file</string>
<string name="download_failed_invalid_file">Download failed: Invalid file</string>
<string name="account_status_tor_unavailable">Tor network unavailable</string> <string name="account_status_tor_unavailable">Tor network unavailable</string>
<string name="account_status_bind_failure">Bind failure</string> <string name="account_status_bind_failure">Bind failure</string>
<string name="account_status_host_unknown">The server is not responsible for this domain</string> <string name="account_status_host_unknown">The server is not responsible for this domain</string>