]> gitweb.stoutner.com Git - PrivacyBrowserAndroid.git/commitdiff
More intelligent parsing of the Content-Disposition header to prevent crashing while...
authorSoren Stoutner <soren@stoutner.com>
Wed, 2 Nov 2016 07:30:21 +0000 (00:30 -0700)
committerSoren Stoutner <soren@stoutner.com>
Wed, 2 Nov 2016 07:30:21 +0000 (00:30 -0700)
app/src/main/java/com/stoutner/privacybrowser/DownloadFile.java

index 1a3b76dae267abaaf7b1deb034e18c4c19aeca66..8e04cf64f29bfa279a96343fc3f929f67415ec26 100644 (file)
@@ -50,9 +50,15 @@ public class DownloadFile extends AppCompatDialogFragment {
         Bundle argumentsBundle = new Bundle();
 
         String fileNameString;
-        if (!contentDisposition.isEmpty()) {  // Extract `fileNameString` from `contentDisposition` using the substring beginning after `filename="` and ending with the next `"`.
+
+        // Parse `filename` from `contentDisposition`.
+        if (contentDisposition.contains("filename=\"")) {  // The file name is contained in a string surrounded by `""`.
             fileNameString = contentDisposition.substring(contentDisposition.indexOf("filename=\"") + 10, contentDisposition.indexOf("\"", contentDisposition.indexOf("filename=\"") + 10));
-        } else {  // `contentDisposition` is empty, so use the last path segment of the URL as the file name.
+        } else if (contentDisposition.contains("filename=") && ((contentDisposition.indexOf(";", contentDisposition.indexOf("filename=") + 9)) > 0 )) {  // The file name is contained in a string beginning with `filename=` and ending with `;`.
+            fileNameString = contentDisposition.substring(contentDisposition.indexOf("filename=") + 9, contentDisposition.indexOf(";", contentDisposition.indexOf("filename=") + 9));
+        } else if (contentDisposition.contains("filename=")) {  // The file name is contained in a string beginning with `filename=` and proceeding to the end of `contentDisposition`.
+            fileNameString = contentDisposition.substring(contentDisposition.indexOf("filename=") + 9, contentDisposition.length());
+        } else {  // `contentDisposition` does not contain the filename, so use the last path segment of the URL.
             Uri downloadUri = Uri.parse(urlString);
             fileNameString = downloadUri.getLastPathSegment();
         }