1 /*
2  * Licensed to the Apache Software Foundation (ASF) under one or more
3  * contributor license agreements.  See the NOTICE file distributed with
4  * this work for additional information regarding copyright ownership.
5  * The ASF licenses this file to You under the Apache License, Version 2.0
6  * (the "License"); you may not use this file except in compliance with
7  * the License.  You may obtain a copy of the License at
8  *
9  *      http://www.apache.org/licenses/LICENSE-2.0
10  *
11  * Unless required by applicable law or agreed to in writing, software
12  * distributed under the License is distributed on an "AS IS" BASIS,
13  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14  * See the License for the specific language governing permissions and
15  * limitations under the License.
16  */
17 
18 package org.apache.commons.codec.binary;
19 
20 import java.math.BigInteger;
21 
22 import org.apache.commons.codec.BinaryDecoder;
23 import org.apache.commons.codec.BinaryEncoder;
24 import org.apache.commons.codec.DecoderException;
25 import org.apache.commons.codec.EncoderException;
26 
27 /**
28  * Provides Base64 encoding and decoding as defined by RFC 2045.
29  *
30  * <p>
31  * This class implements section <cite>6.8. Base64 Content-Transfer-Encoding</cite> from RFC 2045 <cite>Multipurpose
32  * Internet Mail Extensions (MIME) Part One: Format of Internet Message Bodies</cite> by Freed and Borenstein.
33  * </p>
34  * <p>
35  * The class can be parameterized in the following manner with various constructors:
36  * <ul>
37  * <li>URL-safe mode: Default off.</li>
38  * <li>Line length: Default 76. Line length that aren't multiples of 4 will still essentially end up being multiples of
39  * 4 in the encoded data.
40  * <li>Line separator: Default is CRLF ("\r\n")</li>
41  * </ul>
42  * </p>
43  * <p>
44  * Since this class operates directly on byte streams, and not character streams, it is hard-coded to only encode/decode
45  * character encodings which are compatible with the lower 127 ASCII chart (ISO-8859-1, Windows-1252, UTF-8, etc).
46  * </p>
47  *
48  * @see <a href="http://www.ietf.org/rfc/rfc2045.txt">RFC 2045</a>
49  * @since 1.0
50  * @version $Id: Base64.java 801706 2009-08-06 16:27:06Z niallp $
51  */
52 public class Base64Codec implements BinaryEncoder, BinaryDecoder {
53     private static final int DEFAULT_BUFFER_RESIZE_FACTOR = 2;
54 
55     private static final int DEFAULT_BUFFER_SIZE = 8192;
56 
57     /**
58      * Chunk size per RFC 2045 section 6.8.
59      *
60      * <p>
61      * The {@value} character limit does not count the trailing CRLF, but counts all other characters, including any
62      * equal signs.
63      * </p>
64      *
65      * @see <a href="http://www.ietf.org/rfc/rfc2045.txt">RFC 2045 section 6.8</a>
66      */
67     static final int CHUNK_SIZE = 76;
68 
69     /**
70      * Chunk separator per RFC 2045 section 2.1.
71      *
72      * <p>
73      * N.B. The next major release may break compatibility and make this field private.
74      * </p>
75      *
76      * @see <a href="http://www.ietf.org/rfc/rfc2045.txt">RFC 2045 section 2.1</a>
77      */
78     static final byte[] CHUNK_SEPARATOR = {'\r', '\n'};
79 
80     /**
81      * This array is a lookup table that translates 6-bit positive integer index values into their "Base64 Alphabet"
82      * equivalents as specified in Table 1 of RFC 2045.
83      *
84      * Thanks to "commons" project in ws.apache.org for this code.
85      * http://svn.apache.org/repos/asf/webservices/commons/trunk/modules/util/
86      */
87     private static final byte[] STANDARD_ENCODE_TABLE = {
88             'A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'I', 'J', 'K', 'L', 'M',
89             'N', 'O', 'P', 'Q', 'R', 'S', 'T', 'U', 'V', 'W', 'X', 'Y', 'Z',
90             'a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'j', 'k', 'l', 'm',
91             'n', 'o', 'p', 'q', 'r', 's', 't', 'u', 'v', 'w', 'x', 'y', 'z',
92             '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '+', '/'
93     };
94 
95     /**
96      * This is a copy of the STANDARD_ENCODE_TABLE above, but with + and /
97      * changed to - and _ to make the encoded Base64 results more URL-SAFE.
98      * This table is only used when the Base64's mode is set to URL-SAFE.
99      */
100     private static final byte[] URL_SAFE_ENCODE_TABLE = {
101             'A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'I', 'J', 'K', 'L', 'M',
102             'N', 'O', 'P', 'Q', 'R', 'S', 'T', 'U', 'V', 'W', 'X', 'Y', 'Z',
103             'a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'j', 'k', 'l', 'm',
104             'n', 'o', 'p', 'q', 'r', 's', 't', 'u', 'v', 'w', 'x', 'y', 'z',
105             '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '-', '_'
106     };
107 
108     /**
109      * Byte used to pad output.
110      */
111     private static final byte PAD = '=';
112 
113     /**
114      * This array is a lookup table that translates Unicode characters drawn from the "Base64 Alphabet" (as specified in
115      * Table 1 of RFC 2045) into their 6-bit positive integer equivalents. Characters that are not in the Base64
116      * alphabet but fall within the bounds of the array are translated to -1.
117      *
118      * Note: '+' and '-' both decode to 62. '/' and '_' both decode to 63. This means decoder seamlessly handles both
119      * URL_SAFE and STANDARD base64. (The encoder, on the other hand, needs to know ahead of time what to emit).
120      *
121      * Thanks to "commons" project in ws.apache.org for this code.
122      * http://svn.apache.org/repos/asf/webservices/commons/trunk/modules/util/
123      */
124     private static final byte[] DECODE_TABLE = {
125             -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
126             -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
127             -1, -1, -1, -1, -1, -1, -1, -1, -1, 62, -1, 62, -1, 63, 52, 53, 54,
128             55, 56, 57, 58, 59, 60, 61, -1, -1, -1, -1, -1, -1, -1, 0, 1, 2, 3, 4,
129             5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23,
130             24, 25, -1, -1, -1, -1, 63, -1, 26, 27, 28, 29, 30, 31, 32, 33, 34,
131             35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51
132     };
133 
134     /** Mask used to extract 6 bits, used when encoding */
135     private static final int MASK_6BITS = 0x3f;
136 
137     /** Mask used to extract 8 bits, used in decoding base64 bytes */
138     private static final int MASK_8BITS = 0xff;
139 
140     // The static final fields above are used for the original static byte[] methods on Base64.
141     // The private member fields below are used with the new streaming approach, which requires
142     // some state be preserved between calls of encode() and decode().
143 
144     /**
145      * Encode table to use: either STANDARD or URL_SAFE. Note: the DECODE_TABLE above remains static because it is able
146      * to decode both STANDARD and URL_SAFE streams, but the encodeTable must be a member variable so we can switch
147      * between the two modes.
148      */
149     private final byte[] encodeTable;
150 
151     /**
152      * Line length for encoding. Not used when decoding. A value of zero or less implies no chunking of the base64
153      * encoded data.
154      */
155     private final int lineLength;
156 
157     /**
158      * Line separator for encoding. Not used when decoding. Only used if lineLength > 0.
159      */
160     private final byte[] lineSeparator;
161 
162     /**
163      * Convenience variable to help us determine when our buffer is going to run out of room and needs resizing.
164      * <code>decodeSize = 3 + lineSeparator.length;</code>
165      */
166     private final int decodeSize;
167 
168     /**
169      * Convenience variable to help us determine when our buffer is going to run out of room and needs resizing.
170      * <code>encodeSize = 4 + lineSeparator.length;</code>
171      */
172     private final int encodeSize;
173 
174     /**
175      * Buffer for streaming.
176      */
177     private byte[] buffer;
178 
179     /**
180      * Position where next character should be written in the buffer.
181      */
182     private int pos;
183 
184     /**
185      * Position where next character should be read from the buffer.
186      */
187     private int readPos;
188 
189     /**
190      * Variable tracks how many characters have been written to the current line. Only used when encoding. We use it to
191      * make sure each encoded line never goes beyond lineLength (if lineLength > 0).
192      */
193     private int currentLinePos;
194 
195     /**
196      * Writes to the buffer only occur after every 3 reads when encoding, an every 4 reads when decoding. This variable
197      * helps track that.
198      */
199     private int modulus;
200 
201     /**
202      * Boolean flag to indicate the EOF has been reached. Once EOF has been reached, this Base64 object becomes useless,
203      * and must be thrown away.
204      */
205     private boolean eof;
206 
207     /**
208      * Place holder for the 3 bytes we're dealing with for our base64 logic. Bitwise operations store and extract the
209      * base64 encoding or decoding from this variable.
210      */
211     private int x;
212 
213     /**
214      * Creates a Base64 codec used for decoding (all modes) and encoding in URL-unsafe mode.
215      * <p>
216      * When encoding the line length is 76, the line separator is CRLF, and the encoding table is STANDARD_ENCODE_TABLE.
217      * </p>
218      *
219      * <p>
220      * When decoding all variants are supported.
221      * </p>
222      */
Base64Codec()223     public Base64Codec() {
224         this(false);
225     }
226 
227     /**
228      * Creates a Base64 codec used for decoding (all modes) and encoding in the given URL-safe mode.
229      * <p>
230      * When encoding the line length is 76, the line separator is CRLF, and the encoding table is STANDARD_ENCODE_TABLE.
231      * </p>
232      *
233      * <p>
234      * When decoding all variants are supported.
235      * </p>
236      *
237      * @param urlSafe
238      *            if <code>true</code>, URL-safe encoding is used. In most cases this should be set to
239      *            <code>false</code>.
240      * @since 1.4
241      */
Base64Codec(boolean urlSafe)242     public Base64Codec(boolean urlSafe) {
243         this(CHUNK_SIZE, CHUNK_SEPARATOR, urlSafe);
244     }
245 
246     /**
247      * Creates a Base64 codec used for decoding (all modes) and encoding in URL-unsafe mode.
248      * <p>
249      * When encoding the line length is given in the constructor, the line separator is CRLF, and the encoding table is
250      * STANDARD_ENCODE_TABLE.
251      * </p>
252      * <p>
253      * Line lengths that aren't multiples of 4 will still essentially end up being multiples of 4 in the encoded data.
254      * </p>
255      * <p>
256      * When decoding all variants are supported.
257      * </p>
258      *
259      * @param lineLength
260      *            Each line of encoded data will be at most of the given length (rounded down to nearest multiple of 4).
261      *            If lineLength <= 0, then the output will not be divided into lines (chunks). Ignored when decoding.
262      * @since 1.4
263      */
Base64Codec(int lineLength)264     public Base64Codec(int lineLength) {
265         this(lineLength, CHUNK_SEPARATOR);
266     }
267 
268     /**
269      * Creates a Base64 codec used for decoding (all modes) and encoding in URL-unsafe mode.
270      * <p>
271      * When encoding the line length and line separator are given in the constructor, and the encoding table is
272      * STANDARD_ENCODE_TABLE.
273      * </p>
274      * <p>
275      * Line lengths that aren't multiples of 4 will still essentially end up being multiples of 4 in the encoded data.
276      * </p>
277      * <p>
278      * When decoding all variants are supported.
279      * </p>
280      *
281      * @param lineLength
282      *            Each line of encoded data will be at most of the given length (rounded down to nearest multiple of 4).
283      *            If lineLength <= 0, then the output will not be divided into lines (chunks). Ignored when decoding.
284      * @param lineSeparator
285      *            Each line of encoded data will end with this sequence of bytes.
286      * @throws IllegalArgumentException
287      *             Thrown when the provided lineSeparator included some base64 characters.
288      * @since 1.4
289      */
Base64Codec(int lineLength, byte[] lineSeparator)290     public Base64Codec(int lineLength, byte[] lineSeparator) {
291         this(lineLength, lineSeparator, false);
292     }
293 
294     /**
295      * Creates a Base64 codec used for decoding (all modes) and encoding in URL-unsafe mode.
296      * <p>
297      * When encoding the line length and line separator are given in the constructor, and the encoding table is
298      * STANDARD_ENCODE_TABLE.
299      * </p>
300      * <p>
301      * Line lengths that aren't multiples of 4 will still essentially end up being multiples of 4 in the encoded data.
302      * </p>
303      * <p>
304      * When decoding all variants are supported.
305      * </p>
306      *
307      * @param lineLength
308      *            Each line of encoded data will be at most of the given length (rounded down to nearest multiple of 4).
309      *            If lineLength <= 0, then the output will not be divided into lines (chunks). Ignored when decoding.
310      * @param lineSeparator
311      *            Each line of encoded data will end with this sequence of bytes.
312      * @param urlSafe
313      *            Instead of emitting '+' and '/' we emit '-' and '_' respectively. urlSafe is only applied to encode
314      *            operations. Decoding seamlessly handles both modes.
315      * @throws IllegalArgumentException
316      *             The provided lineSeparator included some base64 characters. That's not going to work!
317      * @since 1.4
318      */
Base64Codec(int lineLength, byte[] lineSeparator, boolean urlSafe)319     public Base64Codec(int lineLength, byte[] lineSeparator, boolean urlSafe) {
320         if (lineSeparator == null) {
321             lineLength = 0;  // disable chunk-separating
322             lineSeparator = CHUNK_SEPARATOR;  // this just gets ignored
323         }
324         this.lineLength = lineLength > 0 ? (lineLength / 4) * 4 : 0;
325         this.lineSeparator = new byte[lineSeparator.length];
326         System.arraycopy(lineSeparator, 0, this.lineSeparator, 0, lineSeparator.length);
327         if (lineLength > 0) {
328             this.encodeSize = 4 + lineSeparator.length;
329         } else {
330             this.encodeSize = 4;
331         }
332         this.decodeSize = this.encodeSize - 1;
333         if (containsBase64Byte(lineSeparator)) {
334             String sep = StringUtils.newStringUtf8(lineSeparator);
335             throw new IllegalArgumentException("lineSeperator must not contain base64 characters: [" + sep + "]");
336         }
337         this.encodeTable = urlSafe ? URL_SAFE_ENCODE_TABLE : STANDARD_ENCODE_TABLE;
338     }
339 
340     /**
341      * Returns our current encode mode. True if we're URL-SAFE, false otherwise.
342      *
343      * @return true if we're in URL-SAFE mode, false otherwise.
344      * @since 1.4
345      */
isUrlSafe()346     public boolean isUrlSafe() {
347         return this.encodeTable == URL_SAFE_ENCODE_TABLE;
348     }
349 
350     /**
351      * Returns true if this Base64 object has buffered data for reading.
352      *
353      * @return true if there is Base64 object still available for reading.
354      */
hasData()355     boolean hasData() {
356         return this.buffer != null;
357     }
358 
359     /**
360      * Returns the amount of buffered data available for reading.
361      *
362      * @return The amount of buffered data available for reading.
363      */
avail()364     int avail() {
365         return buffer != null ? pos - readPos : 0;
366     }
367 
368     /** Doubles our buffer. */
resizeBuffer()369     private void resizeBuffer() {
370         if (buffer == null) {
371             buffer = new byte[DEFAULT_BUFFER_SIZE];
372             pos = 0;
373             readPos = 0;
374         } else {
375             byte[] b = new byte[buffer.length * DEFAULT_BUFFER_RESIZE_FACTOR];
376             System.arraycopy(buffer, 0, b, 0, buffer.length);
377             buffer = b;
378         }
379     }
380 
381     /**
382      * Extracts buffered data into the provided byte[] array, starting at position bPos, up to a maximum of bAvail
383      * bytes. Returns how many bytes were actually extracted.
384      *
385      * @param b
386      *            byte[] array to extract the buffered data into.
387      * @param bPos
388      *            position in byte[] array to start extraction at.
389      * @param bAvail
390      *            amount of bytes we're allowed to extract. We may extract fewer (if fewer are available).
391      * @return The number of bytes successfully extracted into the provided byte[] array.
392      */
readResults(byte[] b, int bPos, int bAvail)393     int readResults(byte[] b, int bPos, int bAvail) {
394         if (buffer != null) {
395             int len = Math.min(avail(), bAvail);
396             if (buffer != b) {
397                 System.arraycopy(buffer, readPos, b, bPos, len);
398                 readPos += len;
399                 if (readPos >= pos) {
400                     buffer = null;
401                 }
402             } else {
403                 // Re-using the original consumer's output array is only
404                 // allowed for one round.
405                 buffer = null;
406             }
407             return len;
408         }
409         return eof ? -1 : 0;
410     }
411 
412     /**
413      * Sets the streaming buffer. This is a small optimization where we try to buffer directly to the consumer's output
414      * array for one round (if the consumer calls this method first) instead of starting our own buffer.
415      *
416      * @param out
417      *            byte[] array to buffer directly to.
418      * @param outPos
419      *            Position to start buffering into.
420      * @param outAvail
421      *            Amount of bytes available for direct buffering.
422      */
setInitialBuffer(byte[] out, int outPos, int outAvail)423     void setInitialBuffer(byte[] out, int outPos, int outAvail) {
424         // We can re-use consumer's original output array under
425         // special circumstances, saving on some System.arraycopy().
426         if (out != null && out.length == outAvail) {
427             buffer = out;
428             pos = outPos;
429             readPos = outPos;
430         }
431     }
432 
433     /**
434      * <p>
435      * Encodes all of the provided data, starting at inPos, for inAvail bytes. Must be called at least twice: once with
436      * the data to encode, and once with inAvail set to "-1" to alert encoder that EOF has been reached, so flush last
437      * remaining bytes (if not multiple of 3).
438      * </p>
439      * <p>
440      * Thanks to "commons" project in ws.apache.org for the bitwise operations, and general approach.
441      * http://svn.apache.org/repos/asf/webservices/commons/trunk/modules/util/
442      * </p>
443      *
444      * @param in
445      *            byte[] array of binary data to base64 encode.
446      * @param inPos
447      *            Position to start reading data from.
448      * @param inAvail
449      *            Amount of bytes available from input for encoding.
450      */
encode(byte[] in, int inPos, int inAvail)451     void encode(byte[] in, int inPos, int inAvail) {
452         if (eof) {
453             return;
454         }
455         // inAvail < 0 is how we're informed of EOF in the underlying data we're
456         // encoding.
457         if (inAvail < 0) {
458             eof = true;
459             if (buffer == null || buffer.length - pos < encodeSize) {
460                 resizeBuffer();
461             }
462             switch (modulus) {
463                 case 1 :
464                     buffer[pos++] = encodeTable[(x >> 2) & MASK_6BITS];
465                     buffer[pos++] = encodeTable[(x << 4) & MASK_6BITS];
466                     // URL-SAFE skips the padding to further reduce size.
467                     if (encodeTable == STANDARD_ENCODE_TABLE) {
468                         buffer[pos++] = PAD;
469                         buffer[pos++] = PAD;
470                     }
471                     break;
472 
473                 case 2 :
474                     buffer[pos++] = encodeTable[(x >> 10) & MASK_6BITS];
475                     buffer[pos++] = encodeTable[(x >> 4) & MASK_6BITS];
476                     buffer[pos++] = encodeTable[(x << 2) & MASK_6BITS];
477                     // URL-SAFE skips the padding to further reduce size.
478                     if (encodeTable == STANDARD_ENCODE_TABLE) {
479                         buffer[pos++] = PAD;
480                     }
481                     break;
482             }
483             if (lineLength > 0 && pos > 0) {
484                 System.arraycopy(lineSeparator, 0, buffer, pos, lineSeparator.length);
485                 pos += lineSeparator.length;
486             }
487         } else {
488             for (int i = 0; i < inAvail; i++) {
489                 if (buffer == null || buffer.length - pos < encodeSize) {
490                     resizeBuffer();
491                 }
492                 modulus = (++modulus) % 3;
493                 int b = in[inPos++];
494                 if (b < 0) {
495                     b += 256;
496                 }
497                 x = (x << 8) + b;
498                 if (0 == modulus) {
499                     buffer[pos++] = encodeTable[(x >> 18) & MASK_6BITS];
500                     buffer[pos++] = encodeTable[(x >> 12) & MASK_6BITS];
501                     buffer[pos++] = encodeTable[(x >> 6) & MASK_6BITS];
502                     buffer[pos++] = encodeTable[x & MASK_6BITS];
503                     currentLinePos += 4;
504                     if (lineLength > 0 && lineLength <= currentLinePos) {
505                         System.arraycopy(lineSeparator, 0, buffer, pos, lineSeparator.length);
506                         pos += lineSeparator.length;
507                         currentLinePos = 0;
508                     }
509                 }
510             }
511         }
512     }
513 
514     /**
515      * <p>
516      * Decodes all of the provided data, starting at inPos, for inAvail bytes. Should be called at least twice: once
517      * with the data to decode, and once with inAvail set to "-1" to alert decoder that EOF has been reached. The "-1"
518      * call is not necessary when decoding, but it doesn't hurt, either.
519      * </p>
520      * <p>
521      * Ignores all non-base64 characters. This is how chunked (e.g. 76 character) data is handled, since CR and LF are
522      * silently ignored, but has implications for other bytes, too. This method subscribes to the garbage-in,
523      * garbage-out philosophy: it will not check the provided data for validity.
524      * </p>
525      * <p>
526      * Thanks to "commons" project in ws.apache.org for the bitwise operations, and general approach.
527      * http://svn.apache.org/repos/asf/webservices/commons/trunk/modules/util/
528      * </p>
529      *
530      * @param in
531      *            byte[] array of ascii data to base64 decode.
532      * @param inPos
533      *            Position to start reading data from.
534      * @param inAvail
535      *            Amount of bytes available from input for encoding.
536      */
decode(byte[] in, int inPos, int inAvail)537     void decode(byte[] in, int inPos, int inAvail) {
538         if (eof) {
539             return;
540         }
541         if (inAvail < 0) {
542             eof = true;
543         }
544         for (int i = 0; i < inAvail; i++) {
545             if (buffer == null || buffer.length - pos < decodeSize) {
546                 resizeBuffer();
547             }
548             byte b = in[inPos++];
549             if (b == PAD) {
550                 // We're done.
551                 eof = true;
552                 break;
553             } else {
554                 if (b >= 0 && b < DECODE_TABLE.length) {
555                     int result = DECODE_TABLE[b];
556                     if (result >= 0) {
557                         modulus = (++modulus) % 4;
558                         x = (x << 6) + result;
559                         if (modulus == 0) {
560                             buffer[pos++] = (byte) ((x >> 16) & MASK_8BITS);
561                             buffer[pos++] = (byte) ((x >> 8) & MASK_8BITS);
562                             buffer[pos++] = (byte) (x & MASK_8BITS);
563                         }
564                     }
565                 }
566             }
567         }
568 
569         // Two forms of EOF as far as base64 decoder is concerned: actual
570         // EOF (-1) and first time '=' character is encountered in stream.
571         // This approach makes the '=' padding characters completely optional.
572         if (eof && modulus != 0) {
573             x = x << 6;
574             switch (modulus) {
575                 case 2 :
576                     x = x << 6;
577                     buffer[pos++] = (byte) ((x >> 16) & MASK_8BITS);
578                     break;
579                 case 3 :
580                     buffer[pos++] = (byte) ((x >> 16) & MASK_8BITS);
581                     buffer[pos++] = (byte) ((x >> 8) & MASK_8BITS);
582                     break;
583             }
584         }
585     }
586 
587     /**
588      * Returns whether or not the <code>octet</code> is in the base 64 alphabet.
589      *
590      * @param octet
591      *            The value to test
592      * @return <code>true</code> if the value is defined in the the base 64 alphabet, <code>false</code> otherwise.
593      * @since 1.4
594      */
isBase64(byte octet)595     public static boolean isBase64(byte octet) {
596         return octet == PAD || (octet >= 0 && octet < DECODE_TABLE.length && DECODE_TABLE[octet] != -1);
597     }
598 
599     /**
600      * Tests a given byte array to see if it contains only valid characters within the Base64 alphabet. Currently the
601      * method treats whitespace as valid.
602      *
603      * @param arrayOctet
604      *            byte array to test
605      * @return <code>true</code> if all bytes are valid characters in the Base64 alphabet or if the byte array is empty;
606      *         false, otherwise
607      */
isArrayByteBase64(byte[] arrayOctet)608     public static boolean isArrayByteBase64(byte[] arrayOctet) {
609         for (int i = 0; i < arrayOctet.length; i++) {
610             if (!isBase64(arrayOctet[i]) && !isWhiteSpace(arrayOctet[i])) {
611                 return false;
612             }
613         }
614         return true;
615     }
616 
617     /**
618      * Tests a given byte array to see if it contains only valid characters within the Base64 alphabet.
619      *
620      * @param arrayOctet
621      *            byte array to test
622      * @return <code>true</code> if any byte is a valid character in the Base64 alphabet; false herwise
623      */
containsBase64Byte(byte[] arrayOctet)624     private static boolean containsBase64Byte(byte[] arrayOctet) {
625         for (int i = 0; i < arrayOctet.length; i++) {
626             if (isBase64(arrayOctet[i])) {
627                 return true;
628             }
629         }
630         return false;
631     }
632 
633     /**
634      * Encodes binary data using the base64 algorithm but does not chunk the output.
635      *
636      * @param binaryData
637      *            binary data to encode
638      * @return byte[] containing Base64 characters in their UTF-8 representation.
639      */
encodeBase64(byte[] binaryData)640     public static byte[] encodeBase64(byte[] binaryData) {
641         return encodeBase64(binaryData, false);
642     }
643 
644     /**
645      * Encodes binary data using the base64 algorithm into 76 character blocks separated by CRLF.
646      *
647      * @param binaryData
648      *            binary data to encode
649      * @return String containing Base64 characters.
650      * @since 1.4
651      */
encodeBase64String(byte[] binaryData)652     public static String encodeBase64String(byte[] binaryData) {
653         return StringUtils.newStringUtf8(encodeBase64(binaryData, true));
654     }
655 
656     /**
657      * Encodes binary data using a URL-safe variation of the base64 algorithm but does not chunk the output. The
658      * url-safe variation emits - and _ instead of + and / characters.
659      *
660      * @param binaryData
661      *            binary data to encode
662      * @return byte[] containing Base64 characters in their UTF-8 representation.
663      * @since 1.4
664      */
encodeBase64URLSafe(byte[] binaryData)665     public static byte[] encodeBase64URLSafe(byte[] binaryData) {
666         return encodeBase64(binaryData, false, true);
667     }
668 
669     /**
670      * Encodes binary data using a URL-safe variation of the base64 algorithm but does not chunk the output. The
671      * url-safe variation emits - and _ instead of + and / characters.
672      *
673      * @param binaryData
674      *            binary data to encode
675      * @return String containing Base64 characters
676      * @since 1.4
677      */
encodeBase64URLSafeString(byte[] binaryData)678     public static String encodeBase64URLSafeString(byte[] binaryData) {
679         return StringUtils.newStringUtf8(encodeBase64(binaryData, false, true));
680     }
681 
682     /**
683      * Encodes binary data using the base64 algorithm and chunks the encoded output into 76 character blocks
684      *
685      * @param binaryData
686      *            binary data to encode
687      * @return Base64 characters chunked in 76 character blocks
688      */
encodeBase64Chunked(byte[] binaryData)689     public static byte[] encodeBase64Chunked(byte[] binaryData) {
690         return encodeBase64(binaryData, true);
691     }
692 
693     /**
694      * Decodes an Object using the base64 algorithm. This method is provided in order to satisfy the requirements of the
695      * Decoder interface, and will throw a DecoderException if the supplied object is not of type byte[] or String.
696      *
697      * @param pObject
698      *            Object to decode
699      * @return An object (of type byte[]) containing the binary data which corresponds to the byte[] or String supplied.
700      * @throws DecoderException
701      *             if the parameter supplied is not of type byte[]
702      */
decode(Object pObject)703     public Object decode(Object pObject) throws DecoderException {
704         if (pObject instanceof byte[]) {
705             return decode((byte[]) pObject);
706         } else if (pObject instanceof String) {
707             return decode((String) pObject);
708         } else {
709             throw new DecoderException("Parameter supplied to Base64 decode is not a byte[] or a String");
710         }
711     }
712 
713     /**
714      * Decodes a String containing containing characters in the Base64 alphabet.
715      *
716      * @param pArray
717      *            A String containing Base64 character data
718      * @return a byte array containing binary data
719      * @since 1.4
720      */
decode(String pArray)721     public byte[] decode(String pArray) {
722         return decode(StringUtils.getBytesUtf8(pArray));
723     }
724 
725     /**
726      * Decodes a byte[] containing containing characters in the Base64 alphabet.
727      *
728      * @param pArray
729      *            A byte array containing Base64 character data
730      * @return a byte array containing binary data
731      */
decode(byte[] pArray)732     public byte[] decode(byte[] pArray) {
733         reset();
734         if (pArray == null || pArray.length == 0) {
735             return pArray;
736         }
737         long len = (pArray.length * 3) / 4;
738         byte[] buf = new byte[(int) len];
739         setInitialBuffer(buf, 0, buf.length);
740         decode(pArray, 0, pArray.length);
741         decode(pArray, 0, -1); // Notify decoder of EOF.
742 
743         // Would be nice to just return buf (like we sometimes do in the encode
744         // logic), but we have no idea what the line-length was (could even be
745         // variable).  So we cannot determine ahead of time exactly how big an
746         // array is necessary.  Hence the need to construct a 2nd byte array to
747         // hold the final result:
748 
749         byte[] result = new byte[pos];
750         readResults(result, 0, result.length);
751         return result;
752     }
753 
754     /**
755      * Encodes binary data using the base64 algorithm, optionally chunking the output into 76 character blocks.
756      *
757      * @param binaryData
758      *            Array containing binary data to encode.
759      * @param isChunked
760      *            if <code>true</code> this encoder will chunk the base64 output into 76 character blocks
761      * @return Base64-encoded data.
762      * @throws IllegalArgumentException
763      *             Thrown when the input array needs an output array bigger than {@link Integer#MAX_VALUE}
764      */
encodeBase64(byte[] binaryData, boolean isChunked)765     public static byte[] encodeBase64(byte[] binaryData, boolean isChunked) {
766         return encodeBase64(binaryData, isChunked, false);
767     }
768 
769     /**
770      * Encodes binary data using the base64 algorithm, optionally chunking the output into 76 character blocks.
771      *
772      * @param binaryData
773      *            Array containing binary data to encode.
774      * @param isChunked
775      *            if <code>true</code> this encoder will chunk the base64 output into 76 character blocks
776      * @param urlSafe
777      *            if <code>true</code> this encoder will emit - and _ instead of the usual + and / characters.
778      * @return Base64-encoded data.
779      * @throws IllegalArgumentException
780      *             Thrown when the input array needs an output array bigger than {@link Integer#MAX_VALUE}
781      * @since 1.4
782      */
encodeBase64(byte[] binaryData, boolean isChunked, boolean urlSafe)783     public static byte[] encodeBase64(byte[] binaryData, boolean isChunked, boolean urlSafe) {
784         return encodeBase64(binaryData, isChunked, urlSafe, Integer.MAX_VALUE);
785     }
786 
787     /**
788      * Encodes binary data using the base64 algorithm, optionally chunking the output into 76 character blocks.
789      *
790      * @param binaryData
791      *            Array containing binary data to encode.
792      * @param isChunked
793      *            if <code>true</code> this encoder will chunk the base64 output into 76 character blocks
794      * @param urlSafe
795      *            if <code>true</code> this encoder will emit - and _ instead of the usual + and / characters.
796      * @param maxResultSize
797      *            The maximum result size to accept.
798      * @return Base64-encoded data.
799      * @throws IllegalArgumentException
800      *             Thrown when the input array needs an output array bigger than maxResultSize
801      * @since 1.4
802      */
encodeBase64(byte[] binaryData, boolean isChunked, boolean urlSafe, int maxResultSize)803     public static byte[] encodeBase64(byte[] binaryData, boolean isChunked, boolean urlSafe, int maxResultSize) {
804         if (binaryData == null || binaryData.length == 0) {
805             return binaryData;
806         }
807 
808         long len = getEncodeLength(binaryData, CHUNK_SIZE, CHUNK_SEPARATOR);
809         if (len > maxResultSize) {
810             throw new IllegalArgumentException("Input array too big, the output array would be bigger (" +
811                 len +
812                 ") than the specified maxium size of " +
813                 maxResultSize);
814         }
815 
816         Base64Codec b64 = isChunked ? new Base64Codec(urlSafe) : new Base64Codec(0, CHUNK_SEPARATOR, urlSafe);
817         return b64.encode(binaryData);
818     }
819 
820     /**
821      * Decodes a Base64 String into octets
822      *
823      * @param base64String
824      *            String containing Base64 data
825      * @return Array containing decoded data.
826      * @since 1.4
827      */
decodeBase64(String base64String)828     public static byte[] decodeBase64(String base64String) {
829         return new Base64Codec().decode(base64String);
830     }
831 
832     /**
833      * Decodes Base64 data into octets
834      *
835      * @param base64Data
836      *            Byte array containing Base64 data
837      * @return Array containing decoded data.
838      */
decodeBase64(byte[] base64Data)839     public static byte[] decodeBase64(byte[] base64Data) {
840         return new Base64Codec().decode(base64Data);
841     }
842 
843     /**
844      * Discards any whitespace from a base-64 encoded block.
845      *
846      * @param data
847      *            The base-64 encoded data to discard the whitespace from.
848      * @return The data, less whitespace (see RFC 2045).
849      * @deprecated This method is no longer needed
850      */
discardWhitespace(byte[] data)851     static byte[] discardWhitespace(byte[] data) {
852         byte groomedData[] = new byte[data.length];
853         int bytesCopied = 0;
854         for (int i = 0; i < data.length; i++) {
855             switch (data[i]) {
856                 case ' ' :
857                 case '\n' :
858                 case '\r' :
859                 case '\t' :
860                     break;
861                 default :
862                     groomedData[bytesCopied++] = data[i];
863             }
864         }
865         byte packedData[] = new byte[bytesCopied];
866         System.arraycopy(groomedData, 0, packedData, 0, bytesCopied);
867         return packedData;
868     }
869 
870     /**
871      * Checks if a byte value is whitespace or not.
872      *
873      * @param byteToCheck
874      *            the byte to check
875      * @return true if byte is whitespace, false otherwise
876      */
isWhiteSpace(byte byteToCheck)877     private static boolean isWhiteSpace(byte byteToCheck) {
878         switch (byteToCheck) {
879             case ' ' :
880             case '\n' :
881             case '\r' :
882             case '\t' :
883                 return true;
884             default :
885                 return false;
886         }
887     }
888 
889     // Implementation of the Encoder Interface
890 
891     /**
892      * Encodes an Object using the base64 algorithm. This method is provided in order to satisfy the requirements of the
893      * Encoder interface, and will throw an EncoderException if the supplied object is not of type byte[].
894      *
895      * @param pObject
896      *            Object to encode
897      * @return An object (of type byte[]) containing the base64 encoded data which corresponds to the byte[] supplied.
898      * @throws EncoderException
899      *             if the parameter supplied is not of type byte[]
900      */
encode(Object pObject)901     public Object encode(Object pObject) throws EncoderException {
902         if (!(pObject instanceof byte[])) {
903             throw new EncoderException("Parameter supplied to Base64 encode is not a byte[]");
904         }
905         return encode((byte[]) pObject);
906     }
907 
908     /**
909      * Encodes a byte[] containing binary data, into a String containing characters in the Base64 alphabet.
910      *
911      * @param pArray
912      *            a byte array containing binary data
913      * @return A String containing only Base64 character data
914      * @since 1.4
915      */
encodeToString(byte[] pArray)916     public String encodeToString(byte[] pArray) {
917         return StringUtils.newStringUtf8(encode(pArray));
918     }
919 
920     /**
921      * Encodes a byte[] containing binary data, into a byte[] containing characters in the Base64 alphabet.
922      *
923      * @param pArray
924      *            a byte array containing binary data
925      * @return A byte array containing only Base64 character data
926      */
encode(byte[] pArray)927     public byte[] encode(byte[] pArray) {
928         reset();
929         if (pArray == null || pArray.length == 0) {
930             return pArray;
931         }
932         long len = getEncodeLength(pArray, lineLength, lineSeparator);
933         byte[] buf = new byte[(int) len];
934         setInitialBuffer(buf, 0, buf.length);
935         encode(pArray, 0, pArray.length);
936         encode(pArray, 0, -1); // Notify encoder of EOF.
937         // Encoder might have resized, even though it was unnecessary.
938         if (buffer != buf) {
939             readResults(buf, 0, buf.length);
940         }
941         // In URL-SAFE mode we skip the padding characters, so sometimes our
942         // final length is a bit smaller.
943         if (isUrlSafe() && pos < buf.length) {
944             byte[] smallerBuf = new byte[pos];
945             System.arraycopy(buf, 0, smallerBuf, 0, pos);
946             buf = smallerBuf;
947         }
948         return buf;
949     }
950 
951     /**
952      * Pre-calculates the amount of space needed to base64-encode the supplied array.
953      *
954      * @param pArray byte[] array which will later be encoded
955      * @param chunkSize line-length of the output (<= 0 means no chunking) between each
956      *        chunkSeparator (e.g. CRLF).
957      * @param chunkSeparator the sequence of bytes used to separate chunks of output (e.g. CRLF).
958      *
959      * @return amount of space needed to encoded the supplied array.  Returns
960      *         a long since a max-len array will require Integer.MAX_VALUE + 33%.
961      */
getEncodeLength(byte[] pArray, int chunkSize, byte[] chunkSeparator)962     private static long getEncodeLength(byte[] pArray, int chunkSize, byte[] chunkSeparator) {
963         // base64 always encodes to multiples of 4.
964         chunkSize = (chunkSize / 4) * 4;
965 
966         long len = (pArray.length * 4) / 3;
967         long mod = len % 4;
968         if (mod != 0) {
969             len += 4 - mod;
970         }
971         if (chunkSize > 0) {
972             boolean lenChunksPerfectly = len % chunkSize == 0;
973             len += (len / chunkSize) * chunkSeparator.length;
974             if (!lenChunksPerfectly) {
975                 len += chunkSeparator.length;
976             }
977         }
978         return len;
979     }
980 
981     // Implementation of integer encoding used for crypto
982     /**
983      * Decodes a byte64-encoded integer according to crypto standards such as W3C's XML-Signature
984      *
985      * @param pArray
986      *            a byte array containing base64 character data
987      * @return A BigInteger
988      * @since 1.4
989      */
decodeInteger(byte[] pArray)990     public static BigInteger decodeInteger(byte[] pArray) {
991         return new BigInteger(1, decodeBase64(pArray));
992     }
993 
994     /**
995      * Encodes to a byte64-encoded integer according to crypto standards such as W3C's XML-Signature
996      *
997      * @param bigInt
998      *            a BigInteger
999      * @return A byte array containing base64 character data
1000      * @throws NullPointerException
1001      *             if null is passed in
1002      * @since 1.4
1003      */
encodeInteger(BigInteger bigInt)1004     public static byte[] encodeInteger(BigInteger bigInt) {
1005         if (bigInt == null) {
1006             throw new NullPointerException("encodeInteger called with null parameter");
1007         }
1008         return encodeBase64(toIntegerBytes(bigInt), false);
1009     }
1010 
1011     /**
1012      * Returns a byte-array representation of a <code>BigInteger</code> without sign bit.
1013      *
1014      * @param bigInt
1015      *            <code>BigInteger</code> to be converted
1016      * @return a byte array representation of the BigInteger parameter
1017      */
toIntegerBytes(BigInteger bigInt)1018     static byte[] toIntegerBytes(BigInteger bigInt) {
1019         int bitlen = bigInt.bitLength();
1020         // round bitlen
1021         bitlen = ((bitlen + 7) >> 3) << 3;
1022         byte[] bigBytes = bigInt.toByteArray();
1023 
1024         if (((bigInt.bitLength() % 8) != 0) && (((bigInt.bitLength() / 8) + 1) == (bitlen / 8))) {
1025             return bigBytes;
1026         }
1027         // set up params for copying everything but sign bit
1028         int startSrc = 0;
1029         int len = bigBytes.length;
1030 
1031         // if bigInt is exactly byte-aligned, just skip signbit in copy
1032         if ((bigInt.bitLength() % 8) == 0) {
1033             startSrc = 1;
1034             len--;
1035         }
1036         int startDst = bitlen / 8 - len; // to pad w/ nulls as per spec
1037         byte[] resizedBytes = new byte[bitlen / 8];
1038         System.arraycopy(bigBytes, startSrc, resizedBytes, startDst, len);
1039         return resizedBytes;
1040     }
1041 
1042     /**
1043      * Resets this Base64 object to its initial newly constructed state.
1044      */
reset()1045     private void reset() {
1046         buffer = null;
1047         pos = 0;
1048         readPos = 0;
1049         currentLinePos = 0;
1050         modulus = 0;
1051         eof = false;
1052     }
1053 
1054 }
1055