/*

 * Copyright (c) 1999, 2006, Oracle and/or its affiliates. All rights reserved.

 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.

 *

 * This code is free software; you can redistribute it and/or modify it

 * under the terms of the GNU General Public License version 2 only, as

 * published by the Free Software Foundation.

 *

 * This code is distributed in the hope that it will be useful, but WITHOUT

 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or

 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License

 * version 2 for more details (a copy is included in the LICENSE file that

 * accompanied this code).

 *

 * You should have received a copy of the GNU General Public License version

 * 2 along with this work; if not, write to the Free Software Foundation,

 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.

 *

 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA

 * or visit www.oracle.com if you need additional information or have any

 * questions.

 */

/* @test

 * @bug 4085160 4139951 5005831

 * @summary Test that required character encodings are supported

 */

import java.io.InputStreamReader;

import java.io.OutputStreamWriter;

import java.io.ByteArrayInputStream;

import java.io.ByteArrayOutputStream;

import java.nio.charset.Charset;

import java.nio.charset.UnsupportedCharsetException;

public class Encodings {

    static boolean equals(byte[] a, byte[] b) {

        if (a.length != b.length) return false;

        for (int i = 0; i < a.length; i++)

            if (a[i] != b[i]) return false;

        return true;

    }

    static void go(String enc, String str, final byte[] bytes, boolean bidir)

        throws Exception

    {

        final Charset charset = Charset.forName(enc);

        /* String(byte[] bs, String enc) */

        if (!(new String(bytes, enc).equals(str)))

            throw new Exception(enc + ": String constructor failed");

        /* String(byte[] bs, Charset charset) */

        if (!(new String(bytes, charset).equals(str)))

            throw new Exception(charset + ": String constructor failed");

        /* String(byte[] bs, int off, int len, Charset charset) */

        String start = str.substring(0, 2);

        String end = str.substring(2);

        if (enc.equals("UTF-16BE") || enc.equals("UTF-16LE")) {

            if (!(new String(bytes, 0, 4, charset).equals(start)))

                throw new Exception(charset + ": String constructor failed");

            if (!(new String(bytes, 4, bytes.length - 4, charset).equals(end)))

                throw new Exception(charset + ": String constructor failed");

        } else if (enc.equals("UTF-16")) {

            if (!(new String(bytes, 0, 6, charset).equals(start)))

                throw new Exception(charset + ": String constructor failed");

        } else {

            if (!(new String(bytes, 0, 2, charset).equals(start)))

                throw new Exception(charset + ": String constructor failed");

            if (!(new String(bytes, 2, bytes.length - 2, charset).equals(end)))

                throw new Exception(charset + ": String constructor failed");

        }

        /* InputStreamReader */

        ByteArrayInputStream bi = new ByteArrayInputStream(bytes);

        InputStreamReader r = new InputStreamReader(bi, enc);

        String inEnc = r.getEncoding();

        int n = str.length();

        char[] cs = new char[n];

        for (int i = 0; i < n;) {

            int m;

            if ((m = r.read(cs, i, n - i)) < 0)

                throw new Exception(enc + ": EOF on InputStreamReader");

            i += m;

        }

        if (!(new String(cs).equals(str)))

            throw new Exception(enc + ": InputStreamReader failed");

        if (!bidir) {

            System.err.println(enc + " --> " + inEnc);

            return;

        }

        /* String.getBytes(String enc) */

        byte[] bs = str.getBytes(enc);

        if (!equals(bs, bytes))

            throw new Exception(enc + ": String.getBytes failed");

        /* String.getBytes(Charset charset) */

        bs = str.getBytes(charset);

        if (!equals(bs, bytes))

            throw new Exception(charset + ": String.getBytes failed");

        // Calls to String.getBytes(Charset) shouldn't automatically

        // use the cached thread-local encoder.

        if (charset.name().equals("UTF-16BE")) {

            String s = new String(bytes, charset);

            // Replace the thread-local encoder with this one.

            byte[] bb = s.getBytes(Charset.forName("UTF-16LE"));

            if (bytes.length != bb.length) {

                // Incidental test.

                throw new RuntimeException("unequal length: "

                                           + bytes.length + " != "

                                           + bb.length);

            } else {

                boolean diff = false;

                // Expect different byte[] between UTF-16LE and UTF-16BE

                // even though encoder was previously cached by last call

                // to getBytes().

                for (int i = 0; i < bytes.length; i++) {

                    if (bytes[i] != bb[i])

                        diff = true;

                }

                if (!diff)

                    throw new RuntimeException("byte arrays equal");

            }

        }

        /* OutputStreamWriter */

        ByteArrayOutputStream bo = new ByteArrayOutputStream();

        OutputStreamWriter w = new OutputStreamWriter(bo, enc);

        String outEnc = w.getEncoding();

        w.write(str);

        w.close();

        bs = bo.toByteArray();

        if (!equals(bs, bytes))

            throw new Exception(enc + ": OutputStreamWriter failed");

        System.err.println(enc + " --> " + inEnc + " / " + outEnc);

    }

    static void go(String enc, String str, byte[] bytes) throws Exception {

        go(enc, str, bytes, true);

    }

    public static void main(String[] args) throws Exception {

        go("US-ASCII", "abc", new byte[] { 'a', 'b', 'c' });

        go("us-ascii", "abc", new byte[] { 'a', 'b', 'c' });

        go("ISO646-US", "abc", new byte[] { 'a', 'b', 'c' });

        go("ISO-8859-1", "ab\u00c7", new byte[] { 'a', 'b', (byte)'\u00c7' });

        go("UTF-8", "ab\u1e09",

           new byte[] { 'a', 'b',

                        (byte)(0xe0 | (0x0f & (0x1e09 >> 12))),

                        (byte)(0x80 | (0x3f & (0x1e09 >> 6))),

                        (byte)(0x80 | (0x3f & 0x1e09)) });

        go("UTF-16BE", "ab\u1e09",

           new byte[] { 0, 'a', 0, 'b', 0x1e, 0x09 });

        go("UTF-16LE", "ab\u1e09",

           new byte[] { 'a', 0, 'b', 0, 0x09, 0x1e });

        /* UTF-16 accepts both byte orders on input but always uses big-endian

         * on output, so test all three cases

         */

        go("UTF-16", "ab\u1e09",

           new byte[] { (byte)0xfe, (byte)0xff, 0, 'a', 0, 'b', 0x1e, 0x09 });

        go("UTF-16", "ab\u1e09",

           new byte[] { (byte)0xff, (byte)0xfe, 'a', 0, 'b', 0, 0x09, 0x1e },

           false);

    }

}