Reland of Add GN build rules to allow java_assertion_enabler to enable Java asserts.

build/android/java_assertion_enabler/java/org/chromium/javaassertionenabler/AssertionEnabler.java

 // Copyright 2016 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 package org.chromium.javaassertionenabler;
 
-import com.google.common.io.ByteStreams;
-
 import org.objectweb.asm.ClassReader;
 import org.objectweb.asm.ClassVisitor;
 import org.objectweb.asm.ClassWriter;
 import org.objectweb.asm.MethodVisitor;
 import org.objectweb.asm.Opcodes;
 
 import java.io.BufferedInputStream;
 import java.io.BufferedOutputStream;
+import java.io.ByteArrayOutputStream;
+import java.io.FileInputStream;
 import java.io.FileOutputStream;
 import java.io.IOException;
-import java.util.Collections;
-import java.util.jar.JarEntry;
-import java.util.jar.JarFile;
-import java.util.jar.JarOutputStream;
+import java.io.InputStream;
+import java.nio.file.Files;
+import java.nio.file.Path;
+import java.nio.file.Paths;
+import java.nio.file.StandardCopyOption;
+import java.util.zip.ZipEntry;
+import java.util.zip.ZipInputStream;
+import java.util.zip.ZipOutputStream;
 
 /**
  * An application that enables Java ASSERT statements by modifying Java bytecode. It takes in a JAR
  * file, modifies bytecode of classes that use ASSERT, and outputs the bytecode to a new JAR file.
  */
 class AssertionEnabler {
+    static final String ASSERTION_DISABLED_NAME = "$assertionsDisabled";
     static final String CLASS_FILE_SUFFIX = ".class";
     static final String STATIC_INITIALIZER_NAME = "<clinit>";
-    static final String ASSERTION_DISABLED_NAME = "$assertionsDisabled";
+    static final String TEMPORARY_FILE_SUFFIX = ".temp";
+
+    static final int BUFFER_SIZE = 16384;
 
     static class AssertionEnablerVisitor extends ClassVisitor {
         AssertionEnablerVisitor(ClassWriter writer) {
             super(Opcodes.ASM5, writer);
         }
 
         @Override
         public MethodVisitor visitMethod(final int access, final String name, String desc,
                 String signature, String[] exceptions) {
             // Patch static initializer.
@@ skipping 19 matching lines @@
                         } else {
                             super.visitFieldInsn(opcode, owner, name, desc);
                         }
                     }
                 };
             }
             return super.visitMethod(access, name, desc, signature, exceptions);
         }
     }
 
+    static byte[] readAllBytes(InputStream inputStream) throws IOException {
+        ByteArrayOutputStream buffer = new ByteArrayOutputStream();
+        int numRead = 0;
+        byte[] data = new byte[BUFFER_SIZE];
+        while ((numRead = inputStream.read(data, 0, data.length)) != -1) {
+            buffer.write(data, 0, numRead);
+        }
+
+        return buffer.toByteArray();
+    }
+
     static void enableAssertionInJar(String inputJarPath, String outputJarPath) {
-        try (JarOutputStream outputStream = new JarOutputStream(
-                new BufferedOutputStream(new FileOutputStream(outputJarPath)))) {
-            JarFile jarFile = new JarFile(inputJarPath);
-            for (JarEntry entry : Collections.list(jarFile.entries())) {
-                try (BufferedInputStream inputStream = new BufferedInputStream(
-                        jarFile.getInputStream(entry))) {
-                    byte[] byteCode = ByteStreams.toByteArray(inputStream);
+        String tempJarPath = outputJarPath + TEMPORARY_FILE_SUFFIX;
+        try (ZipInputStream inputStream = new ZipInputStream(
+                    new BufferedInputStream(new FileInputStream(inputJarPath)));
+             ZipOutputStream tempStream = new ZipOutputStream(
+                    new BufferedOutputStream(new FileOutputStream(tempJarPath)))) {
+            ZipEntry entry = null;
 
-                    if (entry.isDirectory() || !entry.getName().endsWith(CLASS_FILE_SUFFIX)) {
-                        outputStream.putNextEntry(entry);
-                        outputStream.write(byteCode);
-                        outputStream.closeEntry();
-                        continue;
-                    }
-                    ClassReader reader = new ClassReader(byteCode);
-                    ClassWriter writer = new ClassWriter(reader, 0);
-                    reader.accept(new AssertionEnablerVisitor(writer), 0);
-                    byte[] patchedByteCode = writer.toByteArray();
-                    outputStream.putNextEntry(new JarEntry(entry.getName()));
-                    outputStream.write(patchedByteCode);
-                    outputStream.closeEntry();
+            while ((entry = inputStream.getNextEntry()) != null) {
+                byte[] byteCode = readAllBytes(inputStream);
+
+                if (entry.isDirectory() || !entry.getName().endsWith(CLASS_FILE_SUFFIX)) {
+                    tempStream.putNextEntry(entry);
+                    tempStream.write(byteCode);
+                    tempStream.closeEntry();
+                    continue;
                 }
+                ClassReader reader = new ClassReader(byteCode);
+                ClassWriter writer = new ClassWriter(reader, 0);
+                reader.accept(new AssertionEnablerVisitor(writer), 0);
+                byte[] patchedByteCode = writer.toByteArray();
+                tempStream.putNextEntry(new ZipEntry(entry.getName()));
+                tempStream.write(patchedByteCode);
+                tempStream.closeEntry();
             }
         } catch (IOException e) {
             throw new RuntimeException(e);
         }
+
+        try {
+            Path src = Paths.get(tempJarPath);
+            Path dest = Paths.get(outputJarPath);
+            Files.move(src, dest, StandardCopyOption.REPLACE_EXISTING);
+        } catch (IOException ioException) {
+            throw new RuntimeException(ioException);
+        }
     }
 
     public static void main(String[] args) {
         if (args.length != 2) {
             System.out.println("Incorrect number of arguments.");
             System.out.println("Example usage: java_assertion_enabler input.jar output.jar");
             System.exit(-1);
         }
         enableAssertionInJar(args[0], args[1]);
     }
 }