1 /*
   2  * Copyright (c) 1999, 2007, Oracle and/or its affiliates. All rights reserved.
   3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
   4  *
   5  * This code is free software; you can redistribute it and/or modify it
   6  * under the terms of the GNU General Public License version 2 only, as
   7  * published by the Free Software Foundation.
   8  *
   9  * This code is distributed in the hope that it will be useful, but WITHOUT
  10  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  11  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  12  * version 2 for more details (a copy is included in the LICENSE file that
  13  * accompanied this code).
  14  *
  15  * You should have received a copy of the GNU General Public License version
  16  * 2 along with this work; if not, write to the Free Software Foundation,
  17  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
  18  *
  19  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
  20  * or visit www.oracle.com if you need additional information or have any
  21  * questions.
  22  */
  23 
  24 /* @test
  25    @bug 4166799
  26    @summary Make sure URL-downloaded jar files (jar_cache files)
  27             will be deleted when VM exits.
  28 
  29    @build DeleteTempJar
  30    @run shell deletetempjar.sh
  31  */
  32 
  33 import java.io.File;
  34 import java.io.FileInputStream;
  35 import java.io.FileOutputStream;
  36 import java.io.OutputStream;
  37 
  38 import java.net.InetAddress;
  39 import java.net.InetSocketAddress;
  40 import java.net.JarURLConnection;
  41 import java.net.URL;
  42 
  43 import java.util.jar.JarEntry;
  44 import java.util.jar.JarFile;
  45 import java.util.jar.JarOutputStream;
  46 
  47 import com.sun.net.httpserver.*;
  48 
  49 
  50 public class DeleteTempJar
  51 {
  52     public static void realMain(String args[]) throws Exception
  53     {
  54         final File zf = File.createTempFile("deletetemp", ".jar");
  55         zf.deleteOnExit();
  56         try (FileOutputStream fos = new FileOutputStream(zf);
  57              JarOutputStream jos = new JarOutputStream(fos))
  58         {
  59             JarEntry je = new JarEntry("entry");
  60             jos.putNextEntry(je);
  61             jos.write("hello, world".getBytes("ASCII"));
  62         }
  63 
  64         HttpServer server = HttpServer.create(
  65                 new InetSocketAddress((InetAddress) null, 0), 0);
  66         HttpContext context = server.createContext("/",
  67             new HttpHandler() {
  68                 public void handle(HttpExchange e) {
  69                     try (FileInputStream fis = new FileInputStream(zf)) {
  70                         e.sendResponseHeaders(200, zf.length());
  71                         OutputStream os = e.getResponseBody();
  72                         byte[] buf = new byte[1024];
  73                         int count = 0;
  74                         while ((count = fis.read(buf)) != -1) {
  75                             os.write(buf, 0, count);
  76                         }
  77                     } catch (Exception ex) {
  78                         unexpected(ex);
  79                     } finally {
  80                         e.close();
  81                     }
  82                 }
  83             });
  84         server.start();
  85 
  86         URL url = new URL("jar:http://localhost:"
  87                           + new Integer(server.getAddress().getPort()).toString()
  88                           + "/deletetemp.jar!/");
  89         JarURLConnection c = (JarURLConnection)url.openConnection();
  90         JarFile f = c.getJarFile();
  91         check(f.getEntry("entry") != null);
  92         System.out.println(f.getName());
  93         server.stop(0);
  94     }
  95 
  96     //--------------------- Infrastructure ---------------------------
  97     static volatile int passed = 0, failed = 0;
  98     static boolean pass() {passed++; return true;}
  99     static boolean fail() {failed++; Thread.dumpStack(); return false;}
 100     static boolean fail(String msg) {System.out.println(msg); return fail();}
 101     static void unexpected(Throwable t) {failed++; t.printStackTrace();}
 102     static boolean check(boolean cond) {if (cond) pass(); else fail(); return cond;}
 103     static boolean equal(Object x, Object y) {
 104         if (x == null ? y == null : x.equals(y)) return pass();
 105         else return fail(x + " not equal to " + y);}
 106     public static void main(String[] args) throws Throwable {
 107         try {realMain(args);} catch (Throwable t) {unexpected(t);}
 108         System.out.println("\nPassed = " + passed + " failed = " + failed);
 109         if (failed > 0) throw new AssertionError("Some tests failed");}
 110 }