Changeset 25053

Timestamp:

2012-02-07T13:54:40+13:00 (12 years ago)

Author:

sjm84

Message:

You can now specify a downloadFile argument in the URL to force the web browser to download a file, even if it is an image. Will need to add some form of security for this

File:

• main/trunk/greenstone3/src/java/org/greenstone/gsdl3/LibraryServlet.java (modified) (3 diffs)

main/trunk/greenstone3/src/java/org/greenstone/gsdl3/LibraryServlet.java

@@ -438 +438 @@
                         }
 
-                        
                         FileChannel source = null;
                         FileChannel destination = null;
@@ -484 +483 @@
             String href = null;
             String rl = null;
-            String[] nameval = new String[2]; // Reuse it for memory efficiency purposes    
 
             for (int i = 0; i < query_arr.length; i++)
@@ -506 +504 @@
                 {
                     rl = query_arr[i].substring(query_arr[i].indexOf("=") + 1, query_arr[i].length());
+                }
+            }
+
+            for (String arg : query_arr)
+            {
+                if (arg.startsWith("downloadFile"))
+                {
+                    int index = arg.indexOf("=");
+                    if (index > -1 && index < arg.length() - 1)
+                    {
+                        String fileLocation = arg.substring(index + 1);
+                        File fileToGet = new File(GlobalProperties.getGSDL3Home() + File.separator + fileLocation);
+
+                        if (fileToGet.exists())
+                        {
+                            response.setContentType("application/octet-stream");
+                            response.addHeader("Content-Disposition","attachment;filename=" + fileToGet.getName());
+                            FileInputStream fis = new FileInputStream(fileToGet);
+                            ServletOutputStream sos = response.getOutputStream();
+
+                            byte[] buffer = new byte[4096];
+                            int len;
+                            while ((len = fis.read(buffer)) != -1)
+                            {
+                                sos.write(buffer, 0, len);
+                            }
+                            sos.flush();
+                            fis.close();
+                            sos.close();
+                            
+                            return;
+                        }
+                    }
                 }
             }