1 /*
   2  * Copyright (c) 2019, 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 import java.nio.ByteBuffer;
  25 import java.util.Base64;
  26 
  27 /**
  28  * @test
  29  * @bug 8210583
  30  * @summary Tests Base64.Encoder.encode/Decoder.decode for the large size
  31  *          of resulting bytes which can not be allocated
  32  * @requires os.maxMemory >= 6g
  33  * @run main/othervm -Xms4g -Xmx6g TestEncodingDecodingLength
  34  *
  35  */
  36 
  37 public class TestEncodingDecodingLength {
  38 
  39     public static void main(String[] args) {
  40         int size = Integer.MAX_VALUE - 2;
  41         byte[] inputBytes = new byte[size];
  42         byte[] outputBytes = new byte[size];
  43 
  44         // Check encoder with large array length
  45         Base64.Encoder encoder = Base64.getEncoder();
  46         checkOOM("encode(byte[])", () -> encoder.encode(inputBytes));
  47         checkIAE("encode(byte[] byte[])", () -> encoder.encode(inputBytes, outputBytes));
  48         checkOOM("encodeToString(byte[])", () -> encoder.encodeToString(inputBytes));
  49         checkOOM("encode(ByteBuffer)", () -> encoder.encode(ByteBuffer.allocate(size)));
  50 
  51         // Check decoder with large array length
  52         Base64.Decoder decoder = Base64.getDecoder();
  53         checkOOM("decode(byte[])", () -> decoder.decode(inputBytes));
  54         checkIAE("decode(byte[], byte[])", () -> decoder.decode(inputBytes, outputBytes));
  55         checkOOM("decode(ByteBuffer)", () -> decoder.decode(ByteBuffer.allocate(size)));
  56     }
  57 
  58     private static final void checkOOM(String methodName, Runnable r) {
  59         try {
  60             r.run();
  61             throw new RuntimeException("OutOfMemoryError should have been thrown by" + methodName);
  62         } catch (OutOfMemoryError er) {}
  63     }
  64 
  65     private static final void checkIAE(String methodName, Runnable r) {
  66         try {
  67             r.run();
  68             throw new RuntimeException("IllegalArgumentException should have been thrown by" + methodName);
  69         } catch (IllegalArgumentException iae) {}
  70     }
  71 }
  72